hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-15 22:06:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: mypy for all type check (#10921)

Browse Source
This commit is contained in:
yihong
2024-12-24 18:38:51 +08:00
committed by GitHub
parent c91e8b1737
commit 56e15d09a9
584 changed files with 3975 additions and 2826 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
api/core/agent/output_parser/cot_output_parser.py
View File

@@ -38,7 +38,7 @@ class CotAgentOutputParser:
except:
return json_str or ""
def extra_json_from_code_block(code_block) -> Generator[Union[dict, str], None, None]:
def extra_json_from_code_block(code_block) -> Generator[Union[str, AgentScratchpadUnit.Action], None, None]:
code_blocks = re.findall(r"```(.*?)```", code_block, re.DOTALL)
if not code_blocks:
return
@@ -67,15 +67,15 @@ class CotAgentOutputParser:
for response in llm_response:
if response.delta.usage:
usage_dict["usage"] = response.delta.usage
response = response.delta.message.content
if not isinstance(response, str):
response_content = response.delta.message.content
if not isinstance(response_content, str):
continue
# stream
index = 0
while index < len(response):
while index < len(response_content):
steps = 1
delta = response[index : index + steps]
delta = response_content[index : index + steps]
yield_delta = False
if delta == "`":