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

Update agents.py

Browse files
Files changed (1) hide show
  1. agents.py +6 -5
agents.py CHANGED
@@ -39,11 +39,12 @@ class SlowLiteLLMModel(LiteLLMModel):
39
 
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
- )
 
47
 
48
 
49
  manager_agent = CodeAgent(
 
39
 
40
 
41
  react_model_name = "gemini-1.5-mini"
42
+ react_model = LiteLLMModel(
43
+ model="gemini-1.5-mini",
44
+ provider="google",
45
+ api_key=os.getenv("GEMINI_KEY"),
46
+ temperature=0.2
47
+ )
48
 
49
 
50
  manager_agent = CodeAgent(