clean up

Former-commit-id: 091e2918f4e623c807fcf0b531547b3b0e225288
2 years ago · c8ffafdb67
parent a25a9f804c
commit c8ffafdb67
1 changed files with 1 additions and 1 deletions
--- a/swarms/swarms.py
+++ b/swarms/swarms.py
@ -80,7 +80,7 @@ class HierarchicalSwarm:
        """
        try: 
            # Initialize language model
-            self.llm = OpenAI(openai_api_key=self.openai_api_key, temperature=self.temperature)
+            self.llm = llm_class(openai_api_key=self.openai_api_key, temperature=self.temperature)
        except Exception as e:
            logging.error(f"Failed to initialize language model: {e}")