shuangwhywhy commited on
Commit
6846f2e
·
verified ·
1 Parent(s): 98c5d3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -4,11 +4,13 @@ import time
4
 
5
  # **配置 OpenAI API Key**
6
  OPENAI_API_KEY = "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" # 👈 **替换这里!**
7
- openai.api_key = OPENAI_API_KEY # 绑定 API Key
 
 
8
 
9
  def generate_task_structure(prompt):
10
  """
11
- 生成任务管理计划
12
  """
13
 
14
  structured_prompt = f"""
@@ -35,7 +37,7 @@ def generate_task_structure(prompt):
35
  ]
36
 
37
  # **2️⃣ 调用 OpenAI API(流式输出)**
38
- response = openai.ChatCompletion.create(
39
  model="gpt-4o",
40
  messages=[
41
  {"role": "system", "content": "你是一个任务管理 AI 助手,帮助用户创建任务计划。"},
@@ -55,10 +57,9 @@ def generate_task_structure(prompt):
55
  # **4️⃣ 流式输出**
56
  full_text = ""
57
  for chunk in response:
58
- if "choices" in chunk and len(chunk["choices"]) > 0:
59
- text_chunk = chunk["choices"][0]["delta"].get("content", "")
60
- full_text += text_chunk
61
- yield full_text # **实时输出**
62
 
63
  yield "✅ 任务已完成!"
64
 
 
4
 
5
  # **配置 OpenAI API Key**
6
  OPENAI_API_KEY = "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" # 👈 **替换这里!**
7
+
8
+ # **使用新版本的 OpenAI Client**
9
+ client = openai.OpenAI(api_key=OPENAI_API_KEY)
10
 
11
  def generate_task_structure(prompt):
12
  """
13
+ 通过 OpenAI GPT-4o 生成任务管理计划,**支持进度条 & 流式输出**
14
  """
15
 
16
  structured_prompt = f"""
 
37
  ]
38
 
39
  # **2️⃣ 调用 OpenAI API(流式输出)**
40
+ response = client.chat.completions.create(
41
  model="gpt-4o",
42
  messages=[
43
  {"role": "system", "content": "你是一个任务管理 AI 助手,帮助用户创建任务计划。"},
 
57
  # **4️⃣ 流式输出**
58
  full_text = ""
59
  for chunk in response:
60
+ text_chunk = chunk.choices[0].delta.content or ""
61
+ full_text += text_chunk
62
+ yield full_text # **实时输出**
 
63
 
64
  yield "✅ 任务已完成!"
65