Updates chat class

.gitignore

@@ -1,3 +1,2 @@
 __pycache__/
-data/
-DS_Store
+.DS_Store

aimakerspace/openai_utils/chatmodel.py

@@ -1,4 +1,4 @@
-from openai import OpenAI
+from openai import OpenAI, AsyncOpenAI
 from dotenv import load_dotenv
 import os
 
@@ -25,3 +25,23 @@ class ChatOpenAI:
             return response.choices[0].message.content
 
         return response
+    
+    async def astream(self, messages, **kwargs):
+        if not isinstance(messages, list):
+            raise ValueError("messages must be a list")
+        
+        client = AsyncOpenAI()
+
+        stream = await client.chat.completions.create(
+            model=self.model_name,
+            messages=messages,
+            stream=True,
+            **kwargs
+        )
+
+        async for chunk in stream:
+            content = chunk.choices[0].delta.content
+            if content is not None:
+                yield content
+
+