Update app.py
Browse files
app.py
CHANGED
|
@@ -76,18 +76,20 @@ async def complete(
|
|
| 76 |
seed=seed,
|
| 77 |
stream=True
|
| 78 |
)
|
|
|
|
| 79 |
for chunk in output:
|
| 80 |
-
|
| 81 |
delta = chunk['choices'][0]['delta']
|
| 82 |
if 'role' in delta:
|
| 83 |
print(delta['role'], end=': ')
|
| 84 |
elif 'content' in delta:
|
| 85 |
print(delta['content'], end='')
|
| 86 |
-
|
| 87 |
print(chunk)
|
|
|
|
| 88 |
et = time()
|
| 89 |
output["time"] = et - st
|
| 90 |
-
|
| 91 |
except Exception as e:
|
| 92 |
logger.error(f"Error in /complete endpoint: {e}")
|
| 93 |
return JSONResponse(
|
|
|
|
| 76 |
seed=seed,
|
| 77 |
stream=True
|
| 78 |
)
|
| 79 |
+
"""
|
| 80 |
for chunk in output:
|
| 81 |
+
|
| 82 |
delta = chunk['choices'][0]['delta']
|
| 83 |
if 'role' in delta:
|
| 84 |
print(delta['role'], end=': ')
|
| 85 |
elif 'content' in delta:
|
| 86 |
print(delta['content'], end='')
|
| 87 |
+
|
| 88 |
print(chunk)
|
| 89 |
+
"""
|
| 90 |
et = time()
|
| 91 |
output["time"] = et - st
|
| 92 |
+
return output
|
| 93 |
except Exception as e:
|
| 94 |
logger.error(f"Error in /complete endpoint: {e}")
|
| 95 |
return JSONResponse(
|