From 1ebb7b95c151511e1b32c39d1e43101ce9bcaada Mon Sep 17 00:00:00 2001 From: Neil Neuwirth Date: Sun, 20 Aug 2023 02:40:35 -0400 Subject: [PATCH] update message assignment (#80) Co-authored-by: Neil Neuwirth --- examples/query_planner_execution/query_planner_execution.py | 2 +- instructor/function_calls.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/query_planner_execution/query_planner_execution.py b/examples/query_planner_execution/query_planner_execution.py index 9a8eb50..b2c3d1a 100644 --- a/examples/query_planner_execution/query_planner_execution.py +++ b/examples/query_planner_execution/query_planner_execution.py @@ -138,7 +138,7 @@ def query_planner(question: str, plan=False) -> QueryPlan: max_tokens=1000, ) - messages.append(completion.choices[0].message) + messages.append(completion['choices'][0]['message']) messages.append( { diff --git a/instructor/function_calls.py b/instructor/function_calls.py index e15a498..50d9249 100644 --- a/instructor/function_calls.py +++ b/instructor/function_calls.py @@ -101,7 +101,7 @@ class openai_function: Returns: result (any): result of the function call """ - message = completion.choices[0].message + message = completion['choices'][0]['message'] if throw_error: assert "function_call" in message, "No function call detected" @@ -159,7 +159,7 @@ class OpenAISchema(BaseModel): Returns: cls (OpenAISchema): An instance of the class """ - message = completion.choices[0].message + message = completion['choices'][0]['message'] if throw_error: assert "function_call" in message, "No function call detected"