Scrapegraph-ai/amazscraper/class_creator.py
2024-01-31 16:53:35 +01:00

17 lines
508 B
Python

import time
base_script = '''
from langchain_core.pydantic_v1 import BaseModel, Field
class _Response(BaseModel):
'''
def create_class(data_dict: dict):
'''
This function creates a class at runtime using the values from the list.
'''
for elem in data_dict:
global base_script
base_script = base_script + f" {elem['title']}: {elem['type']} = Field(description='{elem['description']}')\n"
with open("./utils/pydantic_class.py", "w") as f:
f.write(base_script)