Extend Pyroscope configuration

README.md

@@ -373,7 +373,11 @@ class YourSettings(FastStreamSettings):
 
 ### [Pyroscope](https://pyroscope.io)
 
-To integrate Pyroscope, specify the `pyroscope_endpoint`. The `service_name` will be used as the application name. You can also set `pyroscope_sample_rate` (default is 100).
+To integrate Pyroscope, specify the `pyroscope_endpoint`.
+
+- The `opentelemetry_service_name` will be used as the application name.
+- `service_namespace` tag will be added with `opentelemetry_namespace` value.
+- You can also set `pyroscope_sample_rate`, `pyroscope_auth_token`, `pyroscope_tags` and `pyroscope_additional_params` — params that will be passed to `pyroscope.configure`.
 
 When both Pyroscope and OpenTelemetry are enabled, profile span IDs will be included in traces using [`pyroscope-otel`](https://github.com/grafana/otel-profiling-python) for correlation.
 

microbootstrap/instruments/pyroscope_instrument.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
+import typing
 
-import pydantic  # noqa: TC002
+import pydantic
 
 from microbootstrap.instruments.base import BaseInstrumentConfig, Instrument
 
@@ -13,9 +14,14 @@
 
 class PyroscopeConfig(BaseInstrumentConfig):
     service_name: str = "micro-service"
+    opentelemetry_service_name: str | None = None
+    opentelemetry_namespace: str | None = None
 
     pyroscope_endpoint: pydantic.HttpUrl | None = None
     pyroscope_sample_rate: int = 100
+    pyroscope_auth_token: str | None = None
+    pyroscope_tags: dict[str, str] = pydantic.Field(default_factory=dict)
+    pyroscope_additional_params: dict[str, typing.Any] = pydantic.Field(default_factory=dict)
 
 
 class PyroscopeInstrument(Instrument[PyroscopeConfig]):
@@ -30,9 +36,17 @@ def teardown(self) -> None:
 
     def bootstrap(self) -> None:
         pyroscope.configure(
-            application_name=self.instrument_config.service_name,
+            application_name=self.instrument_config.opentelemetry_service_name or self.instrument_config.service_name,
             server_address=str(self.instrument_config.pyroscope_endpoint),
+            auth_token=self.instrument_config.pyroscope_auth_token or "",
             sample_rate=self.instrument_config.pyroscope_sample_rate,
+            tags=(
+                {"service_namespace": self.instrument_config.opentelemetry_namespace}
+                if self.instrument_config.opentelemetry_namespace
+                else {}
+            )
+            | self.instrument_config.pyroscope_tags,
+            **self.instrument_config.pyroscope_additional_params,
         )
 
     @classmethod