diff --git a/examples/chat_message_history_example.py b/examples/chat_message_history_example.py index 964016457..1f77f6be3 100644 --- a/examples/chat_message_history_example.py +++ b/examples/chat_message_history_example.py @@ -1,4 +1,4 @@ -from crewai import Agent, Task, Crew +from crewai import Agent, Crew, Task from crewai.memory import ChatMessageHistory, MessageRole # Create a chat message history diff --git a/tests/memory/chat_message_history_test.py b/tests/memory/chat_message_history_test.py index 14e33c0ff..bdd303541 100644 --- a/tests/memory/chat_message_history_test.py +++ b/tests/memory/chat_message_history_test.py @@ -1,8 +1,9 @@ -import pytest from datetime import datetime -from crewai.memory.chat_history.chat_message_history import ChatMessageHistory +import pytest + from crewai.memory.chat_history.chat_message import ChatMessage, MessageRole +from crewai.memory.chat_history.chat_message_history import ChatMessageHistory @pytest.fixture