CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-07-18 21:00:31 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: add intrgration for o3min

Browse Source
This commit is contained in:
Marco Vinciguerra 2025-03-13 14:36:05 +01:00
parent 8121c7b7f1
commit fc0a148017
2 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
scrapegraphai/graphs/abstract_graph.py
View File

@ -53,9 +53,6 @@ class AbstractGraph(ABC):
source: Optional[str] = None, source: Optional[str] = None,
schema: Optional[Type[BaseModel]] = None, schema: Optional[Type[BaseModel]] = None,
): ):
if config.get("llm").get("temperature") is None:
config["llm"]["temperature"] = 0
self.prompt = prompt self.prompt = prompt
self.source = source self.source = source
self.config = config self.config = config
@ -125,7 +122,7 @@ class AbstractGraph(ABC):
KeyError: If the model is not supported. KeyError: If the model is not supported.
""" """
llm_defaults = {"temperature": 0, "streaming": False} llm_defaults = {"streaming": False}
llm_params = {**llm_defaults, **llm_config} llm_params = {**llm_defaults, **llm_config}
rate_limit_params = llm_params.pop("rate_limit", {}) rate_limit_params = llm_params.pop("rate_limit", {})

1
scrapegraphai/helpers/models_tokens.py
View File

@ -27,6 +27,7 @@ models_tokens = {
"gpt-4.5-preview": 128000, "gpt-4.5-preview": 128000,
"o1-preview": 128000, "o1-preview": 128000,
"o1-mini": 128000, "o1-mini": 128000,
"o3-mini": 200000,
}, },
"azure_openai": { "azure_openai": {
"gpt-3.5-turbo-0125": 16385, "gpt-3.5-turbo-0125": 16385,