add new groq example

This commit is contained in:
VinciGit00 2024-05-01 12:48:32 +02:00
parent cf038b33ea
commit 5ae67f5d3b
4 changed files with 54 additions and 1 deletions

View File

@ -1 +1,2 @@
GROQ_APIKEY= "your groq key"
GROQ_APIKEY= "your groq key"
OPENAI_APIKEY="your openai api key"

View File

@ -0,0 +1,52 @@
"""
Basic example of scraping pipeline using SmartScraper
"""
import os
from dotenv import load_dotenv
from scrapegraphai.graphs import SmartScraperGraph
from scrapegraphai.utils import prettify_exec_info
load_dotenv()
# ************************************************
# Define the configuration for the graph
# ************************************************
groq_key = os.getenv("GROQ_APIKEY")
openai_key = os.getenv("OPENAI_APIKEY")
graph_config = {
"llm": {
"model": "groq/gemma-7b-it",
"api_key": groq_key,
"temperature": 0
},
"embeddings": {
"api_key": openai_key,
"model": "gpt-3.5-turbo",
},
"headless": False
}
# ************************************************
# Create the SmartScraperGraph instance and run it
# ************************************************
smart_scraper_graph = SmartScraperGraph(
prompt="List me all the projects with their description.",
# also accepts a string with the already downloaded HTML code
source="https://perinim.github.io/projects/",
config=graph_config
)
result = smart_scraper_graph.run()
print(result)
# ************************************************
# Get graph execution info
# ************************************************
graph_exec_info = smart_scraper_graph.get_execution_info()
print(prettify_exec_info(graph_exec_info))