import yaml
from factool.utils.openai_wrapper import OpenAIChat
import os
import pathlib

class pipeline():
    def __init__(self, domain, foundation_model):
        if foundation_model == 'gpt-3.5-turbo' or foundation_model == 'gpt-4':
            self.company = 'openai'
            self.chat = OpenAIChat(model_name=foundation_model)

        self.prompts_path = os.path.join(os.path.dirname(pathlib.Path(__file__)), "../prompts/")
        
        with open(os.path.join(self.prompts_path, "self_check.yaml"), 'r') as file:
            data = yaml.load(file, Loader=yaml.FullLoader)
        self.self_check_prompt = data[domain]