(
self,
id: str | None = None,
role: str = None,
domain: str = "",
llm_name: str = "",
llm_config: KVCommConfig | None = None,
)
| 20 | @AgentRegistry.register('CopyMachine') |
| 21 | class CopyMachine(Node): |
| 22 | def __init__( |
| 23 | self, |
| 24 | id: str | None = None, |
| 25 | role: str = None, |
| 26 | domain: str = "", |
| 27 | llm_name: str = "", |
| 28 | llm_config: KVCommConfig | None = None, |
| 29 | ): |
| 30 | super().__init__(id, "CopyMachine" ,domain, llm_name) |
| 31 | prefix = "" |
| 32 | |
| 33 | self.llm = LLMRegistry.get(llm_name, prefix=prefix, llm_config=llm_config) |
| 34 | self.prompt_set = PromptSetRegistry.get(domain) |
| 35 | self.role = self.prompt_set.get_role() if role is None else role |
| 36 | self.llm.set_id(self.id, self.role) |
| 37 | self.constraint = self.prompt_set.get_analyze_constraint(self.role) |
| 38 | |
| 39 | async def _process_inputs( |
| 40 | self, |
nothing calls this directly
no test coverage detected