Update vector_store_retriever.py
Browse files
vector_store_retriever.py
CHANGED
@@ -6,7 +6,7 @@ from pydantic import BaseModel, Field
|
|
6 |
from typing import Any, Optional, Dict, List, Union
|
7 |
from huggingface_hub import InferenceClient
|
8 |
from langchain.llms.base import LLM
|
9 |
-
from langchain.Images import Images
|
10 |
from langchain.llms.base import LLM
|
11 |
from langchain.embeddings import HuggingFaceInstructEmbeddings, EmbeddingFunction, Embeddings
|
12 |
|
@@ -25,7 +25,7 @@ class HuggingFaceInstructEmbeddings(EmbeddingFunction):
|
|
25 |
self.model = AutoModel.from_pretrained(model_name, **(model_kwargs or {}))
|
26 |
self.tokenizer = AutoTokenizer.from_pretrained(model_name)
|
27 |
|
28 |
-
def __call__(self, input: Union[Documents
|
29 |
if isinstance(input, Documents):
|
30 |
texts = [doc.text for doc in input]
|
31 |
embeddings = self._embed_text(texts)
|
|
|
6 |
from typing import Any, Optional, Dict, List, Union
|
7 |
from huggingface_hub import InferenceClient
|
8 |
from langchain.llms.base import LLM
|
9 |
+
#from langchain.Images import Images
|
10 |
from langchain.llms.base import LLM
|
11 |
from langchain.embeddings import HuggingFaceInstructEmbeddings, EmbeddingFunction, Embeddings
|
12 |
|
|
|
25 |
self.model = AutoModel.from_pretrained(model_name, **(model_kwargs or {}))
|
26 |
self.tokenizer = AutoTokenizer.from_pretrained(model_name)
|
27 |
|
28 |
+
def __call__(self, input: Union[Documents]) -> Embeddings:
|
29 |
if isinstance(input, Documents):
|
30 |
texts = [doc.text for doc in input]
|
31 |
embeddings = self._embed_text(texts)
|