agenta 0.57.0__py3-none-any.whl → 0.65.0__py3-none-any.whl

agenta/sdk/utils/client.py

@@ -0,0 +1,38 @@
+import requests
+
+BASE_TIMEOUT = 10
+
+from agenta.sdk.utils.logging import get_module_logger
+
+import agenta as ag
+
+log = get_module_logger(__name__)
+
+
+def authed_api():
+    """
+    Preconfigured requests for authenticated endpoints (supports all methods).
+    """
+
+    api_url = ag.DEFAULT_AGENTA_SINGLETON_INSTANCE.api_url
+    api_key = ag.DEFAULT_AGENTA_SINGLETON_INSTANCE.api_key
+
+    if not api_url or not api_key:
+        log.error("Please call ag.init() first.")
+        log.error("And don't forget to set AGENTA_API_URL and AGENTA_API_KEY.")
+        raise ValueError("API URL and API Key must be set.")
+
+    def _request(method: str, endpoint: str, **kwargs):
+        url = f"{api_url}{endpoint}"
+        headers = kwargs.pop("headers", {})
+        headers.setdefault("Authorization", f"ApiKey {api_key}")
+
+        return requests.request(
+            method=method,
+            url=url,
+            headers=headers,
+            timeout=BASE_TIMEOUT,
+            **kwargs,
+        )
+
+    return _request

agenta/sdk/utils/helpers.py

@@ -20,25 +20,26 @@ def parse_url(url: str) -> str:
         str: The parsed or rewritten URL suitable for the current environment and Docker network mode.
     """
 
-    # Normalize: remove trailing slash and /api suffix
     url = url.rstrip("/")
-    if url.endswith("/api"):
-        url = url[: -len("/api")]
 
-    if "localhost" not in url:
+    if "localhost" not in url and "0.0.0.0" not in url:
         return url
 
-    internal_url = os.getenv("AGENTA_API_INTERNAL_URL")
-    if internal_url:
-        return internal_url
-
     docker_network_mode = os.getenv("DOCKER_NETWORK_MODE")
 
     if docker_network_mode and docker_network_mode.lower() == "bridge":
-        return url.replace("localhost", "host.docker.internal")
-
-    if not docker_network_mode or docker_network_mode.lower() == "host":
+        return url.replace(
+            "localhost",
+            "host.docker.internal",
+        ).replace(
+            "0.0.0.0",
+            "host.docker.internal",
+        )
+
+    if (
+        not docker_network_mode
+        or (docker_network_mode and docker_network_mode.lower()) == "host"
+    ):
         return url
 
-    # For any other network mode, return the URL unchanged
     return url

agenta/sdk/utils/logging.py

@@ -8,15 +8,6 @@ import logging
 import structlog
 from structlog.typing import EventDict, WrappedLogger, Processor
 
-# from datetime import datetime
-# from logging.handlers import RotatingFileHandler
-
-# from opentelemetry.trace import get_current_span
-# from opentelemetry._logs import set_logger_provider
-# from opentelemetry.sdk._logs import LoggingHandler, LoggerProvider
-# from opentelemetry.sdk._logs.export import BatchLogRecordProcessor
-# from opentelemetry.exporter.otlp.proto.http._log_exporter import OTLPLogExporter
-
 TRACE_LEVEL = 1
 logging.TRACE = TRACE_LEVEL
 logging.addLevelName(TRACE_LEVEL, "TRACE")
@@ -40,15 +31,6 @@ structlog.stdlib.BoundLogger.trace = bound_logger_trace
 AGENTA_LOG_CONSOLE_ENABLED = os.getenv("AGENTA_LOG_CONSOLE_ENABLED", "true") == "true"
 AGENTA_LOG_CONSOLE_LEVEL = os.getenv("AGENTA_LOG_CONSOLE_LEVEL", "TRACE").upper()
 
-# AGENTA_LOG_OTLP_ENABLED = os.getenv("AGENTA_LOG_OTLP_ENABLED", "false") == "true"
-# AGENTA_LOG_OTLP_LEVEL = os.getenv("AGENTA_LOG_OTLP_LEVEL", "INFO").upper()
-
-# AGENTA_LOG_FILE_ENABLED = os.getenv("AGENTA_LOG_FILE_ENABLED", "true") == "true"
-# AGENTA_LOG_FILE_LEVEL = os.getenv("AGENTA_LOG_FILE_LEVEL", "WARNING").upper()
-# AGENTA_LOG_FILE_BASE = os.getenv("AGENTA_LOG_FILE_PATH", "error")
-# LOG_FILE_DATE = datetime.utcnow().strftime("%Y-%m-%d")
-# AGENTA_LOG_FILE_PATH = f"{AGENTA_LOG_FILE_BASE}-{LOG_FILE_DATE}.log"
-
 # COLORS
 LEVEL_COLORS = {
     "TRACE": "\033[97m",
@@ -88,15 +70,6 @@ def process_positional_args(_, __, event_dict: EventDict) -> EventDict:
     return event_dict
 
 
-# def add_trace_context(_, __, event_dict: EventDict) -> EventDict:
-#     span = get_current_span()
-#     if span and span.get_span_context().is_valid:
-#         ctx = span.get_span_context()
-#         event_dict["TraceId"] = format(ctx.trace_id, "032x")
-#         event_dict["SpanId"] = format(ctx.span_id, "016x")
-#     return event_dict
-
-
 def add_logger_info(
     logger: WrappedLogger, method_name: str, event_dict: EventDict
 ) -> EventDict:
@@ -143,36 +116,9 @@ def colored_console_renderer() -> Processor:
     return render
 
 
-# def plain_renderer() -> Processor:
-#     hidden = {
-#         "SeverityText",
-#         "SeverityNumber",
-#         "MethodName",
-#         "logger_factory",
-#         "LoggerName",
-#         "level",
-#     }
-
-#     def render(_, __, event_dict: EventDict) -> str:
-#         ts = event_dict.pop("Timestamp", "")[:23] + "Z"
-#         level = event_dict.get("level", "")
-#         msg = event_dict.pop("event", "")
-#         padded = f"[{level:<5}]"
-#         logger = f"[{event_dict.pop('logger', '')}]"
-#         extras = " ".join(f"{k}={v}" for k, v in event_dict.items() if k not in hidden)
-#         return f"{ts} {padded} {msg} {logger} {extras}"
-
-#     return render
-
-
-# def json_renderer() -> Processor:
-#     return structlog.processors.JSONRenderer()
-
-
 SHARED_PROCESSORS: list[Processor] = [
     structlog.processors.TimeStamper(fmt="iso", utc=True, key="Timestamp"),
     process_positional_args,
-    # add_trace_context,
     add_logger_info,
     structlog.processors.format_exc_info,
     structlog.processors.dict_tracebacks,
@@ -193,35 +139,29 @@ def create_struct_logger(
     )
 
 
+# Guard against double initialization
+_LOGGING_CONFIGURED = False
+
 # CONFIGURE HANDLERS AND STRUCTLOG LOGGERS
 handlers = []
 loggers = []
 
-if AGENTA_LOG_CONSOLE_ENABLED:
-    h = logging.StreamHandler(sys.stdout)
-    h.setLevel(getattr(logging, AGENTA_LOG_CONSOLE_LEVEL, TRACE_LEVEL))
-    h.setFormatter(logging.Formatter("%(message)s"))
-    logging.getLogger("console").addHandler(h)
-    loggers.append(create_struct_logger([colored_console_renderer()], "console"))
+if AGENTA_LOG_CONSOLE_ENABLED and not _LOGGING_CONFIGURED:
+    _LOGGING_CONFIGURED = True
+
+    # Check if console logger already has handlers (from OSS module)
+    console_logger = logging.getLogger("console")
 
-# if AGENTA_LOG_FILE_ENABLED:
-#     h = RotatingFileHandler(AGENTA_LOG_FILE_PATH, maxBytes=10 * 1024 * 1024, backupCount=5)
-#     h.setLevel(getattr(logging, AGENTA_LOG_FILE_LEVEL, logging.WARNING))
-#     h.setFormatter(logging.Formatter("%(message)s"))
-#     logging.getLogger("file").addHandler(h)
-#     loggers.append(create_struct_logger([plain_renderer()], "file"))
-
-# if AGENTA_LOG_OTLP_ENABLED:
-#     provider = LoggerProvider()
-#     exporter = OTLPLogExporter()
-#     provider.add_log_record_processor(BatchLogRecordProcessor(exporter))
-#     set_logger_provider(provider)
-#     h = LoggingHandler(
-#         level=getattr(logging, AGENTA_LOG_OTLP_LEVEL, logging.INFO), logger_provider=provider
-#     )
-#     h.setFormatter(logging.Formatter("%(message)s"))
-#     logging.getLogger("otel").addHandler(h)
-#     loggers.append(create_struct_logger([json_renderer()], "otel"))
+    if not console_logger.handlers:
+        # Only add handler if it doesn't exist yet
+        h = logging.StreamHandler(sys.stdout)
+        h.setLevel(getattr(logging, AGENTA_LOG_CONSOLE_LEVEL, TRACE_LEVEL))
+        h.setFormatter(logging.Formatter("%(message)s"))
+        console_logger.addHandler(h)
+        console_logger.setLevel(TRACE_LEVEL)
+        console_logger.propagate = False
+
+    loggers.append(create_struct_logger([colored_console_renderer()], "console"))
 
 
 class MultiLogger:

agenta/sdk/utils/references.py

@@ -0,0 +1,23 @@
+from uuid import UUID
+import re
+import unicodedata
+
+
+def get_slug_from_name_and_id(
+    name: str,
+    id: UUID,  # pylint: disable=redefined-builtin
+) -> str:
+    # Normalize Unicode (e.g., é → e)
+    name = unicodedata.normalize("NFKD", name)
+    # Remove non-ASCII characters
+    name = name.encode("ascii", "ignore").decode("ascii")
+    # Lowercase and remove non-word characters except hyphens and spaces
+    name = re.sub(r"[^\w\s-]", "", name.lower())
+    # Replace any sequence of hyphens or whitespace with a single hyphen
+    name = re.sub(r"[-\s]+", "-", name)
+    # Trim leading/trailing hyphens
+    name = name.strip("-")
+    # Last 12 characters of the ID
+    slug = f"{name}-{id.hex[-12:]}"
+
+    return slug.lower()