From d14c9b6d9ef418b77ddf9c6b70f57973127335e2 Mon Sep 17 00:00:00 2001
From: Chang She <759245+changhiskhan@users.noreply.github.com>
Date: Sun, 3 Mar 2024 12:33:00 -0800
Subject: [PATCH] feat(python): add model_names() method to openai embedding
 function (#1049)

small QoL improvement
---
 python/python/lancedb/embeddings/openai.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/python/python/lancedb/embeddings/openai.py b/python/python/lancedb/embeddings/openai.py
index 023f3b2f..4af0efce 100644
--- a/python/python/lancedb/embeddings/openai.py
+++ b/python/python/lancedb/embeddings/openai.py
@@ -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":