from models.base import Session
from models.prompt import Prompt


prompts = [
    (
        "setup_block",
        """
Act as a data analyst, your task is to perform data analysis using the provided information.
"""
    ),
    (
        "dataset_blocking",
        """
Information about data sources(DataBase Schema): \n
%(db_schema_str)s
"""
    ),
    (
        "main_task_block",
        """
---
%(main_task)s
---
"""
    ),
    (
        "output_json_format_block",
        """
Output should be json schema as follow %(schema)s.
Only output json, do not output anything else. Json:
"""
    ),
    (
        "domain_knowledge_block",
        """
The relevant domain knowledge:
%(domain_knowledge_embedding)s
"""
    ),
    (
        "session_main_task_block",
        """
MainTask:
%(main_task)s
ChoosedAction:
%(choosed_action)s
SolutionPlan:
%(solution_plan_str)s
"""
    ),
    (
        "chat_logs_block",
        """
---
The relevant chat logs:
%(chat_logs_str)s
---
"""
    ),
]


def init_prompts():
    s = Session()

    for i in prompts:
        name, content = i

        Prompt.create(s, name, content)

    s.commit()


if __name__ == "__main__":
    init_prompts()