admin
d703ab64e3
CI / docs-check (push) Canceled after 0s
CI / backend-test (push) Canceled after 0s
CI / service-test (push) Canceled after 0s
CI / frontend-test (push) Canceled after 0s
CI / rust-core (push) Canceled after 0s
CI / docs-check (pull_request) Canceled after 0s
CI / backend-test (pull_request) Canceled after 0s
CI / service-test (pull_request) Canceled after 0s
CI / frontend-test (pull_request) Canceled after 0s
CI / rust-core (pull_request) Canceled after 0s
12 lines
593 B
Python
12 lines
593 B
Python
"""绑定嵌入结果帧，因此大笔记不会超出管道限制。"""
|
|
import json
|
|
|
|
|
|
def response_lines(response, operation):
|
|
if operation == 'embedding' and 'result' in response and 'error_code' not in response:
|
|
vectors = response['result']
|
|
for offset in range(0, len(vectors), 128):
|
|
yield json.dumps({'embedding_offset': offset, 'embedding_chunk': vectors[offset:offset + 128]}, allow_nan=False) + '\n'
|
|
response = {**response, 'result': [], 'embedding_count': len(vectors)}
|
|
yield json.dumps(response, ensure_ascii=False, allow_nan=False) + '\n'
|