from typing import List, Union
from agentopera.engine.function_call import FunctionCall
from agentopera.engine.types.image import Image
from agentopera.engine.types.models import FunctionExecutionResult, LLMMessage, UserMessage
# Type aliases for convenience
_UserContent = Union[str, List[Union[str, Image]]]
_AssistantContent = Union[str, List[FunctionCall]]
_FunctionExecutionContent = List[FunctionExecutionResult]
_SystemContent = str
[docs]
def content_to_str(content: _UserContent | _AssistantContent | _FunctionExecutionContent | _SystemContent) -> str:
"""Convert the content of an LLMMessage to a string."""
if isinstance(content, str):
return content
else:
result: List[str] = []
for c in content:
if isinstance(c, str):
result.append(c)
elif isinstance(c, Image):
result.append("<image>")
else:
result.append(str(c))
return "\n".join(result)
[docs]
def remove_images(messages: List[LLMMessage]) -> List[LLMMessage]:
"""Remove images from a list of LLMMessages"""
str_messages: List[LLMMessage] = []
for message in messages:
if isinstance(message, UserMessage) and isinstance(message.content, list):
str_messages.append(UserMessage(content=content_to_str(message.content), source=message.source))
else:
str_messages.append(message)
return str_messages