agenta 0.59.7__tar.gz → 0.59.9__tar.gz

{agenta-0.59.7 → agenta-0.59.9}/PKG-INFO

@@ -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 → agenta-0.59.9}/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-0.59.7 → agenta-0.59.9}/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-0.59.7 → agenta-0.59.9}/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 → agenta-0.59.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "agenta"
-version = "0.59.7"
+version = "0.59.9"
 description = "The SDK for agenta is an open-source LLMOps platform."
 readme = "README.md"
 authors = [