From 92bb3a3262ee30cc2287696039ffa5f8d659e4c2 Mon Sep 17 00:00:00 2001 From: Uri Shamay Date: Tue, 26 Mar 2024 23:44:29 +0200 Subject: [PATCH] change headers to optional --- .../open_telemetry/open_telemetry_parser.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/vibraniumdome-shields/vibraniumdome_shields/open_telemetry/open_telemetry_parser.py b/vibraniumdome-shields/vibraniumdome_shields/open_telemetry/open_telemetry_parser.py index ce4abc1..527012e 100644 --- a/vibraniumdome-shields/vibraniumdome_shields/open_telemetry/open_telemetry_parser.py +++ b/vibraniumdome-shields/vibraniumdome_shields/open_telemetry/open_telemetry_parser.py @@ -175,7 +175,8 @@ def _parse_headers(self, llm_interaction, document): llm_interaction["llm.user"] = document.get("llm.user") headers_dict = safe_loads_dictionary_string(document.get("llm.headers")) # parse all meaningful headers here - llm_interaction["session-id"] = headers_dict.get("x-session-id") + if headers_dict: + llm_interaction["session-id"] = headers_dict.get("x-session-id") return llm_interaction def _parse_trace_request(self, export_trace_service_request):