feat: Support newer Gemini models (#188)
This commit is contained in:
@@ -837,7 +837,7 @@ class PromptAgent:
|
|||||||
)
|
)
|
||||||
return response.text
|
return response.text
|
||||||
|
|
||||||
elif self.model == "gemini-1.5-pro-latest":
|
elif self.model.startswith("gemini"):
|
||||||
messages = payload["messages"]
|
messages = payload["messages"]
|
||||||
max_tokens = payload["max_tokens"]
|
max_tokens = payload["max_tokens"]
|
||||||
top_p = payload["top_p"]
|
top_p = payload["top_p"]
|
||||||
|
|||||||
Reference in New Issue
Block a user