vellum-ai 0.14.81__py3-none-any.whl → 0.14.83__py3-none-any.whl

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 (55) hide show
  1. vellum/__init__.py +10 -0
  2. vellum/client/README.md +1 -9
  3. vellum/client/__init__.py +44 -24
  4. vellum/client/core/client_wrapper.py +2 -1
  5. vellum/client/core/http_client.py +0 -2
  6. vellum/client/core/jsonable_encoder.py +0 -1
  7. vellum/client/core/pydantic_utilities.py +1 -3
  8. vellum/client/core/serialization.py +1 -2
  9. vellum/client/reference.md +8 -0
  10. vellum/client/resources/ad_hoc/client.py +20 -12
  11. vellum/client/resources/deployments/client.py +6 -2
  12. vellum/client/resources/documents/client.py +10 -0
  13. vellum/client/types/__init__.py +10 -0
  14. vellum/client/types/container_image_build_config.py +1 -0
  15. vellum/client/types/document_read.py +5 -0
  16. vellum/client/types/execution_thinking_vellum_value.py +31 -0
  17. vellum/client/types/execution_vellum_value.py +2 -0
  18. vellum/client/types/google_vertex_ai_vectorizer_gemini_embedding_001.py +21 -0
  19. vellum/client/types/google_vertex_ai_vectorizer_gemini_embedding_001_request.py +21 -0
  20. vellum/client/types/indexing_config_vectorizer.py +2 -0
  21. vellum/client/types/indexing_config_vectorizer_request.py +2 -0
  22. vellum/client/types/prompt_output.py +4 -1
  23. vellum/client/types/thinking_vellum_value.py +25 -0
  24. vellum/client/types/thinking_vellum_value_request.py +25 -0
  25. vellum/client/types/vellum_value.py +2 -0
  26. vellum/client/types/vellum_value_request.py +2 -0
  27. vellum/client/types/vellum_variable_type.py +1 -0
  28. vellum/client/types/workflow_sandbox_example.py +2 -0
  29. vellum/types/execution_thinking_vellum_value.py +3 -0
  30. vellum/types/google_vertex_ai_vectorizer_gemini_embedding_001.py +3 -0
  31. vellum/types/google_vertex_ai_vectorizer_gemini_embedding_001_request.py +3 -0
  32. vellum/types/thinking_vellum_value.py +3 -0
  33. vellum/types/thinking_vellum_value_request.py +3 -0
  34. vellum/workflows/nodes/displayable/__init__.py +2 -0
  35. vellum/workflows/nodes/displayable/inline_prompt_node/node.py +3 -0
  36. vellum/workflows/nodes/displayable/prompt_deployment_node/node.py +3 -0
  37. vellum/workflows/nodes/displayable/tool_calling_node/__init__.py +3 -0
  38. vellum/workflows/nodes/{experimental → displayable}/tool_calling_node/node.py +4 -21
  39. vellum/workflows/nodes/{experimental → displayable}/tool_calling_node/tests/test_node.py +1 -1
  40. vellum/workflows/nodes/{experimental → displayable}/tool_calling_node/tests/test_utils.py +1 -1
  41. vellum/workflows/nodes/{experimental → displayable}/tool_calling_node/utils.py +22 -57
  42. vellum/workflows/nodes/experimental/__init__.py +1 -1
  43. vellum/workflows/nodes/experimental/tool_calling_node/__init__.py +1 -1
  44. vellum/workflows/state/encoder.py +2 -2
  45. {vellum_ai-0.14.81.dist-info → vellum_ai-0.14.83.dist-info}/METADATA +1 -1
  46. {vellum_ai-0.14.81.dist-info → vellum_ai-0.14.83.dist-info}/RECORD +55 -44
  47. vellum_ee/workflows/display/nodes/vellum/tests/test_tool_calling_node.py +1 -104
  48. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_inline_workflow_serialization.py +1 -6
  49. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_serialization.py +1 -17
  50. vellum_ee/workflows/display/workflows/base_workflow_display.py +10 -5
  51. vellum_ee/workflows/tests/test_serialize_module.py +1 -1
  52. /vellum/workflows/nodes/{experimental → displayable}/tool_calling_node/tests/__init__.py +0 -0
  53. {vellum_ai-0.14.81.dist-info → vellum_ai-0.14.83.dist-info}/LICENSE +0 -0
  54. {vellum_ai-0.14.81.dist-info → vellum_ai-0.14.83.dist-info}/WHEEL +0 -0
  55. {vellum_ai-0.14.81.dist-info → vellum_ai-0.14.83.dist-info}/entry_points.txt +0 -0
vellum/__init__.py CHANGED
@@ -121,6 +121,7 @@ from .types import (
121
121
  ExecutionNumberVellumValue,
122
122
  ExecutionSearchResultsVellumValue,
123
123
  ExecutionStringVellumValue,
124
+ ExecutionThinkingVellumValue,
124
125
  ExecutionVellumValue,
125
126
  ExternalInputDescriptor,
126
127
  ExternalTestCaseExecution,
@@ -169,6 +170,8 @@ from .types import (
169
170
  GenerateStreamResultData,
170
171
  GoogleVertexAiVectorizerConfig,
171
172
  GoogleVertexAiVectorizerConfigRequest,
173
+ GoogleVertexAiVectorizerGeminiEmbedding001,
174
+ GoogleVertexAiVectorizerGeminiEmbedding001Request,
172
175
  GoogleVertexAiVectorizerTextEmbedding004,
173
176
  GoogleVertexAiVectorizerTextEmbedding004Request,
174
177
  GoogleVertexAiVectorizerTextMultilingualEmbedding002,
@@ -487,6 +490,8 @@ from .types import (
487
490
  TestSuiteTestCaseReplacedBulkResult,
488
491
  TestSuiteTestCaseReplacedBulkResultData,
489
492
  TestSuiteTestCaseUpsertBulkOperationRequest,
493
+ ThinkingVellumValue,
494
+ ThinkingVellumValueRequest,
490
495
  TokenOverlappingWindowChunkerConfig,
491
496
  TokenOverlappingWindowChunkerConfigRequest,
492
497
  TokenOverlappingWindowChunking,
@@ -754,6 +759,7 @@ __all__ = [
754
759
  "ExecutionNumberVellumValue",
755
760
  "ExecutionSearchResultsVellumValue",
756
761
  "ExecutionStringVellumValue",
762
+ "ExecutionThinkingVellumValue",
757
763
  "ExecutionVellumValue",
758
764
  "ExternalInputDescriptor",
759
765
  "ExternalTestCaseExecution",
@@ -804,6 +810,8 @@ __all__ = [
804
810
  "GenerateStreamResultData",
805
811
  "GoogleVertexAiVectorizerConfig",
806
812
  "GoogleVertexAiVectorizerConfigRequest",
813
+ "GoogleVertexAiVectorizerGeminiEmbedding001",
814
+ "GoogleVertexAiVectorizerGeminiEmbedding001Request",
807
815
  "GoogleVertexAiVectorizerTextEmbedding004",
808
816
  "GoogleVertexAiVectorizerTextEmbedding004Request",
809
817
  "GoogleVertexAiVectorizerTextMultilingualEmbedding002",
@@ -1127,6 +1135,8 @@ __all__ = [
1127
1135
  "TestSuiteTestCaseReplacedBulkResult",
1128
1136
  "TestSuiteTestCaseReplacedBulkResultData",
1129
1137
  "TestSuiteTestCaseUpsertBulkOperationRequest",
1138
+ "ThinkingVellumValue",
1139
+ "ThinkingVellumValueRequest",
1130
1140
  "TokenOverlappingWindowChunkerConfig",
1131
1141
  "TokenOverlappingWindowChunkerConfigRequest",
1132
1142
  "TokenOverlappingWindowChunking",
vellum/client/README.md CHANGED
@@ -173,6 +173,7 @@ client.execute_prompt(..., request_options={
173
173
 
174
174
  You can override the `httpx` client to customize it for your use-case. Some common use-cases include support for proxies
175
175
  and transports.
176
+
176
177
  ```python
177
178
  import httpx
178
179
  from vellum import Vellum
@@ -186,12 +187,3 @@ client = Vellum(
186
187
  )
187
188
  ```
188
189
 
189
- ## Contributing
190
-
191
- While we value open-source contributions to this SDK, this library is generated programmatically.
192
- Additions made directly to this library would have to be moved over to our generation code,
193
- otherwise they would be overwritten upon the next generated release. Feel free to open a PR as
194
- a proof of concept, but know that we will not be able to merge it as-is. We suggest opening
195
- an issue first to discuss with us!
196
-
197
- On the other hand, contributions to the README are always very welcome!
vellum/client/__init__.py CHANGED
@@ -129,7 +129,9 @@ class Vellum:
129
129
  follow_redirects: typing.Optional[bool] = True,
130
130
  httpx_client: typing.Optional[httpx.Client] = None,
131
131
  ):
132
- _defaulted_timeout = timeout if timeout is not None else None if httpx_client is None else None
132
+ _defaulted_timeout = (
133
+ timeout if timeout is not None else None if httpx_client is None else httpx_client.timeout.read
134
+ )
133
135
  self._client_wrapper = SyncClientWrapper(
134
136
  environment=environment,
135
137
  api_key=api_key,
@@ -210,13 +212,13 @@ class Vellum:
210
212
  "url": url,
211
213
  "method": method,
212
214
  "body": convert_and_respect_annotation_metadata(
213
- object_=body, annotation=ExecuteApiRequestBody, direction="write"
215
+ object_=body, annotation=typing.Optional[ExecuteApiRequestBody], direction="write"
214
216
  ),
215
217
  "headers": convert_and_respect_annotation_metadata(
216
218
  object_=headers, annotation=typing.Dict[str, ExecuteApiRequestHeadersValue], direction="write"
217
219
  ),
218
220
  "bearer_token": convert_and_respect_annotation_metadata(
219
- object_=bearer_token, annotation=ExecuteApiRequestBearerToken, direction="write"
221
+ object_=bearer_token, annotation=typing.Optional[ExecuteApiRequestBearerToken], direction="write"
220
222
  ),
221
223
  },
222
224
  headers={
@@ -423,10 +425,14 @@ class Vellum:
423
425
  "release_tag": release_tag,
424
426
  "external_id": external_id,
425
427
  "expand_meta": convert_and_respect_annotation_metadata(
426
- object_=expand_meta, annotation=PromptDeploymentExpandMetaRequest, direction="write"
428
+ object_=expand_meta,
429
+ annotation=typing.Optional[PromptDeploymentExpandMetaRequest],
430
+ direction="write",
427
431
  ),
428
432
  "raw_overrides": convert_and_respect_annotation_metadata(
429
- object_=raw_overrides, annotation=RawPromptExecutionOverridesRequest, direction="write"
433
+ object_=raw_overrides,
434
+ annotation=typing.Optional[RawPromptExecutionOverridesRequest],
435
+ direction="write",
430
436
  ),
431
437
  "expand_raw": expand_raw,
432
438
  "metadata": metadata,
@@ -576,10 +582,14 @@ class Vellum:
576
582
  "release_tag": release_tag,
577
583
  "external_id": external_id,
578
584
  "expand_meta": convert_and_respect_annotation_metadata(
579
- object_=expand_meta, annotation=PromptDeploymentExpandMetaRequest, direction="write"
585
+ object_=expand_meta,
586
+ annotation=typing.Optional[PromptDeploymentExpandMetaRequest],
587
+ direction="write",
580
588
  ),
581
589
  "raw_overrides": convert_and_respect_annotation_metadata(
582
- object_=raw_overrides, annotation=RawPromptExecutionOverridesRequest, direction="write"
590
+ object_=raw_overrides,
591
+ annotation=typing.Optional[RawPromptExecutionOverridesRequest],
592
+ direction="write",
583
593
  ),
584
594
  "expand_raw": expand_raw,
585
595
  "metadata": metadata,
@@ -723,7 +733,7 @@ class Vellum:
723
733
  object_=inputs, annotation=typing.Sequence[WorkflowRequestInputRequest], direction="write"
724
734
  ),
725
735
  "expand_meta": convert_and_respect_annotation_metadata(
726
- object_=expand_meta, annotation=WorkflowExpandMetaRequest, direction="write"
736
+ object_=expand_meta, annotation=typing.Optional[WorkflowExpandMetaRequest], direction="write"
727
737
  ),
728
738
  "workflow_deployment_id": workflow_deployment_id,
729
739
  "workflow_deployment_name": workflow_deployment_name,
@@ -858,7 +868,7 @@ class Vellum:
858
868
  object_=inputs, annotation=typing.Sequence[WorkflowRequestInputRequest], direction="write"
859
869
  ),
860
870
  "expand_meta": convert_and_respect_annotation_metadata(
861
- object_=expand_meta, annotation=WorkflowExpandMetaRequest, direction="write"
871
+ object_=expand_meta, annotation=typing.Optional[WorkflowExpandMetaRequest], direction="write"
862
872
  ),
863
873
  "workflow_deployment_id": workflow_deployment_id,
864
874
  "workflow_deployment_name": workflow_deployment_name,
@@ -988,7 +998,7 @@ class Vellum:
988
998
  object_=requests, annotation=typing.Sequence[GenerateRequest], direction="write"
989
999
  ),
990
1000
  "options": convert_and_respect_annotation_metadata(
991
- object_=options, annotation=GenerateOptionsRequest, direction="write"
1001
+ object_=options, annotation=typing.Optional[GenerateOptionsRequest], direction="write"
992
1002
  ),
993
1003
  },
994
1004
  headers={
@@ -1116,7 +1126,7 @@ class Vellum:
1116
1126
  object_=requests, annotation=typing.Sequence[GenerateRequest], direction="write"
1117
1127
  ),
1118
1128
  "options": convert_and_respect_annotation_metadata(
1119
- object_=options, annotation=GenerateOptionsRequest, direction="write"
1129
+ object_=options, annotation=typing.Optional[GenerateOptionsRequest], direction="write"
1120
1130
  ),
1121
1131
  },
1122
1132
  headers={
@@ -1245,7 +1255,7 @@ class Vellum:
1245
1255
  "index_name": index_name,
1246
1256
  "query": query,
1247
1257
  "options": convert_and_respect_annotation_metadata(
1248
- object_=options, annotation=SearchRequestOptionsRequest, direction="write"
1258
+ object_=options, annotation=typing.Optional[SearchRequestOptionsRequest], direction="write"
1249
1259
  ),
1250
1260
  "document_index": document_index,
1251
1261
  },
@@ -1505,7 +1515,9 @@ class AsyncVellum:
1505
1515
  follow_redirects: typing.Optional[bool] = True,
1506
1516
  httpx_client: typing.Optional[httpx.AsyncClient] = None,
1507
1517
  ):
1508
- _defaulted_timeout = timeout if timeout is not None else None if httpx_client is None else None
1518
+ _defaulted_timeout = (
1519
+ timeout if timeout is not None else None if httpx_client is None else httpx_client.timeout.read
1520
+ )
1509
1521
  self._client_wrapper = AsyncClientWrapper(
1510
1522
  environment=environment,
1511
1523
  api_key=api_key,
@@ -1594,13 +1606,13 @@ class AsyncVellum:
1594
1606
  "url": url,
1595
1607
  "method": method,
1596
1608
  "body": convert_and_respect_annotation_metadata(
1597
- object_=body, annotation=ExecuteApiRequestBody, direction="write"
1609
+ object_=body, annotation=typing.Optional[ExecuteApiRequestBody], direction="write"
1598
1610
  ),
1599
1611
  "headers": convert_and_respect_annotation_metadata(
1600
1612
  object_=headers, annotation=typing.Dict[str, ExecuteApiRequestHeadersValue], direction="write"
1601
1613
  ),
1602
1614
  "bearer_token": convert_and_respect_annotation_metadata(
1603
- object_=bearer_token, annotation=ExecuteApiRequestBearerToken, direction="write"
1615
+ object_=bearer_token, annotation=typing.Optional[ExecuteApiRequestBearerToken], direction="write"
1604
1616
  ),
1605
1617
  },
1606
1618
  headers={
@@ -1823,10 +1835,14 @@ class AsyncVellum:
1823
1835
  "release_tag": release_tag,
1824
1836
  "external_id": external_id,
1825
1837
  "expand_meta": convert_and_respect_annotation_metadata(
1826
- object_=expand_meta, annotation=PromptDeploymentExpandMetaRequest, direction="write"
1838
+ object_=expand_meta,
1839
+ annotation=typing.Optional[PromptDeploymentExpandMetaRequest],
1840
+ direction="write",
1827
1841
  ),
1828
1842
  "raw_overrides": convert_and_respect_annotation_metadata(
1829
- object_=raw_overrides, annotation=RawPromptExecutionOverridesRequest, direction="write"
1843
+ object_=raw_overrides,
1844
+ annotation=typing.Optional[RawPromptExecutionOverridesRequest],
1845
+ direction="write",
1830
1846
  ),
1831
1847
  "expand_raw": expand_raw,
1832
1848
  "metadata": metadata,
@@ -1984,10 +2000,14 @@ class AsyncVellum:
1984
2000
  "release_tag": release_tag,
1985
2001
  "external_id": external_id,
1986
2002
  "expand_meta": convert_and_respect_annotation_metadata(
1987
- object_=expand_meta, annotation=PromptDeploymentExpandMetaRequest, direction="write"
2003
+ object_=expand_meta,
2004
+ annotation=typing.Optional[PromptDeploymentExpandMetaRequest],
2005
+ direction="write",
1988
2006
  ),
1989
2007
  "raw_overrides": convert_and_respect_annotation_metadata(
1990
- object_=raw_overrides, annotation=RawPromptExecutionOverridesRequest, direction="write"
2008
+ object_=raw_overrides,
2009
+ annotation=typing.Optional[RawPromptExecutionOverridesRequest],
2010
+ direction="write",
1991
2011
  ),
1992
2012
  "expand_raw": expand_raw,
1993
2013
  "metadata": metadata,
@@ -2139,7 +2159,7 @@ class AsyncVellum:
2139
2159
  object_=inputs, annotation=typing.Sequence[WorkflowRequestInputRequest], direction="write"
2140
2160
  ),
2141
2161
  "expand_meta": convert_and_respect_annotation_metadata(
2142
- object_=expand_meta, annotation=WorkflowExpandMetaRequest, direction="write"
2162
+ object_=expand_meta, annotation=typing.Optional[WorkflowExpandMetaRequest], direction="write"
2143
2163
  ),
2144
2164
  "workflow_deployment_id": workflow_deployment_id,
2145
2165
  "workflow_deployment_name": workflow_deployment_name,
@@ -2282,7 +2302,7 @@ class AsyncVellum:
2282
2302
  object_=inputs, annotation=typing.Sequence[WorkflowRequestInputRequest], direction="write"
2283
2303
  ),
2284
2304
  "expand_meta": convert_and_respect_annotation_metadata(
2285
- object_=expand_meta, annotation=WorkflowExpandMetaRequest, direction="write"
2305
+ object_=expand_meta, annotation=typing.Optional[WorkflowExpandMetaRequest], direction="write"
2286
2306
  ),
2287
2307
  "workflow_deployment_id": workflow_deployment_id,
2288
2308
  "workflow_deployment_name": workflow_deployment_name,
@@ -2420,7 +2440,7 @@ class AsyncVellum:
2420
2440
  object_=requests, annotation=typing.Sequence[GenerateRequest], direction="write"
2421
2441
  ),
2422
2442
  "options": convert_and_respect_annotation_metadata(
2423
- object_=options, annotation=GenerateOptionsRequest, direction="write"
2443
+ object_=options, annotation=typing.Optional[GenerateOptionsRequest], direction="write"
2424
2444
  ),
2425
2445
  },
2426
2446
  headers={
@@ -2556,7 +2576,7 @@ class AsyncVellum:
2556
2576
  object_=requests, annotation=typing.Sequence[GenerateRequest], direction="write"
2557
2577
  ),
2558
2578
  "options": convert_and_respect_annotation_metadata(
2559
- object_=options, annotation=GenerateOptionsRequest, direction="write"
2579
+ object_=options, annotation=typing.Optional[GenerateOptionsRequest], direction="write"
2560
2580
  ),
2561
2581
  },
2562
2582
  headers={
@@ -2693,7 +2713,7 @@ class AsyncVellum:
2693
2713
  "index_name": index_name,
2694
2714
  "query": query,
2695
2715
  "options": convert_and_respect_annotation_metadata(
2696
- object_=options, annotation=SearchRequestOptionsRequest, direction="write"
2716
+ object_=options, annotation=typing.Optional[SearchRequestOptionsRequest], direction="write"
2697
2717
  ),
2698
2718
  "document_index": document_index,
2699
2719
  },
@@ -16,9 +16,10 @@ class BaseClientWrapper:
16
16
 
17
17
  def get_headers(self) -> typing.Dict[str, str]:
18
18
  headers: typing.Dict[str, str] = {
19
+ "User-Agent": "vellum-ai/0.14.83",
19
20
  "X-Fern-Language": "Python",
20
21
  "X-Fern-SDK-Name": "vellum-ai",
21
- "X-Fern-SDK-Version": "0.14.81",
22
+ "X-Fern-SDK-Version": "0.14.83",
22
23
  }
23
24
  headers["X-API-KEY"] = self.api_key
24
25
  return headers
@@ -2,7 +2,6 @@
2
2
 
3
3
  import asyncio
4
4
  import email.utils
5
- import json
6
5
  import re
7
6
  import time
8
7
  import typing
@@ -11,7 +10,6 @@ from contextlib import asynccontextmanager, contextmanager
11
10
  from random import random
12
11
 
13
12
  import httpx
14
-
15
13
  from .file import File, convert_file_dict_to_httpx_tuples
16
14
  from .jsonable_encoder import jsonable_encoder
17
15
  from .query_encoder import encode_query
@@ -17,7 +17,6 @@ from types import GeneratorType
17
17
  from typing import Any, Callable, Dict, List, Optional, Set, Union
18
18
 
19
19
  import pydantic
20
-
21
20
  from .datetime_utils import serialize_datetime
22
21
  from .pydantic_utilities import (
23
22
  IS_PYDANTIC_V2,
@@ -5,11 +5,9 @@ import datetime as dt
5
5
  import typing
6
6
  from collections import defaultdict
7
7
 
8
- import typing_extensions
9
-
10
8
  import pydantic
11
9
  import logging
12
-
10
+ import typing_extensions
13
11
  from .datetime_utils import serialize_datetime
14
12
  from .serialization import convert_and_respect_annotation_metadata
15
13
 
@@ -4,9 +4,8 @@ import collections
4
4
  import inspect
5
5
  import typing
6
6
 
7
- import typing_extensions
8
-
9
7
  import pydantic
8
+ import typing_extensions
10
9
 
11
10
 
12
11
  class FieldMetadata:
@@ -3497,6 +3497,14 @@ The current status of the document
3497
3497
  <dl>
3498
3498
  <dd>
3499
3499
 
3500
+ **keywords:** `typing.Optional[typing.Sequence[str]]` — A list of keywords that'll be associated with the document. Used as part of keyword search.
3501
+
3502
+ </dd>
3503
+ </dl>
3504
+
3505
+ <dl>
3506
+ <dd>
3507
+
3500
3508
  **metadata:** `typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]` — A JSON object containing any metadata associated with the document that you'd like to filter upon later.
3501
3509
 
3502
3510
  </dd>
@@ -121,16 +121,18 @@ class AdHocClient:
121
121
  object_=parameters, annotation=PromptParameters, direction="write"
122
122
  ),
123
123
  "settings": convert_and_respect_annotation_metadata(
124
- object_=settings, annotation=PromptSettings, direction="write"
124
+ object_=settings, annotation=typing.Optional[PromptSettings], direction="write"
125
125
  ),
126
126
  "blocks": convert_and_respect_annotation_metadata(
127
127
  object_=blocks, annotation=typing.Sequence[PromptBlock], direction="write"
128
128
  ),
129
129
  "functions": convert_and_respect_annotation_metadata(
130
- object_=functions, annotation=typing.Sequence[FunctionDefinition], direction="write"
130
+ object_=functions,
131
+ annotation=typing.Optional[typing.Sequence[FunctionDefinition]],
132
+ direction="write",
131
133
  ),
132
134
  "expand_meta": convert_and_respect_annotation_metadata(
133
- object_=expand_meta, annotation=AdHocExpandMeta, direction="write"
135
+ object_=expand_meta, annotation=typing.Optional[AdHocExpandMeta], direction="write"
134
136
  ),
135
137
  },
136
138
  headers={
@@ -277,16 +279,18 @@ class AdHocClient:
277
279
  object_=parameters, annotation=PromptParameters, direction="write"
278
280
  ),
279
281
  "settings": convert_and_respect_annotation_metadata(
280
- object_=settings, annotation=PromptSettings, direction="write"
282
+ object_=settings, annotation=typing.Optional[PromptSettings], direction="write"
281
283
  ),
282
284
  "blocks": convert_and_respect_annotation_metadata(
283
285
  object_=blocks, annotation=typing.Sequence[PromptBlock], direction="write"
284
286
  ),
285
287
  "functions": convert_and_respect_annotation_metadata(
286
- object_=functions, annotation=typing.Sequence[FunctionDefinition], direction="write"
288
+ object_=functions,
289
+ annotation=typing.Optional[typing.Sequence[FunctionDefinition]],
290
+ direction="write",
287
291
  ),
288
292
  "expand_meta": convert_and_respect_annotation_metadata(
289
- object_=expand_meta, annotation=AdHocExpandMeta, direction="write"
293
+ object_=expand_meta, annotation=typing.Optional[AdHocExpandMeta], direction="write"
290
294
  ),
291
295
  },
292
296
  headers={
@@ -452,16 +456,18 @@ class AsyncAdHocClient:
452
456
  object_=parameters, annotation=PromptParameters, direction="write"
453
457
  ),
454
458
  "settings": convert_and_respect_annotation_metadata(
455
- object_=settings, annotation=PromptSettings, direction="write"
459
+ object_=settings, annotation=typing.Optional[PromptSettings], direction="write"
456
460
  ),
457
461
  "blocks": convert_and_respect_annotation_metadata(
458
462
  object_=blocks, annotation=typing.Sequence[PromptBlock], direction="write"
459
463
  ),
460
464
  "functions": convert_and_respect_annotation_metadata(
461
- object_=functions, annotation=typing.Sequence[FunctionDefinition], direction="write"
465
+ object_=functions,
466
+ annotation=typing.Optional[typing.Sequence[FunctionDefinition]],
467
+ direction="write",
462
468
  ),
463
469
  "expand_meta": convert_and_respect_annotation_metadata(
464
- object_=expand_meta, annotation=AdHocExpandMeta, direction="write"
470
+ object_=expand_meta, annotation=typing.Optional[AdHocExpandMeta], direction="write"
465
471
  ),
466
472
  },
467
473
  headers={
@@ -616,16 +622,18 @@ class AsyncAdHocClient:
616
622
  object_=parameters, annotation=PromptParameters, direction="write"
617
623
  ),
618
624
  "settings": convert_and_respect_annotation_metadata(
619
- object_=settings, annotation=PromptSettings, direction="write"
625
+ object_=settings, annotation=typing.Optional[PromptSettings], direction="write"
620
626
  ),
621
627
  "blocks": convert_and_respect_annotation_metadata(
622
628
  object_=blocks, annotation=typing.Sequence[PromptBlock], direction="write"
623
629
  ),
624
630
  "functions": convert_and_respect_annotation_metadata(
625
- object_=functions, annotation=typing.Sequence[FunctionDefinition], direction="write"
631
+ object_=functions,
632
+ annotation=typing.Optional[typing.Sequence[FunctionDefinition]],
633
+ direction="write",
626
634
  ),
627
635
  "expand_meta": convert_and_respect_annotation_metadata(
628
- object_=expand_meta, annotation=AdHocExpandMeta, direction="write"
636
+ object_=expand_meta, annotation=typing.Optional[AdHocExpandMeta], direction="write"
629
637
  ),
630
638
  },
631
639
  headers={
@@ -530,7 +530,9 @@ class DeploymentsClient:
530
530
  ),
531
531
  "release_tag": release_tag,
532
532
  "expand_meta": convert_and_respect_annotation_metadata(
533
- object_=expand_meta, annotation=CompilePromptDeploymentExpandMetaRequest, direction="write"
533
+ object_=expand_meta,
534
+ annotation=typing.Optional[CompilePromptDeploymentExpandMetaRequest],
535
+ direction="write",
534
536
  ),
535
537
  },
536
538
  headers={
@@ -1159,7 +1161,9 @@ class AsyncDeploymentsClient:
1159
1161
  ),
1160
1162
  "release_tag": release_tag,
1161
1163
  "expand_meta": convert_and_respect_annotation_metadata(
1162
- object_=expand_meta, annotation=CompilePromptDeploymentExpandMetaRequest, direction="write"
1164
+ object_=expand_meta,
1165
+ annotation=typing.Optional[CompilePromptDeploymentExpandMetaRequest],
1166
+ direction="write",
1163
1167
  ),
1164
1168
  },
1165
1169
  headers={
@@ -192,6 +192,7 @@ class DocumentsClient:
192
192
  *,
193
193
  label: typing.Optional[str] = OMIT,
194
194
  status: typing.Optional[DocumentStatus] = OMIT,
195
+ keywords: typing.Optional[typing.Sequence[str]] = OMIT,
195
196
  metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
196
197
  request_options: typing.Optional[RequestOptions] = None,
197
198
  ) -> DocumentRead:
@@ -210,6 +211,9 @@ class DocumentsClient:
210
211
 
211
212
  * `ACTIVE` - Active
212
213
 
214
+ keywords : typing.Optional[typing.Sequence[str]]
215
+ A list of keywords that'll be associated with the document. Used as part of keyword search.
216
+
213
217
  metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
214
218
  A JSON object containing any metadata associated with the document that you'd like to filter upon later.
215
219
 
@@ -239,6 +243,7 @@ class DocumentsClient:
239
243
  json={
240
244
  "label": label,
241
245
  "status": status,
246
+ "keywords": keywords,
242
247
  "metadata": metadata,
243
248
  },
244
249
  headers={
@@ -573,6 +578,7 @@ class AsyncDocumentsClient:
573
578
  *,
574
579
  label: typing.Optional[str] = OMIT,
575
580
  status: typing.Optional[DocumentStatus] = OMIT,
581
+ keywords: typing.Optional[typing.Sequence[str]] = OMIT,
576
582
  metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
577
583
  request_options: typing.Optional[RequestOptions] = None,
578
584
  ) -> DocumentRead:
@@ -591,6 +597,9 @@ class AsyncDocumentsClient:
591
597
 
592
598
  * `ACTIVE` - Active
593
599
 
600
+ keywords : typing.Optional[typing.Sequence[str]]
601
+ A list of keywords that'll be associated with the document. Used as part of keyword search.
602
+
594
603
  metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
595
604
  A JSON object containing any metadata associated with the document that you'd like to filter upon later.
596
605
 
@@ -628,6 +637,7 @@ class AsyncDocumentsClient:
628
637
  json={
629
638
  "label": label,
630
639
  "status": status,
640
+ "keywords": keywords,
631
641
  "metadata": metadata,
632
642
  },
633
643
  headers={
@@ -125,6 +125,7 @@ from .execution_json_vellum_value import ExecutionJsonVellumValue
125
125
  from .execution_number_vellum_value import ExecutionNumberVellumValue
126
126
  from .execution_search_results_vellum_value import ExecutionSearchResultsVellumValue
127
127
  from .execution_string_vellum_value import ExecutionStringVellumValue
128
+ from .execution_thinking_vellum_value import ExecutionThinkingVellumValue
128
129
  from .execution_vellum_value import ExecutionVellumValue
129
130
  from .external_input_descriptor import ExternalInputDescriptor
130
131
  from .external_test_case_execution import ExternalTestCaseExecution
@@ -173,6 +174,8 @@ from .generate_stream_result import GenerateStreamResult
173
174
  from .generate_stream_result_data import GenerateStreamResultData
174
175
  from .google_vertex_ai_vectorizer_config import GoogleVertexAiVectorizerConfig
175
176
  from .google_vertex_ai_vectorizer_config_request import GoogleVertexAiVectorizerConfigRequest
177
+ from .google_vertex_ai_vectorizer_gemini_embedding_001 import GoogleVertexAiVectorizerGeminiEmbedding001
178
+ from .google_vertex_ai_vectorizer_gemini_embedding_001_request import GoogleVertexAiVectorizerGeminiEmbedding001Request
176
179
  from .google_vertex_ai_vectorizer_text_embedding_004 import GoogleVertexAiVectorizerTextEmbedding004
177
180
  from .google_vertex_ai_vectorizer_text_embedding_004_request import GoogleVertexAiVectorizerTextEmbedding004Request
178
181
  from .google_vertex_ai_vectorizer_text_multilingual_embedding_002 import (
@@ -511,6 +514,8 @@ from .test_suite_test_case_replace_bulk_operation_request import TestSuiteTestCa
511
514
  from .test_suite_test_case_replaced_bulk_result import TestSuiteTestCaseReplacedBulkResult
512
515
  from .test_suite_test_case_replaced_bulk_result_data import TestSuiteTestCaseReplacedBulkResultData
513
516
  from .test_suite_test_case_upsert_bulk_operation_request import TestSuiteTestCaseUpsertBulkOperationRequest
517
+ from .thinking_vellum_value import ThinkingVellumValue
518
+ from .thinking_vellum_value_request import ThinkingVellumValueRequest
514
519
  from .token_overlapping_window_chunker_config import TokenOverlappingWindowChunkerConfig
515
520
  from .token_overlapping_window_chunker_config_request import TokenOverlappingWindowChunkerConfigRequest
516
521
  from .token_overlapping_window_chunking import TokenOverlappingWindowChunking
@@ -741,6 +746,7 @@ __all__ = [
741
746
  "ExecutionNumberVellumValue",
742
747
  "ExecutionSearchResultsVellumValue",
743
748
  "ExecutionStringVellumValue",
749
+ "ExecutionThinkingVellumValue",
744
750
  "ExecutionVellumValue",
745
751
  "ExternalInputDescriptor",
746
752
  "ExternalTestCaseExecution",
@@ -789,6 +795,8 @@ __all__ = [
789
795
  "GenerateStreamResultData",
790
796
  "GoogleVertexAiVectorizerConfig",
791
797
  "GoogleVertexAiVectorizerConfigRequest",
798
+ "GoogleVertexAiVectorizerGeminiEmbedding001",
799
+ "GoogleVertexAiVectorizerGeminiEmbedding001Request",
792
800
  "GoogleVertexAiVectorizerTextEmbedding004",
793
801
  "GoogleVertexAiVectorizerTextEmbedding004Request",
794
802
  "GoogleVertexAiVectorizerTextMultilingualEmbedding002",
@@ -1107,6 +1115,8 @@ __all__ = [
1107
1115
  "TestSuiteTestCaseReplacedBulkResult",
1108
1116
  "TestSuiteTestCaseReplacedBulkResultData",
1109
1117
  "TestSuiteTestCaseUpsertBulkOperationRequest",
1118
+ "ThinkingVellumValue",
1119
+ "ThinkingVellumValueRequest",
1110
1120
  "TokenOverlappingWindowChunkerConfig",
1111
1121
  "TokenOverlappingWindowChunkerConfigRequest",
1112
1122
  "TokenOverlappingWindowChunking",
@@ -9,6 +9,7 @@ import pydantic
9
9
 
10
10
  class ContainerImageBuildConfig(UniversalBaseModel):
11
11
  packages: typing.List[CodeExecutionPackage]
12
+ user_script: typing.Optional[str] = None
12
13
 
13
14
  if IS_PYDANTIC_V2:
14
15
  model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True) # type: ignore # Pydantic v2
@@ -31,6 +31,11 @@ class DocumentRead(UniversalBaseModel):
31
31
  * `ACTIVE` - Active
32
32
  """
33
33
 
34
+ keywords: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
35
+ """
36
+ A list of keywords that'll be associated with the document. Used as part of keyword search.
37
+ """
38
+
34
39
  original_file_url: typing.Optional[str] = None
35
40
  document_to_document_indexes: typing.List[DocumentDocumentToDocumentIndex]
36
41
  metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
@@ -0,0 +1,31 @@
1
+ # This file was auto-generated by Fern from our API Definition.
2
+
3
+ from ..core.pydantic_utilities import UniversalBaseModel
4
+ import pydantic
5
+ import typing
6
+ from .string_vellum_value import StringVellumValue
7
+ from ..core.pydantic_utilities import IS_PYDANTIC_V2
8
+
9
+
10
+ class ExecutionThinkingVellumValue(UniversalBaseModel):
11
+ """
12
+ A value representing Thinking mode output.
13
+ """
14
+
15
+ id: str = pydantic.Field()
16
+ """
17
+ The variable's uniquely identifying internal id.
18
+ """
19
+
20
+ name: str
21
+ type: typing.Literal["THINKING"] = "THINKING"
22
+ value: StringVellumValue
23
+
24
+ if IS_PYDANTIC_V2:
25
+ model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True) # type: ignore # Pydantic v2
26
+ else:
27
+
28
+ class Config:
29
+ frozen = True
30
+ smart_union = True
31
+ extra = pydantic.Extra.allow
@@ -9,6 +9,7 @@ from .execution_search_results_vellum_value import ExecutionSearchResultsVellumV
9
9
  from .execution_error_vellum_value import ExecutionErrorVellumValue
10
10
  from .execution_array_vellum_value import ExecutionArrayVellumValue
11
11
  from .execution_function_call_vellum_value import ExecutionFunctionCallVellumValue
12
+ from .execution_thinking_vellum_value import ExecutionThinkingVellumValue
12
13
 
13
14
  ExecutionVellumValue = typing.Union[
14
15
  ExecutionStringVellumValue,
@@ -19,4 +20,5 @@ ExecutionVellumValue = typing.Union[
19
20
  ExecutionErrorVellumValue,
20
21
  ExecutionArrayVellumValue,
21
22
  ExecutionFunctionCallVellumValue,
23
+ ExecutionThinkingVellumValue,
22
24
  ]