Skip to content

Commit 6b55ce6

Browse files
committed
chore: log raw and parsed JSON in PromptDraftNode for debugging
1 parent b9d2522 commit 6b55ce6

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

api/nodes/new_pipeline/prompt_draft_node.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,7 @@
11
from api.nodes.fetch_summary_node import Node
2+
import logging
3+
logging.basicConfig(level=logging.INFO)
4+
logger = logging.getLogger(__name__)
25

36
@Node(retries=3)
47
def PromptDraftNode(text: str, framework_plan: dict) -> dict[str, list[str]]:
@@ -52,6 +55,7 @@ def PromptDraftNode(text: str, framework_plan: dict) -> dict[str, list[str]]:
5255
)
5356

5457
raw = resp.choices[0].message.content.strip()
58+
logger.info("PromptDraftNode raw LLM output: %s", raw[:1000])
5559
# Strip code fences and extract JSON object
5660
if raw.startswith("```"):
5761
parts = raw.split("```")
@@ -64,6 +68,7 @@ def PromptDraftNode(text: str, framework_plan: dict) -> dict[str, list[str]]:
6468
raw = raw[obj_start:obj_end+1]
6569
try:
6670
prompts_by_cat = json.loads(raw)
71+
logger.info("PromptDraftNode parsed JSON: %r", prompts_by_cat)
6772
except json.JSONDecodeError:
6873
logger.error("PromptDraftNode JSON parse error: %s", raw)
6974
raise

0 commit comments

Comments
 (0)