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
|
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()
|
||||||
|
|||||||
Reference in New Issue
Block a user