From a68c53c55ddc6d741429deaf1952d214c1e87276 Mon Sep 17 00:00:00 2001 From: Kye Date: Wed, 5 Jul 2023 23:14:07 -0400 Subject: [PATCH] init llm clean up --- swarms/swarms.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/swarms/swarms.py b/swarms/swarms.py index d211daa6..7496036d 100644 --- a/swarms/swarms.py +++ b/swarms/swarms.py @@ -9,9 +9,9 @@ class Swarms: def __init__(self, openai_api_key): self.openai_api_key = openai_api_key - def initialize_llm(self, llm_class, temperature=0): + def initialize_llm(self, llm_class): # Initialize language model - return llm_class(temperature=temperature, openai_api_key=self.openai_api_key) + return llm_class(openai_api_key=self.openai_api_key) def initialize_tools(self, llm): # Initialize tools