diff --git a/llama_stack/providers/inline/scoring/braintrust/config.py b/llama_stack/providers/inline/scoring/braintrust/config.py index fae0b17eb..25732b3d7 100644 --- a/llama_stack/providers/inline/scoring/braintrust/config.py +++ b/llama_stack/providers/inline/scoring/braintrust/config.py @@ -11,3 +11,9 @@ class BraintrustScoringConfig(BaseModel): default=None, description="The OpenAI API Key", ) + + @classmethod + def sample_run_config(cls, **kwargs) -> Dict[str, Any]: + return { + "openai_api_key": "${env.OPENAI_API_KEY}", + } diff --git a/llama_stack/templates/bedrock/run.yaml b/llama_stack/templates/bedrock/run.yaml index db0ee9d85..c35ef0847 100644 --- a/llama_stack/templates/bedrock/run.yaml +++ b/llama_stack/templates/bedrock/run.yaml @@ -63,7 +63,8 @@ providers: config: {} - provider_id: braintrust provider_type: inline::braintrust - config: {} + config: + openai_api_key: ${env.OPENAI_API_KEY} metadata_store: namespace: null type: sqlite diff --git a/llama_stack/templates/fireworks/run.yaml b/llama_stack/templates/fireworks/run.yaml index c75db478d..e9c6f5977 100644 --- a/llama_stack/templates/fireworks/run.yaml +++ b/llama_stack/templates/fireworks/run.yaml @@ -65,7 +65,8 @@ providers: config: {} - provider_id: braintrust provider_type: inline::braintrust - config: {} + config: + openai_api_key: ${env.OPENAI_API_KEY} metadata_store: namespace: null type: sqlite diff --git a/llama_stack/templates/hf-endpoint/run-with-safety.yaml b/llama_stack/templates/hf-endpoint/run-with-safety.yaml index 678857201..c5037b37f 100644 --- a/llama_stack/templates/hf-endpoint/run-with-safety.yaml +++ b/llama_stack/templates/hf-endpoint/run-with-safety.yaml @@ -70,7 +70,8 @@ providers: config: {} - provider_id: braintrust provider_type: inline::braintrust - config: {} + config: + openai_api_key: ${env.OPENAI_API_KEY} metadata_store: namespace: null type: sqlite diff --git a/llama_stack/templates/hf-endpoint/run.yaml b/llama_stack/templates/hf-endpoint/run.yaml index c062c6c98..2382971c7 100644 --- a/llama_stack/templates/hf-endpoint/run.yaml +++ b/llama_stack/templates/hf-endpoint/run.yaml @@ -65,7 +65,8 @@ providers: config: {} - provider_id: braintrust provider_type: inline::braintrust - config: {} + config: + openai_api_key: ${env.OPENAI_API_KEY} metadata_store: namespace: null type: sqlite diff --git a/llama_stack/templates/hf-serverless/run-with-safety.yaml b/llama_stack/templates/hf-serverless/run-with-safety.yaml index 4a14ba093..51ddf2d71 100644 --- a/llama_stack/templates/hf-serverless/run-with-safety.yaml +++ b/llama_stack/templates/hf-serverless/run-with-safety.yaml @@ -70,7 +70,8 @@ providers: config: {} - provider_id: braintrust provider_type: inline::braintrust - config: {} + config: + openai_api_key: ${env.OPENAI_API_KEY} metadata_store: namespace: null type: sqlite diff --git a/llama_stack/templates/hf-serverless/run.yaml b/llama_stack/templates/hf-serverless/run.yaml index 268efddc4..23df29f8c 100644 --- a/llama_stack/templates/hf-serverless/run.yaml +++ b/llama_stack/templates/hf-serverless/run.yaml @@ -65,7 +65,8 @@ providers: config: {} - provider_id: braintrust provider_type: inline::braintrust - config: {} + config: + openai_api_key: ${env.OPENAI_API_KEY} metadata_store: namespace: null type: sqlite diff --git a/llama_stack/templates/meta-reference-gpu/run-with-safety.yaml b/llama_stack/templates/meta-reference-gpu/run-with-safety.yaml index 963679665..256191b89 100644 --- a/llama_stack/templates/meta-reference-gpu/run-with-safety.yaml +++ b/llama_stack/templates/meta-reference-gpu/run-with-safety.yaml @@ -72,7 +72,8 @@ providers: config: {} - provider_id: braintrust provider_type: inline::braintrust - config: {} + config: + openai_api_key: ${env.OPENAI_API_KEY} metadata_store: namespace: null type: sqlite diff --git a/llama_stack/templates/meta-reference-gpu/run.yaml b/llama_stack/templates/meta-reference-gpu/run.yaml index a74cde768..1fcba7438 100644 --- a/llama_stack/templates/meta-reference-gpu/run.yaml +++ b/llama_stack/templates/meta-reference-gpu/run.yaml @@ -66,7 +66,8 @@ providers: config: {} - provider_id: braintrust provider_type: inline::braintrust - config: {} + config: + openai_api_key: ${env.OPENAI_API_KEY} metadata_store: namespace: null type: sqlite diff --git a/llama_stack/templates/meta-reference-quantized-gpu/run.yaml b/llama_stack/templates/meta-reference-quantized-gpu/run.yaml index 5aada0fe6..411128ef3 100644 --- a/llama_stack/templates/meta-reference-quantized-gpu/run.yaml +++ b/llama_stack/templates/meta-reference-quantized-gpu/run.yaml @@ -68,7 +68,8 @@ providers: config: {} - provider_id: braintrust provider_type: inline::braintrust - config: {} + config: + openai_api_key: ${env.OPENAI_API_KEY} metadata_store: namespace: null type: sqlite diff --git a/llama_stack/templates/ollama/run-with-safety.yaml b/llama_stack/templates/ollama/run-with-safety.yaml index 2ab0f78f0..63e2964ce 100644 --- a/llama_stack/templates/ollama/run-with-safety.yaml +++ b/llama_stack/templates/ollama/run-with-safety.yaml @@ -64,7 +64,8 @@ providers: config: {} - provider_id: braintrust provider_type: inline::braintrust - config: {} + config: + openai_api_key: ${env.OPENAI_API_KEY} metadata_store: namespace: null type: sqlite diff --git a/llama_stack/templates/ollama/run.yaml b/llama_stack/templates/ollama/run.yaml index c5206c2d0..9bc8b77ac 100644 --- a/llama_stack/templates/ollama/run.yaml +++ b/llama_stack/templates/ollama/run.yaml @@ -64,7 +64,8 @@ providers: config: {} - provider_id: braintrust provider_type: inline::braintrust - config: {} + config: + openai_api_key: ${env.OPENAI_API_KEY} metadata_store: namespace: null type: sqlite diff --git a/llama_stack/templates/tgi/run-with-safety.yaml b/llama_stack/templates/tgi/run-with-safety.yaml index ecd03c36a..eb8c2cef6 100644 --- a/llama_stack/templates/tgi/run-with-safety.yaml +++ b/llama_stack/templates/tgi/run-with-safety.yaml @@ -68,7 +68,8 @@ providers: config: {} - provider_id: braintrust provider_type: inline::braintrust - config: {} + config: + openai_api_key: ${env.OPENAI_API_KEY} metadata_store: namespace: null type: sqlite diff --git a/llama_stack/templates/tgi/run.yaml b/llama_stack/templates/tgi/run.yaml index b93f09042..1e2f8fdbe 100644 --- a/llama_stack/templates/tgi/run.yaml +++ b/llama_stack/templates/tgi/run.yaml @@ -64,7 +64,8 @@ providers: config: {} - provider_id: braintrust provider_type: inline::braintrust - config: {} + config: + openai_api_key: ${env.OPENAI_API_KEY} metadata_store: namespace: null type: sqlite diff --git a/llama_stack/templates/together/run.yaml b/llama_stack/templates/together/run.yaml index 381557816..864e55ae6 100644 --- a/llama_stack/templates/together/run.yaml +++ b/llama_stack/templates/together/run.yaml @@ -65,7 +65,8 @@ providers: config: {} - provider_id: braintrust provider_type: inline::braintrust - config: {} + config: + openai_api_key: ${env.OPENAI_API_KEY} metadata_store: namespace: null type: sqlite diff --git a/llama_stack/templates/vllm-gpu/run.yaml b/llama_stack/templates/vllm-gpu/run.yaml index 1442273f4..e6a80a256 100644 --- a/llama_stack/templates/vllm-gpu/run.yaml +++ b/llama_stack/templates/vllm-gpu/run.yaml @@ -68,7 +68,8 @@ providers: config: {} - provider_id: braintrust provider_type: inline::braintrust - config: {} + config: + openai_api_key: ${env.OPENAI_API_KEY} metadata_store: namespace: null type: sqlite