ddapm-test-agent 1.35.0__py3-none-any.whl → 1.37.0__py3-none-any.whl

ddapm_test_agent/agent.py

@@ -39,7 +39,9 @@ from aiohttp.web import middleware
 from grpc import aio as grpc_aio
 from msgpack.exceptions import ExtraData as MsgPackExtraDataException
 from multidict import CIMultiDict
+from opentelemetry.proto.collector.logs.v1.logs_service_pb2 import ExportLogsServiceResponse
 from opentelemetry.proto.collector.logs.v1.logs_service_pb2_grpc import add_LogsServiceServicer_to_server
+from opentelemetry.proto.collector.metrics.v1.metrics_service_pb2 import ExportMetricsServiceResponse
 from opentelemetry.proto.collector.metrics.v1.metrics_service_pb2_grpc import add_MetricsServiceServicer_to_server
 
 from . import _get_version
@@ -323,12 +325,15 @@ class Agent:
             "/v1.0/traces",
             "/v0.6/stats",
             "/v0.7/config",
+            "/info",
             "/telemetry/proxy/api/v2/apmtelemetry",
             "/v0.1/pipeline_stats",
             "/tracer_flare/v1",
             "/evp_proxy/v2/api/v2/llmobs",
             "/evp_proxy/v2/api/intake/llm-obs/v1/eval-metric",
             "/evp_proxy/v2/api/intake/llm-obs/v2/eval-metric",
+            "/evp_proxy/v2/api/v2/exposures",
+            "/evp_proxy/v4/api/v2/errorsintake",
         ]
 
         # Note that sessions are not cleared at any point since we don't know
@@ -721,7 +726,9 @@ class Agent:
             num_resource_logs,
             total_log_records,
         )
-        return web.HTTPOk()
+        return web.Response(
+            body=ExportLogsServiceResponse().SerializeToString(), status=200, content_type="application/x-protobuf"
+        )
 
     async def handle_v1_metrics(self, request: Request) -> web.Response:
         metrics_data = self._decode_v1_metrics(request)
@@ -736,7 +743,9 @@ class Agent:
             num_resource_metrics,
             total_metrics,
         )
-        return web.HTTPOk()
+        return web.Response(
+            body=ExportMetricsServiceResponse().SerializeToString(), status=200, content_type="application/x-protobuf"
+        )
 
     async def handle_v07_remoteconfig(self, request: Request) -> web.Response:
         """Emulates Remote Config endpoint: /v0.7/config"""
@@ -801,6 +810,12 @@ class Agent:
     async def handle_evp_proxy_v2_llmobs_eval_metric(self, request: Request) -> web.Response:
         return web.HTTPOk()
 
+    async def handle_evp_proxy_v2_api_v2_exposures(self, request: Request) -> web.Response:
+        return web.HTTPOk()
+
+    async def handle_evp_proxy_v4_api_v2_errorsintake(self, request: Request) -> web.Response:
+        return web.HTTPOk()
+
     async def handle_put_tested_integrations(self, request: Request) -> web.Response:
         # we need to store the request manually since this is not a real DD agent endpoint
         await self._store_request(request)
@@ -878,6 +893,7 @@ class Agent:
                     "/v0.7/config",
                     "/tracer_flare/v1",
                     "/evp_proxy/v2/",
+                    "/evp_proxy/v4/",
                 ],
                 "feature_flags": [],
                 "config": {},
@@ -1124,6 +1140,8 @@ class Agent:
                 self.handle_v1_tracer_flare,
                 self.handle_evp_proxy_v2_api_v2_llmobs,
                 self.handle_evp_proxy_v2_llmobs_eval_metric,
+                self.handle_evp_proxy_v2_api_v2_exposures,
+                self.handle_evp_proxy_v4_api_v2_errorsintake,
                 self.handle_v1_logs,
                 self.handle_v1_metrics,
             ):
@@ -1417,6 +1435,8 @@ class Agent:
                 "/evp_proxy/v2/api/v2/llmobs": self.handle_evp_proxy_v2_api_v2_llmobs,
                 "/evp_proxy/v2/api/intake/llm-obs/v1/eval-metric": self.handle_evp_proxy_v2_llmobs_eval_metric,
                 "/evp_proxy/v2/api/intake/llm-obs/v2/eval-metric": self.handle_evp_proxy_v2_llmobs_eval_metric,
+                "/evp_proxy/v2/api/v2/exposures": self.handle_evp_proxy_v2_api_v2_exposures,
+                "/evp_proxy/v4/api/v2/errorsintake": self.handle_evp_proxy_v4_api_v2_errorsintake,
                 "/info": self.handle_info,
                 # Test endpoints
                 "/test/session/start": self.handle_session_start,
@@ -1581,18 +1601,30 @@ def make_app(
     vcr_ci_mode: bool,
     vcr_provider_map: str,
     vcr_ignore_headers: str,
+    enable_web_ui: bool = False,
 ) -> web.Application:
     agent = Agent()
+
+    # Build middleware list conditionally
+    middlewares = []
+    if enable_web_ui:
+        from .web import request_response_capture_middleware
+
+        middlewares.append(request_response_capture_middleware)
+    middlewares.extend(
+        [
+            handle_exception_middleware,
+            agent.check_failure_middleware,
+            agent.store_request_middleware,
+            agent.request_forwarder_middleware,
+            session_token_middleware,
+            agent.vcr_proxy_suffix_middleware,
+        ]
+    )
+
     app = web.Application(
         client_max_size=int(100e6),  # 100MB - arbitrary
-        middlewares=[
-            handle_exception_middleware,  # type: ignore
-            agent.check_failure_middleware,  # type: ignore
-            agent.store_request_middleware,  # type: ignore
-            agent.request_forwarder_middleware,  # type: ignore
-            session_token_middleware,  # type: ignore
-            agent.vcr_proxy_suffix_middleware,  # type: ignore
-        ],
+        middlewares=middlewares,
     )
     app.add_routes(
         [
@@ -1615,6 +1647,8 @@ def make_app(
             web.post("/evp_proxy/v2/api/v2/llmobs", agent.handle_evp_proxy_v2_api_v2_llmobs),
             web.post("/evp_proxy/v2/api/intake/llm-obs/v1/eval-metric", agent.handle_evp_proxy_v2_llmobs_eval_metric),
             web.post("/evp_proxy/v2/api/intake/llm-obs/v2/eval-metric", agent.handle_evp_proxy_v2_llmobs_eval_metric),
+            web.post("/evp_proxy/v2/api/v2/exposures", agent.handle_evp_proxy_v2_api_v2_exposures),
+            web.post("/evp_proxy/v4/api/v2/errorsintake", agent.handle_evp_proxy_v4_api_v2_errorsintake),
             web.get("/info", agent.handle_info),
             web.get("/test/session/start", agent.handle_session_start),
             web.get("/test/session/clear", agent.handle_session_clear),
@@ -1964,6 +1998,18 @@ def main(args: Optional[List[str]] = None) -> None:
         default=os.environ.get("VCR_IGNORE_HEADERS", ""),
         help="Comma-separated list of headers to ignore when recording VCR cassettes.",
     )
+    parser.add_argument(
+        "--web-ui-port",
+        type=int,
+        default=int(os.environ.get("WEB_UI_PORT", 0)),
+        help="Port to serve the optional web UI (default: disabled). Example: --web-ui-port=8080",
+    )
+    parser.add_argument(
+        "--max-requests",
+        type=int,
+        default=int(os.environ.get("MAX_REQUESTS", 200)),
+        help="Maximum number of requests to keep in memory for the UI (default: 200). Older requests are discarded when limit is reached.",
+    )
     parsed_args = parser.parse_args(args=args)
     logging.basicConfig(level=parsed_args.log_level)
 
@@ -2010,6 +2056,7 @@ def main(args: Optional[List[str]] = None) -> None:
         vcr_ci_mode=parsed_args.vcr_ci_mode,
         vcr_provider_map=parsed_args.vcr_provider_map,
         vcr_ignore_headers=parsed_args.vcr_ignore_headers,
+        enable_web_ui=parsed_args.web_ui_port > 0,
     )
 
     # Validate port configuration
@@ -2019,6 +2066,13 @@ def main(args: Optional[List[str]] = None) -> None:
         raise ValueError("APM and OTLP GRPC ports cannot be the same")
     if parsed_args.otlp_http_port == parsed_args.otlp_grpc_port:
         raise ValueError("OTLP HTTP and GRPC ports cannot be the same")
+    if parsed_args.web_ui_port > 0:
+        if parsed_args.web_ui_port == parsed_args.port:
+            raise ValueError("Web UI and APM ports cannot be the same")
+        if parsed_args.web_ui_port == parsed_args.otlp_http_port:
+            raise ValueError("Web UI and OTLP HTTP ports cannot be the same")
+        if parsed_args.web_ui_port == parsed_args.otlp_grpc_port:
+            raise ValueError("Web UI and OTLP GRPC ports cannot be the same")
 
     # Get the shared agent instance from the main app
     agent = app["agent"]
@@ -2036,15 +2090,41 @@ def main(args: Optional[List[str]] = None) -> None:
 
     otlp_http_app = make_otlp_http_app(agent)
 
+    # Create Web UI app if enabled
+    web_ui_app = None
+    if parsed_args.web_ui_port > 0:
+        from .web import WebUI
+
+        # Pass configuration directly to WebUI
+        web_ui_config = {
+            "snapshot_dir": parsed_args.snapshot_dir,
+            "vcr_cassettes_directory": parsed_args.vcr_cassettes_directory,
+            "disable_error_responses": parsed_args.disable_error_responses,
+            "web_ui_port": parsed_args.web_ui_port,
+            "max_requests": parsed_args.max_requests,
+        }
+        web_ui = WebUI(agent, config=web_ui_config)
+        web_ui_app = web_ui.make_app()
+        # Store WebUI instance reference for middleware access
+        web_ui_app._webui_instance = web_ui
+        # Also store on main app for middleware access
+        app._webui_instance = web_ui
+
     async def run_servers():
         """Run APM and OTLP HTTP servers concurrently."""
-        # Create runners for both apps
+        # Create runners for apps
         apm_runner = web.AppRunner(app)
         await apm_runner.setup()
 
         otlp_http_runner = web.AppRunner(otlp_http_app)
         await otlp_http_runner.setup()
 
+        # Create Web UI runner if enabled
+        web_ui_runner = None
+        if web_ui_app is not None:
+            web_ui_runner = web.AppRunner(web_ui_app)
+            await web_ui_runner.setup()
+
         # Start GRPC server if available (async creation)
         otlp_grpc_server = await make_otlp_grpc_server_async(
             agent, parsed_args.otlp_http_port, parsed_args.otlp_grpc_port
@@ -2058,12 +2138,23 @@ def main(args: Optional[List[str]] = None) -> None:
 
         otlp_http_site = web.TCPSite(otlp_http_runner, port=parsed_args.otlp_http_port)
 
-        # Start both servers concurrently
-        await asyncio.gather(apm_site.start(), otlp_http_site.start())
+        # Create Web UI site if enabled
+        web_ui_site = None
+        if web_ui_runner is not None:
+            web_ui_site = web.TCPSite(web_ui_runner, port=parsed_args.web_ui_port)
+
+        # Start servers concurrently
+        sites_to_start = [apm_site.start(), otlp_http_site.start()]
+        if web_ui_site is not None:
+            sites_to_start.append(web_ui_site.start())
+
+        await asyncio.gather(*sites_to_start)
 
         print(f"======== Running APM server on port {parsed_args.port} ========")
         print(f"======== Running OTLP HTTP server on port {parsed_args.otlp_http_port} ========")
         print(f"======== Running OTLP GRPC server on port {parsed_args.otlp_grpc_port} ========")
+        if web_ui_site is not None:
+            print(f"======== Running Web UI on port {parsed_args.web_ui_port} ========")
         print("(Press CTRL+C to quit)")
 
         try:
@@ -2074,6 +2165,8 @@ def main(args: Optional[List[str]] = None) -> None:
         finally:
             await apm_runner.cleanup()
             await otlp_http_runner.cleanup()
+            if web_ui_runner is not None:
+                await web_ui_runner.cleanup()
             await otlp_grpc_server.stop(grace=5.0)
 
     # Run the servers

ddapm_test_agent/remoteconfig.py

@@ -26,7 +26,7 @@ class RemoteConfigServer:
         self._update_response(token, data)
 
     @staticmethod
-    def _build_config_path_response(path: str, msg: str) -> Dict[str, Any]:
+    def _build_config_path_response(path: str, msg: Any) -> Dict[str, Any]:
         expires_date = datetime.datetime.strftime(
             datetime.datetime.now() + datetime.timedelta(days=1), "%Y-%m-%dT%H:%M:%SZ"
         )
@@ -84,7 +84,7 @@ class RemoteConfigServer:
         }
         return remote_config_payload
 
-    def create_config_path_response(self, token: Optional[str], path: str, msg: str) -> None:
+    def create_config_path_response(self, token: Optional[str], path: str, msg: Any) -> None:
         remote_config_payload = self._build_config_path_response(path, msg)
         self.create_config_response(token, remote_config_payload)