claude3_agent_code

This commit is contained in:
lfy79001
2024-03-16 11:57:49 +08:00
parent 684b4a1b7b
commit 505e772463

View File

@@ -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
}