Update app.py
Browse files
app.py
CHANGED
|
@@ -68,13 +68,24 @@ def normalize_headers(request: Request, original_url: str) -> dict:
|
|
| 68 |
return normalized
|
| 69 |
|
| 70 |
|
| 71 |
-
def
|
| 72 |
-
chunk_id = 'chatcmpl-AQ8Lzxlg8eSCB1lgVmboiXwZiexqE'
|
| 73 |
-
timestamp = int(datetime.now().timestamp())
|
| 74 |
try:
|
| 75 |
content = chunk.decode('utf-8')
|
| 76 |
except UnicodeDecodeError:
|
| 77 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 78 |
data = {
|
| 79 |
"id": chunk_id,
|
| 80 |
"object": "chat.completion.chunk",
|
|
@@ -85,7 +96,7 @@ def format_chunk(chunk: bytes, model: str) -> bytes | str | dict:
|
|
| 85 |
{
|
| 86 |
"index": 0,
|
| 87 |
"delta": {
|
| 88 |
-
"content":
|
| 89 |
},
|
| 90 |
"finish_reason": None
|
| 91 |
}
|
|
|
|
| 68 |
return normalized
|
| 69 |
|
| 70 |
|
| 71 |
+
def decode(chunk: bytes) -> str:
|
|
|
|
|
|
|
| 72 |
try:
|
| 73 |
content = chunk.decode('utf-8')
|
| 74 |
except UnicodeDecodeError:
|
| 75 |
+
try:
|
| 76 |
+
content = chunk.decode('latin-1')
|
| 77 |
+
except UnicodeDecodeError:
|
| 78 |
+
try:
|
| 79 |
+
content = chunk.decode('iso-8859-1')
|
| 80 |
+
except UnicodeDecodeError:
|
| 81 |
+
content = chunk.decode(errors='replace')
|
| 82 |
+
return content
|
| 83 |
+
|
| 84 |
+
|
| 85 |
+
def format_chunk(chunk: bytes, model: str) -> bytes | str | dict:
|
| 86 |
+
chunk_id = 'chatcmpl-AQ8Lzxlg8eSCB1lgVmboiXwZiexqE'
|
| 87 |
+
timestamp = int(datetime.now().timestamp())
|
| 88 |
+
|
| 89 |
data = {
|
| 90 |
"id": chunk_id,
|
| 91 |
"object": "chat.completion.chunk",
|
|
|
|
| 96 |
{
|
| 97 |
"index": 0,
|
| 98 |
"delta": {
|
| 99 |
+
"content": decode(chunk)
|
| 100 |
},
|
| 101 |
"finish_reason": None
|
| 102 |
}
|