diff --git a/app-ollama.py b/app-ollama.py index d06c7ce..fe8d14f 100644 --- a/app-ollama.py +++ b/app-ollama.py @@ -3,13 +3,19 @@ from scrapegraphai.graphs import SmartScraperGraph graph_config = { "llm": { # "model": "ollama/mistral-nemo:latest", - "model": "ollama/mistral:latest", + # "model": "ollama/mistral:latest", + + # "model": "ollama/llama3.2:latest", + # "model_tokens": 131072, # ollama show llama3.2:latest + + "model": "ollama/qwen2:7b", + "model_tokens": 32768, # ollama show qwen2:7b + "temperature": 0, # 更准确执行任务 "format": "json", # Ollama 需要显式指定格式 }, "embeddings": { "model": "ollama/nomic-embed-text", - }, "verbose": True, # "headless": False, @@ -17,8 +23,8 @@ graph_config = { # 抓取网页标题 smart_scraper_graph = SmartScraperGraph( - prompt="List all article titles on the page", - # prompt="列出页面上的所有文章标题", + # prompt="List all article titles on the page", + prompt="列出页面上的所有文章标题", source="https://www.aivi.fyi/", config=graph_config )