jina-embeddings-v3 / demo /grequests_demo.py
sanbo
update sth. at 2025-01-16 23:44:43
e397647
raw
history blame contribute delete
955 Bytes
import grequests
import json
def embeddings_run(inputs, url="https://sanbo1200-jina-embeddings-v3.hf.space/api/v1/embeddings", model="jinaai/jina-embeddings-v3"):
headers = {
"Content-Type": "application/json"
}
# 支持单个或批量请求
if isinstance(inputs, str):
inputs = [inputs]
# 创建请求列表
requests = [
grequests.post(
url,
headers=headers,
json={"input": input, "model": model}
) for input in inputs
]
# 并发执行请求
responses = grequests.map(requests)
# 处理响应
results = []
for response in responses:
if response and response.status_code == 200:
results.append(response.json())
return results[0] if len(inputs) == 1 else results
if __name__ == "__main__":
input_text = "Your text string goes here"
print(f"---{embeddings_run(input_text)}")