diff --git a/main.py b/main.py index 8e6ee78cf56e245b96740600c8a2bbc281224d68..7c156c5667f28d653e9ebd88ddae3449c6c97073 100644 --- a/main.py +++ b/main.py @@ -1,20 +1,16 @@ -import requests +import requests,os import json -import os from flask import Flask, request, jsonify app = Flask(__name__) # 请替换为你的 DeepSeek API 地址和 API Key -API_URL = "https://api.deepseek.com/v1" +API_URL = "https://api.deepseek.com" API_KEY = os.getenv('DeepSeek_API_KEY') def call_deepseek(prompt, max_new_tokens=150): """ - 调用 DeepSeek API 生成文本Error: name 'os' is not defined - - - + 调用 DeepSeek API 生成文本 :param prompt: 输入的提示文本 :param max_new_tokens: 生成的最大新 token 数 :return: 生成的文本