alexkueck commited on
Commit
cdfd7b3
1 Parent(s): 834b0ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -62,6 +62,7 @@ HUGGINGFACEHUB_API_TOKEN = os.getenv("HF_ACCESS_READ")
62
  OAI_API_KEY=os.getenv("OPENAI_API_KEY")
63
  HEADERS = {"Authorization": f"Bearer {HUGGINGFACEHUB_API_TOKEN}"}
64
  TAVILY_KEY = os.getenv("TAVILY_KEY")
 
65
 
66
 
67
  ################################################
@@ -233,7 +234,7 @@ def create_assistant_file(prompt, file):
233
  def create_assistant_suche(prompt):
234
  global client, general_assistant_suche
235
 
236
- retriever = TavilySearchAPIRetriever(api_key = TAVILY_KEY, k=4)
237
  result = retriever.invoke(prompt)
238
 
239
  """
 
62
  OAI_API_KEY=os.getenv("OPENAI_API_KEY")
63
  HEADERS = {"Authorization": f"Bearer {HUGGINGFACEHUB_API_TOKEN}"}
64
  TAVILY_KEY = os.getenv("TAVILY_KEY")
65
+ os.environ["TAVILY_API_KEY"] = TAVILY_KEY
66
 
67
 
68
  ################################################
 
234
  def create_assistant_suche(prompt):
235
  global client, general_assistant_suche
236
 
237
+ retriever = TavilySearchAPIRetriever(k=4)
238
  result = retriever.invoke(prompt)
239
 
240
  """