back

Former-commit-id: a62069d394
2 years ago · 5bb49c062d
parent 6523dd2b03
commit 5bb49c062d
1 changed files with 1 additions and 1 deletions
--- 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)