♻️ minor refactor

renlabs-dev · Jun 27, 2024 · f8d82d6 · f8d82d6
1 parent da0f20e
commit f8d82d6
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/synthia/validator/text_validator.py b/src/synthia/validator/text_validator.py
@@ -212,7 +212,6 @@ def __init__(
         if not embedder:
             embedder = OpenAIEmbedder(OpenAISettings())  # type: ignore
         self.embedder = embedder
-        self.val_model = "anthropic/claude-3.5-sonnet"
         self.upload_client = ModuleClient("5.161.229.89", 80, self.key)
         self.call_timeout = call_timeout
         self.provider = provider
@@ -239,13 +238,13 @@ def _get_validation_dataset(self, settings: ValidatorSettings, size: int):
                 claude_settings = AnthropicSettings()  # type: ignore
                 claude_settings.temperature = settings.temperature
                 claude_settings.max_tokens = settings.max_tokens
-                claude_settings.model = self.val_model
+                claude_settings.model = settings.model
                 claude = AnthropicModule(claude_settings)
             case ClaudeProviders.OPENROUTER:
                 claude_settings = OpenrouterSettings()  # type: ignore
                 claude_settings.temperature = settings.temperature
                 claude_settings.max_tokens = settings.max_tokens
-                claude_settings.model = self.val_model
+                claude_settings.model = settings.model
                 claude = OpenrouterModule(claude_settings)
 
         ig = InputGenerator(claude)