dhdSSKJrJrJr SSKJr SSKJr SSKJ r SSK J r J r "SS\ \ 5r g) )AnyListOptional)CallbackManagerForRetrieverRun)Document) BaseRetriever) BaseModelFieldc\rSrSr%SrSr\\\S'\ "SSS9r \\ \S'\ "SS S9r \\ \S '\ "SS S 9r \\\S 'SSjrS\ S\\4SjrS\ S\S\\4SjrSrg)NeedleRetriever a NeedleRetriever retrieves relevant documents or context from a Needle collection based on a search query. 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]): The API key for authenticating with Needle. - `collection_id` (str): The ID of the Needle collection to search in. - `client` (Optional[NeedleClient]): An optional instance of the NeedleClient. - `top_k` (Optional[int]): Maximum number of results to return. Usage: .. code-block:: python from langchain_community.retrievers.needle import NeedleRetriever retriever = NeedleRetriever( needle_api_key="your-api-key", collection_id="your-collection-id", top_k=10 # optional ) results = retriever.retrieve("example query") for doc in results: print(doc.page_content) NclientzNeedle API Key) descriptionneedle_api_key.z,The ID of the Needle collection to search in collection_idz*Maximum number of search results to return)defaultrtop_kreturncSSKJn UR(dU"URS9Ulgg![a [S5ef=f)z} Initialize the NeedleClient with the provided API key. If a client instance is already provided, this method does nothing. r) NeedleClientz0Please install with `pip install needle-python`.)api_keyN) needle.v1r ImportErrorrr)selfrs ]/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/retrievers/needle.py_initialize_client"NeedleRetriever._initialize_client6sJ  R .{{&t/B/BCDK RPQ Q Rs 0AquerycUR5 URc [S5eURRR UR XR S9nUVs/sHn[URS9PM nnU$s snf)z Search the Needle collection for relevant documents. Args: query (str): The search query used to find relevant documents. Returns: List[Document]: A list of documents matching the search query. z4NeedleClient is not initialized. Provide an API key.)rtextr) page_content) rr ValueError collectionssearchrrrcontent)rrresultsresultdocss r_search_collection"NeedleRetriever._search_collectionDs ! ;; ST T++))00,,5 1 ELLG&fnn5GL Ms%B run_managerc$URU5$)z Retrieve relevant documents based on the query. Args: query (str): The query string used to search the collection. Returns: List[Document]: A list of documents relevant to the query. )r))rrr+s r_get_relevant_documents'NeedleRetriever._get_relevant_documentsXs&&u--)r)rN)__name__ __module__ __qualname____firstlineno____doc__rrr__annotations__r rstrrrintrrrr)rr-__static_attributes__r/rr r s D!FHSM ,$)$r?s'&&C-3%\.mY\.r/