both call and generate

Former-commit-id: 86e3476568eaa4390ecb84cddd7461001ea75317
pull/160/head
Kye 2 years ago
parent 2855692106
commit c22324d1ab

@ -17,6 +17,12 @@ class OpenAI:
params=params, params=params,
save_messages=save_messages) save_messages=save_messages)
def __call__(self, message, **kwargs):
try:
return self.ai(message, **kwargs)
except Exception as error:
print(f"Error in OpenAI, {error}")
def generate(self, message, **kwargs): def generate(self, message, **kwargs):
try: try:
return self.ai(message, **kwargs) return self.ai(message, **kwargs)
@ -30,6 +36,6 @@ class OpenAI:
#usage #usage
#from swarms import OpenAI() #from swarms import OpenAI()
#chat = OpenAI() #chat = OpenAI()
#response = chat("Hello World") #response = chat.generate("Hello World")
#print(response) #print(response)

Loading…
Cancel
Save