From a7264cebd28857b4a13e7db2f27e80e5b57e4407 Mon Sep 17 00:00:00 2001 From: Evan Lin Date: Fri, 9 Aug 2024 23:15:01 +0800 Subject: [PATCH] chore: update gemini model to "gemini-pro" --- examples/gemini/pdf_scraper_graph_gemini.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/gemini/pdf_scraper_graph_gemini.py b/examples/gemini/pdf_scraper_graph_gemini.py index 55ce5958..d4b7342a 100644 --- a/examples/gemini/pdf_scraper_graph_gemini.py +++ b/examples/gemini/pdf_scraper_graph_gemini.py @@ -6,6 +6,7 @@ import os, json from dotenv import load_dotenv from scrapegraphai.utils import prettify_exec_info from scrapegraphai.graphs import PDFScraperGraph + load_dotenv() @@ -18,7 +19,7 @@ gemini_key = os.getenv("GOOGLE_APIKEY") graph_config = { "llm": { "api_key": gemini_key, - "model": "gemini-pr", + "model": "gemini-pro", }, }