Azure support multitask.py (#136)

Azure support for Multiple Extractions
2026-06-05 22:50:18 +00:00 · 2023-11-04 09:31:24 -04:00
parent 337fe775b2
commit fc15d8c03a
1 changed files with 5 additions and 3 deletions
@@ -31,9 +31,11 @@ class MultiTaskBase:
    @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):