book_rag_server/test/test_infinity_search.py

import os
import sys
# 添加项目根目录到Python路径
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from utils.infinity import InfinityClient
# import infinity

output_columns = [
                    "file_name",
                    "page_number",
                    "content",
                    "image_path",
                    "dataset_id",
                    "document_id"
                ]
vec = [
                -0.009083136916160583,
                -0.04671327769756317,
                -0.08419924229383469,
                0.008182032033801079,
                -0.03806266933679581,
                0.018454628065228462,
                0.010813258588314056,
                -0.017445391044020653,
                0.09746350347995758,
                0.06949320435523987,
                -0.07180003821849823,
                -0.022635754197835922,
                0.02465422824025154,
                -0.009731932543218136,
                -0.030565477907657623,
                0.017157036811113358,
                0.11303459107875824,
                0.04671327769756317,
                0.004054971970617771,
                -0.03316066041588783,
                -0.0009776988299563527,
                -0.03604419529438019,
                0.044694799929857254,
                -0.06113095581531525,
                0.06805144250392914,
                -0.054787177592515945,
                0.0338815413415432,
                -0.03143053874373436,
                -0.0177337434142828,
                -0.04988516494631767,
                -0.04642492160201073,
                -0.059400834143161774,
                0.0152106499299407,
                -0.05334540829062462,
                -0.01312008686363697,
                0.0006307734292931855,
                0.03618837147951126,
                -0.06372613459825516,
                0.06141930818557739,
                -0.013985147699713707,
                0.0010092374868690968,
                0.005046187434345484,
                0.04036949947476387,
                -0.007821590639650822,
                -0.020040571689605713,
                0.03835102170705795,
                -0.003604419529438019,
                0.015282738953828812,
                0.020040571689605713,
                0.05507552996277809,
                0.05651729553937912,
                0.025375112891197205,
                -0.05046187341213226,
                -0.06372613459825516,
                0.01989639550447464,
                0.0014778119511902332,
                0.047866690903902054,
                0.03301648050546646,
                -0.033304836601018906,
                -0.01787792146205902,
                0.04152291268110275,
                0.022347399964928627,
                0.006848397199064493,
                0.027393588796257973,
                -0.001243524719029665,
                -0.08765947818756104,
                0.009011048823595047,
                0.025807643309235573,
                -0.018598804250359535,
                -0.020040571689605713,
                -0.007677413523197174,
                -0.004289259202778339,
                -0.06257272511720657,
                -0.0031358448322862387,
                0.013624705374240875,
                0.005731027107685804,
                0.020040571689605713,
                0.04036949947476387,
                0.0338815413415432,
                -0.02364499121904373,
                0.004865966271609068,
                -0.02883535623550415,
                -0.023933345451951027,
                0.012615468353033066,
                -0.02768194116652012,
                -0.03719760850071907,
                -0.033304836601018906,
                0.032728128135204315,
                0.010308640077710152,
                -0.06257272511720657,
                0.06228436902165413,
                0.011822495609521866,
                0.07497192174196243,
                -0.01312008686363697,
                0.04354138672351837,
                -0.022347399964928627,
                0.003820684738457203,
                0.030709654092788696,
                0.021193986758589745,
                -0.03604419529438019,
                0.02205904759466648,
                0.02090563252568245,
                0.03503495827317238,
                -0.07554863393306732,
                -0.028691178187727928,
                -0.028547002002596855,
                0.04238797351717949,
                -0.04123456031084061,
                0.006271689664572477,
                0.024942582473158836,
                0.028691178187727928,
                -0.0177337434142828,
                0.016652418300509453,
                -0.003694529877975583,
                -0.010452816262841225,
                -0.021770693361759186,
                0.004145082551985979,
                -0.005226408131420612,
                -0.03979279100894928,
                -0.01384097058326006,
                -0.012255026027560234,
                0.014489766210317612,
                -0.045271508395671844,
                -0.0033701322972774506,
                -0.01283173356205225,
                0.019608041271567345,
                0.029700417071580887,
                0.0041090380400419235,
                0.009443579241633415,
                -0.01888715848326683,
                0.04757833853363991,
                -0.011678319424390793,
                -0.043253034353256226,
                -0.04209962114691734,
                0.012399203144013882,
                0.05075022578239441,
                -0.013985147699713707,
                0.0169407706707716,
                0.035900019109249115,
                -0.052191995084285736,
                0.03835102170705795,
                -0.005478717386722565,
                0.040081143379211426,
                0.026816880330443382,
                -0.0354674868285656,
                0.009083136916160583,
                0.010596993379294872,
                -0.028691178187727928,
                -0.0008470385801047087,
                0.04382973909378052,
                0.03258395195007324,
                -0.0240775216370821,
                -0.031718891113996506,
                -0.024798406288027763,
                0.051326934248209,
                0.004343325272202492,
                0.014273501001298428,
                -0.021338162943720818,
                0.007136750500649214,
                0.0304212998598814,
                0.027393588796257973,
                0.02205904759466648,
                0.014561854302883148,
                0.027393588796257973,
                0.013768882490694523,
                -0.0030637565068900585,
                0.05507552996277809,
                0.042676325887441635,
                0.00275738094933331,
                0.051615286618471146,
                0.03243977576494217,
                0.01650824025273323,
                0.014994384720921516,
                0.03143053874373436,
                -0.04555986076593399,
                -0.011822495609521866,
                -0.0193196889013052,
                -0.00854247435927391,
                0.06228436902165413,
                -0.004181126598268747,
                0.02249157801270485,
                0.008218076080083847,
                0.016796594485640526,
                -0.05305705592036247,
                -0.009083136916160583,
                -0.044694799929857254,
                -0.0017931986367329955,
                0.012615468353033066,
                0.03604419529438019,
                -0.005911247804760933,
                -0.011462054215371609,
                -0.07843216508626938,
                0.020328925922513008,
                -0.005370584782212973,
                0.001351657323539257,
                -0.0048299222253263,
                0.0709349736571312,
                0.060554247349500656,
                0.0029015576001256704,
                -0.06516790390014648,
                0.003892773063853383,
                -0.008001810871064663,
                0.006235645618289709,
                0.009227314032614231,
                -0.0418112650513649,
                -0.04094620421528816,
                -0.01412932388484478,
                0.011317877098917961,
                -0.04354138672351837,
                -0.044694799929857254,
                -0.02768194116652012,
                -0.0033521100413054228,
                -0.029700417071580887,
                0.004271237179636955,
                0.012975909747183323,
                -0.028547002002596855,
                -0.0193196889013052,
                -0.051615286618471146,
                -0.02667270414531231,
                -0.02191486954689026,
                -0.027970295399427414,
                -0.04728998243808746,
                0.007425104267895222,
                -0.05594059079885483,
                -0.02350081503391266,
                0.022779930382966995,
                0.0386393778026104,
                0.001270557870157063,
                -0.0018562760669738054,
                0.025951819494366646,
                -0.0256634671241045,
                -0.030709654092788696,
                -0.017950009554624557,
                -0.02624017372727394,
                0.0963100865483284,
                -0.0481550432741642,
                -0.0064879548735916615,
                -0.02205904759466648,
                -0.03416989743709564,
                -0.0016580329975113273,
                0.0338815413415432,
                0.008830827660858631,
                -0.00785763468593359,
                0.009731932543218136,
                0.011462054215371609,
                0.02782611921429634,
                0.01708494871854782,
                -0.006956529803574085,
                -0.026528527960181236,
                -0.019175510853528976,
                -0.017229124903678894,
                0.011101611889898777,
                -0.015859445556998253,
                -0.037485960870981216,
                -0.03301648050546646,
                -0.03676507994532585,
                0.025951819494366646,
                0.016003621742129326,
                -0.015282738953828812,
                0.001162425265647471,
                0.026961058378219604,
                0.002451005158945918,
                0.017589567229151726,
                0.03892773017287254,
                -0.024221699684858322,
                0.033304836601018906,
                0.005875203758478165,
                0.01888715848326683,
                -0.002649248344823718,
                0.017373301088809967,
                -0.04238797351717949,
                -0.03921608254313469,
                0.009371490217745304,
                0.0019553976599127054,
                0.0009506656206212938,
                -0.05075022578239441,
                0.03243977576494217,
                -0.026095997542142868,
                0.049020104110240936,
                -0.007749502081423998,
                -0.01989639550447464,
                0.018166273832321167,
                -0.051615286618471146,
                -0.02782611921429634,
                0.03402572125196457,
                -0.08881289511919022,
                -0.10150045156478882,
                -0.0045055244117975235,
                0.00029060631641186774,
                -0.059400834143161774,
                0.013552617281675339,
                0.0418112650513649,
                0.01629197597503662,
                0.019752219319343567,
                -0.03575584292411804,
                0.028691178187727928,
                -0.02090563252568245,
                -0.007785546127706766,
                -0.07958558201789856,
                -0.015426915138959885,
                0.026961058378219604,
                -0.0096598444506526,
                -0.0027393586933612823,
                -0.04036949947476387,
                0.022924108430743217,
                -0.02364499121904373,
                0.04152291268110275,
                -0.043253034353256226,
                0.015571092255413532,
                -0.049020104110240936,
                0.010741169564425945,
                0.019752219319343567,
                0.034314073622226715,
                0.005226408131420612,
                0.08073899894952774,
                -0.03186306729912758,
                0.05824741721153259,
                -0.0026852923911064863,
                0.007677413523197174,
                -0.010524905286729336,
                0.012975909747183323,
                0.01629197597503662,
                -0.0014417677884921432,
                0.009047092869877815,
                -0.0014688009396195412,
                0.012182937934994698,
                0.02768194116652012,
                0.01989639550447464,
                0.02926788665354252,
                0.019752219319343567,
                -0.028402825817465782,
                0.05882412567734718,
                0.006127513013780117,
                0.033304836601018906,
                0.016796594485640526,
                -0.022203223779797554,
                0.012687556445598602,
                -0.018454628065228462,
                0.020328925922513008,
                0.009804020635783672,
                -0.011317877098917961,
                0.03460242599248886,
                -0.034314073622226715,
                0.02364499121904373,
                -0.05882412567734718,
                -0.0043973918072879314,
                -0.02350081503391266,
                0.009119180962443352,
                -0.011389965191483498,
                -0.04700163006782532,
                0.047866690903902054,
                0.020184749737381935,
                0.026961058378219604,
                -0.05046187341213226,
                -0.026384349912405014,
                0.03517913445830345,
                -0.02926788665354252,
                0.002613204065710306,
                0.06516790390014648,
                0.014201412908732891,
                -0.00228880625218153,
                0.01283173356205225,
                -0.03892773017287254,
                -0.00032439775532111526,
                -0.006560043431818485,
                0.0338815413415432,
                -0.005767071153968573,
                -0.02724941074848175,
                0.009191269986331463,
                0.04988516494631767,
                0.01708494871854782,
                0.001139897620305419,
                0.04152291268110275,
                0.01730121299624443,
                -0.004685745108872652,
                -0.007353015709668398,
                0.029700417071580887,
                0.017012860625982285,
                0.04123456031084061,
                -0.017445391044020653,
                -0.03777431696653366,
                -0.0304212998598814,
                -0.015354827046394348,
                -0.02148233912885189,
                0.01412932388484478,
                -0.02350081503391266,
                -0.0096598444506526,
                0.028979532420635223,
                0.00016332525410689414,
                0.037485960870981216,
                -0.05565223842859268,
                0.028114471584558487,
                -0.0018382539274170995,
                0.013336352072656155,
                0.003676507854834199,
                -0.01391305960714817,
                -0.01730121299624443,
                0.03157471492886543,
                -0.01888715848326683,
                -0.0031358448322862387,
                -0.019752219319343567,
                -0.007893678732216358,
                0.0772787556052208,
                0.023356638848781586,
                0.024798406288027763,
                0.0338815413415432,
                0.03359318897128105,
                0.016003621742129326,
                -0.007641369476914406,
                0.009299402125179768,
                -0.005154320038855076,
                0.006343778222799301,
                0.05911247804760933,
                0.00915522500872612,
                0.0031178228091448545,
                0.013192174956202507,
                -0.0152106499299407,
                0.002541115740314126,
                -0.051326934248209,
                -0.010308640077710152,
                -0.021770693361759186,
                0.015643181279301643,
                -0.0544988214969635,
                -0.019752219319343567,
                0.0029376018792390823,
                0.049020104110240936,
                -0.05507552996277809,
                -0.05075022578239441,
                -0.02667270414531231,
                0.024221699684858322,
                -0.07958558201789856,
                -0.015499004162847996,
                -0.006848397199064493,
                -0.004379369784146547,
                0.04700163006782532,
                0.03085383027791977,
                -0.03416989743709564,
                0.05911247804760933,
                0.02523093670606613,
                -0.008830827660858631,
                0.01412932388484478,
                -0.010596993379294872,
                0.012471291236579418,
                -0.023212460801005363,
                -0.0386393778026104,
                -0.028979532420635223,
                -0.03359318897128105,
                0.02782611921429634,
                -0.010885346680879593,
                0.04700163006782532,
                -0.027537764981389046,
                -0.024221699684858322,
                -0.03892773017287254,
                0.025807643309235573,
                -0.05911247804760933,
                -0.02047310210764408,
                -0.007497192360460758,
                0.03143053874373436,
                0.033304836601018906,
                0.02249157801270485,
                0.003586397273465991,
                0.030709654092788696,
                -0.007533236872404814,
                -0.07237674295902252,
                0.0041991486214101315,
                -0.01571526937186718,
                0.01989639550447464,
                0.05968918651342392,
                0.036909256130456924,
                0.009227314032614231,
                -0.0027934249956160784,
                -0.013264263980090618,
                0.002378916833549738,
                -0.02249157801270485,
                0.00255913776345551,
                -0.05968918651342392,
                -0.05651729553937912,
                -0.01730121299624443,
                -0.002613204065710306,
                0.023212460801005363,
                -0.06660967320203781,
                -0.01492229662835598,
                -0.010524905286729336,
                -0.05882412567734718,
                0.0044334358535707,
                0.010164462961256504,
                0.004757833667099476,
                -0.05968918651342392,
                0.07295344769954681,
                0.015643181279301643,
                -0.01989639550447464,
                0.0177337434142828,
                0.008254120126366615,
                -0.014057235792279243,
                0.05911247804760933,
                -0.028979532420635223,
                0.028547002002596855,
                0.009876109659671783,
                0.026528527960181236,
                0.03618837147951126,
                0.002171662636101246,
                -0.006632131990045309,
                0.039504438638687134,
                0.040081143379211426,
                -0.05594059079885483,
                0.026816880330443382,
                -0.032728128135204315,
                0.05795906484127045,
                -0.03979279100894928,
                -0.0065960874781012535,
                -0.02523093670606613,
                0.02350081503391266,
                -0.033737365156412125,
                0.0054426733404397964,
                0.020328925922513008,
                -0.010164462961256504,
                0.036332547664642334,
                0.059400834143161774,
                0.028979532420635223,
                -0.04671327769756317,
                -0.005190364085137844,
                0.024510052055120468,
                -0.021338162943720818,
                0.0019734196830540895,
                0.022347399964928627,
                0.025375112891197205,
                0.008722695522010326,
                -0.03460242599248886,
                -0.012543379329144955,
                -0.008975004777312279,
                -0.0003672002349048853,
                -0.016003621742129326,
                -0.03344901278614998,
                -0.04700163006782532,
                -0.00692048529163003,
                0.009011048823595047,
                0.011462054215371609,
                -0.014778119511902332,
                0.012975909747183323,
                -0.0386393778026104,
                0.0084703853353858,
                -0.028691178187727928,
                0.0022347401827573776,
                -0.014489766210317612,
                0.02249157801270485,
                0.0354674868285656,
                -0.0015048451023176312,
                -0.021770693361759186,
                0.032295599579811096,
                -0.026384349912405014,
                0.0027393586933612823,
                0.012687556445598602,
                -0.0032619996927678585,
                0.02364499121904373,
                0.0048299222253263,
                -0.05017351731657982,
                -0.004721789620816708,
                -0.015138561837375164,
                -0.02667270414531231,
                0.04238797351717949,
                -0.030277123674750328,
                0.05911247804760933,
                0.02205904759466648,
                -0.04296467825770378,
                -0.03979279100894928,
                -0.015138561837375164,
                -0.0481550432741642,
                -0.05911247804760933,
                -0.035323310643434525,
                -0.0003469253715593368,
                -0.016796594485640526,
                0.007713457569479942,
                0.0640144869685173,
                0.010524905286729336,
                0.02205904759466648,
                0.003406176343560219,
                -0.009119180962443352,
                -0.018598804250359535,
                0.03921608254313469,
                -0.005226408131420612,
                -0.05795906484127045,
                0.04555986076593399,
                -0.005370584782212973,
                0.060554247349500656,
                -0.0019193533807992935,
                -0.0003536836593411863,
                -0.029123708605766296,
                -0.006091468967497349,
                0.021626517176628113,
                0.007893678732216358,
                -0.08650606870651245,
                0.014345589093863964,
                -0.04728998243808746,
                -0.03662090003490448,
                0.035323310643434525,
                0.018454628065228462,
                -0.02205904759466648,
                0.035323310643434525,
                -0.022347399964928627,
                0.06199601665139198,
                -0.04988516494631767,
                0.03979279100894928,
                0.030132947489619255,
                -0.0069925738498568535,
                0.015138561837375164,
                0.015571092255413532,
                0.020328925922513008,
                -0.03445824980735779,
                0.003838706761598587,
                -0.011245789006352425,
                -0.00255913776345551,
                -0.003730574157088995,
                0.0005248935776762664,
                0.012759644538164139,
                -0.03416989743709564,
                0.039504438638687134,
                -0.06199601665139198,
                -0.016219887882471085,
                0.001459789928048849,
                0.013264263980090618,
                -0.0836225301027298,
                0.04152291268110275,
                -0.01095743477344513,
                0.014417678117752075,
                -0.013696794398128986,
                0.005478717386722565,
                -0.0016219887183979154,
                -0.030565477907657623,
                -0.0016760550206527114,
                0.014273501001298428,
                0.011606230400502682,
                0.025519289076328278,
                0.06430284678936005,
                -0.026095997542142868,
                0.03460242599248886,
                -0.01492229662835598,
                0.00731697166338563,
                -0.011173699982464314,
                -0.0449831560254097,
                0.007533236872404814,
                -0.05046187341213226,
                0.003730574157088995,
                -0.018598804250359535,
                0.02984459325671196,
                0.024798406288027763,
                0.019175510853528976,
                0.04382973909378052,
                -0.0034422206226736307,
                -0.024221699684858322,
                0.02523093670606613,
                -0.01946386508643627,
                -0.034746602177619934,
                0.008182032033801079,
                -0.044694799929857254,
                -0.0007839612080715597,
                -0.008398297242820263,
                -0.015354827046394348,
                -0.032728128135204315,
                0.02306828461587429,
                0.033737365156412125,
                0.03806266933679581,
                0.018022097647190094,
                -0.008578518405556679,
                -0.042676325887441635,
                -0.02205904759466648,
                -0.03604419529438019,
                -0.07180003821849823,
                0.0481550432741642,
                0.01730121299624443,
                -0.01492229662835598,
                -0.006019380409270525,
                0.014706031419336796,
                0.018166273832321167,
                -0.001171436277218163,
                -0.0043973918072879314,
                -0.07439521700143814,
                0.028979532420635223,
                0.06949320435523987,
                -0.03085383027791977,
                -0.008326209150254726,
                0.020617280155420303,
                -0.014778119511902332,
                0.004253215156495571,
                0.0418112650513649,
                0.01463394332677126,
                0.016147799789905548,
                -0.030277123674750328,
                0.0193196889013052,
                0.02883535623550415,
                0.003694529877975583,
                0.022347399964928627,
                -0.07151168584823608,
                0.02926788665354252,
                -0.008686650544404984,
                0.02306828461587429,
                0.03186306729912758,
                -0.0016670440090820193,
                -0.03561166301369667,
                0.053633760660886765,
                -0.01607571169734001,
                -0.0013426461955532432,
                -0.0338815413415432,
                0.0029376018792390823,
                0.017373301088809967,
                -0.005586849991232157,
                -0.01283173356205225,
                0.003820684738457203,
                -0.0005767071270383894,
                0.06372613459825516,
                -0.02191486954689026,
                -0.0481550432741642,
                0.01283173356205225,
                0.0029015576001256704,
                -0.010380728170275688,
                -0.04411809518933296,
                -0.014057235792279243,
                0.029700417071580887,
                -0.05594059079885483,
                0.022635754197835922,
                -0.02306828461587429,
                0.008001810871064663,
                -0.052191995084285736,
                0.021770693361759186,
                0.0304212998598814,
                0.019608041271567345,
                0.02205904759466648,
                0.009299402125179768,
                0.04757833853363991,
                0.005586849991232157,
                -0.009371490217745304,
                0.045271508395671844,
                -0.03618837147951126,
                0.026816880330443382,
                -0.03200724348425865,
                -0.025807643309235573,
                -0.04642492160201073,
                0.01809418573975563,
                0.015787357464432716,
                -0.022635754197835922,
                0.009299402125179768,
                -0.013768882490694523,
                0.06026589497923851,
                0.0018562760669738054,
                -0.017589567229151726,
                -0.007425104267895222,
                -0.05046187341213226,
                0.005983336362987757,
                0.0,
                -0.00032214500242844224,
                -0.0033521100413054228,
                -0.04613656923174858,
                0.009731932543218136,
                -0.05968918651342392,
                0.07439521700143814,
                0.003910795319825411,
                0.024798406288027763,
                -0.03979279100894928,
                -0.005731027107685804,
                0.014561854302883148,
                -0.019031334668397903,
                0.03215142339468002,
                0.01989639550447464,
                0.04757833853363991,
                -0.006956529803574085,
                0.04411809518933296,
                -0.012903821654617786,
                -0.04094620421528816,
                0.014561854302883148,
                -0.0418112650513649,
                -0.01023655105382204,
                -0.00032214500242844224,
                -0.017012860625982285,
                -0.04757833853363991,
                -0.003640463575720787,
                -0.05507552996277809,
                0.03575584292411804,
                0.021626517176628113,
                -0.03517913445830345,
                -0.018742980435490608,
                -0.0019013312412425876,
                0.023789169266819954,
                -0.04382973909378052,
                0.03143053874373436,
                0.0067042200826108456,
                0.002469027414917946,
                -0.0354674868285656,
                -0.016652418300509453,
                0.07381851226091385,
                -0.02465422824025154,
                0.008830827660858631,
                -0.04152291268110275,
                0.0055508059449493885,
                -0.02104981057345867,
                0.045271508395671844,
                -0.026384349912405014,
                0.018382539972662926,
                0.03719760850071907,
                -0.026384349912405014,
                0.008650606498122215,
                0.010741169564425945,
                -0.012543379329144955,
                0.01607571169734001,
                -0.016364064067602158,
                0.0544988214969635,
                -0.0016219887183979154,
                -0.023212460801005363,
                -0.01989639550447464,
                -0.008434341289103031,
                0.01384097058326006,
                0.07612533867359161,
                -0.014201412908732891,
                0.03892773017287254,
                0.0772787556052208,
                0.034890782088041306,
                0.01809418573975563,
                -0.00814598798751831,
                -0.03402572125196457,
                0.03719760850071907,
                -0.09111972153186798,
                -0.03979279100894928,
                0.007965766824781895,
                -0.018598804250359535,
                -0.010380728170275688,
                -0.024221699684858322,
                0.0029556239023804665,
                0.037485960870981216,
                0.03085383027791977,
                -0.024942582473158836,
                -0.008182032033801079,
                -0.03921608254313469,
                0.04123456031084061,
                0.04440644755959511,
                -0.00014023444964550436,
                -0.05190363898873329,
                -0.0169407706707716,
                -0.003946839366108179,
                -0.049020104110240936,
                0.015643181279301643,
                -0.03301648050546646,
                0.06286107748746872,
                -0.01730121299624443,
                0.06199601665139198,
                0.015859445556998253,
                -0.016364064067602158,
                0.0481550432741642,
                0.03921608254313469,
                0.001081325812265277,
                -0.018742980435490608,
                -0.015643181279301643,
                -0.0003424198366701603,
                0.003532330971211195,
                0.001171436277218163,
                -0.035900019109249115,
                0.012543379329144955,
                -0.051038578152656555,
                0.029988769441843033,
                0.005334540735930204,
                0.004685745108872652,
                0.013336352072656155,
                -0.006523999385535717,
                -0.01492229662835598,
                0.0069925738498568535,
                -0.005046187434345484,
                0.019175510853528976,
                -0.005226408131420612,
                -0.020761456340551376,
                -0.020040571689605713,
                -0.008362253196537495,
                0.0034422206226736307,
                0.0354674868285656,
                -0.05046187341213226,
                0.04036949947476387,
                0.0016039666952565312,
                0.008578518405556679,
                0.02883535623550415,
                0.023933345451951027,
                -0.051615286618471146,
                -0.02667270414531231,
                0.012543379329144955,
                0.007497192360460758,
                0.003568375250324607,
                0.009047092869877815,
                -0.05709400400519371,
                0.019175510853528976,
                0.019752219319343567,
                0.022347399964928627,
                0.007641369476914406,
                -0.020184749737381935,
                0.008578518405556679,
                0.0065960874781012535,
                -0.0027213366702198982,
                0.01232711412012577,
                0.003676507854834199,
                -0.03200724348425865,
                -0.07785546034574509,
                -0.007497192360460758,
                0.008182032033801079,
                -0.0031899111345410347,
                -0.03258395195007324,
                -0.0256634671241045,
                0.016796594485640526,
                0.018742980435490608,
                0.012903821654617786,
                -0.043253034353256226,
                0.015282738953828812,
                0.02782611921429634,
                0.036332547664642334,
                0.02508675865828991,
                -0.004487502388656139,
                0.0048299222253263,
                0.016796594485640526,
                -0.007172794546931982,
                0.03618837147951126,
                -0.0096598444506526,
                -0.014273501001298428,
                0.025519289076328278,
                0.05075022578239441,
                -0.05536388233304024,
                -0.03258395195007324,
                0.025519289076328278,
                -0.013336352072656155,
                0.007244883105158806,
                0.02047310210764408,
                -0.032728128135204315,
                -0.048443399369716644,
                0.0028474912978708744,
                0.0240775216370821,
                0.0003581891942303628,
                -0.005406629294157028,
                0.007280927151441574,
                -0.013336352072656155,
                0.020617280155420303,
                0.028979532420635223,
                -0.004469480365514755,
                -0.003946839366108179,
                -0.05017351731657982,
                0.0026852923911064863,
                0.045848216861486435,
                0.010452816262841225,
                -0.011317877098917961,
                0.02465422824025154,
                -0.01730121299624443,
                -0.020761456340551376,
                0.04152291268110275,
                0.026095997542142868,
                -0.033737365156412125,
                -0.026095997542142868,
                -0.0152106499299407,
                -0.002703314647078514,
                0.016796594485640526,
                0.01571526937186718,
                0.034746602177619934,
                -0.0304212998598814,
                0.00994819775223732,
                0.0059472923167049885,
                -0.024942582473158836,
                0.00692048529163003,
                0.03503495827317238,
                0.01989639550447464,
                0.016219887882471085,
                -0.0017841876251623034,
                0.034314073622226715,
                -0.01095743477344513,
                -0.026961058378219604,
                -0.0044334358535707,
                -0.006199601572006941,
                0.011822495609521866,
                0.02883535623550415,
                0.0088668717071414,
                0.04382973909378052,
                0.02465422824025154,
                0.013047998771071434,
                0.023933345451951027,
                0.007497192360460758,
                -0.02249157801270485,
                0.01312008686363697,
                0.008073899894952774,
                0.028547002002596855,
                0.007929722778499126,
                -0.0354674868285656,
                -0.04296467825770378,
                -0.04238797351717949,
                -0.01593153364956379,
                -0.040081143379211426,
                0.037485960870981216,
                0.03344901278614998,
                0.04152291268110275,
                -0.014345589093863964,
                0.024510052055120468,
                0.015066473744809628,
                0.0031718891113996506,
                0.0054426733404397964,
                0.0032439774367958307,
                0.012182937934994698,
                0.04555986076593399,
                0.017589567229151726,
                -0.03416989743709564,
                0.03359318897128105,
                -0.034890782088041306,
                -0.02465422824025154,
                0.056805651634931564,
                -0.04988516494631767,
                -0.003406176343560219,
                0.009299402125179768,
                0.019031334668397903,
                0.04959681257605553,
                -0.0019373755203559995,
                -0.024510052055120468,
                -0.012255026027560234,
                -0.05795906484127045,
                -0.011894584633409977,
                -0.028691178187727928,
                0.007244883105158806,
                -0.0010903369402512908,
                0.0014688009396195412,
                0.002649248344823718,
                -0.011462054215371609,
                -0.007677413523197174,
                -0.022635754197835922,
                -0.004036949947476387,
                -0.003856728784739971,
                -0.045271508395671844,
                0.029700417071580887,
                -0.01989639550447464
            ]


def main():
    # 创建客户端实例
    client = InfinityClient(database="book_image_db")
    
    try:
        print("=== 连接上下文示例 ===") 
            
        # 1. 获取所有数据库
        # databases = conn.list_databases()
        # print(f"1. 所有数据库: {databases}")
            
        # 2. 获取当前数据库的所有表
        tables = client.list_tables()
        print(f"2. 当前数据库表: {tables}")
            

        table_name = "book_page_image"
        # 3. 执行搜索查询
        print(f"5. 查询表 {table_name} 中的数据")
            
        # 构建查询
        search_query = {
            "field": "content",
            "query": "美美",
            "topn": 2
        }
            
        # 执行查询
        result = client.search(table_name, output_columns, search_query).to_pl()
        print(f"   - 查询结果: {result}")
            
        # 4. 执行向量检索（示例，实际需要向量字段）
        print(f"6. 执行向量检索（示例）")
            
        try:
            # 注意：此示例仅用于演示API用法，实际使用需要表中存在向量字段
            vector_query = {
                "vector_field": "dense_vector_1024",  # 假设存在向量字段
                "query_vector": vec,
                "topn": 2
            }
            vector_result = client.vector_search(table_name, output_columns, vector_query).to_pl()
            print(f"   - 向量检索结果: {vector_result}")
        except Exception as e:
            print(f"   - 向量检索示例失败（预期行为，因为表中没有向量字段）: {e}")
            

        # 5. 执行混合检索（示例，实际需要向量字段）
        print(f"6. 执行混合检索（示例）")
            
        try:
            # 注意：此示例仅用于演示API用法，实际使用需要表中存在向量字段
            vector_query = {
                "vector_field": "dense_vector_1024",  # 假设存在向量字段
                "query_vector": vec,
                "topn": 2,
                "field": "content",
                "query": "美美"
            }
            vector_result = client.hybrid_search(table_name, output_columns, vector_query).to_pl()
            print(f"   - 向量检索结果: {vector_result}")
        except Exception as e:
            print(f"   - 向量检索示例失败（预期行为，因为表中没有向量字段）: {e}")

            
        print("\n✅ 所有操作执行完成")
        
        # 8. 连接已自动释放回连接池
        print("\n✅ 连接已自动释放回连接池")
        
    except Exception as e:
        print(f"\n❌ 操作失败: {e}")
    finally:
        # 关闭客户端
        client.close()
        print("✅ 客户端已关闭")

if __name__ == "__main__":
    main()
    # infinity_client = infinity.connect(infinity.NetworkAddress("192.168.16.134", 23817))
    # db = infinity_client.get_database("book_image_db")
    # list_tables = db.list_tables()
    # print(list_tables)