diff --git a/pretor/core/global_state_machine/model_provider/deepseek_provider.py b/pretor/core/global_state_machine/model_provider/deepseek_provider.py index 032c6d9..0ca8a6f 100644 --- a/pretor/core/global_state_machine/model_provider/deepseek_provider.py +++ b/pretor/core/global_state_machine/model_provider/deepseek_provider.py @@ -17,11 +17,11 @@ from pretor.core.global_state_machine.model_provider.base_provider import BasePr import httpx from typing import List -class OpenAIProvider(BaseProvider): +class DeepseekProvider(BaseProvider): @staticmethod async def create_provider(provider_args: ProviderArgs) -> Provider: - provider_models: List[str] = await OpenAIProvider._load_models(provider_args) - provider: Provider = OpenAIProvider._return_provider(provider_args, provider_models) + provider_models: List[str] = await DeepseekProvider._load_models(provider_args) + provider: Provider = DeepseekProvider._return_provider(provider_args, provider_models) return provider @staticmethod