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

agenta/sdk/types.py

@@ -6,6 +6,9 @@ from pydantic import ConfigDict, BaseModel, HttpUrl
 
 from agenta.client.backend.types.agenta_node_dto import AgentaNodeDto
 from agenta.client.backend.types.agenta_nodes_response import AgentaNodesResponse
+from typing import Annotated, List, Union, Optional, Dict, Literal, Any
+from pydantic import BaseModel, Field, model_validator
+from agenta.sdk.assets import supported_llm_models
 
 
 @dataclass
@@ -13,10 +16,17 @@ class MultipleChoice:
     choices: Union[List[str], Dict[str, List[str]]]
 
 
-class InFile:
-    def __init__(self, file_name: str, file_path: str):
-        self.file_name = file_name
-        self.file_path = file_path
+def MCField(  # pylint: disable=invalid-name
+    default: str,
+    choices: Union[List[str], Dict[str, List[str]]],
+) -> Field:
+    field = Field(default=default, description="ID of the model to use")
+    if isinstance(choices, dict):
+        field.json_schema_extra = {"choices": choices, "x-parameter": "grouped_choice"}
+    elif isinstance(choices, list):
+        field.json_schema_extra = {"choices": choices, "x-parameter": "choice"}
+
+    return field
 
 
 class LLMTokenUsage(BaseModel):
@@ -28,7 +38,11 @@ class LLMTokenUsage(BaseModel):
 class BaseResponse(BaseModel):
     version: Optional[str] = "3.0"
     data: Optional[Union[str, Dict[str, Any]]] = None
+    content_type: Optional[str] = "string"
     tree: Optional[AgentaNodesResponse] = None
+    tree_id: Optional[str] = None
+
+    model_config = ConfigDict(use_enum_values=True, exclude_none=True)
 
 
 class DictInput(dict):
@@ -248,3 +262,319 @@ class Prompt(BaseModel):
     top_p: float
     frequency_penalty: float
     presence_penalty: float
+
+
+# -----------------------------------------------------
+# New Prompt model
+# -----------------------------------------------------
+
+
+class ToolCall(BaseModel):
+    id: str
+    type: Literal["function"] = "function"
+    function: Dict[str, str]
+
+
+class Message(BaseModel):
+    role: Literal["system", "user", "assistant", "tool", "function"]
+    content: Optional[str] = None
+    name: Optional[str] = None
+    tool_calls: Optional[List[ToolCall]] = None
+    tool_call_id: Optional[str] = None
+
+
+class ResponseFormatText(BaseModel):
+    type: Literal["text"]
+    """The type of response format being defined: `text`"""
+
+
+class ResponseFormatJSONObject(BaseModel):
+    type: Literal["json_object"]
+    """The type of response format being defined: `json_object`"""
+
+
+class JSONSchema(BaseModel):
+    name: str
+    """The name of the response format."""
+    description: Optional[str] = None
+    """A description of what the response format is for."""
+    schema_: Optional[Dict[str, object]] = Field(alias="schema", default=None)
+    """The schema for the response format, described as a JSON Schema object."""
+    strict: Optional[bool] = None
+    """Whether to enable strict schema adherence."""
+
+    model_config = {
+        "populate_by_name": True,
+        "json_schema_extra": {"required": ["name", "schema"]},
+    }
+
+
+class ResponseFormatJSONSchema(BaseModel):
+    type: Literal["json_schema"]
+    """The type of response format being defined: `json_schema`"""
+    json_schema: JSONSchema
+
+
+ResponseFormat = Union[
+    ResponseFormatText, ResponseFormatJSONObject, ResponseFormatJSONSchema
+]
+
+
+class ModelConfig(BaseModel):
+    """Configuration for model parameters"""
+
+    model: str = MCField(
+        default="gpt-3.5-turbo",
+        choices=supported_llm_models,
+    )
+
+    temperature: Optional[float] = Field(
+        default=None,
+        ge=0.0,
+        le=2.0,
+        description="What sampling temperature to use, between 0 and 2. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic",
+    )
+    max_tokens: Optional[int] = Field(
+        default=None,
+        ge=0,
+        le=4000,
+        description="The maximum number of tokens that can be generated in the chat completion",
+    )
+    top_p: Optional[float] = Field(
+        default=None,
+        ge=0.0,
+        le=1.0,
+        description="An alternative to sampling with temperature, called nucleus sampling, where the model considers the results of the tokens with top_p probability mass",
+    )
+    frequency_penalty: Optional[float] = Field(
+        default=None,
+        ge=-2.0,
+        le=2.0,
+        description="Number between -2.0 and 2.0. Positive values penalize new tokens based on their existing frequency in the text so far",
+    )
+    presence_penalty: Optional[float] = Field(
+        default=None,
+        ge=-2.0,
+        le=2.0,
+        description="Number between -2.0 and 2.0. Positive values penalize new tokens based on whether they appear in the text so far",
+    )
+    response_format: Optional[ResponseFormat] = Field(
+        default=None,
+        description="An object specifying the format that the model must output",
+    )
+    stream: Optional[bool] = Field(
+        default=None, description="If set, partial message deltas will be sent"
+    )
+    tools: Optional[List[Dict]] = Field(
+        default=None,
+        description="A list of tools the model may call. Currently, only functions are supported as a tool",
+    )
+    tool_choice: Optional[Union[Literal["none", "auto"], Dict]] = Field(
+        default=None, description="Controls which (if any) tool is called by the model"
+    )
+
+
+class PromptTemplateError(Exception):
+    """Base exception for all PromptTemplate errors"""
+
+    pass
+
+
+class InputValidationError(PromptTemplateError):
+    """Raised when input validation fails"""
+
+    def __init__(
+        self, message: str, missing: Optional[set] = None, extra: Optional[set] = None
+    ):
+        self.missing = missing
+        self.extra = extra
+        super().__init__(message)
+
+
+class TemplateFormatError(PromptTemplateError):
+    """Raised when template formatting fails"""
+
+    def __init__(self, message: str, original_error: Optional[Exception] = None):
+        self.original_error = original_error
+        super().__init__(message)
+
+
+class PromptTemplate(BaseModel):
+    """A template for generating prompts with formatting capabilities"""
+
+    messages: List[Message] = Field(
+        default=[Message(role="system", content=""), Message(role="user", content="")]
+    )
+    system_prompt: Optional[str] = None
+    user_prompt: Optional[str] = None
+    template_format: Literal["fstring", "jinja2", "curly"] = Field(
+        default="curly",
+        description="Format type for template variables: fstring {var}, jinja2 {{ var }}, or curly {{var}}",
+    )
+    input_keys: Optional[List[str]] = Field(
+        default=None,
+        description="Optional list of input keys for validation. If not provided, any inputs will be accepted",
+    )
+    llm_config: ModelConfig = Field(
+        default_factory=ModelConfig,
+        description="Configuration for the model parameters",
+    )
+
+    model_config = {
+        "json_schema_extra": {
+            "x-parameters": {
+                "prompt": "true",
+            }
+        }
+    }
+
+    @model_validator(mode="before")
+    def init_messages(cls, values):
+        if "messages" not in values:
+            messages = []
+            if "system_prompt" in values and values["system_prompt"]:
+                messages.append(Message(role="system", content=values["system_prompt"]))
+            if "user_prompt" in values and values["user_prompt"]:
+                messages.append(Message(role="user", content=values["user_prompt"]))
+            if messages:
+                values["messages"] = messages
+        return values
+
+    def _format_with_template(self, content: str, kwargs: Dict[str, Any]) -> str:
+        """Internal method to format content based on template_format"""
+        try:
+            if self.template_format == "fstring":
+                return content.format(**kwargs)
+            elif self.template_format == "jinja2":
+                from jinja2 import Template, TemplateError
+
+                try:
+                    return Template(content).render(**kwargs)
+                except TemplateError as e:
+                    raise TemplateFormatError(
+                        f"Jinja2 template error in content: '{content}'. Error: {str(e)}",
+                        original_error=e,
+                    )
+            elif self.template_format == "curly":
+                import re
+
+                result = content
+                for key, value in kwargs.items():
+                    result = re.sub(r"\{\{" + key + r"\}\}", str(value), result)
+                if re.search(r"\{\{.*?\}\}", result):
+                    unreplaced = re.findall(r"\{\{(.*?)\}\}", result)
+                    raise TemplateFormatError(
+                        f"Unreplaced variables in curly template: {unreplaced}"
+                    )
+                return result
+            else:
+                raise TemplateFormatError(
+                    f"Unknown template format: {self.template_format}"
+                )
+        except KeyError as e:
+            key = str(e).strip("'")
+            raise TemplateFormatError(
+                f"Missing required variable '{key}' in template: '{content}'"
+            )
+        except Exception as e:
+            raise TemplateFormatError(
+                f"Error formatting template '{content}': {str(e)}", original_error=e
+            )
+
+    def format(self, **kwargs) -> "PromptTemplate":
+        """
+        Format the template with provided inputs.
+        Only validates against input_keys if they are specified.
+
+        Raises:
+            InputValidationError: If input validation fails
+            TemplateFormatError: If template formatting fails
+        """
+        # Validate inputs if input_keys is set
+        if self.input_keys is not None:
+            missing = set(self.input_keys) - set(kwargs.keys())
+            extra = set(kwargs.keys()) - set(self.input_keys)
+
+            error_parts = []
+            if missing:
+                error_parts.append(
+                    f"Missing required inputs: {', '.join(sorted(missing))}"
+                )
+            if extra:
+                error_parts.append(f"Unexpected inputs: {', '.join(sorted(extra))}")
+
+            if error_parts:
+                raise InputValidationError(
+                    " | ".join(error_parts),
+                    missing=missing if missing else None,
+                    extra=extra if extra else None,
+                )
+
+        new_messages = []
+        for i, msg in enumerate(self.messages):
+            if msg.content:
+                try:
+                    new_content = self._format_with_template(msg.content, kwargs)
+                except TemplateFormatError as e:
+                    raise TemplateFormatError(
+                        f"Error in message {i} ({msg.role}): {str(e)}",
+                        original_error=e.original_error,
+                    )
+            else:
+                new_content = None
+
+            new_messages.append(
+                Message(
+                    role=msg.role,
+                    content=new_content,
+                    name=msg.name,
+                    tool_calls=msg.tool_calls,
+                    tool_call_id=msg.tool_call_id,
+                )
+            )
+
+        return PromptTemplate(
+            messages=new_messages,
+            template_format=self.template_format,
+            llm_config=self.llm_config,
+            input_keys=self.input_keys,
+        )
+
+    def to_openai_kwargs(self) -> dict:
+        """Convert the prompt template to kwargs compatible with litellm/openai"""
+        kwargs = {
+            "model": self.llm_config.model,
+            "messages": [msg.dict(exclude_none=True) for msg in self.messages],
+        }
+
+        # Add optional parameters only if they are set
+        if self.llm_config.temperature is not None:
+            kwargs["temperature"] = self.llm_config.temperature
+
+        if self.llm_config.top_p is not None:
+            kwargs["top_p"] = self.llm_config.top_p
+
+        if self.llm_config.stream is not None:
+            kwargs["stream"] = self.llm_config.stream
+
+        if self.llm_config.max_tokens is not None:
+            kwargs["max_tokens"] = self.llm_config.max_tokens
+
+        if self.llm_config.frequency_penalty is not None:
+            kwargs["frequency_penalty"] = self.llm_config.frequency_penalty
+
+        if self.llm_config.presence_penalty is not None:
+            kwargs["presence_penalty"] = self.llm_config.presence_penalty
+
+        if self.llm_config.response_format:
+            kwargs["response_format"] = self.llm_config.response_format.dict(
+                by_alias=True
+            )
+
+        if self.llm_config.tools:
+            kwargs["tools"] = self.llm_config.tools
+            # Only set tool_choice if tools are present
+            if self.llm_config.tool_choice is not None:
+                kwargs["tool_choice"] = self.llm_config.tool_choice
+
+        return kwargs

{agenta-0.32.0.dist-info → agenta-0.32.0a2.dist-info}/METADATA

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

{agenta-0.32.0.dist-info → agenta-0.32.0a2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-agenta/__init__.py,sha256=BytAaijm7h51NG-H7RmllolVQ3QzLCM9WTlqMdQgYow,2150
+agenta/__init__.py,sha256=A2slx8eFxACWFIcRhm-4WZS1ineUf9cMDgxdd3QhhhQ,2216
 agenta/cli/evaluation_commands.py,sha256=fs6492tprPId9p8eGO02Xy-NCBm2RZNJLZWcUxugwd8,474
 agenta/cli/helper.py,sha256=P97HbNb_qzOyl5CM_MjAqWEBCdgebU6M81G_4UCmF1A,6288
 agenta/cli/main.py,sha256=WJSp-kJ6j0bea64l5QJlnOPpLwTgNcN7Am4X2YZBP1A,7939
@@ -16,7 +16,7 @@ agenta/client/backend/apps/__init__.py,sha256=9mUnTDeA1TxYvkj1l01A1prqsJV0ERRY2t
 agenta/client/backend/apps/client.py,sha256=6ZGBcR37ILwS2VNp8BZR4Tz09W2MlaSChPQ7it0tWoE,54946
 agenta/client/backend/bases/__init__.py,sha256=9mUnTDeA1TxYvkj1l01A1prqsJV0ERRY2tzkY1fA4MQ,64
 agenta/client/backend/bases/client.py,sha256=s8EAwrdxNkOjhDSP2-HIbqp51vNlvhXxS4nGb8YoWPk,6068
-agenta/client/backend/client.py,sha256=jWUJXLKjpi-0J6MPwrLjiXx2gdnRps2JnZ7-m23i3uo,105043
+agenta/client/backend/client.py,sha256=psZK0A7dZJLyQfXXSDEjxPeLXn3svvTGmx04Px_23sE,105267
 agenta/client/backend/configs/__init__.py,sha256=9mUnTDeA1TxYvkj1l01A1prqsJV0ERRY2tzkY1fA4MQ,64
 agenta/client/backend/configs/client.py,sha256=_nuh5K5D5SuCOeckXcGv6P4ZtVdTGWQ7JhcoIclmWfY,19267
 agenta/client/backend/containers/__init__.py,sha256=Haw2PwiPhNvM26PLQN57jY0bN-QqPoDG4VA-P_uGL3A,153
@@ -28,7 +28,7 @@ agenta/client/backend/core/api_error.py,sha256=TtMtCdxXjd7Tasc9c8ooFg124nPrb2MXG
 agenta/client/backend/core/client_wrapper.py,sha256=esOteB9jDmUGJkv6EotT54nPNU42TR_kjF3pPjU6lqY,1817
 agenta/client/backend/core/datetime_utils.py,sha256=BHjt_H3WVslcuPsr6qjJoVif_SsdLvFN0c43ABE5UiQ,1069
 agenta/client/backend/core/file.py,sha256=mUvNH6Wc-k2mSlXCJ9VoYzvgi7NcCkhTzlW2VXxSucg,2710
-agenta/client/backend/core/http_client.py,sha256=Wz6swfW0qdrpVNGEmf8HGdWdDTRq-0bAjsdV3xvbFxE,21004
+agenta/client/backend/core/http_client.py,sha256=e7jy8gDqwvdWiNGoJdY-jX73cOYFiF_p_5at82gAC6Y,21006
 agenta/client/backend/core/jsonable_encoder.py,sha256=SHXw4G4n-f0IPgNkxj_-Fip3kN8NUAI-YrKxdZw8kl0,3662
 agenta/client/backend/core/pydantic_utilities.py,sha256=BZTSULs3wlfRQwTtsEyKlyY6SkHCtf3WxHqfhH9YSG4,12325
 agenta/client/backend/core/query_encoder.py,sha256=8qYl5VPl1jU4cDF0X7oSU_DXjlVWY5ayigFBpNTMGOA,2150
@@ -190,42 +190,47 @@ agenta/docker/docker-assets/entrypoint.sh,sha256=29XK8VQjQsx4hN2j-4JDy-6kQb5y4LC
 agenta/docker/docker-assets/lambda_function.py,sha256=h4UZSSfqwpfsCgERv6frqwm_4JrYu9rLz3I-LxCfeEg,83
 agenta/docker/docker-assets/main.py,sha256=7MI-21n81U7N7A0GxebNi0cmGWtJKcR2sPB6FcH2QfA,251
 agenta/docker/docker_utils.py,sha256=kO1q2_IR0fEAo4M-2Pt_v-zC7GxxnkLogjKFhU869Ps,3555
-agenta/sdk/__init__.py,sha256=ux4l87ko0uGxBv0NqSd237s6icFZ7iAYXkj6JqIY0e0,1993
+agenta/sdk/__init__.py,sha256=Kx86cuRBKOvl8VecWtDV75elDLz5teQ5qkh_CQIPz6g,2022
 agenta/sdk/agenta_init.py,sha256=OZf-pjFyr71hfgcZC7S323t9PJLpNb-qYXOvVRmRlpA,7961
 agenta/sdk/assets.py,sha256=co49Y8T8WWINk2p7ddQPzB39AuwtVfrsyE_AnR1n5TQ,3062
 agenta/sdk/client.py,sha256=trKyBOYFZRk0v5Eptxvh87yPf50Y9CqY6Qgv4Fy-VH4,2142
 agenta/sdk/context/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/sdk/context/exporting.py,sha256=16X8fgMhl58gehSlqANX97FiKxx4TkGiG4d2B0-7ZX0,516
-agenta/sdk/context/routing.py,sha256=6QclrCDXR8Kn7uQf7Q-IzQ6lRAyXrPwFDL5RJxaOWpU,567
+agenta/sdk/context/routing.py,sha256=FEsjw8EttI1SMyUo96ptcUsvHJnhoKwdr1szlkxxJNU,598
 agenta/sdk/context/tracing.py,sha256=zp7T_wLVkR-V1c0k7UAN69rwH9VV7MhoZD_IdNu-_RE,649
 agenta/sdk/decorators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agenta/sdk/decorators/routing.py,sha256=g2F1BCnEufEXW_QRlua9oLY6x4O7_fwmUDJrqKgRTts,33104
-agenta/sdk/decorators/tracing.py,sha256=n5tsbZ2z1G7aIPQEJjvsuVGxbNSstdf4fXUx9bOASEY,8705
+agenta/sdk/decorators/routing.py,sha256=6auINyIdIS7mO7KTDDlaWDjl9iZJ4S2vobPdQVf4nvw,20874
+agenta/sdk/decorators/tracing.py,sha256=Wf7KWR1NknXlQEdB2sjMFrcSI-tSSvO9XubN-Ro5tkU,9216
 agenta/sdk/litellm/__init__.py,sha256=Bpz1gfHQc0MN1yolWcjifLWznv6GjHggvRGQSpxpihM,37
-agenta/sdk/litellm/litellm.py,sha256=Ke0It-jA1z0KQ2770gIlWIEgramZGmt1k0GjmpEnFV4,8793
+agenta/sdk/litellm/litellm.py,sha256=Xx_YJu05BYkmIe6uObjS6DwzjgwfNytGWf807Zh0vcU,10153
+agenta/sdk/litellm/mockllm.py,sha256=8V6dqdv8eA4P-VoXIwHNYlIjHG189P14POSfSfluVw0,678
+agenta/sdk/litellm/mocks/__init__.py,sha256=c-afSm0YkiHyyaLegvtFs6WuWZTl2Q7Fq_iUoxqBbQc,616
 agenta/sdk/managers/__init__.py,sha256=SN-LRwG0pRRDV3u2Q4JiiSTigN3-mYpzGNM35RzT4mc,238
 agenta/sdk/managers/config.py,sha256=8-TJn56SssNjfxCY7NhDwqL4in5gtPeMrsvyEf-W_u4,7421
 agenta/sdk/managers/deployment.py,sha256=SEokjZeh6n7HRKZ92Y0WncdG49hIFx-Z3B3HAl2kmUg,1174
 agenta/sdk/managers/secrets.py,sha256=HAXswBOTbF_LydoNGnQgHr-KVqdFUOvLujUg6S_5taM,878
 agenta/sdk/managers/shared.py,sha256=e53jckQq5PIMpjdxADOonUj7o8aGfzmSvdeH5f43rGs,21497
 agenta/sdk/managers/variant.py,sha256=A5ga3mq3b0weUTXa9HO72MGaspthGcu1uK9K5OnP738,4172
+agenta/sdk/managers/vault.py,sha256=054ce9X_xKa2M4NtQWz-GugO6q_pYVWCP3IxbAJJcRw,337
 agenta/sdk/middleware/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agenta/sdk/middleware/auth.py,sha256=SoVnzdm5ygP03bQgXDd86aTvIdeTfr4uIoeJsPxN7No,5491
+agenta/sdk/middleware/auth.py,sha256=2jV32_JdjaOZFI9VGy3OGVg6tOvgEN64RUpM71AqVZ0,5572
 agenta/sdk/middleware/cache.py,sha256=-_e3_6f4cPfpuUpGRK_ZQvUx_54ez68db4SnBwfuESk,1078
-agenta/sdk/middleware/config.py,sha256=tOMEPwS4PHaxOtH4OII0ymqx4OlP0gvMSrMepoHbtxk,7229
+agenta/sdk/middleware/config.py,sha256=_oBj9XNzxiMyoLo1DLh7Tck5s05H6_udaNCGYe5orZA,7577
 agenta/sdk/middleware/cors.py,sha256=q3r7lGkrIdMcT_vuhsburMcjG7pyl7w0ycxrIrGJ2e8,921
+agenta/sdk/middleware/inline.py,sha256=1bNBEoen9NRF2O3ZAP7z-izaEFUI1hBT22xLPIvof3A,937
+agenta/sdk/middleware/mock.py,sha256=G028gYDMzaHtjULl2yU-zqXrEueMNC9JYmnIqRmjtYM,794
 agenta/sdk/middleware/otel.py,sha256=HB-JYAHejZHHlPuHtp12jIpJ6E7k4CdnMdOhyhO77tU,1030
-agenta/sdk/middleware/vault.py,sha256=H2pquBwRLC8KwAQBPAm7bKWlc407I-vIu-Ox4ncFY1M,5114
+agenta/sdk/middleware/vault.py,sha256=Hyz9Zuq9fm_lvm-z5Ai8-PZectXMuIyaytAlpVOpqDU,4312
 agenta/sdk/router.py,sha256=mOguvtOwl2wmyAgOuWTsf98pQwpNiUILKIo67W_hR3A,119
 agenta/sdk/tracing/__init__.py,sha256=rQNe5-zT5Kt7_CDhq-lnUIi1EYTBVzVf_MbfcIxVD98,41
 agenta/sdk/tracing/attributes.py,sha256=zh8JQZSeYCLBeIRSopKJx6QQ-WEgw08Cr64DS_WOcT8,3833
 agenta/sdk/tracing/conventions.py,sha256=JBtznBXZ3aRkGKkLl7cPwdMNh3w1G-H2Ta2YrAxbr38,950
-agenta/sdk/tracing/exporters.py,sha256=5njI08ezWp2mUXL5mPZFU8qwM9E3RYyQlGvY5iHvEKo,2688
-agenta/sdk/tracing/inline.py,sha256=GI2jo4R9hrKGpiUnQ6nfWNR5DR_kxl2PdEnW23Jc7UU,31354
+agenta/sdk/tracing/exporters.py,sha256=PuSbamuLa4e9eGBM6tp7Smh5R41Jz07VBoBtMnkbtz8,2687
+agenta/sdk/tracing/inline.py,sha256=BQecJkZDlR85aodfvye-LuDhouee3RjLuUl7GMpfsSw,31282
 agenta/sdk/tracing/processors.py,sha256=d7MvJ_DRAP0RAlp8V8XrrY-m8pJ03nLxKoq15RnjedA,3334
 agenta/sdk/tracing/spans.py,sha256=nqUOjjirBxB8Eacv8Qj4Ra_6rknGi3lbJdNyKmk5ODQ,3707
 agenta/sdk/tracing/tracing.py,sha256=iZCIggUkR5f2zVcq0o7T8-to7pcFO349gUEBYXnBMzg,6888
-agenta/sdk/types.py,sha256=_lGsGSEaZJrUT4cVcT3zSpgEqex2jFaPtfpFeUEetbc,7247
+agenta/sdk/types.py,sha256=HvFok4lEFxWow6WT71OrJ23WhTsj_FqqMyzlKByAqYk,19120
 agenta/sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/sdk/utils/constants.py,sha256=zW3R4rjXOo2L5lz6q84l_zYuOM9u4mpPRHw_B1Dr_hI,67
 agenta/sdk/utils/costs.py,sha256=i8C7ud__pThLS55XkN4YW8czXtGeXr2mx7jjcOFeiXg,5955
@@ -251,7 +256,7 @@ agenta/templates/simple_prompt/app.py,sha256=kODgF6lhzsaJPdgL5b21bUki6jkvqjWZzWR
 agenta/templates/simple_prompt/env.example,sha256=g9AE5bYcGPpxawXMJ96gh8oenEPCHTabsiOnfQo3c5k,70
 agenta/templates/simple_prompt/requirements.txt,sha256=ywRglRy7pPkw8bljmMEJJ4aOOQKrt9FGKULZ-DGkoBU,23
 agenta/templates/simple_prompt/template.toml,sha256=DQBtRrF4GU8LBEXOZ-GGuINXMQDKGTEG5y37tnvIUIE,60
-agenta-0.32.0.dist-info/METADATA,sha256=4iT189ml4TdEIGZIDJQlVhxZluAI2GZDSjAX8HYvGaQ,29623
-agenta-0.32.0.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-agenta-0.32.0.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.32.0.dist-info/RECORD,,
+agenta-0.32.0a2.dist-info/METADATA,sha256=iDQKOgsr6EsCMTJDm9G6PxNZJK11iqyKVY_aXiF5jD4,29625
+agenta-0.32.0a2.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+agenta-0.32.0a2.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.32.0a2.dist-info/RECORD,,