Skip to content

Commit 2399b7d

Browse files
feng-95innsd
authored andcommitted
fix: update session id
(cherry picked from commit c64034eac9953a6a94e4f0bfe6bc5454457ed84a)
1 parent ebf4a70 commit 2399b7d

File tree

3 files changed

+5
-6
lines changed

3 files changed

+5
-6
lines changed

agentkit/apps/agent_server_app/agent_server_app.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -104,8 +104,7 @@ async def _invoke_compat(request: Request):
104104
)
105105
session_id = (
106106
headers.get("session_id")
107-
or headers.get("x-session-id")
108-
or "agentkit_sample_session"
107+
or ""
109108
)
110109

111110
# Determine app_name from loader

agentkit/apps/agent_server_app/middleware.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,8 +43,8 @@ async def __call__(self, scope, receive, send):
4343
}
4444

4545
# Currently unable to retrieve user_id and session_id from headers; keep logic for future use
46-
user_id = headers.get("user_id") or headers.get("x-user-id") or ""
47-
session_id = headers.get("session_id") or headers.get("x-session-id") or ""
46+
user_id = headers.get("user_id")
47+
session_id = headers.get("session_id")
4848
headers["user_id"] = user_id
4949
headers["session_id"] = session_id
5050
telemetry.trace_agent_server(

agentkit/apps/agent_server_app/telemetry.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,10 +72,10 @@ def trace_agent_server(
7272
value=safe_serialize_to_json_string(headers),
7373
)
7474

75-
session_id = headers.get("session_id") or headers.get("x-session-id") or ""
75+
session_id = headers.get("session_id")
7676
if session_id:
7777
span.set_attribute(key="gen_ai.session.id", value=session_id)
78-
user_id = headers.get("user_id") or headers.get("x-user-id") or ""
78+
user_id = headers.get("user_id")
7979
if user_id:
8080
span.set_attribute(key="gen_ai.user.id", value=user_id)
8181

0 commit comments

Comments
 (0)