20 lines
649 B
Python
20 lines
649 B
Python
import json
|
|
|
|
from openai.types.chat import ChatCompletion
|
|
|
|
|
|
def func_talk(response: ChatCompletion):
|
|
function_call = response.choices[0].message.function_call
|
|
if function_call:
|
|
function_name = function_call.name
|
|
if function_name == 'talk':
|
|
function_arguments = function_call.arguments
|
|
try:
|
|
j = json.loads(function_arguments)
|
|
return j.get('message')
|
|
except json.decoder.JSONDecodeError:
|
|
# Sometimes the AI doesn't do JSON.
|
|
return function_arguments
|
|
else:
|
|
print('THE AI DID NOT CALL A FUNCTION IN TALK:', response)
|