Azure support multitask.py (#136)

Azure support for Multiple Extractions
instructor-ai · Nov 4, 2023 · fc15d8c · fc15d8c
1 parent 337fe77
commit fc15d8c
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/instructor/dsl/multitask.py b/instructor/dsl/multitask.py
@@ -31,9 +31,11 @@ def tasks_from_chunks(cls, json_chunks):
     @staticmethod
     def extract_json(completion):
         for chunk in completion:
-            delta = chunk["choices"][0]["delta"]
-            if "function_call" in delta:
-                yield delta["function_call"]["arguments"]
+            if chunk["choices"]:
+                delta = chunk["choices"][0]["delta"]
+                if "function_call" in delta:
+                    if "arguments" in delta["function_call"]:
+                        yield delta["function_call"]["arguments"]
 
     @staticmethod
     def get_object(str, stack):