在 Python 项目中实现 Ollama API 的鉴权功能，可以通过多种方式来完成，包括使用中间层服务、添加请求头验证、使用 API 网关或反向代理等。由于 Ollama 本身不提供内置的身份验证机制，因此需要在应用层面或基础设施层面对请求进行控制。 ### 使用请求头验证 API 密钥 一种简单但有效的鉴权方式是在客户端请求时添加自定义请求头（如 `Authorization: Bearer <token>`），并在反向代理或中间服务中验证该请求头是否合法。 以下是一个使用 Python 的 `requests` 库调用 Ollama API 并携带 API 密钥的示例： ```python import requests OLLAMA_URL = "http://localhost:11434/api/generate" API_KEY = "your-secret-api-key" headers = { "Authorization": f"Bearer {API_KEY}", "Content-Type": "application/json" } data = { "model": "llama2", "prompt": "Hello, how can I assist you today?" } response = requests.post(OLLAMA_URL, json=data, headers=headers) print(response.json()) ``` 在服务器端，可以通过反向代理（如 Nginx）或自定义中间服务来验证请求头中的 API 密钥。例如，使用 Flask 构建一个简单的中间层服务来拦截请求并验证 API 密钥： ```python from flask import Flask, request, jsonify import requests app = Flask(__name__) VALID_API_KEY = "your-secret-api-key" @app.route('/generate', methods=['POST']) def proxy(): auth_header = request.headers.get("Authorization") if not auth_header or not auth_header.startswith(f"Bearer {VALID_API_KEY}"): return jsonify({"error": "Unauthorized"}), 401 data = request.json ollama_response = requests.post("http://localhost:11434/api/generate", json=data) return jsonify(ollama_response.json()) if __name__ == '__main__': app.run(host='0.0.0.0', port=5000) ``` ### 使用 Nginx 配置 Basic Auth 若希望在基础设施层面实现鉴权，可以使用 Nginx 配置 Basic Auth 来保护 Ollama 服务。这种方式适合部署在生产环境中，且无需修改应用代码。 Nginx 配置示例如下： ```nginx server { listen 80; server_name yourdomain.com; location / { auth_basic "Restricted Access"; auth_basic_user_file /etc/nginx/.htpasswd; proxy_pass http://localhost:11434; proxy_set_header Host $host; proxy_set_header X-Real-IP $remote_addr; proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for; proxy_set_header X-Forwarded-Proto $scheme; } } ``` 在客户端调用时，需要在请求头中包含 Basic Auth 信息： ```python import requests from requests.auth import HTTPBasicAuth response = requests.post( "http://yourdomain.com/api/generate", json={"model": "llama2", "prompt": "What is AI?"}, auth=HTTPBasicAuth('username', 'password') ) print(response.json()) ``` ### 使用 API 网关或 JWT 鉴权 对于更复杂的场景，可以集成 JWT（JSON Web Token）进行身份验证。客户端在登录后获取一个 token，后续所有请求都需携带该 token。服务端或网关验证 token 的有效性后才转发请求至 Ollama。 若使用 Flask 实现 JWT 鉴权的中间层服务，可以参考如下代码片段： ```python from flask import Flask, request, jsonify import jwt import datetime from functools import wraps app = Flask(__name__) SECRET_KEY = "your-secret-key" def token_required(f): @wraps(f) def decorated(*args, **kwargs): token = request.headers.get('Authorization') if not token: return jsonify({"message": "Token is missing!"}), 403 try: data = jwt.decode(token, SECRET_KEY, algorithms=["HS256"]) except: return jsonify({"message": "Token is invalid!"}), 403 return f(*args, **kwargs) return decorated @app.route('/login', methods=['POST']) def login(): token = jwt.encode({'user': 'test', 'exp': datetime.datetime.utcnow() + datetime.timedelta(minutes=30)}, SECRET_KEY) return jsonify({'token': token}) @app.route('/generate', methods=['POST']) @token_required def proxy(): data = request.json response = requests.post("http://localhost:11434/api/generate", json=data) return jsonify(response.json()) if __name__ == '__main__': app.run(debug=True) ``` 客户端调用流程如下： 1. 先调用 `/login` 获取 token。 2. 在后续请求中携带 token 到 `Authorization` 请求头。 3. 调用 `/generate` 接口时，将由中间服务验证 token 合法性后转发请求。 ### 总结 在 Python 项目中实现 Ollama API 的鉴权功能，可以通过以下方式实现： - 使用 API 密钥验证，通过请求头传递密钥。 - 使用 Nginx 配置 Basic Auth 进行基础身份验证。 - 使用 Flask 构建中间层服务，实现更复杂的鉴权逻辑，如 JWT。 - 通过 API 网关或反向代理统一管理身份验证逻辑。 这些方法均不依赖 Spring 框架，适用于 Python 项目中的实际部署需求。