ys/GenericAgent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: optimize streaming output, add thread safety, and refine system prompts

Browse Source
This commit is contained in:
Jiaqing Liang
2026-02-05 13:40:37 +08:00
parent 16bf93dafa
commit 52558c7f60
5 changed files with 71 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
agentmain.py
View File

@@ -79,11 +79,13 @@ class GeneraticAgent:
raw_query, handler, TOOLS_SCHEMA, max_turns=25)
try:
full_response = ""
full_response = ""; last_pos = 0
for chunk in gen:
if self.stop_sig: break
full_response += chunk
self.display_queue.put({'next': f'{full_response}', 'source': source})
if len(full_response) - last_pos > 50:
self.display_queue.put({'next': f'{full_response}', 'source': source})
last_pos = len(full_response)
if '</summary>' in full_response: full_response = full_response.replace('</summary>', '</summary>\n\n')
if '</file_content>' in full_response: full_response = re.sub(r'<file_content>\s*(.*?)\s*</file_content>', r'\n````\n<file_content>\n\1\n</file_content>\n````', full_response, flags=re.DOTALL)
self.display_queue.put({'done': full_response, 'source': source})