indentation

pull/43/head
Kye 1 year ago
parent d53a5925c1
commit c4da2359c2

@ -10,7 +10,12 @@ class OpenAI:
save_messages=True): save_messages=True):
self.api_key = api_key or self._fetch_api_key() self.api_key = api_key or self._fetch_api_key()
self.system = system or "You are a helpful assistant" self.system = system or "You are a helpful assistant"
self.ai = AIChat(api_key=self.api_key, system=self.system, console=console, model=model, params=params, save_messages=save_messages) self.ai = AIChat(api_key=self.api_key,
system=self.system,
console=console,
model=model,
params=params,
save_messages=save_messages)
def generate(self, message, **kwargs): def generate(self, message, **kwargs):
try: try:
@ -26,4 +31,5 @@ class OpenAI:
#from swarms import OpenAI() #from swarms import OpenAI()
#chat = OpenAI() #chat = OpenAI()
#response = chat("Hello World") #response = chat("Hello World")
#print(response) #print(response)

Loading…
Cancel
Save