16
16
import infinity
17
17
import infinity .index as index
18
18
from infinity .common import REMOTE_HOST
19
+ from infinity .common import ConflictType
19
20
import pandas as pds
20
21
21
22
22
23
def main ():
23
24
infinity_obj = infinity .connect (REMOTE_HOST )
24
25
db = infinity_obj .get_database ("default" )
25
- db .drop_table ("my_table" , if_exists = True )
26
- table = db .create_table (
27
- "my_table" , {"num" : "integer" , "body" : "varchar" , "vec" : "vector,5,float" }, None )
28
- table .insert (
29
- [{"num" : 1 , "body" : "undesirable, unnecessary, and harmful" , "vec" : [1.0 ] * 5 }])
30
- table .insert (
31
- [{"num" : 2 , "body" : "publisher=US National Office for Harmful Algal Blooms" , "vec" : [4.0 ] * 5 }])
32
- table .insert (
33
- [{"num" : 3 , "body" : "in the case of plants, growth and chemical" , "vec" : [7.0 ] * 5 }])
34
-
35
- res = table .output (["*" ]).knn ("vec" , [3.0 ] * 5 , "float" , "ip" , 2 ).to_pl ()
26
+ # Drop my_table if it already exists
27
+ db .drop_table ("my_table" , ConflictType .Ignore )
28
+ # Create a table named "my_table"
29
+ table = db .create_table ("my_table" , {"num" : "integer" , "body" : "varchar" , "vec" : "vector, 4, float" })
30
+ table .insert ([{"num" : 1 , "body" : "unnecessary and harmful" , "vec" : [1.0 , 1.2 , 0.8 , 0.9 ]}])
31
+ table .insert ([{"num" : 2 , "body" : "Office for Harmful Blooms" , "vec" : [4.0 , 4.2 , 4.3 , 4.5 ]}])
32
+
33
+ res = table .output (["*" ]).knn ("vec" , [3.0 , 2.8 , 2.7 , 3.1 ], "float" , "ip" , 2 ).to_pl ()
36
34
print (res )
37
35
38
36
@@ -41,21 +39,18 @@ def test():
41
39
infinity_obj = infinity .connect (REMOTE_HOST )
42
40
db = infinity_obj .get_database ("default" )
43
41
# Drop my_table if it already exists
44
- db .drop_table ("my_table" , if_exists = True )
42
+ db .drop_table ("my_table" , ConflictType . Ignore )
45
43
# Create a table named "my_table"
46
- table = db .create_table ("my_table" , {
47
- "num" : "integer" , "body" : "varchar" , "vec" : "vector, 4, float" }, None )
48
- table .insert (
49
- [{"num" : 1 , "body" : "unnecessary and harmful" , "vec" : [1.0 , 1.2 , 0.8 , 0.9 ]}])
50
- table .insert (
51
- [{"num" : 2 , "body" : "Office for Harmful Blooms" , "vec" : [4.0 , 4.2 , 4.3 , 4.5 ]}])
44
+ table = db .create_table ("my_table" , {"num" : "integer" , "body" : "varchar" , "vec" : "vector, 4, float" })
45
+ table .insert ([{"num" : 1 , "body" : "unnecessary and harmful" , "vec" : [1.0 , 1.2 , 0.8 , 0.9 ]}])
46
+ table .insert ([{"num" : 2 , "body" : "Office for Harmful Blooms" , "vec" : [4.0 , 4.2 , 4.3 , 4.5 ]}])
52
47
53
48
# `create_index()` is required before match() or fusion()
54
- # res = table.create_index("my_index",
55
- # [index.IndexInfo("body",
56
- # index.IndexType.FullText,
57
- # [index.InitParameter("ANALYZER", "segmentation")]),
58
- # ], None )
49
+ res = table .create_index ("my_index" ,
50
+ [index .IndexInfo ("body" ,
51
+ index .IndexType .FullText ,
52
+ [index .InitParameter ("ANALYZER" , "segmentation" )]),
53
+ ], ConflictType . Error )
59
54
# assert res.success
60
55
61
56
res = table .output (["num" , "body" ]).knn (
@@ -97,5 +92,5 @@ def test():
97
92
98
93
99
94
if __name__ == '__main__' :
100
- # main()
95
+ main ()
101
96
test ()
0 commit comments