agenta 0.35.1__tar.gz → 0.36.0__tar.gz

{agenta-0.35.1 → agenta-0.36.0}/PKG-INFO

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

{agenta-0.35.1 → agenta-0.36.0}/agenta/sdk/decorators/routing.py

@@ -674,9 +674,6 @@ class entrypoint:
         """Override config in OpenAPI schema to add agenta-specific metadata."""
         endpoint = endpoint[1:].replace("/", "_")
 
-        schema_key = f"Body_{func_name}_{endpoint}_post"
-        schema_to_override = openapi_schema["components"]["schemas"][schema_key]
-
         # Get the config class name to find its schema
         config_class_name = type(config).__name__
         config_schema = openapi_schema["components"]["schemas"][config_class_name]

{agenta-0.35.1 → agenta-0.36.0}/agenta/sdk/litellm/mocks/__init__.py

@@ -26,20 +26,16 @@ def hello_mock_response(*args, **kwargs) -> MockResponseModel:
         ],
     )
 
+
 def chat_mock_response(*args, **kwargs) -> MockResponseModel:
     return MockResponseModel(
         choices=[
-            MockChoiceModel(
-                message=MockMessageModel(
-                    content="world",
-                    role="assistant"
-                )
-            )
+            MockChoiceModel(message=MockMessageModel(content="world", role="assistant"))
         ],
     )
 
 
 MOCKS: dict[str, Callable[..., MockResponseModel]] = {
     "hello": hello_mock_response,
-    "chat": chat_mock_response
+    "chat": chat_mock_response,
 }

{agenta-0.35.1 → agenta-0.36.0}/agenta/sdk/middleware/vault.py

@@ -102,7 +102,7 @@ class VaultMiddleware(BaseHTTPMiddleware):
                     ),
                 )
 
-                local_secrets.append(secret.model_dump())
+                local_secrets.append(secret.model_dump())  # type: ignore
         except:  # pylint: disable=bare-except
             display_exception("Vault: Local Secrets Exception")
 
@@ -121,7 +121,7 @@ class VaultMiddleware(BaseHTTPMiddleware):
                 else:
                     secrets = response.json()
                     vault_secrets = self._transform_secrets_response_to_secret_dto(
-                        secrets
+                        secrets_list=secrets  # type: ignore
                     )
         except:  # pylint: disable=bare-except
             display_exception("Vault: Vault Secrets Exception")
@@ -130,12 +130,12 @@ class VaultMiddleware(BaseHTTPMiddleware):
 
         if local_secrets:
             for secret in local_secrets:
-                provider = secret["data"]["provider"]
+                provider = secret["data"]["provider"]  # type: ignore
                 merged_secrets[provider] = secret
 
         if vault_secrets:
             for secret in vault_secrets:
-                provider = secret["data"]["provider"]
+                provider = secret["data"]["provider"]  # type: ignore
                 merged_secrets[provider] = secret
 
         secrets = list(merged_secrets.values())

{agenta-0.35.1 → agenta-0.36.0}/pyproject.toml

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