From 48ac97f07073365566d6ccc2aec88824af022f36 Mon Sep 17 00:00:00 2001 From: Kenneth Reitz Date: Thu, 31 Oct 2024 16:29:58 -0400 Subject: [PATCH] Refactor messages parameter in Anthropic provider --- simplemind/providers/anthropic.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/simplemind/providers/anthropic.py b/simplemind/providers/anthropic.py index 5d2938e..5c76160 100644 --- a/simplemind/providers/anthropic.py +++ b/simplemind/providers/anthropic.py @@ -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}, )