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

refactor: Add child link dictionary to LangSmithTracer class

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-21 23:32:51 -03:00
commit bf57cb8e43
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -127,7 +127,7 @@ class TracingService(Service):
self.logs_queue.put_nowait((self._add_log, (trace_name, log))) self.logs_queue.put_nowait((self._add_log, (trace_name, log)))
@asynccontextmanager @asynccontextmanager
def trace_context( async def trace_context(
self, trace_name: str, trace_type: str, inputs: Dict[str, Any] = None, metadata: Dict[str, Any] = None self, trace_name: str, trace_type: str, inputs: Dict[str, Any] = None, metadata: Dict[str, Any] = None
): ):
self._start_traces(trace_name, trace_type, inputs, metadata) self._start_traces(trace_name, trace_type, inputs, metadata)
@ -196,6 +196,7 @@ class LangSmithTracer:
if metadata: if metadata:
child.add_metadata(raw_inputs) child.add_metadata(raw_inputs)
self._children[trace_name] = child self._children[trace_name] = child
self._child_link = {}
def _convert_to_langchain_types(self, io_dict: Dict[str, Any]): def _convert_to_langchain_types(self, io_dict: Dict[str, Any]):
converted = {} converted = {}