102 lines
2.7 KiB
Python
102 lines
2.7 KiB
Python
from typing import List
|
|
|
|
from src.llm_models.model_client.base_client import (
|
|
APIResponse,
|
|
AudioTranscriptionRequest,
|
|
BaseClient,
|
|
ClientProviderRegistration,
|
|
ClientRegistry,
|
|
EmbeddingRequest,
|
|
ResponseRequest,
|
|
)
|
|
|
|
|
|
class DummyClient(BaseClient):
|
|
"""测试用 LLM 客户端。"""
|
|
|
|
async def get_response(self, request: ResponseRequest) -> APIResponse:
|
|
"""获取测试响应。
|
|
|
|
Args:
|
|
request: 统一响应请求。
|
|
|
|
Returns:
|
|
APIResponse: 测试响应。
|
|
"""
|
|
del request
|
|
return APIResponse(content="ok")
|
|
|
|
async def get_embedding(self, request: EmbeddingRequest) -> APIResponse:
|
|
"""获取测试嵌入。
|
|
|
|
Args:
|
|
request: 统一嵌入请求。
|
|
|
|
Returns:
|
|
APIResponse: 测试嵌入响应。
|
|
"""
|
|
del request
|
|
return APIResponse(embedding=[1.0])
|
|
|
|
async def get_audio_transcriptions(self, request: AudioTranscriptionRequest) -> APIResponse:
|
|
"""获取测试音频转写。
|
|
|
|
Args:
|
|
request: 统一音频转写请求。
|
|
|
|
Returns:
|
|
APIResponse: 测试音频转写响应。
|
|
"""
|
|
del request
|
|
return APIResponse(content="audio")
|
|
|
|
def get_support_image_formats(self) -> List[str]:
|
|
"""获取测试支持的图片格式。
|
|
|
|
Returns:
|
|
List[str]: 支持的图片格式列表。
|
|
"""
|
|
return ["png"]
|
|
|
|
|
|
def test_client_registry_rejects_provider_conflict():
|
|
"""同一 client_type 被不同插件注册时应拒绝。"""
|
|
registry = ClientRegistry()
|
|
registry.replace_plugin_providers(
|
|
"plugin.alpha",
|
|
[
|
|
ClientProviderRegistration(
|
|
client_type="example",
|
|
factory=DummyClient,
|
|
owner_plugin_id="plugin.alpha",
|
|
)
|
|
],
|
|
)
|
|
|
|
try:
|
|
registry.validate_plugin_provider_replacement("plugin.beta", ["example"])
|
|
except ValueError as exc:
|
|
assert "冲突" in str(exc)
|
|
else:
|
|
raise AssertionError("不同插件注册相同 client_type 应失败")
|
|
|
|
|
|
def test_client_registry_unregisters_plugin_providers():
|
|
"""插件注销时应移除它拥有的 Provider 注册。"""
|
|
registry = ClientRegistry()
|
|
registry.replace_plugin_providers(
|
|
"plugin.alpha",
|
|
[
|
|
ClientProviderRegistration(
|
|
client_type="example",
|
|
factory=DummyClient,
|
|
owner_plugin_id="plugin.alpha",
|
|
)
|
|
],
|
|
)
|
|
|
|
removed_count = registry.unregister_plugin_providers("plugin.alpha")
|
|
|
|
assert removed_count == 1
|
|
assert "example" not in registry.client_registry
|