mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 08:08:32 +00:00
89 lines
3.0 KiB
Python
89 lines
3.0 KiB
Python
from typing import Any, Dict, List, Union
|
|
|
|
import pytest
|
|
from pydantic import ValidationError
|
|
|
|
from crewai.utilities.events.llm_events import LLMCallStartedEvent
|
|
|
|
|
|
def test_llm_call_started_event_with_multimodal_content():
|
|
"""Test that LLMCallStartedEvent properly handles multimodal content."""
|
|
# Create a multimodal message structure
|
|
multimodal_message = {
|
|
'role': 'user',
|
|
'content': [
|
|
{'type': 'text', 'text': 'Please analyze this image'},
|
|
{
|
|
'type': 'image_url',
|
|
'image_url': {
|
|
'url': 'https://example.com/test-image.jpg',
|
|
},
|
|
},
|
|
],
|
|
}
|
|
|
|
# This should not raise a ValidationError
|
|
event = LLMCallStartedEvent(messages=[multimodal_message])
|
|
|
|
# Verify the event was created correctly
|
|
assert event.messages[0]['role'] == 'user'
|
|
assert isinstance(event.messages[0]['content'], list)
|
|
assert len(event.messages[0]['content']) == 2
|
|
assert event.messages[0]['content'][0]['type'] == 'text'
|
|
assert event.messages[0]['content'][1]['type'] == 'image_url'
|
|
|
|
|
|
def test_llm_call_started_event_with_string_message():
|
|
"""Test that LLMCallStartedEvent still works with string messages."""
|
|
# Create a simple string message
|
|
message = "This is a test message"
|
|
|
|
# This should not raise a ValidationError
|
|
event = LLMCallStartedEvent(messages=message)
|
|
|
|
# Verify the event was created correctly
|
|
assert event.messages == message
|
|
|
|
|
|
def test_llm_call_started_event_with_standard_messages():
|
|
"""Test that LLMCallStartedEvent still works with standard message format."""
|
|
# Create standard messages
|
|
messages = [
|
|
{"role": "system", "content": "You are a helpful assistant"},
|
|
{"role": "user", "content": "Hello, how are you?"}
|
|
]
|
|
|
|
# This should not raise a ValidationError
|
|
event = LLMCallStartedEvent(messages=messages)
|
|
|
|
# Verify the event was created correctly
|
|
assert len(event.messages) == 2
|
|
assert event.messages[0]['role'] == 'system'
|
|
assert event.messages[0]['content'] == 'You are a helpful assistant'
|
|
assert event.messages[1]['role'] == 'user'
|
|
assert event.messages[1]['content'] == 'Hello, how are you?'
|
|
|
|
|
|
def test_llm_call_started_event_with_mixed_content():
|
|
"""Test that LLMCallStartedEvent handles mixed content types."""
|
|
mixed_messages = [
|
|
"Simple string message",
|
|
{
|
|
'role': 'user',
|
|
'content': [
|
|
{'type': 'text', 'text': 'With image'},
|
|
{'type': 'image_url', 'image_url': {'url': 'https://example.com/image.jpg'}},
|
|
],
|
|
}
|
|
]
|
|
|
|
# This should not raise a ValidationError
|
|
event = LLMCallStartedEvent(messages=mixed_messages)
|
|
|
|
# Verify the event was created correctly
|
|
assert isinstance(event.messages, list)
|
|
assert isinstance(event.messages[0], str)
|
|
assert isinstance(event.messages[1], dict)
|
|
assert event.messages[1]['role'] == 'user'
|
|
assert isinstance(event.messages[1]['content'], list)
|