|
14 | 14 | from langchain.text_splitter import RecursiveCharacterTextSplitter |
15 | 15 | from langchain_community.document_loaders.pdf import PyPDFLoader |
16 | 16 | from langchain_openai import OpenAIEmbeddings |
17 | | -from langchain_pinecone import PineconeVectorStore |
| 17 | +from langchain_pinecone import PineconeVectorStore # pylint: disable=import-error |
18 | 18 |
|
19 | 19 | # pinecone integration |
20 | 20 | from pinecone import AwsRegion, CloudProvider, Pinecone, ServerlessSpec, VectorType |
| 21 | + |
| 22 | +# pylint: disable=import-error |
21 | 23 | from pinecone.core.openapi.db_data.models import ( |
22 | 24 | IndexDescription as PineconeIndexDescription, |
23 | 25 | ) |
24 | | -from pinecone.db_control.models import IndexList |
25 | | -from pinecone.db_data import Index |
| 26 | +from pinecone.db_control.models import IndexList # pylint: disable=import-error |
| 27 | +from pinecone.db_data import Index # pylint: disable=import-error |
26 | 28 | from pinecone.exceptions import PineconeApiException |
27 | 29 | from pydantic import SecretStr |
28 | 30 |
|
@@ -105,7 +107,6 @@ def vector_store(self) -> PineconeVectorStore: |
105 | 107 | def openai_embeddings(self) -> OpenAIEmbeddings: |
106 | 108 | """OpenAIEmbeddings lazy read-only property.""" |
107 | 109 | if self._openai_embeddings is None: |
108 | | - # pylint: disable=no-member |
109 | 110 | self._openai_embeddings = OpenAIEmbeddings( |
110 | 111 | api_key=settings.openai_api_key, |
111 | 112 | organization=settings.openai_api_organization, |
@@ -138,7 +139,6 @@ def init_index(self): |
138 | 139 | logging.debug("Index does not exist.") |
139 | 140 | self.create() |
140 | 141 |
|
141 | | - # pylint: disable=no-member |
142 | 142 | def init(self): |
143 | 143 | """Initialize Pinecone.""" |
144 | 144 |
|
|
0 commit comments