phowell
/
pocketflow
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #40 from gwtt/gwtt 

fix pocketflow-workflow yaml parse error
This commit is contained in:
Zachary Huang 2025-04-23 10:09:13 -04:00 committed by GitHub
parent adac24b095 1a64be5031
commit 26fd2d2cfc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
cookbook/pocketflow-workflow/nodes.py
View File

@ -1,3 +1,4 @@
import re
from pocketflow import Node, BatchNode from pocketflow import Node, BatchNode
from utils import call_llm from utils import call_llm
import yaml import yaml
@ -21,6 +22,11 @@ sections:
```""" ```"""
response = call_llm(prompt) response = call_llm(prompt)
yaml_str = response.split("```yaml")[1].split("```")[0].strip() yaml_str = response.split("```yaml")[1].split("```")[0].strip()
yaml_str = re.sub(
r'(?m)^(\s*)-\s*([^"\n]+)',
r'\1- "\2"',
yaml_str
)
structured_result = yaml.safe_load(yaml_str) structured_result = yaml.safe_load(yaml_str)
return structured_result return structured_result