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/workflow_deployments/client.rb CHANGED
@@ -53,6 +53,9 @@ module Vellum
53
53
  unless request_options&.api_key.nil?
54
54
  req.headers["X-API-KEY"] = request_options.api_key
55
55
  end
56
+ unless request_options&.api_version.nil?
57
+ req.headers["X-API-Version"] = request_options.api_version
58
+ end
56
59
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
57
60
  req.params = { **(request_options&.additional_query_parameters || {}), "limit": limit, "offset": offset, "ordering": ordering, "status": status }.compact
58
61
  unless request_options.nil? || request_options&.additional_body_parameters.nil?
@@ -82,6 +85,9 @@ end
82
85
  unless request_options&.api_key.nil?
83
86
  req.headers["X-API-KEY"] = request_options.api_key
84
87
  end
88
+ unless request_options&.api_version.nil?
89
+ req.headers["X-API-Version"] = request_options.api_version
90
+ end
85
91
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
86
92
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
87
93
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -114,6 +120,9 @@ end
114
120
  unless request_options&.api_key.nil?
115
121
  req.headers["X-API-KEY"] = request_options.api_key
116
122
  end
123
+ unless request_options&.api_version.nil?
124
+ req.headers["X-API-Version"] = request_options.api_version
125
+ end
117
126
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
118
127
  req.params = { **(request_options&.additional_query_parameters || {}), "filters": filters, "limit": limit, "offset": offset }.compact
119
128
  unless request_options.nil? || request_options&.additional_body_parameters.nil?
@@ -142,6 +151,9 @@ end
142
151
  unless request_options&.api_key.nil?
143
152
  req.headers["X-API-KEY"] = request_options.api_key
144
153
  end
154
+ unless request_options&.api_version.nil?
155
+ req.headers["X-API-Version"] = request_options.api_version
156
+ end
145
157
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
146
158
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
147
159
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -178,6 +190,9 @@ end
178
190
  unless request_options&.api_key.nil?
179
191
  req.headers["X-API-KEY"] = request_options.api_key
180
192
  end
193
+ unless request_options&.api_version.nil?
194
+ req.headers["X-API-Version"] = request_options.api_version
195
+ end
181
196
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
182
197
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
183
198
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -213,6 +228,9 @@ end
213
228
  unless request_options&.api_key.nil?
214
229
  req.headers["X-API-KEY"] = request_options.api_key
215
230
  end
231
+ unless request_options&.api_version.nil?
232
+ req.headers["X-API-Version"] = request_options.api_version
233
+ end
216
234
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
217
235
  req.params = { **(request_options&.additional_query_parameters || {}), "limit": limit, "offset": offset, "ordering": ordering, "source": source }.compact
218
236
  unless request_options.nil? || request_options&.additional_body_parameters.nil?
@@ -245,6 +263,9 @@ end
245
263
  unless request_options&.api_key.nil?
246
264
  req.headers["X-API-KEY"] = request_options.api_key
247
265
  end
266
+ unless request_options&.api_version.nil?
267
+ req.headers["X-API-Version"] = request_options.api_version
268
+ end
248
269
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
249
270
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
250
271
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -280,6 +301,9 @@ end
280
301
  unless request_options&.api_key.nil?
281
302
  req.headers["X-API-KEY"] = request_options.api_key
282
303
  end
304
+ unless request_options&.api_version.nil?
305
+ req.headers["X-API-Version"] = request_options.api_version
306
+ end
283
307
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
284
308
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
285
309
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -324,6 +348,9 @@ end
324
348
  unless request_options&.api_key.nil?
325
349
  req.headers["X-API-KEY"] = request_options.api_key
326
350
  end
351
+ unless request_options&.api_version.nil?
352
+ req.headers["X-API-Version"] = request_options.api_version
353
+ end
327
354
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
328
355
  req.params = { **(request_options&.additional_query_parameters || {}), "limit": limit, "offset": offset, "ordering": ordering, "status": status }.compact
329
356
  unless request_options.nil? || request_options&.additional_body_parameters.nil?
@@ -355,6 +382,9 @@ end
355
382
  unless request_options&.api_key.nil?
356
383
  req.headers["X-API-KEY"] = request_options.api_key
357
384
  end
385
+ unless request_options&.api_version.nil?
386
+ req.headers["X-API-Version"] = request_options.api_version
387
+ end
358
388
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
359
389
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
360
390
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -389,6 +419,9 @@ end
389
419
  unless request_options&.api_key.nil?
390
420
  req.headers["X-API-KEY"] = request_options.api_key
391
421
  end
422
+ unless request_options&.api_version.nil?
423
+ req.headers["X-API-Version"] = request_options.api_version
424
+ end
392
425
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
393
426
  req.params = { **(request_options&.additional_query_parameters || {}), "filters": filters, "limit": limit, "offset": offset }.compact
394
427
  unless request_options.nil? || request_options&.additional_body_parameters.nil?
@@ -419,6 +452,9 @@ end
419
452
  unless request_options&.api_key.nil?
420
453
  req.headers["X-API-KEY"] = request_options.api_key
421
454
  end
455
+ unless request_options&.api_version.nil?
456
+ req.headers["X-API-Version"] = request_options.api_version
457
+ end
422
458
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
423
459
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
424
460
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -457,6 +493,9 @@ end
457
493
  unless request_options&.api_key.nil?
458
494
  req.headers["X-API-KEY"] = request_options.api_key
459
495
  end
496
+ unless request_options&.api_version.nil?
497
+ req.headers["X-API-Version"] = request_options.api_version
498
+ end
460
499
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
461
500
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
462
501
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -494,6 +533,9 @@ end
494
533
  unless request_options&.api_key.nil?
495
534
  req.headers["X-API-KEY"] = request_options.api_key
496
535
  end
536
+ unless request_options&.api_version.nil?
537
+ req.headers["X-API-Version"] = request_options.api_version
538
+ end
497
539
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
498
540
  req.params = { **(request_options&.additional_query_parameters || {}), "limit": limit, "offset": offset, "ordering": ordering, "source": source }.compact
499
541
  unless request_options.nil? || request_options&.additional_body_parameters.nil?
@@ -528,6 +570,9 @@ end
528
570
  unless request_options&.api_key.nil?
529
571
  req.headers["X-API-KEY"] = request_options.api_key
530
572
  end
573
+ unless request_options&.api_version.nil?
574
+ req.headers["X-API-Version"] = request_options.api_version
575
+ end
531
576
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
532
577
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
533
578
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -565,6 +610,9 @@ end
565
610
  unless request_options&.api_key.nil?
566
611
  req.headers["X-API-KEY"] = request_options.api_key
567
612
  end
613
+ unless request_options&.api_version.nil?
614
+ req.headers["X-API-Version"] = request_options.api_version
615
+ end
568
616
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
569
617
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
570
618
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
data/lib/vellum_ai/workflow_sandboxes/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
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
54
57
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -82,6 +85,9 @@ end
82
85
  unless request_options&.api_key.nil?
83
86
  req.headers["X-API-KEY"] = request_options.api_key
84
87
  end
88
+ unless request_options&.api_version.nil?
89
+ req.headers["X-API-Version"] = request_options.api_version
90
+ end
85
91
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
86
92
  req.params = { **(request_options&.additional_query_parameters || {}), "limit": limit, "offset": offset, "ordering": ordering, "tag": tag }.compact
87
93
  unless request_options.nil? || request_options&.additional_body_parameters.nil?
@@ -134,6 +140,9 @@ end
134
140
  unless request_options&.api_key.nil?
135
141
  req.headers["X-API-KEY"] = request_options.api_key
136
142
  end
143
+ unless request_options&.api_version.nil?
144
+ req.headers["X-API-Version"] = request_options.api_version
145
+ end
137
146
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
138
147
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
139
148
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -169,6 +178,9 @@ end
169
178
  unless request_options&.api_key.nil?
170
179
  req.headers["X-API-KEY"] = request_options.api_key
171
180
  end
181
+ unless request_options&.api_version.nil?
182
+ req.headers["X-API-Version"] = request_options.api_version
183
+ end
172
184
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
173
185
  req.params = { **(request_options&.additional_query_parameters || {}), "limit": limit, "offset": offset, "ordering": ordering, "tag": tag }.compact
174
186
  unless request_options.nil? || request_options&.additional_body_parameters.nil?
data/lib/vellum_ai/workflows/client.rb CHANGED
@@ -37,6 +37,9 @@ module Vellum
37
37
  unless request_options&.api_key.nil?
38
38
  req.headers["X-API-KEY"] = request_options.api_key
39
39
  end
40
+ unless request_options&.api_version.nil?
41
+ req.headers["X-API-Version"] = request_options.api_version
42
+ end
40
43
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
41
44
  req.options.on_data = on_data
42
45
  req.params = { **(request_options&.additional_query_parameters || {}), "exclude_code": exclude_code, "include_json": include_json, "include_sandbox": include_sandbox, "strict": strict }.compact
@@ -58,13 +61,6 @@ end
58
61
  # @param strict [Boolean]
59
62
  # @param request_options [Vellum::RequestOptions]
60
63
  # @return [Vellum::WorkflowPushResponse]
61
- # @example
62
- # api = Vellum::Client.new(
63
- # base_url: "https://api.example.com",
64
- # environment: Vellum::Environment::PRODUCTION,
65
- # api_key: "YOUR_API_KEY"
66
- # )
67
- # api.workflows.push(artifact: my_file.txt)
68
64
  def push(exec_config:, workflow_sandbox_id: nil, deployment_config: nil, artifact: nil, dry_run: nil, strict: nil, request_options: nil)
69
65
  response = @request_client.conn.post do | req |
70
66
  unless request_options&.timeout_in_seconds.nil?
@@ -73,6 +69,9 @@ end
73
69
  unless request_options&.api_key.nil?
74
70
  req.headers["X-API-KEY"] = request_options.api_key
75
71
  end
72
+ unless request_options&.api_version.nil?
73
+ req.headers["X-API-Version"] = request_options.api_version
74
+ end
76
75
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
77
76
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
78
77
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -115,6 +114,9 @@ end
115
114
  unless request_options&.api_key.nil?
116
115
  req.headers["X-API-KEY"] = request_options.api_key
117
116
  end
117
+ unless request_options&.api_version.nil?
118
+ req.headers["X-API-Version"] = request_options.api_version
119
+ end
118
120
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
119
121
  req.options.on_data = on_data
120
122
  req.params = { **(request_options&.additional_query_parameters || {}), "exclude_code": exclude_code, "include_json": include_json, "include_sandbox": include_sandbox, "strict": strict }.compact
@@ -137,13 +139,6 @@ end
137
139
  # @param strict [Boolean]
138
140
  # @param request_options [Vellum::RequestOptions]
139
141
  # @return [Vellum::WorkflowPushResponse]
140
- # @example
141
- # api = Vellum::Client.new(
142
- # base_url: "https://api.example.com",
143
- # environment: Vellum::Environment::PRODUCTION,
144
- # api_key: "YOUR_API_KEY"
145
- # )
146
- # api.workflows.push(artifact: my_file.txt)
147
142
  def push(exec_config:, workflow_sandbox_id: nil, deployment_config: nil, artifact: nil, dry_run: nil, strict: nil, request_options: nil)
148
143
  Async do
149
144
  response = @request_client.conn.post do | req |
@@ -153,6 +148,9 @@ end
153
148
  unless request_options&.api_key.nil?
154
149
  req.headers["X-API-KEY"] = request_options.api_key
155
150
  end
151
+ unless request_options&.api_version.nil?
152
+ req.headers["X-API-Version"] = request_options.api_version
153
+ end
156
154
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
157
155
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
158
156
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
data/lib/vellum_ai/workspace_secrets/client.rb CHANGED
@@ -36,6 +36,9 @@ module Vellum
36
36
  unless request_options&.api_key.nil?
37
37
  req.headers["X-API-KEY"] = request_options.api_key
38
38
  end
39
+ unless request_options&.api_version.nil?
40
+ req.headers["X-API-Version"] = request_options.api_version
41
+ end
39
42
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
40
43
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
41
44
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -69,6 +72,9 @@ end
69
72
  unless request_options&.api_key.nil?
70
73
  req.headers["X-API-KEY"] = request_options.api_key
71
74
  end
75
+ unless request_options&.api_version.nil?
76
+ req.headers["X-API-Version"] = request_options.api_version
77
+ end
72
78
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
73
79
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
74
80
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -110,6 +116,9 @@ end
110
116
  unless request_options&.api_key.nil?
111
117
  req.headers["X-API-KEY"] = request_options.api_key
112
118
  end
119
+ unless request_options&.api_version.nil?
120
+ req.headers["X-API-Version"] = request_options.api_version
121
+ end
113
122
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
114
123
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
115
124
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
@@ -145,6 +154,9 @@ end
145
154
  unless request_options&.api_key.nil?
146
155
  req.headers["X-API-KEY"] = request_options.api_key
147
156
  end
157
+ unless request_options&.api_version.nil?
158
+ req.headers["X-API-Version"] = request_options.api_version
159
+ end
148
160
  req.headers = { **(req.headers || {}), **@request_client.get_headers, **(request_options&.additional_headers || {}) }.compact
149
161
  unless request_options.nil? || request_options&.additional_query_parameters.nil?
150
162
  req.params = { **(request_options&.additional_query_parameters || {}) }.compact
data/lib/vellum_ai/workspaces/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