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 Weston Pace
parent b32b69c993
commit 4b40dad963

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":