lam-ho commited on
Commit
f935155
·
verified ·
1 Parent(s): 3f45924

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +8 -2
agent.py CHANGED
@@ -221,8 +221,14 @@ tools=[
221
  TavilySearchTool()
222
  ]
223
 
224
- model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-exp",
225
- api_key=os.getenv("GEMINI_API_KEY"))
 
 
 
 
 
 
226
 
227
  def delay_execution_10(pagent, **kwargs) -> bool:
228
  """
 
221
  TavilySearchTool()
222
  ]
223
 
224
+ # model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-exp",
225
+ # api_key=os.getenv("GEMINI_API_KEY"))
226
+
227
+ model = LiteLLMModel(
228
+ model_id="ollama_chat/gemma3:27b", # Or try other Ollama-supported models
229
+ api_base="http://127.0.0.1:11434", # Default Ollama local server
230
+ num_ctx=8192,
231
+ )
232
 
233
  def delay_execution_10(pagent, **kwargs) -> bool:
234
  """