From 05cf9adaa7bed093c5700cd2feb47b01ab88402f Mon Sep 17 00:00:00 2001 From: amosdinh Date: Fri, 26 Jul 2024 12:10:14 +0200 Subject: [PATCH] Ollama: Use no json format when creating the search query --- scrapegraphai/nodes/search_internet_node.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/scrapegraphai/nodes/search_internet_node.py b/scrapegraphai/nodes/search_internet_node.py index 97fed67b..2474ab60 100644 --- a/scrapegraphai/nodes/search_internet_node.py +++ b/scrapegraphai/nodes/search_internet_node.py @@ -10,6 +10,7 @@ from langchain.prompts import PromptTemplate from ..utils.logging import get_logger from ..utils.research_web import search_on_web from .base_node import BaseNode +from ..models import Ollama class SearchInternetNode(BaseNode): @@ -94,7 +95,14 @@ class SearchInternetNode(BaseNode): # Execute the chain to get the search query search_answer = search_prompt | self.llm_model | output_parser - search_query = search_answer.invoke({"user_prompt": user_prompt})[0] + + # Ollama: Use no json format when creating the search query + if isinstance(self.llm_model, Ollama) and self.llm_model.format == 'json': + self.llm_model.format = None + search_query = search_answer.invoke({"user_prompt": user_prompt})[0] + self.llm_model.format = 'json' + else: + search_query = search_answer.invoke({"user_prompt": user_prompt})[0] self.logger.info(f"Search Query: {search_query}")