unique_toolkit 2026.34.0.dev6__py3-none-any.whl → 2026.34.0.dev8__py3-none-any.whl

unique_toolkit/agentic/feature_flags/__init__.py

@@ -1,6 +1,7 @@
 from unique_toolkit.agentic.feature_flags.feature_flags import (
+    FeatureFlagNames,
     FeatureFlags,
     feature_flags,
 )
 
-__all__ = ["FeatureFlags", "feature_flags"]
+__all__ = ["FeatureFlagNames", "FeatureFlags", "feature_flags"]

unique_toolkit/agentic/feature_flags/feature_flags.py

@@ -1,11 +1,41 @@
+from enum import StrEnum
 from typing import Any
 
 from pydantic import Field, field_validator
 from pydantic_settings import BaseSettings, SettingsConfigDict
+from typing_extensions import deprecated
 
 from unique_toolkit._common.config_checker import register_config
 
 
+class FeatureFlagNames(StrEnum):
+    """Flag name constants used by unique_toolkit's own agentic features.
+
+    Values match both the env-var fallback names (``FeatureFlags`` below) and
+    the configuration-backend registry keys used by
+    ``unique_toolkit.experimental.resources.feature_flags.FeatureFlagClient``.
+    """
+
+    enable_selected_uploaded_files_un_18215 = (
+        "FEATURE_FLAG_ENABLE_SELECTED_UPLOADED_FILES_UN_18215"
+    )
+    enable_mcp_metadata_fallback_un_19145 = (
+        "FEATURE_FLAG_ENABLE_MCP_METADATA_FALLBACK_UN_19145"
+    )
+    enable_html_rendering_un_15131 = "FEATURE_FLAG_ENABLE_HTML_RENDERING_UN_15131"
+    enable_code_execution_fence_un_17972 = (
+        "FEATURE_FLAG_ENABLE_CODE_EXECUTION_FENCE_UN_17972"
+    )
+    enable_html_with_fence_un_17927 = "FEATURE_FLAG_ENABLE_HTML_WITH_FENCE_UN_17927"
+    enable_web_search_argument_screening_un_18741 = (
+        "FEATURE_FLAG_ENABLE_WEB_SEARCH_ARGUMENT_SCREENING_UN_18741"
+    )
+    enable_new_answers_ui_un_14411 = "FEATURE_FLAG_ENABLE_NEW_ANSWERS_UI_UN_14411"
+
+
+@deprecated(
+    "Use unique_toolkit.experimental.resources.feature_flags.is_flag_enabled instead"
+)
 class FeatureFlag:
     """A feature flag that can be enabled globally or for specific company IDs.
 
@@ -45,6 +75,9 @@ class FeatureFlag:
         return f"FeatureFlag({self.value})"
 
 
+@deprecated(
+    "Use unique_toolkit.experimental.resources.feature_flags.is_flag_enabled instead"
+)
 @register_config()
 class FeatureFlags(BaseSettings):
     """Feature flags loaded from environment variables.
@@ -145,5 +178,7 @@ class FeatureFlags(BaseSettings):
         return FeatureFlag(False)
 
 
-# Initialize once at module load - import this where needed
+# Deprecated, unused elsewhere in this toolkit — kept only for external
+# backward compatibility. Use unique_toolkit.experimental.resources.feature_flags
+# .is_flag_enabled instead.
 feature_flags = FeatureFlags()

unique_toolkit/agentic/history_manager/loop_token_reducer.py

@@ -109,7 +109,19 @@ class LoopTokenReducer:
         self._max_db_source_number: int = -1
         self._db_source_map: dict[int, ContentChunk] = {}
         self._enable_tool_call_persistence = enable_tool_call_persistence
-        self._selected_content_ids = get_selected_uploaded_content_ids(event)
+        self._event = event
+        # Lazily resolved on first async use since the flag check is async;
+        # `_resolved` distinguishes "not yet checked" from a resolved `None` (flag off).
+        self._selected_content_ids: set[str] | None = None
+        self._selected_content_ids_resolved = False
+
+    async def _get_selected_content_ids(self) -> set[str] | None:
+        if not self._selected_content_ids_resolved:
+            self._selected_content_ids = await get_selected_uploaded_content_ids(
+                self._event
+            )
+            self._selected_content_ids_resolved = True
+        return self._selected_content_ids
 
     @property
     def max_db_source_number(self) -> int:
@@ -354,6 +366,7 @@ class LoopTokenReducer:
         Returns:
             list[LanguageModelMessage]: The history
         """
+        selected_content_ids = await self._get_selected_content_ids()
         if self._enable_tool_call_persistence:
             (
                 full_history,
@@ -365,7 +378,7 @@ class LoopTokenReducer:
                 chat_service=self._chat_service,
                 content_service=self._content_service,
                 file_content_serializer=self._file_content_serializer,
-                selected_content_ids=self._selected_content_ids,
+                selected_content_ids=selected_content_ids,
             )
             self._max_db_source_number = max_src
             self._db_source_map = src_map
@@ -376,7 +389,7 @@ class LoopTokenReducer:
                 chat_service=self._chat_service,
                 content_service=self._content_service,
                 file_content_serializer=self._file_content_serializer,
-                selected_content_ids=self._selected_content_ids,
+                selected_content_ids=selected_content_ids,
             )
 
         if remove_from_text is not None:

unique_toolkit/agentic/history_manager/utils.py

@@ -3,10 +3,11 @@ import logging
 from copy import deepcopy
 from typing import Any
 
-from unique_toolkit.agentic.feature_flags import feature_flags
+from unique_toolkit.agentic.feature_flags import FeatureFlagNames
 from unique_toolkit.agentic.tools.schemas import Source
 from unique_toolkit.app.schemas import ChatEvent
 from unique_toolkit.content.schemas import ContentChunk
+from unique_toolkit.experimental.resources.feature_flags import is_flag_enabled
 from unique_toolkit.language_model.schemas import (
     LanguageModelAssistantMessage,
     LanguageModelMessage,
@@ -16,15 +17,16 @@ from unique_toolkit.language_model.schemas import (
 logger = logging.getLogger(__name__)
 
 
-def get_selected_uploaded_content_ids(event: ChatEvent) -> set[str] | None:
+async def get_selected_uploaded_content_ids(event: ChatEvent) -> set[str] | None:
     """Derive the set of content IDs the user explicitly selected.
 
     Returns ``None`` when the feature flag is disabled (meaning *all*
     uploaded images should be included), or a ``set[str]`` of IDs when
     only a subset should be attached.
     """
-    if not feature_flags.enable_selected_uploaded_files_un_18215.is_enabled(
-        event.company_id
+    if not await is_flag_enabled(
+        FeatureFlagNames.enable_selected_uploaded_files_un_18215,
+        company_id=event.company_id,
     ):
         return None
 

unique_toolkit/agentic/tools/a2a/tool/service.py

@@ -22,7 +22,7 @@ from unique_toolkit._common.referencing import (
 )
 from unique_toolkit._common.utils.jinja.render import render_template
 from unique_toolkit.agentic.evaluation.schemas import EvaluationMetricName
-from unique_toolkit.agentic.feature_flags import feature_flags
+from unique_toolkit.agentic.feature_flags import FeatureFlagNames
 from unique_toolkit.agentic.tools.a2a.response_watcher import SubAgentResponseWatcher
 from unique_toolkit.agentic.tools.a2a.tool._memory import (
     get_sub_agent_short_term_memory_manager,
@@ -53,6 +53,7 @@ from unique_toolkit.chat.schemas import (
 )
 from unique_toolkit.chat.service import ChatService
 from unique_toolkit.content import ContentChunk, ContentReference
+from unique_toolkit.experimental.resources.feature_flags import is_flag_enabled
 from unique_toolkit.language_model import (
     LanguageModelFunction,
     LanguageModelService,
@@ -431,11 +432,9 @@ class SubAgentTool(Tool[SubAgentToolConfig]):
         message: str,
         state: ProgressState,
     ) -> None:
-        if (
-            self._tool_progress_reporter is not None
-            and not feature_flags.enable_new_answers_ui_un_14411.is_enabled(
-                self._company_id
-            )
+        if self._tool_progress_reporter is not None and not await is_flag_enabled(
+            FeatureFlagNames.enable_new_answers_ui_un_14411,
+            company_id=self._company_id or "",
         ):
             await self._tool_progress_reporter.notify_from_tool_call(
                 tool_call=tool_call,

unique_toolkit/agentic/tools/mcp/tool_wrapper.py

@@ -9,7 +9,7 @@ from typing import Any, Dict
 import unique_sdk
 
 from unique_toolkit.agentic.evaluation.schemas import EvaluationMetricName
-from unique_toolkit.agentic.feature_flags import feature_flags
+from unique_toolkit.agentic.feature_flags import FeatureFlagNames
 from unique_toolkit.agentic.tools.mcp.models import MCPToolConfig
 from unique_toolkit.agentic.tools.schemas import ToolCallResponse
 from unique_toolkit.agentic.tools.tool import Tool
@@ -21,6 +21,7 @@ from unique_toolkit.app.schemas import ChatEvent, McpServer, McpTool
 from unique_toolkit.chat.schemas import MessageLog, MessageLogStatus
 from unique_toolkit.chat.service import ChatService
 from unique_toolkit.content.functions import upload_content_from_bytes
+from unique_toolkit.experimental.resources.feature_flags import is_flag_enabled
 from unique_toolkit.language_model import LanguageModelService
 from unique_toolkit.language_model.schemas import (
     LanguageModelFunction,
@@ -133,11 +134,9 @@ class MCPToolWrapper(Tool[MCPToolConfig]):
         )
 
         # Notify progress if reporter is available
-        if (
-            self._tool_progress_reporter
-            and not feature_flags.enable_new_answers_ui_un_14411.is_enabled(
-                self._event.company_id
-            )
+        if self._tool_progress_reporter and not await is_flag_enabled(
+            FeatureFlagNames.enable_new_answers_ui_un_14411,
+            company_id=self._event.company_id,
         ):
             await self._tool_progress_reporter.notify_from_tool_call(
                 tool_call=tool_call,
@@ -172,11 +171,9 @@ class MCPToolWrapper(Tool[MCPToolConfig]):
             )
 
             # Notify completion
-            if (
-                self._tool_progress_reporter
-                and not feature_flags.enable_new_answers_ui_un_14411.is_enabled(
-                    self._event.company_id
-                )
+            if self._tool_progress_reporter and not await is_flag_enabled(
+                FeatureFlagNames.enable_new_answers_ui_un_14411,
+                company_id=self._event.company_id,
             ):
                 await self._tool_progress_reporter.notify_from_tool_call(
                     tool_call=tool_call,
@@ -198,11 +195,9 @@ class MCPToolWrapper(Tool[MCPToolConfig]):
             self.logger.error(f"Error executing MCP tool {self.name}: {e}")
 
             # Notify failure
-            if (
-                self._tool_progress_reporter
-                and not feature_flags.enable_new_answers_ui_un_14411.is_enabled(
-                    self._event.company_id
-                )
+            if self._tool_progress_reporter and not await is_flag_enabled(
+                FeatureFlagNames.enable_new_answers_ui_un_14411,
+                company_id=self._event.company_id,
             ):
                 await self._tool_progress_reporter.notify_from_tool_call(
                     tool_call=tool_call,

unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/code_display.py

@@ -6,11 +6,15 @@ from typing import override
 from pydantic import BaseModel, Field
 from pydantic.json_schema import SkipJsonSchema
 
-from unique_toolkit.agentic.feature_flags.feature_flags import feature_flags
+from unique_toolkit.agentic.feature_flags import FeatureFlagNames
 from unique_toolkit.agentic.postprocessor.postprocessor_manager import (
     ResponsesApiPostprocessor,
 )
 from unique_toolkit.agentic.tools.config import get_configuration_dict
+from unique_toolkit.experimental.resources.feature_flags import (
+    COMPANY_ID_PLACEHOLDER,
+    is_flag_enabled,
+)
 from unique_toolkit.language_model.schemas import ResponsesLanguageModelStreamResponse
 
 _TEMPLATE = """
@@ -56,15 +60,15 @@ class ShowExecutedCodePostprocessor(ResponsesApiPostprocessor):
         super().__init__(self.__class__.__name__)
         self._config = config
         self._company_id = company_id
-        self._is_enabled = (
-            self._config.enable
-            and not feature_flags.enable_code_execution_fence_un_17972.is_enabled(
-                company_id
-            )
-        )
+        # Resolved in run() (before apply_postprocessing_to_response) since flag evaluation is async.
+        self._is_enabled = False
 
     @override
     async def run(self, loop_response: ResponsesLanguageModelStreamResponse) -> None:
+        self._is_enabled = self._config.enable and not await is_flag_enabled(
+            FeatureFlagNames.enable_code_execution_fence_un_17972,
+            company_id=self._company_id or COMPANY_ID_PLACEHOLDER,
+        )
         if self._is_enabled:
             await asyncio.sleep(self._config.sleep_time_before_display)
 

unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/generated_files.py

@@ -21,7 +21,7 @@ from tenacity import (
 
 from unique_toolkit import ChatService
 from unique_toolkit._common.execution import failsafe_async
-from unique_toolkit.agentic.feature_flags.feature_flags import feature_flags
+from unique_toolkit.agentic.feature_flags import FeatureFlagNames
 from unique_toolkit.agentic.postprocessor.postprocessor_manager import (
     ResponsesApiPostprocessor,
 )
@@ -39,6 +39,10 @@ from unique_toolkit.agentic.tools.openai_builtin.code_interpreter.schemas import
 )
 from unique_toolkit.content.schemas import ContentReference
 from unique_toolkit.content.service import ContentService
+from unique_toolkit.experimental.resources.feature_flags import (
+    COMPANY_ID_PLACEHOLDER,
+    is_flag_enabled,
+)
 from unique_toolkit.language_model.schemas import ResponsesLanguageModelStreamResponse
 from unique_toolkit.services.knowledge_base import KnowledgeBaseService
 from unique_toolkit.short_term_memory.service import ShortTermMemoryService
@@ -389,6 +393,10 @@ class DisplayCodeInterpreterFilesPostProcessor(
         # Cleared at the start of each run(); holds this-turn download sizes.
         self._file_size_map: dict[str, int] = {}
 
+        # Resolved in run() (before apply_postprocessing_to_response) since flag evaluation is async.
+        self._fence_ff_on = False
+        self._html_fence_ff_on = False
+
     def _build_retry(self) -> AsyncRetrying:
         """Build a tenacity retry policy from the current config.
 
@@ -409,6 +417,17 @@ class DisplayCodeInterpreterFilesPostProcessor(
         run_t0 = time.monotonic()
         self._log.info("run() started — fetching and uploading code interpreter files")
 
+        self._fence_ff_on = await is_flag_enabled(
+            FeatureFlagNames.enable_code_execution_fence_un_17972,
+            company_id=self._company_id or COMPANY_ID_PLACEHOLDER,
+        )
+        # htmlWithSource requires BOTH the fence FF and the HTML-fence FF on; default
+        # (FF off) keeps HtmlRendering, so existing deployments are unaffected.
+        self._html_fence_ff_on = await is_flag_enabled(
+            FeatureFlagNames.enable_html_with_fence_un_17927,
+            company_id=self._company_id or COMPANY_ID_PLACEHOLDER,
+        )
+
         container_files = loop_response.container_files
         self._log.info(
             "run() found %d container file annotation(s): %s",
@@ -554,9 +573,7 @@ class DisplayCodeInterpreterFilesPostProcessor(
         self._log.info(
             "apply_postprocessing started — %d file(s) in content_map, fence_ff=%s",
             len(self._content_map),
-            feature_flags.enable_code_execution_fence_un_17972.is_enabled(
-                self._company_id
-            ),
+            self._fence_ff_on,
         )
 
         if loop_response.message.references is None:
@@ -566,18 +583,6 @@ class DisplayCodeInterpreterFilesPostProcessor(
 
         ref_number = _get_next_ref_number(loop_response.message.references)
         changed = False
-        fence_ff_on = feature_flags.enable_code_execution_fence_un_17972.is_enabled(
-            self._company_id
-        )
-        # HTML uses htmlWithSource only when BOTH the code-execution fence FF AND the
-        # dedicated HTML-fence FF are on. Default (FF off) keeps HtmlRendering so
-        # existing deployments are unaffected.
-        html_fence_ff_on = (
-            fence_ff_on
-            and feature_flags.enable_html_with_fence_un_17927.is_enabled(
-                self._company_id
-            )
-        )
 
         replaced_files: list[str] = []
         missed_files: list[str] = []
@@ -611,7 +616,7 @@ class DisplayCodeInterpreterFilesPostProcessor(
 
             # HTML rendered as HtmlRendering block unless the dedicated html-fence FF
             # is on (enable_html_with_fence_un_17927). Default keeps HtmlRendering.
-            elif is_html and not html_fence_ff_on:
+            elif is_html and not self._html_fence_ff_on:
                 loop_response.message.text, replaced = _replace_container_html_citation(
                     text=loop_response.message.text or "",
                     filename=filename,
@@ -619,7 +624,7 @@ class DisplayCodeInterpreterFilesPostProcessor(
                 )
                 changed |= replaced
 
-            # Non-HTML files, or HTML when html_fence_ff_on → inline content link for
+            # Non-HTML files, or HTML when self._html_fence_ff_on → inline content link for
             # subsequent fence injection (htmlWithSource / imgWithSource / fileWithSource)
             else:
                 loop_response.message.text, replaced = _replace_container_file_citation(
@@ -627,7 +632,7 @@ class DisplayCodeInterpreterFilesPostProcessor(
                     filename=filename,
                     content_id=content_id,
                     ref_number=ref_number,
-                    use_content_link=fence_ff_on,
+                    use_content_link=self._fence_ff_on,
                 )
                 changed |= replaced
 
@@ -644,7 +649,7 @@ class DisplayCodeInterpreterFilesPostProcessor(
 
             # HtmlRendering and htmlWithSource both embed contentId directly — no ContentReference needed
             is_html_rendered = is_html
-            if replaced and not (is_image or is_html_rendered or fence_ff_on):
+            if replaced and not (is_image or is_html_rendered or self._fence_ff_on):
                 loop_response.message.references.append(
                     ContentReference(
                         sequence_number=ref_number,
@@ -663,9 +668,9 @@ class DisplayCodeInterpreterFilesPostProcessor(
             error_files,
         )
 
-        if fence_ff_on:
+        if self._fence_ff_on:
             code_blocks = _build_code_blocks(
-                loop_response, self._content_map, include_html=html_fence_ff_on
+                loop_response, self._content_map, include_html=self._html_fence_ff_on
             )
             self._log.info(
                 "Fence injection — %d code block(s), files: %s",
@@ -673,7 +678,7 @@ class DisplayCodeInterpreterFilesPostProcessor(
                 [f.filename for b in code_blocks for f in b.files],
             )
             _warn_unmatched_code_blocks(
-                self._content_map, code_blocks, include_html=html_fence_ff_on
+                self._content_map, code_blocks, include_html=self._html_fence_ff_on
             )
             text_before = loop_response.message.text
             loop_response.message.text = _inject_code_execution_fences(

unique_toolkit/experimental/resources/feature_flags/__init__.py

@@ -12,10 +12,15 @@ Required env vars::
 
 Quick start — explicit IDs::
 
-    from unique_toolkit.experimental.resources.feature_flags import FeatureFlagClient
+    from unique_toolkit.experimental.resources.feature_flags import FeatureFlagClient, is_flag_enabled
 
-    client = FeatureFlagClient.from_settings()  # singleton; call at startup or first request
-    enabled = await client.is_enabled("FEATURE_FLAG_ENABLE_X", company_id=company_id)
+    # Requires CONFIGURATION_BACKEND_URL / FEATURE_FLAG_SERVICE_ID to be set;
+    # falls back to the env-var value (or stale cache) on transport errors.
+    enabled = await is_flag_enabled("FEATURE_FLAG_ENABLE_X", company_id=company_id)
+
+    # Or manage the singleton directly:
+    client = FeatureFlagClient.from_settings()  # raises if URL/service_id absent
+    client = get_feature_flag_client()          # same — thin passthrough to from_settings()
 
 Per-request binding (when you have UniqueSettings / AuthContext)::
 
@@ -32,6 +37,7 @@ See README.md for the full evaluation-order diagram and adoption steps.
 """
 
 from .client import (
+    COMPANY_ID_PLACEHOLDER,
     BoundFeatureFlagClient,
     FeatureFlagClient,
     get_feature_flag_client,
@@ -41,6 +47,7 @@ from .schemas import FlagEvaluation
 from .settings import FeatureFlagSettings
 
 __all__ = [
+    "COMPANY_ID_PLACEHOLDER",
     "BoundFeatureFlagClient",
     "FeatureFlagClient",
     "FlagEvaluation",

unique_toolkit/experimental/resources/feature_flags/client.py

@@ -177,21 +177,15 @@ class FeatureFlagClient:
 
     @staticmethod
     def _env_fallback(flag: str, company_id: str | None = None) -> bool:
-        """Read *flag* from env vars using the same semantics as ``FeatureFlag``.
-
-        Supports both plain booleans (``"true"`` / ``"false"``) and
-        comma-separated company-ID allowlists (``"company1,company2"``),
-        consistent with ``unique_toolkit.agentic.feature_flags.FeatureFlags``.
-        """
-        raw = os.getenv(flag, "false").strip()
-        raw_lower = raw.lower()
+        """Read *flag* from env vars: plain booleans or comma-separated company-ID allowlists."""
+        raw = os.getenv(flag, "false")
+        raw_lower = raw.strip().lower()
         if raw_lower in ("true", "1", "yes"):
             return True
         if raw_lower in ("false", "0", "no", ""):
             return False
-        # Comma-separated company-ID allowlist
-        allowed = {part.strip() for part in raw.split(",") if part.strip()}
-        return company_id in allowed if company_id else False
+        allowlist = [c.strip() for c in raw.split(",") if c.strip()]
+        return company_id in allowlist if company_id else False
 
 
 class BoundFeatureFlagClient:
@@ -215,12 +209,18 @@ class BoundFeatureFlagClient:
         return (await self.evaluate(flag)).value
 
 
+# For is_flag_enabled() callers with no real company_id. Not a real company:
+# bool-configured flags still resolve correctly; allowlist-configured ones safely read as off.
+COMPANY_ID_PLACEHOLDER = "company_id_placeholder"
+
+
 def get_feature_flag_client() -> FeatureFlagClient:
     """Return the process-level :class:`FeatureFlagClient` singleton.
 
-    Built lazily via :meth:`FeatureFlagClient.from_settings` which reads
-    ``CONFIGURATION_BACKEND_URL`` and ``FEATURE_FLAG_SERVICE_ID`` from env.
-    Falls back to env-var defaults on missing config or transport errors.
+    Built lazily via :meth:`FeatureFlagClient.from_settings`, which reads
+    ``CONFIGURATION_BACKEND_URL`` and ``FEATURE_FLAG_SERVICE_ID`` from env and
+    raises :class:`ValueError` if either is missing. Evaluations fall back to
+    env-var defaults (or a stale cached value) only on *transport* errors.
     """
     return FeatureFlagClient.from_settings()
 

unique_toolkit/monitoring/__init__.py

@@ -4,6 +4,8 @@ from unique_toolkit.monitoring.tracing import (
     TracingSettings,
     configure_tracing,
     inject_trace_headers,
+    instrument_fastapi_app,
+    instrument_requests,
 )
 
 __all__ = [
@@ -12,6 +14,8 @@ __all__ = [
     "TracingSettings",
     "configure_tracing",
     "inject_trace_headers",
+    "instrument_fastapi_app",
+    "instrument_requests",
 ]
 
 _MONITORING_AVAILABLE = False

unique_toolkit/monitoring/tracing.py

@@ -9,7 +9,10 @@ from typing import TYPE_CHECKING, ClassVar, Self, TypeAlias, cast
 from pydantic import AliasChoices, Field
 from pydantic_settings import BaseSettings, SettingsConfigDict
 
+from unique_toolkit.app.find_env_file import find_env_file
+
 if TYPE_CHECKING:
+    from fastapi import FastAPI  # pyright: ignore[reportMissingImports]
     from opentelemetry.trace.propagation.tracecontext import (
         TraceContextTextMapPropagator,
     )
@@ -25,6 +28,10 @@ _OTEL_EXTRA_MESSAGE = (
     "OpenTelemetry tracing requires the unique_toolkit[otel] extra. "
     "Install it with: uv add 'unique-toolkit[otel]'"
 )
+_FASTAPI_EXTRA_MESSAGE = (
+    "FastAPI instrumentation requires the unique_toolkit[fastapi] extra. "
+    "Install it with: uv add 'unique-toolkit[fastapi]'"
+)
 
 
 class TraceExporter(StrEnum):
@@ -42,6 +49,8 @@ class TracingSettings(BaseSettings):
         env_prefix="OTEL_",
         env_ignore_empty=True,
         extra="ignore",
+        env_file=find_env_file(".env", required=False),
+        env_file_encoding="utf-8",
     )
 
     traces_exporter: TraceExporter | None = None
@@ -90,6 +99,11 @@ def _missing_otel_extra() -> ImportError:
     return ImportError(_OTEL_EXTRA_MESSAGE)
 
 
+def _missing_fastapi_extra() -> ImportError:
+    """Create a consistent error for an unavailable optional FastAPI dependency."""
+    return ImportError(_FASTAPI_EXTRA_MESSAGE)
+
+
 def _trace_context_propagator() -> TraceContextTextMapPropagator:
     """Return a W3C trace-context propagator without forwarding baggage."""
     try:
@@ -165,9 +179,12 @@ def configure_tracing(
 
     try:
         from opentelemetry import trace
+        from opentelemetry.baggage.propagation import W3CBaggagePropagator
         from opentelemetry.exporter.otlp.proto.http.trace_exporter import (
             OTLPSpanExporter,
         )
+        from opentelemetry.propagate import set_global_textmap
+        from opentelemetry.propagators.composite import CompositePropagator
         from opentelemetry.sdk.resources import Resource
         from opentelemetry.sdk.trace import TracerProvider
         from opentelemetry.sdk.trace.export import (
@@ -176,6 +193,9 @@ def configure_tracing(
             SimpleSpanProcessor,
         )
         from opentelemetry.trace import ProxyTracerProvider
+        from opentelemetry.trace.propagation.tracecontext import (
+            TraceContextTextMapPropagator,
+        )
     except ImportError as error:
         raise _missing_otel_extra() from error
 
@@ -196,6 +216,54 @@ def configure_tracing(
     if exporter is TraceExporter.CONSOLE:
         provider.add_span_processor(SimpleSpanProcessor(ConsoleSpanExporter()))
     else:
-        provider.add_span_processor(BatchSpanProcessor(OTLPSpanExporter()))
+        # Sizing pinned to match the Node services' instrumentation.ts.
+        provider.add_span_processor(
+            BatchSpanProcessor(
+                OTLPSpanExporter(),
+                max_queue_size=2048,
+                max_export_batch_size=512,
+                schedule_delay_millis=5000,
+            )
+        )
     trace.set_tracer_provider(provider)
+    # Trace-context + baggage, the shared subset of the Node services' propagator.
+    set_global_textmap(
+        CompositePropagator([TraceContextTextMapPropagator(), W3CBaggagePropagator()])
+    )
     return True
+
+
+def instrument_fastapi_app(app: FastAPI, *, excluded_urls: str | None = None) -> None:
+    """Trace inbound requests for an already-created FastAPI app instance.
+
+    Uses ``FastAPIInstrumentor.instrument_app`` rather than the global
+    ``FastAPIInstrumentor().instrument()``: the global form swaps out the
+    ``fastapi.FastAPI`` class, so it only takes effect for app objects created
+    afterward. Instrumenting the instance has no such ordering requirement.
+    """
+    try:
+        import fastapi  # noqa: F401  # pyright: ignore[reportMissingImports, reportUnusedImport]
+    except ImportError as error:
+        # opentelemetry-instrumentation-fastapi imports fastapi internally, so without
+        # this explicit check its own ImportError would be misreported below as a
+        # missing otel extra when fastapi itself is what's actually missing.
+        raise _missing_fastapi_extra() from error
+
+    try:
+        from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
+    except ImportError as error:
+        raise _missing_otel_extra() from error
+    FastAPIInstrumentor.instrument_app(app, excluded_urls=excluded_urls)
+
+
+def instrument_requests() -> None:
+    """Trace outbound calls made with the ``requests`` library.
+
+    Patches ``requests`` at module level, so callers that already imported it
+    are still covered.
+    """
+    try:
+        from opentelemetry.instrumentation.requests import RequestsInstrumentor
+    except ImportError as error:
+        raise _missing_otel_extra() from error
+    RequestsInstrumentor().instrument()

unique_toolkit/services/chat_service.py

@@ -22,7 +22,7 @@ from unique_toolkit._common.metadata_filter_scope import (
     merge_scope_clause_into_metadata_filter,
 )
 from unique_toolkit._common.utils.files import is_file_content, is_image_content
-from unique_toolkit.agentic.feature_flags import feature_flags
+from unique_toolkit.agentic.feature_flags import FeatureFlagNames
 from unique_toolkit.agentic.message_log_order import next_message_log_order
 from unique_toolkit.app.unique_settings import UniqueContext, UniqueSettings
 from unique_toolkit.chat.cancellation import CancellationWatcher
@@ -100,6 +100,7 @@ from unique_toolkit.content.schemas import (
     ContentSearchType,
 )
 from unique_toolkit.elicitation.service import ElicitationService
+from unique_toolkit.experimental.resources.feature_flags import is_flag_enabled
 from unique_toolkit.language_model.constants import (
     DEFAULT_COMPLETE_TEMPERATURE,
     DEFAULT_COMPLETE_TIMEOUT,
@@ -1805,8 +1806,9 @@ class ChatService(ChatServiceDeprecated):
 
             if not rate_limit_retry_config.log_message_on_retry:
                 return
-            if not feature_flags.enable_new_answers_ui_un_14411.is_enabled(
-                self._company_id
+            if not await is_flag_enabled(
+                FeatureFlagNames.enable_new_answers_ui_un_14411,
+                company_id=self._company_id,
             ):
                 return
 

{unique_toolkit-2026.34.0.dev6.dist-info → unique_toolkit-2026.34.0.dev8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.34.0.dev6
+Version: 2026.34.0.dev8
 Summary: 
 Author: Cedric Klinkert, Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri, Ejiroghene Oki-Peter, Rami Azouz, Ricardo Visini, Sanzio Monti
 Author-email: Cedric Klinkert <cedric.klinkert@unique.ch>, Martin Fadler <martin.fadler@unique.ch>, Sadique Sheik <sadique@unique.ch>, Fabian Schläpfer <fabian@unique.ch>, Pascal Hauri <pascal@unique.ch>, Ejiroghene Oki-Peter <oki-peter.ejiroghene@unique.ai>, Rami Azouz <rami.ext@unique.ch>, Ricardo Visini <ricardo.visini@unique.ch>, Sanzio Monti <sanzio.monti@unique.ai>
@@ -44,6 +44,8 @@ Requires-Dist: langchain-core>=1.0.0,<2 ; extra == 'langchain'
 Requires-Dist: prometheus-client>=0.24.1,<1 ; extra == 'monitoring'
 Requires-Dist: opentelemetry-sdk>=1.27,<2 ; extra == 'otel'
 Requires-Dist: opentelemetry-exporter-otlp-proto-http>=1.27,<2 ; extra == 'otel'
+Requires-Dist: opentelemetry-instrumentation-fastapi>=0.65b0,<1 ; extra == 'otel'
+Requires-Dist: opentelemetry-instrumentation-requests>=0.65b0,<1 ; extra == 'otel'
 Requires-Python: >=3.12
 Provides-Extra: fastapi
 Provides-Extra: langchain

{unique_toolkit-2026.34.0.dev6.dist-info → unique_toolkit-2026.34.0.dev8.dist-info}/RECORD

@@ -111,13 +111,13 @@ unique_toolkit/agentic/evaluation/tests/test_hallucination_utils.py,sha256=ogds7
 unique_toolkit/agentic/evaluation/tests/test_output_parser.py,sha256=KfltytmvqnPWLhmZpBXqcRmnlYorw_USwM5rkLVv8so,5179
 unique_toolkit/agentic/evaluation/tests/test_prompt_loaders.py,sha256=zBREdlKf5tdDyB8XSaNgpQv3-tuZJoYteeJrp6WMWDM,11897
 unique_toolkit/agentic/evaluation/utils.py,sha256=HmyPaDV8wdW-_gOjjW-wDaMKgdrsP5-SHP7OqTmGI_A,264
-unique_toolkit/agentic/feature_flags/__init__.py,sha256=LhE2cHoa9AYBOR7TjiIToOn46sttm9paKcrzE7gnDPM,149
-unique_toolkit/agentic/feature_flags/feature_flags.py,sha256=ul9-StpKO-4Kzq0GxTR4UlM-fRQsOD4BGr5aNmAiyBo,5315
+unique_toolkit/agentic/feature_flags/__init__.py,sha256=fBR9FaLrc9e8sdYDe1LwP9aNJt5aLLtAq3nIphVWkIk,191
+unique_toolkit/agentic/feature_flags/feature_flags.py,sha256=IqJMvWEJ49gRuIKkLfg4ZBcEPjaXO1KBuJ-SnRfLakg,6756
 unique_toolkit/agentic/history_manager/history_construction_with_contents.py,sha256=70lm8pYmQpb0HRpqzPTb4ikh8vVbvhfGR5S-D4Ftbbc,25306
 unique_toolkit/agentic/history_manager/history_manager.py,sha256=RIVeQxlCq25JwUY6WWQzz5zoOCouAK7shbxUztpgZPs,15903
-unique_toolkit/agentic/history_manager/loop_token_reducer.py,sha256=rzZztBfacFil2aX30Z23Pqhhf2nAWabi4StGvIcUmJs,30130
+unique_toolkit/agentic/history_manager/loop_token_reducer.py,sha256=lgo5zAq6gYLo7FFTNWe5sc6o80vnadKsr-cgEDP6hM8,30754
 unique_toolkit/agentic/history_manager/settings.py,sha256=Lrx3a8yXeU0MXrWlV6RZSHsgYWF9Zrs8LyR96OpqQ_k,1216
-unique_toolkit/agentic/history_manager/utils.py,sha256=hYgaahQq9gKm7V2Q5qnyZtqw5opO-MKH7wTGq9-iEgE,6455
+unique_toolkit/agentic/history_manager/utils.py,sha256=DiBip8xtah3I9-G-_Ul3OjlYKunrXR-5RJtzREHWV_0,6579
 unique_toolkit/agentic/loop_runner/__init__.py,sha256=91539BPsDVJ8qG0fyGPZ47vSLBupNnbjxkQf7RzqdBs,1412
 unique_toolkit/agentic/loop_runner/_iteration_handler_utils.py,sha256=y3tjTAtpkyL_Y7279_3v0fGaq9FUYN3UquwdIB3ZhgI,3831
 unique_toolkit/agentic/loop_runner/_responses_iteration_handler_utils.py,sha256=DuQRsToV0KLs7YcFjJVnEmrryhlX5q4isztx2jUxZcI,2064
@@ -173,7 +173,7 @@ unique_toolkit/agentic/tools/a2a/tool/__init__.py,sha256=9x8nGZsMcpF8ch0RKoelGPl
 unique_toolkit/agentic/tools/a2a/tool/_memory.py,sha256=w8bxjokrqHQZgApd55b5rHXF-DpgJwaKTg4CvLBLamc,1034
 unique_toolkit/agentic/tools/a2a/tool/_schema.py,sha256=wMwyunViTnxaURvenkATEvyfXn5LvLaP0HxbYqdZGls,158
 unique_toolkit/agentic/tools/a2a/tool/config.py,sha256=v4rZQU42RjUTC5Md-Cye1rUIWDlHpuhc3MzcyvpGQNw,9748
-unique_toolkit/agentic/tools/a2a/tool/service.py,sha256=j3kQZ1P-gYyXE2SmbbyNIVC1tIgTm8pJkCwrfUB9YU8,24173
+unique_toolkit/agentic/tools/a2a/tool/service.py,sha256=_N6dxN3qsps77de4t3VhOxH5GLD5cGM83j4I8Pp2-3A,24257
 unique_toolkit/agentic/tools/a2a/tool/test/test_service_utils.py,sha256=0w3721KiJot54PIy105DNiOGviLiiKNHi6kECDGvO1k,27199
 unique_toolkit/agentic/tools/agent_chunks_hanlder.py,sha256=eO0g46wZOQlmWqYYVGiszRVe8K2LX-9qW30ZCdGg5eE,1943
 unique_toolkit/agentic/tools/config.py,sha256=AOtBn2tL1rIoPPzl1csAHLD5aetYqPL-jAqkIoSBxyE,7468
@@ -201,7 +201,7 @@ unique_toolkit/agentic/tools/factory.py,sha256=YK9uh2EqotQFb9stsXs-A8lsNaxklTCd_
 unique_toolkit/agentic/tools/mcp/__init__.py,sha256=RLF_p-LDRC7GhiB3fdCi4u3bh6V9PY_w26fg61BLyco,122
 unique_toolkit/agentic/tools/mcp/manager.py,sha256=EfWbfNhB8Lnx2EqBgUXhK7klFLaBv2eezvYq8J0TXyo,3808
 unique_toolkit/agentic/tools/mcp/models.py,sha256=DaBVzwJ4dqtgzm8Mv0-NOGBU04zyVe-RgWntBhkpmnE,742
-unique_toolkit/agentic/tools/mcp/tool_wrapper.py,sha256=6rrJ83dSzyiteOPXl5ZTca8hSPkH8aOGNjI41IcRp78,15544
+unique_toolkit/agentic/tools/mcp/tool_wrapper.py,sha256=fE5nVALJ-wVZkkbbJ_ySRLBhAAR6sCAAu1lUxQQVWC4,15596
 unique_toolkit/agentic/tools/names.py,sha256=MDF0ONLrxeC-iz--EtyrHAox_RN6zasxo5MSbXQYGyw,412
 unique_toolkit/agentic/tools/openai_builtin/__init__.py,sha256=Q6d9mhsLRVF3erT2yMxLld62D3FTZ8nbovI4vLv4Nso,701
 unique_toolkit/agentic/tools/openai_builtin/base.py,sha256=QWF-GL_fkJqzGPpxvtg69-TB_IjsrxokeJVQESzH-18,1415
@@ -217,8 +217,8 @@ unique_toolkit/agentic/tools/openai_builtin/code_interpreter/builder/builder.py,
 unique_toolkit/agentic/tools/openai_builtin/code_interpreter/config.py,sha256=C4Vrj_xLgLB8AEUz-cp0JpXAIwYwz-VaSRh-xAG7PW8,2381
 unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/__init__.py,sha256=SzEiUqxIw8D4_51SvTt-8oTOYTGirYkmH4J6SgojzME,634
 unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/artifacts.py,sha256=wAgDrcXZbCdDbNWNpbMigBUXySzbKQXSfg3KsZfAWfs,3115
-unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/code_display.py,sha256=Q4crOkYG-C_uTSc7rW14cP1H4IWxsKpfuftafDHbjEo,2897
-unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/generated_files.py,sha256=OIbaRhjorsNe6LhVJYVvLxxMKwphc6ot13vi58axnnM,62861
+unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/code_display.py,sha256=WC__H2lhprchVVKsQzPeckhq-EcYg1ePYtrHIAoVzpQ,3165
+unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/generated_files.py,sha256=CJ9qVZ720ekyYUyUezsXpQEteNcmYJxw_Jb-JRdhSqI,63127
 unique_toolkit/agentic/tools/openai_builtin/code_interpreter/schemas.py,sha256=W7egCmYrPrzic0aXsu9wY5fvQ8y_cl4Tzq_cjVs0mZs,542
 unique_toolkit/agentic/tools/openai_builtin/code_interpreter/tool/__init__.py,sha256=r6yXmGTLJzXNq5laHOlre9veu--VbCQVlDRVm2QUGzI,326
 unique_toolkit/agentic/tools/openai_builtin/code_interpreter/tool/config.py,sha256=sbLV-fYt_veBAdonqAHY9sm-eHxpjIZgc1KVbm03pZY,19042
@@ -358,10 +358,10 @@ unique_toolkit/experimental/resources/facades/__init__.py,sha256=imKXMz2nsQcg5zm
 unique_toolkit/experimental/resources/facades/identity/__init__.py,sha256=7r-1_gr81Idc6AJ1IRO6GCHx7psmC81BtD15BWxrlYs,1640
 unique_toolkit/experimental/resources/facades/identity/service.py,sha256=A5I7QkNy-Su9FmOkrl8z1iwWhqzoo6KgxEgcdvnWN2M,4287
 unique_toolkit/experimental/resources/feature_flags/README.md,sha256=ChPgYn4-SzY1UyF3OQgBxHzMk6JH_hVKfzHsTcKhcXo,6122
-unique_toolkit/experimental/resources/feature_flags/__init__.py,sha256=eAY6OUfQbxHIyW9Vm5hDAQKkiXkPsyyuhnmV7yaI870,1627
+unique_toolkit/experimental/resources/feature_flags/__init__.py,sha256=C8k2YyJBTEJyKP0ZLH3zvcGwVIHShiz2XHleKbgFIl0,1978
 unique_toolkit/experimental/resources/feature_flags/_graphql_client.py,sha256=a0KxF0EJk7baYChVKFy8bllIJShHKELyQReFd_B9bXs,1743
 unique_toolkit/experimental/resources/feature_flags/_ttl_cache.py,sha256=lPEdWX1JkP2985M1AqQ0uYyWGlagXB0LvOtQPltbgJo,2520
-unique_toolkit/experimental/resources/feature_flags/client.py,sha256=TduDaRLDd8jnpopWdpy0mXL9C_lP5xNXIaElCGbd8HI,8701
+unique_toolkit/experimental/resources/feature_flags/client.py,sha256=NAGCjsQpolJNaSJk0dZbpCouUIYbKbFzOOpEqodvmE8,8739
 unique_toolkit/experimental/resources/feature_flags/schemas.py,sha256=chLsBNkIL4BOvGrwrGeixy7BmNSi2BagXD6Jr2fZcqs,602
 unique_toolkit/experimental/resources/feature_flags/settings.py,sha256=3L_c-0zxj4cpuHd3NL3mZ2kN35V6npgh6_hSwu6Gk00,1071
 unique_toolkit/experimental/resources/group/__init__.py,sha256=qPi2DWgWLZw1ku0zi-UsPdornfl8pczW6QSYT5P8CzQ,2098
@@ -402,14 +402,14 @@ unique_toolkit/language_model/service.py,sha256=hDoJBBgG_wWs8WuC_c8tnmK8cNu8o5Eh
 unique_toolkit/language_model/settings.py,sha256=7szohExI-qmTfUf2Zp5AB5TzrsZy1UekAGyXx_UaPMc,1524
 unique_toolkit/language_model/stream_transform.py,sha256=iNNGpLY_7G6Bpp9Y4KtbIkQm3WB4yZt75BXcqx0DVkI,2775
 unique_toolkit/language_model/utils.py,sha256=bPQ4l6_YO71w-zaIPanUUmtbXC1_hCvLK0tAFc3VCRc,1902
-unique_toolkit/monitoring/__init__.py,sha256=auxvvOBwYrCD3TqEX0sIupDYx9nbCt2u4XN0Sy55qKI,1376
+unique_toolkit/monitoring/__init__.py,sha256=XHRzPx4LKcbKruRrf_gNv50amhvhy8Go2IkOoHMuLsw,1486
 unique_toolkit/monitoring/gunicorn.py,sha256=urTFFS5zYHKoxb1TEQHcLqb2byi-aYgFRHGwgYRzYYI,844
 unique_toolkit/monitoring/middleware.py,sha256=MNsUuKgt74mSnUwttXTB_AxX546Pz2zKoAP9OewVrwQ,4531
 unique_toolkit/monitoring/registry.py,sha256=De2Bx5QBBRJQY2MM3xtntiJ2JU2FoKkVT7rNMG7ewgA,5874
-unique_toolkit/monitoring/tracing.py,sha256=86yUIF8HrHVrkegOBOM7GqpbmVrYP07AZ4HtP-3X9QY,7128
+unique_toolkit/monitoring/tracing.py,sha256=0lmLuQjyyC3iFpNTw8rOyCpEZ6PTvp16nVbzcO9d2sQ,10043
 unique_toolkit/protocols/support.py,sha256=x5edzbBmKAUuvm3oLFh52gkgQsqR5LU7M0t8RHK0DZk,5161
 unique_toolkit/services/__init__.py,sha256=b9Wl7Wvz1JmZDbDexIZzBoXZ0wscXuczGAH8LoL-Qak,572
-unique_toolkit/services/chat_service.py,sha256=ofhNHCW7kz6xXN4pUTBdDEYovZ7PXhUglPDgez8Qov0,99142
+unique_toolkit/services/chat_service.py,sha256=6lVMSwvi2ztPebK6meo9BVLOL_Wdje5wPSUBp5dMZIg,99268
 unique_toolkit/services/factory.py,sha256=7gNeanyHkgX23Iz7RMCQTCkE9gI5asCI11nXxM0jjx4,9832
 unique_toolkit/services/knowledge_base.py,sha256=6dM4abcXR9y2VYoR6SK79gwjrzb83Wm8jMPPIt5z95o,43461
 unique_toolkit/short_term_memory/__init__.py,sha256=2mI3AUrffgH7Yt-xS57EGqnHf7jnn6xquoKEhJqk3Wg,185
@@ -420,6 +420,6 @@ unique_toolkit/short_term_memory/service.py,sha256=2JMBEBn6GoyK-a0xvdcWpPft-3C74
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=vRe0aMJWuVIK-VsqomgxQImzke7MDmwgSxxJfYCa7zw,1386
 unique_toolkit/test_utilities/events.py,sha256=FLrhMh0A_mQutAUjoD0FdMcVQ4Gq31r35OVGzRjvLro,6760
-unique_toolkit-2026.34.0.dev6.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-unique_toolkit-2026.34.0.dev6.dist-info/METADATA,sha256=0dze18eXJM17YPbcVoW6o81JxC5ovpkH8z__PuEopsg,7656
-unique_toolkit-2026.34.0.dev6.dist-info/RECORD,,
+unique_toolkit-2026.34.0.dev8.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+unique_toolkit-2026.34.0.dev8.dist-info/METADATA,sha256=xKbzXHZ9elQ5mo-UZHvCPfjCK1Hfr3ds0Xum8Z1v8Ns,7821
+unique_toolkit-2026.34.0.dev8.dist-info/RECORD,,