fix Pydantic v2 errors

Significant-Gravitas · Jul 4, 2024 · b7da031 · b7da031
1 parent c9501bd
commit b7da031
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/forge/forge/llm/providers/llamafile/llamafile.py b/forge/forge/llm/providers/llamafile/llamafile.py
@@ -57,7 +57,7 @@ class LlamafileConfiguration(ModelProviderConfiguration):
 
 
 class LlamafileCredentials(ModelProviderCredentials):
-    api_key: SecretStr = SecretStr("sk-no-key-required")
+    api_key: Optional[SecretStr] = SecretStr("sk-no-key-required")
     api_base: SecretStr = UserConfigurable(  # type: ignore
         default=SecretStr("http://localhost:8080/v1"), from_env="LLAMAFILE_API_BASE"
     )

diff --git a/forge/forge/llm/providers/multi.py b/forge/forge/llm/providers/multi.py
@@ -151,7 +151,7 @@ def _get_provider(self, provider_name: ModelProviderName) -> ChatModelProvider:
                 try:
                     settings.credentials = Credentials.from_env()
                 except ValidationError as e:
-                    if credentials_field.required:
+                    if credentials_field.is_required():
                         self._logger.debug(
                             f"Could not load (required) {Credentials.__name__}"
                         )