Bhumi14 commited on
Commit
c84e21f
Β·
verified Β·
1 Parent(s): 15a0045

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -3
agent.py CHANGED
@@ -15,11 +15,11 @@ load_dotenv()
15
 
16
  # Initialize the model
17
  model = LiteLLMModel(
18
- model_id="gemini-1.5-flash", # AI Studio model, api_key=os.getenv("GEMINI_API_KEY"),
19
- provider="google_ai"
 
20
  )
21
 
22
-
23
  class ExcelToTextTool(Tool):
24
  """Render an Excel worksheet as a Markdown table."""
25
 
 
15
 
16
  # Initialize the model
17
  model = LiteLLMModel(
18
+ model_id="gemini/gemini-1.5-flash", # πŸ‘ˆ use gemini prefix
19
+ api_key=os.getenv("GEMINI_API_KEY"), # πŸ‘ˆ pulls from Hugging Face secrets
20
+ provider="gemini" # πŸ‘ˆ not google_ai
21
  )
22
 
 
23
  class ExcelToTextTool(Tool):
24
  """Render an Excel worksheet as a Markdown table."""
25