fix: handle missing name in on_retriever_start

lunary-ai · Dec 11, 2024 · e79c629 · e79c629
1 parent d802e14
commit e79c629
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lunary/__init__.py b/lunary/__init__.py
@@ -1402,6 +1402,7 @@ def on_retriever_start(
             query: str,
             run_id: Optional[UUID] = None,
             parent_run_id: Optional[UUID] = None,
+            name: Union[str, None] = None,
             **kwargs: Any,
         ) -> None:
             try:
@@ -1412,7 +1413,8 @@ def on_retriever_start(
                 user_id = _get_user_id(kwargs.get("metadata"))
                 user_props = _get_user_props(kwargs.get("metadata"))
 
-                name = serialized.get("name")
+                if name is None and serialized:
+                    name = serialized.get("name")
 
                 self.__track_event(
                     "retriever",