agenta 0.30.0a2__py3-none-any.whl → 0.30.0a4__py3-none-any.whl

agenta/sdk/agenta_init.py

@@ -6,8 +6,9 @@ from importlib.metadata import version
 from agenta.sdk.utils.logging import log
 from agenta.sdk.utils.globals import set_global
 from agenta.client.backend.client import AgentaApi, AsyncAgentaApi
+
 from agenta.sdk.tracing import Tracing
-from agenta.client.exceptions import APIRequestError
+from agenta.sdk.context.routing import routing_context
 
 
 class AgentaSingleton:
@@ -59,9 +60,7 @@ class AgentaSingleton:
             ValueError: If `app_id` is not specified either as an argument, in the config file, or in the environment variables.
         """
 
-        log.info("---------------------------")
-        log.info("Agenta SDK - using version: %s", version("agenta"))
-        log.info("---------------------------")
+        log.info("Agenta - SDK version: %s", version("agenta"))
 
         config = {}
         if config_fname:
@@ -86,6 +85,13 @@ class AgentaSingleton:
 
         self.api_key = api_key or getenv("AGENTA_API_KEY") or config.get("api_key")
 
+        self.base_id = getenv("AGENTA_BASE_ID")
+
+        self.service_id = getenv("AGENTA_SERVICE_ID") or self.base_id
+
+        log.info("Agenta - Service ID: %s", self.service_id)
+        log.info("Agenta - Application ID: %s", self.app_id)
+
         self.tracing = Tracing(
             url=f"{self.host}/api/observability/v1/otlp/traces",  # type: ignore
             redact=redact,
@@ -94,6 +100,7 @@ class AgentaSingleton:
 
         self.tracing.configure(
             api_key=self.api_key,
+            service_id=self.service_id,
             # DEPRECATING
             app_id=self.app_id,
         )
@@ -108,8 +115,6 @@ class AgentaSingleton:
             api_key=self.api_key if self.api_key else "",
         )
 
-        self.base_id = getenv("AGENTA_BASE_ID")
-
         self.config = Config(
             host=self.host,
             base_id=self.base_id,
@@ -120,28 +125,43 @@ class AgentaSingleton:
 class Config:
     def __init__(
         self,
-        host: str,
+        # LEGACY
+        host: Optional[str] = None,
         base_id: Optional[str] = None,
-        api_key: Optional[str] = "",
+        api_key: Optional[str] = None,
+        # LEGACY
+        **kwargs,
     ):
-        self.host = host
+        self.default_parameters = {**kwargs}
+
+    def set_default(self, **kwargs):
+        self.default_parameters.update(kwargs)
+
+    def get_default(self):
+        return self.default_parameters
+
+    def __getattr__(self, key):
+        context = routing_context.get()
+
+        parameters = context.parameters
+
+        if not parameters:
+            return None
+
+        if key in parameters:
+            value = parameters[key]
+
+            if isinstance(value, dict):
+                nested_config = Config()
+                nested_config.set_default(**value)
 
-        self.base_id = base_id
+                return nested_config
 
-        if self.base_id is None:
-            # print(
-            #    "Warning: Your configuration will not be saved permanently since base_id is not provided.\n"
-            # )
-            pass
+            return value
 
-        if base_id is None or host is None:
-            self.persist = False
-        else:
-            self.persist = True
-            self.client = AgentaApi(
-                base_url=self.host + "/api",
-                api_key=api_key if api_key else "",
-            )
+        return None
+
+    ### --- LEGACY --- ###
 
     def register_default(self, overwrite=False, **kwargs):
         """alias for default"""
@@ -153,104 +173,13 @@ class Config:
             overwrite: Whether to overwrite the existing configuration or not
             **kwargs: A dict containing the parameters
         """
-        self.set(
-            **kwargs
-        )  # In case there is no connectivity, we still can use the default values
-        try:
-            self.push(config_name="default", overwrite=overwrite, **kwargs)
-        except Exception as ex:
-            log.warning(
-                "Unable to push the default configuration to the server. %s", str(ex)
-            )
-
-    def push(self, config_name: str, overwrite=True, **kwargs):
-        """Pushes the parameters for the app variant to the server
-        Args:
-            config_name: Name of the configuration to push to
-            overwrite: Whether to overwrite the existing configuration or not
-            **kwargs: A dict containing the parameters
-        """
-        if not self.persist:
-            return
-        try:
-            self.client.configs.save_config(
-                base_id=self.base_id,
-                config_name=config_name,
-                parameters=kwargs,
-                overwrite=overwrite,
-            )
-        except Exception as ex:
-            log.warning(
-                "Failed to push the configuration to the server with error: %s", ex
-            )
-
-    def pull(
-        self, config_name: str = "default", environment_name: Optional[str] = None
-    ):
-        """Pulls the parameters for the app variant from the server and sets them to the config"""
-        if not self.persist and (
-            config_name != "default" or environment_name is not None
-        ):
-            raise ValueError(
-                "Cannot pull the configuration from the server since the app_name and base_name are not provided."
-            )
-        if self.persist:
-            try:
-                if environment_name:
-                    config = self.client.configs.get_config(
-                        base_id=self.base_id, environment_name=environment_name
-                    )
-
-                else:
-                    config = self.client.configs.get_config(
-                        base_id=self.base_id,
-                        config_name=config_name,
-                    )
-            except Exception as ex:
-                log.warning(
-                    "Failed to pull the configuration from the server with error: %s",
-                    str(ex),
-                )
-        try:
-            self.set(**{"current_version": config.current_version, **config.parameters})
-        except Exception as ex:
-            log.warning("Failed to set the configuration with error: %s", str(ex))
+        self.set(**kwargs)
 
-    def all(self):
-        """Returns all the parameters for the app variant"""
-        return {
-            k: v
-            for k, v in self.__dict__.items()
-            if k
-            not in [
-                "app_name",
-                "base_name",
-                "host",
-                "base_id",
-                "api_key",
-                "persist",
-                "client",
-            ]
-        }
-
-    # function to set the parameters for the app variant
     def set(self, **kwargs):
-        """Sets the parameters for the app variant
-
-        Args:
-            **kwargs: A dict containing the parameters
-        """
-        for key, value in kwargs.items():
-            setattr(self, key, value)
-
-    def dump(self):
-        """Returns all the information about the current version in the configuration.
+        self.set_default(**kwargs)
 
-        Raises:
-            NotImplementedError: _description_
-        """
-
-        raise NotImplementedError()
+    def all(self):
+        return self.default_parameters
 
 
 def init(

agenta/sdk/assets.py

@@ -1,23 +1,9 @@
 supported_llm_models = {
-    "Mistral AI": [
-        "mistral/mistral-tiny",
-        "mistral/mistral-small",
-        "mistral/mistral-medium",
-        "mistral/mistral-large-latest",
-    ],
-    "Open AI": [
-        "gpt-3.5-turbo-1106",
-        "gpt-3.5-turbo",
-        "gpt-4",
-        "gpt-4o",
-        "gpt-4o-mini",
-        "gpt-4-1106-preview",
-    ],
-    "Gemini": ["gemini/gemini-1.5-pro-latest", "gemini/gemini-1.5-flash"],
-    "Cohere": [
-        "cohere/command-light",
-        "cohere/command-r-plus",
-        "cohere/command-nightly",
+    "Aleph Alpha": [
+        "luminous-base",
+        "luminous-base-control",
+        "luminous-extended-control",
+        "luminous-supreme",
     ],
     "Anthropic": [
         "anthropic/claude-3-5-sonnet-20240620",
@@ -33,11 +19,10 @@ supported_llm_models = {
         "anyscale/meta-llama/Llama-2-13b-chat-hf",
         "anyscale/meta-llama/Llama-2-70b-chat-hf",
     ],
-    "Perplexity AI": [
-        "perplexity/pplx-7b-chat",
-        "perplexity/pplx-70b-chat",
-        "perplexity/pplx-7b-online",
-        "perplexity/pplx-70b-online",
+    "Cohere": [
+        "cohere/command-light",
+        "cohere/command-r-plus",
+        "cohere/command-nightly",
     ],
     "DeepInfra": [
         "deepinfra/meta-llama/Llama-2-70b-chat-hf",
@@ -46,6 +31,46 @@ supported_llm_models = {
         "deepinfra/mistralai/Mistral-7B-Instruct-v0.1",
         "deepinfra/jondurbin/airoboros-l2-70b-gpt4-1.4.1",
     ],
+    "Gemini": [
+        "gemini/gemini-1.5-pro-latest",
+        "gemini/gemini-1.5-flash",
+    ],
+    "Groq": [
+        "groq/llama3-8b-8192",
+        "groq/llama3-70b-8192",
+        "groq/llama2-70b-4096",
+        "groq/mixtral-8x7b-32768",
+        "groq/gemma-7b-it",
+    ],
+    "Mistral AI": [
+        "mistral/mistral-tiny",
+        "mistral/mistral-small",
+        "mistral/mistral-medium",
+        "mistral/mistral-large-latest",
+    ],
+    "Open AI": [
+        "gpt-3.5-turbo-1106",
+        "gpt-3.5-turbo",
+        "gpt-4",
+        "gpt-4o",
+        "gpt-4o-mini",
+        "gpt-4-1106-preview",
+    ],
+    "OpenRouter": [
+        "openrouter/openai/gpt-3.5-turbo",
+        "openrouter/openai/gpt-3.5-turbo-16k",
+        "openrouter/anthropic/claude-instant-v1",
+        "openrouter/google/palm-2-chat-bison",
+        "openrouter/google/palm-2-codechat-bison",
+        "openrouter/meta-llama/llama-2-13b-chat",
+        "openrouter/meta-llama/llama-2-70b-chat",
+    ],
+    "Perplexity AI": [
+        "perplexity/pplx-7b-chat",
+        "perplexity/pplx-70b-chat",
+        "perplexity/pplx-7b-online",
+        "perplexity/pplx-70b-online",
+    ],
     "Together AI": [
         "together_ai/togethercomputer/llama-2-70b-chat",
         "together_ai/togethercomputer/llama-2-70b",
@@ -59,26 +84,12 @@ supported_llm_models = {
         "together_ai/NousResearch/Nous-Hermes-Llama2-13b",
         "together_ai/Austism/chronos-hermes-13b",
     ],
-    "Aleph Alpha": [
-        "luminous-base",
-        "luminous-base-control",
-        "luminous-extended-control",
-        "luminous-supreme",
-    ],
-    "OpenRouter": [
-        "openrouter/openai/gpt-3.5-turbo",
-        "openrouter/openai/gpt-3.5-turbo-16k",
-        "openrouter/anthropic/claude-instant-v1",
-        "openrouter/google/palm-2-chat-bison",
-        "openrouter/google/palm-2-codechat-bison",
-        "openrouter/meta-llama/llama-2-13b-chat",
-        "openrouter/meta-llama/llama-2-70b-chat",
-    ],
-    "Groq": [
-        "groq/llama3-8b-8192",
-        "groq/llama3-70b-8192",
-        "groq/llama2-70b-4096",
-        "groq/mixtral-8x7b-32768",
-        "groq/gemma-7b-it",
-    ],
+}
+
+providers_list = list(supported_llm_models.keys())
+
+model_to_provider_mapping = {
+    model: provider
+    for provider, models in supported_llm_models.items()
+    for model in models
 }

agenta/sdk/context/exporting.py

@@ -0,0 +1,25 @@
+from typing import Optional
+
+from contextlib import contextmanager
+from contextvars import ContextVar
+
+from pydantic import BaseModel
+
+
+class ExportingContext(BaseModel):
+    credentials: Optional[str] = None
+
+
+exporting_context = ContextVar("exporting_context", default=ExportingContext())
+
+
+@contextmanager
+def exporting_context_manager(
+    *,
+    context: Optional[ExportingContext] = None,
+):
+    token = exporting_context.set(context)
+    try:
+        yield
+    finally:
+        exporting_context.reset(token)

agenta/sdk/context/routing.py

@@ -1,24 +1,24 @@
+from typing import Any, Dict, List, Optional
+
 from contextlib import contextmanager
 from contextvars import ContextVar
-from typing import Any, Dict, Optional
 
-routing_context = ContextVar("routing_context", default={})
+from pydantic import BaseModel
+
+
+class RoutingContext(BaseModel):
+    parameters: Optional[Dict[str, Any]] = None
+    secrets: Optional[List[Any]] = None
+
+
+routing_context = ContextVar("routing_context", default=RoutingContext())
 
 
 @contextmanager
 def routing_context_manager(
     *,
-    config: Optional[Dict[str, Any]] = None,
-    application: Optional[Dict[str, Any]] = None,
-    variant: Optional[Dict[str, Any]] = None,
-    environment: Optional[Dict[str, Any]] = None,
+    context: Optional[RoutingContext] = None,
 ):
-    context = {
-        "config": config,
-        "application": application,
-        "variant": variant,
-        "environment": environment,
-    }
     token = routing_context.set(context)
     try:
         yield

agenta/sdk/context/tracing.py

@@ -1,3 +1,28 @@
+from typing import Any, Dict, Optional
+
+from contextlib import contextmanager
 from contextvars import ContextVar
 
-tracing_context = ContextVar("tracing_context", default={})
+from pydantic import BaseModel
+
+
+class TracingContext(BaseModel):
+    credentials: Optional[str] = None
+    parameters: Optional[Dict[str, Any]] = None
+    references: Optional[Dict[str, Any]] = None
+    link: Optional[Dict[str, Any]] = None
+
+
+tracing_context = ContextVar("tracing_context", default=TracingContext())
+
+
+@contextmanager
+def tracing_context_manager(
+    *,
+    context: Optional[TracingContext] = None,
+):
+    token = tracing_context.set(context)
+    try:
+        yield
+    finally:
+        tracing_context.reset(token)