From 6e147a7c3fdfb1dbf10c5df5d17c326eb5fa13c1 Mon Sep 17 00:00:00 2001 From: dhirenmathur Date: Fri, 20 Dec 2024 19:56:24 +0530 Subject: [PATCH] Update EOL check --- app/modules/intelligence/agents/agents/code_gen_agent.py | 2 +- app/modules/intelligence/agents/agents/debug_rag_agent.py | 2 +- .../intelligence/agents/agents/low_level_design_agent.py | 2 +- app/modules/intelligence/agents/agents/rag_agent.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/app/modules/intelligence/agents/agents/code_gen_agent.py b/app/modules/intelligence/agents/agents/code_gen_agent.py index 59e58440..aa25cf56 100644 --- a/app/modules/intelligence/agents/agents/code_gen_agent.py +++ b/app/modules/intelligence/agents/agents/code_gen_agent.py @@ -302,7 +302,7 @@ async def kickoff(): if not line: break if final_answer_streaming: - if line.endswith("\\x1b[00m\\n"): + if line.endswith("\x1b[00m\n"): yield line[:-6] else: yield line diff --git a/app/modules/intelligence/agents/agents/debug_rag_agent.py b/app/modules/intelligence/agents/agents/debug_rag_agent.py index e5cb6d0d..62a523d1 100644 --- a/app/modules/intelligence/agents/agents/debug_rag_agent.py +++ b/app/modules/intelligence/agents/agents/debug_rag_agent.py @@ -274,7 +274,7 @@ async def kickoff(): if not line: break if final_answer_streaming: - if line.endswith("\\x1b[00m\\n"): + if line.endswith("\x1b[00m\n"): yield line[:-6] else: yield line diff --git a/app/modules/intelligence/agents/agents/low_level_design_agent.py b/app/modules/intelligence/agents/agents/low_level_design_agent.py index ec6cfe15..8c36585c 100644 --- a/app/modules/intelligence/agents/agents/low_level_design_agent.py +++ b/app/modules/intelligence/agents/agents/low_level_design_agent.py @@ -196,7 +196,7 @@ async def kickoff(): if not line: break if final_answer_streaming: - if line.endswith("\\x1b[00m\\n"): + if line.endswith("\x1b[00m\n"): yield line[:-6] else: yield line diff --git a/app/modules/intelligence/agents/agents/rag_agent.py b/app/modules/intelligence/agents/agents/rag_agent.py index 6c9f6204..c9e23c42 100644 --- a/app/modules/intelligence/agents/agents/rag_agent.py +++ b/app/modules/intelligence/agents/agents/rag_agent.py @@ -298,7 +298,7 @@ async def kickoff(): break else: if final_answer_streaming: - if line.endswith("\\x1b[00m\\n"): + if line.endswith("\x1b[00m\n"): yield line[:-6] else: yield line