akiko19191 commited on
Commit
7394b15
·
verified ·
1 Parent(s): 5879708

Update utils/llms.py

Browse files
Files changed (1) hide show
  1. utils/llms.py +1 -1
utils/llms.py CHANGED
@@ -123,7 +123,7 @@ def gpt4stream(messages,model,api_keys):
123
  try:
124
  os.environ["GEMINI_API_KEY"] =key
125
 
126
- response = completion(model="gemini/gemini-2.0-flash-thinking-exp-01-21", messages=messages, stream=True)
127
 
128
  cunk=""
129
  for part in response:
 
123
  try:
124
  os.environ["GEMINI_API_KEY"] =key
125
 
126
+ response = completion(model="gemini/gemini-2.5-pro-exp-03-25", messages=messages, stream=True)
127
 
128
  cunk=""
129
  for part in response: