LLMUnity/Runtime/RAG/usearch/NativeMethods.cs at e7612980e4473dda800faa2185aa219fa2e05db0 · undreamai/LLMUnity · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
using System.Runtime.InteropServices;

using usearch_index_t = System.IntPtr;
using usearch_key_t = System.UInt64;
using usearch_error_t = System.IntPtr;
using size_t = System.UIntPtr;
using void_ptr_t = System.IntPtr;
using usearch_distance_t = System.Single;

namespace Cloud.Unum.USearch
{
    public static class NativeMethods
    {
#if UNITY_IOS
        private const string LibraryName = "__Internal";
#else
        private const string LibraryName = "libusearch_c";
#endif

        [DllImport(LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern usearch_index_t usearch_init(ref IndexOptions options, out usearch_error_t error);

        [DllImport(LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern void usearch_free(usearch_index_t index, out usearch_error_t error);

        [DllImport(LibraryName, CharSet = CharSet.Ansi, CallingConvention = CallingConvention.Cdecl)]
        public static extern void usearch_save(usearch_index_t index, [MarshalAs(UnmanagedType.LPStr)] string path, out usearch_error_t error);

        [DllImport(LibraryName, CharSet = CharSet.Ansi, CallingConvention = CallingConvention.Cdecl)]
        public static extern void usearch_load(usearch_index_t index, [MarshalAs(UnmanagedType.LPStr)] string path, out usearch_error_t error);

        [DllImport(LibraryName, CharSet = CharSet.Ansi, CallingConvention = CallingConvention.Cdecl)]
        public static extern void usearch_view(usearch_index_t index, [MarshalAs(UnmanagedType.LPStr)] string path, out usearch_error_t error);

        [DllImport(LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern size_t usearch_size(usearch_index_t index, out usearch_error_t error);

        [DllImport(LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern size_t usearch_capacity(usearch_index_t index, out usearch_error_t error);

        [DllImport(LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern size_t usearch_dimensions(usearch_index_t index, out usearch_error_t error);

        [DllImport(LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern size_t usearch_connectivity(usearch_index_t index, out usearch_error_t error);

        [DllImport(LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern void usearch_reserve(usearch_index_t index, size_t capacity, out usearch_error_t error);

        [DllImport(LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern void usearch_add(
            usearch_index_t index,
            usearch_key_t key,
            [In] float[] vector,
            ScalarKind vector_kind,
            out usearch_error_t error
        );

        [DllImport(LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern void usearch_add(
            usearch_index_t index,
            usearch_key_t key,
            [In] double[] vector,
            ScalarKind vector_kind,
            out usearch_error_t error
        );

        [DllImport(LibraryName, CallingConvention = CallingConvention.Cdecl)]
        [return : MarshalAs(UnmanagedType.I1)]
        public static extern bool usearch_contains(usearch_index_t index, usearch_key_t key, out usearch_error_t error);

        [DllImport(LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern size_t usearch_count(usearch_index_t index, usearch_key_t key, out usearch_error_t error);

        [DllImport(LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern size_t usearch_search(
            usearch_index_t index,
            void_ptr_t query_vector,
            ScalarKind query_kind,
            size_t count,
            [Out] usearch_key_t[] found_keys,
            [Out] usearch_distance_t[] found_distances,
            out usearch_error_t error
        );

        [DllImport(LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern size_t usearch_search(
            usearch_index_t index,
            [In] float[] query_vector,
            ScalarKind query_kind,
            size_t count,
            [Out] usearch_key_t[] found_keys,
            [Out] usearch_distance_t[] found_distances,
            out usearch_error_t error
        );

        [DllImport(LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern size_t usearch_search(
            usearch_index_t index,
            [In] double[] query_vector,
            ScalarKind query_kind,
            size_t count,
            [Out] usearch_key_t[] found_keys,
            [Out] usearch_distance_t[] found_distances,
            out usearch_error_t error
        );


        [DllImport(LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern size_t usearch_get(
            usearch_index_t index,
            usearch_key_t key,
            size_t count,
            [Out] float[] vector,
            ScalarKind vector_kind,
            out usearch_error_t error
        );

        [DllImport(LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern size_t usearch_get(
            usearch_index_t index,
            usearch_key_t key,
            size_t count,
            [Out] double[] vector,
            ScalarKind vector_kind,
            out usearch_error_t error
        );

        [DllImport(LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern size_t usearch_remove(usearch_index_t index, usearch_key_t key, out usearch_error_t error);

        [DllImport(LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern size_t usearch_rename(usearch_index_t index, usearch_key_t key_from, usearch_key_t key_to, out usearch_error_t error);

        //========================== Additional methods from LLMUnity ==========================//

        [UnmanagedFunctionPointer(CallingConvention.Cdecl)]
        public delegate int FilterCallback(int key, void_ptr_t filterState);

        [DllImport(LibraryName, CharSet = CharSet.Ansi, CallingConvention = CallingConvention.Cdecl)]
        public static extern void usearch_load_buffer(usearch_index_t index, void_ptr_t buffer, size_t length, out usearch_error_t error);

        [DllImport(LibraryName, CharSet = CharSet.Ansi, CallingConvention = CallingConvention.Cdecl)]
        public static extern void usearch_view_buffer(usearch_index_t index, void_ptr_t buffer, size_t length, out usearch_error_t error);

        [DllImport(LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern size_t usearch_filtered_search(
            usearch_index_t index,
            void_ptr_t query_vector,
            ScalarKind query_kind,
            size_t count,
            FilterCallback filter,
            void_ptr_t filterState,
            [Out] usearch_key_t[] found_keys,
            [Out] usearch_distance_t[] found_distances,
            out usearch_error_t error
        );
    }
}