agenta 0.35.2__py3-none-any.whl → 0.36.0__py3-none-any.whl

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/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/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.2.dist-info → agenta-0.36.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agenta
-Version: 0.35.2
+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.2.dist-info → agenta-0.36.0.dist-info}/RECORD

@@ -205,12 +205,12 @@ agenta/sdk/context/exporting.py,sha256=16X8fgMhl58gehSlqANX97FiKxx4TkGiG4d2B0-7Z
 agenta/sdk/context/routing.py,sha256=FEsjw8EttI1SMyUo96ptcUsvHJnhoKwdr1szlkxxJNU,598
 agenta/sdk/context/tracing.py,sha256=xjErrXP1Nq1AfL-Cif1l-lNEfs12eQ3v_VCRgoKe7nY,743
 agenta/sdk/decorators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agenta/sdk/decorators/routing.py,sha256=a6llLTQTs-eGZ1023hyboNVJBeYcqI4zKEShRLvhTXk,25374
+agenta/sdk/decorators/routing.py,sha256=PqKopSYfgt_HGJcBn7-QveudZEKBQHcGrMiPZQ23gUo,25235
 agenta/sdk/decorators/tracing.py,sha256=tIi5gpwzvXd26pFW-7kYQtTFO8w1zzwFeAPY2Ldq4P4,9952
 agenta/sdk/litellm/__init__.py,sha256=Bpz1gfHQc0MN1yolWcjifLWznv6GjHggvRGQSpxpihM,37
 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=UvunEgoDcVobBZXrqTnHt9ZEjJEFtqFHGiSCKfNiuHU,952
+agenta/sdk/litellm/mocks/__init__.py,sha256=0bGjOOWXz8BxE0rvRWciPP7SAegO2mgi_UtWw8jTlDk,866
 agenta/sdk/managers/__init__.py,sha256=SN-LRwG0pRRDV3u2Q4JiiSTigN3-mYpzGNM35RzT4mc,238
 agenta/sdk/managers/apps.py,sha256=RSE5ER-a0IHo7iZrcQLshQZMemPTfAVsBfR8H1J-ryQ,1713
 agenta/sdk/managers/config.py,sha256=2LFl7aBF7FKd6UKhqcbnDXzlP8Rsj7hnCu5wBllVQzc,7753
@@ -226,7 +226,7 @@ agenta/sdk/middleware/cors.py,sha256=q3r7lGkrIdMcT_vuhsburMcjG7pyl7w0ycxrIrGJ2e8
 agenta/sdk/middleware/inline.py,sha256=ee8E4XBGcRSrHTvblqX1yRXuTN_sxLm7lY1jnywrBG8,901
 agenta/sdk/middleware/mock.py,sha256=bCUN9iJBxePyN9MBwBpJs-_iCNkUQeUjIIu3WElS1oQ,759
 agenta/sdk/middleware/otel.py,sha256=ShaB7iVaUjdq6lOXf0vmZthQsDUjAgIrseyPeJzVkEU,840
-agenta/sdk/middleware/vault.py,sha256=R4uLcGWuS0oigAjF-80jvqCScrF1_fpGpdYmA2D5A5A,4233
+agenta/sdk/middleware/vault.py,sha256=xGCDI0PXtDHaxz_Mv3ee3l_0JQgwnwTskzX5RTaYaBU,4310
 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
@@ -249,7 +249,7 @@ agenta/sdk/utils/logging.py,sha256=eFzEFuYpggfIhEKv09JZRqcDzkmZ482a_E2G-X0FK7Y,4
 agenta/sdk/utils/preinit.py,sha256=YlJL7RLfel0R7DFp-jK7OV-z4ZIQJM0oupYlk7g8b5o,1278
 agenta/sdk/utils/singleton.py,sha256=17Ph7LGnnV8HkPjImruKita2ni03Ari5jr0jqm__4sc,312
 agenta/sdk/utils/timing.py,sha256=rmBPSBuUnIu-epocUCVk0KcM2r36HuDoxkFqOZgfPhc,1507
-agenta-0.35.2.dist-info/METADATA,sha256=WCd4YnVt6pcc0AD6IC1CRYyWlRoAsqDqU_Cs8dWD6xE,29827
-agenta-0.35.2.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-agenta-0.35.2.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.35.2.dist-info/RECORD,,
+agenta-0.36.0.dist-info/METADATA,sha256=B-HORFIEKQe86deRWc1uE_pwfjwm5Fuu1jXulVmmZyo,29827
+agenta-0.36.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+agenta-0.36.0.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.36.0.dist-info/RECORD,,