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,