Fmt

libs/experimental/langchain_experimental/open_clip/open_clip.py

@@ -1,7 +1,5 @@
 from typing import Any, Dict, List
 
-import numpy as np
-
 from langchain.pydantic_v1 import BaseModel, root_validator
 from langchain.schema.embeddings import Embeddings
 
@@ -43,7 +41,7 @@ class OpenCLIPEmbeddings(BaseModel, Embeddings):
     def embed_query(self, text: str) -> List[float]:
         return self.embed_documents([text])[0]
 
-    def embed_image(self, images: List[np.ndarray]) -> List[List[float]]:
+    def embed_image(self, images: List[List[float]]) -> List[List[float]]:
         try:
             from PIL import Image as _PILImage
         except ImportError:

libs/langchain/langchain/embeddings/__init__.py

@@ -53,7 +53,6 @@ from langchain.embeddings.mosaicml import MosaicMLInstructorEmbeddings
 from langchain.embeddings.nlpcloud import NLPCloudEmbeddings
 from langchain.embeddings.octoai_embeddings import OctoAIEmbeddings
 from langchain.embeddings.ollama import OllamaEmbeddings
-from langchain.embeddings.open_clip import OpenCLIPEmbeddings
 from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.embeddings.sagemaker_endpoint import SagemakerEndpointEmbeddings
 from langchain.embeddings.self_hosted import SelfHostedEmbeddings
@@ -118,7 +117,6 @@ __all__ = [
     "QianfanEmbeddingsEndpoint",
     "JohnSnowLabsEmbeddings",
     "VoyageEmbeddings",
-    "OpenCLIPEmbeddings",
 ]
 
 

libs/langchain/tests/unit_tests/embeddings/test_imports.py

@@ -48,7 +48,6 @@ EXPECTED_ALL = [
     "QianfanEmbeddingsEndpoint",
     "JohnSnowLabsEmbeddings",
     "VoyageEmbeddings",
-    "OpenCLIPEmbeddings",
 ]