agenta 0.32.0__py3-none-any.whl → 0.32.0a2__py3-none-any.whl

agenta/__init__.py

@@ -5,10 +5,10 @@ from .sdk.utils.preinit import PreInitObject
 import agenta.client.backend.types as client_types  # pylint: disable=wrong-import-order
 
 from .sdk.types import (
+    MCField,
     DictInput,
     MultipleChoice,
     FloatParam,
-    InFile,
     IntParam,
     MultipleChoiceParam,
     GroupedMultipleChoiceParam,
@@ -17,6 +17,7 @@ from .sdk.types import (
     FileInputURL,
     BinaryParam,
     Prompt,
+    PromptTemplate,
 )
 
 from .sdk.utils.logging import log as logging
@@ -28,6 +29,7 @@ from .sdk.agenta_init import Config, AgentaSingleton, init as _init
 from .sdk.utils.costs import calculate_token_usage
 from .sdk.client import Agenta
 from .sdk.litellm import litellm as callbacks
+from .sdk.managers.vault import VaultManager
 from .sdk.managers.secrets import SecretsManager
 from .sdk.managers.config import ConfigManager
 from .sdk.managers.variant import VariantManager

agenta/client/backend/client.py

@@ -99,13 +99,17 @@ class AgentaApi:
         self._client_wrapper = SyncClientWrapper(
             base_url=base_url,
             api_key=api_key,
-            httpx_client=httpx_client
-            if httpx_client is not None
-            else httpx.Client(
-                timeout=_defaulted_timeout, follow_redirects=follow_redirects
-            )
-            if follow_redirects is not None
-            else httpx.Client(timeout=_defaulted_timeout),
+            httpx_client=(
+                httpx_client
+                if httpx_client is not None
+                else (
+                    httpx.Client(
+                        timeout=_defaulted_timeout, follow_redirects=follow_redirects
+                    )
+                    if follow_redirects is not None
+                    else httpx.Client(timeout=_defaulted_timeout)
+                )
+            ),
             timeout=_defaulted_timeout,
         )
         self.observability = ObservabilityClient(client_wrapper=self._client_wrapper)
@@ -1635,13 +1639,17 @@ class AsyncAgentaApi:
         self._client_wrapper = AsyncClientWrapper(
             base_url=base_url,
             api_key=api_key,
-            httpx_client=httpx_client
-            if httpx_client is not None
-            else httpx.AsyncClient(
-                timeout=_defaulted_timeout, follow_redirects=follow_redirects
-            )
-            if follow_redirects is not None
-            else httpx.AsyncClient(timeout=_defaulted_timeout),
+            httpx_client=(
+                httpx_client
+                if httpx_client is not None
+                else (
+                    httpx.AsyncClient(
+                        timeout=_defaulted_timeout, follow_redirects=follow_redirects
+                    )
+                    if follow_redirects is not None
+                    else httpx.AsyncClient(timeout=_defaulted_timeout)
+                )
+            ),
             timeout=_defaulted_timeout,
         )
         self.observability = AsyncObservabilityClient(

agenta/client/backend/core/http_client.py

@@ -148,9 +148,9 @@ def get_request_body(
         json_body = maybe_filter_request_body(json, request_options, omit)
 
     # If you have an empty JSON body, you should just send None
-    return (
-        json_body if json_body != {} else None
-    ), data_body if data_body != {} else None
+    return (json_body if json_body != {} else None), (
+        data_body if data_body != {} else None
+    )
 
 
 class HttpClient:

agenta/sdk/__init__.py

@@ -8,7 +8,6 @@ from .types import (
     DictInput,
     MultipleChoice,
     FloatParam,
-    InFile,
     IntParam,
     MultipleChoiceParam,
     GroupedMultipleChoiceParam,
@@ -27,6 +26,7 @@ from .tracing.conventions import Reference
 from .decorators.routing import entrypoint, app, route
 from .agenta_init import Config, AgentaSingleton, init as _init
 from .utils.costs import calculate_token_usage
+from .managers.vault import VaultManager
 from .managers.secrets import SecretsManager
 from .managers.config import ConfigManager
 from .managers.variant import VariantManager

agenta/sdk/context/routing.py

@@ -9,6 +9,7 @@ from pydantic import BaseModel
 class RoutingContext(BaseModel):
     parameters: Optional[Dict[str, Any]] = None
     secrets: Optional[List[Any]] = None
+    mock: Optional[str] = None
 
 
 routing_context = ContextVar("routing_context", default=RoutingContext())