diff --git a/ai/utils/show_prompt.py b/ai/utils/show_prompt.py index 025b2a8..356c130 100644 --- a/ai/utils/show_prompt.py +++ b/ai/utils/show_prompt.py @@ -144,4 +144,7 @@ class ShowAI: ) # print(dict(response)) # print(response.choices[0].message.content) - return response.choices[0].message.content.model_dump_json(), response.model_dump_json(), response.usage.model_dump_json() + res_json = response.model_dump_json() + res_obj = json.loads(res_json) + print(res_obj) + return res_obj['choices'][0]['.message']['content'], res_json, response.usage.model_dump_json()