CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-07-04 21:00:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: update gemini model to "gemini-pro"

Browse Source
This commit is contained in:
Evan Lin 2024-08-09 23:15:01 +08:00
parent 78593e303a
commit a7264cebd2
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
examples/gemini/pdf_scraper_graph_gemini.py
View File

@ -6,6 +6,7 @@ import os, json
from dotenv import load_dotenv from dotenv import load_dotenv
from scrapegraphai.utils import prettify_exec_info from scrapegraphai.utils import prettify_exec_info
from scrapegraphai.graphs import PDFScraperGraph from scrapegraphai.graphs import PDFScraperGraph
load_dotenv() load_dotenv()
@ -18,7 +19,7 @@ gemini_key = os.getenv("GOOGLE_APIKEY")
graph_config = { graph_config = {
"llm": { "llm": {
"api_key": gemini_key, "api_key": gemini_key,
"model": "gemini-pr", "model": "gemini-pro",
}, },
} }