From 6c3b37ab001b80c09ea9ffb56d4c3df338e33a7a Mon Sep 17 00:00:00 2001 From: Marco Vinciguerra Date: Sat, 10 Aug 2024 11:51:37 +0200 Subject: [PATCH] feat: add refactoring of default temperature --- examples/local_models/smart_scraper_ollama.py | 1 - scrapegraphai/graphs/abstract_graph.py | 7 +++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/examples/local_models/smart_scraper_ollama.py b/examples/local_models/smart_scraper_ollama.py index b161cd0f..d5585ff7 100644 --- a/examples/local_models/smart_scraper_ollama.py +++ b/examples/local_models/smart_scraper_ollama.py @@ -14,7 +14,6 @@ graph_config = { "format": "json", # Ollama needs the format to be specified explicitly # "base_url": "http://localhost:11434", # set ollama URL arbitrarily }, - "verbose": True, "headless": False } diff --git a/scrapegraphai/graphs/abstract_graph.py b/scrapegraphai/graphs/abstract_graph.py index 83b532bc..2ccf14b2 100644 --- a/scrapegraphai/graphs/abstract_graph.py +++ b/scrapegraphai/graphs/abstract_graph.py @@ -53,6 +53,9 @@ class AbstractGraph(ABC): def __init__(self, prompt: str, config: dict, source: Optional[str] = None, schema: Optional[BaseModel] = None): + if config.get("llm").get("temperature") is None: + config["llm"]["temperature"] = 0 + self.prompt = prompt self.source = source self.config = config @@ -212,7 +215,7 @@ class AbstractGraph(ABC): print("model not found, using default token size (8192)") self.model_token = 8192 return ErnieBotChat(llm_params) - + if "oneapi" in llm_params["model"]: # take the model after the last dash llm_params["model"] = llm_params["model"].split("/")[-1] @@ -221,7 +224,7 @@ class AbstractGraph(ABC): except KeyError as exc: raise KeyError("Model not supported") from exc return OneApi(llm_params) - + if "nvidia" in llm_params["model"]: try: self.model_token = models_tokens["nvidia"][llm_params["model"].split("/")[-1]]