fix: anthropic

This commit is contained in:
Timothy J. Baek 2024-06-02 16:59:43 -07:00
parent c2f5200906
commit 0f516bd95e

View File

@ -63,6 +63,13 @@ class Pipeline:
self, user_message: str, model_id: str, messages: List[dict], body: dict
) -> Union[str, Generator, Iterator]:
try:
if "user" in body:
del body["user"]
if "chat_id" in body:
del body["chat_id"]
if "title" in body:
del body["title"]
if body.get("stream", False):
return self.stream_response(model_id, messages, body)
else: