🎨 style(embeddings): improve code formatting and consistency

    This class handles both the embedding model lifecycle and storage initialization.

    """

    def __init__(

        self,

        config: AiConfig

    ) -> None:

    def __init__(self, config: AiConfig) -> None:

        """Initialize embeddings manager.

        Args:

        Raises:

            RuntimeError: If neither storage nor model was provided at init.

        """

        if self._storage is None:

            self._storage = AiStorage(self.config.ai_cache_dir, embedding_dimension=self.dimension)

            logger.info(f"Initialized AiStorage at {self.config.ai_cache_dir} with dimension {self.dimension}")

import litellm

from tdoc_ai.operations.embeddings import EmbeddingsManager

from tdoc_ai.config import AiConfig

from tdoc_ai.models import DocumentSummary, LlmConfigError, SummarizeResult

from tdoc_ai.operations.embeddings import EmbeddingsManager

from tdoc_ai.operations.workspace_names import normalize_workspace_name

from tdoc_ai.storage import AiStorage

from tdoc_crawler.tdocs.sources.whatthespec import resolve_via_whatthespec