_utils/langchain_utils/Chain_class.py

class Chain:
    def __init__(self, prompt, model):
        self.prompt = prompt
        self.model = model

    def create_prompt_model_chain(self):
        return self.prompt | self.model

    def invoke_prompt_model_chain(self, invoke_params):
        chain = self.create_prompt_model_chain()
        return chain.invoke(invoke_params)