From 50b1edfca400875252c56385b2377184de7edf4a Mon Sep 17 00:00:00 2001 From: Kye Date: Sat, 29 Jul 2023 11:25:42 -0400 Subject: [PATCH] clean up --- swarms/swarms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/swarms/swarms.py b/swarms/swarms.py index a746459b..15da612a 100644 --- 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}")