diff --git a/src/backend/base/langflow/base/models/model.py b/src/backend/base/langflow/base/models/model.py index 48fc99fe3..b32540ef6 100644 --- a/src/backend/base/langflow/base/models/model.py +++ b/src/backend/base/langflow/base/models/model.py @@ -166,7 +166,7 @@ class LCModelComponent(Component): 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 880af91df..f3e9ee23f 100644 --- a/src/backend/base/langflow/custom/custom_component/component.py +++ b/src/backend/base/langflow/custom/custom_component/component.py @@ -682,3 +682,8 @@ class Component(CustomComponent): from langflow.base.tools.component_tool import ComponentTool return ComponentTool(component=self) + + def get_project_name(self): + if hasattr(self, "_tracing_service"): + return self._tracing_service.project_name + return "Langflow"