Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -15,11 +15,11 @@ load_dotenv()
|
|
| 15 |
|
| 16 |
# Initialize the model
|
| 17 |
model = LiteLLMModel(
|
| 18 |
-
model_id="gemini-1.5-flash", #
|
| 19 |
-
|
|
|
|
| 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 |
|