From 25b4f94b33f6f4393a80f2077c79d3bbc16120cf Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@langflow.org>
Date: Wed, 19 Jun 2024 19:01:27 -0300
Subject: [PATCH] chore: Add trace_type attribute to LCModelComponent

---
 src/backend/base/langflow/base/models/model.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/backend/base/langflow/base/models/model.py b/src/backend/base/langflow/base/models/model.py
index d767d637c..89973faab 100644
--- a/src/backend/base/langflow/base/models/model.py
+++ b/src/backend/base/langflow/base/models/model.py
@@ -14,6 +14,7 @@ from langflow.template.field.base import Output
 class LCModelComponent(Component):
     display_name: str = "Model Name"
     description: str = "Model Description"
+    trace_type = "llm"
 
     outputs = [
         Output(display_name="Text", name="text_output", method="text_response"),
@@ -138,6 +139,9 @@ class LCModelComponent(Component):
                 messages.append(HumanMessage(content=input_value))
         inputs: Union[list, dict] = messages or {}
         try:
+            runnable = runnable.with_config(
+                {"run_name": self.display_name, "project_name": self.tracing_service.project_name}
+            )
             if stream:
                 return runnable.stream(inputs)
             else: