File size: 632 Bytes
d195d4f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
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] |