chore: distrogen enables telemetry by default

# What does this PR do?


## Test Plan
Telemetry provider was added to all distributions in the latest commit
but the protocol mapping was missing, causing a KeyError when the stack
tried to validate provider compliance.
This commit is contained in:
Eric Huang 2025-10-16 10:47:35 -07:00
parent f22aaef42f
commit b7c276ea6d
14 changed files with 33 additions and 0 deletions

View file

@ -26,6 +26,7 @@ from llama_stack.apis.safety import Safety
from llama_stack.apis.scoring import Scoring
from llama_stack.apis.scoring_functions import ScoringFunctions
from llama_stack.apis.shields import Shields
from llama_stack.apis.telemetry import Telemetry
from llama_stack.apis.tools import ToolGroups, ToolRuntime
from llama_stack.apis.vector_io import VectorIO
from llama_stack.apis.version import LLAMA_STACK_API_V1ALPHA
@ -94,6 +95,7 @@ def api_protocol_map(external_apis: dict[Api, ExternalApiSpec] | None = None) ->
Api.files: Files,
Api.prompts: Prompts,
Api.conversations: Conversations,
Api.telemetry: Telemetry,
}
if external_apis:

View file

@ -237,3 +237,5 @@ tool_groups:
provider_id: rag-runtime
server:
port: 8321
telemetry:
enabled: true

View file

@ -122,3 +122,5 @@ tool_groups:
provider_id: rag-runtime
server:
port: 8321
telemetry:
enabled: true

View file

@ -113,3 +113,5 @@ tool_groups:
provider_id: rag-runtime
server:
port: 8321
telemetry:
enabled: true

View file

@ -135,3 +135,5 @@ tool_groups:
provider_id: rag-runtime
server:
port: 8321
telemetry:
enabled: true

View file

@ -120,3 +120,5 @@ tool_groups:
provider_id: rag-runtime
server:
port: 8321
telemetry:
enabled: true

View file

@ -118,3 +118,5 @@ tool_groups:
provider_id: rag-runtime
server:
port: 8321
telemetry:
enabled: true

View file

@ -97,3 +97,5 @@ tool_groups:
provider_id: rag-runtime
server:
port: 8321
telemetry:
enabled: true

View file

@ -233,3 +233,5 @@ tool_groups:
provider_id: rag-runtime
server:
port: 8321
telemetry:
enabled: true

View file

@ -104,3 +104,5 @@ tool_groups:
provider_id: rag-runtime
server:
port: 8321
telemetry:
enabled: true

View file

@ -240,3 +240,5 @@ tool_groups:
provider_id: rag-runtime
server:
port: 8321
telemetry:
enabled: true

View file

@ -237,3 +237,5 @@ tool_groups:
provider_id: rag-runtime
server:
port: 8321
telemetry:
enabled: true

View file

@ -25,6 +25,7 @@ from llama_stack.core.datatypes import (
ModelInput,
Provider,
ShieldInput,
TelemetryConfig,
ToolGroupInput,
)
from llama_stack.core.distribution import get_provider_registry
@ -182,6 +183,11 @@ class RunConfigSettings(BaseModel):
metadata_store: dict | None = None
inference_store: dict | None = None
conversations_store: dict | None = None
telemetry: TelemetryConfig | None = None
def model_post_init(self, __context__: Any) -> None:
if self.telemetry is None:
self.telemetry = TelemetryConfig(enabled=True)
def run_config(
self,
@ -256,6 +262,7 @@ class RunConfigSettings(BaseModel):
"server": {
"port": 8321,
},
"telemetry": self.telemetry.model_dump(exclude_none=True) if self.telemetry else None,
}

View file

@ -114,3 +114,5 @@ tool_groups:
provider_id: rag-runtime
server:
port: 8321
telemetry:
enabled: true