try to fix dynamic schema issues

Author: voorhs

src/autointent/_optimization_config.py

@@ -1,8 +1,16 @@
 from typing import Any
 
-from pydantic import BaseModel, PositiveInt
+from pydantic import BaseModel, Field, PositiveInt, field_validator
 
-from .configs import CrossEncoderConfig, DataConfig, EmbedderConfig, HFModelConfig, HPOConfig, LoggingConfig
+from .configs import (
+    CrossEncoderConfig,
+    DataConfig,
+    EmbedderConfig,
+    HFModelConfig,
+    HPOConfig,
+    LoggingConfig,
+    initialize_embedder_config,
+)
 
 
 class OptimizationConfig(BaseModel):
@@ -20,7 +28,13 @@ class OptimizationConfig(BaseModel):
     logging_config: LoggingConfig = LoggingConfig()
     """See tutorial on logging configuration."""
 
-    embedder_config: EmbedderConfig = EmbedderConfig()
+    embedder_config: EmbedderConfig = Field(default_factory=lambda: initialize_embedder_config(None))
+
+    @field_validator("embedder_config", mode="before")
+    @classmethod
+    def validate_embedder_config(cls, v: Any) -> EmbedderConfig:  # noqa: ANN401
+        """Validate and convert embedder config to proper type."""
+        return initialize_embedder_config(v)
 
     cross_encoder_config: CrossEncoderConfig = CrossEncoderConfig()
 

src/autointent/modules/embedding/_logreg.py

@@ -67,7 +67,7 @@ def __init__(
     def from_context(
         cls,
         context: Context,
-        embedder_config: EmbedderConfig | str | None = None,
+        embedder_config: EmbedderConfig | str | dict[str, Any] | None = None,
         ft_config: EmbedderFineTuningConfig | dict[str, Any] | None = None,
         cv: PositiveInt = 3,
     ) -> "LogregAimedEmbedding":

src/autointent/modules/embedding/_retrieval.py

@@ -71,7 +71,7 @@ def __init__(
     def from_context(
         cls,
         context: Context,
-        embedder_config: EmbedderConfig | str | None = None,
+        embedder_config: EmbedderConfig | str | dict[str, Any] | None = None,
         k: PositiveInt = 10,
         ft_config: EmbedderFineTuningConfig | dict[str, Any] | None = None,
     ) -> "RetrievalAimedEmbedding":

src/autointent/modules/scoring/_description/bi_encoder.py

@@ -73,7 +73,7 @@ def from_context(
         cls,
         context: Context,
         temperature: PositiveFloat = 1.0,
-        embedder_config: EmbedderConfig | str | None = None,
+        embedder_config: EmbedderConfig | str | dict[str, Any] | None = None,
     ) -> "BiEncoderDescriptionScorer":
         """Create a BiEncoderDescriptionScorer instance using a Context object.
 

src/autointent/modules/scoring/_linear.py

@@ -73,7 +73,7 @@ def from_context(
         cls,
         context: Context,
         cv: PositiveInt = 3,
-        embedder_config: EmbedderConfig | str | None = None,
+        embedder_config: EmbedderConfig | str | dict[str, Any] | None = None,
     ) -> "LinearScorer":
         """Create a LinearScorer instance using a Context object.