demo

app-ollama.py

@@ -0,0 +1,28 @@
+from scrapegraphai.graphs import SmartScraperGraph
+
+graph_config = {
+    "llm": {
+        # "model": "ollama/mistral-nemo:latest",
+        "model": "ollama/mistral:latest",
+        "temperature": 0,  # 更准确执行任务
+        "format": "json",  # Ollama 需要显式指定格式
+    },
+    "embeddings": {
+        "model": "ollama/nomic-embed-text",
+
+    },
+    "verbose": True,
+    # "headless": False,
+}
+
+# 抓取网页标题
+smart_scraper_graph = SmartScraperGraph(
+    prompt="List all article titles on the page",
+    # prompt="列出页面上的所有文章标题",
+    source="https://www.aivi.fyi/",
+    config=graph_config
+)
+
+# Run the pipeline
+result = smart_scraper_graph.run()
+print(result)