vellum-workflow-server 1.7.5.post2__tar.gz → 1.7.5.post4__tar.gz

{vellum_workflow_server-1.7.5.post2 → vellum_workflow_server-1.7.5.post4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-workflow-server
-Version: 1.7.5.post2
+Version: 1.7.5.post4
 Summary: 
 License: AGPL
 Requires-Python: >=3.9.0,<4

{vellum_workflow_server-1.7.5.post2 → vellum_workflow_server-1.7.5.post4}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-workflow-server"
 
 [tool.poetry]
 name = "vellum-workflow-server"
-version = "1.7.5.post2"
+version = "1.7.5.post4"
 description = ""
 readme = "README.md"
 authors = []

{vellum_workflow_server-1.7.5.post2 → vellum_workflow_server-1.7.5.post4}/src/workflow_server/api/workflow_view.py

@@ -22,6 +22,7 @@ from vellum_ee.workflows.server.virtual_file_loader import VirtualFileFinder
 
 from vellum.workflows.exceptions import WorkflowInitializationException
 from vellum.workflows.nodes import BaseNode
+from vellum.workflows.vellum_client import create_vellum_client
 from workflow_server.config import ENABLE_PROCESS_WRAPPER, MEMORY_LIMIT_MB
 from workflow_server.core.events import (
     SPAN_ID_EVENT,
@@ -47,7 +48,6 @@ from workflow_server.core.workflow_executor_context import (
     DEFAULT_TIMEOUT_SECONDS,
     NodeExecutorContext,
     WorkflowExecutorContext,
-    create_vellum_client,
 )
 from workflow_server.utils.oom_killer import get_is_oom_killed
 from workflow_server.utils.system_utils import (

{vellum_workflow_server-1.7.5.post2 → vellum_workflow_server-1.7.5.post4}/src/workflow_server/config.py

@@ -27,6 +27,11 @@ MEMORY_LIMIT_MB = int(os.getenv("MEMORY_LIMIT_MB", "2048"))
 PORT = os.getenv("PORT", "8000")
 VELLUM_API_URL_HOST = os.getenv("VELLUM_API_URL_HOST", "localhost")
 VELLUM_API_URL_PORT = os.getenv("VELLUM_API_URL_PORT", 8000)
+
+# Set VELLUM_API_URL for SDK when using local Vellum API
+if os.getenv("USE_LOCAL_VELLUM_API") == "true" and "VELLUM_API_URL" not in os.environ:
+    os.environ["VELLUM_API_URL"] = "http://{}:{}".format(VELLUM_API_URL_HOST, VELLUM_API_URL_PORT)
+
 CONCURRENCY = int(os.getenv("CONCURRENCY", "8"))
 CONTAINER_IMAGE = os.getenv("CONTAINER_IMAGE", "python-workflow-runtime:latest")
 ENABLE_PROCESS_WRAPPER = os.getenv("ENABLE_PROCESS_WRAPPER", "true").lower() == "true"

{vellum_workflow_server-1.7.5.post2 → vellum_workflow_server-1.7.5.post4}/src/workflow_server/core/workflow_executor_context.py

@@ -9,54 +9,15 @@ from typing_extensions import Self
 from flask import has_request_context, request
 from pydantic import Field, field_validator, model_validator
 
-from vellum import ApiVersionEnum, Vellum, VellumEnvironment
+from vellum import ApiVersionEnum, Vellum
 from vellum.client.core import UniversalBaseModel
 from vellum.workflows.context import ExecutionContext
-from workflow_server.config import IS_VPC, VELLUM_API_URL_HOST, VELLUM_API_URL_PORT
+from vellum.workflows.vellum_client import create_vellum_client
 from workflow_server.utils.utils import convert_json_inputs_to_vellum
 
 DEFAULT_TIMEOUT_SECONDS = int(os.getenv("MAX_WORKFLOW_RUNTIME_SECONDS", 1800))
 
 
-def create_vellum_client(
-    api_key: str,
-    api_version: Optional[ApiVersionEnum] = None,
-) -> Vellum:
-    """
-    Create a VellumClient with proper environment configuration.
-
-    Args:
-        api_key: The API key for the Vellum client
-        api_version: Optional API version to use
-
-    Returns:
-        Configured Vellum client instance
-
-    Note: Ideally we replace this with `vellum.workflows.vellum_client.create_vellum_client`
-    """
-    if IS_VPC:
-        environment = VellumEnvironment(
-            default=os.getenv("VELLUM_DEFAULT_API_URL", VellumEnvironment.PRODUCTION.default),
-            documents=os.getenv("VELLUM_DOCUMENTS_API_URL", VellumEnvironment.PRODUCTION.documents),
-            predict=os.getenv("VELLUM_PREDICT_API_URL", VellumEnvironment.PRODUCTION.predict),
-        )
-    elif os.getenv("USE_LOCAL_VELLUM_API") == "true":
-        VELLUM_API_URL = f"http://{VELLUM_API_URL_HOST}:{VELLUM_API_URL_PORT}"
-        environment = VellumEnvironment(
-            default=VELLUM_API_URL,
-            documents=VELLUM_API_URL,
-            predict=VELLUM_API_URL,
-        )
-    else:
-        environment = VellumEnvironment.PRODUCTION
-
-    return Vellum(
-        api_key=api_key,
-        environment=environment,
-        api_version=api_version,
-    )
-
-
 class BaseExecutorContext(UniversalBaseModel):
     inputs: dict = Field(default_factory=dict)
     state: Optional[dict] = None