Get the embedding dimension.
(self)
| 241 | |
| 242 | |
| 243 | def get_embedding_dim(self) -> int: |
| 244 | """Get the embedding dimension.""" |
| 245 | if self.embedding_model == "text-embedding-ada-002": |
| 246 | embedding_dim = 1536 |
| 247 | elif self.embedding_model == "text-embedding-3-small": |
| 248 | embedding_dim = 1536 |
| 249 | elif self.embedding_model == "text-embedding-3-large": |
| 250 | embedding_dim = 3072 |
| 251 | else: |
| 252 | raise ValueError(f"Unknown embedding model: {self.embedding_model}") |
| 253 | return embedding_dim |
| 254 | |
| 255 | |
| 256 | def create_completion( |