From c22324d1abe0f7d31f64ed5ae76930c67146eecc Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Tue, 8 Aug 2023 11:41:11 -0400
Subject: [PATCH] both call and generate

Former-commit-id: 86e3476568eaa4390ecb84cddd7461001ea75317
---
 swarms/agents/models/openai.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/swarms/agents/models/openai.py b/swarms/agents/models/openai.py
index 71dfd126..6958cd1a 100644
--- a/swarms/agents/models/openai.py
+++ b/swarms/agents/models/openai.py
@@ -17,6 +17,12 @@ class OpenAI:
             params=params, 
             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):
         try:
             return self.ai(message, **kwargs)
@@ -30,6 +36,6 @@ class OpenAI:
 #usage
 #from swarms import OpenAI()
 #chat = OpenAI()
-#response = chat("Hello World")
+#response = chat.generate("Hello World")
 #print(response)