vellum-ai 0.14.89__py3-none-any.whl → 1.0.1__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 (62) hide show
  1. vellum/__init__.py +7 -1
  2. vellum/client/__init__.py +2 -2
  3. vellum/client/core/client_wrapper.py +2 -2
  4. vellum/client/types/__init__.py +6 -0
  5. vellum/client/types/organization_limit_config.py +24 -0
  6. vellum/client/types/organization_read.py +2 -0
  7. vellum/client/types/quota.py +21 -0
  8. vellum/client/types/vembda_service_tier_enum.py +5 -0
  9. vellum/client/types/workflow_execution_actual.py +1 -1
  10. vellum/evaluations/resources.py +5 -5
  11. vellum/types/organization_limit_config.py +3 -0
  12. vellum/types/quota.py +3 -0
  13. vellum/types/vembda_service_tier_enum.py +3 -0
  14. vellum/workflows/events/node.py +1 -1
  15. vellum/workflows/events/types.py +1 -1
  16. vellum/workflows/events/workflow.py +1 -1
  17. vellum/workflows/nodes/bases/tests/test_base_node.py +1 -1
  18. vellum/workflows/nodes/displayable/api_node/node.py +1 -0
  19. vellum/workflows/nodes/displayable/api_node/tests/test_api_node.py +30 -0
  20. vellum/workflows/nodes/displayable/bases/api_node/node.py +27 -10
  21. vellum/workflows/nodes/displayable/bases/base_prompt_node/node.py +1 -1
  22. vellum/workflows/nodes/displayable/bases/search_node.py +1 -1
  23. vellum/workflows/nodes/displayable/code_execution_node/node.py +1 -1
  24. vellum/workflows/nodes/displayable/guardrail_node/node.py +1 -1
  25. vellum/workflows/nodes/displayable/subworkflow_deployment_node/node.py +1 -1
  26. vellum/workflows/nodes/displayable/tool_calling_node/node.py +5 -0
  27. vellum/workflows/nodes/displayable/tool_calling_node/tests/test_node.py +3 -0
  28. vellum/workflows/nodes/displayable/tool_calling_node/utils.py +86 -114
  29. vellum/workflows/nodes/experimental/__init__.py +2 -2
  30. vellum/workflows/state/base.py +1 -1
  31. vellum/workflows/state/delta.py +1 -1
  32. {vellum_ai-0.14.89.dist-info → vellum_ai-1.0.1.dist-info}/METADATA +1 -1
  33. {vellum_ai-0.14.89.dist-info → vellum_ai-1.0.1.dist-info}/RECORD +59 -55
  34. vellum_cli/config.py +1 -1
  35. vellum_cli/push.py +1 -1
  36. vellum_cli/tests/test_ping.py +1 -0
  37. vellum_ee/workflows/display/base.py +0 -54
  38. vellum_ee/workflows/display/nodes/base_node_display.py +24 -0
  39. vellum_ee/workflows/display/nodes/vellum/api_node.py +20 -1
  40. vellum_ee/workflows/display/nodes/vellum/conditional_node.py +16 -2
  41. vellum_ee/workflows/display/nodes/vellum/inline_prompt_node.py +0 -26
  42. vellum_ee/workflows/display/nodes/vellum/tests/test_api_node.py +31 -0
  43. vellum_ee/workflows/display/nodes/vellum/tests/test_utils.py +3 -3
  44. vellum_ee/workflows/display/tests/test_base_workflow_display.py +2 -4
  45. vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/conftest.py +2 -2
  46. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_api_node_serialization.py +10 -0
  47. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_conditional_node_serialization.py +11 -11
  48. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_inline_workflow_serialization.py +21 -0
  49. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_serialization.py +21 -0
  50. vellum_ee/workflows/display/vellum.py +2 -128
  51. vellum_ee/workflows/display/workflows/__init__.py +0 -1
  52. vellum_ee/workflows/display/workflows/base_workflow_display.py +15 -3
  53. vellum_ee/workflows/display/workflows/tests/test_workflow_display.py +127 -0
  54. vellum_ee/workflows/tests/local_workflow/display/nodes/final_output.py +1 -1
  55. vellum_ee/workflows/tests/local_workflow/display/nodes/templating_node.py +1 -1
  56. vellum_ee/workflows/tests/local_workflow/display/workflow.py +11 -14
  57. vellum/workflows/nodes/experimental/tool_calling_node/__init__.py +0 -3
  58. vellum/workflows/nodes/experimental/tool_calling_node/node.py +0 -3
  59. vellum_ee/workflows/display/workflows/vellum_workflow_display.py +0 -9
  60. {vellum_ai-0.14.89.dist-info → vellum_ai-1.0.1.dist-info}/LICENSE +0 -0
  61. {vellum_ai-0.14.89.dist-info → vellum_ai-1.0.1.dist-info}/WHEEL +0 -0
  62. {vellum_ai-0.14.89.dist-info → vellum_ai-1.0.1.dist-info}/entry_points.txt +0 -0
vellum_ee/workflows/tests/local_workflow/display/nodes/final_output.py CHANGED
@@ -1,8 +1,8 @@
1
1
  from uuid import UUID
2
2
 
3
+ from vellum_ee.workflows.display.editor import NodeDisplayData, NodeDisplayPosition
3
4
  from vellum_ee.workflows.display.nodes import BaseFinalOutputNodeDisplay
4
5
  from vellum_ee.workflows.display.nodes.types import NodeOutputDisplay
5
- from vellum_ee.workflows.display.vellum import NodeDisplayData, NodeDisplayPosition
6
6
 
7
7
  from ...nodes.final_output import FinalOutput
8
8
 
vellum_ee/workflows/tests/local_workflow/display/nodes/templating_node.py CHANGED
@@ -1,8 +1,8 @@
1
1
  from uuid import UUID
2
2
 
3
+ from vellum_ee.workflows.display.editor import NodeDisplayData, NodeDisplayPosition
3
4
  from vellum_ee.workflows.display.nodes import BaseTemplatingNodeDisplay
4
5
  from vellum_ee.workflows.display.nodes.types import NodeOutputDisplay, PortDisplayOverrides
5
- from vellum_ee.workflows.display.vellum import NodeDisplayData, NodeDisplayPosition
6
6
 
7
7
  from ...nodes.templating_node import TemplatingNode
8
8
 
vellum_ee/workflows/tests/local_workflow/display/workflow.py CHANGED
@@ -1,16 +1,15 @@
1
1
  from uuid import UUID
2
2
 
3
- from vellum_ee.workflows.display.base import EdgeDisplay
4
- from vellum_ee.workflows.display.vellum import (
5
- EntrypointVellumDisplayOverrides,
6
- NodeDisplayData,
7
- NodeDisplayPosition,
3
+ from vellum_ee.workflows.display.base import (
4
+ EdgeDisplay,
5
+ EntrypointDisplay,
8
6
  WorkflowDisplayData,
9
7
  WorkflowDisplayDataViewport,
10
- WorkflowInputsVellumDisplayOverrides,
11
- WorkflowMetaVellumDisplayOverrides,
12
- WorkflowOutputVellumDisplayOverrides,
8
+ WorkflowInputsDisplay,
9
+ WorkflowMetaDisplay,
10
+ WorkflowOutputDisplay,
13
11
  )
12
+ from vellum_ee.workflows.display.editor import NodeDisplayData, NodeDisplayPosition
14
13
  from vellum_ee.workflows.display.workflows import BaseWorkflowDisplay
15
14
 
16
15
  from ..inputs import Inputs
@@ -20,7 +19,7 @@ from ..workflow import Workflow
20
19
 
21
20
 
22
21
  class WorkflowDisplay(BaseWorkflowDisplay[Workflow]):
23
- workflow_display = WorkflowMetaVellumDisplayOverrides(
22
+ workflow_display = WorkflowMetaDisplay(
24
23
  entrypoint_node_id=UUID("0bf86989-13f2-438c-ab9c-d172e5771d31"),
25
24
  entrypoint_node_source_handle_id=UUID("23448f09-81ad-4378-abbd-1cccff350627"),
26
25
  entrypoint_node_display=NodeDisplayData(position=NodeDisplayPosition(x=1545, y=330), width=124, height=48),
@@ -29,12 +28,10 @@ class WorkflowDisplay(BaseWorkflowDisplay[Workflow]):
29
28
  ),
30
29
  )
31
30
  inputs_display = {
32
- Inputs.input_value: WorkflowInputsVellumDisplayOverrides(
33
- id=UUID("2268a996-bd17-4832-b3ff-f5662d54b306"), name="input-value", required=True
34
- )
31
+ Inputs.input_value: WorkflowInputsDisplay(id=UUID("2268a996-bd17-4832-b3ff-f5662d54b306"), name="input-value")
35
32
  }
36
33
  entrypoint_displays = {
37
- TemplatingNode: EntrypointVellumDisplayOverrides(
34
+ TemplatingNode: EntrypointDisplay(
38
35
  id=UUID("0bf86989-13f2-438c-ab9c-d172e5771d31"),
39
36
  edge_display=EdgeDisplay(id=UUID("38532a0e-9432-4ed2-8a34-48a29fd6984d")),
40
37
  )
@@ -43,7 +40,7 @@ class WorkflowDisplay(BaseWorkflowDisplay[Workflow]):
43
40
  (TemplatingNode.Ports.default, FinalOutput): EdgeDisplay(id=UUID("417c56a4-cdc1-4f9d-a10c-b535163f51e8"))
44
41
  }
45
42
  output_displays = {
46
- Workflow.Outputs.final_output: WorkflowOutputVellumDisplayOverrides(
43
+ Workflow.Outputs.final_output: WorkflowOutputDisplay(
47
44
  id=UUID("5469b810-6ea6-4362-9e79-e360d44a1405"),
48
45
  name="final-output",
49
46
  )
vellum/workflows/nodes/experimental/tool_calling_node/__init__.py DELETED
@@ -1,3 +0,0 @@
1
- from vellum.workflows.nodes.displayable.tool_calling_node import ToolCallingNode
2
-
3
- __all__ = ["ToolCallingNode"]
vellum/workflows/nodes/experimental/tool_calling_node/node.py DELETED
@@ -1,3 +0,0 @@
1
- from vellum.workflows.nodes.displayable.tool_calling_node import ToolCallingNode
2
-
3
- __all__ = ["ToolCallingNode"]
vellum_ee/workflows/display/workflows/vellum_workflow_display.py DELETED
@@ -1,9 +0,0 @@
1
- from typing import Generic
2
-
3
- from vellum.workflows.types.generics import WorkflowType
4
- from vellum_ee.workflows.display.workflows.base_workflow_display import BaseWorkflowDisplay
5
-
6
-
7
- # DEPRECATED: Use BaseWorkflowDisplay instead - This file will be removed in 0.15.0
8
- class VellumWorkflowDisplay(BaseWorkflowDisplay[WorkflowType], Generic[WorkflowType]):
9
- pass