26 lines
1.0 KiB
Python
26 lines
1.0 KiB
Python
import time
|
||
import logging
|
||
from server.chat import utils
|
||
from fastapi.responses import JSONResponse
|
||
|
||
def stop(message_id: str):
|
||
try:
|
||
# start_time = time.time()
|
||
# tatus = None
|
||
# # 重试十秒
|
||
# while time.time() - start_time < 10:
|
||
status = utils.get_shared_variable(message_id)
|
||
# if status is not None:
|
||
# break
|
||
# logging.info(f"status未赋值, messageId: {message_id}")
|
||
# time.sleep(0.5) # 等待0.5秒后重试
|
||
|
||
if status is None:
|
||
return JSONResponse(content={"error": "未找到messegeId", "messageId": message_id}, status_code=500)
|
||
|
||
status["status"] = False
|
||
utils.set_shared_variable(message_id, status)
|
||
logging.info(f"停止请求成功，messageId: {message_id}")
|
||
return JSONResponse(content={"event": "请求已成功停止", "messageId": message_id})
|
||
except Exception as e:
|
||
return JSONResponse(content={"error": str(e),"messageId": message_id}, status_code=500) |