vellum-ai 1.5.4__py3-none-any.whl → 1.5.6__py3-none-any.whl

vellum/client/resources/integrations/raw_client.py

@@ -10,8 +10,11 @@ from ...core.jsonable_encoder import jsonable_encoder
 from ...core.pydantic_utilities import parse_obj_as
 from ...core.request_options import RequestOptions
 from ...errors.bad_request_error import BadRequestError
+from ...errors.forbidden_error import ForbiddenError
 from ...types.components_schemas_composio_execute_tool_response import ComponentsSchemasComposioExecuteToolResponse
 from ...types.components_schemas_composio_tool_definition import ComponentsSchemasComposioToolDefinition
+from ...types.integration_read import IntegrationRead
+from ...types.paginated_slim_integration_read_list import PaginatedSlimIntegrationReadList
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -23,8 +26,8 @@ class RawIntegrationsClient:
 
     def retrieve_integration_tool_definition(
         self,
-        integration: str,
-        provider: str,
+        integration_name: str,
+        integration_provider: str,
         tool_name: str,
         *,
         request_options: typing.Optional[RequestOptions] = None,
@@ -32,10 +35,10 @@ class RawIntegrationsClient:
         """
         Parameters
         ----------
-        integration : str
+        integration_name : str
             The integration name
 
-        provider : str
+        integration_provider : str
             The integration provider name
 
         tool_name : str
@@ -50,7 +53,7 @@ class RawIntegrationsClient:
 
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"integrations/v1/providers/{jsonable_encoder(provider)}/integrations/{jsonable_encoder(integration)}/tools/{jsonable_encoder(tool_name)}",
+            f"integrations/v1/providers/{jsonable_encoder(integration_provider)}/integrations/{jsonable_encoder(integration_name)}/tools/{jsonable_encoder(tool_name)}",
             base_url=self._client_wrapper.get_environment().default,
             method="GET",
             request_options=request_options,
@@ -72,8 +75,8 @@ class RawIntegrationsClient:
 
     def execute_integration_tool(
         self,
-        integration: str,
-        provider: str,
+        integration_name: str,
+        integration_provider: str,
         tool_name: str,
         *,
         arguments: typing.Dict[str, typing.Optional[typing.Any]],
@@ -82,10 +85,10 @@ class RawIntegrationsClient:
         """
         Parameters
         ----------
-        integration : str
+        integration_name : str
             The integration name
 
-        provider : str
+        integration_provider : str
             The integration provider name
 
         tool_name : str
@@ -102,7 +105,7 @@ class RawIntegrationsClient:
 
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"integrations/v1/providers/{jsonable_encoder(provider)}/integrations/{jsonable_encoder(integration)}/tools/{jsonable_encoder(tool_name)}/execute",
+            f"integrations/v1/providers/{jsonable_encoder(integration_provider)}/integrations/{jsonable_encoder(integration_name)}/tools/{jsonable_encoder(tool_name)}/execute",
             base_url=self._client_wrapper.get_environment().default,
             method="POST",
             json={
@@ -136,6 +139,123 @@ class RawIntegrationsClient:
                         ),
                     ),
                 )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    def list(
+        self,
+        *,
+        integration_provider: typing.Optional[typing.Literal["COMPOSIO"]] = None,
+        limit: typing.Optional[int] = None,
+        offset: typing.Optional[int] = None,
+        ordering: typing.Optional[str] = None,
+        search: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[PaginatedSlimIntegrationReadList]:
+        """
+        List all integrations
+
+        Parameters
+        ----------
+        integration_provider : typing.Optional[typing.Literal["COMPOSIO"]]
+            * `COMPOSIO` - Composio
+
+        limit : typing.Optional[int]
+            Number of results to return per page.
+
+        offset : typing.Optional[int]
+            The initial index from which to return the results.
+
+        ordering : typing.Optional[str]
+            Which field to use when ordering the results.
+
+        search : typing.Optional[str]
+            A search term.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[PaginatedSlimIntegrationReadList]
+
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/integrations",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            params={
+                "integration_provider": integration_provider,
+                "limit": limit,
+                "offset": offset,
+                "ordering": ordering,
+                "search": search,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    PaginatedSlimIntegrationReadList,
+                    parse_obj_as(
+                        type_=PaginatedSlimIntegrationReadList,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    def retrieve(
+        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> HttpResponse[IntegrationRead]:
+        """
+        Retrieve an integration
+
+        Parameters
+        ----------
+        id : str
+            A UUID string identifying this integration.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[IntegrationRead]
+
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/integrations/{jsonable_encoder(id)}",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    IntegrationRead,
+                    parse_obj_as(
+                        type_=IntegrationRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
@@ -148,8 +268,8 @@ class AsyncRawIntegrationsClient:
 
     async def retrieve_integration_tool_definition(
         self,
-        integration: str,
-        provider: str,
+        integration_name: str,
+        integration_provider: str,
         tool_name: str,
         *,
         request_options: typing.Optional[RequestOptions] = None,
@@ -157,10 +277,10 @@ class AsyncRawIntegrationsClient:
         """
         Parameters
         ----------
-        integration : str
+        integration_name : str
             The integration name
 
-        provider : str
+        integration_provider : str
             The integration provider name
 
         tool_name : str
@@ -175,7 +295,7 @@ class AsyncRawIntegrationsClient:
 
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"integrations/v1/providers/{jsonable_encoder(provider)}/integrations/{jsonable_encoder(integration)}/tools/{jsonable_encoder(tool_name)}",
+            f"integrations/v1/providers/{jsonable_encoder(integration_provider)}/integrations/{jsonable_encoder(integration_name)}/tools/{jsonable_encoder(tool_name)}",
             base_url=self._client_wrapper.get_environment().default,
             method="GET",
             request_options=request_options,
@@ -197,8 +317,8 @@ class AsyncRawIntegrationsClient:
 
     async def execute_integration_tool(
         self,
-        integration: str,
-        provider: str,
+        integration_name: str,
+        integration_provider: str,
         tool_name: str,
         *,
         arguments: typing.Dict[str, typing.Optional[typing.Any]],
@@ -207,10 +327,10 @@ class AsyncRawIntegrationsClient:
         """
         Parameters
         ----------
-        integration : str
+        integration_name : str
             The integration name
 
-        provider : str
+        integration_provider : str
             The integration provider name
 
         tool_name : str
@@ -227,7 +347,7 @@ class AsyncRawIntegrationsClient:
 
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"integrations/v1/providers/{jsonable_encoder(provider)}/integrations/{jsonable_encoder(integration)}/tools/{jsonable_encoder(tool_name)}/execute",
+            f"integrations/v1/providers/{jsonable_encoder(integration_provider)}/integrations/{jsonable_encoder(integration_name)}/tools/{jsonable_encoder(tool_name)}/execute",
             base_url=self._client_wrapper.get_environment().default,
             method="POST",
             json={
@@ -261,6 +381,123 @@ class AsyncRawIntegrationsClient:
                         ),
                     ),
                 )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    async def list(
+        self,
+        *,
+        integration_provider: typing.Optional[typing.Literal["COMPOSIO"]] = None,
+        limit: typing.Optional[int] = None,
+        offset: typing.Optional[int] = None,
+        ordering: typing.Optional[str] = None,
+        search: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[PaginatedSlimIntegrationReadList]:
+        """
+        List all integrations
+
+        Parameters
+        ----------
+        integration_provider : typing.Optional[typing.Literal["COMPOSIO"]]
+            * `COMPOSIO` - Composio
+
+        limit : typing.Optional[int]
+            Number of results to return per page.
+
+        offset : typing.Optional[int]
+            The initial index from which to return the results.
+
+        ordering : typing.Optional[str]
+            Which field to use when ordering the results.
+
+        search : typing.Optional[str]
+            A search term.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[PaginatedSlimIntegrationReadList]
+
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/integrations",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            params={
+                "integration_provider": integration_provider,
+                "limit": limit,
+                "offset": offset,
+                "ordering": ordering,
+                "search": search,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    PaginatedSlimIntegrationReadList,
+                    parse_obj_as(
+                        type_=PaginatedSlimIntegrationReadList,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    async def retrieve(
+        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[IntegrationRead]:
+        """
+        Retrieve an integration
+
+        Parameters
+        ----------
+        id : str
+            A UUID string identifying this integration.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[IntegrationRead]
+
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/integrations/{jsonable_encoder(id)}",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    IntegrationRead,
+                    parse_obj_as(
+                        type_=IntegrationRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)

vellum/client/resources/test_suite_runs/client.py

@@ -139,9 +139,9 @@ class TestSuiteRunsClient:
 
         expand : typing.Optional[typing.Union[str, typing.Sequence[str]]]
             The response fields to expand for more information.
-            - 'results.metric_results.metric_label' expands the metric label for each metric result.
-            - 'results.metric_results.metric_definition' expands the metric definition for each metric result.
-            - 'results.metric_results.metric_definition.name' expands the metric definition name for each metric result.
+            - 'metric_results.metric_label' expands the metric label for each metric result.
+            - 'metric_results.metric_definition' expands the metric definition for each metric result.
+            - 'metric_results.metric_definition.name' expands the metric definition name for each metric result.
 
         limit : typing.Optional[int]
             Number of results to return per page.
@@ -313,9 +313,9 @@ class AsyncTestSuiteRunsClient:
 
         expand : typing.Optional[typing.Union[str, typing.Sequence[str]]]
             The response fields to expand for more information.
-            - 'results.metric_results.metric_label' expands the metric label for each metric result.
-            - 'results.metric_results.metric_definition' expands the metric definition for each metric result.
-            - 'results.metric_results.metric_definition.name' expands the metric definition name for each metric result.
+            - 'metric_results.metric_label' expands the metric label for each metric result.
+            - 'metric_results.metric_definition' expands the metric definition for each metric result.
+            - 'metric_results.metric_definition.name' expands the metric definition name for each metric result.
 
         limit : typing.Optional[int]
             Number of results to return per page.

vellum/client/resources/test_suite_runs/raw_client.py

@@ -141,9 +141,9 @@ class RawTestSuiteRunsClient:
 
         expand : typing.Optional[typing.Union[str, typing.Sequence[str]]]
             The response fields to expand for more information.
-            - 'results.metric_results.metric_label' expands the metric label for each metric result.
-            - 'results.metric_results.metric_definition' expands the metric definition for each metric result.
-            - 'results.metric_results.metric_definition.name' expands the metric definition name for each metric result.
+            - 'metric_results.metric_label' expands the metric label for each metric result.
+            - 'metric_results.metric_definition' expands the metric definition for each metric result.
+            - 'metric_results.metric_definition.name' expands the metric definition name for each metric result.
 
         limit : typing.Optional[int]
             Number of results to return per page.
@@ -309,9 +309,9 @@ class AsyncRawTestSuiteRunsClient:
 
         expand : typing.Optional[typing.Union[str, typing.Sequence[str]]]
             The response fields to expand for more information.
-            - 'results.metric_results.metric_label' expands the metric label for each metric result.
-            - 'results.metric_results.metric_definition' expands the metric definition for each metric result.
-            - 'results.metric_results.metric_definition.name' expands the metric definition name for each metric result.
+            - 'metric_results.metric_label' expands the metric label for each metric result.
+            - 'metric_results.metric_definition' expands the metric definition for each metric result.
+            - 'metric_results.metric_definition.name' expands the metric definition name for each metric result.
 
         limit : typing.Optional[int]
             Number of results to return per page.

vellum/client/types/__init__.py

@@ -74,12 +74,15 @@ from .compile_prompt_deployment_expand_meta_request import CompilePromptDeployme
 from .compile_prompt_meta import CompilePromptMeta
 from .components_schemas_composio_execute_tool_request import ComponentsSchemasComposioExecuteToolRequest
 from .components_schemas_composio_execute_tool_response import ComponentsSchemasComposioExecuteToolResponse
+from .components_schemas_composio_integration_exec_config import ComponentsSchemasComposioIntegrationExecConfig
 from .components_schemas_composio_tool_definition import ComponentsSchemasComposioToolDefinition
 from .components_schemas_pdf_search_result_meta_source import ComponentsSchemasPdfSearchResultMetaSource
 from .components_schemas_pdf_search_result_meta_source_request import ComponentsSchemasPdfSearchResultMetaSourceRequest
 from .components_schemas_prompt_version_build_config_sandbox import ComponentsSchemasPromptVersionBuildConfigSandbox
+from .components_schemas_slim_composio_tool_definition import ComponentsSchemasSlimComposioToolDefinition
 from .composio_execute_tool_request import ComposioExecuteToolRequest
 from .composio_execute_tool_response import ComposioExecuteToolResponse
+from .composio_integration_exec_config import ComposioIntegrationExecConfig
 from .composio_tool_definition import ComposioToolDefinition
 from .condition_combinator import ConditionCombinator
 from .conditional_node_result import ConditionalNodeResult
@@ -227,6 +230,12 @@ from .initiated_prompt_execution_meta import InitiatedPromptExecutionMeta
 from .initiated_workflow_node_result_event import InitiatedWorkflowNodeResultEvent
 from .instructor_vectorizer_config import InstructorVectorizerConfig
 from .instructor_vectorizer_config_request import InstructorVectorizerConfigRequest
+from .integration_auth_config_integration import IntegrationAuthConfigIntegration
+from .integration_auth_config_integration_credential import IntegrationAuthConfigIntegrationCredential
+from .integration_credential_access_type import IntegrationCredentialAccessType
+from .integration_name import IntegrationName
+from .integration_provider import IntegrationProvider
+from .integration_read import IntegrationRead
 from .invoked_port import InvokedPort
 from .iteration_state_enum import IterationStateEnum
 from .jinja_prompt_block import JinjaPromptBlock
@@ -346,6 +355,9 @@ from .paginated_document_index_read_list import PaginatedDocumentIndexReadList
 from .paginated_folder_entity_list import PaginatedFolderEntityList
 from .paginated_slim_deployment_read_list import PaginatedSlimDeploymentReadList
 from .paginated_slim_document_list import PaginatedSlimDocumentList
+from .paginated_slim_integration_auth_config_read_list import PaginatedSlimIntegrationAuthConfigReadList
+from .paginated_slim_integration_read_list import PaginatedSlimIntegrationReadList
+from .paginated_slim_tool_definition_list import PaginatedSlimToolDefinitionList
 from .paginated_slim_workflow_deployment_list import PaginatedSlimWorkflowDeploymentList
 from .paginated_test_suite_run_execution_list import PaginatedTestSuiteRunExecutionList
 from .paginated_test_suite_test_case_list import PaginatedTestSuiteTestCaseList
@@ -436,9 +448,12 @@ from .sentence_chunker_config import SentenceChunkerConfig
 from .sentence_chunker_config_request import SentenceChunkerConfigRequest
 from .sentence_chunking import SentenceChunking
 from .sentence_chunking_request import SentenceChunkingRequest
+from .slim_composio_tool_definition import SlimComposioToolDefinition
 from .slim_deployment_read import SlimDeploymentRead
 from .slim_document import SlimDocument
 from .slim_document_document_to_document_index import SlimDocumentDocumentToDocumentIndex
+from .slim_integration_auth_config_read import SlimIntegrationAuthConfigRead
+from .slim_integration_read import SlimIntegrationRead
 from .slim_release_review import SlimReleaseReview
 from .slim_workflow_deployment import SlimWorkflowDeployment
 from .slim_workflow_execution_read import SlimWorkflowExecutionRead
@@ -578,6 +593,7 @@ from .token_overlapping_window_chunker_config import TokenOverlappingWindowChunk
 from .token_overlapping_window_chunker_config_request import TokenOverlappingWindowChunkerConfigRequest
 from .token_overlapping_window_chunking import TokenOverlappingWindowChunking
 from .token_overlapping_window_chunking_request import TokenOverlappingWindowChunkingRequest
+from .tool_definition_integration import ToolDefinitionIntegration
 from .unit_enum import UnitEnum
 from .upload_document_response import UploadDocumentResponse
 from .upsert_test_suite_test_case_request import UpsertTestSuiteTestCaseRequest
@@ -768,12 +784,15 @@ __all__ = [
     "CompilePromptMeta",
     "ComponentsSchemasComposioExecuteToolRequest",
     "ComponentsSchemasComposioExecuteToolResponse",
+    "ComponentsSchemasComposioIntegrationExecConfig",
     "ComponentsSchemasComposioToolDefinition",
     "ComponentsSchemasPdfSearchResultMetaSource",
     "ComponentsSchemasPdfSearchResultMetaSourceRequest",
     "ComponentsSchemasPromptVersionBuildConfigSandbox",
+    "ComponentsSchemasSlimComposioToolDefinition",
     "ComposioExecuteToolRequest",
     "ComposioExecuteToolResponse",
+    "ComposioIntegrationExecConfig",
     "ComposioToolDefinition",
     "ConditionCombinator",
     "ConditionalNodeResult",
@@ -917,6 +936,12 @@ __all__ = [
     "InitiatedWorkflowNodeResultEvent",
     "InstructorVectorizerConfig",
     "InstructorVectorizerConfigRequest",
+    "IntegrationAuthConfigIntegration",
+    "IntegrationAuthConfigIntegrationCredential",
+    "IntegrationCredentialAccessType",
+    "IntegrationName",
+    "IntegrationProvider",
+    "IntegrationRead",
     "InvokedPort",
     "IterationStateEnum",
     "JinjaPromptBlock",
@@ -1036,6 +1061,9 @@ __all__ = [
     "PaginatedFolderEntityList",
     "PaginatedSlimDeploymentReadList",
     "PaginatedSlimDocumentList",
+    "PaginatedSlimIntegrationAuthConfigReadList",
+    "PaginatedSlimIntegrationReadList",
+    "PaginatedSlimToolDefinitionList",
     "PaginatedSlimWorkflowDeploymentList",
     "PaginatedTestSuiteRunExecutionList",
     "PaginatedTestSuiteTestCaseList",
@@ -1126,9 +1154,12 @@ __all__ = [
     "SentenceChunkerConfigRequest",
     "SentenceChunking",
     "SentenceChunkingRequest",
+    "SlimComposioToolDefinition",
     "SlimDeploymentRead",
     "SlimDocument",
     "SlimDocumentDocumentToDocumentIndex",
+    "SlimIntegrationAuthConfigRead",
+    "SlimIntegrationRead",
     "SlimReleaseReview",
     "SlimWorkflowDeployment",
     "SlimWorkflowExecutionRead",
@@ -1252,6 +1283,7 @@ __all__ = [
     "TokenOverlappingWindowChunkerConfigRequest",
     "TokenOverlappingWindowChunking",
     "TokenOverlappingWindowChunkingRequest",
+    "ToolDefinitionIntegration",
     "UnitEnum",
     "UploadDocumentResponse",
     "UpsertTestSuiteTestCaseRequest",

vellum/client/types/components_schemas_composio_integration_exec_config.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .composio_integration_exec_config import ComposioIntegrationExecConfig
+
+ComponentsSchemasComposioIntegrationExecConfig = ComposioIntegrationExecConfig

vellum/client/types/components_schemas_slim_composio_tool_definition.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .slim_composio_tool_definition import SlimComposioToolDefinition
+
+ComponentsSchemasSlimComposioToolDefinition = SlimComposioToolDefinition

vellum/client/types/composio_integration_exec_config.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class ComposioIntegrationExecConfig(UniversalBaseModel):
+    type: typing.Literal["COMPOSIO"] = "COMPOSIO"
+    slug: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/composio_tool_definition.py

@@ -4,6 +4,7 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .tool_definition_integration import ToolDefinitionIntegration
 
 
 class ComposioToolDefinition(UniversalBaseModel):
@@ -12,9 +13,11 @@ class ComposioToolDefinition(UniversalBaseModel):
     """
 
     provider: typing.Literal["COMPOSIO"] = "COMPOSIO"
+    integration: ToolDefinitionIntegration
     name: str
     description: str
-    parameters: typing.Dict[str, typing.Optional[typing.Any]]
+    input_parameters: typing.Dict[str, typing.Optional[typing.Any]]
+    output_parameters: typing.Dict[str, typing.Optional[typing.Any]]
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/integration_auth_config_integration.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .integration_name import IntegrationName
+from .integration_provider import IntegrationProvider
+
+
+class IntegrationAuthConfigIntegration(UniversalBaseModel):
+    id: str
+    provider: IntegrationProvider = "COMPOSIO"
+    name: IntegrationName
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/integration_auth_config_integration_credential.py

@@ -0,0 +1,19 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class IntegrationAuthConfigIntegrationCredential(UniversalBaseModel):
+    id: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/integration_credential_access_type.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+IntegrationCredentialAccessType = typing.Union[typing.Literal["ENVIRONMENT", "USER"], typing.Any]