examples/ScrapeGraphAI-experiments
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新模型配置

Browse Source
This commit is contained in:
严浩
2025-01-14 14:06:45 +08:00
parent 13590cda71
commit 653bd96d35
1 changed files with 17 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
app-ollama.py
View File

@ -2,14 +2,17 @@ from scrapegraphai.graphs import SmartScraperGraph
graph_config = {
"llm": {
# "model": "ollama/mistral-nemo:latest",
# "model": "ollama/mistral:latest",
"model": "ollama/mistral-nemo:12b",
"model_tokens": 1024000,
# "model": "ollama/llama3.2:latest",
# "model": "ollama/mistral:7b",
# "model_tokens": 32768,
# "model": "ollama/llama3.2:3b",
# "model_tokens": 131072, # ollama show llama3.2:latest
"model": "ollama/qwen2:7b",
"model_tokens": 32768, # ollama show qwen2:7b
# "model": "ollama/qwen2:7b",
# "model_tokens": 32768, # ollama show qwen2:7b
"temperature": 0, # 更准确执行任务
"format": "json", # Ollama 需要显式指定格式
@ -21,11 +24,16 @@ graph_config = {
# "headless": False,
}
# 抓取网页标题
# smart_scraper_graph = SmartScraperGraph(
# # prompt="List all article titles on the page",
# prompt="列出页面上的所有文章标题",
# source="https://www.aivi.fyi/",
# config=graph_config
# )
smart_scraper_graph = SmartScraperGraph(
# prompt="List all article titles on the page",
prompt="列出页面上的所有文章标题",
source="https://www.aivi.fyi/",
prompt="List all Stations on the page. the data is in tbody",
source="https://network.satnogs.org/stations/",
config=graph_config
)