vellum-ai 1.3.0__py3-none-any.whl → 1.3.2__py3-none-any.whl

vellum/client/resources/ad_hoc/raw_client.py

@@ -16,10 +16,10 @@ from ...errors.forbidden_error import ForbiddenError
 from ...errors.internal_server_error import InternalServerError
 from ...types.ad_hoc_execute_prompt_event import AdHocExecutePromptEvent
 from ...types.ad_hoc_expand_meta import AdHocExpandMeta
+from ...types.deprecated_prompt_request_input import DeprecatedPromptRequestInput
 from ...types.function_definition import FunctionDefinition
 from ...types.prompt_block import PromptBlock
 from ...types.prompt_parameters import PromptParameters
-from ...types.prompt_request_input import PromptRequestInput
 from ...types.prompt_settings import PromptSettings
 from ...types.vellum_variable import VellumVariable
 
@@ -35,7 +35,7 @@ class RawAdHocClient:
         self,
         *,
         ml_model: str,
-        input_values: typing.Sequence[PromptRequestInput],
+        input_values: typing.Sequence[DeprecatedPromptRequestInput],
         input_variables: typing.Sequence[VellumVariable],
         parameters: PromptParameters,
         blocks: typing.Sequence[PromptBlock],
@@ -49,7 +49,7 @@ class RawAdHocClient:
         ----------
         ml_model : str
 
-        input_values : typing.Sequence[PromptRequestInput]
+        input_values : typing.Sequence[DeprecatedPromptRequestInput]
 
         input_variables : typing.Sequence[VellumVariable]
 
@@ -78,7 +78,7 @@ class RawAdHocClient:
             json={
                 "ml_model": ml_model,
                 "input_values": convert_and_respect_annotation_metadata(
-                    object_=input_values, annotation=typing.Sequence[PromptRequestInput], direction="write"
+                    object_=input_values, annotation=typing.Sequence[DeprecatedPromptRequestInput], direction="write"
                 ),
                 "input_variables": convert_and_respect_annotation_metadata(
                     object_=input_variables, annotation=typing.Sequence[VellumVariable], direction="write"
@@ -160,7 +160,7 @@ class RawAdHocClient:
         self,
         *,
         ml_model: str,
-        input_values: typing.Sequence[PromptRequestInput],
+        input_values: typing.Sequence[DeprecatedPromptRequestInput],
         input_variables: typing.Sequence[VellumVariable],
         parameters: PromptParameters,
         blocks: typing.Sequence[PromptBlock],
@@ -174,7 +174,7 @@ class RawAdHocClient:
         ----------
         ml_model : str
 
-        input_values : typing.Sequence[PromptRequestInput]
+        input_values : typing.Sequence[DeprecatedPromptRequestInput]
 
         input_variables : typing.Sequence[VellumVariable]
 
@@ -203,7 +203,7 @@ class RawAdHocClient:
             json={
                 "ml_model": ml_model,
                 "input_values": convert_and_respect_annotation_metadata(
-                    object_=input_values, annotation=typing.Sequence[PromptRequestInput], direction="write"
+                    object_=input_values, annotation=typing.Sequence[DeprecatedPromptRequestInput], direction="write"
                 ),
                 "input_variables": convert_and_respect_annotation_metadata(
                     object_=input_variables, annotation=typing.Sequence[VellumVariable], direction="write"
@@ -306,7 +306,7 @@ class AsyncRawAdHocClient:
         self,
         *,
         ml_model: str,
-        input_values: typing.Sequence[PromptRequestInput],
+        input_values: typing.Sequence[DeprecatedPromptRequestInput],
         input_variables: typing.Sequence[VellumVariable],
         parameters: PromptParameters,
         blocks: typing.Sequence[PromptBlock],
@@ -320,7 +320,7 @@ class AsyncRawAdHocClient:
         ----------
         ml_model : str
 
-        input_values : typing.Sequence[PromptRequestInput]
+        input_values : typing.Sequence[DeprecatedPromptRequestInput]
 
         input_variables : typing.Sequence[VellumVariable]
 
@@ -349,7 +349,7 @@ class AsyncRawAdHocClient:
             json={
                 "ml_model": ml_model,
                 "input_values": convert_and_respect_annotation_metadata(
-                    object_=input_values, annotation=typing.Sequence[PromptRequestInput], direction="write"
+                    object_=input_values, annotation=typing.Sequence[DeprecatedPromptRequestInput], direction="write"
                 ),
                 "input_variables": convert_and_respect_annotation_metadata(
                     object_=input_variables, annotation=typing.Sequence[VellumVariable], direction="write"
@@ -431,7 +431,7 @@ class AsyncRawAdHocClient:
         self,
         *,
         ml_model: str,
-        input_values: typing.Sequence[PromptRequestInput],
+        input_values: typing.Sequence[DeprecatedPromptRequestInput],
         input_variables: typing.Sequence[VellumVariable],
         parameters: PromptParameters,
         blocks: typing.Sequence[PromptBlock],
@@ -445,7 +445,7 @@ class AsyncRawAdHocClient:
         ----------
         ml_model : str
 
-        input_values : typing.Sequence[PromptRequestInput]
+        input_values : typing.Sequence[DeprecatedPromptRequestInput]
 
         input_variables : typing.Sequence[VellumVariable]
 
@@ -474,7 +474,7 @@ class AsyncRawAdHocClient:
             json={
                 "ml_model": ml_model,
                 "input_values": convert_and_respect_annotation_metadata(
-                    object_=input_values, annotation=typing.Sequence[PromptRequestInput], direction="write"
+                    object_=input_values, annotation=typing.Sequence[DeprecatedPromptRequestInput], direction="write"
                 ),
                 "input_variables": convert_and_respect_annotation_metadata(
                     object_=input_variables, annotation=typing.Sequence[VellumVariable], direction="write"

vellum/client/resources/events/client.py

@@ -4,8 +4,8 @@ import typing
 
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
+from ...types.create_workflow_event_request import CreateWorkflowEventRequest
 from ...types.event_create_response import EventCreateResponse
-from ...types.workflow_event import WorkflowEvent
 from .raw_client import AsyncRawEventsClient, RawEventsClient
 
 # this is used as the default value for optional parameters
@@ -32,14 +32,14 @@ class EventsClient:
         return self._raw_client
 
     def create(
-        self, *, request: WorkflowEvent, request_options: typing.Optional[RequestOptions] = None
+        self, *, request: CreateWorkflowEventRequest, request_options: typing.Optional[RequestOptions] = None
     ) -> EventCreateResponse:
         """
-        Accept an event and publish it to ClickHouse for analytics processing.
+        Accept an event or list of events and publish them to ClickHouse for analytics processing.
 
         Parameters
         ----------
-        request : WorkflowEvent
+        request : CreateWorkflowEventRequest
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -65,22 +65,40 @@ class EventsClient:
             api_key="YOUR_API_KEY",
         )
         client.events.create(
-            request=NodeExecutionInitiatedEvent(
-                body=NodeExecutionInitiatedBody(
-                    node_definition=VellumCodeResourceDefinition(
-                        name="name",
-                        module=["module", "module"],
-                        id="id",
+            request=[
+                NodeExecutionInitiatedEvent(
+                    body=NodeExecutionInitiatedBody(
+                        node_definition=VellumCodeResourceDefinition(
+                            name="name",
+                            module=["module", "module"],
+                            id="id",
+                        ),
+                        inputs={"inputs": {"key": "value"}},
+                    ),
+                    id="id",
+                    timestamp=datetime.datetime.fromisoformat(
+                        "2024-01-15 09:30:00+00:00",
                     ),
-                    inputs={"inputs": {"key": "value"}},
+                    trace_id="trace_id",
+                    span_id="span_id",
                 ),
-                id="id",
-                timestamp=datetime.datetime.fromisoformat(
-                    "2024-01-15 09:30:00+00:00",
+                NodeExecutionInitiatedEvent(
+                    body=NodeExecutionInitiatedBody(
+                        node_definition=VellumCodeResourceDefinition(
+                            name="name",
+                            module=["module", "module"],
+                            id="id",
+                        ),
+                        inputs={"inputs": {"key": "value"}},
+                    ),
+                    id="id",
+                    timestamp=datetime.datetime.fromisoformat(
+                        "2024-01-15 09:30:00+00:00",
+                    ),
+                    trace_id="trace_id",
+                    span_id="span_id",
                 ),
-                trace_id="trace_id",
-                span_id="span_id",
-            ),
+            ],
         )
         """
         _response = self._raw_client.create(request=request, request_options=request_options)
@@ -103,14 +121,14 @@ class AsyncEventsClient:
         return self._raw_client
 
     async def create(
-        self, *, request: WorkflowEvent, request_options: typing.Optional[RequestOptions] = None
+        self, *, request: CreateWorkflowEventRequest, request_options: typing.Optional[RequestOptions] = None
     ) -> EventCreateResponse:
         """
-        Accept an event and publish it to ClickHouse for analytics processing.
+        Accept an event or list of events and publish them to ClickHouse for analytics processing.
 
         Parameters
         ----------
-        request : WorkflowEvent
+        request : CreateWorkflowEventRequest
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -140,22 +158,40 @@ class AsyncEventsClient:
 
         async def main() -> None:
             await client.events.create(
-                request=NodeExecutionInitiatedEvent(
-                    body=NodeExecutionInitiatedBody(
-                        node_definition=VellumCodeResourceDefinition(
-                            name="name",
-                            module=["module", "module"],
-                            id="id",
+                request=[
+                    NodeExecutionInitiatedEvent(
+                        body=NodeExecutionInitiatedBody(
+                            node_definition=VellumCodeResourceDefinition(
+                                name="name",
+                                module=["module", "module"],
+                                id="id",
+                            ),
+                            inputs={"inputs": {"key": "value"}},
                         ),
-                        inputs={"inputs": {"key": "value"}},
+                        id="id",
+                        timestamp=datetime.datetime.fromisoformat(
+                            "2024-01-15 09:30:00+00:00",
+                        ),
+                        trace_id="trace_id",
+                        span_id="span_id",
                     ),
-                    id="id",
-                    timestamp=datetime.datetime.fromisoformat(
-                        "2024-01-15 09:30:00+00:00",
+                    NodeExecutionInitiatedEvent(
+                        body=NodeExecutionInitiatedBody(
+                            node_definition=VellumCodeResourceDefinition(
+                                name="name",
+                                module=["module", "module"],
+                                id="id",
+                            ),
+                            inputs={"inputs": {"key": "value"}},
+                        ),
+                        id="id",
+                        timestamp=datetime.datetime.fromisoformat(
+                            "2024-01-15 09:30:00+00:00",
+                        ),
+                        trace_id="trace_id",
+                        span_id="span_id",
                     ),
-                    trace_id="trace_id",
-                    span_id="span_id",
-                ),
+                ],
             )
 
 

vellum/client/resources/events/raw_client.py

@@ -13,9 +13,9 @@ from ...errors.bad_request_error import BadRequestError
 from ...errors.forbidden_error import ForbiddenError
 from ...errors.too_many_requests_error import TooManyRequestsError
 from ...errors.unauthorized_error import UnauthorizedError
+from ...types.create_workflow_event_request import CreateWorkflowEventRequest
 from ...types.error_detail_response import ErrorDetailResponse
 from ...types.event_create_response import EventCreateResponse
-from ...types.workflow_event import WorkflowEvent
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -26,14 +26,14 @@ class RawEventsClient:
         self._client_wrapper = client_wrapper
 
     def create(
-        self, *, request: WorkflowEvent, request_options: typing.Optional[RequestOptions] = None
+        self, *, request: CreateWorkflowEventRequest, request_options: typing.Optional[RequestOptions] = None
     ) -> HttpResponse[EventCreateResponse]:
         """
-        Accept an event and publish it to ClickHouse for analytics processing.
+        Accept an event or list of events and publish them to ClickHouse for analytics processing.
 
         Parameters
         ----------
-        request : WorkflowEvent
+        request : CreateWorkflowEventRequest
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -47,7 +47,9 @@ class RawEventsClient:
             "monitoring/v1/events",
             base_url=self._client_wrapper.get_environment().default,
             method="POST",
-            json=convert_and_respect_annotation_metadata(object_=request, annotation=WorkflowEvent, direction="write"),
+            json=convert_and_respect_annotation_metadata(
+                object_=request, annotation=CreateWorkflowEventRequest, direction="write"
+            ),
             headers={
                 "content-type": "application/json",
             },
@@ -119,14 +121,14 @@ class AsyncRawEventsClient:
         self._client_wrapper = client_wrapper
 
     async def create(
-        self, *, request: WorkflowEvent, request_options: typing.Optional[RequestOptions] = None
+        self, *, request: CreateWorkflowEventRequest, request_options: typing.Optional[RequestOptions] = None
     ) -> AsyncHttpResponse[EventCreateResponse]:
         """
-        Accept an event and publish it to ClickHouse for analytics processing.
+        Accept an event or list of events and publish them to ClickHouse for analytics processing.
 
         Parameters
         ----------
-        request : WorkflowEvent
+        request : CreateWorkflowEventRequest
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -140,7 +142,9 @@ class AsyncRawEventsClient:
             "monitoring/v1/events",
             base_url=self._client_wrapper.get_environment().default,
             method="POST",
-            json=convert_and_respect_annotation_metadata(object_=request, annotation=WorkflowEvent, direction="write"),
+            json=convert_and_respect_annotation_metadata(
+                object_=request, annotation=CreateWorkflowEventRequest, direction="write"
+            ),
             headers={
                 "content-type": "application/json",
             },

vellum/client/types/__init__.py

@@ -82,6 +82,7 @@ from .container_image_build_config import ContainerImageBuildConfig
 from .container_image_container_image_tag import ContainerImageContainerImageTag
 from .container_image_read import ContainerImageRead
 from .create_test_suite_test_case_request import CreateTestSuiteTestCaseRequest
+from .create_workflow_event_request import CreateWorkflowEventRequest
 from .delimiter_chunker_config import DelimiterChunkerConfig
 from .delimiter_chunker_config_request import DelimiterChunkerConfigRequest
 from .delimiter_chunking import DelimiterChunking
@@ -92,6 +93,7 @@ from .deployment_provider_payload_response_payload import DeploymentProviderPayl
 from .deployment_read import DeploymentRead
 from .deployment_release_tag_deployment_history_item import DeploymentReleaseTagDeploymentHistoryItem
 from .deployment_release_tag_read import DeploymentReleaseTagRead
+from .deprecated_prompt_request_input import DeprecatedPromptRequestInput
 from .docker_service_token import DockerServiceToken
 from .document_chat_message_content import DocumentChatMessageContent
 from .document_chat_message_content_request import DocumentChatMessageContentRequest
@@ -648,6 +650,7 @@ from .workflow_execution_view_online_eval_metric_result import WorkflowExecution
 from .workflow_execution_workflow_result_event import WorkflowExecutionWorkflowResultEvent
 from .workflow_expand_meta_request import WorkflowExpandMetaRequest
 from .workflow_initialization_error import WorkflowInitializationError
+from .workflow_input import WorkflowInput
 from .workflow_node_result_data import WorkflowNodeResultData
 from .workflow_node_result_event import WorkflowNodeResultEvent
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
@@ -765,6 +768,7 @@ __all__ = [
     "ContainerImageContainerImageTag",
     "ContainerImageRead",
     "CreateTestSuiteTestCaseRequest",
+    "CreateWorkflowEventRequest",
     "DelimiterChunkerConfig",
     "DelimiterChunkerConfigRequest",
     "DelimiterChunking",
@@ -775,6 +779,7 @@ __all__ = [
     "DeploymentRead",
     "DeploymentReleaseTagDeploymentHistoryItem",
     "DeploymentReleaseTagRead",
+    "DeprecatedPromptRequestInput",
     "DockerServiceToken",
     "DocumentChatMessageContent",
     "DocumentChatMessageContentRequest",
@@ -1311,6 +1316,7 @@ __all__ = [
     "WorkflowExecutionWorkflowResultEvent",
     "WorkflowExpandMetaRequest",
     "WorkflowInitializationError",
+    "WorkflowInput",
     "WorkflowNodeResultData",
     "WorkflowNodeResultEvent",
     "WorkflowNodeResultEventState",

vellum/client/types/create_workflow_event_request.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .workflow_event import WorkflowEvent
+
+CreateWorkflowEventRequest = typing.Union[typing.List[WorkflowEvent], WorkflowEvent]

vellum/client/types/deprecated_prompt_request_input.py

@@ -0,0 +1,8 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .prompt_request_input import PromptRequestInput
+from .workflow_input import WorkflowInput
+
+DeprecatedPromptRequestInput = typing.Union[WorkflowInput, PromptRequestInput]

vellum/client/types/event_create_response.py

@@ -16,6 +16,11 @@ class EventCreateResponse(UniversalBaseModel):
     Indicates whether the event was published successfully.
     """
 
+    count: int = pydantic.Field()
+    """
+    Number of events processed
+    """
+
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:

vellum/client/types/logical_operator.py

@@ -22,6 +22,7 @@ LogicalOperator = typing.Union[
         "notIn",
         "between",
         "notBetween",
+        "concat",
         "blank",
         "notBlank",
         "coalesce",

vellum/client/types/processing_failure_reason_enum.py

@@ -2,4 +2,6 @@
 
 import typing
 
-ProcessingFailureReasonEnum = typing.Union[typing.Literal["EXCEEDED_CHARACTER_LIMIT", "INVALID_FILE"], typing.Any]
+ProcessingFailureReasonEnum = typing.Union[
+    typing.Literal["EXCEEDED_CHARACTER_LIMIT", "INVALID_FILE", "INVALID_CREDENTIALS"], typing.Any
+]

vellum/client/types/slim_document.py

@@ -39,6 +39,7 @@ class SlimDocument(UniversalBaseModel):
     
     * `EXCEEDED_CHARACTER_LIMIT` - Exceeded Character Limit
     * `INVALID_FILE` - Invalid File
+    * `INVALID_CREDENTIALS` - Invalid Credentials
     """
 
     status: typing.Optional[DocumentStatus] = pydantic.Field(default=None)

vellum/client/types/workflow_input.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .array_input import ArrayInput
+from .audio_input import AudioInput
+from .chat_history_input import ChatHistoryInput
+from .document_input import DocumentInput
+from .error_input import ErrorInput
+from .function_call_input import FunctionCallInput
+from .image_input import ImageInput
+from .json_input import JsonInput
+from .number_input import NumberInput
+from .search_results_input import SearchResultsInput
+from .string_input import StringInput
+from .video_input import VideoInput
+
+WorkflowInput = typing.Union[
+    StringInput,
+    JsonInput,
+    ChatHistoryInput,
+    NumberInput,
+    SearchResultsInput,
+    ErrorInput,
+    ArrayInput,
+    FunctionCallInput,
+    AudioInput,
+    VideoInput,
+    ImageInput,
+    DocumentInput,
+]

vellum/types/create_workflow_event_request.py

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

vellum/types/deprecated_prompt_request_input.py

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

vellum/types/workflow_input.py

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

vellum/workflows/constants.py

@@ -22,6 +22,9 @@ class _UndefMeta(type):
 
         return super().__getattribute__(name)
 
+    def __str__(cls) -> str:
+        return ""
+
     def __bool__(cls) -> bool:
         return False
 

vellum/workflows/events/node.py

@@ -141,6 +141,7 @@ class NodeExecutionFulfilledEvent(_BaseNodeEvent, Generic[OutputsType]):
 
 class NodeExecutionRejectedBody(_BaseNodeExecutionBody):
     error: WorkflowError
+    traceback: Optional[str] = None
 
 
 class NodeExecutionRejectedEvent(_BaseNodeEvent):

vellum/workflows/events/tests/test_event.py

@@ -261,6 +261,7 @@ mock_node_uuid = str(uuid4_from_hash(MockNode.__qualname__))
                         "message": "Workflow failed",
                         "code": "USER_DEFINED_ERROR",
                     },
+                    "traceback": None,
                 },
                 "parent": None,
                 "links": None,

vellum/workflows/events/workflow.py

@@ -1,3 +1,4 @@
+import logging
 from uuid import UUID
 from typing import TYPE_CHECKING, Any, Dict, Generic, Iterable, Literal, Optional, Type, Union
 from typing_extensions import TypeGuard
@@ -25,6 +26,8 @@ from .types import BaseEvent, default_serializer
 if TYPE_CHECKING:
     from vellum.workflows.workflows.base import BaseWorkflow
 
+logger = logging.getLogger(__name__)
+
 
 class _BaseWorkflowExecutionBody(UniversalBaseModel):
     workflow_definition: Type["BaseWorkflow"]
@@ -107,8 +110,12 @@ class WorkflowExecutionInitiatedEvent(_BaseWorkflowEvent, Generic[InputsType, St
     ) -> WorkflowExecutionInitiatedBody[InputsType, StateType]:
         context = info.context if info and hasattr(info, "context") else {}
         if context and "event_enricher" in context and callable(context["event_enricher"]):
-            event = context["event_enricher"](self)
-            return event.body
+            try:
+                event = context["event_enricher"](self)
+                return event.body
+            except Exception as e:
+                logger.exception(f"Error in event_enricher: {e}")
+                return body
         else:
             return body
 
@@ -149,6 +156,7 @@ class WorkflowExecutionFulfilledEvent(_BaseWorkflowEvent, Generic[OutputsType]):
 
 class WorkflowExecutionRejectedBody(_BaseWorkflowExecutionBody):
     error: WorkflowError
+    traceback: Optional[str] = None
 
 
 class WorkflowExecutionRejectedEvent(_BaseWorkflowEvent):

vellum/workflows/nodes/core/templating_node/tests/test_templating_node.py

@@ -446,3 +446,19 @@ def test_templating_node__conditional_type_checking():
 
     # THEN conditional type checking works
     assert outputs.result == "test string"
+
+
+def test_templating_node__dict_wrapper_nonexistent_attribute_is_none():
+    """Test that non-existent attributes on DictWrapper evaluate to None."""
+
+    # GIVEN a templating node with nonexistent attr in the template
+    class TemplateNode(TemplatingNode[BaseState, str]):
+        template = "{% if data.nonexistent_attr is none %}none_value{% else %}{{ data.nonexistent_attr }}{% endif %}"
+        inputs = {"data": {"existing_key": "existing_value"}}
+
+    # WHEN the node is run
+    node = TemplateNode()
+    outputs = node.run()
+
+    # THEN it should recognize the non-existent attribute as ""
+    assert outputs.result == ""

vellum/workflows/nodes/displayable/code_execution_node/tests/test_node.py

@@ -847,21 +847,11 @@ def main(arg1: list) -> str:
         runtime = "PYTHON_3_11_6"
 
     # WHEN we run the node
-    with pytest.raises(NodeException) as exc_info:
-        node = ExampleCodeExecutionNode()
-        node.run()
+    node = ExampleCodeExecutionNode()
+    outputs = node.run()
 
     # AND the result should be the correct output
-    assert (
-        exc_info.value.message
-        == """\
-Traceback (most recent call last):
-  File "ExampleCodeExecutionNode.code.py", line 2, in main
-    return arg1["invalid"]
-
-AttributeError: dict has no key: 'invalid'
-"""
-    )
+    assert outputs == {"result": "", "log": ""}
 
 
 def test_run_node__execute_code__value_key_access():

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

@@ -54,12 +54,23 @@ class FunctionCallNodeMixin:
                 return function_call.value.arguments or {}
         return {}
 
+    def _extract_function_call_id(self) -> Optional[str]:
+        """Extract function call ID from function call output."""
+        current_index = getattr(self, "state").current_prompt_output_index
+        if self.function_call_output and len(self.function_call_output) > current_index:
+            function_call = self.function_call_output[current_index]
+            if function_call.type == "FUNCTION_CALL" and function_call.value is not None:
+                return function_call.value.id
+        return None
+
     def _add_function_result_to_chat_history(self, result: Any, state: ToolCallingState) -> None:
         """Add function execution result to chat history."""
+        function_call_id = self._extract_function_call_id()
         state.chat_history.append(
             ChatMessage(
                 role="FUNCTION",
                 content=StringChatMessageContent(value=json.dumps(result, cls=DefaultStateEncoder)),
+                source=function_call_id,
             )
         )
         with state.__quiet__():