vellum-ai 1.5.4__py3-none-any.whl → 1.5.6__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 (66) hide show
  1. vellum/__init__.py +36 -0
  2. vellum/client/__init__.py +6 -0
  3. vellum/client/core/client_wrapper.py +2 -2
  4. vellum/client/reference.md +497 -11
  5. vellum/client/resources/__init__.py +4 -0
  6. vellum/client/resources/integration_auth_configs/__init__.py +4 -0
  7. vellum/client/resources/integration_auth_configs/client.py +186 -0
  8. vellum/client/resources/integration_auth_configs/raw_client.py +171 -0
  9. vellum/client/resources/integration_providers/__init__.py +4 -0
  10. vellum/client/resources/integration_providers/client.py +279 -0
  11. vellum/client/resources/integration_providers/raw_client.py +255 -0
  12. vellum/client/resources/integrations/client.py +226 -28
  13. vellum/client/resources/integrations/raw_client.py +257 -20
  14. vellum/client/resources/test_suite_runs/client.py +6 -6
  15. vellum/client/resources/test_suite_runs/raw_client.py +6 -6
  16. vellum/client/types/__init__.py +32 -0
  17. vellum/client/types/components_schemas_composio_integration_exec_config.py +5 -0
  18. vellum/client/types/components_schemas_slim_composio_tool_definition.py +5 -0
  19. vellum/client/types/composio_integration_exec_config.py +20 -0
  20. vellum/client/types/composio_tool_definition.py +4 -1
  21. vellum/client/types/integration_auth_config_integration.py +23 -0
  22. vellum/client/types/integration_auth_config_integration_credential.py +19 -0
  23. vellum/client/types/integration_credential_access_type.py +5 -0
  24. vellum/client/types/integration_name.py +20 -0
  25. vellum/client/types/integration_provider.py +5 -0
  26. vellum/client/types/integration_read.py +30 -0
  27. vellum/client/types/paginated_slim_integration_auth_config_read_list.py +23 -0
  28. vellum/client/types/paginated_slim_integration_read_list.py +23 -0
  29. vellum/client/types/paginated_slim_tool_definition_list.py +23 -0
  30. vellum/client/types/slim_composio_tool_definition.py +23 -0
  31. vellum/client/types/slim_integration_auth_config_read.py +29 -0
  32. vellum/client/types/slim_integration_read.py +25 -0
  33. vellum/client/types/tool_definition_integration.py +23 -0
  34. vellum/resources/integration_auth_configs/__init__.py +3 -0
  35. vellum/resources/integration_auth_configs/client.py +3 -0
  36. vellum/resources/integration_auth_configs/raw_client.py +3 -0
  37. vellum/resources/integration_providers/__init__.py +3 -0
  38. vellum/resources/integration_providers/client.py +3 -0
  39. vellum/resources/integration_providers/raw_client.py +3 -0
  40. vellum/types/components_schemas_composio_integration_exec_config.py +3 -0
  41. vellum/types/components_schemas_slim_composio_tool_definition.py +3 -0
  42. vellum/types/composio_integration_exec_config.py +3 -0
  43. vellum/types/integration_auth_config_integration.py +3 -0
  44. vellum/types/integration_auth_config_integration_credential.py +3 -0
  45. vellum/types/integration_credential_access_type.py +3 -0
  46. vellum/types/integration_name.py +3 -0
  47. vellum/types/integration_provider.py +3 -0
  48. vellum/types/integration_read.py +3 -0
  49. vellum/types/paginated_slim_integration_auth_config_read_list.py +3 -0
  50. vellum/types/paginated_slim_integration_read_list.py +3 -0
  51. vellum/types/paginated_slim_tool_definition_list.py +3 -0
  52. vellum/types/slim_composio_tool_definition.py +3 -0
  53. vellum/types/slim_integration_auth_config_read.py +3 -0
  54. vellum/types/slim_integration_read.py +3 -0
  55. vellum/types/tool_definition_integration.py +3 -0
  56. vellum/workflows/integrations/tests/test_vellum_integration_service.py +26 -19
  57. vellum/workflows/integrations/vellum_integration_service.py +5 -5
  58. vellum/workflows/nodes/displayable/tool_calling_node/node.py +2 -4
  59. vellum/workflows/types/definition.py +1 -1
  60. vellum/workflows/types/tests/test_definition.py +40 -2
  61. {vellum_ai-1.5.4.dist-info → vellum_ai-1.5.6.dist-info}/METADATA +1 -1
  62. {vellum_ai-1.5.4.dist-info → vellum_ai-1.5.6.dist-info}/RECORD +66 -21
  63. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_vellum_integration_serialization.py +68 -0
  64. {vellum_ai-1.5.4.dist-info → vellum_ai-1.5.6.dist-info}/LICENSE +0 -0
  65. {vellum_ai-1.5.4.dist-info → vellum_ai-1.5.6.dist-info}/WHEEL +0 -0
  66. {vellum_ai-1.5.4.dist-info → vellum_ai-1.5.6.dist-info}/entry_points.txt +0 -0
vellum/__init__.py CHANGED
@@ -70,12 +70,15 @@ from .client.types import (
70
70
  CompilePromptMeta,
71
71
  ComponentsSchemasComposioExecuteToolRequest,
72
72
  ComponentsSchemasComposioExecuteToolResponse,
73
+ ComponentsSchemasComposioIntegrationExecConfig,
73
74
  ComponentsSchemasComposioToolDefinition,
74
75
  ComponentsSchemasPdfSearchResultMetaSource,
75
76
  ComponentsSchemasPdfSearchResultMetaSourceRequest,
76
77
  ComponentsSchemasPromptVersionBuildConfigSandbox,
78
+ ComponentsSchemasSlimComposioToolDefinition,
77
79
  ComposioExecuteToolRequest,
78
80
  ComposioExecuteToolResponse,
81
+ ComposioIntegrationExecConfig,
79
82
  ComposioToolDefinition,
80
83
  ConditionCombinator,
81
84
  ConditionalNodeResult,
@@ -219,6 +222,12 @@ from .client.types import (
219
222
  InitiatedWorkflowNodeResultEvent,
220
223
  InstructorVectorizerConfig,
221
224
  InstructorVectorizerConfigRequest,
225
+ IntegrationAuthConfigIntegration,
226
+ IntegrationAuthConfigIntegrationCredential,
227
+ IntegrationCredentialAccessType,
228
+ IntegrationName,
229
+ IntegrationProvider,
230
+ IntegrationRead,
222
231
  InvokedPort,
223
232
  IterationStateEnum,
224
233
  JinjaPromptBlock,
@@ -338,6 +347,9 @@ from .client.types import (
338
347
  PaginatedFolderEntityList,
339
348
  PaginatedSlimDeploymentReadList,
340
349
  PaginatedSlimDocumentList,
350
+ PaginatedSlimIntegrationAuthConfigReadList,
351
+ PaginatedSlimIntegrationReadList,
352
+ PaginatedSlimToolDefinitionList,
341
353
  PaginatedSlimWorkflowDeploymentList,
342
354
  PaginatedTestSuiteRunExecutionList,
343
355
  PaginatedTestSuiteTestCaseList,
@@ -428,9 +440,12 @@ from .client.types import (
428
440
  SentenceChunkerConfigRequest,
429
441
  SentenceChunking,
430
442
  SentenceChunkingRequest,
443
+ SlimComposioToolDefinition,
431
444
  SlimDeploymentRead,
432
445
  SlimDocument,
433
446
  SlimDocumentDocumentToDocumentIndex,
447
+ SlimIntegrationAuthConfigRead,
448
+ SlimIntegrationRead,
434
449
  SlimReleaseReview,
435
450
  SlimWorkflowDeployment,
436
451
  SlimWorkflowExecutionRead,
@@ -554,6 +569,7 @@ from .client.types import (
554
569
  TokenOverlappingWindowChunkerConfigRequest,
555
570
  TokenOverlappingWindowChunking,
556
571
  TokenOverlappingWindowChunkingRequest,
572
+ ToolDefinitionIntegration,
557
573
  UnitEnum,
558
574
  UploadDocumentResponse,
559
575
  UpsertTestSuiteTestCaseRequest,
@@ -702,6 +718,8 @@ from .resources import (
702
718
  documents,
703
719
  events,
704
720
  folder_entities,
721
+ integration_auth_configs,
722
+ integration_providers,
705
723
  integrations,
706
724
  metric_definitions,
707
725
  ml_models,
@@ -788,12 +806,15 @@ __all__ = [
788
806
  "CompilePromptMeta",
789
807
  "ComponentsSchemasComposioExecuteToolRequest",
790
808
  "ComponentsSchemasComposioExecuteToolResponse",
809
+ "ComponentsSchemasComposioIntegrationExecConfig",
791
810
  "ComponentsSchemasComposioToolDefinition",
792
811
  "ComponentsSchemasPdfSearchResultMetaSource",
793
812
  "ComponentsSchemasPdfSearchResultMetaSourceRequest",
794
813
  "ComponentsSchemasPromptVersionBuildConfigSandbox",
814
+ "ComponentsSchemasSlimComposioToolDefinition",
795
815
  "ComposioExecuteToolRequest",
796
816
  "ComposioExecuteToolResponse",
817
+ "ComposioIntegrationExecConfig",
797
818
  "ComposioToolDefinition",
798
819
  "ConditionCombinator",
799
820
  "ConditionalNodeResult",
@@ -941,6 +962,12 @@ __all__ = [
941
962
  "InitiatedWorkflowNodeResultEvent",
942
963
  "InstructorVectorizerConfig",
943
964
  "InstructorVectorizerConfigRequest",
965
+ "IntegrationAuthConfigIntegration",
966
+ "IntegrationAuthConfigIntegrationCredential",
967
+ "IntegrationCredentialAccessType",
968
+ "IntegrationName",
969
+ "IntegrationProvider",
970
+ "IntegrationRead",
944
971
  "InternalServerError",
945
972
  "InvokedPort",
946
973
  "IterationStateEnum",
@@ -1065,6 +1092,9 @@ __all__ = [
1065
1092
  "PaginatedFolderEntityList",
1066
1093
  "PaginatedSlimDeploymentReadList",
1067
1094
  "PaginatedSlimDocumentList",
1095
+ "PaginatedSlimIntegrationAuthConfigReadList",
1096
+ "PaginatedSlimIntegrationReadList",
1097
+ "PaginatedSlimToolDefinitionList",
1068
1098
  "PaginatedSlimWorkflowDeploymentList",
1069
1099
  "PaginatedTestSuiteRunExecutionList",
1070
1100
  "PaginatedTestSuiteTestCaseList",
@@ -1155,9 +1185,12 @@ __all__ = [
1155
1185
  "SentenceChunkerConfigRequest",
1156
1186
  "SentenceChunking",
1157
1187
  "SentenceChunkingRequest",
1188
+ "SlimComposioToolDefinition",
1158
1189
  "SlimDeploymentRead",
1159
1190
  "SlimDocument",
1160
1191
  "SlimDocumentDocumentToDocumentIndex",
1192
+ "SlimIntegrationAuthConfigRead",
1193
+ "SlimIntegrationRead",
1161
1194
  "SlimReleaseReview",
1162
1195
  "SlimWorkflowDeployment",
1163
1196
  "SlimWorkflowExecutionRead",
@@ -1282,6 +1315,7 @@ __all__ = [
1282
1315
  "TokenOverlappingWindowChunking",
1283
1316
  "TokenOverlappingWindowChunkingRequest",
1284
1317
  "TooManyRequestsError",
1318
+ "ToolDefinitionIntegration",
1285
1319
  "UnauthorizedError",
1286
1320
  "UnitEnum",
1287
1321
  "UploadDocumentResponse",
@@ -1418,6 +1452,8 @@ __all__ = [
1418
1452
  "documents",
1419
1453
  "events",
1420
1454
  "folder_entities",
1455
+ "integration_auth_configs",
1456
+ "integration_providers",
1421
1457
  "integrations",
1422
1458
  "metric_definitions",
1423
1459
  "ml_models",
vellum/client/__init__.py CHANGED
@@ -16,6 +16,8 @@ from .resources.document_indexes.client import AsyncDocumentIndexesClient, Docum
16
16
  from .resources.documents.client import AsyncDocumentsClient, DocumentsClient
17
17
  from .resources.events.client import AsyncEventsClient, EventsClient
18
18
  from .resources.folder_entities.client import AsyncFolderEntitiesClient, FolderEntitiesClient
19
+ from .resources.integration_auth_configs.client import AsyncIntegrationAuthConfigsClient, IntegrationAuthConfigsClient
20
+ from .resources.integration_providers.client import AsyncIntegrationProvidersClient, IntegrationProvidersClient
19
21
  from .resources.integrations.client import AsyncIntegrationsClient, IntegrationsClient
20
22
  from .resources.metric_definitions.client import AsyncMetricDefinitionsClient, MetricDefinitionsClient
21
23
  from .resources.ml_models.client import AsyncMlModelsClient, MlModelsClient
@@ -138,6 +140,8 @@ class Vellum:
138
140
  self.document_indexes = DocumentIndexesClient(client_wrapper=self._client_wrapper)
139
141
  self.documents = DocumentsClient(client_wrapper=self._client_wrapper)
140
142
  self.folder_entities = FolderEntitiesClient(client_wrapper=self._client_wrapper)
143
+ self.integration_auth_configs = IntegrationAuthConfigsClient(client_wrapper=self._client_wrapper)
144
+ self.integration_providers = IntegrationProvidersClient(client_wrapper=self._client_wrapper)
141
145
  self.metric_definitions = MetricDefinitionsClient(client_wrapper=self._client_wrapper)
142
146
  self.ml_models = MlModelsClient(client_wrapper=self._client_wrapper)
143
147
  self.organizations = OrganizationsClient(client_wrapper=self._client_wrapper)
@@ -1013,6 +1017,8 @@ class AsyncVellum:
1013
1017
  self.document_indexes = AsyncDocumentIndexesClient(client_wrapper=self._client_wrapper)
1014
1018
  self.documents = AsyncDocumentsClient(client_wrapper=self._client_wrapper)
1015
1019
  self.folder_entities = AsyncFolderEntitiesClient(client_wrapper=self._client_wrapper)
1020
+ self.integration_auth_configs = AsyncIntegrationAuthConfigsClient(client_wrapper=self._client_wrapper)
1021
+ self.integration_providers = AsyncIntegrationProvidersClient(client_wrapper=self._client_wrapper)
1016
1022
  self.metric_definitions = AsyncMetricDefinitionsClient(client_wrapper=self._client_wrapper)
1017
1023
  self.ml_models = AsyncMlModelsClient(client_wrapper=self._client_wrapper)
1018
1024
  self.organizations = AsyncOrganizationsClient(client_wrapper=self._client_wrapper)
@@ -27,10 +27,10 @@ class BaseClientWrapper:
27
27
 
28
28
  def get_headers(self) -> typing.Dict[str, str]:
29
29
  headers: typing.Dict[str, str] = {
30
- "User-Agent": "vellum-ai/1.5.4",
30
+ "User-Agent": "vellum-ai/1.5.6",
31
31
  "X-Fern-Language": "Python",
32
32
  "X-Fern-SDK-Name": "vellum-ai",
33
- "X-Fern-SDK-Version": "1.5.4",
33
+ "X-Fern-SDK-Version": "1.5.6",
34
34
  **(self.get_custom_headers() or {}),
35
35
  }
36
36
  if self._api_version is not None: