fix none varible defs and trace ingest

kasarolzzw · kasarolzzw · commit 70a3cb97b97a · 2025-06-30T12:17:35.000+08:00
Change-Id: Iab054fc781aed268418a2aa59528900fdccd10a5

fix none varible defs and trace ingest

Change-Id: I08bfbcb39557821c6d88cf59883b38c5a7677269
diff --git a/cozeloop/internal/prompt/prompt.py b/cozeloop/internal/prompt/prompt.py
@@ -162,7 +162,7 @@ def _format_normal_messages(
             variables: Dict[str, PromptVariable]
     ) -> List[Message]:
         results = []
-        variable_def_map = {var_def.key: var_def for var_def in variable_defs if var_def}
+        variable_def_map = {var_def.key: var_def for var_def in variable_defs if var_def} if variable_defs else {}
 
         for message in messages:
             if message is None:
diff --git a/cozeloop/internal/trace/span.py b/cozeloop/internal/trace/span.py
@@ -150,7 +150,8 @@ def set_input(self, input_data):
             if isinstance(input_data, ModelInput):
                 m_content = input_data
                 for message in input_data.messages:
-                    message_parts.extend(message.parts)
+                    if message.parts:
+                        message_parts.extend(message.parts)
 
             is_multi_modality = self.parse_model_message_parts(message_parts)
 
diff --git a/examples/prompt/prompt_hub.py b/examples/prompt/prompt_hub.py
@@ -11,6 +11,19 @@
 from cozeloop.spec.tracespec import CALL_OPTIONS, ModelCallOption, ModelMessage, ModelInput
 
 
+def convert_model_input(messages: List[Message]) -> ModelInput:
+    model_messages = []
+    for message in messages:
+        model_messages.append(ModelMessage(
+            role=str(message.role),
+            content=message.content if message.content is not None else ""
+        ))
+
+    return ModelInput(
+        messages=model_messages
+    )
+
+
 class LLMRunner:
     def __init__(self, client):
         self.client = client
@@ -49,14 +62,14 @@ def llm_call(self, input_data):
             span.set_output_tokens(output_token)
             # set tag key: `model_name`, e.g., gpt-4-1106-preview, etc.
             span.set_model_name("gpt-4-1106-preview")
-            span.set_tags(CALL_OPTIONS, ModelCallOption(
+            span.set_tags({CALL_OPTIONS: ModelCallOption(
                 temperature=0.5,
                 top_p=0.5,
                 top_k=10,
                 presence_penalty=0.5,
                 frequency_penalty=0.5,
                 max_tokens=1024,
-            ))
+            )})
 
             return None
         except Exception as e:
@@ -124,16 +137,3 @@ def llm_call(self, input_data):
     # Note that flush will block and wait for the report to complete, and it may cause frequent reporting,
     # affecting performance.
     client.flush()
-
-
-def convert_model_input(messages: List[Message]) -> ModelInput:
-    model_messages = []
-    for message in messages:
-        model_messages.append(ModelMessage(
-            role=str(message.role),
-            content=message.content if message.content is not None else ""
-        ))
-
-    return ModelInput(
-        messages=model_messages
-    )
