jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

🐛 fix(callback.py): remove "Prompt after formatting:\n" from text to improve formatting

Browse source 
🐛 fix(util.py): add remove_ansi_escape_codes function to remove ANSI escape codes from text
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-23 10:05:56 -03:00
commit e4646b1e5f
2 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/api/v1/callback.py
View file

@ -11,6 +11,7 @@ from fastapi import WebSocket
from langchain.schema import AgentAction, LLMResult, AgentFinish from langchain.schema import AgentAction, LLMResult, AgentFinish
from langflow.utils.logger import logger from langflow.utils.logger import logger
from langflow.utils.util import remove_ansi_escape_codes
# https://github.com/hwchase17/chat-langchain/blob/master/callback.py # https://github.com/hwchase17/chat-langchain/blob/master/callback.py
@ -105,7 +106,8 @@ class AsyncStreamingLLMCallbackHandler(AsyncCallbackHandler):
# to the LLM, adding it will send the final prompt # to the LLM, adding it will send the final prompt
# to the frontend # to the frontend
if "Prompt after formatting" in text: if "Prompt after formatting" in text:
text = text.replace("Prompt after formatting", "") text = text.replace("Prompt after formatting:\n", "")
text = remove_ansi_escape_codes(text)
resp = ChatResponse( resp = ChatResponse(
message="", message="",
type="stream", type="stream",

4
src/backend/langflow/utils/util.py
View file

@ -12,6 +12,10 @@ from langflow.utils.logger import logger
from multiprocess import cpu_count # type: ignore from multiprocess import cpu_count # type: ignore
def remove_ansi_escape_codes(text):
return re.sub(r"\x1b\[[0-9;]*[a-zA-Z]", "", text)
def build_template_from_function( def build_template_from_function(
name: str, type_to_loader_dict: Dict, add_function: bool = False name: str, type_to_loader_dict: Dict, add_function: bool = False
): ):