patricksamuel commited on
Commit
df8feef
·
verified ·
1 Parent(s): 143a3a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -15,10 +15,10 @@ from pydantic import BaseModel, Field
15
  import nest_asyncio
16
  import requests
17
  import os, re
18
- from agents.extensions.models.litellm_model import LitellmModel
19
 
20
- #os.getenv("OPENAI_API_KEY")
21
- os.getenv("GEMINI_API_KEY")
22
  os.getenv("TAVILY_API_KEY")
23
 
24
 
@@ -182,9 +182,10 @@ When you've found the answer, stop reasoning and return ONLY the answer.
182
  tavily_search,
183
  visit_website
184
  ],
185
- model=LitellmModel(
186
- model="gemini/gemini-2.5-flash",
187
- api_key=os.getenv("GEMINI_API_KEY")
 
188
  )
189
 
190
  )
 
15
  import nest_asyncio
16
  import requests
17
  import os, re
18
+ #from agents.extensions.models.litellm_model import LitellmModel
19
 
20
+ os.getenv("OPENAI_API_KEY")
21
+ #os.getenv("GEMINI_API_KEY")
22
  os.getenv("TAVILY_API_KEY")
23
 
24
 
 
182
  tavily_search,
183
  visit_website
184
  ],
185
+ model="gpt-4o-mini"
186
+ # model=LitellmModel(
187
+ # model="gemini/gemini-2.5-flash",
188
+ # api_key=os.getenv("GEMINI_API_KEY")
189
  )
190
 
191
  )