jackkuo commited on
Commit
10ac639
·
verified ·
1 Parent(s): 4bf2c6c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -95,6 +95,7 @@ def extract_pdf_md(pdf_dir):
95
  return content
96
 
97
 
 
98
  def openai_api(messages, model="claude-sonnet-4-5-20250929", temperature=0.1):
99
  print("use model:", model, "temperature:", temperature)
100
  try:
@@ -105,8 +106,12 @@ def openai_api(messages, model="claude-sonnet-4-5-20250929", temperature=0.1):
105
  max_tokens=8192,
106
  stream=True
107
  )
108
- response = ''.join(
109
- [chunk.choices[0].delta.content if chunk.choices[0].delta.content else "" for chunk in completion])
 
 
 
 
110
  return response
111
  except Exception as ex:
112
  print("API error:", ex)
 
95
  return content
96
 
97
 
98
+
99
  def openai_api(messages, model="claude-sonnet-4-5-20250929", temperature=0.1):
100
  print("use model:", model, "temperature:", temperature)
101
  try:
 
106
  max_tokens=8192,
107
  stream=True
108
  )
109
+ response = ""
110
+ for chunk in completion:
111
+ if chunk.choices and len(chunk.choices) > 0:
112
+ delta = chunk.choices[0].delta
113
+ if delta.content:
114
+ response += delta.content
115
  return response
116
  except Exception as ex:
117
  print("API error:", ex)