ollama embedding兼容OpenAI格式

ollama embeddings 端点:/api/embed,而非/embeddings,输出的格式与OpenAI不兼容,下面给出兼容的方式。推荐模型: bge-m3

import json

# Ollama 嵌入数据
ollama_embeddings = {
    "model": "all-minilm",
    "embeddings": [
        [
            0.010071029, -0.0017594862, 0.05007221, 0.04692972, 0.054916814,
            0.008599704, 0.105441414, -0.025878139, 0.12958129, 0.031952348
        ],
        [
            -0.0098027075, 0.06042469, 0.025257962, -0.006364387, 0.07272725,
            0.017194884, 0.09032035, -0.051705178, 0.09951512, 0.09072481
        ]
    ]
}

# 转换为 OpenAI 嵌入输出格式
openai_embeddings = {
    "data": [
        {
            "embedding": embedding,
            "index": index,
            "object": "embedding"
        }
        for index, embedding in enumerate(ollama_embeddings["embeddings"])
    ]
}

# 打印转换后的 JSON 格式
print(json.dumps(openai_embeddings, indent=2))

输出结果:

{
  "data": [
    {
      "embeddin
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

jacky_wxl(微信同号)

喜欢作者

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值