Refactor messages parameter in Anthropic provider

This commit is contained in:
2024-10-31 16:29:58 -04:00
parent c41a3f00fb
commit 48ac97f070
+7
View File
@@ -68,8 +68,15 @@ class Anthropic(BaseProvider):
) -> T:
model = llm_model or self.DEFAULT_MODEL
# Extract the prompt from kwargs if it exists
prompt = kwargs.pop("prompt", kwargs.pop("messages", ""))
# Format the messages properly
messages = [{"role": "user", "content": prompt}]
response = self.structured_client.messages.create(
model=model,
messages=messages, # Add the messages parameter
response_model=response_model,
**{**self.DEFAULT_KWARGS, **kwargs},
)