pass in pipeline_id
This commit is contained in:
@@ -135,7 +135,7 @@ class Pipeline:
|
|||||||
if isinstance(chunk, str):
|
if isinstance(chunk, str):
|
||||||
yield chunk
|
yield chunk
|
||||||
else:
|
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 def _astream_res(self, out, conv_id:str=None):
|
||||||
"""Async version of _stream_res for async generators."""
|
"""Async version of _stream_res for async generators."""
|
||||||
@@ -143,7 +143,7 @@ class Pipeline:
|
|||||||
if isinstance(chunk, str):
|
if isinstance(chunk, str):
|
||||||
yield chunk
|
yield chunk
|
||||||
else:
|
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'):
|
def chat(self, inp:str, as_stream:bool=False, as_raw:bool=False, thread_id:str = '3'):
|
||||||
"""
|
"""
|
||||||
|
|||||||
Reference in New Issue
Block a user