Realistvik0 commited on
Commit
f664f59
·
verified ·
1 Parent(s): a7a72a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -279,24 +279,25 @@ tools = [
279
  # =========================================================
280
  # Model
281
  # =========================================================
 
282
  model = HfApiModel(
283
  model_id=MODEL_ID,
284
- #provider="auto", # or "together", "nebius", "fireworks-ai", etc.
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