feat(python): add model_names() method to openai embedding function (#1049)

small QoL improvement
This commit is contained in:
Chang She
2024-03-03 12:33:00 -08:00
committed by GitHub
parent c1af53b787
commit d14c9b6d9e

View File

@@ -36,6 +36,14 @@ class OpenAIEmbeddings(TextEmbeddingFunction):
def ndims(self):
return self._ndims
@staticmethod
def model_names():
return [
"text-embedding-ada-002",
"text-embedding-3-large",
"text-embedding-3-small",
]
@cached_property
def _ndims(self):
if self.name == "text-embedding-ada-002":