update message assignment (#80)

Co-authored-by: Neil Neuwirth <neil.neuwirth@blackstone.com>
This commit is contained in:
Neil Neuwirth
2023-08-20 02:40:35 -04:00
committed by GitHub
parent dd81bf60e9
commit 1ebb7b95c1
2 changed files with 3 additions and 3 deletions
@@ -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(
{
+2 -2
View File
@@ -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"