From fc15d8c03aaf391823e86a959ea4fa62b6504e47 Mon Sep 17 00:00:00 2001 From: Zac Date: Sat, 4 Nov 2023 09:31:24 -0400 Subject: [PATCH] Azure support multitask.py (#136) Azure support for Multiple Extractions --- instructor/dsl/multitask.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/instructor/dsl/multitask.py b/instructor/dsl/multitask.py index c43101f..47c81be 100644 --- a/instructor/dsl/multitask.py +++ b/instructor/dsl/multitask.py @@ -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):