Update app.py
Browse files
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 =
|
| 109 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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)
|