From 5c25da2fe64b4b64a00f1879f3d5dcfbf1512848 Mon Sep 17 00:00:00 2001 From: Marco Vinciguerra Date: Fri, 13 Sep 2024 14:20:23 +0200 Subject: [PATCH] feat: integration of o1 --- examples/openai/smart_scraper_openai.py | 2 +- scrapegraphai/helpers/models_tokens.py | 7 +++++-- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/examples/openai/smart_scraper_openai.py b/examples/openai/smart_scraper_openai.py index 2962f51b..4d9d6330 100644 --- a/examples/openai/smart_scraper_openai.py +++ b/examples/openai/smart_scraper_openai.py @@ -18,7 +18,7 @@ load_dotenv() graph_config = { "llm": { "api_key": os.getenv("OPENAI_API_KEY"), - "model": "openai/gpt-4o", + "model": "openai/o1-preview", }, "verbose": True, "headless": False, diff --git a/scrapegraphai/helpers/models_tokens.py b/scrapegraphai/helpers/models_tokens.py index ea35ed71..13cc82b2 100644 --- a/scrapegraphai/helpers/models_tokens.py +++ b/scrapegraphai/helpers/models_tokens.py @@ -23,7 +23,8 @@ models_tokens = { "gpt-4o-2024-08-06": 128000, "gpt-4o-2024-05-13": 128000, "gpt-4o-mini":128000, - + "o1-preview":128000, + "o1-mini":128000 }, "azure_openai": { "gpt-3.5-turbo-0125": 16385, @@ -43,7 +44,9 @@ models_tokens = { "gpt-4-32k-0613": 32768, "gpt-4o": 128000, "gpt-4o-mini":128000, - "chatgpt-4o-latest": 128000 + "chatgpt-4o-latest": 128000, + "o1-preview":128000, + "o1-mini":128000 }, "google_genai": { "gemini-pro": 128000,