Skip to content

Commit 38fff84

Browse files
committed
fix type in StrayCat.user_data
1 parent b04bcfa commit 38fff84

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

Diff for: core/cat/looking_glass/stray_cat.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
from cat.memory.working_memory import WorkingMemory
1919
from cat.convo.messages import CatMessage, UserMessage, MessageWhy, Role, EmbedderModelInteraction
2020
from cat.agents import AgentOutput
21+
from cat.auth.permissions import AuthUserInfo
2122
from cat import utils
2223
from websockets.exceptions import ConnectionClosedOK
2324

@@ -32,7 +33,7 @@ def __init__(
3233
self,
3334
user_id: str,
3435
main_loop,
35-
user_data: dict = {},
36+
user_data: AuthUserInfo = None,
3637
ws: WebSocket = None,
3738
):
3839
self.__user_id = user_id

Diff for: core/tests/routes/test_session.py

+2
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11

22
from cat.looking_glass.stray_cat import StrayCat
3+
from cat.auth.permissions import AuthUserInfo
34

45
from tests.utils import send_websocket_message
56

@@ -18,6 +19,7 @@ def test_session_creation_from_websocket(client):
1819
assert isinstance(strays["Alice"], StrayCat)
1920
assert strays["Alice"].user_id == "Alice"
2021
assert hasattr(strays["Alice"], "user_data")
22+
assert isinstance(strays["Alice"].user_data, AuthUserInfo)
2123
assert strays["Alice"].user_data.id == "Alice"
2224
convo = strays["Alice"].working_memory.history
2325
assert len(convo) == 2

0 commit comments

Comments
 (0)