(self, agent_config, model)
| 27 | self.sum_agent = Agent(config.sum_agent, self.models[config.sum_agent.model.class_name]) |
| 28 | |
| 29 | def add_agent(self, agent_config, model): |
| 30 | module = importlib.import_module(agent_config.agent.module_name) |
| 31 | agent_class = getattr(module, agent_config.agent.class_name) |
| 32 | agent:Agent = agent_class(agent_config, model) |
| 33 | self.agents.append(agent) |
| 34 | |
| 35 | def predict(self, question, texts, images): |
| 36 | '''Implement the method in the subclass''' |