change headers to optional

genia-dev · Mar 26, 2024 · 92bb3a3 · 92bb3a3
1 parent 10e5e7b
commit 92bb3a3
Showing 1 changed file with 2 additions and 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
@@ -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):