Merge pull request #41 from bucolucas/fix-anthropic-provider

Fix AnthropicProvider: Add missing methods
This commit is contained in:
2024-08-18 13:51:18 -05:00
committed by GitHub
+12
View File
@@ -65,6 +65,18 @@ class AnthropicProvider(AIProvider):
for function in functions 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): class OpenAIProvider(AIProvider):
def __init__(self, use_smart_model=True): def __init__(self, use_smart_model=True):
self.client = OpenAI() self.client = OpenAI()