Scrapegraph-ai/examples/extras/conditional_usage.py
Marco Vinciguerra dc0a138a7e run pre commit
2025-01-12 16:35:31 +01:00

40 lines
980 B
Python

"""
Basic example of scraping pipeline using SmartScraperMultiConcatGraph with Groq
"""
import json
import os
from dotenv import load_dotenv
from scrapegraphai.graphs import SmartScraperMultiGraph
load_dotenv()
# ************************************************
# Define the configuration for the graph
# ************************************************
graph_config = {
"llm": {
"api_key": os.getenv("OPENAI_API_KEY"),
"model": "openai/gpt-4o",
},
"verbose": True,
"headless": False,
}
# *******************************************************
# Create the SmartScraperMultiCondGraph 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))