Skip to content

Commit 6a8c7f6

Browse files
committed
Fix mypy
1 parent eee38c9 commit 6a8c7f6

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

plugins/getstream/tests/test_stream_conversation.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,12 @@
77
from unittest.mock import Mock, AsyncMock
88
from dotenv import load_dotenv
99

10-
from getstream.models import MessageRequest, ChannelInput, MessagePaginationParams
10+
from getstream.models import (
11+
Message,
12+
MessageRequest,
13+
ChannelInput,
14+
MessagePaginationParams,
15+
)
1116
from getstream import AsyncStream
1217

1318
from vision_agents.plugins.getstream.stream_conversation import StreamConversation
@@ -48,7 +53,7 @@ def mock_channel(self):
4853
def stream_conversation(self, mock_channel):
4954
"""Create a StreamConversation instance with mocked dependencies."""
5055
instructions = "You are a helpful assistant."
51-
messages = []
56+
messages: list[Message] = []
5257
conversation = StreamConversation(
5358
instructions=instructions, messages=messages, channel=mock_channel
5459
)

0 commit comments

Comments
 (0)