Initialize the EmbeddingClient. Args: model (str): The OpenAI embedding model name to use.
(self, model_name: str = "text-embedding-3-small")
| 30 | |
| 31 | class EmbeddingClient: |
| 32 | def __init__(self, model_name: str = "text-embedding-3-small"): |
| 33 | """ |
| 34 | Initialize the EmbeddingClient. |
| 35 | |
| 36 | Args: |
| 37 | model (str): The OpenAI embedding model name to use. |
| 38 | """ |
| 39 | self.client, self.model = self._get_client_model(model_name) |
| 40 | |
| 41 | def _get_client_model(self, model_name: str) -> tuple[openai.OpenAI, str]: |
| 42 | if model_name in OPENAI_EMBEDDING_MODELS: |
nothing calls this directly
no test coverage detected