From f2a9ba55c5f2db08e32ab21de65eb29dec576509 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 22 Aug 2024 13:20:27 -0300 Subject: [PATCH] feat: Update project name retrieval in LCModelComponent and CustomComponent Return the project name using get_project_name method in both classes. --- src/backend/base/langflow/base/models/model.py | 2 +- .../base/langflow/custom/custom_component/component.py | 5 +++++ 2 files changed, 6 insertions(+), 1 deletion(-) 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 880af91dfb9e..f3e9ee23f693 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 @@ def to_tool(self): 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"