33 lines
964 B
Python
33 lines
964 B
Python
from providers.factory.default import DefaultRealtimeServiceFactory
|
|
from providers.llm.dify import DifyLLMService
|
|
from providers.llm.openai import OpenAILLMService
|
|
from runtime.ports import LLMServiceSpec
|
|
|
|
|
|
def test_create_llm_service_dify_returns_dify_provider():
|
|
factory = DefaultRealtimeServiceFactory()
|
|
service = factory.create_llm_service(
|
|
LLMServiceSpec(
|
|
provider="dify",
|
|
model="dify",
|
|
api_key="test-key",
|
|
base_url="https://dify.example/v1",
|
|
)
|
|
)
|
|
|
|
assert isinstance(service, DifyLLMService)
|
|
|
|
|
|
def test_create_llm_service_openai_returns_openai_provider():
|
|
factory = DefaultRealtimeServiceFactory()
|
|
service = factory.create_llm_service(
|
|
LLMServiceSpec(
|
|
provider="openai",
|
|
model="gpt-4o-mini",
|
|
api_key="test-key",
|
|
base_url="https://api.openai.com/v1",
|
|
)
|
|
)
|
|
|
|
assert isinstance(service, OpenAILLMService)
|