diff --git a/mm_agents/agent.py b/mm_agents/agent.py index 28b5f11..7599b02 100644 --- a/mm_agents/agent.py +++ b/mm_agents/agent.py @@ -491,6 +491,8 @@ class PromptAgent: elif self.model.startswith("claude"): messages = payload["messages"] max_tokens = payload["max_tokens"] + top_p = payload["top_p"] + temperature = payload["temperature"] claude_messages = [] @@ -528,7 +530,7 @@ class PromptAgent: } payload = { - "model": "claude-3-opus-20240229", + "model": self.model, "max_tokens": max_tokens, "messages": claude_messages }