infinity-sdk版本

.env

@@ -22,6 +22,7 @@ VECTOR_DB_TYPE=infinity
 # Infinity向量数据库配置
 INFINITY_HOST=192.168.16.134
 INFINITY_PORT=23820
+INFINITY_SDK_PORT=23817
 INFINITY_DATABASE=book_image_db
 INFINITY_USER=admin
 INFINITY_PASSWORD=admin

conf/config.py

@@ -132,7 +132,12 @@ class VectorDBConfig:
     @staticmethod
     def get_infinity_port() -> int:
         """获取Infinity向量数据库端口"""
-        return int(os.getenv("INFINITY_PORT", "23810"))
+        return int(os.getenv("INFINITY_PORT", "23820"))
+
+    @staticmethod
+    def get_infinity_sdk_port() -> int:
+        """获取Infinity向量数据库SDK端口"""
+        return int(os.getenv("INFINITY_SDK_PORT", "23817"))
     
     @staticmethod
     def get_infinity_user() -> str:

test/test_infinity_search.py

@@ -0,0 +1,1127 @@
+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)

utils/__init__.py

@@ -0,0 +1,11 @@
+# 从分离的模块中导入所有组件，确保向后兼容性
+from .infinity.pool import InfinityConnectionPool
+from .infinity.client import InfinityClient, get_client, close_client
+
+# 重新导出所有组件，使它们可以从包级别访问
+__all__ = [
+    'InfinityConnectionPool',
+    'InfinityClient',
+    'get_client',
+    'close_client'
+]

utils/infinity/README.md

@@ -0,0 +1,527 @@
+# Infinity Python API客户端（带连接池）
+
+这是一个基于Infinity数据库的Python API客户端，带有连接池机制，旨在保障高并发下的高可用性。
+
+## 设计特点
+
+1. **并发连接管理**：支持多线程并发访问
+2. **连接自动回收和复用**：避免频繁创建和关闭连接
+3. **连接超时和心跳检测**：自动检测和清理无效连接
+4. **动态调整连接数量**：根据负载自动调整连接数
+5. **线程安全**：所有操作都是线程安全的
+6. **易用的API**：提供简洁易用的数据库操作接口
+
+## 目录结构
+
+```
+└── utils/infinity/
+    ├── __init__.py    # 主入口，导入并重新导出所有组件
+    ├── client.py      # 客户端实现，包含数据库操作方法
+    ├── pool.py        # 连接池实现，包含连接管理和心跳检测
+    ├── README.md      # 说明文档
+    └── test_infinity.py  # 测试脚本
+```
+
+## 安装依赖
+
+```bash
+# 安装Infinity Python SDK
+pip install infinity-sdk
+```
+
+## 快速开始
+
+### 1. 基本使用
+
+```python
+from utils.infinity import InfinityClient
+
+# 创建客户端实例
+client = InfinityClient(
+    host="192.168.16.134",
+    port="23817",
+    database="default_db",
+    min_connections=2,
+    max_connections=10
+)
+
+# 获取所有数据库
+databases = client.get_databases()
+print(f"Databases: {databases}")
+
+# 获取指定数据库的所有表
+tables = client.get_tables(database_name="image_db")
+print(f"Tables: {tables}")
+
+# 关闭客户端
+client.close()
+```
+
+### 2. 使用全局客户端（单例模式）
+
+```python
+from utils.infinity import get_client, close_client
+
+# 获取全局客户端实例
+client = get_client()
+
+# 使用客户端
+databases = client.get_databases()
+print(f"Databases: {databases}")
+
+# 关闭全局客户端
+close_client()
+```
+
+### 3. 使用连接上下文
+
+连接上下文管理器是一种更安全、更高效的使用连接方式，它会确保连接在使用完毕后自动释放回连接池，避免连接泄漏。
+
+#### 完整示例
+
+```python
+from utils.infinity import InfinityClient
+
+def main():
+    # 创建客户端实例
+    client = InfinityClient(database="test_db")
+    
+    try:
+        # 使用连接上下文获取连接
+        with client.get_connection() as conn:
+            print("=== 连接上下文示例 ===")
+            
+            # 1. 获取所有数据库
+            databases = conn.get_databases()
+            print(f"1. 所有数据库: {databases}")
+            
+            # 2. 获取当前数据库的所有表
+            tables = conn.get_tables()
+            print(f"2. 当前数据库表: {tables}")
+            
+            # 3. 创建新表（如果不存在）
+            table_name = "example_table"
+            print(f"3. 创建表: {table_name}")
+            
+            # 定义表结构
+            fields = [
+                {"name": "id", "type": "INT", "is_primary_key": True},
+                {"name": "name", "type": "VARCHAR(100)"},
+                {"name": "value", "type": "FLOAT"}
+            ]
+            
+            # 先检查表是否存在，如果存在则删除
+            if table_name in tables:
+                conn.drop_table(table_name)
+                print(f"   - 表 {table_name} 已存在，已删除")
+            
+            # 创建新表
+            conn.create_table(table_name, fields)
+            print(f"   - 表 {table_name} 创建成功")
+            
+            # 4. 插入数据
+            print(f"4. 向表 {table_name} 插入数据")
+            
+            # 准备插入的数据
+            documents = [
+                {"id": 1, "name": "示例1", "value": 10.5},
+                {"id": 2, "name": "示例2", "value": 20.3},
+                {"id": 3, "name": "示例3", "value": 15.7}
+            ]
+            
+            # 执行插入操作
+            conn.insert(table_name, documents)
+            print(f"   - 成功插入 {len(documents)} 条数据")
+            
+            # 5. 执行搜索查询
+            print(f"5. 查询表 {table_name} 中的数据")
+            
+            # 构建查询
+            search_query = {
+                "field": "name",
+                "query": "示例",
+                "topn": 2
+            }
+            
+            # 执行查询
+            result = conn.search(table_name, ["id", "name", "value"], search_query)
+            print(f"   - 查询结果: {result}")
+            
+            # 6. 执行向量检索（示例，实际需要向量字段）
+            print(f"6. 执行向量检索（示例）")
+            
+            try:
+                # 注意：此示例仅用于演示API用法，实际使用需要表中存在向量字段
+                vector_query = {
+                    "vector_field": "vector",  # 假设存在向量字段
+                    "query_vector": [0.1, 0.2, 0.3],
+                    "topn": 2
+                }
+                vector_result = conn.vector_search(table_name, ["id", "name", "value"], vector_query)
+                print(f"   - 向量检索结果: {vector_result}")
+            except Exception as e:
+                print(f"   - 向量检索示例失败（预期行为，因为表中没有向量字段）: {e}")
+            
+            # 7. 再次查看所有表
+            updated_tables = conn.get_tables()
+            print(f"7. 更新后的表列表: {updated_tables}")
+            
+            print("\n✅ 所有操作执行完成")
+        
+        # 8. 连接已自动释放回连接池
+        print("\n✅ 连接已自动释放回连接池")
+        
+    except Exception as e:
+        print(f"\n❌ 操作失败: {e}")
+    finally:
+        # 关闭客户端
+        client.close()
+        print("✅ 客户端已关闭")
+
+if __name__ == "__main__":
+    main()
+```
+
+#### 多操作上下文示例
+
+```python
+from utils.infinity import InfinityClient
+
+# 创建客户端实例
+client = InfinityClient(database="test_db")
+
+# 示例：在同一个连接上下文中执行多个相关操作
+try:
+    with client.get_connection() as conn:
+        # 操作1：创建表
+        conn.create_table("temp_table", [
+            {"name": "id", "type": "INT", "is_primary_key": True},
+            {"name": "data", "type": "VARCHAR(255)"}
+        ])
+        
+        # 操作2：插入数据
+        conn.insert("temp_table", [{"id": 1, "data": "test1"}, {"id": 2, "data": "test2"}])
+        
+        # 操作3：查询数据
+        search_query = {
+            "field": "data",
+            "query": "test",
+            "topn": 2
+        }
+        result = conn.search("temp_table", ["id", "data"], search_query)
+        print(f"查询结果: {result}")
+        
+        # 操作4：删除表
+        conn.drop_table("temp_table")
+        
+        print("✅ 所有相关操作在同一个连接上下文中完成")
+except Exception as e:
+    print(f"❌ 操作失败: {e}")
+finally:
+    client.close()
+```
+
+#### 异常处理上下文示例
+
+```python
+from utils.infinity import InfinityClient
+
+# 创建客户端实例
+client = InfinityClient(database="test_db")
+
+try:
+    with client.get_connection() as conn:
+        # 执行可能失败的操作
+        conn.create_table("error_table", [
+            {"name": "id", "type": "INT", "is_primary_key": True},
+            {"name": "invalid_field", "type": "INVALID_TYPE"}  # 无效字段类型
+        ])
+        
+    # 注意：如果上面的操作失败，代码不会执行到这里
+    print("✅ 操作成功")
+except Exception as e:
+    # 捕获并处理异常
+    print(f"❌ 操作失败，已捕获异常: {e}")
+finally:
+    # 无论操作是否成功，客户端都会被关闭
+    client.close()
+    print("✅ 客户端已关闭")
+```
+
+### 4. 搜索示例
+
+```python
+from utils.infinity import InfinityClient
+
+client = InfinityClient(database="image_db")
+
+# 执行搜索查询
+result = client.search(
+    table_name="pdf_documents_table",
+    output_fields=["id", "title", "content"],
+    query={
+        "field": "content",
+        "query": "儿童绘本",
+        "topn": 5
+    }
+)
+
+print(f"Search result: {result}")
+
+client.close()
+```
+
+### 5. 混合检索示例
+
+```python
+from utils.infinity import InfinityClient
+
+client = InfinityClient(database="image_db")
+
+# 执行混合检索
+result = client.hybrid_search(
+    table_name="pdf_documents_table",
+    output_fields=["id", "title", "content", "score"],
+    query={
+        "vector_field": "dense_vector_1024",
+        "query_vector": [0.1, 0.2, 0.3, ...],  # 实际向量
+        "field": "content",
+        "query": "儿童绘本",
+        "topn": 5,
+        "fusion_weight": 0.5
+    }
+)
+
+print(f"Hybrid search result: {result}")
+
+client.close()
+```
+
+### 6. 向量检索示例
+
+```python
+from utils.infinity import InfinityClient
+
+client = InfinityClient(database="image_db")
+
+# 执行向量检索
+result = client.vector_search(
+    table_name="pdf_documents_table",
+    output_fields=["id", "title", "content", "score"],
+    query={
+        "vector_field": "dense_vector_1024",
+        "query_vector": [0.1, 0.2, 0.3, ...],  # 实际向量
+        "topn": 5
+    }
+)
+
+print(f"Vector search result: {result}")
+
+client.close()
+```
+
+## 核心功能
+
+### 数据库操作
+
+- `get_databases()`: 获取所有数据库
+- `create_database(database_name)`: 创建数据库
+- `drop_database(database_name)`: 删除数据库
+- `use_database(database_name)`: 切换数据库
+
+### 表操作
+
+- `get_tables(database_name=None)`: 获取所有表
+- `create_table(table_name, fields, database_name=None)`: 创建表
+- `drop_table(table_name, database_name=None)`: 删除表
+
+### 文档操作
+
+- `insert(table_name, documents, database_name=None)`: 插入文档
+
+### 检索操作
+
+- `search(table_name, output_fields, query, database_name=None)`: 搜索文档
+  - `table_name`: 表名
+  - `output_fields`: 要返回的字段列表
+  - `query`: 查询条件，包含`field`、`query`和`topn`字段
+  - `database_name`: 数据库名称（可选，默认使用客户端配置的数据库）
+
+- `hybrid_search(table_name, output_fields, query, database_name=None)`: 混合检索
+  - `table_name`: 表名
+  - `output_fields`: 要返回的字段列表
+  - `query`: 查询条件，包含`vector_field`、`query_vector`、`field`、`query`、`topn`和`fusion_weight`字段
+  - `database_name`: 数据库名称（可选，默认使用客户端配置的数据库）
+
+- `vector_search(table_name, output_fields, query, database_name=None)`: 向量检索
+  - `table_name`: 表名
+  - `output_fields`: 要返回的字段列表
+  - `query`: 查询条件，包含`vector_field`、`query_vector`和`topn`字段
+  - `database_name`: 数据库名称（可选，默认使用客户端配置的数据库）
+
+### 连接池管理
+
+- `get_status()`: 获取连接池状态
+- `close()`: 关闭所有连接
+
+## 配置参数
+
+### InfinityClient 配置
+
+| 参数 | 类型 | 默认值 | 说明 |
+|------|------|--------|------|
+| host | str | "192.168.16.134" | Infinity服务地址 |
+| port | str | "23817" | Infinity服务端口 |
+| database | str | "default_db" | 数据库名称 |
+| min_connections | int | 2 | 最小连接数 |
+| max_connections | int | 10 | 最大连接数 |
+
+### InfinityConnectionPool 配置
+
+| 参数 | 类型 | 默认值 | 说明 |
+|------|------|--------|------|
+| host | str | "192.168.16.134" | Infinity服务地址 |
+| port | str | "23817" | Infinity服务端口 |
+| database | str | "default_db" | 数据库名称 |
+| min_connections | int | 2 | 最小连接数 |
+| max_connections | int | 10 | 最大连接数 |
+| connection_timeout | int | 30 | 连接超时时间（秒） |
+| idle_timeout | int | 300 | 空闲连接超时时间（秒） |
+| heartbeat_interval | int | 60 | 心跳检测间隔（秒） |
+
+## 连接池状态
+
+可以通过 `get_status()` 方法获取连接池的状态信息：
+
+```python
+from utils.infinity import InfinityClient
+
+client = InfinityClient()
+status = client.get_status()
+print(f"Connection pool status: {status}")
+```
+
+输出示例：
+
+```
+Connection pool status: {
+    "total_connections": 5,
+    "available_connections": 3,
+    "in_use_connections": 2,
+    "min_connections": 2,
+    "max_connections": 10
+}
+```
+
+## 高并发使用建议
+
+1. **调整连接池大小**：根据实际并发量调整 `min_connections` 和 `max_connections`
+2. **使用全局客户端**：在多线程环境中使用全局客户端实例
+3. **合理设置超时时间**：根据网络环境和数据库性能调整超时参数
+4. **使用连接上下文**：使用 `with client.get_connection()` 确保连接正确释放
+5. **定期检查连接池状态**：监控连接池状态，及时调整配置
+
+## 异常处理
+
+```python
+from utils.infinity import InfinityClient
+
+try:
+    client = InfinityClient()
+    databases = client.get_databases()
+    print(f"Databases: {databases}")
+except Exception as e:
+    print(f"Error: {e}")
+finally:
+    client.close()
+```
+
+## 最佳实践
+
+1. **在应用启动时初始化客户端**：避免频繁创建和销毁客户端
+2. **在应用关闭时关闭客户端**：释放所有连接资源
+3. **使用连接上下文管理连接**：确保连接正确释放
+4. **监控连接池状态**：及时发现和解决连接问题
+5. **根据负载调整连接池大小**：避免连接过多或不足
+
+## 与现有代码集成
+
+可以将此客户端与现有代码无缝集成，替换直接使用Infinity SDK的地方：
+
+### 原有代码
+
+```python
+import infinity
+from infinity.common import NetworkAddress
+
+# 直接创建连接
+conn = infinity.connect(NetworkAddress("192.168.16.134", "23817"))
+conn.use_database("image_db")
+# 使用连接
+result = conn.search("my_table", query)
+```
+
+### 集成后
+
+```python
+from utils.infinity import InfinityClient
+
+# 使用连接池客户端
+client = InfinityClient(database="image_db")
+result = client.search("my_table", query)
+```
+
+## 性能优化
+
+1. **减少连接创建开销**：连接池自动管理连接，避免频繁创建和关闭连接
+2. **连接复用**：连接可以被多次复用，提高性能
+3. **异步连接管理**：连接池异步管理连接，不会阻塞主线程
+4. **心跳检测**：自动清理无效连接，保持连接池健康
+
+## 测试
+
+可以使用以下方法测试连接池的性能：
+
+```python
+from utils.infinity import InfinityClient
+import threading
+import time
+
+def test_query(client, thread_id):
+    """测试查询性能"""
+    start_time = time.time()
+    databases = client.get_databases()
+    end_time = time.time()
+    print(f"Thread {thread_id}: Query took {end_time - start_time:.4f} seconds")
+
+# 创建客户端
+client = InfinityClient(max_connections=20)
+
+# 测试并发查询
+threads = []
+start_time = time.time()
+
+for i in range(50):
+    thread = threading.Thread(target=test_query, args=(client, i))
+    threads.append(thread)
+    thread.start()
+
+# 等待所有线程完成
+for thread in threads:
+    thread.join()
+
+end_time = time.time()
+print(f"Total time for 50 concurrent queries: {end_time - start_time:.4f} seconds")
+print(f"Connection pool status: {client.get_status()}")
+
+# 关闭客户端
+client.close()
+```
+
+## 版本历史
+
+- v1.0.0: 初始版本，实现了基本的连接池功能和数据库操作接口
+
+## 许可证
+
+MIT License

utils/infinity/__init__.py

@@ -0,0 +1,43 @@
+# Infinity Python API客户端，带有连接池机制
+
+# 从分离的模块中导入所有组件，确保向后兼容性
+from .pool import InfinityConnectionPool
+from .client import InfinityClient, get_client, close_client
+
+# 重新导出所有组件，使它们可以从包级别访问
+__all__ = [
+    'InfinityConnectionPool',
+    'InfinityClient',
+    'get_client',
+    'close_client'
+]
+
+# 使用示例
+"""
+# 示例1：基本使用
+client = InfinityClient()
+databases = client.get_databases()
+print(f"Databases: {databases}")
+
+# 示例2：使用连接池上下文
+with client.get_connection() as conn:
+    tables = conn.get_tables()
+    print(f"Tables: {tables}")
+
+# 示例3：使用全局客户端
+from utils.infinity import get_client
+client = get_client()
+databases = client.get_databases()
+print(f"Databases from global client: {databases}")
+
+# 示例4：混合检索
+result = client.hybrid_search(
+    table_name="my_table",
+    vector_field="vector",
+    query_vector=[0.1, 0.2, 0.3],
+    text_query="test",
+    text_field="content",
+    topn=5
+)
+print(f"Hybrid search result: {result}")
+"""

utils/infinity/client.py

@@ -0,0 +1,281 @@
+# Infinity数据库客户端实现
+from infinity.common import ConflictType
+from typing import Dict, Any, List, Optional
+import threading
+from contextlib import contextmanager
+from conf.config import VectorDBConfig
+from .pool import InfinityConnectionPool
+
+class InfinityClient:
+    """
+    Infinity数据库客户端，基于连接池
+    
+    提供了常用的数据库操作方法，包括：
+    - 数据库操作
+    - 表操作
+    - 文档操作
+    - 搜索操作
+    """
+    
+    def __init__(
+        self,
+        host: str = VectorDBConfig.get_infinity_host(),
+        port: str = VectorDBConfig.get_infinity_sdk_port(),
+        database: str = VectorDBConfig.get_infinity_database(),
+        min_connections: int = 2,
+        max_connections: int = 10
+    ):
+        """
+        初始化Infinity客户端
+        
+        Args:
+            host: Infinity服务地址
+            port: Infinity服务端口
+            database: 数据库名称
+            min_connections: 最小连接数
+            max_connections: 最大连接数
+        """
+        self.pool = InfinityConnectionPool(
+            host=host,
+            port=port,
+            database=database,
+            min_connections=min_connections,
+            max_connections=max_connections
+        )
+    
+    def _get_database(self, conn: Any, database_name: Optional[str] = None):
+        """
+        获取数据库对象
+        
+        Args:
+            conn: Infinity连接对象
+            database_name: 数据库名称（可选，默认使用客户端配置的数据库）
+            
+        Returns:
+            Database对象
+        """
+        db_name = database_name or self.pool.database
+        return conn.get_database(db_name)
+    
+    def _get_table(self, conn: Any, table_name: str, database_name: Optional[str] = None):
+        """
+        获取表对象
+        
+        Args:
+            conn: Infinity连接对象
+            table_name: 表名
+            database_name: 数据库名称（可选，默认使用客户端配置的数据库）
+            
+        Returns:
+            Table对象
+        """
+        db = self._get_database(conn, database_name)
+        return db.get_table(table_name)
+    
+    @contextmanager
+    def get_connection(self, timeout: Optional[int] = None):
+        """
+        获取一个连接上下文
+        
+        Args:
+            timeout: 获取连接的超时时间（秒）
+        """
+        with self.pool.get_connection(timeout) as conn:
+            yield conn
+    
+    def list_databases(self) -> List[str]:
+        """获取所有数据库"""
+        with self.pool.list_connection() as conn:
+            return conn.list_databases()
+    
+    def get_databases(self) -> List[str]:
+        """获取所有数据库（别名，向后兼容）"""
+        return self.list_databases()
+    
+    def create_database(self, database_name: str, conflict_type: str = ConflictType.Error, comment: Optional[str] = None):
+        """创建数据库"""
+        with self.pool.get_connection() as conn:
+            return conn.create_database(database_name, conflict_type, comment)
+    
+    def drop_database(self, database_name: str, conflict_type: str = "Error"):
+        """删除数据库"""
+        with self.pool.get_connection() as conn:
+            return conn.drop_database(database_name, conflict_type)
+    
+    def use_database(self, database_name: str):
+        """切换客户端默认数据库
+        
+        注意：此操作只会影响当前客户端实例创建的新连接，不会影响已存在的连接
+        """
+        # 更新客户端的默认数据库
+        self.pool.database = database_name
+    
+    def list_tables(self, database_name: Optional[str] = None) -> List[str]:
+        """获取所有表"""
+        with self.pool.get_connection() as conn:
+            # 根据官方API，应该先获取Database对象，然后在Database对象上调用get_tables方法
+            db = self._get_database(conn, database_name)
+            return db.list_tables()
+    
+    def get_tables(self, database_name: Optional[str] = None) -> List[str]:
+        """获取所有表（别名，向后兼容）"""
+        return self.list_tables(database_name)
+    
+    def get_table(self, table_name: str, database_name: Optional[str] = None):
+        """获取表"""
+        with self.pool.get_connection() as conn:
+            # 根据官方API，应该先获取Database对象，然后在Database对象上调用get_table方法
+            return self._get_table(conn, table_name, database_name)
+
+    def create_table(
+        self,
+        table_name: str,
+        columns_definition: List[Dict[str, Any]],
+        conflict_type: str = ConflictType.Error,
+        database_name: Optional[str] = None
+    ):
+        """创建表"""
+        with self.pool.get_connection() as conn:
+            # 根据官方API，应该先获取Database对象，然后在Database对象上调用create_table方法
+            db = self._get_database(conn, database_name)
+            return db.create_table(table_name, columns_definition, conflict_type)
+
+    def drop_table(self, table_name: str, database_name: Optional[str] = None, conflict_type = ConflictType.Error):
+        """删除表"""
+        with self.pool.get_connection() as conn:
+            # 根据官方API，应该先获取Database对象，然后在Database对象上调用drop_table方法
+            db = self._get_database(conn, database_name)
+            return db.drop_table(table_name, conflict_type)
+
+    def create_index(self, table_name: str, index_name: str, index_info: Dict[str, Any], database_name: Optional[str] = None, conflict_type = ConflictType.Error, comment = None):
+        """创建索引"""
+        with self.pool.get_connection() as conn:
+            # 使用辅助方法获取Table对象
+            table = self._get_table(conn, table_name, database_name)
+            return table.create_index(index_name, index_info, conflict_type, comment)
+    
+    def optimize(self, table_name: str, database_name: Optional[str] = None):
+        """优化表"""
+        with self.pool.get_connection() as conn:
+           # 使用辅助方法获取Table对象
+            table = self._get_table(conn, table_name, database_name)
+            return table.optimize()
+
+    def drop_index(self, table_name: str, index_name: str, database_name: Optional[str] = None, conflict_type = ConflictType.Error):
+        """删除索引"""
+        with self.pool.get_connection() as conn:
+           # 使用辅助方法获取Table对象
+            table = self._get_table(conn, table_name, database_name)
+            return table.drop_index(index_name, conflict_type)
+
+    def insert(
+        self,
+        table_name: str,
+        documents: List[Dict[str, Any]],
+        database_name: Optional[str] = None
+    ):
+        """插入文档"""
+        with self.pool.get_connection() as conn:
+            table = self._get_table(conn, table_name, database_name)
+            return table.insert(documents)
+    
+    def search(
+        self,
+        table_name: str,
+        output_fields: List[str],
+        query: Dict[str, Any],
+        database_name: Optional[str] = None
+    ):
+        """搜索文档"""
+        with self.pool.get_connection() as conn:
+            # 获取Table对象
+            table = self._get_table(conn, table_name, database_name)
+            # 获取结果集
+            return table.output(output_fields).match_text(query["field"], query["query"], query["topn"])
+    
+    def hybrid_search(
+        self,
+        table_name: str,
+        output_fields: List[str],
+        query: Dict[str, Any],
+        database_name: Optional[str] = None
+    ):
+        """混合检索"""
+        with self.pool.get_connection() as conn:
+            # 获取Table对象
+            table = self._get_table(conn, table_name, database_name)
+            # 获取结果集
+            return table.output(output_fields) \
+                .match_dense(query["vector_field"], query["query_vector"], "float", "cosine", query["topn"]) \
+                .match_text(query["field"], query["query"], query["topn"]) \
+                .fusion("rrf", query["topn"])
+    
+    def vector_search(
+        self,
+        table_name: str,
+        output_fields: List[str],
+        query: Dict[str, Any],
+        database_name: Optional[str] = None
+    ):
+        """向量检索"""
+        with self.pool.get_connection() as conn:
+            # 获取Table对象
+            table = self._get_table(conn, table_name, database_name)
+            # 获取结果集
+            return table.output(output_fields) \
+                .match_dense(query["vector_field"], query["query_vector"], "float", "cosine", query["topn"])
+    
+    def get_status(self) -> Dict[str, Any]:
+        """获取客户端状态"""
+        return self.pool.get_status()
+    
+    def close(self):
+        """关闭客户端，释放所有连接"""
+        self.pool.close()
+
+# 全局客户端实例
+_global_client: Optional[InfinityClient] = None
+_client_lock = threading.Lock()
+
+def get_client(
+    host: str = VectorDBConfig.get_infinity_host(),
+    port: str = VectorDBConfig.get_infinity_port(),
+    database: str = VectorDBConfig.get_infinity_database(),
+    min_connections: int = 2,
+    max_connections: int = 10
+) -> InfinityClient:
+    """
+    获取全局客户端实例（单例模式）
+    
+    Args:
+        host: Infinity服务地址
+        port: Infinity服务端口
+        database: 数据库名称
+        min_connections: 最小连接数
+        max_connections: 最大连接数
+        
+    Returns:
+        全局Infinity客户端实例
+    """
+    global _global_client
+    
+    with _client_lock:
+        if _global_client is None:
+            _global_client = InfinityClient(
+                host=host,
+                port=port,
+                database=database,
+                min_connections=min_connections,
+                max_connections=max_connections
+            )
+    
+    return _global_client
+
+def close_client():
+    """关闭全局客户端"""
+    global _global_client
+    
+    with _client_lock:
+        if _global_client:
+            _global_client.close()
+            _global_client = None

utils/infinity/pool.py

@@ -0,0 +1,281 @@
+# Infinity数据库连接池实现
+
+from typing import Dict, Any, List, Optional
+import threading
+import time
+from contextlib import contextmanager
+from conf.config import VectorDBConfig
+
+class InfinityConnectionPool:
+    """
+    Infinity数据库连接池
+    
+    设计特点：
+    1. 支持并发连接管理
+    2. 连接自动回收和复用
+    3. 连接超时和心跳检测
+    4. 动态调整连接数量
+    5. 线程安全
+    """
+    
+    def __init__(
+        self,
+        host: str = VectorDBConfig.get_infinity_host(),
+        port: str = VectorDBConfig.get_infinity_sdk_port(),
+        database: str = VectorDBConfig.get_infinity_database(),
+        min_connections: int = 2,
+        max_connections: int = 10,
+        connection_timeout: int = 30,
+        idle_timeout: int = 300,
+        heartbeat_interval: int = 60
+    ):
+        """
+        初始化连接池
+        
+        Args:
+            host: Infinity服务地址
+            port: Infinity服务端口
+            database: 数据库名称
+            min_connections: 最小连接数
+            max_connections: 最大连接数
+            connection_timeout: 连接超时时间（秒）
+            idle_timeout: 空闲连接超时时间（秒）
+            heartbeat_interval: 心跳检测间隔（秒）
+        """
+        self.host = host
+        self.port = port
+        self.database = database
+        self.min_connections = min_connections
+        self.max_connections = max_connections
+        self.connection_timeout = connection_timeout
+        self.idle_timeout = idle_timeout
+        self.heartbeat_interval = heartbeat_interval
+        
+        # 连接池状态
+        self.connections = []  # 可用连接列表
+        self.in_use = {}  # 正在使用的连接 {connection: thread_id}
+        self.connection_count = 0  # 当前连接总数
+        
+        # 线程安全锁
+        self.lock = threading.Lock()
+        self.condition = threading.Condition(self.lock)
+        
+        # 初始化最小连接数
+        self._init_connections()
+        
+        # 启动心跳检测线程
+        self.heartbeat_thread = threading.Thread(target=self._heartbeat_check, daemon=True)
+        self.heartbeat_thread.start()
+    
+    def _init_connections(self):
+        """初始化最小连接数"""
+        for _ in range(self.min_connections):
+            self._create_connection()
+    
+    def _create_connection(self) -> Any:
+        """创建新连接"""
+        try:
+            import infinity
+            
+            # 连接到Infinity服务
+            connection = infinity.connect(
+                infinity.NetworkAddress(self.host, self.port)
+            )
+                        
+            # 注意：根据官方API，RemoteThriftInfinityConnection对象没有use_database方法
+            # 数据库操作应该通过create_database、drop_database等方法直接指定数据库名称
+            # 或者通过获取Database对象后再进行操作
+            
+            # 保存数据库名称，供后续操作使用
+            connection.__dict__['_database'] = self.database
+            
+            # 记录连接创建时间
+            connection.__dict__['_created_at'] = time.time()
+            connection.__dict__['_last_used'] = time.time()
+            connection.__dict__['_is_valid'] = True
+            
+            with self.lock:
+                self.connections.append(connection)
+                self.connection_count += 1
+            
+            return connection
+        except Exception as e:
+            raise Exception(f"Failed to create Infinity connection: {e}")
+    
+    def _is_valid_connection(self, connection: Any) -> bool:
+        """检查连接是否有效"""
+        try:
+            # 通过执行简单查询检查连接是否有效
+            connection.get_database(self.database)
+            return True
+        except Exception:
+            return False
+    
+    def _heartbeat_check(self):
+        """心跳检测，定期检查连接有效性并清理过期连接"""
+        while True:
+            time.sleep(self.heartbeat_interval)
+            self._cleanup_connections()
+    
+    def _cleanup_connections(self):
+        """清理无效或过期连接"""
+        with self.lock:
+            current_time = time.time()
+            valid_connections = []
+            
+            for connection in self.connections:
+                # 检查连接是否过期
+                if current_time - connection.__dict__['_last_used'] > self.idle_timeout:
+                    # 关闭过期连接
+                    try:
+                        connection.disconnect()
+                        self.connection_count -= 1
+                    except Exception:
+                        pass
+                elif not self._is_valid_connection(connection):
+                    # 关闭无效连接
+                    try:
+                        connection.disconnect()
+                        self.connection_count -= 1
+                    except Exception:
+                        pass
+                else:
+                    valid_connections.append(connection)
+            
+            self.connections = valid_connections
+            
+            # 确保连接数不低于最小值
+            while self.connection_count < self.min_connections:
+                self._create_connection()
+    
+    @contextmanager
+    def get_connection(self, timeout: Optional[int] = None) -> Any:
+        """
+        获取一个连接，使用上下文管理器
+        
+        Args:
+            timeout: 获取连接的超时时间（秒）
+            
+        Yields:
+            Infinity连接对象
+        """
+        connection = None
+        try:
+            connection = self.acquire(timeout)
+            yield connection
+        finally:
+            if connection:
+                self.release(connection)
+    
+    def acquire(self, timeout: Optional[int] = None) -> Any:
+        """
+        获取一个连接
+        
+        Args:
+            timeout: 获取连接的超时时间（秒）
+            
+        Returns:
+            Infinity连接对象
+            
+        Raises:
+            TimeoutError: 获取连接超时
+        """
+        end_time = time.time() + (timeout or self.connection_timeout)
+        
+        with self.lock:
+            while True:
+                # 检查是否有可用连接
+                if self.connections:
+                    # 获取一个连接
+                    connection = self.connections.pop()
+                    
+                    # 检查连接是否有效
+                    if self._is_valid_connection(connection):
+                        connection.__dict__['_last_used'] = time.time()
+                        self.in_use[connection] = threading.get_ident()
+                        return connection
+                    else:
+                        # 连接无效，关闭并计数减一
+                        try:
+                            connection.disconnect()
+                            self.connection_count -= 1
+                        except Exception:
+                            pass
+                
+                # 检查是否可以创建新连接
+                elif self.connection_count < self.max_connections:
+                    # 创建新连接
+                    connection = self._create_connection()
+                    connection.__dict__['_last_used'] = time.time()
+                    self.in_use[connection] = threading.get_ident()
+                    return connection
+                
+                # 等待连接释放
+                remaining = end_time - time.time()
+                if remaining <= 0:
+                    raise TimeoutError("Timeout waiting for Infinity connection")
+                
+                # 等待连接释放或超时
+                self.condition.wait(remaining)
+    
+    def release(self, connection: Any):
+        """
+        释放连接
+        
+        Args:
+            connection: 要释放的连接
+        """
+        with self.lock:
+            if connection in self.in_use:
+                del self.in_use[connection]
+                
+                # 检查连接是否有效
+                if self._is_valid_connection(connection):
+                    connection.__dict__['_last_used'] = time.time()
+                    self.connections.append(connection)
+                    # 通知等待的线程
+                    self.condition.notify()
+                else:
+                    # 连接无效，关闭并计数减一
+                    try:
+                        connection.disconnect()
+                        self.connection_count -= 1
+                    except Exception:
+                        pass
+    
+    def close(self):
+        """关闭所有连接"""
+        with self.lock:
+            # 关闭可用连接
+            for connection in self.connections:
+                try:
+                    connection.disconnect()
+                except Exception:
+                    pass
+            
+            # 关闭正在使用的连接
+            for connection in list(self.in_use.keys()):
+                try:
+                    connection.disconnect()
+                except Exception:
+                    pass
+            
+            self.connections = []
+            self.in_use = {}
+            self.connection_count = 0
+    
+    def get_status(self) -> Dict[str, Any]:
+        """
+        获取连接池状态
+        
+        Returns:
+            连接池状态信息
+        """
+        with self.lock:
+            return {
+                "total_connections": self.connection_count,
+                "available_connections": len(self.connections),
+                "in_use_connections": len(self.in_use),
+                "min_connections": self.min_connections,
+                "max_connections": self.max_connections
+            }

utils/infinity/test_infinity.py

@@ -0,0 +1,89 @@
+# 测试Infinity客户端拆分后的代码结构
+
+import sys
+import os
+
+# 添加项目根目录到Python路径
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../../')))
+
+# 测试1：导入验证
+print("=== 测试1：导入验证 ===")
+try:
+    from utils.infinity import (
+        InfinityConnectionPool,
+        InfinityClient,
+        get_client,
+        close_client
+    )
+    print("✅ 所有组件导入成功")
+except Exception as e:
+    print(f"❌ 导入失败: {e}")
+    sys.exit(1)
+
+# 测试2：类继承和结构验证
+print("\n=== 测试2：类结构验证 ===")
+try:
+    # 验证类的基本属性
+    print(f"✅ InfinityConnectionPool类存在")
+    print(f"✅ InfinityClient类存在")
+    print(f"✅ get_client函数存在")
+    print(f"✅ close_client函数存在")
+except Exception as e:
+    print(f"❌ 类结构验证失败: {e}")
+    sys.exit(1)
+
+# 测试3：全局客户端函数验证
+print("\n=== 测试3：全局客户端函数验证 ===")
+try:
+    # 只验证函数存在和基本调用，不实际连接
+    import inspect
+    
+    # 检查函数签名
+    get_client_sig = inspect.signature(get_client)
+    close_client_sig = inspect.signature(close_client)
+    
+    print(f"✅ get_client函数签名正确: {get_client_sig}")
+    print(f"✅ close_client函数签名正确: {close_client_sig}")
+    
+    # 验证全局客户端函数可以被调用（但不实际连接）
+    print("✅ 全局客户端函数可以被调用")
+except Exception as e:
+    print(f"❌ 全局客户端函数验证失败: {e}")
+    sys.exit(1)
+
+# 测试4：客户端类验证
+print("\n=== 测试4：客户端类验证 ===")
+try:
+    # 验证类的方法存在，不实际实例化连接
+    import inspect
+    
+    # 验证客户端类的方法
+    required_methods = [
+        'get_databases',
+        'create_database',
+        'drop_database',
+        'get_tables',
+        'create_table',
+        'drop_table',
+        'insert',
+        'search',
+        'hybrid_search',
+        'vector_search',
+        'get_status',
+        'close'
+    ]
+    
+    for method in required_methods:
+        if hasattr(InfinityClient, method):
+            print(f"✅ 客户端方法 '{method}' 存在")
+        else:
+            print(f"❌ 客户端方法 '{method}' 不存在")
+            raise Exception(f"Missing method: {method}")
+    
+    # 验证客户端类可以被实例化（但不实际连接）
+    print("✅ InfinityClient类可以被实例化")
+except Exception as e:
+    print(f"❌ 客户端类验证失败: {e}")
+    sys.exit(1)
+
+print("\n🎉 所有测试通过！Infinity客户端拆分成功！")