vellum-ai 1.5.4__tar.gz → 1.5.5__tar.gz

{vellum_ai-1.5.4 → vellum_ai-1.5.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 1.5.4
+Version: 1.5.5
 Summary: 
 License: MIT
 Requires-Python: >=3.9,<4.0

vellum_ai-1.5.5/ee/vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_vellum_integration_serialization.py

@@ -0,0 +1,68 @@
+from vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import get_workflow_display
+
+from tests.workflows.basic_tool_calling_node_with_vellum_integration_tool.workflow import (
+    BasicToolCallingNodeWithVellumIntegrationToolWorkflow,
+)
+
+
+def test_serialize_workflow():
+    # GIVEN a Workflow that uses a tool calling node with a vellum integration tool
+    # WHEN we serialize it
+    workflow_display = get_workflow_display(workflow_class=BasicToolCallingNodeWithVellumIntegrationToolWorkflow)
+
+    serialized_workflow: dict = workflow_display.serialize()
+
+    # THEN we should get a serialized representation of the Workflow
+    assert serialized_workflow.keys() == {
+        "workflow_raw_data",
+        "input_variables",
+        "state_variables",
+        "output_variables",
+    }
+
+    # AND its input variables should be what we expect
+    input_variables = serialized_workflow["input_variables"]
+    assert len(input_variables) == 1
+    assert input_variables[0]["key"] == "query"
+    assert input_variables[0]["type"] == "STRING"
+
+    # AND its output variables should be what we expect
+    output_variables = serialized_workflow["output_variables"]
+    assert len(output_variables) == 2
+
+    # Find the text and chat_history outputs
+    text_output = next(var for var in output_variables if var["key"] == "text")
+    chat_history_output = next(var for var in output_variables if var["key"] == "chat_history")
+
+    assert text_output["type"] == "STRING"
+    assert chat_history_output["type"] == "CHAT_HISTORY"
+
+    # AND its raw data should be what we expect
+    workflow_raw_data = serialized_workflow["workflow_raw_data"]
+    tool_calling_node = workflow_raw_data["nodes"][1]
+
+    # Tool calling nodes serialize as GENERIC nodes
+    assert tool_calling_node["type"] == "GENERIC"
+    assert tool_calling_node["label"] == "Vellum Integration Tool Calling Node"
+
+    # Verify the node base class
+    assert tool_calling_node["base"]["name"] == "ToolCallingNode"
+
+    # Verify the node has the expected attributes with function definitions
+    assert "attributes" in tool_calling_node
+    attributes = tool_calling_node["attributes"]
+
+    # Find the functions attribute
+    functions_attr = next(attr for attr in attributes if attr["name"] == "functions")
+    assert functions_attr["value"]["type"] == "CONSTANT_VALUE"
+
+    # Verify the function definition details
+    functions_value = functions_attr["value"]["value"]["value"]
+    assert len(functions_value) == 1
+
+    function = functions_value[0]
+    assert function["type"] == "VELLUM_INTEGRATION"
+    assert function["provider"] == "COMPOSIO"  # VellumIntegrationProviderType.COMPOSIO
+    assert function["integration"] == "GITHUB"
+    assert function["name"] == "create_issue"
+    assert function["description"] == "Create a new issue in a GitHub repository"

{vellum_ai-1.5.4 → vellum_ai-1.5.5}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-ai"
 
 [tool.poetry]
 name = "vellum-ai"
-version = "1.5.4"
+version = "1.5.5"
 description = ""
 readme = "README.md"
 authors = []

{vellum_ai-1.5.4 → vellum_ai-1.5.5}/src/vellum/__init__.py

@@ -70,12 +70,15 @@ from .client.types import (
     CompilePromptMeta,
     ComponentsSchemasComposioExecuteToolRequest,
     ComponentsSchemasComposioExecuteToolResponse,
+    ComponentsSchemasComposioIntegrationExecConfig,
     ComponentsSchemasComposioToolDefinition,
     ComponentsSchemasPdfSearchResultMetaSource,
     ComponentsSchemasPdfSearchResultMetaSourceRequest,
     ComponentsSchemasPromptVersionBuildConfigSandbox,
+    ComponentsSchemasSlimComposioToolDefinition,
     ComposioExecuteToolRequest,
     ComposioExecuteToolResponse,
+    ComposioIntegrationExecConfig,
     ComposioToolDefinition,
     ConditionCombinator,
     ConditionalNodeResult,
@@ -219,6 +222,11 @@ from .client.types import (
     InitiatedWorkflowNodeResultEvent,
     InstructorVectorizerConfig,
     InstructorVectorizerConfigRequest,
+    IntegrationAuthConfigIntegrationCredential,
+    IntegrationCredentialAccessType,
+    IntegrationName,
+    IntegrationProvider,
+    IntegrationRead,
     InvokedPort,
     IterationStateEnum,
     JinjaPromptBlock,
@@ -338,6 +346,9 @@ from .client.types import (
     PaginatedFolderEntityList,
     PaginatedSlimDeploymentReadList,
     PaginatedSlimDocumentList,
+    PaginatedSlimIntegrationAuthConfigReadList,
+    PaginatedSlimIntegrationReadList,
+    PaginatedSlimToolDefinitionList,
     PaginatedSlimWorkflowDeploymentList,
     PaginatedTestSuiteRunExecutionList,
     PaginatedTestSuiteTestCaseList,
@@ -428,9 +439,12 @@ from .client.types import (
     SentenceChunkerConfigRequest,
     SentenceChunking,
     SentenceChunkingRequest,
+    SlimComposioToolDefinition,
     SlimDeploymentRead,
     SlimDocument,
     SlimDocumentDocumentToDocumentIndex,
+    SlimIntegrationAuthConfigRead,
+    SlimIntegrationRead,
     SlimReleaseReview,
     SlimWorkflowDeployment,
     SlimWorkflowExecutionRead,
@@ -554,6 +568,7 @@ from .client.types import (
     TokenOverlappingWindowChunkerConfigRequest,
     TokenOverlappingWindowChunking,
     TokenOverlappingWindowChunkingRequest,
+    ToolDefinitionIntegration,
     UnitEnum,
     UploadDocumentResponse,
     UpsertTestSuiteTestCaseRequest,
@@ -702,6 +717,8 @@ from .resources import (
     documents,
     events,
     folder_entities,
+    integration_auth_configs,
+    integration_providers,
     integrations,
     metric_definitions,
     ml_models,
@@ -788,12 +805,15 @@ __all__ = [
     "CompilePromptMeta",
     "ComponentsSchemasComposioExecuteToolRequest",
     "ComponentsSchemasComposioExecuteToolResponse",
+    "ComponentsSchemasComposioIntegrationExecConfig",
     "ComponentsSchemasComposioToolDefinition",
     "ComponentsSchemasPdfSearchResultMetaSource",
     "ComponentsSchemasPdfSearchResultMetaSourceRequest",
     "ComponentsSchemasPromptVersionBuildConfigSandbox",
+    "ComponentsSchemasSlimComposioToolDefinition",
     "ComposioExecuteToolRequest",
     "ComposioExecuteToolResponse",
+    "ComposioIntegrationExecConfig",
     "ComposioToolDefinition",
     "ConditionCombinator",
     "ConditionalNodeResult",
@@ -941,6 +961,11 @@ __all__ = [
     "InitiatedWorkflowNodeResultEvent",
     "InstructorVectorizerConfig",
     "InstructorVectorizerConfigRequest",
+    "IntegrationAuthConfigIntegrationCredential",
+    "IntegrationCredentialAccessType",
+    "IntegrationName",
+    "IntegrationProvider",
+    "IntegrationRead",
     "InternalServerError",
     "InvokedPort",
     "IterationStateEnum",
@@ -1065,6 +1090,9 @@ __all__ = [
     "PaginatedFolderEntityList",
     "PaginatedSlimDeploymentReadList",
     "PaginatedSlimDocumentList",
+    "PaginatedSlimIntegrationAuthConfigReadList",
+    "PaginatedSlimIntegrationReadList",
+    "PaginatedSlimToolDefinitionList",
     "PaginatedSlimWorkflowDeploymentList",
     "PaginatedTestSuiteRunExecutionList",
     "PaginatedTestSuiteTestCaseList",
@@ -1155,9 +1183,12 @@ __all__ = [
     "SentenceChunkerConfigRequest",
     "SentenceChunking",
     "SentenceChunkingRequest",
+    "SlimComposioToolDefinition",
     "SlimDeploymentRead",
     "SlimDocument",
     "SlimDocumentDocumentToDocumentIndex",
+    "SlimIntegrationAuthConfigRead",
+    "SlimIntegrationRead",
     "SlimReleaseReview",
     "SlimWorkflowDeployment",
     "SlimWorkflowExecutionRead",
@@ -1282,6 +1313,7 @@ __all__ = [
     "TokenOverlappingWindowChunking",
     "TokenOverlappingWindowChunkingRequest",
     "TooManyRequestsError",
+    "ToolDefinitionIntegration",
     "UnauthorizedError",
     "UnitEnum",
     "UploadDocumentResponse",
@@ -1418,6 +1450,8 @@ __all__ = [
     "documents",
     "events",
     "folder_entities",
+    "integration_auth_configs",
+    "integration_providers",
     "integrations",
     "metric_definitions",
     "ml_models",

{vellum_ai-1.5.4 → vellum_ai-1.5.5}/src/vellum/client/__init__.py

@@ -16,6 +16,8 @@ from .resources.document_indexes.client import AsyncDocumentIndexesClient, Docum
 from .resources.documents.client import AsyncDocumentsClient, DocumentsClient
 from .resources.events.client import AsyncEventsClient, EventsClient
 from .resources.folder_entities.client import AsyncFolderEntitiesClient, FolderEntitiesClient
+from .resources.integration_auth_configs.client import AsyncIntegrationAuthConfigsClient, IntegrationAuthConfigsClient
+from .resources.integration_providers.client import AsyncIntegrationProvidersClient, IntegrationProvidersClient
 from .resources.integrations.client import AsyncIntegrationsClient, IntegrationsClient
 from .resources.metric_definitions.client import AsyncMetricDefinitionsClient, MetricDefinitionsClient
 from .resources.ml_models.client import AsyncMlModelsClient, MlModelsClient
@@ -138,6 +140,8 @@ class Vellum:
         self.document_indexes = DocumentIndexesClient(client_wrapper=self._client_wrapper)
         self.documents = DocumentsClient(client_wrapper=self._client_wrapper)
         self.folder_entities = FolderEntitiesClient(client_wrapper=self._client_wrapper)
+        self.integration_auth_configs = IntegrationAuthConfigsClient(client_wrapper=self._client_wrapper)
+        self.integration_providers = IntegrationProvidersClient(client_wrapper=self._client_wrapper)
         self.metric_definitions = MetricDefinitionsClient(client_wrapper=self._client_wrapper)
         self.ml_models = MlModelsClient(client_wrapper=self._client_wrapper)
         self.organizations = OrganizationsClient(client_wrapper=self._client_wrapper)
@@ -1013,6 +1017,8 @@ class AsyncVellum:
         self.document_indexes = AsyncDocumentIndexesClient(client_wrapper=self._client_wrapper)
         self.documents = AsyncDocumentsClient(client_wrapper=self._client_wrapper)
         self.folder_entities = AsyncFolderEntitiesClient(client_wrapper=self._client_wrapper)
+        self.integration_auth_configs = AsyncIntegrationAuthConfigsClient(client_wrapper=self._client_wrapper)
+        self.integration_providers = AsyncIntegrationProvidersClient(client_wrapper=self._client_wrapper)
         self.metric_definitions = AsyncMetricDefinitionsClient(client_wrapper=self._client_wrapper)
         self.ml_models = AsyncMlModelsClient(client_wrapper=self._client_wrapper)
         self.organizations = AsyncOrganizationsClient(client_wrapper=self._client_wrapper)

{vellum_ai-1.5.4 → vellum_ai-1.5.5}/src/vellum/client/core/client_wrapper.py

@@ -27,10 +27,10 @@ class BaseClientWrapper:
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
-            "User-Agent": "vellum-ai/1.5.4",
+            "User-Agent": "vellum-ai/1.5.5",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "1.5.4",
+            "X-Fern-SDK-Version": "1.5.5",
             **(self.get_custom_headers() or {}),
         }
         if self._api_version is not None: