Tangledup-ai/lang-agent
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

pipeline to normal

Browse Source
This commit is contained in:
goulustis
2025-11-07 20:33:43 +08:00
parent 248e8e2338
commit ac535ce26b
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lang_agent/pipeline.py
View File

@@ -118,9 +118,7 @@ class Pipeline:
# If streaming, yield chunks from the generator # If streaming, yield chunks from the generator
if kwargs.get("as_stream"): if kwargs.get("as_stream"):
for chunk in out: return self._stream_res(out)
yield chunk
return
# Non-streaming path # Non-streaming path
if kwargs.get("as_raw"): if kwargs.get("as_raw"):
@@ -165,6 +163,9 @@ class Pipeline:
def get_ws_url(self): def get_ws_url(self):
return f"ws://{self.config.host}:{self.config.port}" return f"ws://{self.config.host}:{self.config.port}"
def _stream_res(self, out:list):
for chunk in out:
yield chunk
def chat(self, inp:str, as_stream:bool=False, as_raw:bool=False, thread_id:int = None): def chat(self, inp:str, as_stream:bool=False, as_raw:bool=False, thread_id:int = None):
# NOTE: this prompt will be overwritten by 'configs/route_sys_prompts/chat_prompt.txt' for route graph # NOTE: this prompt will be overwritten by 'configs/route_sys_prompts/chat_prompt.txt' for route graph
@@ -178,7 +179,6 @@ class Pipeline:
if as_stream: if as_stream:
# Yield chunks from the generator # Yield chunks from the generator
for chunk in out: return self._stream_res(out)
yield chunk
else: else:
return out return out