From 35bea5933fff978cd4dc5e74ae289ddfbead9d5b Mon Sep 17 00:00:00 2001 From: bucolucas Date: Sun, 18 Aug 2024 13:50:37 -0500 Subject: [PATCH] Add format_assistant_reply and get_reply_text methods to AnthropicProvider --- ai_providers.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) 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()