mirror of
https://github.com/VinciGit00/Scrapegraph-ai.git
synced 2026-06-25 21:11:11 +08:00
24 lines
523 B
Python
24 lines
523 B
Python
import os
|
|
from dotenv import load_dotenv
|
|
from yosoai.getter import remover
|
|
from yosoai.class_generator import Generator
|
|
|
|
load_dotenv()
|
|
|
|
MY_ENV_VAR = os.getenv('API_KEY')
|
|
|
|
values = [
|
|
{
|
|
"title": "title_website",
|
|
"type": "str",
|
|
"description": "Find the main header title of the website"
|
|
}
|
|
]
|
|
|
|
if __name__ == "__main__":
|
|
|
|
generator_instance = Generator(values, MY_ENV_VAR, 0, "gpt-3.5-turbo")
|
|
|
|
res = generator_instance.invocation(remover("https://www.mockupworld.co"))
|
|
|
|
print(res) |