Mehedi2 commited on
Commit
7cb06ba
·
verified ·
1 Parent(s): 26c1163

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +7 -4
agent.py CHANGED
@@ -19,14 +19,17 @@ from smolagents.tools import Tool
19
  # Load environment variables
20
  load_dotenv()
21
 
22
- # Initialize the model with Google AI Studio Gemini
 
 
23
  model = LiteLLMModel(
24
- model_id="gemini-2.5-flash-preview-09-2025", # short name only
25
- api_key=os.getenv("GEMINI_API_KEY"), # set in Hugging Face secrets
26
- provider="google_ai", # force AI Studio, not Vertex
27
  )
28
 
29
 
 
30
  class ExcelToTextTool(Tool):
31
  """Render an Excel worksheet as a Markdown table."""
32
 
 
19
  # Load environment variables
20
  load_dotenv()
21
 
22
+ from smolagents import LiteLLMModel
23
+ import os
24
+
25
  model = LiteLLMModel(
26
+ model_id="gemini-2.5-flash-preview-09-2025", # no vertex_ai/ prefix
27
+ provider="google_ai", # force Google AI Studio
28
+ api_key=os.getenv("GEMINI_API_KEY"), # from HF secrets
29
  )
30
 
31
 
32
+
33
  class ExcelToTextTool(Tool):
34
  """Render an Excel worksheet as a Markdown table."""
35