天翼云星辰智能体知识库 Python SDK

这是一个用于对接天翼云星辰智能体平台知识库服务的Python SDK。

🎉 测试状态

• ✅ API连接成功 - 已验证可用
• ✅ 文件上传 - 功能正常
• ✅ 知识库创建 - 功能正常
• ✅ 文档检索 - 功能正常，响应快速
• ✅ 混合检索 - 效果良好

测试日期: 2026-01-24
当前API Key: dataset-5sB0vOWsdenDq7mhlz1nRZFc
详细测试结果: 查看 API_TEST_RESULTS.md

功能特性

• ✅ 文件上传
• ✅ 知识库创建
• ✅ 文档检索
• ✅ 知识库管理
• ✅ 混合检索（向量+关键词）
• ✅ 重排序支持

安装依赖

pip install -r requirements.txt

快速开始

1. 配置API Key

编辑 .env 文件：

TELEAI_API_KEY=dataset-your-api-key-here
TELEAI_BASE_URL=https://agent.teleai.com.cn/v1

2. 上传文件并创建知识库

from teleai_kb import TeleAIKnowledgeBase

# 初始化客户端
kb = TeleAIKnowledgeBase()

# 上传文件
file_id = kb.upload_file("document.pdf")

# 创建知识库
dataset_id = kb.create_dataset(
    name="我的知识库",
    description="产品文档知识库",
    file_ids=[file_id]
)

3. 检索文档

# 检索相关文档
results = kb.retrieve(
    dataset_id=dataset_id,
    query="如何重置密码？",
    top_k=3
)

for result in results:
    print(f"内容: {result['content']}")
    print(f"分数: {result['score']}")

API文档

详见 docs/API.md

示例

查看 examples/ 目录下的示例代码。

许可证

MIT License