Viking 长期记忆支持通过 Python SDK 操作记忆库,Python SDK 提供了高层次的抽象和易于使用的接口,简化开发人员的工作。在使用 Python SDK 之前,需要先完成安装 SDK 和 初始化 SDK 步骤。

首次使用可通过 pip 安装 SDK for Python,支持Python 3.7 及其以上版本。
pip install vikingdb-python-sdk
如需更新SDK版本,可使用以下命令:
pip install --upgrade vikingdb-python-sdk
配置安全凭证,根据记忆库名和项目名或资源ID初始化客户端。
说明
代码中 your_ak 及 your_sk 需要分别替换为您的 AK 及 SK。
import os from vikingdb import IAM from vikingdb.memory import VikingMem _auth = IAM( ak=os.getenv("VIKINGDB_AK", "your_ak"), sk=os.getenv("VIKINGDB_SK", "your_sk"), ) client = VikingMem( host="api-knowledgebase.mlp.cn-beijing.volces.com", region="cn-beijing", auth=_auth, scheme="http", ) print("Client initialized successfully") if __name__ == "__main__": client.ping() ## 例子 1: 用collection_name和project_name collection1 = client.get_collection( collection_name="sdk_test", # 替换为您的记忆库名称 project_name="default" ) ## 例子 2: 用resource_id collection2 = client.get_collection( resource_id="col-abc123xyz" )