From 7eda6bc06bc4c32850029f54b9b4c22f3124296e Mon Sep 17 00:00:00 2001 From: Marco Vinciguerra Date: Sun, 22 Sep 2024 18:55:05 +0200 Subject: [PATCH] fix: issue about parser --- scrapegraphai/nodes/generate_answer_node.py | 10 ++++------ scrapegraphai/utils/research_web.py | 4 +++- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/scrapegraphai/nodes/generate_answer_node.py b/scrapegraphai/nodes/generate_answer_node.py index 96ca0238..d380d7b8 100644 --- a/scrapegraphai/nodes/generate_answer_node.py +++ b/scrapegraphai/nodes/generate_answer_node.py @@ -96,14 +96,12 @@ class GenerateAnswerNode(BaseNode): output_parser = get_structured_output_parser(self.node_config["schema"]) format_instructions = "NA" else: - if not isinstance(self.llm_model, ChatBedrock): - output_parser = get_pydantic_output_parser(self.node_config["schema"]) - format_instructions = output_parser.get_format_instructions() + output_parser = get_pydantic_output_parser(self.node_config["schema"]) + format_instructions = output_parser.get_format_instructions() else: - if not isinstance(self.llm_model, ChatBedrock): - output_parser = JsonOutputParser() - format_instructions = output_parser.get_format_instructions() + output_parser = JsonOutputParser() + format_instructions = output_parser.get_format_instructions() if isinstance(self.llm_model, (ChatOpenAI, AzureChatOpenAI)) \ and not self.script_creator \ diff --git a/scrapegraphai/utils/research_web.py b/scrapegraphai/utils/research_web.py index 0a10c8f2..7e978ffd 100644 --- a/scrapegraphai/utils/research_web.py +++ b/scrapegraphai/utils/research_web.py @@ -60,7 +60,9 @@ def search_on_web(query: str, search_engine: str = "Google", elif search_engine.lower() == "searxng": url = f"http://localhost:{port}" - params = {"q": query, "format": "json"} + params = {"q": query, + "format": "json", + "engines": "google,duckduckgo,brave,qwant,bing"} response = requests.get(url, params=params)