fixes

AbanteAI · Mar 8, 2024 · 4ae3272 · 4ae3272
1 parent a53ceff
commit 4ae3272
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/spice.py b/spice.py
@@ -4,6 +4,7 @@
 from dotenv import load_dotenv
 from openai import OpenAI
 
+load_dotenv()
 
 
 class SpiceClient:
@@ -17,7 +18,6 @@ def __init__(self, model):
 
         self.model = model
 
-        load_dotenv()
         if self._provider == "openai":
             self._client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
         elif self._provider == "anthropic":
@@ -64,5 +64,6 @@ def _stream_generator(self, stream):
                     content = chunk.delta.text
             else:
                 content = chunk.choices[0].delta.content
-            if content is not None:
-                yield content
+                if content is None:
+                    content = ""
+            yield content