Karim0111 commited on
Commit
fe6a398
·
verified ·
1 Parent(s): d17cfab

Update agents.py

Browse files
Files changed (1) hide show
  1. agents.py +1 -0
agents.py CHANGED
@@ -40,6 +40,7 @@ class SlowLiteLLMModel(LiteLLMModel):
40
 
41
  react_model_name = "gemini-1.5-mini"
42
  react_model = LiteLLMModel(model_id=react_model_name,
 
43
  api_key=os.getenv("GEMINI_KEY"),
44
  temperature=0.2
45
  )
 
40
 
41
  react_model_name = "gemini-1.5-mini"
42
  react_model = LiteLLMModel(model_id=react_model_name,
43
+ provider="google",
44
  api_key=os.getenv("GEMINI_KEY"),
45
  temperature=0.2
46
  )