agenta 0.85.2__tar.gz → 0.85.4__tar.gz

{agenta-0.85.2 → agenta-0.85.4}/PKG-INFO

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

{agenta-0.85.2 → agenta-0.85.4}/agenta/__init__.py

@@ -7,43 +7,42 @@ from .sdk import assets as assets
 
 # evaluations
 from .sdk import testsets as testsets
-from .sdk import tracer
 from .sdk.agenta_init import AgentaSingleton
 from .sdk.agenta_init import init as _init
-from .sdk.context.running import workflow_mode_enabled
+from .sdk.context.running import workflow_mode_enabled  # noqa: F401
 from .sdk.decorators.running import (
-    application,
-    evaluator,
-    workflow,
+    application,  # noqa: F401
+    evaluator,  # noqa: F401
+    workflow,  # noqa: F401
 )
-from .sdk.decorators.serving import app, route, create_app
-from .sdk.decorators.tracing import instrument
-from .sdk.managers.apps import AppManager
-from .sdk.managers.config import ConfigManager
-from .sdk.managers.deployment import DeploymentManager
-from .sdk.managers.secrets import SecretsManager
-from .sdk.managers.variant import VariantManager
-from .sdk.managers.vault import VaultManager
-from .sdk.tracing import Tracing, get_tracer
-from .sdk.tracing.conventions import Reference
+from .sdk.decorators.serving import app, route, create_app  # noqa: F401
+from .sdk.decorators.tracing import instrument  # noqa: F401
+from .sdk.managers.apps import AppManager  # noqa: F401
+from .sdk.managers.config import ConfigManager  # noqa: F401
+from .sdk.managers.deployment import DeploymentManager  # noqa: F401
+from .sdk.managers.secrets import SecretsManager  # noqa: F401
+from .sdk.managers.variant import VariantManager  # noqa: F401
+from .sdk.managers.vault import VaultManager  # noqa: F401
+from .sdk.tracing import Tracing, get_tracer  # noqa: F401
+from .sdk.tracing.conventions import Reference  # noqa: F401
 from .sdk.types import (
-    BinaryParam,
-    DictInput,
-    FileInputURL,
-    FloatParam,
-    GroupedMultipleChoiceParam,
-    IntParam,
-    MCField,
-    MessagesInput,
-    MultipleChoice,
-    MultipleChoiceParam,
-    Prompt,
-    PromptTemplate,
-    TextParam,
+    BinaryParam,  # noqa: F401
+    DictInput,  # noqa: F401
+    FileInputURL,  # noqa: F401
+    FloatParam,  # noqa: F401
+    GroupedMultipleChoiceParam,  # noqa: F401
+    IntParam,  # noqa: F401
+    MCField,  # noqa: F401
+    MessagesInput,  # noqa: F401
+    MultipleChoice,  # noqa: F401
+    MultipleChoiceParam,  # noqa: F401
+    Prompt,  # noqa: F401
+    PromptTemplate,  # noqa: F401
+    TextParam,  # noqa: F401
 )
-from .sdk.utils.costs import calculate_token_usage
-from .sdk.utils.logging import get_module_logger
-from .sdk.utils.preinit import PreInitObject
+from .sdk.utils.costs import calculate_token_usage  # noqa: F401
+from .sdk.utils.logging import get_module_logger  # noqa: F401
+from .sdk.utils.preinit import PreInitObject  # noqa: F401
 
 DEFAULT_AGENTA_SINGLETON_INSTANCE = AgentaSingleton()
 

{agenta-0.85.2 → agenta-0.85.4}/agenta/client/backend/core/pydantic_utilities.py

@@ -43,9 +43,9 @@ else:
     from pydantic.typing import is_literal_type as is_literal_type  # type: ignore[no-redef]
     from pydantic.typing import is_union as is_union  # type: ignore[no-redef]
 
-from .datetime_utils import serialize_datetime
-from .serialization import convert_and_respect_annotation_metadata
-from typing_extensions import TypeAlias
+from .datetime_utils import serialize_datetime  # noqa: E402
+from .serialization import convert_and_respect_annotation_metadata  # noqa: E402
+from typing_extensions import TypeAlias  # noqa: E402
 
 T = TypeVar("T")
 Model = TypeVar("Model", bound=pydantic.BaseModel)
@@ -406,7 +406,7 @@ def _get_model_fields(model: Type["Model"]) -> Mapping[str, PydanticField]:
 def _get_field_default(field: PydanticField) -> Any:
     try:
         value = field.get_default()  # type: ignore[union-attr]
-    except:
+    except:  # noqa: E722
         value = field.default
     if IS_PYDANTIC_V2:
         from pydantic_core import PydanticUndefined

{agenta-0.85.2 → agenta-0.85.4}/agenta/client/backend/core/serialization.py

@@ -74,10 +74,10 @@ def convert_and_respect_annotation_metadata(
 
     if (
         typing_extensions.get_origin(clean_type) == typing.Dict
-        or typing_extensions.get_origin(clean_type) == dict
+        or typing_extensions.get_origin(clean_type) == dict  # noqa: E721
         or clean_type == typing.Dict
     ) and isinstance(object_, typing.Dict):
-        key_type = typing_extensions.get_args(clean_type)[0]
+        key_type = typing_extensions.get_args(clean_type)[0]  # noqa: F841
         value_type = typing_extensions.get_args(clean_type)[1]
 
         return {
@@ -94,7 +94,7 @@ def convert_and_respect_annotation_metadata(
     if not isinstance(object_, str):
         if (
             typing_extensions.get_origin(clean_type) == typing.Set
-            or typing_extensions.get_origin(clean_type) == set
+            or typing_extensions.get_origin(clean_type) == set  # noqa: E721
             or clean_type == typing.Set
         ) and isinstance(object_, typing.Set):
             inner_type = typing_extensions.get_args(clean_type)[0]
@@ -110,7 +110,7 @@ def convert_and_respect_annotation_metadata(
         elif (
             (
                 typing_extensions.get_origin(clean_type) == typing.List
-                or typing_extensions.get_origin(clean_type) == list
+                or typing_extensions.get_origin(clean_type) == list  # noqa: E721
                 or clean_type == typing.List
             )
             and isinstance(object_, typing.List)

{agenta-0.85.2 → agenta-0.85.4}/agenta/client/types.py

@@ -1,4 +1,4 @@
 # Compatibility layer to maintain backward compatibility
 # Re-exports all types from backend.types
 
-from .backend.types import *
+from .backend.types import *  # noqa: F403

{agenta-0.85.2 → agenta-0.85.4}/agenta/sdk/__init__.py

@@ -1,6 +1,6 @@
 from typing import Optional, Callable, Any
 
-from .utils.preinit import PreInitObject  # always the first import!
+from .utils.preinit import PreInitObject  # always the first import!  # noqa: F401
 
 __all__ = [
     # Decorators
@@ -60,8 +60,8 @@ from .types import (
     FileInputURL,
     BinaryParam,
     Prompt,
-    AgentaNodeDto,
-    AgentaNodesResponse,
+    AgentaNodeDto,  # noqa: F401
+    AgentaNodesResponse,  # noqa: F401
 )
 
 from .tracing import Tracing, get_tracer

{agenta-0.85.2 → agenta-0.85.4}/agenta/sdk/agenta_init.py

@@ -4,7 +4,6 @@ from typing import Any, Callable, Optional
 
 import httpx
 from agenta.client.backend.client import AgentaApi, AsyncAgentaApi
-from agenta.sdk.contexts.routing import RoutingContext
 from agenta.sdk.tracing import Tracing
 from agenta.sdk.utils.globals import set_global
 from agenta.sdk.utils.helpers import parse_url

agenta-0.85.4/agenta/sdk/decorators/__init__.py

@@ -0,0 +1 @@
+from .running import application, evaluator  # noqa: F401

{agenta-0.85.2 → agenta-0.85.4}/agenta/sdk/decorators/routing.py

@@ -20,8 +20,7 @@ from agenta.sdk.middlewares.routing.cors import CORSMiddleware
 from agenta.sdk.middlewares.routing.auth import AuthMiddleware
 from agenta.sdk.middlewares.routing.otel import OTelMiddleware
 from agenta.sdk.middleware.vault import VaultMiddleware
-from agenta.sdk.contexts.running import running_context_manager, RunningContext
-from agenta.sdk.contexts.tracing import tracing_context_manager, TracingContext
+from agenta.sdk.contexts.tracing import TracingContext
 from agenta.sdk.decorators.running import auto_workflow, Workflow
 from agenta.sdk.workflows.errors import ErrorStatus
 
@@ -210,18 +209,20 @@ class route:
         path: str = "/",
         app: Optional[FastAPI] = None,
         router: Optional[APIRouter] = None,
+        flags: Optional[dict] = None,
     ):
         path = path.rstrip("/")
         path = path if path else "/"
         path = path if path.startswith("/") else "/" + path
         self.path = path
         self.root = app or router or default_app
+        self.flags = flags
 
     def __call__(self, foo: Optional[Union[Callable[..., Any], Workflow]] = None):
         if foo is None:
             return self
 
-        workflow = auto_workflow(foo)
+        workflow = auto_workflow(foo, flags=self.flags)
 
         async def invoke_endpoint(req: Request, request: WorkflowServiceRequest):
             credentials = req.state.auth.get("credentials")

{agenta-0.85.2 → agenta-0.85.4}/agenta/sdk/decorators/running.py

@@ -1,17 +1,14 @@
 # /agenta/sdk/decorators/running.py
 
-from typing import Any, Callable, Optional, Protocol, Union, Dict
+from typing import Any, Callable, Optional, Protocol, Union, Dict, cast
 from functools import update_wrapper, wraps
-from typing import Callable, Any
 from inspect import signature
-from uuid import UUID, uuid4
+from uuid import UUID
 
 from agenta.sdk.utils.logging import get_module_logger
 from agenta.sdk.models.workflows import (
-    WorkflowRevisionData,
     WorkflowRevision,
     WorkflowServiceRequestData,
-    WorkflowServiceResponseData,
     WorkflowServiceRequest,
     WorkflowServiceInterface,
     WorkflowServiceConfiguration,
@@ -32,7 +29,6 @@ from agenta.sdk.middlewares.running.resolver import (
 )
 from agenta.sdk.middlewares.running.vault import (
     VaultMiddleware,
-    get_secrets,
 )
 from agenta.sdk.decorators.tracing import auto_instrument
 from agenta.sdk.workflows.utils import (
@@ -458,14 +454,52 @@ def is_workflow(obj: Any) -> bool:
 
 
 def auto_workflow(obj: Any, **kwargs) -> Workflow:
+    flags = kwargs.get("flags")
+
+    def _merge_flags(
+        existing: Optional[dict], incoming: Optional[dict]
+    ) -> Optional[dict]:
+        if incoming is None:
+            return existing
+        merged = dict(existing or {})
+        merged.update(incoming)
+        return merged
+
+    def _apply_flags_to_existing_workflow(
+        target: Any, incoming_flags: Optional[dict]
+    ) -> None:
+        if not incoming_flags:
+            return
+
+        # Common case: object exposes the underlying workflow decorator instance.
+        try:
+            wf = getattr(target, "workflow", None)
+            if isinstance(wf, workflow):
+                wf.flags = _merge_flags(wf.flags, incoming_flags)
+                return
+        except Exception:
+            pass
+
+        # If this is an extended Workflow wrapper, its `.invoke` is typically a bound
+        # method whose `__self__` is the originating `workflow` instance.
+        try:
+            invoke = getattr(target, "invoke", None)
+            wf_self = getattr(invoke, "__self__", None)
+            if isinstance(wf_self, workflow):
+                wf_self.flags = _merge_flags(wf_self.flags, incoming_flags)
+        except Exception:
+            pass
+
     if is_workflow(obj):
-        return obj
+        _apply_flags_to_existing_workflow(obj, flags)
+        return cast(Workflow, obj)
+
     if isinstance(obj, workflow):
-        return obj()
-    if isinstance(getattr(obj, "workflow", None), workflow):
-        return obj
+        if flags is not None:
+            obj.flags = _merge_flags(obj.flags, flags)
+        return cast(Workflow, obj())
 
-    return workflow(**kwargs)(obj)
+    return cast(Workflow, workflow(**kwargs)(obj))
 
 
 async def invoke_workflow(
@@ -548,12 +582,11 @@ class application(workflow):
             # is_human=False,  # None / False / missing is the same
         )
 
-        if not "references" in kwargs or not isinstance(kwargs["references"], dict):
+        if "references" not in kwargs or not isinstance(kwargs["references"], dict):
             kwargs["references"] = dict()
 
-        for key in kwargs["references"]:
-            if key.startswith("evaluator_"):
-                del kwargs["references"][key]
+        for key in [k for k in kwargs["references"] if k.startswith("evaluator_")]:
+            del kwargs["references"][key]
 
         if slug is not None:
             kwargs["references"]["application"] = {"slug": slug}
@@ -651,12 +684,11 @@ class evaluator(workflow):
             # is_human=False,  # None / False / missing is the same
         )
 
-        if not "references" in kwargs or not isinstance(kwargs["references"], dict):
+        if "references" not in kwargs or not isinstance(kwargs["references"], dict):
             kwargs["references"] = dict()
 
-        for key in kwargs["references"]:
-            if key.startswith("application_"):
-                del kwargs["references"][key]
+        for key in [k for k in kwargs["references"] if k.startswith("application_")]:
+            del kwargs["references"][key]
 
         if slug is not None:
             kwargs["references"]["evaluator"] = {"slug": slug}

{agenta-0.85.2 → agenta-0.85.4}/agenta/sdk/decorators/serving.py

@@ -15,8 +15,7 @@ from pydantic import BaseModel, HttpUrl, ValidationError
 from os import environ
 
 if TYPE_CHECKING:
-    from fastapi import Request, HTTPException, Body
-    from starlette.responses import Response as StarletteResponse, StreamingResponse
+    from fastapi import FastAPI, Request, HTTPException, Body
 else:
     # Lazy imports - only loaded when @entrypoint or @route is used
     Request = None
@@ -31,7 +30,6 @@ from agenta.sdk.contexts.tracing import (
     tracing_context_manager,
     TracingContext,
 )
-from agenta.sdk.router import router
 from agenta.sdk.utils.exceptions import suppress, display_exception
 from agenta.sdk.utils.logging import get_module_logger
 from agenta.sdk.utils.helpers import get_current_version
@@ -40,7 +38,6 @@ from agenta.sdk.types import (
     MultipleChoice,
     BaseResponse,
     StreamResponse,
-    MCField,
 )
 
 import agenta as ag
@@ -82,7 +79,7 @@ class PathValidator(BaseModel):
     url: HttpUrl
 
 
-def _add_middleware_to_app(target_app: "FastAPI") -> None:
+def _add_middleware_to_app(target_app: "FastAPI") -> None:  # noqa: F821
     """
     Add all required middleware to a FastAPI app.
 
@@ -165,8 +162,10 @@ def create_app():
             self,
             path: Optional[str] = "/",
             config_schema: Optional[BaseModel] = None,
+            flags: Optional[Dict[str, Any]] = None,
         ):
             self.config_schema = config_schema
+            self.flags = dict(flags or {})
             path = "/" + path.strip("/").strip()
             path = "" if path == "/" else path
             PathValidator(url=f"http://example.com{path}")
@@ -178,6 +177,7 @@ def create_app():
                 f,
                 route_path=self.route_path,
                 config_schema=self.config_schema,
+                flags=self.flags,
                 target_app=new_app,
                 app_routes=isolated_routes,
             )
@@ -196,8 +196,10 @@ class route:  # pylint: disable=invalid-name
         self,
         path: Optional[str] = "/",
         config_schema: Optional[BaseModel] = None,
+        flags: Optional[Dict[str, Any]] = None,
     ):
         self.config_schema: BaseModel = config_schema
+        self.flags = dict(flags or {})
         path = "/" + path.strip("/").strip()
         path = "" if path == "/" else path
         PathValidator(url=f"http://example.com{path}")
@@ -211,6 +213,7 @@ class route:  # pylint: disable=invalid-name
             f,
             route_path=self.route_path,
             config_schema=self.config_schema,
+            flags=self.flags,
         )
 
         return f
@@ -262,6 +265,7 @@ class entrypoint:
         func: Callable[..., Any],
         route_path: str = "",
         config_schema: Optional[BaseModel] = None,
+        flags: Optional[Dict[str, Any]] = None,
         target_app: Optional[Any] = None,
         app_routes: Optional[List[Dict[str, Any]]] = None,
     ):
@@ -271,6 +275,7 @@ class entrypoint:
         self.func = func
         self.route_path = route_path
         self.config_schema = config_schema
+        self.flags = dict(flags or {})
 
         # Use provided app/routes or fall back to global defaults
         target_app = target_app if target_app is not None else app
@@ -325,6 +330,16 @@ class entrypoint:
             response_model_exclude_none=True,
         )(run_wrapper)
 
+        app_routes.append(
+            {
+                "func": func.__name__,
+                "endpoint": run_route,
+                "params": signature_parameters,
+                "config": None,
+                "flags": self.flags,
+            }
+        )
+
         # LEGACY
         # TODO: Removing this implies breaking changes in :
         # - calls to /generate_deployed must be replaced with calls to /run
@@ -381,6 +396,7 @@ class entrypoint:
                 "endpoint": test_route,
                 "params": signature_parameters,
                 "config": config,
+                "flags": self.flags,
             }
         )
 
@@ -397,12 +413,15 @@ class entrypoint:
                         else signature_parameters
                     ),
                     "config": config,
+                    "flags": self.flags,
                 }
             )
         # LEGACY
 
         openapi_schema = _generate_openapi(target_app)
 
+        self.add_flags_to_schema(openapi_schema=openapi_schema, app_routes=app_routes)
+
         for _route in app_routes:
             if _route["config"] is not None:
                 self.override_config_in_schema(
@@ -540,7 +559,7 @@ class entrypoint:
                     result.headers.setdefault("x-ag-span-id", span_id)
 
                 return result
-        except:
+        except Exception:
             return result
 
         try:
@@ -552,7 +571,7 @@ class entrypoint:
                     trace_id=trace_id,
                     span_id=span_id,
                 )
-        except:
+        except Exception:
             return StreamingResponse(
                 result,
                 media_type="text/event-stream",
@@ -567,7 +586,7 @@ class entrypoint:
                 trace_id=trace_id,
                 span_id=span_id,
             )
-        except:  # pylint: disable=bare-except
+        except Exception:
             try:
                 return BaseResponse(
                     data=data,
@@ -576,7 +595,7 @@ class entrypoint:
                     trace_id=trace_id,
                     span_id=span_id,
                 )
-            except:  # pylint: disable=bare-except
+            except Exception:
                 return BaseResponse(
                     data=data,
                     content_type=content_type,
@@ -882,6 +901,39 @@ class entrypoint:
 
         return openapi_schema
 
+    def add_flags_to_schema(
+        self,
+        openapi_schema: Dict[str, Any],
+        app_routes: List[Dict[str, Any]],
+    ) -> None:
+        paths = openapi_schema.get("paths")
+        if not paths:
+            return
+
+        for route in app_routes:
+            endpoint = route.get("endpoint")
+            if not endpoint or ("/run" not in endpoint and "/test" not in endpoint):
+                continue
+
+            methods = paths.get(endpoint)
+            if not methods:
+                continue
+
+            flags = dict(route.get("flags") or {})
+            for method_data in methods.values():
+                # Prefer a single vendor-extension namespace we can evolve over time.
+                existing = method_data.get("x-agenta")
+                if not isinstance(existing, dict):
+                    existing = {}
+
+                existing_flags = existing.get("flags")
+                if not isinstance(existing_flags, dict):
+                    existing_flags = {}
+
+                # Route-level flags override any previously set flags.
+                existing["flags"] = {**existing_flags, **flags}
+                method_data["x-agenta"] = existing
+
     def override_config_in_schema(
         self,
         openapi_schema: dict,

{agenta-0.85.2 → agenta-0.85.4}/agenta/sdk/decorators/tracing.py

@@ -520,14 +520,14 @@ class instrument:  # pylint: disable=invalid-name
         if self.redact is not None:
             try:
                 io = self.redact(name, field, io)
-            except:  # pylint: disable=bare-except
+            except Exception:  # pylint: disable=bare-except
                 if self.redact_on_error:
                     io = {}
 
         if ag.tracing.redact is not None:
             try:
                 io = ag.tracing.redact(name, field, io)
-            except:  # pylint: disable=bare-except
+            except Exception:  # pylint: disable=bare-except
                 if ag.tracing.redact_on_error:
                     io = {}
 

{agenta-0.85.2 → agenta-0.85.4}/agenta/sdk/engines/running/registry.py

@@ -27,7 +27,7 @@ async def exact_match_v1(
             reference_outputs = dumps(reference_outputs, sort_keys=True)
             success = outputs == reference_outputs
 
-    except:  # pylint: disable=bare-except
+    except Exception:  # pylint: disable=bare-except
         log.error("Error in exact_match_v1", exc_info=True)
 
     return {"success": success}

agenta-0.85.4/agenta/sdk/engines/tracing/__init__.py

@@ -0,0 +1 @@
+from .tracing import Tracing, get_tracer  # noqa: F401

{agenta-0.85.2 → agenta-0.85.4}/agenta/sdk/engines/tracing/attributes.py

@@ -1,4 +1,4 @@
-from json import loads, dumps
+from json import dumps
 from typing import Optional, Union, Sequence, Any, Dict
 
 Primitive = Union[str, int, float, bool, bytes]

{agenta-0.85.2 → agenta-0.85.4}/agenta/sdk/engines/tracing/inline.py

@@ -31,11 +31,10 @@ class LifecycleDTO(BaseModel):
 ### services.observability.dtos ###
 ### --------------------------- ###
 
-from typing import List, Dict, Any, Union, Optional
+from typing import List, Dict, Any, Union, Optional  # noqa: E402
 
-from enum import Enum
-from datetime import datetime
-from uuid import UUID
+from datetime import datetime  # noqa: E402
+from uuid import UUID  # noqa: E402
 
 
 class TimeDTO(BaseModel):
@@ -218,7 +217,7 @@ class OTelSpanDTO(BaseModel):
 ### services.observability.utils ###
 ### ---------------------------- ###
 
-from typing import List, Dict, OrderedDict
+from typing import List, Dict  # noqa: E402
 
 
 def parse_span_dtos_to_span_idx(
@@ -415,7 +414,7 @@ def _connect_tree_dfs(
 ### apis.fastapi.observability.opentelemetry.semconv ###
 ### ------------------------------------------------ ###
 
-from json import loads
+from json import loads  # noqa: E402
 
 VERSION = "0.4.1"
 
@@ -525,11 +524,10 @@ CODEX = {"maps": MAPS[VERSION], "keys": KEYS[VERSION]}
 ### apis.fastapi.observability.utils ###
 ### -------------------------------- ###
 
-from typing import Optional, Union, Tuple, Any, List, Dict
-from uuid import UUID
-from collections import OrderedDict
-from json import loads, JSONDecodeError, dumps
-from copy import copy
+from typing import Optional, Union, Tuple, Any, List, Dict  # noqa: E402
+from uuid import UUID  # noqa: E402
+from collections import OrderedDict  # noqa: E402
+from copy import copy  # noqa: E402
 
 
 def _unmarshal_attributes(
@@ -698,7 +696,7 @@ def _parse_from_semconv(
 
                         attributes[new_key] = new_value
 
-                    except:  # pylint: disable=bare-except
+                    except Exception:  # pylint: disable=bare-except
                         pass
 
 
@@ -821,7 +819,7 @@ def parse_from_otel_span_dto(
     node_type = NodeType.TASK
     try:
         node_type = NodeType(types.get("node", "").lower())
-    except:  # pylint: disable=bare-except
+    except Exception:  # pylint: disable=bare-except
         pass
 
     node = NodeDTO(
@@ -957,11 +955,11 @@ def parse_to_agenta_span_dto(
 ########################################
 
 
-from opentelemetry.sdk.trace import ReadableSpan
+from opentelemetry.sdk.trace import ReadableSpan  # noqa: E402
 
-from agenta.sdk.utils.lazy import _load_litellm
+from agenta.sdk.utils.lazy import _load_litellm  # noqa: E402
 
-from agenta.sdk.types import AgentaNodeDto, AgentaNodesResponse
+from agenta.sdk.types import AgentaNodeDto, AgentaNodesResponse  # noqa: E402
 
 
 def parse_inline_trace(
@@ -1157,5 +1155,5 @@ def calculate_costs(span_idx: Dict[str, SpanDTO]):
                 span.metrics["unit.costs.completion"] = completion_cost
                 span.metrics["unit.costs.total"] = total_cost
 
-            except:  # pylint: disable=bare-except
+            except Exception:  # pylint: disable=bare-except
                 pass