dan/memory
1
0
Fork 0
mirror of https://github.com/mruwnik/memory.git synced 2025-11-13 00:04:05 +01:00
Code Issues 1 Packages Projects Releases Wiki Activity

properly handle images by anthropic

Browse Source
This commit is contained in:
Daniel O'Connell 2025-11-02 12:08:46 +01:00
parent a4f42e656a
commit 6bd7df8ee3
1 changed files with 17 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src/memory/common/llms/anthropic_provider.py
View File

@ -71,15 +71,23 @@ class AnthropicProvider(BaseLLMProvider):
}
def _convert_message(self, message: Message) -> dict[str, Any]:
converted = message.to_dict()
if converted["role"] == MessageRole.ASSISTANT and isinstance(
converted["content"], list
):
content = sorted(
converted["content"], key=lambda x: x["type"] != "thinking"
)
return converted | {"content": content}
return converted
# Handle string content directly
if isinstance(message.content, str):
return {"role": message.role.value, "content": message.content}
# Convert content items, handling ImageContent specially
content_list = []
for item in message.content:
if isinstance(item, ImageContent):
content_list.append(self._convert_image_content(item))
else:
content_list.append(item.to_dict())
# Sort assistant messages to put thinking last
if message.role == MessageRole.ASSISTANT:
content_list = sorted(content_list, key=lambda x: x["type"] != "thinking")
return {"role": message.role.value, "content": content_list}
def _should_include_message(self, message: Message) -> bool:
"""Filter out system messages (handled separately in Anthropic)."""