diff --git a/ai/utils/show_process.py b/ai/utils/show_process.py index 7fdf8bf..d60ee1f 100644 --- a/ai/utils/show_process.py +++ b/ai/utils/show_process.py @@ -41,7 +41,7 @@ def show_main_process(zz_code=None, day_delta=3): print('message:', message) print('tokens:', tokens) # 获取排片数据 - result_obj = json.loads(result) + result_obj = result # 预测排片数据 _show = result_obj['show'].replace('```', '').replace('csv', '').strip() # 预测销售数据 diff --git a/ai/utils/show_prompt.py b/ai/utils/show_prompt.py index 70a2579..025b2a8 100644 --- a/ai/utils/show_prompt.py +++ b/ai/utils/show_prompt.py @@ -144,4 +144,4 @@ class ShowAI: ) # print(dict(response)) # print(response.choices[0].message.content) - return response.choices[0].message.content, response.model_dump_json(), response.usage.model_dump_json() + return response.choices[0].message.content.model_dump_json(), response.model_dump_json(), response.usage.model_dump_json()