File tree Expand file tree Collapse file tree 1 file changed +13
-4
lines changed
apis/python/src/tiledb/vector_search/embeddings Expand file tree Collapse file tree 1 file changed +13
-4
lines changed Original file line number Diff line number Diff line change 88# class LangChainEmbedding(ObjectEmbedding):
99class LangChainEmbedding :
1010 """
11- Embedding functions from `langchain.embeddings` package.
11+ Embedding functions from Langchain.
12+
13+ This attempts to import the embedding_class from the following modules:
14+ - langchain_openai
15+ - langchain.embeddings
1216 """
1317
1418 def __init__ (
@@ -37,9 +41,14 @@ def vector_type(self) -> np.dtype:
3741 def load (self ) -> None :
3842 import importlib
3943
40- embeddings_module = importlib .import_module ("langchain.embeddings" )
41- embedding_class_ = getattr (embeddings_module , self .embedding_class )
42- self .embedding = embedding_class_ (** self .embedding_kwargs )
44+ try :
45+ embeddings_module = importlib .import_module ("langchain_openai" )
46+ embedding_class_ = getattr (embeddings_module , self .embedding_class )
47+ self .embedding = embedding_class_ (** self .embedding_kwargs )
48+ except ImportError :
49+ embeddings_module = importlib .import_module ("langchain.embeddings" )
50+ embedding_class_ = getattr (embeddings_module , self .embedding_class )
51+ self .embedding = embedding_class_ (** self .embedding_kwargs )
4352
4453 def embed (self , objects : OrderedDict , metadata : OrderedDict ) -> np .ndarray :
4554 return np .array (
You can’t perform that action at this time.
0 commit comments