From e8cabfd1ae7cc93abc04745948db1f6933fd2e26 Mon Sep 17 00:00:00 2001 From: aziz-ullah-khan Date: Sun, 3 Nov 2024 20:23:29 +0500 Subject: [PATCH] fix: resolved outparser issue --- .ipynb_checkpoints/Untitled-checkpoint.ipynb | 6 ++++++ scrapegraphai/nodes/generate_answer_node.py | 6 +++--- 2 files changed, 9 insertions(+), 3 deletions(-) create mode 100644 .ipynb_checkpoints/Untitled-checkpoint.ipynb diff --git a/.ipynb_checkpoints/Untitled-checkpoint.ipynb b/.ipynb_checkpoints/Untitled-checkpoint.ipynb new file mode 100644 index 00000000..363fcab7 --- /dev/null +++ b/.ipynb_checkpoints/Untitled-checkpoint.ipynb @@ -0,0 +1,6 @@ +{ + "cells": [], + "metadata": {}, + "nbformat": 4, + "nbformat_minor": 5 +} diff --git a/scrapegraphai/nodes/generate_answer_node.py b/scrapegraphai/nodes/generate_answer_node.py index 30058ec5..56d57d09 100644 --- a/scrapegraphai/nodes/generate_answer_node.py +++ b/scrapegraphai/nodes/generate_answer_node.py @@ -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