fixed ollama embedder params

Author: PeriniM

examples/local_models/Docker/smart_scraper_docker.py

@@ -10,14 +10,14 @@
 
 graph_config = {
     "llm": {
-        "model": "ollama/mistral",
+        "model": "ollama/llama3",
         "temperature": 0,
         "format": "json",
-        # "model_tokens": 2000, # set context length arbitrarily,
+        "model_tokens": 2000, # set context length arbitrarily,
     },
     "embeddings": {
         "model": "ollama/nomic-embed-text",
-        "temperature": 0,
+        "base_url": "http://localhost:11434",
     }
 }
 

scrapegraphai/nodes/rag_node.py

@@ -92,7 +92,13 @@ def execute(self, state):
         elif isinstance(embedding_model, AzureOpenAI):
             embeddings = AzureOpenAIEmbeddings()
         elif isinstance(embedding_model, Ollama):
-            embeddings = OllamaEmbeddings(model=embedding_model.model)
+            # unwrap the kwargs from the model whihc is a dict
+            params = embedding_model._lc_kwargs
+            # remove streaming and temperature
+            params.pop("streaming", None)
+            params.pop("temperature", None)
+            
+            embeddings = OllamaEmbeddings(**params)
         elif isinstance(embedding_model, HuggingFace):
             embeddings = HuggingFaceHubEmbeddings(model=embedding_model.model)
         else: