mirror of
https://github.com/kennethreitz/instructor.git
synced 2026-06-05 22:50:18 +00:00
update message assignment (#80)
Co-authored-by: Neil Neuwirth <neil.neuwirth@blackstone.com>
This commit is contained in:
@@ -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(
|
||||
{
|
||||
|
||||
@@ -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"
|
||||
|
||||
Reference in New Issue
Block a user