From b5a4c95cbbd11dbf4cc477ea3546a35224e62715 Mon Sep 17 00:00:00 2001 From: practicer2015 Date: Fri, 18 Apr 2025 04:33:00 +0800 Subject: [PATCH] Fri Apr 18 04:33:00 CST 2025 inscode --- main.py | 74 --------------------------------------------------------- 1 file changed, 74 deletions(-) diff --git a/main.py b/main.py index 2a7be1d..e69de29 100644 --- a/main.py +++ b/main.py @@ -1,74 +0,0 @@ -import requests,os -import json -from flask import Flask, request, jsonify - -app = Flask(__name__) - -# 请替换为你的 DeepSeek API 地址和 API Key -API_URL = "https://api.deepseek.com" -API_KEY = os.getenv('DeepSeek_API_KEY') - -def call_deepseek(prompt, model="deepseek-chat", max_tokens=2048, temperature=0.7): - """ - 调用 DeepSeek API 的函数 - - 参数: - - prompt: 输入的提示文本 - - model: 使用的模型名称(默认为 deepseek-chat) - - max_tokens: 生成的最大token数 - - temperature: 控制生成随机性的参数(0-1) - - 返回: - - API的响应内容 - """ - # API端点(请根据实际API文档替换) - api_url = "https://api.deepseek.com/v1/chat/completions" - - # 你的API密钥(请替换为实际的密钥) - api_key = "sk-b36c4122bb874a3f83f250d2e9c8a705" - - # 请求头 - headers = { - "Content-Type": "application/json", - "Authorization": f"Bearer {api_key}" - } - - # 请求体 - payload = { - "model": model, - "messages": [{"role": "user", "content": prompt}], - "max_tokens": max_tokens, - "temperature": temperature - } - - try: - # 发送POST请求 - response = requests.post(api_url, headers=headers, json=payload) - response.raise_for_status() # 检查是否有错误 - - # 返回API响应 - return response.json() - - except requests.exceptions.RequestException as e: - print(f"请求出错: {e}") - return None - -@app.route('/generate', methods=['POST']) -def generate_text(): - data = request.get_json() - prompt = data.get('prompt') - max_new_tokens = data.get('max_new_tokens', 150) - - if not prompt: - return jsonify({"error": "Prompt is required"}), 400 - - response = call_deepseek(prompt, max_new_tokens) - if response: - return jsonify({"generated_text": response}) - else: - return jsonify({"error": "Failed to generate text"}), 500 - -if __name__ == "__main__": - app.run(debug=True) - - -- GitLab