From 9d1eeaeec56d368b84b065cc2fc4292bd7c7de3e Mon Sep 17 00:00:00 2001 From: goulustis Date: Wed, 4 Mar 2026 14:41:58 +0800 Subject: [PATCH] pass in pipeline_id --- lang_agent/pipeline.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lang_agent/pipeline.py b/lang_agent/pipeline.py index 3375a68..9b77db6 100644 --- a/lang_agent/pipeline.py +++ b/lang_agent/pipeline.py @@ -135,7 +135,7 @@ class Pipeline: if isinstance(chunk, str): yield chunk else: - conv_store.CONV_STORE.record_message_list(conv_id, chunk) + conv_store.CONV_STORE.record_message_list(conv_id, chunk, pipeline_id=self.config.pipeline_id) async def _astream_res(self, out, conv_id:str=None): """Async version of _stream_res for async generators.""" @@ -143,7 +143,7 @@ class Pipeline: if isinstance(chunk, str): yield chunk else: - conv_store.CONV_STORE.record_message_list(conv_id, chunk) + conv_store.CONV_STORE.record_message_list(conv_id, chunk, pipeline_id=self.config.pipeline_id) def chat(self, inp:str, as_stream:bool=False, as_raw:bool=False, thread_id:str = '3'): """