AIServ/utils/request.py

import requests
import json
import time

# 基础配置 start ---------------------------------------->
# pre环境
URL = "https://apipre1.bjzxtw.org.cn/"
# master环境
# URL = "https://flzxw.bjzxtw.org.cn/"
# 服务器本地
# URL = "http://127.0.0.1:9501/"
# 基础配置 end ---------------------------------------->

def post_request(url, data, token):
    url = URL + url
    response = requests.post(url, json=data, headers={'Token': token})
    return response.json()

# 异常响应 start ---------------------------------------->
# 当php服务端出现异常的时候，或者python端出现异常的时候，汇报前端
def generate_error(message):
    for char in message:
        data = {
            'code': 0,
            'type': 'error',
            'data': char,
            'finished': False,
            'isComplete': False
        }
        yield f"data: {json.dumps(data, ensure_ascii=False)}\n\n"
        time.sleep(0.05)

    # 发送结束信号
    end_data = {
        'code': 200,
        'data': '',
        'finished': True,
        'full_text': "处理结果已发送完毕！"
    }
    yield f"data: {json.dumps(end_data, ensure_ascii=False)}\n\n"
# 异常响应 end ---------------------------------------->