dh<SSKJr SSKrSSKrSSKJrJrJrJr SSK r SSK J r SSK Jr SSKJr SSKJr SSKJr \R*"\5r"S S \5r"S S \5rg) ) annotationsN)AnyListOptionalTuple)Document) Embeddings) VectorStore)Self)maximal_marginal_relevancecF\rSrSrSrSSSjjrSrg)VikingDBConfigavikingdb connection config See the following documentation for details: https://www.volcengine.com/docs/6459/1167770 Attribute: host(str):The access address of the vector database server that the client needs to connect to. region(str):"cn-shanghai" or "cn-beijing" ak(str):Access Key ID, security credentials for accessing Volcano Engine services. sk(str):Secret Access Key, security credentials for accessing Volcano Engine services. scheme(str):http or https, defaulting to http. c@XlX lX0lX@lXPlgN)hostregionakskscheme)selfrrrrrs a/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/vectorstores/vikingdb.py__init__VikingDBConfig.__init__#s   )rrrrrN)rrrrhttp) rstrrrrrrrrrreturnNone)__name__ __module__ __qualname____firstlineno____doc__r__static_attributes__rrrrs\$            rrc\rSrSrSrSSSjjr\SSj5rSSSjjrSSjr SSSjjr SSS jjr SSS jjr SSS jjr SSS jjrSSS jjrSS SjjrSS!Sjjr\S"S#Sjj5rSrg)$VikingDB2zvikingdb as a vector store In order to use this you need to have a database instance. See the following documentation for details: https://www.volcengine.com/docs/6459/1167774 Nc @SSKJnJn XlX lSUlX0lX@lXPl U"URURURURUR5UlURR!U5n XlSUlUR$b5URR)UR UR 5UlU(a[+UR$U5(asURR-U5n U H(n URR/X+R 5 M* URR1U5 SUlSUlggg![a [S5ef=f!["a Sn GNf=f)Nr) CollectionVikingDBServicefCould not import volcengine python package. Please install it with `pip install --upgrade volcengine`.LangChainIndex)volcengine.viking_dbr+r, ImportErrorembedding_funccollection_name index_nameconnection_args index_paramsdrop_oldrrrrrserviceget_collection Exception collectionindex get_index isinstance list_indexes drop_indexdrop_collection) rembedding_functionr2r4r5r6kwargsr+r,colindexesr;s rrVikingDB.__init__:sf  H 1.*.( &   " "        " "    ,,--o>C ?? &//0D0DdooVDJ  4??J??ll//@G  ''9I9IJ! LL ( ( 9"DODJ @89 M  * C sE43F 4F  FFcUR$r)r1)rs r embeddingsVikingDB.embeddingsis"""rcSSKJnJn [ US5n/nU(GaUSR 5GHupx[ U[5(a#URU"XtR55 M>[ U[5(a#URU"XtR55 Mv[ U[5(a#URU"XtR55 M[ U[5(a:[SU55(a#URU"XtR 55 M[ U[5(a;[SU55(a$URU"XtR"55 GMM[ U[$5(a$URU"XtR&55 GM[)S5e URU"SUR&55 URU"SURS S 95 URU"S UR*US 95 UR,R/UR0U5Ulg![a [S5ef=f) Nr)Field FieldTyper-c3@# UHn[U[5v M g7fr)r=r.0items r .VikingDB._create_collection..56;dJtS))ec3@# UHn[U[5v M g7fr)r=intrMs rrPrQrRrSz>metadatas value is invalidplease change the type of metadatas.text primary_keyT)is_primary_keyvector)dim)r/rJrKr0lenitemsr=rappendStringrUInt64boolBoollistall List_String List_Int64bytesText ValueErrorVectorr7create_collectionr2r:) rrG metadatasrJrKrZfieldskeyvalues r_create_collectionVikingDB._create_collectionms  = *Q-  'l002 eS))MM%-=-=">?s++MM%__"=>t,,MM%^^"<=t,,56;522MM%-B-B"CDt,,56;522MM%-A-A"BCu--MM%^^"<=$X'3.  eFINN34 eM9+;+;DQR eHi&6&6C@A,,889M9MvVK M  s II.c >SSKJn SnU"5nSnSnURbURR S5bURSnURR S5bURSnURR S5bURSnURR S 5bURS nUR R URURUUUUS 9Ul g![a [S5ef=f) Nr)VectorIndexParamsr- cpu_quota vector_index partition_by scalar_index)rvrurwrx) r/rrr0r5getr7 create_indexr2r3r;)rrrrurvrwrxs r _create_indexVikingDB._create_indexs&  >  (*      (  $$[1= --k:   $$^4@#00@   $$^4@#00@   $$^4@#00@ \\..  OO%%% /  ' M  s DDc SSKJn [U5nURR U5n[U5S:Xa[RS5 /$URc!URXb5 UR5 /n/n [[U55Hn [[ R""55n U R%U 5 XU XjS.n UbHU [U5:a9[X*R'55n U HnX*R)U5X'M UR%U"U 55 M [U5n[SX5H2n[+UU-U5nUUUnURR-U5 M4 U $![a [S5ef=f![ a3 UVs/sHopRRU5PM Os snfnnGNf=f)zInsert text data into VikingDB.r)Datar-zNothing to insert, skipping.)rVrWrY)r/r~r0rbr1embed_documentsNotImplementedError embed_queryr[loggerdebugr:ror{rangeruuiduuid4r]keysrymin upsert_data)rtextsrk batch_sizerBr~rGxdatapksr;rWfieldnamesname total_countiend insert_datas r add_textsVikingDB.add_textss  1 U  M,,<Perform a search and return results that are reordered by MMR.)rk lambda_multrr&)r1r'max_marginal_relevance_search_by_vector)rrrrrrBrs rmax_marginal_relevance_search&VikingDB.max_marginal_relevance_search?sH''33E: ;; #     rc jURc[RS5 /$SnSnSnSn Ub\URS5bUSnURS5bUSnURS5bUSnURS5bUSn URR UUUUU S 9n /n /n U Hn S U R ;d S U R ;dS U R ;aM5U RU R RS 55 U R RS 5 U R RS 5n[XR S 9nU RU5 M [[R"U5XUS9n/nUH nUS:Xa U$URU U5 M" U$)rNrrrrrrrrrYrWrVr)rr) r:rrryr;rrlr]rrr nparray)rrrrrrBrrrrr documentsordered_result_embeddingsrOrr new_orderingrrs rr0VikingDB.max_marginal_relevance_search_by_vectorQs ?? " LL< =I   zz(#/)zz'".wzz/*6 & 7 zz+&2";/ jj)) ' *  $&!D +  3, % , ,T[[__X-F G KKOOM *;;??62L {{KC   S !2 HHY !:[ ABw  9Q<(   rc ~URc[RS5 URRU5 g)Nr)r:rr delete_data)ridsrBs rdeleteVikingDB.deletes- ?? " LL< = ##C(rc \Uc [S5eU"SUUUUUS.UD6n U RXS9 U $)z0Create a collection, indexes it and insert data.zVikingDBConfig does not exists)rAr2r4r5r6)rrkr&)r9r) clsrrr4rkr2r5r6rB vector_dbs r from_textsVikingDB.from_textssW  "<= = (++%     %=r) r:r2r4r6r1r;r3r5r7)LangChainCollectionNNF) rAr r2rr4Optional[VikingDBConfig]r5Optional[dict]r6zOptional[bool]rBr)rr r)rGrrkOptional[List[dict]]rr)rr)Ni) r List[str]rkrrrUrBrrr)rrrrrBrrList[Document])rrrrrBrrList[Tuple[Document, float]])r List[float]rrrBrrr)rrrrrBrrr)g?N) rrrrUrfloatrrrBrrr) rrrrUrrrrrBrrr)rzOptional[List[str]]rBrrr)NNrNF)rrrr r4rrkrr2rr5rr6r`rBrrr )r r!r"r#r$rpropertyrGror{rrrrrrrr classmethodrr%r&rrr(r(2s 548'+#( -&--2 - % - ! --^##CG*W*W+?*W *WX D+/ 22(2 2  2  2n"&''' '  '"&     & ""& ' ' ' '  '"&--- - & -d !%             * !% ;; ; ;  ;  ; ;~$() ))  ) 59*.4'+2  (   % rr() __future__rloggingrtypingrrrrnumpyrlangchain_core.documentsrlangchain_core.embeddingsr langchain_core.vectorstoresr typing_extensionsr &langchain_community.vectorstores.utilsr getLoggerr robjectrr(r&rrrsQ" ---03"M   8 $V@}{}r