Scrapegraph-ai/utils/class_creator.py
2024-01-30 10:36:02 +01:00

15 lines
478 B
Python

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