kennethreitz/simplemind
1
0
Fork 0
mirror of https://github.com/kennethreitz/simplemind.git synced 2026-06-05 06:46:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Revert "Merge pull request #37 from lucianosrp/fix-sys-role-anthropic"

Browse Source 
This reverts commit 5770c37edf, reversing
changes made to a5c7486dfc.
This commit is contained in:
Kenneth Reitz
2024-11-04 11:21:07 -05:00
parent 5770c37edf
commit 44fd3468fa
1 changed files with 2 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
simplemind/providers/anthropic.py
+2 -11
View File
@@ -49,24 +49,15 @@ class Anthropic(BaseProvider):
return instructor.from_anthropic(self.client) return instructor.from_anthropic(self.client)
@logger @logger
def send_conversation( def send_conversation(self, conversation: "Conversation", **kwargs) -> "Message":
self, conversation: "Conversation", **kwargs
) -> "Message":
"""Send a conversation to the Anthropic API.""" """Send a conversation to the Anthropic API."""
from ..models import Message 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 = [ messages = [
{"role": msg.role, "content": msg.text} {"role": msg.role, "content": msg.text} for msg in conversation.messages
for msg in conversation.messages
if msg.role != "system"
] ]
response = self.client.messages.create( response = self.client.messages.create(
system=next(system_prompt, None),
model=conversation.llm_model or self.DEFAULT_MODEL, model=conversation.llm_model or self.DEFAULT_MODEL,
messages=messages, messages=messages,
**{**self.DEFAULT_KWARGS, **kwargs}, **{**self.DEFAULT_KWARGS, **kwargs},