agenta 0.52.3__tar.gz → 0.52.5__tar.gz

{agenta-0.52.3 → agenta-0.52.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agenta
-Version: 0.52.3
+Version: 0.52.5
 Summary: The SDK for agenta is an open-source LLMOps platform.
 Keywords: LLMOps,LLM,evaluation,prompt engineering
 Author: Mahmoud Mabrouk

{agenta-0.52.3 → agenta-0.52.5}/agenta/sdk/middleware/base.py

@@ -1,4 +1,4 @@
-from typing import Protocol, Callable, Any
+from typing import Protocol, Callable, Any, Union
 
 from agenta.sdk.workflows.types import (
     WorkflowServiceRequest,
@@ -22,7 +22,7 @@ WorkflowMiddlewareDecorator = Callable[[WorkflowServiceHandler], WorkflowService
 
 
 def middleware_as_decorator(
-    middleware: WorkflowMiddleware | type[WorkflowMiddleware],
+    middleware: Union[WorkflowMiddleware, type[WorkflowMiddleware]],
 ) -> WorkflowMiddlewareDecorator:
     middleware = middleware() if isinstance(middleware, type) else middleware
 

{agenta-0.52.3 → agenta-0.52.5}/agenta/sdk/workflows/registry.py

@@ -1,3 +1,4 @@
+from typing import Union
 from json import dumps
 
 from agenta.sdk.utils.logging import get_module_logger
@@ -11,7 +12,7 @@ async def exact_match_v1(
     *,
     parameters: Data,
     inputs: Data,
-    outputs: Data | str,
+    outputs: Union[Data, str],
 ) -> Data:
     success = False
 

{agenta-0.52.3 → agenta-0.52.5}/agenta/sdk/workflows/types.py

@@ -401,7 +401,9 @@ class WorkflowServiceVersion(BaseModel):
 class WorkflowServiceInterface(WorkflowServiceVersion):
     uri: Optional[str] = None  # str (Enum) w/ validation
     url: Optional[str] = None  # str w/ validation
-    headers: Optional[Dict[str, Reference | str]] = None  # either hardcoded or a secret
+    headers: Optional[
+        Dict[str, Union[Reference, str]]  # either hardcoded or a secret
+    ] = None
 
     schemas: Optional[Schema] = None  # json-schema instead of pydantic
     mappings: Optional[Mappings] = None  # used in the workflow interface
@@ -432,11 +434,11 @@ class WorkflowRevision(
 
 class WorkflowServiceData(BaseModel):
     inputs: Optional[Data] = None
-    outputs: Optional[str | Data] = None
+    outputs: Optional[Union[str, Data]] = None
     trace: Optional[Trace] = None
-    trace_outputs: Optional[str | Data] = None
+    trace_outputs: Optional[Union[str, Data]] = None
     traces: Optional[Dict[str, Trace]] = None
-    traces_outputs: Optional[Dict[str, str | Data]] = None
+    traces_outputs: Optional[Dict[str, Union[str, Data]]] = None
 
 
 class WorkflowServiceRequest(Version, Metadata):

{agenta-0.52.3 → agenta-0.52.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "agenta"
-version = "0.52.3"
+version = "0.52.5"
 description = "The SDK for agenta is an open-source LLMOps platform."
 readme = "README.md"
 authors = ["Mahmoud Mabrouk <mahmoud@agenta.ai>"]