dh:#SSKJr SSKrSSKJr SSKJrJrJrJ r J r J r J r SSK Jr SSKJr SSKJr "SS \5rg) ) annotationsN)repeat)AnyDictIterableListOptionalTupleType)Document) Embeddings) VectorStorec\rSrSrSrSSjr\SSj5rSSSjjrSSSjjr SSSjjr \ SSS jj5r \ SSS jj5rSSS jjrSSS jjrSS jrSSSjjrS SjrS!S"SjjrSrg)#XataVectorStore z`Xata` vector store. It assumes you have a Xata database created with the right schema. See the guide at: https://integrations.langchain.com/vectorstores?integration_name=XataVectorStore cSSKJn U"XS9UlX0lU=(d SUlg![a [S5ef=f)zInitialize with Xata client.r) XataClientzPCould not import xata python package. Please install it with `pip install xata`.)api_keydb_urlvectorsN) xata.clientr ImportError_client _embedding _table_name)selfrr embedding table_namers ]/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/vectorstores/xata.py__init__XataVectorStore.__init__sM  . "'A &/%2 =  s *AcUR$N)r)rs r embeddingsXataVectorStore.embeddings(s Nc&URXU5$r#) _add_vectors)rr documentsidss r add_vectorsXataVectorStore.add_vectors,s   S99r&c UnURX5nURR[U55nUR XeU5$r#)_texts_to_documentsrembed_documentslistr+)rtexts metadatasr*kwargsdocsrs r add_textsXataVectorStore.add_texts4sD''9//11$u+>s33r&c,/n[U5HfupVX%RUS.nU(aX5US'X%RR5HupUS;dM XU'M UR U5 Mh Sn /n [ S[ U5U 5H~n XLX-n URR5RURSU 05nURS:wa[SURS U35eU RUS 5 M U $) z!Add vectors to the Xata database.)contentrid)r9r8rirrecordszError adding vectors to Xata:  recordIDs) enumerate page_contentmetadataitemsappendrangelenrr: bulk_insertr status_code Exceptionextend)rrr)r*rowsidxrrowkeyval chunk_sizeid_listichunkrs rr(XataVectorStore._add_vectorsAs&('0NC$>66&CHD %N3399;<<"H< KK 1 q#d)Z0AQ^,E $$&2243C3CiQVEWXA}}#"@qQRPS TUU NN1[> * 1r&c xUc [05n[X5VVs/sHup#[X#S9PM nnnU$s snnf)z:Return list of Documents from list of texts and metadatas.r?r@)rzipr )r1r2textr@r4s rr.#XataVectorStore._texts_to_documentscsM  r I#&e"7 "7 $ :"7    s6c U(aU(d [S5eURU5n SnURX5n U"UUUUS9n U RXU5 U $)z9Return VectorStore initialized from texts and embeddings.z$Xata api_key and db_url must be set.N)rrrr) ValueErrorr/r.r() clsr1rr2rrrr*r3r$r4 vector_dbs r from_textsXataVectorStore.from_textssshfCD D..u5 &&u8!   z5r&c VURXUS9nUVs/sHofSPM nnU$s snf)zReturn docs most similar to query. Args: query: Text to look up documents similar to. k: Number of Documents to return. Defaults to 4. Returns: List of Documents most similar to the query. )filterr)similarity_search_with_score)rquerykr`r3docs_and_scoresdr)s rsimilarity_search!XataVectorStore.similarity_searchs:;;EV;T#23?aqT? 34s &c URRU5nUSUS.nU(aX6S'URR5R UR US9nUR S:wa[SUR SU35eUSnUV s/sH'n [U S URU 5S 9U S S 4PM) n n U $s sn f) a\Run similarity search with Chroma with distance. Args: query (str): Query text to search for. k (int): Number of results to return. Defaults to 4. filter (Optional[dict]): Filter by metadata. Defaults to None. Returns: List[Tuple[Document, float]]: List of documents most similar to the query text with distance in float. r) queryVectorcolumnsizer`payloadr;z!Error running similarity search: r<r:r8rUxatascore) r embed_queryrdata vector_searchrrFrGr _extractMetadata) rrbrcr`r3rrmrRhitshitrds rra,XataVectorStore.similarity_search_with_scoresOO//6 $!   &H  LL    - -d.>.> - P ==C ? aPQsST T|   !$Y!2237F G$       s-CcT0nUR5Hup4US;dM XBU'M U$)z:Extract metadata from a record. Filters out known columns.)r9r8rrn)rA)rrecordr@rLrMs rrs XataVectorStore._extractMetadatas0 HC@@ # 'r&c TU(a UR5 URSS9 gUboSn[S[U5U5HRnXXT-nUVs/sHnSURUS.0PM nnUR R 5RSU0S9 MT g[S 5es snf) znDelete by vector IDs. Args: ids: List of ids to delete. delete_all: Delete all records in the table. r)ndocsNideletetabler9 operationsrlz%Either ids or delete_all must be set.) _delete_allwait_for_indexingrCrDrrr: transactionrZ) rr* delete_allr3rNrPrQr9rs rr|XataVectorStore.deletes       " " " + _J1c#h 3/PUPU"X)9)9DEPU $$&22L*;U2V 4DE E sB%cURR5RURSS/0S9nURS:wa[ SURSU35eUSVs/sHo"SPM nn[ U5S:Xag UVs/sHnS URUS .0PM nnURR5RS U0S9 Ms snfs snf) z Delete all records in the table.columnsr9rlr;zError running query: r<r:rr|r}rN) rrqrbrrFrGrDr:r)rrRrecr*r9rs rrXataVectorStore._delete_alls !!#))$*:*:YQUPVDW)XA}}#"7 as KLL()) 5 t9 C53x1}LOLObT%5%5R@AC  LL " . . j7Q . R6s ) C Cc[R"5nURR5RURSSS0S.S9nUR S:wa[ SUR SU35eUS U:Xag [R"5U- U:a [ S 5e[R"S 5 M) zUWait for the search index to contain a certain number of documents. Useful in tests. rkr)rbpagerlr;zError running search: r< totalCountz+Timed out waiting for indexing to complete.g?N)timerrq search_tablerrFrGsleep)rtimeoutr{startrRs rr!XataVectorStore.wait_for_indexings  !!#00  B *L1A}}#"8q LMM%'yy{U"W, MNN JJsOr&)rrr) rstrrrrr rrreturnNone)rr r#)rzList[List[float]]r)List[Document]r*Optional[List[str]]r List[str])NN) r1 Iterable[str]r2zOptional[List[Dict[Any, Any]]]r*rr3rrr)r1rr2z"Optional[Iterable[Dict[Any, Any]]]rr)NNNrN)r[zType['XataVectorStore']r1rrr r2zOptional[List[dict]]r Optional[str]rrrrr*rr3rrz'XataVectorStore')N) rbrrcintr`Optional[dict]r3rrr) rbrrcrr`rr3rrzList[Tuple[Document, float]])rxdictrr)r*rrzOptional[bool]r3rrr)rr))rfloatr{rrr)__name__ __module__ __qualname____firstlineno____doc__r propertyr$r+r5r( staticmethodr. classmethodr]rfrarsr|rr__static_attributes__r&rrr s\333 3  3  3&$( :":":! :  :59#' 4 42 4! 4  4  4"$( " " !   D9=  5     +/!% $##' $(     ! >@D .<OR "@D$$ $.<$OR$ %$L$(%)F F#F F  F4 Sr&r) __future__rr itertoolsrtypingrrrrr r r langchain_core.documentsr langchain_core.embeddingsr langchain_core.vectorstoresrrrr&rrs." CCC-03{k{r&