Public/koboldcpp
1
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2026-09-19 01:05:09 +02:00
Code Issues Packages Projects Releases Wiki Activity

embeddings handle base64

Browse Source
This commit is contained in:
Concedo
2025-06-09 00:26:40 +08:00
parent 7de88802f9
commit 82d7c53b85
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
koboldcpp.py
+6 -1
View File
@@ -3770,7 +3770,12 @@ Change Mode<br>
outdatas = []
odidx = 0
for od in gen["data"]:
outdatas.append({"object":"embedding","index":odidx,"embedding":od})
if genparams.get("encoding_format", "")=="base64":
binary_data = struct.pack('<' + 'f' * len(od), *od)
b64_string = base64.b64encode(binary_data).decode('utf-8')
outdatas.append({"object":"embedding","index":odidx,"embedding":b64_string})
else:
outdatas.append({"object":"embedding","index":odidx,"embedding":od})
odidx += 1
genresp = (json.dumps({"object":"list","data":outdatas,"model":friendlyembeddingsmodelname,"usage":{"prompt_tokens":gen["count"],"total_tokens":gen["count"]}}).encode())
self.send_response(200)