Fri Apr 18 04:33:00 CST 2025 inscode

main.py

-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)
-    
-