Update app.py
Browse files
app.py
CHANGED
|
@@ -279,24 +279,25 @@ tools = [
|
|
| 279 |
# =========================================================
|
| 280 |
# Model
|
| 281 |
# =========================================================
|
|
|
|
| 282 |
model = HfApiModel(
|
| 283 |
model_id=MODEL_ID,
|
| 284 |
-
|
| 285 |
token=os.environ["GEMINI_API_KEY"],
|
| 286 |
max_tokens=MAX_TOKENS,
|
| 287 |
temperature=TEMPERATURE,
|
| 288 |
timeout=120,
|
| 289 |
)
|
| 290 |
-
|
| 291 |
"""
|
|
|
|
| 292 |
model = LiteLLMModel(
|
| 293 |
model_id=MODEL_ID,
|
| 294 |
api_key=os.environ["GEMINI_API_KEY"],
|
| 295 |
max_tokens=MAX_TOKENS,
|
| 296 |
temperature=TEMPERATURE,
|
| 297 |
-
tool_config={"function_calling_config": {"mode": "NONE"}},
|
| 298 |
)
|
| 299 |
-
|
| 300 |
|
| 301 |
# =========================================================
|
| 302 |
# Agent
|
|
|
|
| 279 |
# =========================================================
|
| 280 |
# Model
|
| 281 |
# =========================================================
|
| 282 |
+
"""
|
| 283 |
model = HfApiModel(
|
| 284 |
model_id=MODEL_ID,
|
| 285 |
+
provider="auto", # or "together", "nebius", "fireworks-ai", etc.
|
| 286 |
token=os.environ["GEMINI_API_KEY"],
|
| 287 |
max_tokens=MAX_TOKENS,
|
| 288 |
temperature=TEMPERATURE,
|
| 289 |
timeout=120,
|
| 290 |
)
|
|
|
|
| 291 |
"""
|
| 292 |
+
|
| 293 |
model = LiteLLMModel(
|
| 294 |
model_id=MODEL_ID,
|
| 295 |
api_key=os.environ["GEMINI_API_KEY"],
|
| 296 |
max_tokens=MAX_TOKENS,
|
| 297 |
temperature=TEMPERATURE,
|
| 298 |
+
#tool_config={"function_calling_config": {"mode": "NONE"}},
|
| 299 |
)
|
| 300 |
+
|
| 301 |
|
| 302 |
# =========================================================
|
| 303 |
# Agent
|