Agents: validate auth

cohere-ai · Jul 19, 2024 · 02318a6 · 02318a6
1 parent 8283345
commit 02318a6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/src/backend/config/routers.py b/src/backend/config/routers.py
@@ -99,8 +99,7 @@ class RouterName(StrEnum):
         ],
         "auth": [
             Depends(get_session),
-            # TODO: Add if the router's have to have authorization
-            # Depends(validate_authorization),
+            Depends(validate_authorization),
         ],
     },
     RouterName.DEFAULT_AGENT: {
@@ -109,8 +108,7 @@ class RouterName(StrEnum):
         ],
         "auth": [
             Depends(get_session),
-            # TODO: Add if the router's have to have authorization
-            # Depends(validate_authorization),
+            Depends(validate_authorization),
         ],
     },
     RouterName.SNAPSHOT: {

diff --git a/src/backend/routers/chat.py b/src/backend/routers/chat.py
@@ -46,7 +46,6 @@ async def chat_stream(
     trace_id = None
     if hasattr(request.state, "trace_id"):
         trace_id = request.state.trace_id
-    print("trace_id", trace_id)
 
     user_id = request.headers.get("User-Id", None)
     agent_id = chat_request.agent_id