Class: TrieveRubyClient::FileApi

Inherits:
Object
  • Object
show all
Defined in:
lib/trieve_ruby_client/api/file_api.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(api_client = ApiClient.default) ⇒ FileApi

Returns a new instance of FileApi.



19
20
21
 
# File 'lib/trieve_ruby_client/api/file_api.rb', line 19

def initialize(api_client = ApiClient.default)
  @api_client = api_client
end

Instance Attribute Details

#api_client ⇒ Object

Returns the value of attribute api_client.



17
18
19
 
# File 'lib/trieve_ruby_client/api/file_api.rb', line 17

def api_client
  @api_client
end

Instance Method Details

#delete_file_handler(tr_dataset, file_id, opts = {}) ⇒ nil

Delete File Delete a file from S3 attached to the server based on its id. This will disassociate chunks from the file, but only delete them all together if you specify delete_chunks to be true. Auth’ed user or api key must have an admin or owner role for the specified dataset’s organization.

Parameters:

  • tr_dataset (String) —

    The dataset id to use for the request

  • file_id (String) —

    The id of the file to delete

  • opts (Hash) (defaults to: {}) —

    the optional parameters

Returns:

  • (nil) 


28
29
30
31
 
# File 'lib/trieve_ruby_client/api/file_api.rb', line 28

def delete_file_handler(tr_dataset, file_id, opts = {})
  delete_file_handler_with_http_info(tr_dataset, file_id, opts)
  nil
end

#delete_file_handler_with_http_info(tr_dataset, file_id, opts = {}) ⇒ Array<(nil, Integer, Hash)>

Delete File Delete a file from S3 attached to the server based on its id. This will disassociate chunks from the file, but only delete them all together if you specify delete_chunks to be true. Auth&#39;ed user or api key must have an admin or owner role for the specified dataset&#39;s organization.

Parameters:

  • tr_dataset (String) —

    The dataset id to use for the request

  • file_id (String) —

    The id of the file to delete

  • opts (Hash) (defaults to: {}) —

    the optional parameters

Returns:

  • (Array<(nil, Integer, Hash)>) —

    nil, response status code and response headers



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
 
# File 'lib/trieve_ruby_client/api/file_api.rb', line 39

def delete_file_handler_with_http_info(tr_dataset, file_id, opts = {})
  if @api_client.config.debugging
    @api_client.config.logger.debug 'Calling API: FileApi.delete_file_handler ...'
  end
  # verify the required parameter 'tr_dataset' is set
  if @api_client.config.client_side_validation && tr_dataset.nil?
    fail ArgumentError, "Missing the required parameter 'tr_dataset' when calling FileApi.delete_file_handler"
  end
  # verify the required parameter 'file_id' is set
  if @api_client.config.client_side_validation && file_id.nil?
    fail ArgumentError, "Missing the required parameter 'file_id' when calling FileApi.delete_file_handler"
  end
  # resource path
  local_var_path = '/api/file/{file_id}'.sub('{' + 'file_id' + '}', CGI.escape(file_id.to_s))

  # query parameters
  query_params = opts[:query_params] || {}

  # header parameters
  header_params = opts[:header_params] || {}
  # HTTP header 'Accept' (if needed)
  header_params['Accept'] = @api_client.select_header_accept(['application/json'])
  header_params[:'TR-Dataset'] = tr_dataset

  # form parameters
  form_params = opts[:form_params] || {}

  # http body (model)
  post_body = opts[:debug_body]

  # return_type
  return_type = opts[:debug_return_type]

  # auth_names
  auth_names = opts[:debug_auth_names] || ['ApiKey']

  new_options = opts.merge(
    :operation => :"FileApi.delete_file_handler",
    :header_params => header_params,
    :query_params => query_params,
    :form_params => form_params,
    :body => post_body,
    :auth_names => auth_names,
    :return_type => return_type
  )

  data, status_code, headers = @api_client.call_api(:DELETE, local_var_path, new_options)
  if @api_client.config.debugging
    @api_client.config.logger.debug "API called: FileApi#delete_file_handler\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
  end
  return data, status_code, headers
end

#get_dataset_files_handler(tr_dataset, dataset_id, page, opts = {}) ⇒ Array<File>

Get Files for Dataset Get all files which belong to a given dataset specified by the dataset_id parameter. 10 files are returned per page.

Parameters:

  • tr_dataset (String) —

    The dataset id to use for the request

  • dataset_id (String) —

    The id of the dataset to fetch files for.

  • page (Integer) —

    The page number of files you wish to fetch. Each page contains at most 10 files.

  • opts (Hash) (defaults to: {}) —

    the optional parameters

Returns:



99
100
101
102
 
# File 'lib/trieve_ruby_client/api/file_api.rb', line 99

def get_dataset_files_handler(tr_dataset, dataset_id, page, opts = {})
  data, _status_code, _headers = get_dataset_files_handler_with_http_info(tr_dataset, dataset_id, page, opts)
  data
end

#get_dataset_files_handler_with_http_info(tr_dataset, dataset_id, page, opts = {}) ⇒ Array<(Array<File>, Integer, Hash)>

Get Files for Dataset Get all files which belong to a given dataset specified by the dataset_id parameter. 10 files are returned per page.

Parameters:

  • tr_dataset (String) —

    The dataset id to use for the request

  • dataset_id (String) —

    The id of the dataset to fetch files for.

  • page (Integer) —

    The page number of files you wish to fetch. Each page contains at most 10 files.

  • opts (Hash) (defaults to: {}) —

    the optional parameters

Returns:

  • (Array<(Array<File>, Integer, Hash)>) —

    Array<File> data, response status code and response headers



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
160
161
162
163
164
165
166
167
168
169
170
 
# File 'lib/trieve_ruby_client/api/file_api.rb', line 111

def get_dataset_files_handler_with_http_info(tr_dataset, dataset_id, page, opts = {})
  if @api_client.config.debugging
    @api_client.config.logger.debug 'Calling API: FileApi.get_dataset_files_handler ...'
  end
  # verify the required parameter 'tr_dataset' is set
  if @api_client.config.client_side_validation && tr_dataset.nil?
    fail ArgumentError, "Missing the required parameter 'tr_dataset' when calling FileApi.get_dataset_files_handler"
  end
  # verify the required parameter 'dataset_id' is set
  if @api_client.config.client_side_validation && dataset_id.nil?
    fail ArgumentError, "Missing the required parameter 'dataset_id' when calling FileApi.get_dataset_files_handler"
  end
  # verify the required parameter 'page' is set
  if @api_client.config.client_side_validation && page.nil?
    fail ArgumentError, "Missing the required parameter 'page' when calling FileApi.get_dataset_files_handler"
  end
  if @api_client.config.client_side_validation && page < 0
    fail ArgumentError, 'invalid value for "page" when calling FileApi.get_dataset_files_handler, must be greater than or equal to 0.'
  end

  # resource path
  local_var_path = '/api/dataset/files/{dataset_id}/{page}'.sub('{' + 'dataset_id' + '}', CGI.escape(dataset_id.to_s)).sub('{' + 'page' + '}', CGI.escape(page.to_s))

  # query parameters
  query_params = opts[:query_params] || {}

  # header parameters
  header_params = opts[:header_params] || {}
  # HTTP header 'Accept' (if needed)
  header_params['Accept'] = @api_client.select_header_accept(['application/json'])
  header_params[:'TR-Dataset'] = tr_dataset

  # form parameters
  form_params = opts[:form_params] || {}

  # http body (model)
  post_body = opts[:debug_body]

  # return_type
  return_type = opts[:debug_return_type] || 'Array<File>'

  # auth_names
  auth_names = opts[:debug_auth_names] || ['ApiKey']

  new_options = opts.merge(
    :operation => :"FileApi.get_dataset_files_handler",
    :header_params => header_params,
    :query_params => query_params,
    :form_params => form_params,
    :body => post_body,
    :auth_names => auth_names,
    :return_type => return_type
  )

  data, status_code, headers = @api_client.call_api(:GET, local_var_path, new_options)
  if @api_client.config.debugging
    @api_client.config.logger.debug "API called: FileApi#get_dataset_files_handler\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
  end
  return data, status_code, headers
end

#get_file_handler(tr_dataset, file_id, opts = {}) ⇒ FileDTO

Get File Download a file based on its id.

Parameters:

  • tr_dataset (String) —

    The dataset id to use for the request

  • file_id (String) —

    The id of the file to fetch

  • opts (Hash) (defaults to: {}) —

    the optional parameters

Returns:



178
179
180
181
 
# File 'lib/trieve_ruby_client/api/file_api.rb', line 178

def get_file_handler(tr_dataset, file_id, opts = {})
  data, _status_code, _headers = get_file_handler_with_http_info(tr_dataset, file_id, opts)
  data
end

#get_file_handler_with_http_info(tr_dataset, file_id, opts = {}) ⇒ Array<(FileDTO, Integer, Hash)>

Get File Download a file based on its id.

Parameters:

  • tr_dataset (String) —

    The dataset id to use for the request

  • file_id (String) —

    The id of the file to fetch

  • opts (Hash) (defaults to: {}) —

    the optional parameters

Returns:

  • (Array<(FileDTO, Integer, Hash)>) —

    FileDTO data, response status code and response headers



189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
 
# File 'lib/trieve_ruby_client/api/file_api.rb', line 189

def get_file_handler_with_http_info(tr_dataset, file_id, opts = {})
  if @api_client.config.debugging
    @api_client.config.logger.debug 'Calling API: FileApi.get_file_handler ...'
  end
  # verify the required parameter 'tr_dataset' is set
  if @api_client.config.client_side_validation && tr_dataset.nil?
    fail ArgumentError, "Missing the required parameter 'tr_dataset' when calling FileApi.get_file_handler"
  end
  # verify the required parameter 'file_id' is set
  if @api_client.config.client_side_validation && file_id.nil?
    fail ArgumentError, "Missing the required parameter 'file_id' when calling FileApi.get_file_handler"
  end
  # resource path
  local_var_path = '/api/file/{file_id}'.sub('{' + 'file_id' + '}', CGI.escape(file_id.to_s))

  # query parameters
  query_params = opts[:query_params] || {}

  # header parameters
  header_params = opts[:header_params] || {}
  # HTTP header 'Accept' (if needed)
  header_params['Accept'] = @api_client.select_header_accept(['application/json'])
  header_params[:'TR-Dataset'] = tr_dataset

  # form parameters
  form_params = opts[:form_params] || {}

  # http body (model)
  post_body = opts[:debug_body]

  # return_type
  return_type = opts[:debug_return_type] || 'FileDTO'

  # auth_names
  auth_names = opts[:debug_auth_names] || ['ApiKey']

  new_options = opts.merge(
    :operation => :"FileApi.get_file_handler",
    :header_params => header_params,
    :query_params => query_params,
    :form_params => form_params,
    :body => post_body,
    :auth_names => auth_names,
    :return_type => return_type
  )

  data, status_code, headers = @api_client.call_api(:GET, local_var_path, new_options)
  if @api_client.config.debugging
    @api_client.config.logger.debug "API called: FileApi#get_file_handler\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
  end
  return data, status_code, headers
end

#upload_file_handler(tr_dataset, upload_file_req_payload, opts = {}) ⇒ UploadFileResult

Upload File Upload a file to S3 attached to the server. The file will be converted to HTML with tika and chunked algorithmically, images will be OCR’ed with tesseract. The resulting chunks will be indexed and searchable. Optionally, you can only upload the file and manually create chunks associated to the file after. See docs.trieve.ai and/or contact us for more details and tips. Auth’ed user must be an admin or owner of the dataset’s organization to upload a file.

Parameters:

  • tr_dataset (String) —

    The dataset id to use for the request

  • upload_file_req_payload (UploadFileReqPayload) —

    JSON request payload to upload a file

  • opts (Hash) (defaults to: {}) —

    the optional parameters

Returns:



248
249
250
251
 
# File 'lib/trieve_ruby_client/api/file_api.rb', line 248

def upload_file_handler(tr_dataset, upload_file_req_payload, opts = {})
  data, _status_code, _headers = upload_file_handler_with_http_info(tr_dataset, upload_file_req_payload, opts)
  data
end

#upload_file_handler_with_http_info(tr_dataset, upload_file_req_payload, opts = {}) ⇒ Array<(UploadFileResult, Integer, Hash)>

Upload File Upload a file to S3 attached to the server. The file will be converted to HTML with tika and chunked algorithmically, images will be OCR&#39;ed with tesseract. The resulting chunks will be indexed and searchable. Optionally, you can only upload the file and manually create chunks associated to the file after. See docs.trieve.ai and/or contact us for more details and tips. Auth&#39;ed user must be an admin or owner of the dataset&#39;s organization to upload a file.

Parameters:

  • tr_dataset (String) —

    The dataset id to use for the request

  • upload_file_req_payload (UploadFileReqPayload) —

    JSON request payload to upload a file

  • opts (Hash) (defaults to: {}) —

    the optional parameters

Returns:

  • (Array<(UploadFileResult, Integer, Hash)>) —

    UploadFileResult data, response status code and response headers



259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
 
# File 'lib/trieve_ruby_client/api/file_api.rb', line 259

def upload_file_handler_with_http_info(tr_dataset, upload_file_req_payload, opts = {})
  if @api_client.config.debugging
    @api_client.config.logger.debug 'Calling API: FileApi.upload_file_handler ...'
  end
  # verify the required parameter 'tr_dataset' is set
  if @api_client.config.client_side_validation && tr_dataset.nil?
    fail ArgumentError, "Missing the required parameter 'tr_dataset' when calling FileApi.upload_file_handler"
  end
  # verify the required parameter 'upload_file_req_payload' is set
  if @api_client.config.client_side_validation && upload_file_req_payload.nil?
    fail ArgumentError, "Missing the required parameter 'upload_file_req_payload' when calling FileApi.upload_file_handler"
  end
  # resource path
  local_var_path = '/api/file'

  # query parameters
  query_params = opts[:query_params] || {}

  # header parameters
  header_params = opts[:header_params] || {}
  # HTTP header 'Accept' (if needed)
  header_params['Accept'] = @api_client.select_header_accept(['application/json'])
  # HTTP header 'Content-Type'
  content_type = @api_client.select_header_content_type(['application/json'])
  if !content_type.nil?
      header_params['Content-Type'] = content_type
  end
  header_params[:'TR-Dataset'] = tr_dataset

  # form parameters
  form_params = opts[:form_params] || {}

  # http body (model)
  post_body = opts[:debug_body] || @api_client.object_to_http_body(upload_file_req_payload)

  # return_type
  return_type = opts[:debug_return_type] || 'UploadFileResult'

  # auth_names
  auth_names = opts[:debug_auth_names] || ['ApiKey']

  new_options = opts.merge(
    :operation => :"FileApi.upload_file_handler",
    :header_params => header_params,
    :query_params => query_params,
    :form_params => form_params,
    :body => post_body,
    :auth_names => auth_names,
    :return_type => return_type
  )

  data, status_code, headers = @api_client.call_api(:POST, local_var_path, new_options)
  if @api_client.config.debugging
    @api_client.config.logger.debug "API called: FileApi#upload_file_handler\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
  end
  return data, status_code, headers
end