Scrapegraph-ai/examples/oneapi/smart_scraper_multi_oneapi.py
2024-05-28 09:11:14 +02:00

37 lines
980 B
Python

"""
Basic example of scraping pipeline using SmartScraper
"""
import json
from dotenv import load_dotenv
from scrapegraphai.graphs import SmartScraperMultiGraph
# ************************************************
# Define the configuration for the graph
# ************************************************
graph_config = {
"llm": {
"api_key": "***************************",
"model": "oneapi/qwen-turbo",
"base_url": "http://127.0.0.1:3000/v1", # 设置 OneAPI URL
}
}
# *******************************************************
# Create the SmartScraperMultiGraph instance and run it
# *******************************************************
multiple_search_graph = SmartScraperMultiGraph(
prompt="Who is Marco Perini?",
source= [
"https://perinim.github.io/",
"https://perinim.github.io/cv/"
],
schema=None,
config=graph_config
)
result = multiple_search_graph.run()
print(json.dumps(result, indent=4))