From 505e772463699cdb45ec98f488f66e6d6b3d477c Mon Sep 17 00:00:00 2001
From: lfy79001 <843265183@qq.com>
Date: Sat, 16 Mar 2024 11:57:49 +0800
Subject: [PATCH] claude3_agent_code

---
 mm_agents/agent.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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
             }