agenta 0.59.7__py3-none-any.whl → 0.59.9__py3-none-any.whl

agenta/sdk/assets.py

@@ -1,5 +1,5 @@
 supported_llm_models = {
-    "Anthropic": [
+    "anthropic": [
         "anthropic/claude-sonnet-4-5",
         "anthropic/claude-haiku-4-5",
         "anthropic/claude-opus-4-1",
@@ -15,18 +15,18 @@ supported_llm_models = {
         "anthropic/claude-2.1",
         "anthropic/claude-2",
     ],
-    "Cohere": [
+    "cohere": [
         "cohere/command-light",
         "cohere/command-r-plus",
         "cohere/command-nightly",
     ],
-    "DeepInfra": [
+    "deepinfra": [
         "deepinfra/meta-llama/Llama-2-70b-chat-hf",
         "deepinfra/meta-llama/Llama-2-13b-chat-hf",
         "deepinfra/codellama/CodeLlama-34b-Instruct-hf",
         "deepinfra/mistralai/Mistral-7B-Instruct-v0.1",
     ],
-    "Gemini": [
+    "gemini": [
         "gemini/gemini-2.5-pro",
         "gemini/gemini-2.5-pro-preview-05-06",
         "gemini/gemini-2.5-flash",
@@ -40,7 +40,7 @@ supported_llm_models = {
         "gemini/gemini-2.0-flash-lite",
         "gemini/gemini-2.0-flash-lite-preview-02-05",
     ],
-    "Groq": [
+    "groq": [
         "groq/deepseek-r1-distill-llama-70b",
         "groq/deepseek-r1-distill-llama-70b-specdec",
         "groq/gemma2-9b-it",
@@ -56,13 +56,13 @@ supported_llm_models = {
         "groq/llama3-8b-8192",
         "groq/mixtral-8x7b-32768",
     ],
-    "Mistral": [
+    "mistral": [
         "mistral/mistral-tiny",
         "mistral/mistral-small",
         "mistral/mistral-medium",
         "mistral/mistral-large-latest",
     ],
-    "OpenAI": [
+    "openai": [
         "gpt-5",
         "gpt-5-mini",
         "gpt-5-nano",
@@ -78,7 +78,7 @@ supported_llm_models = {
         "gpt-4.1-nano",
         "o4-mini",
     ],
-    "OpenRouter": [
+    "openrouter": [
         "openrouter/qwen/qwen3-235b-a22b",
         "openrouter/qwen/qwen3-32b",
         "openrouter/qwen/qwen3-30b-a3b",
@@ -148,13 +148,13 @@ supported_llm_models = {
         "openrouter/google/gemini-2.0-flash-001",
         "openrouter/perplexity/sonar-reasoning",
     ],
-    "Perplexity AI": [
+    "perplexity": [
         "perplexity/sonar",
         "perplexity/sonar-pro",
         "perplexity/sonar-reasoning",
         "perplexity/sonar-reasoning-pro",
     ],
-    "togetherai": [
+    "together_ai": [
         "together_ai/deepseek-ai/DeepSeek-R1",
         "together_ai/deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
         "together_ai/deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B",

agenta/sdk/managers/secrets.py

@@ -119,11 +119,17 @@ class SecretsManager:
         # The reason is that custom providers are in fact openai compatible providers
         # They need to be passed in litellm as openai/modelname
 
-        if "custom" in model:
-            modified_model = model.replace(f"{provider_slug}/custom/", "openai/")
-            return modified_model.replace(f"{provider_slug}/", "")
+        modified_model = model
 
-        return model.replace(f"{provider_slug}/", "")
+        if "custom" in modified_model:
+            modified_model = modified_model.replace(
+                f"{provider_slug}/custom/", "openai/"
+            )
+
+        if provider_slug:
+            modified_model = modified_model.replace(f"{provider_slug}/", "")
+
+        return modified_model
 
     @staticmethod
     def get_provider_settings(model: str) -> Optional[Dict]:
@@ -137,6 +143,8 @@ class SecretsManager:
             Dict: A dictionary containing all parameters needed for litellm.completion
         """
 
+        request_provider_model = model
+
         # STEP 1: get vault secrets from route context and transform it
         secrets = SecretsManager.get_from_route()
         if not secrets:
@@ -146,11 +154,11 @@ class SecretsManager:
         secrets = SecretsManager._parse_secrets(secrets=secrets)
 
         # STEP 2: check model exists in supported standard models
-        provider = _standard_providers.get(model)
+        provider = _standard_providers.get(request_provider_model)
         if not provider:
             # check and get provider kind if model exists in custom provider models
             provider = SecretsManager._custom_providers_get(
-                model=model,
+                model=request_provider_model,
                 secrets=secrets,
             )
 
@@ -159,16 +167,19 @@ class SecretsManager:
             return None
 
         # STEP 2c: get litellm compatible model
-        provider_slug = SecretsManager._custom_provider_slug_get(
-            model=model, secrets=secrets
+        request_provider_slug = (
+            SecretsManager._custom_provider_slug_get(
+                model=request_provider_model, secrets=secrets
+            )
+            or ""
         )
-        model = SecretsManager._get_compatible_model(
-            model=model, provider_slug=provider_slug
+        compatible_provider_model = SecretsManager._get_compatible_model(
+            model=request_provider_model, provider_slug=request_provider_slug
         )
 
         # STEP 3: initialize provider settings and simplify provider name
-        provider_settings = {"model": model}
-        provider_name = re.sub(
+        provider_settings = dict(model=compatible_provider_model)
+        request_provider_kind = re.sub(
             r"[\s-]+", "", provider.lower()
         )  # normalizing other special characters too (azure-openai)
 
@@ -180,9 +191,9 @@ class SecretsManager:
             # i). Extract API key if present
             # (for standard models -- openai/anthropic/gemini, etc)
             if secret.get("kind") == "provider_key":
-                provider_kind = secret_data.get("kind", "")
+                secret_provider_kind = secret_data.get("kind", "")
 
-                if provider_kind == provider_name:
+                if request_provider_kind == secret_provider_kind:
                     if "key" in provider_info:
                         provider_settings["api_key"] = provider_info["key"]
                 continue
@@ -190,13 +201,20 @@ class SecretsManager:
             # ii). Extract Credentials if present
             # (for custom providers -- aws bedrock/sagemaker, vertex_ai, etc)
             elif secret.get("kind") == "custom_provider":
-                provider_kind = provider_info.get("kind", "").lower().replace(" ", "")
-                provider_slug = secret_data.get("provider_slug", "")
-                provider_extras = provider_info.get("extras", {})
-
-                if provider_kind == provider_name or provider_slug == provider_name:
-                    if provider_extras:
-                        provider_settings.update(provider_extras)
+                secret_provider_kind = (
+                    provider_info.get("kind", "").lower().replace(" ", "")
+                )
+                secret_provider_slug = secret_data.get("provider_slug", "")
+                secret_provider_models = secret_data.get("models", "")
+                secret_provider_extras = provider_info.get("extras", {})
+
+                if (
+                    request_provider_kind == secret_provider_kind
+                    and request_provider_slug == secret_provider_slug
+                    and request_provider_model in secret_provider_models
+                ):
+                    if secret_provider_extras:
+                        provider_settings.update(secret_provider_extras)
                 continue
 
         return provider_settings

agenta/sdk/middleware/vault.py

@@ -85,7 +85,7 @@ class VaultMiddleware(BaseHTTPMiddleware):
                     continue
 
                 secret = SecretDTO(
-                    kind="provider_kind",  # type: ignore
+                    kind="provider_key",  # type: ignore
                     data=StandardProviderDTO(
                         kind=provider,
                         provider=StandardProviderSettingsDTO(key=key),

{agenta-0.59.7.dist-info → agenta-0.59.9.dist-info}/METADATA

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

{agenta-0.59.7.dist-info → agenta-0.59.9.dist-info}/RECORD

@@ -307,7 +307,7 @@ agenta/config.py,sha256=0VrTqduB4g8Mt_Ll7ffFcEjKF5qjTUIxmUtTPW2ygWw,653
 agenta/config.toml,sha256=sIORbhnyct2R9lJrquxhNL4pHul3O0R7iaipCoja5MY,193
 agenta/sdk/__init__.py,sha256=6LWzIVeXpkBKgjIXyOFOJwSDDcAf9JC_3kmxaBxjB_w,2170
 agenta/sdk/agenta_init.py,sha256=Jkfm5Fs5qy1gMY_AWiMhr_EnvVBddNRqYb8GtN19kkc,6610
-agenta/sdk/assets.py,sha256=Np_zZbnRUSws-NxlrW1mMLyJPGM8Y-uKN6Q8Qd1dW_o,8643
+agenta/sdk/assets.py,sha256=GRnOTMcNqMNCVtOviD3xnuXrEM8k1j3i9-U0ZjHv6o4,8641
 agenta/sdk/context/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/sdk/context/running.py,sha256=3gEuUdQrJwcuN93MlXFZ6aHXNxUW6dUk_EudgaxOkCU,907
 agenta/sdk/context/serving.py,sha256=f0Iq8Sv_CBJF7838tDecy_89K6HI7k6joWBkWHTM7yE,598
@@ -324,7 +324,7 @@ agenta/sdk/managers/__init__.py,sha256=SN-LRwG0pRRDV3u2Q4JiiSTigN3-mYpzGNM35RzT4
 agenta/sdk/managers/apps.py,sha256=BeAlTJlOOM0Wh_XWgjve65Mt-LwFgib_swu-1wvGQi4,2250
 agenta/sdk/managers/config.py,sha256=yyFxjZVf3_XfNAkZpl-qWzljYz4XB8LgPYc55JXEwG0,7494
 agenta/sdk/managers/deployment.py,sha256=SEokjZeh6n7HRKZ92Y0WncdG49hIFx-Z3B3HAl2kmUg,1174
-agenta/sdk/managers/secrets.py,sha256=3fd-WKJo2vs8dA_Nk4NLqf8RYG9FcvFTrnSmKBQqSm0,7459
+agenta/sdk/managers/secrets.py,sha256=pD0WIYfCz-OvSk7RGubHwj7mxAMREZhkF-CluAta_XU,8107
 agenta/sdk/managers/shared.py,sha256=kI3w74E4rS3YDORG2e-_r0Pz5KslTRzRBK7vgyOeaJA,21559
 agenta/sdk/managers/variant.py,sha256=A5ga3mq3b0weUTXa9HO72MGaspthGcu1uK9K5OnP738,4172
 agenta/sdk/managers/vault.py,sha256=tOJsRWSEpmNT_ysWVcK5KQMdQO4pwiK90H3rsr9BUbg,337
@@ -338,7 +338,7 @@ agenta/sdk/middleware/flags.py,sha256=okku9RBmcD4Qka8XIsVqtroMDg-9IXevUwDiAA9whJ
 agenta/sdk/middleware/inline.py,sha256=ee8E4XBGcRSrHTvblqX1yRXuTN_sxLm7lY1jnywrBG8,901
 agenta/sdk/middleware/mock.py,sha256=bCUN9iJBxePyN9MBwBpJs-_iCNkUQeUjIIu3WElS1oQ,759
 agenta/sdk/middleware/otel.py,sha256=lHzhGUv4fq2RPuTPH2keJ16v-_cBUrLrTqWzHUmEVdI,1041
-agenta/sdk/middleware/vault.py,sha256=sF2CmWQdCvXJcmZppT4duV0KgqHW4c4wg4Fs_i1SvNs,4059
+agenta/sdk/middleware/vault.py,sha256=HPiEV7mqq96haUG_fzaI_3QphbS-U80kduH2PPpIyHA,4058
 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=Brqle9hGV3DaEJjeYCBq7MDlbvHMAIwmkUj2Lni8dF4,5563
@@ -366,6 +366,6 @@ agenta/sdk/workflows/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 agenta/sdk/workflows/registry.py,sha256=4FRSeU4njMmP6xFCIteF5f_W6NVlqFTx1AM7hsaGAQk,975
 agenta/sdk/workflows/types.py,sha256=SjYeT8FWVgwaIIC8sI3fRjKERLEA_oxuBGvYSaFqNg8,11720
 agenta/sdk/workflows/utils.py,sha256=ILfY8DSBWLrdWIuKg6mq7rANwKiiY6sxEeFiBFhjLYM,413
-agenta-0.59.7.dist-info/METADATA,sha256=2uIEfhZCpN4AKWkMHiku3SLW64jWheVmOmjuO26_x4Q,31792
-agenta-0.59.7.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-agenta-0.59.7.dist-info/RECORD,,
+agenta-0.59.9.dist-info/METADATA,sha256=A-GxoCEoeZ_1nwo38s1ChmRQ2kUue7ECCDhjTMTk61k,31792
+agenta-0.59.9.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+agenta-0.59.9.dist-info/RECORD,,