本文介紹如何通過Python SDK向Collection中插入或更新Doc。
前提條件
已創建Cluster：創建Cluster。
已獲得API-KEY：API-KEY管理。
已安裝最新版SDK：安裝DashVector SDK。
接口定義
Collection.upsert(
docs: Union[Doc, List[Doc], Tuple, List[Tuple]],
partition: Optional[str] = None,
async_req: False
) -> DashVectorResponse使用示例
需要使用您的api-key替換示例中的YOUR_API_KEY、您的Cluster Endpoint替換示例中的YOUR_CLUSTER_ENDPOINT，代碼才能正常運行。
本示例需要參考新建Collection-使用示例提前創建好名稱為
quickstart的Collection。
import dashvector
from dashvector import Doc
import numpy as np
client = dashvector.Client(
api_key='YOUR_API_KEY',
endpoint='YOUR_CLUSTER_ENDPOINT'
)
collection = client.get(name='quickstart')插入或更新Doc
# 通過Doc對象upsert
ret = collection.upsert(
Doc(
id='1',
vector=[0.1, 0.2, 0.3, 0.4]
)
)
# 判斷upsert是否成功
assert ret
# 簡化形式：通過Tuple upsert
ret = collection.upsert(
('2', [0.1, 0.1, 0.1, 0.1]) # (id, vector)
)插入或更新不帶有Id的Doc
# 通過Doc對象upsert
ret = collection.upsert(
Doc(vector=[0.1, 0.2, 0.3, 0.4])
)
# 簡化形式：通過Tuple upsert
ret = collection.upsert(
([0.1, 0.1, 0.1, 0.1],)
)插入或更新帶有Fields的Doc
# upsert單條數據，并設置Fields Value
ret = collection.upsert(
Doc(
id='3',
vector=np.random.rand(4),
fields={
# 設置創建Collection時預定義的Fileds Value
'name': 'zhangsan', 'weight':70.0, 'age':30,
# 設置Schema-Free的Field & Value
'anykey1': 'str-value', 'anykey2': 1,
'anykey3': True, 'anykey4': 3.1415926
}
)
)
# upsert單條數據，并設置Fields Value
ret = collection.upsert(
('4', np.random.rand(4), {'foo': 'bar'}) # (id, vector, fields)
)批量插入或更新Doc
# 通過Doc對象，批量upsert 10條數據
ret = collection.upsert(
[
Doc(id=str(i+5), vector=np.random.rand(4)) for i in range(10)
]
)
# 簡化形式：通過Tuple，批量upsert 3條數據
ret = collection.upsert(
[
('15', [0.2,0.7,0.8,1.3], {'age': 20}),
('16', [0.3,0.6,0.9,1.2], {'age': 30}),
('17', [0.4,0.5,1.0,1.1], {'age': 40})
] # List[(id, vector, fields)]
)
# 判斷批量upsert是否成功
assert ret異步插入或更新Doc
# 異步批量upsert 10條數據
ret_funture = collection.upsert(
[
Doc(id=str(i+18), vector=np.random.rand(4), fields={'name': 'foo' + str(i)}) for i in range(10)
],
async_req=True
)
# 等待并獲取異步upsert結果
ret = ret_funture.get()插入或更新帶有Sparse Vector的Doc
ret = collection.upsert(
Doc(
id='28',
vector=[0.1, 0.2, 0.3, 0.4],
sparse_vector={1:0.4, 10000:0.6, 222222:0.8}
)
)入參描述
參數 | 類型 | 默認值 | 說明 |
docs | - | upsert的數據或數據list | |
partition（可選） | Optional[str] | None | Partition名稱 |
async_req（可選） | bool | False | 是否異步 |
docs參數，id為非必填項。當類型為Tuple時，可以為
(vector)、(id, vector)、(vector, fields)或(id, vector, fields)順序，等同于Doc對象。Doc對象的fields參數，可自由設置“任意”的KeyValue數據，Key必須為
str類型，Value必須為str or int or bool or float。當Key在創建Collection時預先定義過，則Value的類型必須為預定義時的類型
當Key未在創建Collection時預先定義過，則Value的類型可為
str or int or bool or float
是否預先定義Fields，可參考Schema Free。
出參描述
返回結果為DashVectorResponse對象，DashVectorResponse對象中可獲取本次操作結果信息，如下表所示。
字段 | 類型 | 描述 | 示例 |
code | int | 返回值，參考返回狀態碼說明 | 0 |
message | str | 返回消息 | success |
request_id | str | 請求唯一id | 19215409-ea66-4db9-8764-26ce2eb5bb99 |
output | List[DocOpResult] | 返回插入或更新Doc的結果 | |
usage | 對Serverless實例（按量付費）集合的Doc插入或更新請求，成功后返回實際消耗的寫請求單元數 |