vellum-ai 1.0.5__py3-none-any.whl → 1.0.7__py3-none-any.whl

vellum/client/types/organization_limit_config.py

@@ -1,25 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from .vembda_service_tier_enum import VembdaServiceTierEnum
-from .quota import Quota
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-
-
-class OrganizationLimitConfig(UniversalBaseModel):
-    vembda_service_tier: typing.Optional[VembdaServiceTierEnum] = None
-    prompt_executions_quota: typing.Optional[Quota] = None
-    workflow_executions_quota: typing.Optional[Quota] = None
-    workflow_runtime_seconds_quota: typing.Optional[Quota] = None
-    max_workflow_runtime_seconds: typing.Optional[int] = None
-
-    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/quota.py

@@ -1,22 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from ..core.pydantic_utilities import UniversalBaseModel
-from .name_enum import NameEnum
-import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-
-
-class Quota(UniversalBaseModel):
-    name: NameEnum
-    value: typing.Optional[int] = None
-    period_seconds: typing.Optional[int] = None
-
-    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/vembda_service_tier_enum.py

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

vellum/types/name_enum.py

@@ -1,3 +0,0 @@
-# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
-
-from vellum.client.types.name_enum import *

vellum/types/organization_limit_config.py

@@ -1,3 +0,0 @@
-# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
-
-from vellum.client.types.organization_limit_config import *

vellum/types/quota.py

@@ -1,3 +0,0 @@
-# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
-
-from vellum.client.types.quota import *

vellum/types/vembda_service_tier_enum.py

@@ -1,3 +0,0 @@
-# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
-
-from vellum.client.types.vembda_service_tier_enum import *

vellum/workflows/nodes/displayable/tool_calling_node/composio_service.py

@@ -1,83 +0,0 @@
-from dataclasses import dataclass
-from typing import Any, Dict, List
-
-from composio import Action, Composio
-from composio_client import Composio as ComposioClient
-
-
-@dataclass
-class ConnectionInfo:
-    """Information about a user's authorized connection"""
-
-    connection_id: str
-    integration_name: str
-    created_at: str
-    updated_at: str
-    status: str = "ACTIVE"  # TODO: Use enum if we end up supporting integrations that the user has not yet connected to
-
-
-class ComposioAccountService:
-    """Manages user authorized connections using composio-client"""
-
-    def __init__(self, api_key: str):
-        self.client = ComposioClient(api_key=api_key)
-
-    def get_user_connections(self) -> List[ConnectionInfo]:
-        """Get all authorized connections for the user"""
-        response = self.client.connected_accounts.list()
-
-        return [
-            ConnectionInfo(
-                connection_id=item.id,
-                integration_name=item.toolkit.slug,
-                status=item.status,
-                created_at=item.created_at,
-                updated_at=item.updated_at,
-            )
-            for item in response.items
-        ]
-
-
-class ComposioCoreService:
-    """Handles tool execution using composio-core"""
-
-    def __init__(self, api_key: str):
-        self.client = Composio(api_key=api_key)
-
-    def execute_tool(self, tool_name: str, arguments: Dict[str, Any]) -> Any:
-        """Execute a tool using composio-core
-
-        Args:
-            tool_name: The name of the tool to execute (e.g., "HACKERNEWS_GET_USER")
-            arguments: Dictionary of arguments to pass to the tool
-
-        Returns:
-            The result of the tool execution
-        """
-        # Convert tool name string to Action enum
-        action = getattr(Action, tool_name)
-        return self.client.actions.execute(action, params=arguments)
-
-
-class ComposioService:
-    """Unified interface for Composio operations"""
-
-    def __init__(self, api_key: str):
-        self.accounts = ComposioAccountService(api_key)
-        self.core = ComposioCoreService(api_key)
-
-    def get_user_connections(self) -> List[ConnectionInfo]:
-        """Get user's authorized connections"""
-        return self.accounts.get_user_connections()
-
-    def execute_tool(self, tool_name: str, arguments: Dict[str, Any]) -> Any:
-        """Execute a tool using composio-core
-
-        Args:
-            tool_name: The name of the tool to execute (e.g., "HACKERNEWS_GET_USER")
-            arguments: Dictionary of arguments to pass to the tool
-
-        Returns:
-            The result of the tool execution
-        """
-        return self.core.execute_tool(tool_name, arguments)