Merge pull request #215 from potpie-ai/stream_agent

Update EOL check in response
potpie-ai · Dec 20, 2024 · 9f68a2b · 9f68a2b
2 parents 275f0d0 + 6e147a7
commit 9f68a2b
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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