Skip to content

Commit fc0a148

Browse files
committed
feat: add intrgration for o3min
1 parent 8121c7b commit fc0a148

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

scrapegraphai/graphs/abstract_graph.py

+1-4
Original file line numberDiff line numberDiff line change
@@ -53,9 +53,6 @@ def __init__(
5353
source: Optional[str] = None,
5454
schema: Optional[Type[BaseModel]] = None,
5555
):
56-
if config.get("llm").get("temperature") is None:
57-
config["llm"]["temperature"] = 0
58-
5956
self.prompt = prompt
6057
self.source = source
6158
self.config = config
@@ -125,7 +122,7 @@ def _create_llm(self, llm_config: dict) -> object:
125122
KeyError: If the model is not supported.
126123
"""
127124

128-
llm_defaults = {"temperature": 0, "streaming": False}
125+
llm_defaults = {"streaming": False}
129126
llm_params = {**llm_defaults, **llm_config}
130127
rate_limit_params = llm_params.pop("rate_limit", {})
131128

scrapegraphai/helpers/models_tokens.py

+1
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
"gpt-4.5-preview": 128000,
2828
"o1-preview": 128000,
2929
"o1-mini": 128000,
30+
"o3-mini": 200000,
3031
},
3132
"azure_openai": {
3233
"gpt-3.5-turbo-0125": 16385,

0 commit comments

Comments
 (0)