各位大佬怎么糊接口

整了一个别的站的api,但是格式和open ai不同,没办法接入one api,目前本人会python

问问gpt,糊一个

给个调用格式看看,纯 api 接口转换不比 chat2api 容易多了

参考
https://linux.do/t/topic/92524

大概会是这种样子,可以参考着修改一下

from flask import Flask, request, jsonify
import time

app = Flask(__name__)

@app.route('/chat/completions', methods=['POST'])
def my_endpoint():
    url = "https://api.anakin.ai/v1/apps/1344/chatbot/chat"
    headers = {
        'Content-Type': 'application/json',
        'Authorization': 'Bearer ANAKINAI_API_ACCESS_TOKEN'
    }
    data = {
        "content": request.json["messages"][-1]["content"]
    }
    response = request.post(url, headers=headers, json=data)
    if response.status_code == 200:
        response_transformed = {
            "id": f"chatcmpl-{response.json['threadId']}",
            "object": "chat.completion",
            "created": int(time.time()),
            "model": request.json["model"],
            "system_fingerprint": "fp_44709d6fcb",
            "choices": [{
                "index": 0,
                "message": {
                "role": "assistant",
                "content": response.json['content'],
                },
                "logprobs": None,
                "finish_reason": "stop"
            }],
            "usage": {
                "prompt_tokens": len(data["content"]),
                "completion_tokens": len(response.json["content"]),
                "total_tokens":  len(data["content"])+ len(response.json["content"]),
            }
        }
        return jsonify(response_transformed)
    return jsonify({"error": {"message": "An error occurred", "type": "server_error", "code": 500}})

if __name__ == '__main__':
    app.run(port=5000)  # 将服务绑定到5000端口
2 个赞