chromadb-experimental 0.1.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (97) hide show
  1. checksums.yaml +7 -0
  2. data/lib/chromadb/admin_client.rb +6 -0
  3. data/lib/chromadb/client.rb +317 -0
  4. data/lib/chromadb/collection.rb +573 -0
  5. data/lib/chromadb/embedding_functions/chroma_bm25.rb +459 -0
  6. data/lib/chromadb/embedding_functions/chroma_cloud_qwen.rb +139 -0
  7. data/lib/chromadb/embedding_functions/chroma_cloud_splade.rb +121 -0
  8. data/lib/chromadb/embedding_functions.rb +121 -0
  9. data/lib/chromadb/errors.rb +120 -0
  10. data/lib/chromadb/http_client.rb +142 -0
  11. data/lib/chromadb/openapi/lib/chromadb/api/default_api.rb +2349 -0
  12. data/lib/chromadb/openapi/lib/chromadb/api_client.rb +392 -0
  13. data/lib/chromadb/openapi/lib/chromadb/api_error.rb +58 -0
  14. data/lib/chromadb/openapi/lib/chromadb/configuration.rb +295 -0
  15. data/lib/chromadb/openapi/lib/chromadb/models/add_collection_records_payload.rb +260 -0
  16. data/lib/chromadb/openapi/lib/chromadb/models/attach_function_request.rb +250 -0
  17. data/lib/chromadb/openapi/lib/chromadb/models/attach_function_response.rb +235 -0
  18. data/lib/chromadb/openapi/lib/chromadb/models/attached_function_api_response.rb +361 -0
  19. data/lib/chromadb/openapi/lib/chromadb/models/attached_function_info.rb +240 -0
  20. data/lib/chromadb/openapi/lib/chromadb/models/bool_inverted_index_type.rb +229 -0
  21. data/lib/chromadb/openapi/lib/chromadb/models/bool_value_type.rb +221 -0
  22. data/lib/chromadb/openapi/lib/chromadb/models/checklist_response.rb +245 -0
  23. data/lib/chromadb/openapi/lib/chromadb/models/collection.rb +315 -0
  24. data/lib/chromadb/openapi/lib/chromadb/models/collection_configuration.rb +240 -0
  25. data/lib/chromadb/openapi/lib/chromadb/models/create_collection_payload.rb +260 -0
  26. data/lib/chromadb/openapi/lib/chromadb/models/create_database_payload.rb +220 -0
  27. data/lib/chromadb/openapi/lib/chromadb/models/create_tenant_payload.rb +220 -0
  28. data/lib/chromadb/openapi/lib/chromadb/models/database.rb +240 -0
  29. data/lib/chromadb/openapi/lib/chromadb/models/detach_function_request.rb +221 -0
  30. data/lib/chromadb/openapi/lib/chromadb/models/detach_function_response.rb +220 -0
  31. data/lib/chromadb/openapi/lib/chromadb/models/embedding_function_new_configuration.rb +230 -0
  32. data/lib/chromadb/openapi/lib/chromadb/models/error_response.rb +230 -0
  33. data/lib/chromadb/openapi/lib/chromadb/models/float_inverted_index_type.rb +229 -0
  34. data/lib/chromadb/openapi/lib/chromadb/models/float_list_value_type.rb +221 -0
  35. data/lib/chromadb/openapi/lib/chromadb/models/float_value_type.rb +221 -0
  36. data/lib/chromadb/openapi/lib/chromadb/models/fork_collection_payload.rb +220 -0
  37. data/lib/chromadb/openapi/lib/chromadb/models/fts_index_type.rb +229 -0
  38. data/lib/chromadb/openapi/lib/chromadb/models/get_attached_function_response.rb +224 -0
  39. data/lib/chromadb/openapi/lib/chromadb/models/get_response.rb +270 -0
  40. data/lib/chromadb/openapi/lib/chromadb/models/get_tenant_response.rb +230 -0
  41. data/lib/chromadb/openapi/lib/chromadb/models/get_user_identity_response.rb +246 -0
  42. data/lib/chromadb/openapi/lib/chromadb/models/heartbeat_response.rb +235 -0
  43. data/lib/chromadb/openapi/lib/chromadb/models/hnsw_configuration.rb +330 -0
  44. data/lib/chromadb/openapi/lib/chromadb/models/hnsw_index_config.rb +371 -0
  45. data/lib/chromadb/openapi/lib/chromadb/models/include.rb +210 -0
  46. data/lib/chromadb/openapi/lib/chromadb/models/int_inverted_index_type.rb +229 -0
  47. data/lib/chromadb/openapi/lib/chromadb/models/int_value_type.rb +221 -0
  48. data/lib/chromadb/openapi/lib/chromadb/models/query_response.rb +280 -0
  49. data/lib/chromadb/openapi/lib/chromadb/models/raw_where_fields.rb +230 -0
  50. data/lib/chromadb/openapi/lib/chromadb/models/schema.rb +258 -0
  51. data/lib/chromadb/openapi/lib/chromadb/models/search_payload.rb +256 -0
  52. data/lib/chromadb/openapi/lib/chromadb/models/search_payload_filter.rb +230 -0
  53. data/lib/chromadb/openapi/lib/chromadb/models/search_payload_group_by.rb +230 -0
  54. data/lib/chromadb/openapi/lib/chromadb/models/search_payload_limit.rb +230 -0
  55. data/lib/chromadb/openapi/lib/chromadb/models/search_payload_select.rb +220 -0
  56. data/lib/chromadb/openapi/lib/chromadb/models/search_request_payload.rb +220 -0
  57. data/lib/chromadb/openapi/lib/chromadb/models/search_response.rb +270 -0
  58. data/lib/chromadb/openapi/lib/chromadb/models/space.rb +210 -0
  59. data/lib/chromadb/openapi/lib/chromadb/models/spann_configuration.rb +420 -0
  60. data/lib/chromadb/openapi/lib/chromadb/models/spann_index_config.rb +536 -0
  61. data/lib/chromadb/openapi/lib/chromadb/models/sparse_vector.rb +244 -0
  62. data/lib/chromadb/openapi/lib/chromadb/models/sparse_vector_index_config.rb +242 -0
  63. data/lib/chromadb/openapi/lib/chromadb/models/sparse_vector_index_type.rb +234 -0
  64. data/lib/chromadb/openapi/lib/chromadb/models/sparse_vector_value_type.rb +221 -0
  65. data/lib/chromadb/openapi/lib/chromadb/models/string_inverted_index_type.rb +229 -0
  66. data/lib/chromadb/openapi/lib/chromadb/models/string_value_type.rb +231 -0
  67. data/lib/chromadb/openapi/lib/chromadb/models/update_collection_configuration.rb +240 -0
  68. data/lib/chromadb/openapi/lib/chromadb/models/update_collection_payload.rb +240 -0
  69. data/lib/chromadb/openapi/lib/chromadb/models/update_collection_records_payload.rb +260 -0
  70. data/lib/chromadb/openapi/lib/chromadb/models/update_hnsw_configuration.rb +345 -0
  71. data/lib/chromadb/openapi/lib/chromadb/models/update_spann_configuration.rb +260 -0
  72. data/lib/chromadb/openapi/lib/chromadb/models/update_tenant_payload.rb +220 -0
  73. data/lib/chromadb/openapi/lib/chromadb/models/upsert_collection_records_payload.rb +260 -0
  74. data/lib/chromadb/openapi/lib/chromadb/models/value_types.rb +271 -0
  75. data/lib/chromadb/openapi/lib/chromadb/models/vector_index_config.rb +261 -0
  76. data/lib/chromadb/openapi/lib/chromadb/models/vector_index_type.rb +234 -0
  77. data/lib/chromadb/openapi/lib/chromadb/version.rb +15 -0
  78. data/lib/chromadb/openapi/lib/chromadb.rb +102 -0
  79. data/lib/chromadb/openapi.rb +6 -0
  80. data/lib/chromadb/schema.rb +744 -0
  81. data/lib/chromadb/schemas/chroma-cloud-qwen.json +61 -0
  82. data/lib/chromadb/schemas/chroma-cloud-splade.json +31 -0
  83. data/lib/chromadb/schemas/chroma_bm25.json +37 -0
  84. data/lib/chromadb/search/key.rb +94 -0
  85. data/lib/chromadb/search/limit.rb +41 -0
  86. data/lib/chromadb/search/rank.rb +425 -0
  87. data/lib/chromadb/search/search.rb +73 -0
  88. data/lib/chromadb/search/select.rb +54 -0
  89. data/lib/chromadb/search/where.rb +157 -0
  90. data/lib/chromadb/search.rb +8 -0
  91. data/lib/chromadb/types/results.rb +96 -0
  92. data/lib/chromadb/types/sparse_vector.rb +86 -0
  93. data/lib/chromadb/types/validation.rb +519 -0
  94. data/lib/chromadb/types.rb +13 -0
  95. data/lib/chromadb/version.rb +5 -0
  96. data/lib/chromadb.rb +15 -0
  97. metadata +233 -0
metadata ADDED
@@ -0,0 +1,233 @@
1
+ --- !ruby/object:Gem::Specification
2
+ name: chromadb-experimental
3
+ version: !ruby/object:Gem::Version
4
+ version: 0.1.0
5
+ platform: ruby
6
+ authors:
7
+ - Chroma Core
8
+ bindir: bin
9
+ cert_chain: []
10
+ date: 2025-12-31 00:00:00.000000000 Z
11
+ dependencies:
12
+ - !ruby/object:Gem::Dependency
13
+ name: faraday
14
+ requirement: !ruby/object:Gem::Requirement
15
+ requirements:
16
+ - - "~>"
17
+ - !ruby/object:Gem::Version
18
+ version: '2.0'
19
+ type: :runtime
20
+ prerelease: false
21
+ version_requirements: !ruby/object:Gem::Requirement
22
+ requirements:
23
+ - - "~>"
24
+ - !ruby/object:Gem::Version
25
+ version: '2.0'
26
+ - !ruby/object:Gem::Dependency
27
+ name: base64
28
+ requirement: !ruby/object:Gem::Requirement
29
+ requirements:
30
+ - - "~>"
31
+ - !ruby/object:Gem::Version
32
+ version: '0.2'
33
+ type: :runtime
34
+ prerelease: false
35
+ version_requirements: !ruby/object:Gem::Requirement
36
+ requirements:
37
+ - - "~>"
38
+ - !ruby/object:Gem::Version
39
+ version: '0.2'
40
+ - !ruby/object:Gem::Dependency
41
+ name: json_schemer
42
+ requirement: !ruby/object:Gem::Requirement
43
+ requirements:
44
+ - - "~>"
45
+ - !ruby/object:Gem::Version
46
+ version: '0.2'
47
+ type: :runtime
48
+ prerelease: false
49
+ version_requirements: !ruby/object:Gem::Requirement
50
+ requirements:
51
+ - - "~>"
52
+ - !ruby/object:Gem::Version
53
+ version: '0.2'
54
+ - !ruby/object:Gem::Dependency
55
+ name: porter2stemmer
56
+ requirement: !ruby/object:Gem::Requirement
57
+ requirements:
58
+ - - "~>"
59
+ - !ruby/object:Gem::Version
60
+ version: '1.0'
61
+ type: :runtime
62
+ prerelease: false
63
+ version_requirements: !ruby/object:Gem::Requirement
64
+ requirements:
65
+ - - "~>"
66
+ - !ruby/object:Gem::Version
67
+ version: '1.0'
68
+ - !ruby/object:Gem::Dependency
69
+ name: rspec
70
+ requirement: !ruby/object:Gem::Requirement
71
+ requirements:
72
+ - - "~>"
73
+ - !ruby/object:Gem::Version
74
+ version: '3.13'
75
+ type: :development
76
+ prerelease: false
77
+ version_requirements: !ruby/object:Gem::Requirement
78
+ requirements:
79
+ - - "~>"
80
+ - !ruby/object:Gem::Version
81
+ version: '3.13'
82
+ - !ruby/object:Gem::Dependency
83
+ name: rubocop-rails-omakase
84
+ requirement: !ruby/object:Gem::Requirement
85
+ requirements:
86
+ - - "~>"
87
+ - !ruby/object:Gem::Version
88
+ version: '1.1'
89
+ type: :development
90
+ prerelease: false
91
+ version_requirements: !ruby/object:Gem::Requirement
92
+ requirements:
93
+ - - "~>"
94
+ - !ruby/object:Gem::Version
95
+ version: '1.1'
96
+ - !ruby/object:Gem::Dependency
97
+ name: webmock
98
+ requirement: !ruby/object:Gem::Requirement
99
+ requirements:
100
+ - - "~>"
101
+ - !ruby/object:Gem::Version
102
+ version: '3.23'
103
+ type: :development
104
+ prerelease: false
105
+ version_requirements: !ruby/object:Gem::Requirement
106
+ requirements:
107
+ - - "~>"
108
+ - !ruby/object:Gem::Version
109
+ version: '3.23'
110
+ description: Ruby client for Chroma's HTTP API (dense + sparse embeddings).
111
+ email:
112
+ - support@trychroma.com
113
+ executables: []
114
+ extensions: []
115
+ extra_rdoc_files: []
116
+ files:
117
+ - lib/chromadb.rb
118
+ - lib/chromadb/admin_client.rb
119
+ - lib/chromadb/client.rb
120
+ - lib/chromadb/collection.rb
121
+ - lib/chromadb/embedding_functions.rb
122
+ - lib/chromadb/embedding_functions/chroma_bm25.rb
123
+ - lib/chromadb/embedding_functions/chroma_cloud_qwen.rb
124
+ - lib/chromadb/embedding_functions/chroma_cloud_splade.rb
125
+ - lib/chromadb/errors.rb
126
+ - lib/chromadb/http_client.rb
127
+ - lib/chromadb/openapi.rb
128
+ - lib/chromadb/openapi/lib/chromadb.rb
129
+ - lib/chromadb/openapi/lib/chromadb/api/default_api.rb
130
+ - lib/chromadb/openapi/lib/chromadb/api_client.rb
131
+ - lib/chromadb/openapi/lib/chromadb/api_error.rb
132
+ - lib/chromadb/openapi/lib/chromadb/configuration.rb
133
+ - lib/chromadb/openapi/lib/chromadb/models/add_collection_records_payload.rb
134
+ - lib/chromadb/openapi/lib/chromadb/models/attach_function_request.rb
135
+ - lib/chromadb/openapi/lib/chromadb/models/attach_function_response.rb
136
+ - lib/chromadb/openapi/lib/chromadb/models/attached_function_api_response.rb
137
+ - lib/chromadb/openapi/lib/chromadb/models/attached_function_info.rb
138
+ - lib/chromadb/openapi/lib/chromadb/models/bool_inverted_index_type.rb
139
+ - lib/chromadb/openapi/lib/chromadb/models/bool_value_type.rb
140
+ - lib/chromadb/openapi/lib/chromadb/models/checklist_response.rb
141
+ - lib/chromadb/openapi/lib/chromadb/models/collection.rb
142
+ - lib/chromadb/openapi/lib/chromadb/models/collection_configuration.rb
143
+ - lib/chromadb/openapi/lib/chromadb/models/create_collection_payload.rb
144
+ - lib/chromadb/openapi/lib/chromadb/models/create_database_payload.rb
145
+ - lib/chromadb/openapi/lib/chromadb/models/create_tenant_payload.rb
146
+ - lib/chromadb/openapi/lib/chromadb/models/database.rb
147
+ - lib/chromadb/openapi/lib/chromadb/models/detach_function_request.rb
148
+ - lib/chromadb/openapi/lib/chromadb/models/detach_function_response.rb
149
+ - lib/chromadb/openapi/lib/chromadb/models/embedding_function_new_configuration.rb
150
+ - lib/chromadb/openapi/lib/chromadb/models/error_response.rb
151
+ - lib/chromadb/openapi/lib/chromadb/models/float_inverted_index_type.rb
152
+ - lib/chromadb/openapi/lib/chromadb/models/float_list_value_type.rb
153
+ - lib/chromadb/openapi/lib/chromadb/models/float_value_type.rb
154
+ - lib/chromadb/openapi/lib/chromadb/models/fork_collection_payload.rb
155
+ - lib/chromadb/openapi/lib/chromadb/models/fts_index_type.rb
156
+ - lib/chromadb/openapi/lib/chromadb/models/get_attached_function_response.rb
157
+ - lib/chromadb/openapi/lib/chromadb/models/get_response.rb
158
+ - lib/chromadb/openapi/lib/chromadb/models/get_tenant_response.rb
159
+ - lib/chromadb/openapi/lib/chromadb/models/get_user_identity_response.rb
160
+ - lib/chromadb/openapi/lib/chromadb/models/heartbeat_response.rb
161
+ - lib/chromadb/openapi/lib/chromadb/models/hnsw_configuration.rb
162
+ - lib/chromadb/openapi/lib/chromadb/models/hnsw_index_config.rb
163
+ - lib/chromadb/openapi/lib/chromadb/models/include.rb
164
+ - lib/chromadb/openapi/lib/chromadb/models/int_inverted_index_type.rb
165
+ - lib/chromadb/openapi/lib/chromadb/models/int_value_type.rb
166
+ - lib/chromadb/openapi/lib/chromadb/models/query_response.rb
167
+ - lib/chromadb/openapi/lib/chromadb/models/raw_where_fields.rb
168
+ - lib/chromadb/openapi/lib/chromadb/models/schema.rb
169
+ - lib/chromadb/openapi/lib/chromadb/models/search_payload.rb
170
+ - lib/chromadb/openapi/lib/chromadb/models/search_payload_filter.rb
171
+ - lib/chromadb/openapi/lib/chromadb/models/search_payload_group_by.rb
172
+ - lib/chromadb/openapi/lib/chromadb/models/search_payload_limit.rb
173
+ - lib/chromadb/openapi/lib/chromadb/models/search_payload_select.rb
174
+ - lib/chromadb/openapi/lib/chromadb/models/search_request_payload.rb
175
+ - lib/chromadb/openapi/lib/chromadb/models/search_response.rb
176
+ - lib/chromadb/openapi/lib/chromadb/models/space.rb
177
+ - lib/chromadb/openapi/lib/chromadb/models/spann_configuration.rb
178
+ - lib/chromadb/openapi/lib/chromadb/models/spann_index_config.rb
179
+ - lib/chromadb/openapi/lib/chromadb/models/sparse_vector.rb
180
+ - lib/chromadb/openapi/lib/chromadb/models/sparse_vector_index_config.rb
181
+ - lib/chromadb/openapi/lib/chromadb/models/sparse_vector_index_type.rb
182
+ - lib/chromadb/openapi/lib/chromadb/models/sparse_vector_value_type.rb
183
+ - lib/chromadb/openapi/lib/chromadb/models/string_inverted_index_type.rb
184
+ - lib/chromadb/openapi/lib/chromadb/models/string_value_type.rb
185
+ - lib/chromadb/openapi/lib/chromadb/models/update_collection_configuration.rb
186
+ - lib/chromadb/openapi/lib/chromadb/models/update_collection_payload.rb
187
+ - lib/chromadb/openapi/lib/chromadb/models/update_collection_records_payload.rb
188
+ - lib/chromadb/openapi/lib/chromadb/models/update_hnsw_configuration.rb
189
+ - lib/chromadb/openapi/lib/chromadb/models/update_spann_configuration.rb
190
+ - lib/chromadb/openapi/lib/chromadb/models/update_tenant_payload.rb
191
+ - lib/chromadb/openapi/lib/chromadb/models/upsert_collection_records_payload.rb
192
+ - lib/chromadb/openapi/lib/chromadb/models/value_types.rb
193
+ - lib/chromadb/openapi/lib/chromadb/models/vector_index_config.rb
194
+ - lib/chromadb/openapi/lib/chromadb/models/vector_index_type.rb
195
+ - lib/chromadb/openapi/lib/chromadb/version.rb
196
+ - lib/chromadb/schema.rb
197
+ - lib/chromadb/schemas/chroma-cloud-qwen.json
198
+ - lib/chromadb/schemas/chroma-cloud-splade.json
199
+ - lib/chromadb/schemas/chroma_bm25.json
200
+ - lib/chromadb/search.rb
201
+ - lib/chromadb/search/key.rb
202
+ - lib/chromadb/search/limit.rb
203
+ - lib/chromadb/search/rank.rb
204
+ - lib/chromadb/search/search.rb
205
+ - lib/chromadb/search/select.rb
206
+ - lib/chromadb/search/where.rb
207
+ - lib/chromadb/types.rb
208
+ - lib/chromadb/types/results.rb
209
+ - lib/chromadb/types/sparse_vector.rb
210
+ - lib/chromadb/types/validation.rb
211
+ - lib/chromadb/version.rb
212
+ homepage: https://github.com/chroma-core/chroma
213
+ licenses:
214
+ - Apache-2.0
215
+ metadata: {}
216
+ rdoc_options: []
217
+ require_paths:
218
+ - lib
219
+ required_ruby_version: !ruby/object:Gem::Requirement
220
+ requirements:
221
+ - - ">="
222
+ - !ruby/object:Gem::Version
223
+ version: '3.0'
224
+ required_rubygems_version: !ruby/object:Gem::Requirement
225
+ requirements:
226
+ - - ">="
227
+ - !ruby/object:Gem::Version
228
+ version: '0'
229
+ requirements: []
230
+ rubygems_version: 3.6.2
231
+ specification_version: 4
232
+ summary: Chroma Ruby client
233
+ test_files: []