6225310600
- 新增 MilvusGraphVectorStore 和 KnowledgeGraphRepository - 新增文件大小从 MinIO 补全的逻辑（上传时 + 启动时） - 重构 FileTable、GraphDetailPanel、KnowledgeGraphSection 等前端组件 - 新增文件大小回退测试和 kb_utils 测试 - 优化 file_utils 工具函数 - 更新 roadmap 文档
34 lines
1.0 KiB
Python
34 lines
1.0 KiB
Python
from yuxi.knowledge.utils.kb_utils import prepare_item_metadata
|
|
|
|
|
|
async def test_prepare_item_metadata_preserves_uploaded_file_size():
|
|
item = "minio://knowledgebases/db/upload/demo.txt"
|
|
params = {
|
|
"content_hashes": {item: "hash"},
|
|
"file_sizes": {item: 1234},
|
|
}
|
|
|
|
metadata = await prepare_item_metadata(item, "file", "db", params=params)
|
|
|
|
assert metadata["size"] == 1234
|
|
assert "file_sizes" not in (metadata.get("processing_params") or {})
|
|
|
|
|
|
async def test_prepare_item_metadata_preserves_preprocessed_file_size():
|
|
item = "minio://knowledgebases/db/upload/page.html"
|
|
params = {
|
|
"_preprocessed_map": {
|
|
item: {
|
|
"path": item,
|
|
"content_hash": "hash",
|
|
"filename": "https://example.com",
|
|
"file_size": 5678,
|
|
}
|
|
}
|
|
}
|
|
|
|
metadata = await prepare_item_metadata(item, "file", "db", params=params)
|
|
|
|
assert metadata["size"] == 5678
|
|
assert "_preprocessed_map" not in (metadata.get("processing_params") or {})
|