diff --git a/swarms/boss/boss_node.py b/swarms/boss/boss_node.py index 3bfcf9b5..8cf9d144 100644 --- a/swarms/boss/boss_node.py +++ b/swarms/boss/boss_node.py @@ -129,7 +129,7 @@ class BossNode: if not self.api_key: raise ValueError("[BossNode][ValueError][API KEY must be provided either as an argument or as an environment variable API_KEY]") - self.llm = llm_class + self.llm = self.initialize_llm(self.llm_class) todo_prompt = PromptTemplate.from_template(boss_system_prompt) todo_chain = LLMChain(llm=self.llm, prompt=todo_prompt)