jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

chore: Add trace_type attribute to LCModelComponent

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-19 19:01:27 -03:00
commit 25b4f94b33
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/base/models/model.py
View file

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