本文介绍如何通过[百川智能向量化模型](https://platform.baichuan-ai.com/docs/text-Embedding)将 **文本转换为向量** ,并入库至向量检索服务DashVector中进行向量检索。
前提条件
---------------------
* DashVector:
* 已创建Cluster
* 已获得API-KEY
* 已安装最新版SDK
* 百川智能:
* 已获得API密钥:[百川智能向量化模型](https://platform.baichuan-ai.com/docs/text-Embedding)
**百川智能向量化模型**
------------------------------
### **简介**
![image.png](https://static.golangjob.cn/240905/75e2aa2b2a3bfa84dff4e75344d6f2cd.png)
**说明**
关于百川智能向量化模型更多信息请参考:[百川智能向量化模型](https://platform.baichuan-ai.com/docs/text-Embedding)。
### **使用示例**
**说明**
需要进行如下替换代码才能正常运行:
1. DashVector api-key替换示例中的{your-dashvector-api-key}
2. DashVector Cluster Endpoint替换示例中的{your-dashvector-cluster-endpoint}
3. 百川智能api-key替换示例中的{your-baichuan-api-key}
Python
```python
from dashvector import Client
import requests
from typing import List
# 调用百川智能向量化模型服务,将文本embedding为向量
def generate_embeddings(texts: List[str]):
headers = {
'Content-Type': 'application/json',
'Authorization': 'Bearer {your-baichuan-api-key}'
}
data = {'input': texts, 'model': 'Baichuan-Text-Embedding'}
response = requests.post('http://api.baichuan-ai.com/v1/embeddings', headers=headers, json=data)
return [record["embedding"] for record in response.json()["data"]]
# 创建DashVector Client
client = Client(
api_key='{your-dashvector-api-key}',
endpoint='{your-dashvector-cluster-endpoint}'
)
# 创建DashVector Collection
rsp = client.create('baichuan-text-embedding', 1024)
assert rsp
collection = client.get('baichuan-text-embedding')
assert collection
# 向量入库DashVector
collection.insert(
('ID1', generate_embeddings(['阿里云向量检索服务DashVector是性能、性价比具佳的向量数据库之一'])[0])
)
# 向量检索
docs = collection.query(
generate_embeddings(['The best vector database'])[0]
)
print(docs)
```
有疑问加站长微信联系(非本文作者))