trieve_ruby_client 0.8.0 → 0.10.7

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 (229) hide show
  1. checksums.yaml +4 -4
  2. data/README.md +39 -30
  3. data/docs/ApiKeyRespBody.md +32 -0
  4. data/docs/AutocompleteData.md +42 -0
  5. data/docs/AutocompleteReqPayload.md +50 -0
  6. data/docs/BoostPhrase.md +20 -0
  7. data/docs/ChunkApi.md +140 -66
  8. data/docs/ChunkData.md +2 -0
  9. data/docs/ChunkFilter.md +3 -3
  10. data/docs/ChunkGroup.md +1 -1
  11. data/docs/ChunkGroupApi.md +35 -33
  12. data/docs/ChunkMetadata.md +5 -3
  13. data/docs/ChunkMetadataStringTagSet.md +46 -0
  14. data/docs/ChunkMetadataTypes.md +51 -0
  15. data/docs/ChunkMetadataWithScore.md +0 -2
  16. data/docs/ChunkReqPayload.md +50 -0
  17. data/docs/ClientDatasetConfiguration.md +0 -2
  18. data/docs/ConditionType.md +49 -0
  19. data/docs/ContentChunkMetadata.md +32 -0
  20. data/docs/CreateChunkGroupReqPayload.md +28 -0
  21. data/docs/CreateChunkReqPayloadEnum.md +49 -0
  22. data/docs/CreateDatasetRequest.md +2 -2
  23. data/docs/CreateMessageData.md +2 -2
  24. data/docs/CreateMessageReqPayload.md +42 -0
  25. data/docs/CreateTopicReqPayload.md +22 -0
  26. data/docs/Dataset.md +2 -0
  27. data/docs/DatasetApi.md +236 -12
  28. data/docs/DateRange.md +24 -0
  29. data/docs/EditMessageReqPayload.md +44 -0
  30. data/docs/FieldCondition.md +2 -0
  31. data/docs/File.md +1 -1
  32. data/docs/FileApi.md +7 -7
  33. data/docs/GenerateChunksRequest.md +2 -0
  34. data/docs/GetDatasetsPagination.md +20 -0
  35. data/docs/GroupScoreSlimChunks.md +4 -0
  36. data/docs/HasIDCondition.md +20 -0
  37. data/docs/InvitationApi.md +1 -1
  38. data/docs/MatchCondition.md +3 -1
  39. data/docs/MessageApi.md +42 -42
  40. data/docs/Organization.md +2 -0
  41. data/docs/OrganizationApi.md +24 -25
  42. data/docs/RangeCondition.md +1 -3
  43. data/docs/RecommendChunksRequest.md +1 -1
  44. data/docs/RecommendGroupChunksRequest.md +1 -1
  45. data/docs/RegenerateMessageReqPayload.md +40 -0
  46. data/docs/RemoveChunkFromGroupReqPayload.md +18 -0
  47. data/docs/ScoreChunkDTO.md +3 -1
  48. data/docs/SearchChunkData.md +12 -6
  49. data/docs/SearchChunksReqPayload.md +54 -0
  50. data/docs/SearchOverGroupsData.md +12 -4
  51. data/docs/SearchWithinGroupData.md +16 -6
  52. data/docs/SetUserApiKeyRequest.md +4 -0
  53. data/docs/SlimChunkMetadata.md +4 -0
  54. data/docs/SuggestedQueriesReqPayload.md +18 -0
  55. data/docs/TopicApi.md +16 -16
  56. data/docs/UpdateChunkData.md +2 -0
  57. data/docs/UpdateChunkReqPayload.md +46 -0
  58. data/docs/UpdateDatasetRequest.md +2 -0
  59. data/docs/UpdateGroupByTrackingIDReqPayload.md +26 -0
  60. data/docs/UpdateTopicReqPayload.md +20 -0
  61. data/docs/UpdateUserOrgRoleData.md +1 -1
  62. data/docs/UploadFileReqPayload.md +40 -0
  63. data/docs/UserApi.md +7 -8
  64. data/lib/trieve_ruby_client/api/auth_api.rb +1 -1
  65. data/lib/trieve_ruby_client/api/chunk_api.rb +158 -83
  66. data/lib/trieve_ruby_client/api/chunk_group_api.rb +58 -51
  67. data/lib/trieve_ruby_client/api/dataset_api.rb +224 -15
  68. data/lib/trieve_ruby_client/api/events_api.rb +1 -1
  69. data/lib/trieve_ruby_client/api/file_api.rb +12 -12
  70. data/lib/trieve_ruby_client/api/health_api.rb +1 -1
  71. data/lib/trieve_ruby_client/api/invitation_api.rb +3 -3
  72. data/lib/trieve_ruby_client/api/message_api.rb +50 -50
  73. data/lib/trieve_ruby_client/api/organization_api.rb +27 -27
  74. data/lib/trieve_ruby_client/api/stripe_api.rb +1 -1
  75. data/lib/trieve_ruby_client/api/topic_api.rb +27 -27
  76. data/lib/trieve_ruby_client/api/user_api.rb +8 -8
  77. data/lib/trieve_ruby_client/api_client.rb +1 -1
  78. data/lib/trieve_ruby_client/api_error.rb +1 -1
  79. data/lib/trieve_ruby_client/configuration.rb +1 -1
  80. data/lib/trieve_ruby_client/models/add_chunk_to_group_data.rb +1 -1
  81. data/lib/trieve_ruby_client/models/api_key_dto.rb +1 -1
  82. data/lib/trieve_ruby_client/models/api_key_resp_body.rb +325 -0
  83. data/lib/trieve_ruby_client/models/auth_query.rb +1 -1
  84. data/lib/trieve_ruby_client/models/autocomplete_data.rb +378 -0
  85. data/lib/trieve_ruby_client/models/autocomplete_req_payload.rb +467 -0
  86. data/lib/trieve_ruby_client/models/batch_queued_chunk_response.rb +1 -1
  87. data/lib/trieve_ruby_client/models/bookmark_data.rb +1 -1
  88. data/lib/trieve_ruby_client/models/bookmark_group_result.rb +1 -1
  89. data/lib/trieve_ruby_client/models/boost_phrase.rb +240 -0
  90. data/lib/trieve_ruby_client/models/chat_message_proxy.rb +1 -1
  91. data/lib/trieve_ruby_client/models/chunk_data.rb +15 -2
  92. data/lib/trieve_ruby_client/models/chunk_filter.rb +4 -4
  93. data/lib/trieve_ruby_client/models/chunk_group.rb +5 -3
  94. data/lib/trieve_ruby_client/models/chunk_group_and_file.rb +1 -1
  95. data/lib/trieve_ruby_client/models/chunk_metadata.rb +28 -20
  96. data/lib/trieve_ruby_client/models/chunk_metadata_string_tag_set.rb +387 -0
  97. data/lib/trieve_ruby_client/models/chunk_metadata_types.rb +106 -0
  98. data/lib/trieve_ruby_client/models/chunk_metadata_with_score.rb +2 -18
  99. data/lib/trieve_ruby_client/models/chunk_req_payload.rb +400 -0
  100. data/lib/trieve_ruby_client/models/client_dataset_configuration.rb +2 -18
  101. data/lib/trieve_ruby_client/models/condition_type.rb +105 -0
  102. data/lib/trieve_ruby_client/models/content_chunk_metadata.rb +299 -0
  103. data/lib/trieve_ruby_client/models/create_chunk_data.rb +1 -1
  104. data/lib/trieve_ruby_client/models/create_chunk_group_data.rb +1 -1
  105. data/lib/trieve_ruby_client/models/create_chunk_group_req_payload.rb +273 -0
  106. data/lib/trieve_ruby_client/models/create_chunk_req_payload_enum.rb +105 -0
  107. data/lib/trieve_ruby_client/models/create_dataset_request.rb +3 -3
  108. data/lib/trieve_ruby_client/models/create_message_data.rb +13 -13
  109. data/lib/trieve_ruby_client/models/create_message_req_payload.rb +378 -0
  110. data/lib/trieve_ruby_client/models/create_organization_data.rb +1 -1
  111. data/lib/trieve_ruby_client/models/create_topic_data.rb +1 -1
  112. data/lib/trieve_ruby_client/models/create_topic_req_payload.rb +244 -0
  113. data/lib/trieve_ruby_client/models/dataset.rb +18 -2
  114. data/lib/trieve_ruby_client/models/dataset_and_usage.rb +1 -1
  115. data/lib/trieve_ruby_client/models/dataset_dto.rb +1 -1
  116. data/lib/trieve_ruby_client/models/dataset_usage_count.rb +1 -1
  117. data/lib/trieve_ruby_client/models/date_range.rb +245 -0
  118. data/lib/trieve_ruby_client/models/delete_topic_data.rb +1 -1
  119. data/lib/trieve_ruby_client/models/delete_user_api_key_request.rb +1 -1
  120. data/lib/trieve_ruby_client/models/edit_message_data.rb +1 -1
  121. data/lib/trieve_ruby_client/models/edit_message_req_payload.rb +395 -0
  122. data/lib/trieve_ruby_client/models/error_response_body.rb +1 -1
  123. data/lib/trieve_ruby_client/models/event.rb +1 -1
  124. data/lib/trieve_ruby_client/models/event_return.rb +1 -1
  125. data/lib/trieve_ruby_client/models/field_condition.rb +12 -2
  126. data/lib/trieve_ruby_client/models/file.rb +5 -3
  127. data/lib/trieve_ruby_client/models/file_dto.rb +1 -1
  128. data/lib/trieve_ruby_client/models/generate_chunks_request.rb +13 -2
  129. data/lib/trieve_ruby_client/models/geo_info.rb +1 -1
  130. data/lib/trieve_ruby_client/models/geo_types.rb +1 -1
  131. data/lib/trieve_ruby_client/models/get_chunks_data.rb +1 -1
  132. data/lib/trieve_ruby_client/models/get_datasets_pagination.rb +225 -0
  133. data/lib/trieve_ruby_client/models/get_events_data.rb +1 -1
  134. data/lib/trieve_ruby_client/models/get_groups_for_chunks_data.rb +1 -1
  135. data/lib/trieve_ruby_client/models/get_tracking_chunks_data.rb +1 -1
  136. data/lib/trieve_ruby_client/models/group_data.rb +1 -1
  137. data/lib/trieve_ruby_client/models/group_score_chunk.rb +1 -1
  138. data/lib/trieve_ruby_client/models/group_score_slim_chunks.rb +22 -2
  139. data/lib/trieve_ruby_client/models/has_id_condition.rb +229 -0
  140. data/lib/trieve_ruby_client/models/invitation_data.rb +1 -1
  141. data/lib/trieve_ruby_client/models/location_bounding_box.rb +1 -1
  142. data/lib/trieve_ruby_client/models/location_polygon.rb +1 -1
  143. data/lib/trieve_ruby_client/models/location_radius.rb +1 -1
  144. data/lib/trieve_ruby_client/models/match_condition.rb +2 -1
  145. data/lib/trieve_ruby_client/models/message.rb +1 -1
  146. data/lib/trieve_ruby_client/models/organization.rb +18 -2
  147. data/lib/trieve_ruby_client/models/organization_usage_count.rb +1 -1
  148. data/lib/trieve_ruby_client/models/range.rb +1 -1
  149. data/lib/trieve_ruby_client/models/range_condition.rb +2 -3
  150. data/lib/trieve_ruby_client/models/recommend_chunks_request.rb +2 -2
  151. data/lib/trieve_ruby_client/models/recommend_group_chunks_request.rb +2 -2
  152. data/lib/trieve_ruby_client/models/regenerate_message_data.rb +1 -1
  153. data/lib/trieve_ruby_client/models/regenerate_message_req_payload.rb +361 -0
  154. data/lib/trieve_ruby_client/models/remove_chunk_from_group_req_payload.rb +222 -0
  155. data/lib/trieve_ruby_client/models/return_queued_chunk.rb +1 -1
  156. data/lib/trieve_ruby_client/models/score_chunk_dto.rb +15 -3
  157. data/lib/trieve_ruby_client/models/score_slim_chunks.rb +1 -1
  158. data/lib/trieve_ruby_client/models/search_chunk_data.rb +49 -14
  159. data/lib/trieve_ruby_client/models/search_chunk_query_response_body.rb +1 -1
  160. data/lib/trieve_ruby_client/models/search_chunks_req_payload.rb +504 -0
  161. data/lib/trieve_ruby_client/models/search_over_groups_data.rb +95 -6
  162. data/lib/trieve_ruby_client/models/search_over_groups_results.rb +1 -1
  163. data/lib/trieve_ruby_client/models/search_over_groups_slim_results.rb +1 -1
  164. data/lib/trieve_ruby_client/models/search_slim_chunk_query_response_body.rb +1 -1
  165. data/lib/trieve_ruby_client/models/search_within_group_data.rb +119 -17
  166. data/lib/trieve_ruby_client/models/search_within_group_results.rb +1 -1
  167. data/lib/trieve_ruby_client/models/search_within_group_slim_results.rb +1 -1
  168. data/lib/trieve_ruby_client/models/set_user_api_key_request.rb +28 -2
  169. data/lib/trieve_ruby_client/models/set_user_api_key_response.rb +1 -1
  170. data/lib/trieve_ruby_client/models/single_queued_chunk_response.rb +1 -1
  171. data/lib/trieve_ruby_client/models/slim_chunk_metadata.rb +24 -2
  172. data/lib/trieve_ruby_client/models/slim_chunk_metadata_with_score.rb +1 -1
  173. data/lib/trieve_ruby_client/models/slim_group.rb +1 -1
  174. data/lib/trieve_ruby_client/models/slim_user.rb +1 -1
  175. data/lib/trieve_ruby_client/models/stripe_plan.rb +1 -1
  176. data/lib/trieve_ruby_client/models/suggested_queries_req_payload.rb +222 -0
  177. data/lib/trieve_ruby_client/models/suggested_queries_request.rb +1 -1
  178. data/lib/trieve_ruby_client/models/suggested_queries_response.rb +1 -1
  179. data/lib/trieve_ruby_client/models/topic.rb +1 -1
  180. data/lib/trieve_ruby_client/models/update_chunk_by_tracking_id_data.rb +1 -1
  181. data/lib/trieve_ruby_client/models/update_chunk_data.rb +15 -2
  182. data/lib/trieve_ruby_client/models/update_chunk_group_data.rb +1 -1
  183. data/lib/trieve_ruby_client/models/update_chunk_req_payload.rb +376 -0
  184. data/lib/trieve_ruby_client/models/update_dataset_request.rb +13 -2
  185. data/lib/trieve_ruby_client/models/update_group_by_tracking_id_data.rb +1 -1
  186. data/lib/trieve_ruby_client/models/update_group_by_tracking_id_req_payload.rb +268 -0
  187. data/lib/trieve_ruby_client/models/update_organization_data.rb +1 -1
  188. data/lib/trieve_ruby_client/models/update_topic_data.rb +1 -1
  189. data/lib/trieve_ruby_client/models/update_topic_req_payload.rb +239 -0
  190. data/lib/trieve_ruby_client/models/update_user_org_role_data.rb +2 -2
  191. data/lib/trieve_ruby_client/models/upload_file_data.rb +1 -1
  192. data/lib/trieve_ruby_client/models/upload_file_req_payload.rb +368 -0
  193. data/lib/trieve_ruby_client/models/upload_file_result.rb +1 -1
  194. data/lib/trieve_ruby_client/models/user_organization.rb +1 -1
  195. data/lib/trieve_ruby_client/version.rb +2 -2
  196. data/lib/trieve_ruby_client.rb +25 -20
  197. data/openapi-generator.yaml +1 -1
  198. data/openapi.json +1 -7784
  199. data/spec/models/api_key_resp_body_spec.rb +78 -0
  200. data/spec/models/autocomplete_data_spec.rb +108 -0
  201. data/spec/models/autocomplete_req_payload_spec.rb +132 -0
  202. data/spec/models/boost_phrase_spec.rb +42 -0
  203. data/spec/models/chunk_metadata_string_tag_set_spec.rb +120 -0
  204. data/spec/models/chunk_metadata_types_spec.rb +32 -0
  205. data/spec/models/chunk_req_payload_spec.rb +132 -0
  206. data/spec/models/condition_type_spec.rb +32 -0
  207. data/spec/models/content_chunk_metadata_spec.rb +72 -0
  208. data/spec/models/create_chunk_group_req_payload_spec.rb +66 -0
  209. data/spec/models/create_chunk_req_payload_enum_spec.rb +32 -0
  210. data/spec/models/create_message_req_payload_spec.rb +108 -0
  211. data/spec/models/create_topic_req_payload_spec.rb +48 -0
  212. data/spec/models/date_range_spec.rb +54 -0
  213. data/spec/models/edit_message_req_payload_spec.rb +114 -0
  214. data/spec/models/get_datasets_pagination_spec.rb +42 -0
  215. data/spec/models/has_id_condition_spec.rb +42 -0
  216. data/spec/models/regenerate_message_req_payload_spec.rb +102 -0
  217. data/spec/models/remove_chunk_from_group_req_payload_spec.rb +36 -0
  218. data/spec/models/search_chunks_req_payload_spec.rb +144 -0
  219. data/spec/models/suggested_queries_req_payload_spec.rb +36 -0
  220. data/spec/models/update_chunk_req_payload_spec.rb +120 -0
  221. data/spec/models/update_group_by_tracking_id_req_payload_spec.rb +60 -0
  222. data/spec/models/update_topic_req_payload_spec.rb +42 -0
  223. data/spec/models/upload_file_req_payload_spec.rb +102 -0
  224. data/spec/spec_helper.rb +1 -1
  225. data/trieve_ruby_client-0.8.0.gem +0 -0
  226. data/trieve_ruby_client-0.8.2.gem +0 -0
  227. data/trieve_ruby_client-0.8.5.gem +0 -0
  228. data/trieve_ruby_client.gemspec +1 -1
  229. metadata +105 -2
@@ -3,7 +3,7 @@
3
3
 
4
4
  #Trieve OpenAPI Specification. This document describes all of the operations available through the Trieve API.
5
5
 
6
- The version of the OpenAPI document: 0.8.0
6
+ The version of the OpenAPI document: 0.10.7
7
7
  Contact: developers@trieve.ai
8
8
  Generated by: https://openapi-generator.tech
9
9
  Generator version: 7.4.0
@@ -20,18 +20,30 @@ module TrieveRubyClient
20
20
  # Set get_collisions to true to get the collisions for each chunk. This will only apply if environment variable COLLISIONS_ENABLED is set to true.
21
21
  attr_accessor :get_collisions
22
22
 
23
- # Get total page count for the query accounting for the applied filters. Defaults to true, but can be set to false to reduce latency in edge cases performance.
23
+ # Get total page count for the query accounting for the applied filters. Defaults to false, but can be set to true when the latency penalty is acceptable (typically 50-200ms).
24
24
  attr_accessor :get_total_pages
25
25
 
26
26
  # Group_size is the number of chunks to fetch for each group. The default is 3. If a group has less than group_size chunks, all chunks will be returned. If this is set to a large number, we recommend setting slim_chunks to true to avoid returning the content and chunk_html of the chunks so as to lower the amount of time required for content download and serialization.
27
27
  attr_accessor :group_size
28
28
 
29
- # Set highlight_delimiters to a list of strings to use as delimiters for highlighting. If not specified, this defaults to [\"?\", \",\", \".\", \"!\"].
29
+ # Set highlight_delimiters to a list of strings to use as delimiters for highlighting. If not specified, this defaults to [\"?\", \",\", \".\", \"!\"]. These are the characters that will be used to split the chunk_html into splits for highlighting.
30
30
  attr_accessor :highlight_delimiters
31
31
 
32
- # Set highlight_results to true to highlight the results. If not specified, this defaults to true.
32
+ # Set highlight_max_length to control the maximum number of tokens (typically whitespace separated strings, but sometimes also word stems) which can be present within a single highlight. If not specified, this defaults to 8. This is useful to shorten large splits which may have low scores due to length compared to the query. Set to something very large like 100 to highlight entire splits.
33
+ attr_accessor :highlight_max_length
34
+
35
+ # Set highlight_max_num to control the maximum number of highlights per chunk. If not specified, this defaults to 3. It may be less than 3 if no snippets score above the highlight_threshold.
36
+ attr_accessor :highlight_max_num
37
+
38
+ # Set highlight_results to false for a slight latency improvement (1-10ms). If not specified, this defaults to true. This will add `<b><mark>` tags to the chunk_html of the chunks to highlight matching splits and return the highlights on each scored chunk in the response.
33
39
  attr_accessor :highlight_results
34
40
 
41
+ # Set highlight_threshold to a lower or higher value to adjust the sensitivity of the highlights applied to the chunk html. If not specified, this defaults to 0.8. The range is 0.0 to 1.0.
42
+ attr_accessor :highlight_threshold
43
+
44
+ # Set highlight_window to a number to control the amount of words that are returned around the matched phrases. If not specified, this defaults to 0. This is useful for when you want to show more context around the matched words. When specified, window/2 whitespace separated words are added before and after each highlight in the response's highlights array. If an extended highlight overlaps with another highlight, the overlapping words are only included once.
45
+ attr_accessor :highlight_window
46
+
35
47
  # Page of group results to fetch. Page is 1-indexed.
36
48
  attr_accessor :page
37
49
 
@@ -47,7 +59,7 @@ module TrieveRubyClient
47
59
  # Can be either \"semantic\", \"fulltext\", or \"hybrid\". \"hybrid\" will pull in one page (10 chunks) of both semantic and full-text results then re-rank them using BAAI/bge-reranker-large. \"semantic\" will pull in one page (10 chunks) of the nearest cosine distant vectors. \"fulltext\" will pull in one page (10 chunks) of full-text results based on SPLADE.
48
60
  attr_accessor :search_type
49
61
 
50
- # Set slim_chunks to true to avoid returning the content and chunk_html of the chunks. This is useful for when you want to reduce amount of data over the wire for latency improvement. Default is false.
62
+ # Set slim_chunks to true to avoid returning the content and chunk_html of the chunks. This is useful for when you want to reduce amount of data over the wire for latency improvement (typicall 10-50ms). Default is false.
51
63
  attr_accessor :slim_chunks
52
64
 
53
65
  # Attribute mapping from ruby-style variable name to JSON key.
@@ -58,7 +70,11 @@ module TrieveRubyClient
58
70
  :'get_total_pages' => :'get_total_pages',
59
71
  :'group_size' => :'group_size',
60
72
  :'highlight_delimiters' => :'highlight_delimiters',
73
+ :'highlight_max_length' => :'highlight_max_length',
74
+ :'highlight_max_num' => :'highlight_max_num',
61
75
  :'highlight_results' => :'highlight_results',
76
+ :'highlight_threshold' => :'highlight_threshold',
77
+ :'highlight_window' => :'highlight_window',
62
78
  :'page' => :'page',
63
79
  :'page_size' => :'page_size',
64
80
  :'query' => :'query',
@@ -81,7 +97,11 @@ module TrieveRubyClient
81
97
  :'get_total_pages' => :'Boolean',
82
98
  :'group_size' => :'Integer',
83
99
  :'highlight_delimiters' => :'Array<String>',
100
+ :'highlight_max_length' => :'Integer',
101
+ :'highlight_max_num' => :'Integer',
84
102
  :'highlight_results' => :'Boolean',
103
+ :'highlight_threshold' => :'Float',
104
+ :'highlight_window' => :'Integer',
85
105
  :'page' => :'Integer',
86
106
  :'page_size' => :'Integer',
87
107
  :'query' => :'String',
@@ -99,7 +119,11 @@ module TrieveRubyClient
99
119
  :'get_total_pages',
100
120
  :'group_size',
101
121
  :'highlight_delimiters',
122
+ :'highlight_max_length',
123
+ :'highlight_max_num',
102
124
  :'highlight_results',
125
+ :'highlight_threshold',
126
+ :'highlight_window',
103
127
  :'page',
104
128
  :'page_size',
105
129
  :'score_threshold',
@@ -144,10 +168,26 @@ module TrieveRubyClient
144
168
  end
145
169
  end
146
170
 
171
+ if attributes.key?(:'highlight_max_length')
172
+ self.highlight_max_length = attributes[:'highlight_max_length']
173
+ end
174
+
175
+ if attributes.key?(:'highlight_max_num')
176
+ self.highlight_max_num = attributes[:'highlight_max_num']
177
+ end
178
+
147
179
  if attributes.key?(:'highlight_results')
148
180
  self.highlight_results = attributes[:'highlight_results']
149
181
  end
150
182
 
183
+ if attributes.key?(:'highlight_threshold')
184
+ self.highlight_threshold = attributes[:'highlight_threshold']
185
+ end
186
+
187
+ if attributes.key?(:'highlight_window')
188
+ self.highlight_window = attributes[:'highlight_window']
189
+ end
190
+
151
191
  if attributes.key?(:'page')
152
192
  self.page = attributes[:'page']
153
193
  end
@@ -186,6 +226,18 @@ module TrieveRubyClient
186
226
  invalid_properties.push('invalid value for "group_size", must be greater than or equal to 0.')
187
227
  end
188
228
 
229
+ if !@highlight_max_length.nil? && @highlight_max_length < 0
230
+ invalid_properties.push('invalid value for "highlight_max_length", must be greater than or equal to 0.')
231
+ end
232
+
233
+ if !@highlight_max_num.nil? && @highlight_max_num < 0
234
+ invalid_properties.push('invalid value for "highlight_max_num", must be greater than or equal to 0.')
235
+ end
236
+
237
+ if !@highlight_window.nil? && @highlight_window < 0
238
+ invalid_properties.push('invalid value for "highlight_window", must be greater than or equal to 0.')
239
+ end
240
+
189
241
  if !@page.nil? && @page < 0
190
242
  invalid_properties.push('invalid value for "page", must be greater than or equal to 0.')
191
243
  end
@@ -210,6 +262,9 @@ module TrieveRubyClient
210
262
  def valid?
211
263
  warn '[DEPRECATED] the `valid?` method is obsolete'
212
264
  return false if !@group_size.nil? && @group_size < 0
265
+ return false if !@highlight_max_length.nil? && @highlight_max_length < 0
266
+ return false if !@highlight_max_num.nil? && @highlight_max_num < 0
267
+ return false if !@highlight_window.nil? && @highlight_window < 0
213
268
  return false if !@page.nil? && @page < 0
214
269
  return false if !@page_size.nil? && @page_size < 0
215
270
  return false if @query.nil?
@@ -227,6 +282,36 @@ module TrieveRubyClient
227
282
  @group_size = group_size
228
283
  end
229
284
 
285
+ # Custom attribute writer method with validation
286
+ # @param [Object] highlight_max_length Value to be assigned
287
+ def highlight_max_length=(highlight_max_length)
288
+ if !highlight_max_length.nil? && highlight_max_length < 0
289
+ fail ArgumentError, 'invalid value for "highlight_max_length", must be greater than or equal to 0.'
290
+ end
291
+
292
+ @highlight_max_length = highlight_max_length
293
+ end
294
+
295
+ # Custom attribute writer method with validation
296
+ # @param [Object] highlight_max_num Value to be assigned
297
+ def highlight_max_num=(highlight_max_num)
298
+ if !highlight_max_num.nil? && highlight_max_num < 0
299
+ fail ArgumentError, 'invalid value for "highlight_max_num", must be greater than or equal to 0.'
300
+ end
301
+
302
+ @highlight_max_num = highlight_max_num
303
+ end
304
+
305
+ # Custom attribute writer method with validation
306
+ # @param [Object] highlight_window Value to be assigned
307
+ def highlight_window=(highlight_window)
308
+ if !highlight_window.nil? && highlight_window < 0
309
+ fail ArgumentError, 'invalid value for "highlight_window", must be greater than or equal to 0.'
310
+ end
311
+
312
+ @highlight_window = highlight_window
313
+ end
314
+
230
315
  # Custom attribute writer method with validation
231
316
  # @param [Object] page Value to be assigned
232
317
  def page=(page)
@@ -257,7 +342,11 @@ module TrieveRubyClient
257
342
  get_total_pages == o.get_total_pages &&
258
343
  group_size == o.group_size &&
259
344
  highlight_delimiters == o.highlight_delimiters &&
345
+ highlight_max_length == o.highlight_max_length &&
346
+ highlight_max_num == o.highlight_max_num &&
260
347
  highlight_results == o.highlight_results &&
348
+ highlight_threshold == o.highlight_threshold &&
349
+ highlight_window == o.highlight_window &&
261
350
  page == o.page &&
262
351
  page_size == o.page_size &&
263
352
  query == o.query &&
@@ -275,7 +364,7 @@ module TrieveRubyClient
275
364
  # Calculates hash code according to all attributes.
276
365
  # @return [Integer] Hash code
277
366
  def hash
278
- [filters, get_collisions, get_total_pages, group_size, highlight_delimiters, highlight_results, page, page_size, query, score_threshold, search_type, slim_chunks].hash
367
+ [filters, get_collisions, get_total_pages, group_size, highlight_delimiters, highlight_max_length, highlight_max_num, highlight_results, highlight_threshold, highlight_window, page, page_size, query, score_threshold, search_type, slim_chunks].hash
279
368
  end
280
369
 
281
370
  # Builds the object from hash
@@ -3,7 +3,7 @@
3
3
 
4
4
  #Trieve OpenAPI Specification. This document describes all of the operations available through the Trieve API.
5
5
 
6
- The version of the OpenAPI document: 0.8.0
6
+ The version of the OpenAPI document: 0.10.7
7
7
  Contact: developers@trieve.ai
8
8
  Generated by: https://openapi-generator.tech
9
9
  Generator version: 7.4.0
@@ -3,7 +3,7 @@
3
3
 
4
4
  #Trieve OpenAPI Specification. This document describes all of the operations available through the Trieve API.
5
5
 
6
- The version of the OpenAPI document: 0.8.0
6
+ The version of the OpenAPI document: 0.8.2
7
7
  Contact: developers@trieve.ai
8
8
  Generated by: https://openapi-generator.tech
9
9
  Generator version: 7.4.0
@@ -3,7 +3,7 @@
3
3
 
4
4
  #Trieve OpenAPI Specification. This document describes all of the operations available through the Trieve API.
5
5
 
6
- The version of the OpenAPI document: 0.8.0
6
+ The version of the OpenAPI document: 0.8.2
7
7
  Contact: developers@trieve.ai
8
8
  Generated by: https://openapi-generator.tech
9
9
  Generator version: 7.4.0
@@ -3,7 +3,7 @@
3
3
 
4
4
  #Trieve OpenAPI Specification. This document describes all of the operations available through the Trieve API.
5
5
 
6
- The version of the OpenAPI document: 0.8.0
6
+ The version of the OpenAPI document: 0.10.7
7
7
  Contact: developers@trieve.ai
8
8
  Generated by: https://openapi-generator.tech
9
9
  Generator version: 7.4.0
@@ -15,12 +15,9 @@ require 'time'
15
15
 
16
16
  module TrieveRubyClient
17
17
  class SearchWithinGroupData
18
- # Set date_bias to true to bias search results towards more recent chunks. This will work best in hybrid search mode.
19
- attr_accessor :date_bias
20
-
21
18
  attr_accessor :filters
22
19
 
23
- # Get total page count for the query accounting for the applied filters. Defaults to true, but can be set to false to reduce latency in edge cases performance.
20
+ # Get total page count for the query accounting for the applied filters. Defaults to false, but can be set to true when the latency penalty is acceptable (typically 50-200ms).
24
21
  attr_accessor :get_total_pages
25
22
 
26
23
  # Group specifies the group to search within. Results will only consist of chunks which are bookmarks within the specified group.
@@ -29,12 +26,24 @@ module TrieveRubyClient
29
26
  # Group_tracking_id specifies the group to search within by tracking id. Results will only consist of chunks which are bookmarks within the specified group. If both group_id and group_tracking_id are provided, group_id will be used.
30
27
  attr_accessor :group_tracking_id
31
28
 
32
- # Set highlight_delimiters to a list of strings to use as delimiters for highlighting. If not specified, this defaults to [\"?\", \",\", \".\", \"!\"].
29
+ # Set highlight_delimiters to a list of strings to use as delimiters for highlighting. If not specified, this defaults to [\"?\", \",\", \".\", \"!\"]. These are the characters that will be used to split the chunk_html into splits for highlighting.
33
30
  attr_accessor :highlight_delimiters
34
31
 
35
- # Set highlight_results to true to highlight the results. If not specified, this defaults to true.
32
+ # Set highlight_max_length to control the maximum number of tokens (typically whitespace separated strings, but sometimes also word stems) which can be present within a single highlight. If not specified, this defaults to 8. This is useful to shorten large splits which may have low scores due to length compared to the query. Set to something very large like 100 to highlight entire splits.
33
+ attr_accessor :highlight_max_length
34
+
35
+ # Set highlight_max_num to control the maximum number of highlights per chunk. If not specified, this defaults to 3. It may be less than 3 if no snippets score above the highlight_threshold.
36
+ attr_accessor :highlight_max_num
37
+
38
+ # Set highlight_results to false for a slight latency improvement (1-10ms). If not specified, this defaults to true. This will add `<b><mark>` tags to the chunk_html of the chunks to highlight matching splits and return the highlights on each scored chunk in the response.
36
39
  attr_accessor :highlight_results
37
40
 
41
+ # Set highlight_threshold to a lower or higher value to adjust the sensitivity of the highlights applied to the chunk html. If not specified, this defaults to 0.8. The range is 0.0 to 1.0.
42
+ attr_accessor :highlight_threshold
43
+
44
+ # Set highlight_window to a number to control the amount of words that are returned around the matched phrases. If not specified, this defaults to 0. This is useful for when you want to show more context around the matched words. When specified, window/2 whitespace separated words are added before and after each highlight in the response's highlights array. If an extended highlight overlaps with another highlight, the overlapping words are only included once.
45
+ attr_accessor :highlight_window
46
+
38
47
  # The page of chunks to fetch. Page is 1-indexed.
39
48
  attr_accessor :page
40
49
 
@@ -44,34 +53,45 @@ module TrieveRubyClient
44
53
  # The query is the search query. This can be any string. The query will be used to create an embedding vector and/or SPLADE vector which will be used to find the result set.
45
54
  attr_accessor :query
46
55
 
56
+ # Recency Bias lets you determine how much of an effect the recency of chunks will have on the search results. If not specified, this defaults to 0.0.
57
+ attr_accessor :recency_bias
58
+
47
59
  # Set score_threshold to a float to filter out chunks with a score below the threshold.
48
60
  attr_accessor :score_threshold
49
61
 
50
62
  # Search_type can be either \"semantic\", \"fulltext\", or \"hybrid\". \"hybrid\" will pull in one page (10 chunks) of both semantic and full-text results then re-rank them using BAAI/bge-reranker-large. \"semantic\" will pull in one page (10 chunks) of the nearest cosine distant vectors. \"fulltext\" will pull in one page (10 chunks) of full-text results based on SPLADE.
51
63
  attr_accessor :search_type
52
64
 
53
- # Set slim_chunks to true to avoid returning the content and chunk_html of the chunks. This is useful for when you want to reduce amount of data over the wire for latency improvement. Default is false.
65
+ # Set slim_chunks to true to avoid returning the content and chunk_html of the chunks. This is useful for when you want to reduce amount of data over the wire for latency improvement (typicall 10-50ms). Default is false.
54
66
  attr_accessor :slim_chunks
55
67
 
68
+ # Tag weights is a JSON object which can be used to boost the ranking of chunks with certain tags. This is useful for when you want to be able to bias towards chunks with a certain tag on the fly. The keys are the tag names and the values are the weights.
69
+ attr_accessor :tag_weights
70
+
56
71
  # Set use_weights to true to use the weights of the chunks in the result set in order to sort them. If not specified, this defaults to true.
57
72
  attr_accessor :use_weights
58
73
 
59
74
  # Attribute mapping from ruby-style variable name to JSON key.
60
75
  def self.attribute_map
61
76
  {
62
- :'date_bias' => :'date_bias',
63
77
  :'filters' => :'filters',
64
78
  :'get_total_pages' => :'get_total_pages',
65
79
  :'group_id' => :'group_id',
66
80
  :'group_tracking_id' => :'group_tracking_id',
67
81
  :'highlight_delimiters' => :'highlight_delimiters',
82
+ :'highlight_max_length' => :'highlight_max_length',
83
+ :'highlight_max_num' => :'highlight_max_num',
68
84
  :'highlight_results' => :'highlight_results',
85
+ :'highlight_threshold' => :'highlight_threshold',
86
+ :'highlight_window' => :'highlight_window',
69
87
  :'page' => :'page',
70
88
  :'page_size' => :'page_size',
71
89
  :'query' => :'query',
90
+ :'recency_bias' => :'recency_bias',
72
91
  :'score_threshold' => :'score_threshold',
73
92
  :'search_type' => :'search_type',
74
93
  :'slim_chunks' => :'slim_chunks',
94
+ :'tag_weights' => :'tag_weights',
75
95
  :'use_weights' => :'use_weights'
76
96
  }
77
97
  end
@@ -84,19 +104,24 @@ module TrieveRubyClient
84
104
  # Attribute type mapping.
85
105
  def self.openapi_types
86
106
  {
87
- :'date_bias' => :'Boolean',
88
107
  :'filters' => :'ChunkFilter',
89
108
  :'get_total_pages' => :'Boolean',
90
109
  :'group_id' => :'String',
91
110
  :'group_tracking_id' => :'String',
92
111
  :'highlight_delimiters' => :'Array<String>',
112
+ :'highlight_max_length' => :'Integer',
113
+ :'highlight_max_num' => :'Integer',
93
114
  :'highlight_results' => :'Boolean',
115
+ :'highlight_threshold' => :'Float',
116
+ :'highlight_window' => :'Integer',
94
117
  :'page' => :'Integer',
95
118
  :'page_size' => :'Integer',
96
119
  :'query' => :'String',
120
+ :'recency_bias' => :'Float',
97
121
  :'score_threshold' => :'Float',
98
122
  :'search_type' => :'String',
99
123
  :'slim_chunks' => :'Boolean',
124
+ :'tag_weights' => :'Hash<String, Float>',
100
125
  :'use_weights' => :'Boolean'
101
126
  }
102
127
  end
@@ -104,17 +129,22 @@ module TrieveRubyClient
104
129
  # List of attributes with nullable: true
105
130
  def self.openapi_nullable
106
131
  Set.new([
107
- :'date_bias',
108
132
  :'filters',
109
133
  :'get_total_pages',
110
134
  :'group_id',
111
135
  :'group_tracking_id',
112
136
  :'highlight_delimiters',
137
+ :'highlight_max_length',
138
+ :'highlight_max_num',
113
139
  :'highlight_results',
140
+ :'highlight_threshold',
141
+ :'highlight_window',
114
142
  :'page',
115
143
  :'page_size',
144
+ :'recency_bias',
116
145
  :'score_threshold',
117
146
  :'slim_chunks',
147
+ :'tag_weights',
118
148
  :'use_weights'
119
149
  ])
120
150
  end
@@ -134,10 +164,6 @@ module TrieveRubyClient
134
164
  h[k.to_sym] = v
135
165
  }
136
166
 
137
- if attributes.key?(:'date_bias')
138
- self.date_bias = attributes[:'date_bias']
139
- end
140
-
141
167
  if attributes.key?(:'filters')
142
168
  self.filters = attributes[:'filters']
143
169
  end
@@ -160,10 +186,26 @@ module TrieveRubyClient
160
186
  end
161
187
  end
162
188
 
189
+ if attributes.key?(:'highlight_max_length')
190
+ self.highlight_max_length = attributes[:'highlight_max_length']
191
+ end
192
+
193
+ if attributes.key?(:'highlight_max_num')
194
+ self.highlight_max_num = attributes[:'highlight_max_num']
195
+ end
196
+
163
197
  if attributes.key?(:'highlight_results')
164
198
  self.highlight_results = attributes[:'highlight_results']
165
199
  end
166
200
 
201
+ if attributes.key?(:'highlight_threshold')
202
+ self.highlight_threshold = attributes[:'highlight_threshold']
203
+ end
204
+
205
+ if attributes.key?(:'highlight_window')
206
+ self.highlight_window = attributes[:'highlight_window']
207
+ end
208
+
167
209
  if attributes.key?(:'page')
168
210
  self.page = attributes[:'page']
169
211
  end
@@ -178,6 +220,10 @@ module TrieveRubyClient
178
220
  self.query = nil
179
221
  end
180
222
 
223
+ if attributes.key?(:'recency_bias')
224
+ self.recency_bias = attributes[:'recency_bias']
225
+ end
226
+
181
227
  if attributes.key?(:'score_threshold')
182
228
  self.score_threshold = attributes[:'score_threshold']
183
229
  end
@@ -192,6 +238,12 @@ module TrieveRubyClient
192
238
  self.slim_chunks = attributes[:'slim_chunks']
193
239
  end
194
240
 
241
+ if attributes.key?(:'tag_weights')
242
+ if (value = attributes[:'tag_weights']).is_a?(Hash)
243
+ self.tag_weights = value
244
+ end
245
+ end
246
+
195
247
  if attributes.key?(:'use_weights')
196
248
  self.use_weights = attributes[:'use_weights']
197
249
  end
@@ -202,6 +254,18 @@ module TrieveRubyClient
202
254
  def list_invalid_properties
203
255
  warn '[DEPRECATED] the `list_invalid_properties` method is obsolete'
204
256
  invalid_properties = Array.new
257
+ if !@highlight_max_length.nil? && @highlight_max_length < 0
258
+ invalid_properties.push('invalid value for "highlight_max_length", must be greater than or equal to 0.')
259
+ end
260
+
261
+ if !@highlight_max_num.nil? && @highlight_max_num < 0
262
+ invalid_properties.push('invalid value for "highlight_max_num", must be greater than or equal to 0.')
263
+ end
264
+
265
+ if !@highlight_window.nil? && @highlight_window < 0
266
+ invalid_properties.push('invalid value for "highlight_window", must be greater than or equal to 0.')
267
+ end
268
+
205
269
  if !@page.nil? && @page < 0
206
270
  invalid_properties.push('invalid value for "page", must be greater than or equal to 0.')
207
271
  end
@@ -225,6 +289,9 @@ module TrieveRubyClient
225
289
  # @return true if the model is valid
226
290
  def valid?
227
291
  warn '[DEPRECATED] the `valid?` method is obsolete'
292
+ return false if !@highlight_max_length.nil? && @highlight_max_length < 0
293
+ return false if !@highlight_max_num.nil? && @highlight_max_num < 0
294
+ return false if !@highlight_window.nil? && @highlight_window < 0
228
295
  return false if !@page.nil? && @page < 0
229
296
  return false if !@page_size.nil? && @page_size < 0
230
297
  return false if @query.nil?
@@ -232,6 +299,36 @@ module TrieveRubyClient
232
299
  true
233
300
  end
234
301
 
302
+ # Custom attribute writer method with validation
303
+ # @param [Object] highlight_max_length Value to be assigned
304
+ def highlight_max_length=(highlight_max_length)
305
+ if !highlight_max_length.nil? && highlight_max_length < 0
306
+ fail ArgumentError, 'invalid value for "highlight_max_length", must be greater than or equal to 0.'
307
+ end
308
+
309
+ @highlight_max_length = highlight_max_length
310
+ end
311
+
312
+ # Custom attribute writer method with validation
313
+ # @param [Object] highlight_max_num Value to be assigned
314
+ def highlight_max_num=(highlight_max_num)
315
+ if !highlight_max_num.nil? && highlight_max_num < 0
316
+ fail ArgumentError, 'invalid value for "highlight_max_num", must be greater than or equal to 0.'
317
+ end
318
+
319
+ @highlight_max_num = highlight_max_num
320
+ end
321
+
322
+ # Custom attribute writer method with validation
323
+ # @param [Object] highlight_window Value to be assigned
324
+ def highlight_window=(highlight_window)
325
+ if !highlight_window.nil? && highlight_window < 0
326
+ fail ArgumentError, 'invalid value for "highlight_window", must be greater than or equal to 0.'
327
+ end
328
+
329
+ @highlight_window = highlight_window
330
+ end
331
+
235
332
  # Custom attribute writer method with validation
236
333
  # @param [Object] page Value to be assigned
237
334
  def page=(page)
@@ -257,19 +354,24 @@ module TrieveRubyClient
257
354
  def ==(o)
258
355
  return true if self.equal?(o)
259
356
  self.class == o.class &&
260
- date_bias == o.date_bias &&
261
357
  filters == o.filters &&
262
358
  get_total_pages == o.get_total_pages &&
263
359
  group_id == o.group_id &&
264
360
  group_tracking_id == o.group_tracking_id &&
265
361
  highlight_delimiters == o.highlight_delimiters &&
362
+ highlight_max_length == o.highlight_max_length &&
363
+ highlight_max_num == o.highlight_max_num &&
266
364
  highlight_results == o.highlight_results &&
365
+ highlight_threshold == o.highlight_threshold &&
366
+ highlight_window == o.highlight_window &&
267
367
  page == o.page &&
268
368
  page_size == o.page_size &&
269
369
  query == o.query &&
370
+ recency_bias == o.recency_bias &&
270
371
  score_threshold == o.score_threshold &&
271
372
  search_type == o.search_type &&
272
373
  slim_chunks == o.slim_chunks &&
374
+ tag_weights == o.tag_weights &&
273
375
  use_weights == o.use_weights
274
376
  end
275
377
 
@@ -282,7 +384,7 @@ module TrieveRubyClient
282
384
  # Calculates hash code according to all attributes.
283
385
  # @return [Integer] Hash code
284
386
  def hash
285
- [date_bias, filters, get_total_pages, group_id, group_tracking_id, highlight_delimiters, highlight_results, page, page_size, query, score_threshold, search_type, slim_chunks, use_weights].hash
387
+ [filters, get_total_pages, group_id, group_tracking_id, highlight_delimiters, highlight_max_length, highlight_max_num, highlight_results, highlight_threshold, highlight_window, page, page_size, query, recency_bias, score_threshold, search_type, slim_chunks, tag_weights, use_weights].hash
286
388
  end
287
389
 
288
390
  # Builds the object from hash
@@ -3,7 +3,7 @@
3
3
 
4
4
  #Trieve OpenAPI Specification. This document describes all of the operations available through the Trieve API.
5
5
 
6
- The version of the OpenAPI document: 0.8.0
6
+ The version of the OpenAPI document: 0.10.7
7
7
  Contact: developers@trieve.ai
8
8
  Generated by: https://openapi-generator.tech
9
9
  Generator version: 7.4.0
@@ -3,7 +3,7 @@
3
3
 
4
4
  #Trieve OpenAPI Specification. This document describes all of the operations available through the Trieve API.
5
5
 
6
- The version of the OpenAPI document: 0.8.0
6
+ The version of the OpenAPI document: 0.8.2
7
7
  Contact: developers@trieve.ai
8
8
  Generated by: https://openapi-generator.tech
9
9
  Generator version: 7.4.0
@@ -3,7 +3,7 @@
3
3
 
4
4
  #Trieve OpenAPI Specification. This document describes all of the operations available through the Trieve API.
5
5
 
6
- The version of the OpenAPI document: 0.8.0
6
+ The version of the OpenAPI document: 0.10.7
7
7
  Contact: developers@trieve.ai
8
8
  Generated by: https://openapi-generator.tech
9
9
  Generator version: 7.4.0
@@ -15,16 +15,24 @@ require 'time'
15
15
 
16
16
  module TrieveRubyClient
17
17
  class SetUserApiKeyRequest
18
+ # The dataset ids which the api key will have access to. If not provided or empty, the api key will have access to all datasets the auth'ed user has access to. If both dataset_ids and organization_ids are provided, the api key will have access to the intersection of the datasets and organizations.
19
+ attr_accessor :dataset_ids
20
+
18
21
  # The name which will be assigned to the new api key.
19
22
  attr_accessor :name
20
23
 
24
+ # The organization ids which the api key will have access to. If not provided or empty, the api key will have access to all organizations the auth'ed user has access to.
25
+ attr_accessor :organization_ids
26
+
21
27
  # The role which will be assigned to the new api key. Either 0 (read), 1 (read and write at the level of the currently auth'ed user). The auth'ed user must have a role greater than or equal to the role being assigned which means they must be an admin (1) or owner (2) of the organization to assign write permissions with a role of 1.
22
28
  attr_accessor :role
23
29
 
24
30
  # Attribute mapping from ruby-style variable name to JSON key.
25
31
  def self.attribute_map
26
32
  {
33
+ :'dataset_ids' => :'dataset_ids',
27
34
  :'name' => :'name',
35
+ :'organization_ids' => :'organization_ids',
28
36
  :'role' => :'role'
29
37
  }
30
38
  end
@@ -37,7 +45,9 @@ module TrieveRubyClient
37
45
  # Attribute type mapping.
38
46
  def self.openapi_types
39
47
  {
48
+ :'dataset_ids' => :'Array<String>',
40
49
  :'name' => :'String',
50
+ :'organization_ids' => :'Array<String>',
41
51
  :'role' => :'Integer'
42
52
  }
43
53
  end
@@ -45,6 +55,8 @@ module TrieveRubyClient
45
55
  # List of attributes with nullable: true
46
56
  def self.openapi_nullable
47
57
  Set.new([
58
+ :'dataset_ids',
59
+ :'organization_ids',
48
60
  ])
49
61
  end
50
62
 
@@ -63,12 +75,24 @@ module TrieveRubyClient
63
75
  h[k.to_sym] = v
64
76
  }
65
77
 
78
+ if attributes.key?(:'dataset_ids')
79
+ if (value = attributes[:'dataset_ids']).is_a?(Array)
80
+ self.dataset_ids = value
81
+ end
82
+ end
83
+
66
84
  if attributes.key?(:'name')
67
85
  self.name = attributes[:'name']
68
86
  else
69
87
  self.name = nil
70
88
  end
71
89
 
90
+ if attributes.key?(:'organization_ids')
91
+ if (value = attributes[:'organization_ids']).is_a?(Array)
92
+ self.organization_ids = value
93
+ end
94
+ end
95
+
72
96
  if attributes.key?(:'role')
73
97
  self.role = attributes[:'role']
74
98
  else
@@ -106,7 +130,9 @@ module TrieveRubyClient
106
130
  def ==(o)
107
131
  return true if self.equal?(o)
108
132
  self.class == o.class &&
133
+ dataset_ids == o.dataset_ids &&
109
134
  name == o.name &&
135
+ organization_ids == o.organization_ids &&
110
136
  role == o.role
111
137
  end
112
138
 
@@ -119,7 +145,7 @@ module TrieveRubyClient
119
145
  # Calculates hash code according to all attributes.
120
146
  # @return [Integer] Hash code
121
147
  def hash
122
- [name, role].hash
148
+ [dataset_ids, name, organization_ids, role].hash
123
149
  end
124
150
 
125
151
  # Builds the object from hash
@@ -3,7 +3,7 @@
3
3
 
4
4
  #Trieve OpenAPI Specification. This document describes all of the operations available through the Trieve API.
5
5
 
6
- The version of the OpenAPI document: 0.8.0
6
+ The version of the OpenAPI document: 0.10.7
7
7
  Contact: developers@trieve.ai
8
8
  Generated by: https://openapi-generator.tech
9
9
  Generator version: 7.4.0