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

fix: stream option is not working with tracing enabled (#2602)

Browse source 
* fix: cannot create 'generator' instances

* fix: cannot create 'generator' instances

* [autofix.ci] apply automated fixes

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
Nicolò Boschi 2024-07-09 15:37:47 +02:00 • committed by GitHub
commit 21adbd5531
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 40 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

17
src/backend/base/langflow/schema/message.py
View file

@ -86,24 +86,21 @@ class Message(Data):
# they are: "text", "sender"
if self.text is None or not self.sender:
logger.warning("Missing required keys ('text', 'sender') in Message, defaulting to HumanMessage.")
if not isinstance(self.text, str):
text = ""
else:
text = self.text
if self.sender == "User" or not self.sender:
if self.files:
contents = [{"type": "text", "text": self.text}]
contents = [{"type": "text", "text": text}]
contents.extend(self.get_file_content_dicts())
human_message = HumanMessage(content=contents) # type: ignore
else:
if not isinstance(self.text, str):
text = ""
else:
text = self.text
human_message = HumanMessage(
content=text,
)
human_message = HumanMessage(content=text)
return human_message
return AIMessage(content=self.text) # type: ignore
return AIMessage(content=text) # type: ignore
@classmethod
def from_data(cls, data: "Data") -> "Message":

13
src/backend/base/langflow/services/tracing/service.py
View file

@ -1,6 +1,7 @@
import asyncio
import os
import traceback
import types
from collections import defaultdict
from contextlib import asynccontextmanager
from datetime import datetime, timezone
@ -214,11 +215,8 @@ class LangSmithTracer(BaseTracer):
):
if not self._ready:
return
raw_inputs = {}
processed_inputs = {}
if inputs:
raw_inputs = inputs.copy()
raw_inputs |= metadata or {}
processed_inputs = self._convert_to_langchain_types(inputs)
child = self._run_tree.create_child(
name=trace_name,
@ -226,7 +224,7 @@ class LangSmithTracer(BaseTracer):
inputs=processed_inputs,
)
if metadata:
child.add_metadata(metadata)
child.add_metadata(self._convert_to_langchain_types(metadata))
self._children[trace_name] = child
self._child_link: dict[str, str] = {}
@ -254,6 +252,9 @@ class LangSmithTracer(BaseTracer):
value = value.to_lc_document()
elif isinstance(value, Data):
value = value.to_lc_document()
elif isinstance(value, types.GeneratorType):
# generator is not serializable, also we can't consume it
value = str(value)
return value
def end_trace(
@ -270,8 +271,8 @@ class LangSmithTracer(BaseTracer):
raw_outputs = outputs
processed_outputs = self._convert_to_langchain_types(outputs)
if logs:
child.add_metadata({"logs": {log.get("name"): log for log in logs}})
child.add_metadata({"outputs": raw_outputs})
child.add_metadata(self._convert_to_langchain_types({"logs": {log.get("name"): log for log in logs}}))
child.add_metadata(self._convert_to_langchain_types({"outputs": raw_outputs}))
child.end(outputs=processed_outputs, error=error)
if error:
child.patch()