Add format_assistant_reply and get_reply_text methods to AnthropicProvider
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user