Aakash010 commited on
Commit
68608b9
·
verified ·
1 Parent(s): 14dac12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -8,6 +8,8 @@ from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool,Pyt
8
  from langchain_openai import ChatOpenAI
9
  from dotenv import load_dotenv
10
  load_dotenv()
 
 
11
  # (Keep Constants as is)
12
  # --- Constants ---
13
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -17,7 +19,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
17
  class BasicAgent:
18
  def __init__(self):
19
  model =model = LiteLLMModel(
20
- model_id="gemini/gemini-2.0-flash",
21
  api_key=os.getenv("GEMINI_API_KEY")
22
  )
23
  self.agent = CodeAgent(
 
8
  from langchain_openai import ChatOpenAI
9
  from dotenv import load_dotenv
10
  load_dotenv()
11
+ import litellm
12
+ litellm._turn_on_debug()
13
  # (Keep Constants as is)
14
  # --- Constants ---
15
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
19
  class BasicAgent:
20
  def __init__(self):
21
  model =model = LiteLLMModel(
22
+ model_id="gemini/gemini-1.5-flash",
23
  api_key=os.getenv("GEMINI_API_KEY")
24
  )
25
  self.agent = CodeAgent(