diff --git a/src/backend/base/langflow/base/models/model.py b/src/backend/base/langflow/base/models/model.py index 48fc99fe3c96..b32540ef6648 100644 --- a/src/backend/base/langflow/base/models/model.py +++ b/src/backend/base/langflow/base/models/model.py @@ -166,7 +166,7 @@ def get_chat_result( runnable = runnable.with_config( # type: ignore { "run_name": self.display_name, - "project_name": self._tracing_service.project_name, # type: ignore + "project_name": self.get_project_name(), "callbacks": self.get_langchain_callbacks(), } ) diff --git a/src/backend/base/langflow/custom/custom_component/component.py b/src/backend/base/langflow/custom/custom_component/component.py index 6f8767c3209f..baf7c9d82573 100644 --- a/src/backend/base/langflow/custom/custom_component/component.py +++ b/src/backend/base/langflow/custom/custom_component/component.py @@ -715,3 +715,7 @@ def log(self, message: LoggableType | list[LoggableType], name: Optional[str] = data["output"] = self._current_output data["component_id"] = self._id self._log_callback(event_name, data) + def get_project_name(self): + if hasattr(self, "_tracing_service"): + return self._tracing_service.project_name + return "Langflow"