CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-07-12 21:01:56 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: resolved outparser issue

Browse Source
This commit is contained in:
aziz-ullah-khan 2024-11-03 20:23:29 +05:00
parent 65d39bbaf0
commit e8cabfd1ae
2 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
.ipynb_checkpoints/Untitled-checkpoint.ipynb Normal file
View File

@ -0,0 +1,6 @@
{
"cells": [],
"metadata": {},
"nbformat": 4,
"nbformat_minor": 5
}

6
scrapegraphai/nodes/generate_answer_node.py
View File

@ -122,11 +122,11 @@ class GenerateAnswerNode(BaseNode):
partial_variables={"context": doc, "format_instructions": format_instructions}
)
chain = prompt | self.llm_model
raw_response = str((prompt | self.llm_model).invoke({"question": user_prompt}))
raw_response = chain.invoke({"question": user_prompt})
if output_parser:
try:
answer = output_parser.parse(raw_response)
answer = output_parser.parse(raw_response.content)
except JSONDecodeError:
lines = raw_response.split('\n')
if lines[0].strip().startswith('```'):
@ -136,7 +136,7 @@ class GenerateAnswerNode(BaseNode):
cleaned_response = '\n'.join(lines)
answer = output_parser.parse(cleaned_response)
else:
answer = raw_response
answer = raw_response.content
state.update({self.output[0]: answer})
return state