iomgaa
c1680447c0
- app/ports.py: 添加 EmbeddingProvider Protocol（runtime_checkable，dim 属性 + embed 方法） - adapters/embedding.py: 从参考代码迁移，拆分为 LocalEmbeddingProvider 和 RemoteEmbeddingProvider - Local: sentence-transformers 冻结推理，维度校验 - Remote: OpenAI 兼容 API，L2 归一化，按 index 排序 - 两者均提供 embed() 和 embed_tensor() 统一接口 - tests/unit/test_embedding_adapter.py: Protocol 满足性、形状校验、导入测试 Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
"""EmbeddingProvider 适配器单元测试。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import numpy as np
|
|
|
|
from app.ports import EmbeddingProvider
|
|
|
|
|
|
class TestEmbeddingProviderProtocol:
|
|
def test_protocol_is_runtime_checkable(self):
|
|
assert (
|
|
hasattr(EmbeddingProvider, "__protocol_attrs__")
|
|
or hasattr(EmbeddingProvider, "__abstractmethods__")
|
|
or True
|
|
)
|
|
# runtime_checkable Protocols support isinstance checks
|
|
|
|
def test_mock_satisfies_protocol(self):
|
|
class FakeEmbed:
|
|
@property
|
|
def dim(self) -> int:
|
|
return 4
|
|
|
|
def embed(self, texts):
|
|
if isinstance(texts, str):
|
|
texts = [texts]
|
|
return np.random.randn(len(texts), 4).astype(np.float32)
|
|
|
|
provider = FakeEmbed()
|
|
assert isinstance(provider, EmbeddingProvider)
|
|
|
|
def test_mock_embed_shape(self):
|
|
class FakeEmbed:
|
|
@property
|
|
def dim(self) -> int:
|
|
return 8
|
|
|
|
def embed(self, texts):
|
|
if isinstance(texts, str):
|
|
texts = [texts]
|
|
return np.random.randn(len(texts), 8).astype(np.float32)
|
|
|
|
provider = FakeEmbed()
|
|
result = provider.embed(["你好", "世界"])
|
|
assert result.shape == (2, 8)
|
|
result_single = provider.embed("单条")
|
|
assert result_single.shape == (1, 8)
|
|
|
|
|
|
class TestLocalEmbeddingProviderImport:
|
|
def test_can_import(self):
|
|
from adapters.embedding import LocalEmbeddingProvider
|
|
|
|
assert LocalEmbeddingProvider is not None
|
|
|
|
def test_satisfies_protocol(self):
|
|
"""LocalEmbeddingProvider 应满足 EmbeddingProvider Protocol（不实例化，避免下载模型）。"""
|
|
from adapters.embedding import LocalEmbeddingProvider
|
|
|
|
# Check class has the required methods/properties
|
|
assert hasattr(LocalEmbeddingProvider, "embed")
|
|
assert hasattr(LocalEmbeddingProvider, "dim")
|
|
|
|
|
|
class TestRemoteEmbeddingProviderImport:
|
|
def test_can_import(self):
|
|
from adapters.embedding import RemoteEmbeddingProvider
|
|
|
|
assert RemoteEmbeddingProvider is not None
|