dhm/SSKJr SSKrSSKrSSKJrJrJrJrJ r SSK r SSK J r SSKJr SSKJr \R$"\5r"SS\5rg) ) annotationsN)AnyIterableListOptionalType)Document) Embeddings) VectorStorecx\rSrSr%SrSrS\S'SSSjjr\SSj5r SSS jjr SS jr SSS jjr \ SSS jj5r\ SSS jj5rSrg)AtlasDBa`Atlas` vector store. Atlas is the `Nomic's` neural database and `rhizomatic` instrument. To use, you should have the ``nomic`` python package installed. Example: .. code-block:: python from langchain_community.vectorstores import AtlasDB from langchain_community.embeddings.openai import OpenAIEmbeddings embeddings = OpenAIEmbeddings() vectorstore = AtlasDB("my_project", embeddings.embed_query) atlas_idstr_ATLAS_DEFAULT_ID_FIELDNc (SSKnSSKJn Uc [S5eUR U5 X lSn UR bSn U"UUU UU[ RS9UlURR5 g![a [S5ef=f) a Initialize the Atlas Client Args: name (str): The name of your project. If the project already exists, it will be loaded. embedding_function (Optional[Embeddings]): An optional function used for embedding your data. If None, data will be embedded with Nomic's embed model. api_key (str): Your nomic API key description (str): A description for your project. is_public (bool): Whether your project is publicly accessible. True by default. reset_project_if_exists (bool): Whether to reset this project if it already exists. Default False. Generally useful during development and testing. rN) AtlasProjectzRCould not import nomic python package. Please install it with `pip install nomic`.z/No API key provided. Sign up at atlas.nomic.ai!text embedding)name descriptionmodality is_publicreset_project_if_existsunique_id_field) nomicr ImportError ValueErrorlogin_embedding_functionr rproject_latest_project_state) selfrembedding_functionapi_keyrrrrrrs ^/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/vectorstores/atlas.py__init__AtlasDB.__init__"s4   * ?NO O G#5   # # /"H$#$;#;;    **,1 >  s A;;BcUR$)N)r )r#s r& embeddingsAtlasDB.embeddingsYs'''c Ub1[U5S:a"SUSR5;a [S5e[U5nUc-UVs/sH!n[ [ R "55PM# nnURGb URRU5n[R"U5nUc5[U5V Vs/sHup[RX9SX0PM n n nO=[[U55H#n X9X)[R'XX)S'M% Un URR!/[RU S9 URR#5 URR%XS9 SSS5 OUc3[U5V V s/sHupSU [RX90PM n n n O4[U5H#upXU S'X9X)[R'M% Un URR!/[RU S9 URR#5 URR'U 5 SSS5 U(ab[URR(5S:a?URR#5 URR+5 SSS5 U$U$s snfs snn f!,(df  N=fs sn n f!,(df  N=f!,(df  U$=f)aRun more texts through the embeddings and add to the vectorstore. Args: texts (Iterable[str]): Texts to add to the vectorstore. metadatas (Optional[List[dict]], optional): Optional list of metadatas. ids (Optional[List[str]]): An optional list of ids. refresh(bool): Whether or not to refresh indices with the updated data. Default True. Returns: List[str]: List of IDs of the added texts. Nrrz#Cannot accept key text in metadata!)id_fielddata)r*r/)lenkeysrlistruuiduuid4r embed_documentsnpstack enumerater rranger!_validate_map_data_inputswait_for_project_lockadd_embeddingsadd_textindices rebuild_maps) r#texts metadatasidsrefreshkwargs_ _embeddingsr*ir/rs r& add_textsAtlasDB.add_texts]s*  !I")A,++--BC CU  ;.34e3tzz|$eC4  # # /22BB5IK+.J !*% 0 044cffehO 0 s9~.ADGFIL!@!@A+08IL(/! LL 2 2W<<4 3 335 ++z+M65 $-U#3#3T7#B#BCFK#3 )/GA+0aL(DGFIL!@!@A 0! LL 2 2W<<4 3 335 %%d+6 4<<''(1,\\779LL--/: s _565 65 :9 s6'K#K 'K!K"7K(K9 K( K69 Lc URR5 URR"S0UD6sSSS5 $!,(df  g=f)zCreates an index in your project. See https://docs.nomic.ai/atlas_api.html#nomic.project.AtlasProject.create_index for full detail. N)r!r; create_index)r#rDs r&rLAtlasDB.create_indexs5\\ / / 1<<,,6v62 1 1s A Ac URc [S5eURRU/5Sn[R"U5R SS5nUR R5 UR RSRXRS9upgUR RUSS9nSSS5 [W5V V s/sHup[WU SXS 9PM n n n U $!,(df  N>=fs sn n f) zRun similarity search with AtlasDB Args: query (str): Query text to search for. k (int): Number of results to return. Defaults to 4. Returns: List[Document]: List of documents most similar to the query text. NzBAtlasDB requires an embedding_function for text similarity search!r)queriesk)rBr) page_contentmetadata) r NotImplementedErrorr5r6arrayreshaper!r; projections vector_searchget_datar8r ) r#queryrRrD _embeddingr neighborsrEr/rGneighbordocss r&similarity_searchAtlasDB.similarity_searchs  # # +%T --==ugFqI HHZ(00B7 \\ / / 1<<33A6DD!ELI<<((Yq\(:D 2 )3 3  $q'&/DG D3   2 1  s8AC5D5 Dc <UbUc [S5eUS-SS.n U bU R5H upXU 'M U"UUUSUU S9nURR5 UR XUS9 UR "S 0U D6 SSS5 U$!,(df  U$=f) aCreate an AtlasDB vectorstore from a raw documents. Args: texts (List[str]): The list of texts to ingest. name (str): Name of the project to create. api_key (str): Your nomic API key, embedding (Optional[Embeddings]): Embedding function. Defaults to None. metadatas (Optional[List[dict]]): List of metadatas. Defaults to None. ids (Optional[List[str]]): Optional list of document IDs. If None, ids will be auto created description (str): A description for your project. is_public (bool): Whether your project is publicly accessible. True by default. reset_project_if_exists (bool): Whether to reset this project if it already exists. Default False. Generally useful during development and testing. index_kwargs (Optional[dict]): Dict of kwargs for index creation. See https://docs.nomic.ai/atlas_api.html Returns: AtlasDB: Nomic's neural database and finest rhizomatic instrument N$`name` and `api_key` cannot be None._indexr)r indexed_fieldA description for your project)r$r%rrr)r@rArBrK)ritemsr!r;rHrL)clsr@rrArBrr%rrr index_kwargsrDall_index_kwargsrRvatlasDBs r& from_textsAtlasDB.from_textssJ <7?CD D%)8OfM  #$**,&'#- (8$;  __ 2 2 4   EC  H  4#3 455 4s #B  Bc UbUc [S5eUV s/sHoRPM n n UV s/sHoRPM nn URUUU UUUUUU U S9 $s sn fs sn f)aCreate an AtlasDB vectorstore from a list of documents. Args: name (str): Name of the collection to create. api_key (str): Your nomic API key, documents (List[Document]): List of documents to add to the vectorstore. embedding (Optional[Embeddings]): Embedding function. Defaults to None. ids (Optional[List[str]]): Optional list of document IDs. If None, ids will be auto created description (str): A description for your project. is_public (bool): Whether your project is publicly accessible. True by default. reset_project_if_exists (bool): Whether to reset this project if it already exists. Default False. Generally useful during development and testing. index_kwargs (Optional[dict]): Dict of kwargs for index creation. See https://docs.nomic.ai/atlas_api.html Returns: AtlasDB: Nomic's neural database and finest rhizomatic instrument rc) rr%r@rrArBrrrri)rrSrTrm)rh documentsrrBrr%persist_directoryrrrrirDdocr@rAs r&from_documentsAtlasDB.from_documentssH <7?CD D-67Yc!!Y7-67Yc\\Y 7~~#$;%  87s AA#)r r!)NNrfTF)rrr$Optional[Embeddings]r% Optional[str]rrrboolrrwreturnNone)rxru)NNT) r@z Iterable[str]rAOptional[List[dict]]rBOptional[List[str]]rCrwrDrrx List[str])rDrrxr))r[rrRintrDrrxList[Document]) NNNNNrfTFN)rh Type[AtlasDB]r@r|rrurArzrBr{rrvr%rvrrrrwrrwriOptional[dict]rDrrxr )rhrrprrrurBr{rrvr%rvrqrvrrrrwrrwrirrDrrxr )__name__ __module__ __qualname____firstlineno____doc__r__annotations__r'propertyr*rHrLr` classmethodrmrs__static_attributes__rKr,r&r r sj $.S- 48!%;(-5-5-15- 5-  5-  5-"&5- 5-n(( +/#' LL(L! L  L  L L\7       D+/*.#'"!%;(-'+9 99(9( 9 ! 9  9999"&9%99 99v+/#'"!%+/;(-'+2 2 !2 (2 ! 2  2  2 )2 2 2 "&2 %2 2  2 2 r,r ) __future__rloggingr3typingrrrrrnumpyr6langchain_core.documentsr langchain_core.embeddingsr langchain_core.vectorstoresr getLoggerrloggerr rKr,r&rs>" 66-03   8 $w kw r,