(self)
| 24 | self.init_flag = False |
| 25 | |
| 26 | def init_model(self): |
| 27 | llm = self.create_openai_model(self.openai_api_key, self.model_name) |
| 28 | prompt = self.create_prompt() |
| 29 | self.chain = LLMChain(llm=llm, prompt=prompt) |
| 30 | self.init_flag = True |
| 31 | |
| 32 | def __call__(self, inputs: Dict[str, str]) -> Dict[str, str]: |
| 33 | if not self.init_flag: |
no test coverage detected