diff --git a/ai_providers.py b/ai_providers.py index 6b92b95..74d7cd9 100644 --- a/ai_providers.py +++ b/ai_providers.py @@ -65,6 +65,18 @@ class AnthropicProvider(AIProvider): for function in functions ] + def format_assistant_reply(self, response): + for message in response.content: + if message.type == "text": + return message.text + return "" + + def get_reply_text(self, response): + return self.format_assistant_reply(response) + + def get_model(self): + return self.model + class OpenAIProvider(AIProvider): def __init__(self, use_smart_model=True): self.client = OpenAI()