vellum_ai 0.14.88 → 0.14.89

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 (23) hide show
  1. checksums.yaml +4 -4
  2. data/lib/requests.rb +27 -8
  3. data/lib/vellum_ai/ad_hoc/client.rb +14 -8
  4. data/lib/vellum_ai/container_images/client.rb +30 -6
  5. data/lib/vellum_ai/deployments/client.rb +50 -2
  6. data/lib/vellum_ai/document_indexes/client.rb +56 -8
  7. data/lib/vellum_ai/documents/client.rb +30 -14
  8. data/lib/vellum_ai/folder_entities/client.rb +12 -0
  9. data/lib/vellum_ai/metric_definitions/client.rb +14 -2
  10. data/lib/vellum_ai/ml_models/client.rb +6 -0
  11. data/lib/vellum_ai/organizations/client.rb +6 -0
  12. data/lib/vellum_ai/prompts/client.rb +14 -2
  13. data/lib/vellum_ai/release_reviews/client.rb +6 -0
  14. data/lib/vellum_ai/sandboxes/client.rb +20 -10
  15. data/lib/vellum_ai/test_suite_runs/client.rb +18 -0
  16. data/lib/vellum_ai/test_suites/client.rb +20 -2
  17. data/lib/vellum_ai/workflow_deployments/client.rb +48 -0
  18. data/lib/vellum_ai/workflow_sandboxes/client.rb +12 -0
  19. data/lib/vellum_ai/workflows/client.rb +12 -14
  20. data/lib/vellum_ai/workspace_secrets/client.rb +12 -0
  21. data/lib/vellum_ai/workspaces/client.rb +6 -0
  22. data/lib/vellum_ai.rb +77 -24
  23. metadata +1 -1
data/lib/vellum_ai/documents/client.rb CHANGED
@@ -49,6 +49,9 @@ module Vellum
49
49
  unless request_options&.api_key.nil?
50
50
  req.headers["X-API-KEY"] = request_options.api_key
51
51
  end
52
+ unless request_options&.api_version.nil?
53
+ req.headers["X-API-Version"] = request_options.api_version
54
+ end
52
55
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
53
56
  req.params = { **(request_options&.additional_query_parameters || {}), "document_index_id": document_index_id, "limit": limit, "offset": offset, "ordering": ordering, "search": search }.compact
54
57
  unless request_options.nil? || request_options&.additional_body_parameters.nil?
@@ -79,6 +82,9 @@ end
79
82
  unless request_options&.api_key.nil?
80
83
  req.headers["X-API-KEY"] = request_options.api_key
81
84
  end
85
+ unless request_options&.api_version.nil?
86
+ req.headers["X-API-Version"] = request_options.api_version
87
+ end
82
88
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
83
89
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
84
90
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -111,6 +117,9 @@ end
111
117
  unless request_options&.api_key.nil?
112
118
  req.headers["X-API-KEY"] = request_options.api_key
113
119
  end
120
+ unless request_options&.api_version.nil?
121
+ req.headers["X-API-Version"] = request_options.api_version
122
+ end
114
123
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
115
124
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
116
125
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -150,6 +159,9 @@ end
150
159
  unless request_options&.api_key.nil?
151
160
  req.headers["X-API-KEY"] = request_options.api_key
152
161
  end
162
+ unless request_options&.api_version.nil?
163
+ req.headers["X-API-Version"] = request_options.api_version
164
+ end
153
165
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
154
166
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
155
167
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -175,13 +187,6 @@ end
175
187
  # that you'd like to filter upon later.
176
188
  # @param request_options [Vellum::RequestOptions]
177
189
  # @return [Vellum::UploadDocumentResponse]
178
- # @example
179
- # api = Vellum::Client.new(
180
- # base_url: "https://api.example.com",
181
- # environment: Vellum::Environment::PRODUCTION,
182
- # api_key: "YOUR_API_KEY"
183
- # )
184
- # api.documents.upload(contents: my_file.txt)
185
190
  def upload(add_to_index_names: nil, external_id: nil, label:, contents:, keywords: nil, metadata: nil, request_options: nil)
186
191
  response = @request_client.conn.post do | req |
187
192
  unless request_options&.timeout_in_seconds.nil?
@@ -190,6 +195,9 @@ end
190
195
  unless request_options&.api_key.nil?
191
196
  req.headers["X-API-KEY"] = request_options.api_key
192
197
  end
198
+ unless request_options&.api_version.nil?
199
+ req.headers["X-API-Version"] = request_options.api_version
200
+ end
193
201
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
194
202
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
195
203
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -237,6 +245,9 @@ end
237
245
  unless request_options&.api_key.nil?
238
246
  req.headers["X-API-KEY"] = request_options.api_key
239
247
  end
248
+ unless request_options&.api_version.nil?
249
+ req.headers["X-API-Version"] = request_options.api_version
250
+ end
240
251
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
241
252
  req.params = { **(request_options&.additional_query_parameters || {}), "document_index_id": document_index_id, "limit": limit, "offset": offset, "ordering": ordering, "search": search }.compact
242
253
  unless request_options.nil? || request_options&.additional_body_parameters.nil?
@@ -269,6 +280,9 @@ end
269
280
  unless request_options&.api_key.nil?
270
281
  req.headers["X-API-KEY"] = request_options.api_key
271
282
  end
283
+ unless request_options&.api_version.nil?
284
+ req.headers["X-API-Version"] = request_options.api_version
285
+ end
272
286
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
273
287
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
274
288
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -303,6 +317,9 @@ end
303
317
  unless request_options&.api_key.nil?
304
318
  req.headers["X-API-KEY"] = request_options.api_key
305
319
  end
320
+ unless request_options&.api_version.nil?
321
+ req.headers["X-API-Version"] = request_options.api_version
322
+ end
306
323
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
307
324
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
308
325
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -344,6 +361,9 @@ end
344
361
  unless request_options&.api_key.nil?
345
362
  req.headers["X-API-KEY"] = request_options.api_key
346
363
  end
364
+ unless request_options&.api_version.nil?
365
+ req.headers["X-API-Version"] = request_options.api_version
366
+ end
347
367
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
348
368
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
349
369
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -370,13 +390,6 @@ end
370
390
  # that you'd like to filter upon later.
371
391
  # @param request_options [Vellum::RequestOptions]
372
392
  # @return [Vellum::UploadDocumentResponse]
373
- # @example
374
- # api = Vellum::Client.new(
375
- # base_url: "https://api.example.com",
376
- # environment: Vellum::Environment::PRODUCTION,
377
- # api_key: "YOUR_API_KEY"
378
- # )
379
- # api.documents.upload(contents: my_file.txt)
380
393
  def upload(add_to_index_names: nil, external_id: nil, label:, contents:, keywords: nil, metadata: nil, request_options: nil)
381
394
  Async do
382
395
  response = @request_client.conn.post do | req |
@@ -386,6 +399,9 @@ end
386
399
  unless request_options&.api_key.nil?
387
400
  req.headers["X-API-KEY"] = request_options.api_key
388
401
  end
402
+ unless request_options&.api_version.nil?
403
+ req.headers["X-API-Version"] = request_options.api_version
404
+ end
389
405
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
390
406
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
391
407
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
data/lib/vellum_ai/folder_entities/client.rb CHANGED
@@ -52,6 +52,9 @@ module Vellum
52
52
  unless request_options&.api_key.nil?
53
53
  req.headers["X-API-KEY"] = request_options.api_key
54
54
  end
55
+ unless request_options&.api_version.nil?
56
+ req.headers["X-API-Version"] = request_options.api_version
57
+ end
55
58
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
56
59
  req.params = { **(request_options&.additional_query_parameters || {}), "entity_status": entity_status, "limit": limit, "offset": offset, "ordering": ordering, "parent_folder_id": parent_folder_id }.compact
57
60
  unless request_options.nil? || request_options&.additional_body_parameters.nil?
@@ -90,6 +93,9 @@ end
90
93
  unless request_options&.api_key.nil?
91
94
  req.headers["X-API-KEY"] = request_options.api_key
92
95
  end
96
+ unless request_options&.api_version.nil?
97
+ req.headers["X-API-Version"] = request_options.api_version
98
+ end
93
99
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
94
100
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
95
101
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -145,6 +151,9 @@ end
145
151
  unless request_options&.api_key.nil?
146
152
  req.headers["X-API-KEY"] = request_options.api_key
147
153
  end
154
+ unless request_options&.api_version.nil?
155
+ req.headers["X-API-Version"] = request_options.api_version
156
+ end
148
157
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
149
158
  req.params = { **(request_options&.additional_query_parameters || {}), "entity_status": entity_status, "limit": limit, "offset": offset, "ordering": ordering, "parent_folder_id": parent_folder_id }.compact
150
159
  unless request_options.nil? || request_options&.additional_body_parameters.nil?
@@ -185,6 +194,9 @@ end
185
194
  unless request_options&.api_key.nil?
186
195
  req.headers["X-API-KEY"] = request_options.api_key
187
196
  end
197
+ unless request_options&.api_version.nil?
198
+ req.headers["X-API-Version"] = request_options.api_version
199
+ end
188
200
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
189
201
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
190
202
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
data/lib/vellum_ai/metric_definitions/client.rb CHANGED
@@ -29,7 +29,7 @@ module Vellum
29
29
  # environment: Vellum::Environment::PRODUCTION,
30
30
  # api_key: "YOUR_API_KEY"
31
31
  # )
32
- # api.metric_definitions.execute_metric_definition(id: "id", inputs: [{ name: "name", type: "STRING", value: "value" }])
32
+ # api.metric_definitions.execute_metric_definition(id: "id", inputs: [{ name: "x", type: "STRING", value: "value" }, { name: "x", type: "STRING", value: "value" }])
33
33
  def execute_metric_definition(id:, inputs:, release_tag: nil, request_options: nil)
34
34
  response = @request_client.conn.post do | req |
35
35
  unless request_options&.timeout_in_seconds.nil?
@@ -38,6 +38,9 @@ module Vellum
38
38
  unless request_options&.api_key.nil?
39
39
  req.headers["X-API-KEY"] = request_options.api_key
40
40
  end
41
+ unless request_options&.api_version.nil?
42
+ req.headers["X-API-Version"] = request_options.api_version
43
+ end
41
44
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
42
45
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
43
46
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -68,6 +71,9 @@ end
68
71
  unless request_options&.api_key.nil?
69
72
  req.headers["X-API-KEY"] = request_options.api_key
70
73
  end
74
+ unless request_options&.api_version.nil?
75
+ req.headers["X-API-Version"] = request_options.api_version
76
+ end
71
77
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
72
78
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
73
79
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -101,7 +107,7 @@ end
101
107
  # environment: Vellum::Environment::PRODUCTION,
102
108
  # api_key: "YOUR_API_KEY"
103
109
  # )
104
- # api.metric_definitions.execute_metric_definition(id: "id", inputs: [{ name: "name", type: "STRING", value: "value" }])
110
+ # api.metric_definitions.execute_metric_definition(id: "id", inputs: [{ name: "x", type: "STRING", value: "value" }, { name: "x", type: "STRING", value: "value" }])
105
111
  def execute_metric_definition(id:, inputs:, release_tag: nil, request_options: nil)
106
112
  Async do
107
113
  response = @request_client.conn.post do | req |
@@ -111,6 +117,9 @@ end
111
117
  unless request_options&.api_key.nil?
112
118
  req.headers["X-API-KEY"] = request_options.api_key
113
119
  end
120
+ unless request_options&.api_version.nil?
121
+ req.headers["X-API-Version"] = request_options.api_version
122
+ end
114
123
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
115
124
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
116
125
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -143,6 +152,9 @@ end
143
152
  unless request_options&.api_key.nil?
144
153
  req.headers["X-API-KEY"] = request_options.api_key
145
154
  end
155
+ unless request_options&.api_version.nil?
156
+ req.headers["X-API-Version"] = request_options.api_version
157
+ end
146
158
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
147
159
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
148
160
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
data/lib/vellum_ai/ml_models/client.rb CHANGED
@@ -35,6 +35,9 @@ module Vellum
35
35
  unless request_options&.api_key.nil?
36
36
  req.headers["X-API-KEY"] = request_options.api_key
37
37
  end
38
+ unless request_options&.api_version.nil?
39
+ req.headers["X-API-Version"] = request_options.api_version
40
+ end
38
41
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
39
42
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
40
43
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -78,6 +81,9 @@ end
78
81
  unless request_options&.api_key.nil?
79
82
  req.headers["X-API-KEY"] = request_options.api_key
80
83
  end
84
+ unless request_options&.api_version.nil?
85
+ req.headers["X-API-Version"] = request_options.api_version
86
+ end
81
87
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
82
88
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
83
89
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
data/lib/vellum_ai/organizations/client.rb CHANGED
@@ -34,6 +34,9 @@ module Vellum
34
34
  unless request_options&.api_key.nil?
35
35
  req.headers["X-API-KEY"] = request_options.api_key
36
36
  end
37
+ unless request_options&.api_version.nil?
38
+ req.headers["X-API-Version"] = request_options.api_version
39
+ end
37
40
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
38
41
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
39
42
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -76,6 +79,9 @@ end
76
79
  unless request_options&.api_key.nil?
77
80
  req.headers["X-API-KEY"] = request_options.api_key
78
81
  end
82
+ unless request_options&.api_version.nil?
83
+ req.headers["X-API-Version"] = request_options.api_version
84
+ end
79
85
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
80
86
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
81
87
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
data/lib/vellum_ai/prompts/client.rb CHANGED
@@ -39,6 +39,9 @@ module Vellum
39
39
  unless request_options&.api_key.nil?
40
40
  req.headers["X-API-KEY"] = request_options.api_key
41
41
  end
42
+ unless request_options&.api_version.nil?
43
+ req.headers["X-API-Version"] = request_options.api_version
44
+ end
42
45
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}), "Accept": "application/json" }.compact
43
46
  req.params = { **(request_options&.additional_query_parameters || {}), "prompt_variant_id": prompt_variant_id }.compact
44
47
  unless request_options.nil? || request_options&.additional_body_parameters.nil?
@@ -80,7 +83,7 @@ end
80
83
  # environment: Vellum::Environment::PRODUCTION,
81
84
  # api_key: "YOUR_API_KEY"
82
85
  # )
83
- # api.prompts.push(exec_config: { ml_model: "ml_model", input_variables: [{ id: "id", key: "key", type: STRING }], parameters: { }, blocks: [{ block_type: "JINJA", template: "template" }] })
86
+ # api.prompts.push(exec_config: { ml_model: "ml_model", input_variables: [{ id: "x", key: "key", type: STRING }, { id: "x", key: "key", type: STRING }], parameters: { }, blocks: [{ block_type: "JINJA", template: "template" }, { block_type: "JINJA", template: "template" }] })
84
87
  def push(prompt_variant_id: nil, prompt_variant_label: nil, prompt_sandbox_id: nil, exec_config:, request_options: nil)
85
88
  response = @request_client.conn.post do | req |
86
89
  unless request_options&.timeout_in_seconds.nil?
@@ -89,6 +92,9 @@ end
89
92
  unless request_options&.api_key.nil?
90
93
  req.headers["X-API-KEY"] = request_options.api_key
91
94
  end
95
+ unless request_options&.api_version.nil?
96
+ req.headers["X-API-Version"] = request_options.api_version
97
+ end
92
98
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
93
99
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
94
100
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -132,6 +138,9 @@ end
132
138
  unless request_options&.api_key.nil?
133
139
  req.headers["X-API-KEY"] = request_options.api_key
134
140
  end
141
+ unless request_options&.api_version.nil?
142
+ req.headers["X-API-Version"] = request_options.api_version
143
+ end
135
144
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}), "Accept": "application/json" }.compact
136
145
  req.params = { **(request_options&.additional_query_parameters || {}), "prompt_variant_id": prompt_variant_id }.compact
137
146
  unless request_options.nil? || request_options&.additional_body_parameters.nil?
@@ -174,7 +183,7 @@ end
174
183
  # environment: Vellum::Environment::PRODUCTION,
175
184
  # api_key: "YOUR_API_KEY"
176
185
  # )
177
- # api.prompts.push(exec_config: { ml_model: "ml_model", input_variables: [{ id: "id", key: "key", type: STRING }], parameters: { }, blocks: [{ block_type: "JINJA", template: "template" }] })
186
+ # api.prompts.push(exec_config: { ml_model: "ml_model", input_variables: [{ id: "x", key: "key", type: STRING }, { id: "x", key: "key", type: STRING }], parameters: { }, blocks: [{ block_type: "JINJA", template: "template" }, { block_type: "JINJA", template: "template" }] })
178
187
  def push(prompt_variant_id: nil, prompt_variant_label: nil, prompt_sandbox_id: nil, exec_config:, request_options: nil)
179
188
  Async do
180
189
  response = @request_client.conn.post do | req |
@@ -184,6 +193,9 @@ end
184
193
  unless request_options&.api_key.nil?
185
194
  req.headers["X-API-KEY"] = request_options.api_key
186
195
  end
196
+ unless request_options&.api_version.nil?
197
+ req.headers["X-API-Version"] = request_options.api_version
198
+ end
187
199
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
188
200
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
189
201
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
data/lib/vellum_ai/release_reviews/client.rb CHANGED
@@ -39,6 +39,9 @@ module Vellum
39
39
  unless request_options&.api_key.nil?
40
40
  req.headers["X-API-KEY"] = request_options.api_key
41
41
  end
42
+ unless request_options&.api_version.nil?
43
+ req.headers["X-API-Version"] = request_options.api_version
44
+ end
42
45
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
43
46
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
44
47
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -86,6 +89,9 @@ end
86
89
  unless request_options&.api_key.nil?
87
90
  req.headers["X-API-KEY"] = request_options.api_key
88
91
  end
92
+ unless request_options&.api_version.nil?
93
+ req.headers["X-API-Version"] = request_options.api_version
94
+ end
89
95
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
90
96
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
91
97
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
data/lib/vellum_ai/sandboxes/client.rb CHANGED
@@ -50,6 +50,9 @@ module Vellum
50
50
  unless request_options&.api_key.nil?
51
51
  req.headers["X-API-KEY"] = request_options.api_key
52
52
  end
53
+ unless request_options&.api_version.nil?
54
+ req.headers["X-API-Version"] = request_options.api_version
55
+ end
53
56
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
54
57
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
55
58
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -81,11 +84,7 @@ end
81
84
  # environment: Vellum::Environment::PRODUCTION,
82
85
  # api_key: "YOUR_API_KEY"
83
86
  # )
84
- # api.sandboxes.upsert_sandbox_scenario(
85
- # id: "id",
86
- # label: "Scenario 1",
87
- # inputs: [{ type: "STRING", value: "Hello, world!", name: "var_1" }, { type: "STRING", value: "Why hello, there!", name: "var_2" }]
88
- # )
87
+ # api.sandboxes.upsert_sandbox_scenario(id: "id", inputs: [{ type: "STRING", name: "x" }, { type: "STRING", name: "x" }])
89
88
  def upsert_sandbox_scenario(id:, label: nil, inputs:, scenario_id: nil, request_options: nil)
90
89
  response = @request_client.conn.post do | req |
91
90
  unless request_options&.timeout_in_seconds.nil?
@@ -94,6 +93,9 @@ end
94
93
  unless request_options&.api_key.nil?
95
94
  req.headers["X-API-KEY"] = request_options.api_key
96
95
  end
96
+ unless request_options&.api_version.nil?
97
+ req.headers["X-API-Version"] = request_options.api_version
98
+ end
97
99
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
98
100
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
99
101
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -125,6 +127,9 @@ end
125
127
  unless request_options&.api_key.nil?
126
128
  req.headers["X-API-KEY"] = request_options.api_key
127
129
  end
130
+ unless request_options&.api_version.nil?
131
+ req.headers["X-API-Version"] = request_options.api_version
132
+ end
128
133
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
129
134
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
130
135
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -178,6 +183,9 @@ end
178
183
  unless request_options&.api_key.nil?
179
184
  req.headers["X-API-KEY"] = request_options.api_key
180
185
  end
186
+ unless request_options&.api_version.nil?
187
+ req.headers["X-API-Version"] = request_options.api_version
188
+ end
181
189
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
182
190
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
183
191
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -210,11 +218,7 @@ end
210
218
  # environment: Vellum::Environment::PRODUCTION,
211
219
  # api_key: "YOUR_API_KEY"
212
220
  # )
213
- # api.sandboxes.upsert_sandbox_scenario(
214
- # id: "id",
215
- # label: "Scenario 1",
216
- # inputs: [{ type: "STRING", value: "Hello, world!", name: "var_1" }, { type: "STRING", value: "Why hello, there!", name: "var_2" }]
217
- # )
221
+ # api.sandboxes.upsert_sandbox_scenario(id: "id", inputs: [{ type: "STRING", name: "x" }, { type: "STRING", name: "x" }])
218
222
  def upsert_sandbox_scenario(id:, label: nil, inputs:, scenario_id: nil, request_options: nil)
219
223
  Async do
220
224
  response = @request_client.conn.post do | req |
@@ -224,6 +228,9 @@ end
224
228
  unless request_options&.api_key.nil?
225
229
  req.headers["X-API-KEY"] = request_options.api_key
226
230
  end
231
+ unless request_options&.api_version.nil?
232
+ req.headers["X-API-Version"] = request_options.api_version
233
+ end
227
234
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
228
235
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
229
236
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -257,6 +264,9 @@ end
257
264
  unless request_options&.api_key.nil?
258
265
  req.headers["X-API-KEY"] = request_options.api_key
259
266
  end
267
+ unless request_options&.api_version.nil?
268
+ req.headers["X-API-Version"] = request_options.api_version
269
+ end
260
270
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
261
271
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
262
272
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
data/lib/vellum_ai/test_suite_runs/client.rb CHANGED
@@ -41,6 +41,9 @@ module Vellum
41
41
  unless request_options&.api_key.nil?
42
42
  req.headers["X-API-KEY"] = request_options.api_key
43
43
  end
44
+ unless request_options&.api_version.nil?
45
+ req.headers["X-API-Version"] = request_options.api_version
46
+ end
44
47
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
45
48
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
46
49
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -70,6 +73,9 @@ end
70
73
  unless request_options&.api_key.nil?
71
74
  req.headers["X-API-KEY"] = request_options.api_key
72
75
  end
76
+ unless request_options&.api_version.nil?
77
+ req.headers["X-API-Version"] = request_options.api_version
78
+ end
73
79
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
74
80
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
75
81
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -108,6 +114,9 @@ end
108
114
  unless request_options&.api_key.nil?
109
115
  req.headers["X-API-KEY"] = request_options.api_key
110
116
  end
117
+ unless request_options&.api_version.nil?
118
+ req.headers["X-API-Version"] = request_options.api_version
119
+ end
111
120
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
112
121
  req.params = { **(request_options&.additional_query_parameters || {}), "expand": expand, "limit": limit, "offset": offset }.compact
113
122
  unless request_options.nil? || request_options&.additional_body_parameters.nil?
@@ -151,6 +160,9 @@ end
151
160
  unless request_options&.api_key.nil?
152
161
  req.headers["X-API-KEY"] = request_options.api_key
153
162
  end
163
+ unless request_options&.api_version.nil?
164
+ req.headers["X-API-Version"] = request_options.api_version
165
+ end
154
166
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
155
167
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
156
168
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -182,6 +194,9 @@ end
182
194
  unless request_options&.api_key.nil?
183
195
  req.headers["X-API-KEY"] = request_options.api_key
184
196
  end
197
+ unless request_options&.api_version.nil?
198
+ req.headers["X-API-Version"] = request_options.api_version
199
+ end
185
200
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
186
201
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
187
202
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -222,6 +237,9 @@ end
222
237
  unless request_options&.api_key.nil?
223
238
  req.headers["X-API-KEY"] = request_options.api_key
224
239
  end
240
+ unless request_options&.api_version.nil?
241
+ req.headers["X-API-Version"] = request_options.api_version
242
+ end
225
243
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
226
244
  req.params = { **(request_options&.additional_query_parameters || {}), "expand": expand, "limit": limit, "offset": offset }.compact
227
245
  unless request_options.nil? || request_options&.additional_body_parameters.nil?
data/lib/vellum_ai/test_suites/client.rb CHANGED
@@ -41,6 +41,9 @@ module Vellum
41
41
  unless request_options&.api_key.nil?
42
42
  req.headers["X-API-KEY"] = request_options.api_key
43
43
  end
44
+ unless request_options&.api_version.nil?
45
+ req.headers["X-API-Version"] = request_options.api_version
46
+ end
44
47
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
45
48
  req.params = { **(request_options&.additional_query_parameters || {}), "limit": limit, "offset": offset }.compact
46
49
  unless request_options.nil? || request_options&.additional_body_parameters.nil?
@@ -74,7 +77,7 @@ end
74
77
  # environment: Vellum::Environment::PRODUCTION,
75
78
  # api_key: "YOUR_API_KEY"
76
79
  # )
77
- # api.test_suites.upsert_test_suite_test_case(id: "id", request: { label: "Test Case 1", input_values: [{ type: "STRING", value: "What are your favorite colors?", name: "var_1" }], evaluation_values: [{ type: "ARRAY", value: [{ type: "STRING", value: "Red" }, { type: "STRING", value: "Green" }, { type: "STRING", value: "Blue" }], name: "var_2" }] })
80
+ # api.test_suites.upsert_test_suite_test_case(id: "id", request: { input_values: [{ type: "STRING", name: "x" }, { type: "STRING", name: "x" }], evaluation_values: [{ type: "STRING", name: "x" }, { type: "STRING", name: "x" }] })
78
81
  def upsert_test_suite_test_case(id:, request:, request_options: nil)
79
82
  response = @request_client.conn.post do | req |
80
83
  unless request_options&.timeout_in_seconds.nil?
@@ -83,6 +86,9 @@ end
83
86
  unless request_options&.api_key.nil?
84
87
  req.headers["X-API-KEY"] = request_options.api_key
85
88
  end
89
+ unless request_options&.api_version.nil?
90
+ req.headers["X-API-Version"] = request_options.api_version
91
+ end
86
92
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
87
93
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
88
94
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -113,6 +119,9 @@ end
113
119
  unless request_options&.api_key.nil?
114
120
  req.headers["X-API-KEY"] = request_options.api_key
115
121
  end
122
+ unless request_options&.api_version.nil?
123
+ req.headers["X-API-Version"] = request_options.api_version
124
+ end
116
125
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
117
126
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
118
127
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -157,6 +166,9 @@ end
157
166
  unless request_options&.api_key.nil?
158
167
  req.headers["X-API-KEY"] = request_options.api_key
159
168
  end
169
+ unless request_options&.api_version.nil?
170
+ req.headers["X-API-Version"] = request_options.api_version
171
+ end
160
172
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
161
173
  req.params = { **(request_options&.additional_query_parameters || {}), "limit": limit, "offset": offset }.compact
162
174
  unless request_options.nil? || request_options&.additional_body_parameters.nil?
@@ -191,7 +203,7 @@ end
191
203
  # environment: Vellum::Environment::PRODUCTION,
192
204
  # api_key: "YOUR_API_KEY"
193
205
  # )
194
- # api.test_suites.upsert_test_suite_test_case(id: "id", request: { label: "Test Case 1", input_values: [{ type: "STRING", value: "What are your favorite colors?", name: "var_1" }], evaluation_values: [{ type: "ARRAY", value: [{ type: "STRING", value: "Red" }, { type: "STRING", value: "Green" }, { type: "STRING", value: "Blue" }], name: "var_2" }] })
206
+ # api.test_suites.upsert_test_suite_test_case(id: "id", request: { input_values: [{ type: "STRING", name: "x" }, { type: "STRING", name: "x" }], evaluation_values: [{ type: "STRING", name: "x" }, { type: "STRING", name: "x" }] })
195
207
  def upsert_test_suite_test_case(id:, request:, request_options: nil)
196
208
  Async do
197
209
  response = @request_client.conn.post do | req |
@@ -201,6 +213,9 @@ end
201
213
  unless request_options&.api_key.nil?
202
214
  req.headers["X-API-KEY"] = request_options.api_key
203
215
  end
216
+ unless request_options&.api_version.nil?
217
+ req.headers["X-API-Version"] = request_options.api_version
218
+ end
204
219
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
205
220
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
206
221
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -233,6 +248,9 @@ end
233
248
  unless request_options&.api_key.nil?
234
249
  req.headers["X-API-KEY"] = request_options.api_key
235
250
  end
251
+ unless request_options&.api_version.nil?
252
+ req.headers["X-API-Version"] = request_options.api_version
253
+ end
236
254
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
237
255
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
238
256
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact