dhJSSKJrJrJrJr SSKJr SSKJr "SS\5r g))DictIteratorListOptional) BaseLoader)Documentc^\rSrSrSrSS\\S\\SS4U4SjjjrSSjrS \ \\4SS4S jr S\ \ 4S jr S\ \ 4S jrS\\ 4S jrSrU=r$) NeedleLoadera NeedleLoader is a document loader for managing documents stored in a collection. Setup: Install the `needle-python` library and set your Needle API key. .. code-block:: bash pip install needle-python export NEEDLE_API_KEY="your-api-key" Key init args: - `needle_api_key` (Optional[str]): API key for authenticating with Needle. - `collection_id` (str): Needle collection to load documents from. Usage: .. code-block:: python from langchain_community.document_loaders.needle import NeedleLoader loader = NeedleLoader( needle_api_key="your-api-key", collection_id="your-collection-id" ) # Load documents documents = loader.load() for doc in documents: print(doc.metadata) # Lazy load documents for doc in loader.lazy_load(): print(doc.metadata) Nneedle_api_key collection_idreturnc >SSKJn [TU]5 XlX lSUlUR (aU"UR S9UlUR (d [S5eg![a [S5ef=f)a[ Initializes the NeedleLoader with API key and collection ID. Args: needle_api_key (Optional[str]): API key for authenticating with Needle. collection_id (Optional[str]): Identifier for the Needle collection. Raises: ImportError: If the `needle-python` library is not installed. ValueError: If the collection ID is not provided. r) NeedleClientzDPlease install with `pip install needle-python` to use NeedleLoader.N)api_keyCollection ID must be provided.) needle.v1r ImportErrorsuper__init__r r client ValueError)selfr r r __class__s c/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/document_loaders/needle.pyrNeedleLoader.__init__+s   . ,*.2   &t/B/BCDK!!>? ?" V  s A--BclURc [S5eUR(d [S5eg)z Ensures the Needle collection is set and the client is initialized. Raises: ValueError: If the Needle client is not initialized or if the collection ID is missing. Nz9NeedleClient is not initialized. Provide a valid API key.r)rrr rs r_get_collectionNeedleLoader._get_collectionMs: ;; K !!>? ?"filesc VSSKJn UR5 URcS5eUR 5VVs/sH up4U"X4S9PM nnnURR RRURUS9 g![a [S5ef=fs snnf)a. Adds files to the Needle collection. Args: files (Dict[str, str]): Dictionary where keys are file names and values are file URLs. Raises: ImportError: If the `needle-python` library is not installed. ValueError: If the collection is not properly initialized. r) FileToAddz=Please install with `pip install needle-python` to add files.N!NeedleClient must be initialized.)nameurl)r r") needle.v1.modelsr$rrritems collectionsr"addr )rr"r$r&r' files_to_adds r add_filesNeedleLoader.add_files\s  2 {{&K(KK&GL{{}U})$ t5} U %%)),,L *  O  VsB B% B"c fUR5 URcS5eURRRR UR 5nUVs/sHBnUR S:XdM[SURUR[USS5S.S9PMD nnU$s snf)z Fetches metadata for documents from the Needle collection. Returns: List[Document]: A list of documents with metadata. Content is excluded. Raises: ValueError: If the collection is not properly initialized. Nr%indexedsize)sourcetitler2) page_contentmetadata) rrr*r"listr statusrr'r&getattr)rr"filedocss r_fetch_documentsNeedleLoader._fetch_documentsxs {{&K(KK& ''--2243E3EF  {{i' H"hh!YY#D&$7      s "B.91B.c"UR5$)zx Loads all documents from the Needle collection. Returns: List[Document]: A list of documents from the collection. r<rs rloadNeedleLoader.loads$$&&r!c#@# UR5ShvN gN7f)zu Lazily loads documents from the Needle collection. Yields: Iterator[Document]: An iterator over the documents. Nr?rs r lazy_loadNeedleLoader.lazy_loads((***s )rr r )NN)rN)__name__ __module__ __qualname____firstlineno____doc__rstrrrrr-rrr<r@rrC__static_attributes__ __classcell__)rs@rr r s!J)-'+ @   @ } @  @ @D @ tCH~ $ 8$x.8'd8n'+8H-++r!r N) typingrrrr$langchain_core.document_loaders.baserlangchain_core.documentsrr r!rrQs11;-]+:]+r!