fix tests

api/app/endpoints/stream.py

@@ -241,7 +241,7 @@ def start_stream(
     if db_stream is None:
         raise HTTPException(404, detail="Stream not found")
 
-    if current_user.id not in (db_stream.user_id, chat.user_id):
+    if current_user.id not in (db_stream.user_id, getattr(db_stream.chat, "user_id", None)):
         raise HTTPException(403, detail="Forbidden")
 
     # Get and configure the request parameters
@@ -260,8 +260,8 @@ def start_stream(
     should_sids = db_stream.should_sids
     must_not_sids = db_stream.must_not_sids
     postprocessing = db_stream.postprocessing
-    operators = chat.operators
-    themes = chat.themes
+    operators = getattr(chat, "operators", None)
+    themes = getattr(chat, "themes", None)
 
     sources = None
     if db_stream.sources:

api/app/tests/cases/archive.json

@@ -2,7 +2,8 @@
   "chat_name": "string",
   "chat_type": "meeting",
   "id": 1,
-  "operator": null,
+  "operators": ["test_operator"],
+  "themes": ["test_theme"],
   "stream_count": 2,
   "streams": [
     {
@@ -64,7 +65,6 @@
       "with_history": true
     }
   ],
-  "themes": null,
   "user_id": 1
 }
 

api/app/tests/endpoints/test_chat.py

@@ -22,7 +22,7 @@ def test_chat(self, client: TestClient):
         assert response.status_code == 200
 
         # Create Chat:
-        response = chat.create_chat(client, token, "qa")
+        response = chat.create_chat(client, token, "evaluations")
         assert response.status_code == 200
         chat_id = response.json()["id"]
 

api/app/tests/endpoints/test_stream.py

@@ -58,7 +58,7 @@ def test_chat_stream(self, client: TestClient, db):
         assert response.status_code == 200
 
         # Create Chat:
-        response = chat.create_chat(client, token, "meeting")
+        response = chat.create_chat(client, token, "meeting", operators=["test_operator"], themes=["test_theme"])
         assert response.status_code == 200
         chat_id = response.json()["id"]
 

api/app/tests/utils/chat.py

@@ -9,13 +9,11 @@ def read_chats(client: TestClient, token):
     return client.get(f"{ROOT_PATH}/chats", headers={"Authorization": f"Bearer {token}"})
 
 
-def create_chat(client: TestClient, token, chat_type):
+def create_chat(client: TestClient, token, chat_type, operators=None, themes=None):
     return client.post(
         f"{ROOT_PATH}/chat",
         headers={"Authorization": f"Bearer {token}"},
-        json={
-            "chat_type": chat_type,
-        },
+        json={"chat_type": chat_type, "operators": operators, "themes": themes},
     )