小白也能看懂的 大模型api 接口转发函数 转MD5简单版本

from datetime import datetime
导包缺啥  补啥这个不讲 主打一个简单明了 拿代码即可

def generate_sign():
    today = datetime.now().strftime(r"%Y%m%d")
    print("Today is:", today)  今天的日期

    sign_str = f"{today}*117"
    print("Sign string:", sign_str)  签名字符串
    md5 = hashlib.md5()
    md5.update(sign_str.encode('utf-8'))
    return md5.hexdigest()


# 假设今天是2024年5月11日，为了模拟这个情况，我们可以直接设置today的值

# 但是为了使用datetime.now()获取当前日期 系统日期应该设置为2024年5月11日


signature = generate_sign()
print("Generated MD5 signature:", signature)

class Query(BaseModel):

    data: str

    session_id: str = None

# 接口转发函数

@app.post("/aiForwarding")

async def ai_forwarding(item: ForwardItem):

    # 验证签名

    expected_sign = generate_sign()

    if item.sign != expected_sign:

        return {"code": "4003", "msg": "请求失败，无权访问", "data": ""}

   

    if item.urlName == "aiSscrmTalk":

        # 直接使用提供的Query数据

        return await handle_query(Query(**item.data))

    if item.urlName == "aiSscrmTalk2":

        return await handle_query2(Query(**item.data))

  

    else:

        return {"code": "4004", "msg": "无效的urlName", "data": ""}

假设大模型1（LLM） handle_query（）
 

假设大模型2  (LLM2) handle_query2 ()