From 1528dc2a21be02ca99d8a05bb64c1999c234dbe7 Mon Sep 17 00:00:00 2001 From: Kenneth Reitz Date: Wed, 6 Nov 2024 07:42:55 -0500 Subject: [PATCH] Revert "Merge pull request #39 from lucianosrp/fix-sys-role-anthropic" This reverts commit 46cd19ea9024dddf326d690aef206d0b4849f5a3, reversing changes made to 3e8d5662d24fffd4d1cfd6643e1ffc81ba3ca685. --- simplemind/providers/anthropic.py | 21 +++------------------ 1 file changed, 3 insertions(+), 18 deletions(-) diff --git a/simplemind/providers/anthropic.py b/simplemind/providers/anthropic.py index fbf80e8..ffa776d 100644 --- a/simplemind/providers/anthropic.py +++ b/simplemind/providers/anthropic.py @@ -1,5 +1,5 @@ from functools import cached_property -from typing import TYPE_CHECKING, Iterator, Type, TypeVar +from typing import TYPE_CHECKING, Type, TypeVar, Iterator import instructor from pydantic import BaseModel @@ -49,30 +49,15 @@ class Anthropic(BaseProvider): return instructor.from_anthropic(self.client) @logger - def send_conversation( - self, conversation: "Conversation", **kwargs - ) -> "Message": + def send_conversation(self, conversation: "Conversation", **kwargs) -> "Message": """Send a conversation to the Anthropic API.""" from ..models import Message - system_messages = [ - msg for msg in conversation.messages if msg.role == "system" - ] - if len(system_messages) > 1: - logger.warning( - "Multiple system messages found. Using the first one." - ) - - system_prompt = system_messages[0] if system_messages else None - messages = [ - {"role": msg.role, "content": msg.text} - for msg in conversation.messages - if msg.role != "system" + {"role": msg.role, "content": msg.text} for msg in conversation.messages ] response = self.client.messages.create( - system=system_prompt.text if system_prompt else None, model=conversation.llm_model or self.DEFAULT_MODEL, messages=messages, **{**self.DEFAULT_KWARGS, **kwargs},