fix: make lint

libs/community/langchain_community/document_loaders/needle.py

@@ -93,6 +93,7 @@ def add_files(self, files: dict) -> None:
             ValueError: If the collection is not properly initialized.
         """
         self._get_collection()
+        assert self.client is not None, "NeedleClient must be initialized."
 
         files_to_add = []
         for name, url in files.items():
@@ -113,6 +114,7 @@ def _fetch_documents(self) -> List[Document]:
             ValueError: If the collection is not properly initialized.
         """
         self._get_collection()
+        assert self.client is not None, "NeedleClient must be initialized."
 
         files = self.client.collections.files.list(self.collection_id)
         docs = []
@@ -129,6 +131,7 @@ def _fetch_documents(self) -> List[Document]:
                 docs.append(doc)
         return docs
 
+
     def load(self) -> List[Document]:
         """
         Loads all documents from the Needle collection.

libs/community/langchain_community/retrievers/needle.py

@@ -66,6 +66,9 @@ def _search_collection(self, query: str) -> List[Document]:
             List[Document]: A list of documents matching the search query.
         """
         self._initialize_client()
+        if self.client is None:
+            raise ValueError("Provide a valid API key.")
+
         results = self.client.collections.search(
             collection_id=self.collection_id, text=query
         )