Merge pull request #490 from AmosDinh/pre/beta

Ollama: Use no json format when creating the search query

Author: VinciGit00

scrapegraphai/nodes/search_internet_node.py

@@ -10,6 +10,7 @@
 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 @@ def execute(self, state: dict) -> dict:
 
         # 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}")