diff --git a/scrapegraphai/graphs/abstract_graph.py b/scrapegraphai/graphs/abstract_graph.py index d9ed45a2..3908545d 100644 --- a/scrapegraphai/graphs/abstract_graph.py +++ b/scrapegraphai/graphs/abstract_graph.py @@ -53,9 +53,6 @@ class AbstractGraph(ABC): source: Optional[str] = None, schema: Optional[Type[BaseModel]] = None, ): - if config.get("llm").get("temperature") is None: - config["llm"]["temperature"] = 0 - self.prompt = prompt self.source = source self.config = config @@ -125,7 +122,7 @@ class AbstractGraph(ABC): KeyError: If the model is not supported. """ - llm_defaults = {"temperature": 0, "streaming": False} + llm_defaults = {"streaming": False} llm_params = {**llm_defaults, **llm_config} rate_limit_params = llm_params.pop("rate_limit", {}) diff --git a/scrapegraphai/helpers/models_tokens.py b/scrapegraphai/helpers/models_tokens.py index 828cefc0..eedf1cb9 100644 --- a/scrapegraphai/helpers/models_tokens.py +++ b/scrapegraphai/helpers/models_tokens.py @@ -27,6 +27,7 @@ models_tokens = { "gpt-4.5-preview": 128000, "o1-preview": 128000, "o1-mini": 128000, + "o3-mini": 200000, }, "azure_openai": { "gpt-3.5-turbo-0125": 16385,