ORromu commited on
Commit
abc8f08
·
verified ·
1 Parent(s): 41a92e3

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -1
agent.py CHANGED
@@ -42,7 +42,9 @@ sys_msg = SystemMessage(content=system_prompt)
42
 
43
 
44
  # Loading the assistant
45
- chat = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
 
 
46
 
47
  tools = [add,
48
  substract,
 
42
 
43
 
44
  # Loading the assistant
45
+ #chat = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
46
+ chat = ChatGoogleGenerativeAI(model="gemini-1.5-pro", temperature=0)
47
+
48
 
49
  tools = [add,
50
  substract,