Skip to content

Commit 5c25da2

Browse files
committed
feat: integration of o1
1 parent 62912c2 commit 5c25da2

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

examples/openai/smart_scraper_openai.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
graph_config = {
1919
"llm": {
2020
"api_key": os.getenv("OPENAI_API_KEY"),
21-
"model": "openai/gpt-4o",
21+
"model": "openai/o1-preview",
2222
},
2323
"verbose": True,
2424
"headless": False,

scrapegraphai/helpers/models_tokens.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,8 @@
2323
"gpt-4o-2024-08-06": 128000,
2424
"gpt-4o-2024-05-13": 128000,
2525
"gpt-4o-mini":128000,
26-
26+
"o1-preview":128000,
27+
"o1-mini":128000
2728
},
2829
"azure_openai": {
2930
"gpt-3.5-turbo-0125": 16385,
@@ -43,7 +44,9 @@
4344
"gpt-4-32k-0613": 32768,
4445
"gpt-4o": 128000,
4546
"gpt-4o-mini":128000,
46-
"chatgpt-4o-latest": 128000
47+
"chatgpt-4o-latest": 128000,
48+
"o1-preview":128000,
49+
"o1-mini":128000
4750
},
4851
"google_genai": {
4952
"gemini-pro": 128000,

0 commit comments

Comments
 (0)