SYMBOL INDEX (6974 symbols across 820 files)

FILE: benchmarking/k8s-benchmark/openai-mock-server.py
  function get_models (line 27) | def get_models():
  function generate_random_text (line 39) | def generate_random_text(length=50):
  function list_models (line 86) | def list_models():
  function chat_completions (line 93) | def chat_completions():
  function handle_non_streaming_completion (line 109) | def handle_non_streaming_completion(model, messages):
  function handle_streaming_completion (line 132) | def handle_streaming_completion(model, messages):
  function health (line 185) | def health():

FILE: benchmarking/k8s-benchmark/scripts/generate_charts.py
  function extract_setup_name (line 26) | def extract_setup_name(filename: str) -> str:
  function parse_txt_file (line 80) | def parse_txt_file(filepath: str) -> list[tuple[float, float, float, flo...
  function generate_charts (line 148) | def generate_charts(benchmark_dir: str = "results"):

FILE: benchmarking/vertical-scaling/locustfile.py
  class ChatCompletionUser (line 12) | class ChatCompletionUser(HttpUser):
    method on_start (line 20) | def on_start(self):
    method chat_completion (line 28) | def chat_completion(self):

FILE: benchmarking/vertical-scaling/mock-server.py
  class MockHandler (line 34) | class MockHandler(BaseHTTPRequestHandler):
    method do_POST (line 35) | def do_POST(self):  # noqa: N802
    method do_GET (line 47) | def do_GET(self):  # noqa: N802
    method log_message (line 57) | def log_message(self, format, *args):

FILE: client-sdks/openapi/build_hierarchy.py
  function build_hierarchy_from_tags (line 39) | def build_hierarchy_from_tags(tags: list[str], hierarchy: dict) -> None:
  function get_children_from_hierarchy (line 51) | def get_children_from_hierarchy(tag: str, hierarchy: dict) -> list[str]:
  function convert_oneof_const_to_enum (line 56) | def convert_oneof_const_to_enum(schema: dict) -> dict:
  function fix_oneof_const_schemas (line 83) | def fix_oneof_const_schemas(obj: dict[str, Any]) -> dict[str, Any]:
  function remove_defaults_from_required (line 98) | def remove_defaults_from_required(spec: dict[str, Any]) -> int:
  function mark_unwrappable_list_responses (line 130) | def mark_unwrappable_list_responses(spec: dict[str, Any]) -> int:
  function mark_streaming_operations (line 165) | def mark_streaming_operations(spec: dict[str, Any]) -> int:
  function process_openapi (line 202) | def process_openapi(input_file: str, output_file: str, hierarchy_file: s...
  function main (line 351) | def main():

FILE: client-sdks/openapi/merge_stainless_config.py
  function parse_endpoint (line 49) | def parse_endpoint(endpoint_str: str) -> tuple[str, str]:
  function extract_resources (line 67) | def extract_resources(stainless_config: dict[str, Any]) -> tuple[dict[st...
  function enrich_openapi_spec (line 175) | def enrich_openapi_spec(
  function get_nested_value (line 259) | def get_nested_value(obj: Any, path: str) -> tuple[Any, Any, str]:
  function apply_patches (line 289) | def apply_patches(openapi_spec: dict[str, Any], patch_config: dict[str, ...
  function main (line 376) | def main():

FILE: client-sdks/openapi/patch_hierarchy.py
  function to_snake_case (line 35) | def to_snake_case(name: str) -> str:
  function to_pascal_case (line 42) | def to_pascal_case(name: str) -> str:
  function extract_parent_child_pairs (line 48) | def extract_parent_child_pairs(hierarchy: dict, parent: str | None = Non...
  function patch_api_file (line 59) | def patch_api_file(api_file: Path, child_tag: str, package_name: str) ->...
  function patch_optional_import (line 120) | def patch_optional_import(api_file: Path) -> bool:
  function patch_llama_stack_client (line 154) | def patch_llama_stack_client(client_file: Path, pairs: list[tuple[str, s...
  function patch_proxy_methods (line 215) | def patch_proxy_methods(api_dir: Path, proxy_methods: list[dict]) -> int:
  function patch_apis (line 273) | def patch_apis(hierarchy_file: str, sdk_dir: str, package_name: str = "l...
  function main (line 323) | def main():

FILE: client-sdks/openapi/templates/python/lib/_utils.py
  function pascal_to_snake_case (line 10) | def pascal_to_snake_case(name: str) -> str:

FILE: client-sdks/openapi/templates/python/lib/agents/agent.py
  class ToolResponsePayload (line 31) | class ToolResponsePayload(TypedDict, total=False):
  class ToolUtils (line 41) | class ToolUtils:
    method coerce_tool_content (line 43) | def coerce_tool_content(content: Any) -> str:
    method parse_tool_arguments (line 56) | def parse_tool_arguments(arguments: Any) -> dict[str, Any]:
    method normalize_tool_response (line 75) | def normalize_tool_response(tool_response: Any) -> ToolResponsePayload:
  class Agent (line 101) | class Agent:
    method __init__ (line 102) | def __init__(
    method create_session (line 130) | def create_session(self, session_name: str) -> str:
    method _run_tool_calls (line 138) | def _run_tool_calls(self, tool_calls: list[ToolCall]) -> list[ToolResp...
    method _run_single_tool (line 145) | def _run_single_tool(self, tool_call: ToolCall) -> Any:
    method create_turn (line 169) | def create_turn(
    method _create_turn_streaming (line 191) | def _create_turn_streaming(
  class AsyncAgent (line 286) | class AsyncAgent:
    method __init__ (line 287) | def __init__(
    method create_session (line 316) | async def create_session(self, session_name: str) -> str:
    method create_turn (line 324) | async def create_turn(
    method _run_tool_calls (line 340) | async def _run_tool_calls(self, tool_calls: list[ToolCall]) -> list[To...
    method _run_single_tool (line 347) | async def _run_single_tool(self, tool_call: ToolCall) -> Any:
    method _create_turn_streaming (line 371) | async def _create_turn_streaming(
  class AgentUtils (line 466) | class AgentUtils:
    method get_client_tools (line 468) | def get_client_tools(
    method get_tool_calls (line 479) | def get_tool_calls(chunk: AgentStreamChunk, tool_parser: ToolParser | ...
    method get_turn_id (line 505) | def get_turn_id(chunk: AgentStreamChunk) -> str | None:
    method normalize_tools (line 509) | def normalize_tools(

FILE: client-sdks/openapi/templates/python/lib/agents/client_tool.py
  class JSONSchema (line 24) | class JSONSchema(TypedDict, total=False):
  class ClientTool (line 30) | class ClientTool:
    method get_name (line 45) | def get_name(self) -> str:
    method get_description (line 49) | def get_description(self) -> str:
    method get_input_schema (line 53) | def get_input_schema(self) -> JSONSchema:
    method get_instruction_string (line 56) | def get_instruction_string(self) -> str:
    method get_tool_definition (line 59) | def get_tool_definition(self) -> ToolDefinition:
    method run (line 67) | def run(
    method async_run (line 95) | async def async_run(
    method run_impl (line 123) | def run_impl(self, **kwargs) -> Any:
    method async_run_impl (line 132) | def async_run_impl(self, **kwargs):
  function _python_type_to_json_schema_type (line 136) | def _python_type_to_json_schema_type(type_hint: Any) -> str:
  function client_tool (line 167) | def client_tool(func: Callable) -> ClientTool:

FILE: client-sdks/openapi/templates/python/lib/agents/event_logger.py
  class AgentEventLogger (line 31) | class AgentEventLogger:
    method log (line 48) | def log(self, event_generator: Iterator[AgentStreamChunk]) -> Iterator...

FILE: client-sdks/openapi/templates/python/lib/agents/event_synthesizer.py
  class _ToolCallState (line 43) | class _ToolCallState:
    method update (line 50) | def update(self, *, delta: str | None = None, final: str | None = None...
    method as_tool_call (line 56) | def as_tool_call(self) -> ToolCall:
  class TurnEventSynthesizer (line 61) | class TurnEventSynthesizer:
    method __init__ (line 64) | def __init__(self, session_id: str, turn_id: str):
    method _next_step_id (line 83) | def _next_step_id(self) -> str:
    method _maybe_emit_turn_started (line 88) | def _maybe_emit_turn_started(self) -> Iterator[AgentEvent]:
    method _start_inference_step (line 93) | def _start_inference_step(
    method _complete_inference_step (line 106) | def _complete_inference_step(self, *, stop_reason: str, response_id: s...
    method _start_tool_execution_step (line 142) | def _start_tool_execution_step(self, call_state: _ToolCallState) -> It...
    method _complete_tool_execution_step (line 163) | def _complete_tool_execution_step(self, call_state: _ToolCallState) ->...
    method _coerce_arguments (line 182) | def _coerce_arguments(payload: Any) -> str | None:
    method _register_tool_call (line 192) | def _register_tool_call(
    method _classify_tool_type (line 212) | def _classify_tool_type(self, tool_name: str) -> str:
    method process_raw_stream (line 236) | def process_raw_stream(self, events: Iterable[Any]) -> Iterator[AgentE...
    method _handle_output_item_added (line 311) | def _handle_output_item_added(self, item: Any) -> Iterator[AgentEvent]:
    method _handle_output_item_delta (line 362) | def _handle_output_item_delta(self, delta: Any) -> Iterator[AgentEvent]:
    method _handle_output_item_done (line 401) | def _handle_output_item_done(self, item: Any) -> Iterator[AgentEvent]:
    method finish_turn (line 438) | def finish_turn(self) -> Iterator[AgentEvent]:

FILE: client-sdks/openapi/templates/python/lib/agents/react/agent.py
  function _tool_definition_from_mapping (line 20) | def _tool_definition_from_mapping(tool: Mapping[str, Any]) -> dict[str, ...
  function _collect_tool_definitions (line 31) | def _collect_tool_definitions(
  function get_default_react_instructions (line 47) | def get_default_react_instructions(tool_defs: list[dict[str, Any]]) -> str:
  class ReActAgent (line 61) | class ReActAgent(Agent):
    method __init__ (line 62) | def __init__(

FILE: client-sdks/openapi/templates/python/lib/agents/react/tool_parser.py
  class Param (line 16) | class Param(BaseModel):
  class Action (line 21) | class Action(BaseModel):
  class ReActOutput (line 26) | class ReActOutput(BaseModel):
  class ReActToolParser (line 32) | class ReActToolParser(ToolParser):
    method get_tool_calls (line 33) | def get_tool_calls(self, output_message: CompletionMessage) -> list[To...

FILE: client-sdks/openapi/templates/python/lib/agents/tool_parser.py
  class ToolParser (line 12) | class ToolParser:
    method get_tool_calls (line 41) | def get_tool_calls(self, output_message: CompletionMessage) -> list[To...

FILE: client-sdks/openapi/templates/python/lib/agents/turn_events.py
  class TurnStarted (line 49) | class TurnStarted:
  class TurnCompleted (line 62) | class TurnCompleted:
  class TurnFailed (line 78) | class TurnFailed:
  class StepStarted (line 94) | class StepStarted:
  class TextDelta (line 113) | class TextDelta:
  class ToolCallIssuedDelta (line 124) | class ToolCallIssuedDelta:
  class ToolCallDelta (line 141) | class ToolCallDelta:
  class ToolCallCompletedDelta (line 154) | class ToolCallCompletedDelta:
  class StepProgress (line 176) | class StepProgress:
  class InferenceStepResult (line 194) | class InferenceStepResult:
  class ToolExecutionStepResult (line 212) | class ToolExecutionStepResult:
  class StepCompleted (line 229) | class StepCompleted:
  class AgentStreamChunk (line 251) | class AgentStreamChunk:

FILE: client-sdks/openapi/templates/python/lib/agents/types.py
  class ToolCall (line 16) | class ToolCall:
  class ToolResponse (line 25) | class ToolResponse:
  class CompletionMessage (line 35) | class CompletionMessage:
  class ToolDefinition (line 47) | class ToolDefinition(TypedDict, total=False):
  class FunctionTool (line 56) | class FunctionTool(Protocol):
    method get_name (line 59) | def get_name(self) -> str: ...
    method get_description (line 61) | def get_description(self) -> str: ...
    method get_input_schema (line 63) | def get_input_schema(self) -> dict[str, Any]: ...

FILE: client-sdks/openapi/templates/python/lib/cli/common/utils.py
  function create_bar_chart (line 13) | def create_bar_chart(data, labels, title=""):
  function handle_client_errors (line 36) | def handle_client_errors(operation_name):

FILE: client-sdks/openapi/templates/python/lib/cli/configure.py
  function get_config (line 18) | def get_config():
  function configure (line 30) | def configure(endpoint: str | None, api_key: str | None):

FILE: client-sdks/openapi/templates/python/lib/cli/constants.py
  function get_config_file_path (line 13) | def get_config_file_path():

FILE: client-sdks/openapi/templates/python/lib/cli/datasets/datasets.py
  function datasets (line 15) | def datasets():

FILE: client-sdks/openapi/templates/python/lib/cli/datasets/list.py
  function list_datasets (line 18) | def list_datasets(ctx):

FILE: client-sdks/openapi/templates/python/lib/cli/datasets/register.py
  function data_url_from_file (line 18) | def data_url_from_file(file_path: str) -> str:
  function register (line 48) | def register(

FILE: client-sdks/openapi/templates/python/lib/cli/datasets/unregister.py
  function unregister (line 16) | def unregister(ctx, dataset_id: str):

FILE: client-sdks/openapi/templates/python/lib/cli/eval/eval.py
  function eval (line 16) | def eval():

FILE: client-sdks/openapi/templates/python/lib/cli/eval/run_benchmark.py
  function run_benchmark (line 81) | def run_benchmark(

FILE: client-sdks/openapi/templates/python/lib/cli/eval/run_scoring.py
  function run_scoring (line 55) | def run_scoring(

FILE: client-sdks/openapi/templates/python/lib/cli/eval/utils.py
  function aggregate_categorical_count (line 11) | def aggregate_categorical_count(
  function aggregate_average (line 19) | def aggregate_average(
  function aggregate_weighted_average (line 28) | def aggregate_weighted_average(
  function aggregate_median (line 41) | def aggregate_median(
  function aggregate_accuracy (line 49) | def aggregate_accuracy(

FILE: client-sdks/openapi/templates/python/lib/cli/eval_tasks/eval_tasks.py
  function eval_tasks (line 19) | def eval_tasks():
  function register (line 33) | def register(

FILE: client-sdks/openapi/templates/python/lib/cli/eval_tasks/list.py
  function list_eval_tasks (line 18) | def list_eval_tasks(ctx):

FILE: client-sdks/openapi/templates/python/lib/cli/inference/inference.py
  function inference (line 18) | def inference():
  function chat_completion (line 30) | def chat_completion(ctx, message: str, stream: bool, session: bool, mode...
  function chat_session (line 62) | def chat_session(client, model_id: str | None, messages: list[dict[str, ...

FILE: client-sdks/openapi/templates/python/lib/cli/inspect/inspect.py
  function inspect (line 14) | def inspect():

FILE: client-sdks/openapi/templates/python/lib/cli/inspect/version.py
  function inspect_version (line 17) | def inspect_version(ctx):

FILE: client-sdks/openapi/templates/python/lib/cli/llama_stack_client.py
  function _get_version (line 16) | def _get_version():
  function llama_stack_client (line 48) | def llama_stack_client(ctx, endpoint: str, api_key: str, config: str | N...
  function main (line 108) | def main():

FILE: client-sdks/openapi/templates/python/lib/cli/models/models.py
  function models (line 18) | def models():
  function list_models (line 26) | def list_models(ctx):
  function get_model (line 65) | def get_model(ctx, model_id: str):
  class JSONParamType (line 89) | class JSONParamType(click.ParamType):
    method convert (line 92) | def convert(self, value, param, ctx):
  function register_model (line 108) | def register_model(
  function unregister_model (line 136) | def unregister_model(ctx, model_id: str):

FILE: client-sdks/openapi/templates/python/lib/cli/post_training/post_training.py
  function post_training (line 18) | def post_training():
  function supervised_fine_tune (line 31) | def supervised_fine_tune(
  function get_training_jobs (line 60) | def get_training_jobs(ctx):
  function get_training_job_status (line 74) | def get_training_job_status(ctx, job_uuid: str):
  function get_training_job_artifacts (line 88) | def get_training_job_artifacts(ctx, job_uuid: str):
  function cancel_training_job (line 102) | def cancel_training_job(ctx, job_uuid: str):

FILE: client-sdks/openapi/templates/python/lib/cli/providers/inspect.py
  function inspect_provider (line 18) | def inspect_provider(ctx, provider_id):

FILE: client-sdks/openapi/templates/python/lib/cli/providers/list.py
  function list_providers (line 18) | def list_providers(ctx):

FILE: client-sdks/openapi/templates/python/lib/cli/providers/providers.py
  function providers (line 15) | def providers():

FILE: client-sdks/openapi/templates/python/lib/cli/scoring_functions/list.py
  function list_scoring_functions (line 18) | def list_scoring_functions(ctx):

FILE: client-sdks/openapi/templates/python/lib/cli/scoring_functions/scoring_functions.py
  function scoring_functions (line 17) | def scoring_functions():
  function register (line 30) | def register(

FILE: client-sdks/openapi/templates/python/lib/cli/shields/shields.py
  function shields (line 18) | def shields():
  function list (line 26) | def list(ctx):
  function register (line 69) | def register(

FILE: client-sdks/openapi/templates/python/lib/cli/toolgroups/toolgroups.py
  function toolgroups (line 19) | def toolgroups():
  function list_toolgroups (line 27) | def list_toolgroups(ctx):
  function get_toolgroup (line 49) | def get_toolgroup(ctx, toolgroup_id: str):
  function register_toolgroup (line 86) | def register_toolgroup(
  function unregister_toolgroup (line 116) | def unregister_toolgroup(ctx, toolgroup_id: str):

FILE: client-sdks/openapi/templates/python/lib/cli/vector_stores/vector_stores.py
  function vector_stores (line 18) | def vector_stores():
  function list (line 26) | def list(ctx):
  function register (line 87) | def register(
  function unregister (line 118) | def unregister(ctx, vector_db_id: str):

FILE: client-sdks/openapi/templates/python/lib/inference/event_logger.py
  class InferenceStreamPrintableEvent (line 13) | class InferenceStreamPrintableEvent:
    method __init__ (line 14) | def __init__(
    method print (line 24) | def print(self, flush=True):
  class InferenceStreamLogEventPrinter (line 28) | class InferenceStreamLogEventPrinter:
    method __init__ (line 29) | def __init__(self):
    method yield_printable_events (line 32) | def yield_printable_events(
    method _handle_chat_completion_chunk (line 38) | def _handle_chat_completion_chunk(
  class EventLogger (line 54) | class EventLogger:
    method log (line 55) | def log(self, event_generator):

FILE: client-sdks/openapi/templates/python/lib/inference/utils.py
  class MessageAttachment (line 11) | class MessageAttachment:
    method base64 (line 14) | def base64(cls, file_path: str) -> str:
    method data_url (line 20) | def data_url(cls, media_type: str, file_path: str) -> str:

FILE: client-sdks/openapi/templates/python/lib/stream_printer.py
  class EventStreamPrinter (line 11) | class EventStreamPrinter:
    method gen (line 13) | def gen(cls, event_generator):

FILE: client-sdks/openapi/templates/python/lib/tools/mcp_oauth.py
  class McpOAuthHelper (line 26) | class McpOAuthHelper:
    method __init__ (line 29) | def __init__(self, server_url):
    method discover_auth_endpoints (line 64) | def discover_auth_endpoints(self):
    method register_client (line 78) | def register_client(self, registration_endpoint):
    method generate_pkce_values (line 105) | def generate_pkce_values(self):
    method stop_server (line 119) | def stop_server(self):
    method start_callback_server (line 124) | def start_callback_server(self):
    method exchange_code_for_token (line 139) | def exchange_code_for_token(self, auth_code, token_endpoint):
    method initiate_auth_flow (line 163) | def initiate_auth_flow(self):
  function get_base_url (line 210) | def get_base_url(url):
  function find_available_port (line 215) | def find_available_port(start_port, end_port):
  class CallbackServer (line 227) | class CallbackServer(HTTPServer):
    class OAuthCallbackHandler (line 228) | class OAuthCallbackHandler(BaseHTTPRequestHandler):
      method do_GET (line 229) | def do_GET(self):  # noqa: N802
      method log_message (line 262) | def log_message(self):
    method __init__ (line 266) | def __init__(self, server_address, auth_code_callback):
  function get_oauth_token_for_mcp_server (line 271) | def get_oauth_token_for_mcp_server(url: str) -> str | None:
  function run_main (line 276) | async def run_main(url: str):
  function main (line 298) | def main(url: str):

FILE: conftest.py
  function pytest_configure (line 18) | def pytest_configure(config):

FILE: docs/blog/self_improving_agent.py
  function fn_to_tool_schema (line 43) | def fn_to_tool_schema(fn) -> dict:
  class ScoreLedger (line 94) | class ScoreLedger:
    method __init__ (line 95) | def __init__(self, db_path: str = "prompt_optimizer.db"):
    method log (line 107) | def log(self, prompt_id: str, version: int, score: float, reasoning: s...
    method history (line 114) | def history(self, prompt_id: str) -> list[dict]:
  class ResearchAgent (line 134) | class ResearchAgent:
    method __init__ (line 135) | def __init__(
    method from_files (line 166) | def from_files(
    method _read_local_file (line 214) | def _read_local_file(
    method _index_document (line 221) | def _index_document(
    method _list_local_files (line 240) | def _list_local_files(
    method _tool_schemas (line 253) | def _tool_schemas(self) -> list[dict]:
    method _run_query (line 260) | def _run_query(self, question: str, system_prompt: str) -> str:
    method research (line 290) | def research(self, question: str) -> str:
    method evaluate_self (line 309) | def evaluate_self(self) -> dict:
    method improve_self (line 344) | def improve_self(self) -> dict:
    method optimize (line 368) | def optimize(self, max_iterations: int = 5):
    method best_prompt (line 377) | def best_prompt(self) -> dict:

FILE: docs/scripts/sync-files.js
  function getRequestedFiles (line 10) | function getRequestedFiles() {
  function trackFileUsage (line 26) | function trackFileUsage(filePath) {
  function filterContent (line 51) | function filterContent(content, filePath) {
  function syncFile (line 102) | function syncFile(filePath) {

FILE: docs/src/components/CodeFromFile.jsx
  function CodeFromFile (line 4) | function CodeFromFile({
  function getLanguageFromExtension (line 69) | function getLanguageFromExtension(filename) {

FILE: docs/src/components/HomepageFeatures/index.js
  function Feature (line 38) | function Feature({Svg, title, description}) {
  function HomepageFeatures (line 52) | function HomepageFeatures() {

FILE: docs/src/pages/index.js
  function HomepageHeader (line 8) | function HomepageHeader() {
  function QuickStart (line 41) | function QuickStart() {
  function Ecosystem (line 116) | function Ecosystem() {
  function CommunityLinks (line 170) | function CommunityLinks() {
  function Home (line 209) | function Home() {

FILE: scripts/cleanup_recordings.py
  function _unique_configs (line 47) | def _unique_configs(entries):
  function iter_all_ci_configs (line 59) | def iter_all_ci_configs() -> list[dict[str, str]]:
  function collect_ci_tests (line 67) | def collect_ci_tests():
  function get_base_test_id (line 139) | def get_base_test_id(test_id: str) -> str:
  function find_all_recordings (line 149) | def find_all_recordings():
  function analyze_recordings (line 154) | def analyze_recordings(ci_test_ids, dry_run=True):
  function main (line 233) | def main():

FILE: scripts/diagnose_recordings.py
  function find_hash (line 43) | def find_hash(hash_value: str, base_dir: Path | None = None, test_id: st...
  function show_recording (line 143) | def show_recording(file_path: Path):
  function list_test_recordings (line 196) | def list_test_recordings(test_id: str, base_dir: Path | None = None):
  function explain_paths (line 244) | def explain_paths(test_id: str | None = None, base_dir: Path | None = No...
  function compute_hash (line 281) | def compute_hash(endpoint: str, method: str, body_json: str, test_id: st...
  function main (line 315) | def main():

FILE: scripts/distro_codegen.py
  class ChangedPathTracker (line 21) | class ChangedPathTracker:
    method __init__ (line 24) | def __init__(self):
    method add_paths (line 27) | def add_paths(self, *paths):
    method changed_paths (line 33) | def changed_paths(self):
  function find_distro_dirs (line 37) | def find_distro_dirs(distro_dir: Path) -> Iterable[Path]:
  function process_distro (line 45) | def process_distro(distro_dir: Path, progress, change_tracker: ChangedPa...
  function check_for_changes (line 73) | def check_for_changes(change_tracker: ChangedPathTracker) -> bool:
  function pre_import_distros (line 88) | def pre_import_distros(distro_dirs: list[Path]) -> None:
  function main (line 95) | def main():

FILE: scripts/gen-ci-docs.py
  function parse_workflow_file (line 15) | def parse_workflow_file(file_path):
  function generate_ci_docs (line 30) | def generate_ci_docs():

FILE: scripts/generate_ci_matrix.py
  function generate_matrix (line 27) | def generate_matrix(schedule="", test_setup="", matrix_key="default"):

FILE: scripts/generate_openai_coverage_docs.py
  function _extract_property_name (line 31) | def _extract_property_name(property_path: str) -> str | None:
  function _load_test_coverage (line 46) | def _load_test_coverage() -> tuple[set[str], float | None, dict[str, tup...
  function generate_docs (line 71) | def generate_docs(
  function main (line 293) | def main():

FILE: scripts/generate_prompt_format.py
  function run_main (line 26) | def run_main(
  function main (line 63) | def main():

FILE: scripts/get_setup_env.py
  function get_setup_config (line 19) | def get_setup_config(setup_name, suite_name=None):
  function main (line 50) | def main():

FILE: scripts/normalize_recordings.py
  function normalize_response_data (line 24) | def normalize_response_data(data: dict, request_hash: str) -> dict:
  function normalize_recording_file (line 53) | def normalize_recording_file(file_path: Path, dry_run: bool = False) -> ...
  function main (line 92) | def main():

FILE: scripts/openai_coverage.py
  function _load_spec (line 35) | def _load_spec(spec_path: Path) -> dict[str, Any]:
  function _count_schema_properties (line 43) | def _count_schema_properties(schema: dict[str, Any], spec: dict[str, Any...
  function _count_endpoint_properties (line 93) | def _count_endpoint_properties(spec: dict[str, Any], paths: list[str]) -...
  function _load_endpoint_categories (line 136) | def _load_endpoint_categories(openai_spec: Path) -> dict[str, list[str]]:
  function _categorize_path (line 160) | def _categorize_path(path: str, categories: dict[str, list[str]]) -> str:
  function _check_oasdiff_installed (line 170) | def _check_oasdiff_installed() -> bool:
  function _run_oasdiff (line 175) | def _run_oasdiff(openai_spec: Path, llama_spec: Path) -> dict[str, Any]:
  function _extract_issues (line 206) | def _extract_issues(obj: Any, path: str = "") -> dict[str, Any]:
  function _merge_diffs (line 312) | def _merge_diffs(base_diff: dict[str, Any], overlay_diff: dict[str, Any]...
  function calculate_coverage (line 345) | def calculate_coverage(
  function print_summary (line 515) | def print_summary(report: dict[str, Any]) -> None:
  function main (line 562) | def main():

FILE: scripts/openapi_generator/__init__.py
  function __getattr__ (line 17) | def __getattr__(name: str):

FILE: scripts/openapi_generator/app.py
  function _get_protocol_method (line 23) | def _get_protocol_method(api: Api, method_name: str) -> Any | None:
  function create_llama_stack_app (line 65) | def create_llama_stack_app() -> FastAPI:

FILE: scripts/openapi_generator/endpoints.py
  function _to_pascal_case (line 33) | def _to_pascal_case(segment: str) -> str:
  function _compose_request_model_name (line 38) | def _compose_request_model_name(api: Api, method_name: str, variant: str...
  function _extract_path_parameters (line 53) | def _extract_path_parameters(path: str) -> list[dict[str, Any]]:
  function _create_endpoint_with_request_model (line 68) | def _create_endpoint_with_request_model(
  function _build_field_definitions (line 81) | def _build_field_definitions(query_parameters: list[QueryParameter], use...
  function _create_dynamic_request_model (line 127) | def _create_dynamic_request_model(
  function _build_signature_params (line 148) | def _build_signature_params(
  function _extract_operation_description_from_docstring (line 167) | def _extract_operation_description_from_docstring(api: Api, method_name:...
  function _extract_response_description_from_docstring (line 186) | def _extract_response_description_from_docstring(webmethod, response_mod...
  function _get_tag_from_api (line 198) | def _get_tag_from_api(api: Api) -> str:
  function _is_file_or_form_param (line 203) | def _is_file_or_form_param(param_type: Any) -> bool:
  function _is_extra_body_field (line 218) | def _is_extra_body_field(metadata_item: Any) -> bool:
  function _should_embed_parameter (line 225) | def _should_embed_parameter(param_type: Any) -> bool:
  function _is_async_iterator_type (line 238) | def _is_async_iterator_type(type_obj: Any) -> bool:
  function _extract_response_models_from_union (line 251) | def _extract_response_models_from_union(union_type: Any) -> tuple[type |...
  function _find_models_for_endpoint (line 286) | def _find_models_for_endpoint(
  function _create_fastapi_endpoint (line 454) | def _create_fastapi_endpoint(app: FastAPI, route, webmethod, api: Api):

FILE: scripts/openapi_generator/main.py
  function generate_openapi_spec (line 22) | def generate_openapi_spec(output_dir: str) -> dict[str, Any]:
  function main (line 213) | def main():

FILE: scripts/openapi_generator/schema_collection.py
  function _ensure_components_schemas (line 14) | def _ensure_components_schemas(openapi_schema: dict[str, Any]) -> None:
  function _extract_and_fix_defs (line 22) | def _extract_and_fix_defs(schema: dict[str, Any], openapi_schema: dict[s...
  function _ensure_json_schema_types_included (line 49) | def _ensure_json_schema_types_included(openapi_schema: dict[str, Any]) -...

FILE: scripts/openapi_generator/schema_filtering.py
  function _get_all_json_schema_type_names (line 21) | def _get_all_json_schema_type_names() -> set[str]:
  function _get_explicit_schema_names (line 31) | def _get_explicit_schema_names(openapi_schema: dict[str, Any]) -> set[str]:
  function _find_schema_refs_in_object (line 38) | def _find_schema_refs_in_object(obj: Any) -> set[str]:
  function _add_transitive_references (line 58) | def _add_transitive_references(
  function _find_schemas_referenced_by_paths (line 87) | def _find_schemas_referenced_by_paths(filtered_paths: dict[str, Any], op...
  function _filter_schemas_by_references (line 134) | def _filter_schemas_by_references(
  function _path_starts_with_version (line 159) | def _path_starts_with_version(path: str, version: str) -> bool:
  function _is_stable_path (line 164) | def _is_stable_path(path: str) -> bool:
  function _is_experimental_path (line 173) | def _is_experimental_path(path: str) -> bool:
  function _is_path_deprecated (line 180) | def _is_path_deprecated(path_item: dict[str, Any]) -> bool:
  function _filter_schema_by_version (line 190) | def _filter_schema_by_version(
  function _filter_deprecated_schema (line 240) | def _filter_deprecated_schema(openapi_schema: dict[str, Any]) -> dict[st...
  function _filter_combined_schema (line 261) | def _filter_combined_schema(openapi_schema: dict[str, Any]) -> dict[str,...

FILE: scripts/openapi_generator/schema_transforms.py
  function _fix_ref_references (line 33) | def _fix_ref_references(openapi_schema: dict[str, Any]) -> dict[str, Any]:
  function _normalize_empty_responses (line 54) | def _normalize_empty_responses(openapi_schema: dict[str, Any]) -> dict[s...
  function _eliminate_defs_section (line 81) | def _eliminate_defs_section(openapi_schema: dict[str, Any]) -> dict[str,...
  function _add_error_responses (line 142) | def _add_error_responses(openapi_schema: dict[str, Any]) -> dict[str, Any]:
  function _fix_path_parameters (line 207) | def _fix_path_parameters(openapi_schema: dict[str, Any]) -> dict[str, Any]:
  function _get_schema_title (line 237) | def _get_schema_title(item: dict[str, Any]) -> str | None:
  function _add_titles_to_unions (line 285) | def _add_titles_to_unions(obj: Any, parent_key: str | None = None) -> None:
  function _convert_anyof_const_to_enum (line 361) | def _convert_anyof_const_to_enum(obj: Any) -> None:
  function _fix_schema_recursive (line 410) | def _fix_schema_recursive(obj: Any) -> None:
  function _clean_description (line 425) | def _clean_description(description: str) -> str:
  function _clean_schema_descriptions (line 455) | def _clean_schema_descriptions(openapi_schema: dict[str, Any]) -> dict[s...
  function _add_extra_body_params_extension (line 468) | def _add_extra_body_params_extension(openapi_schema: dict[str, Any]) -> ...
  function _remove_query_params_from_body_endpoints (line 560) | def _remove_query_params_from_body_endpoints(openapi_schema: dict[str, A...
  function _remove_request_bodies_from_get_endpoints (line 601) | def _remove_request_bodies_from_get_endpoints(openapi_schema: dict[str, ...
  function _extract_duplicate_union_types (line 689) | def _extract_duplicate_union_types(openapi_schema: dict[str, Any]) -> di...
  function _dedupe_create_response_request_input_union_for_stainless (line 820) | def _dedupe_create_response_request_input_union_for_stainless(openapi_sc...
  function _convert_multiline_strings_to_literal (line 904) | def _convert_multiline_strings_to_literal(obj: Any) -> Any:
  function _write_yaml_file (line 921) | def _write_yaml_file(file_path: Path, schema: dict[str, Any]) -> None:
  function _apply_legacy_sorting (line 949) | def _apply_legacy_sorting(openapi_schema: dict[str, Any]) -> dict[str, A...
  function _remove_type_object_from_openai_schemas (line 1011) | def _remove_type_object_from_openai_schemas(openapi_schema: dict[str, An...
  function _fix_schema_issues (line 1026) | def _fix_schema_issues(openapi_schema: dict[str, Any]) -> dict[str, Any]:
  function validate_openapi_schema (line 1039) | def validate_openapi_schema(schema: dict[str, Any], schema_name: str = "...

FILE: scripts/openapi_generator/stainless_config/generate_config.py
  class Endpoint (line 593) | class Endpoint:
    method from_config (line 599) | def from_config(cls, value: Any) -> Endpoint:
    method _from_parts (line 612) | def _from_parts(cls, method: str, path: str) -> Endpoint:
    method to_config (line 621) | def to_config(self) -> Any:
    method route_key (line 628) | def route_key(self) -> str:
  class Resource (line 633) | class Resource:
    method from_dict (line 639) | def from_dict(cls, data: dict[str, Any]) -> Resource:
    method to_config (line 645) | def to_config(self) -> dict[str, Any]:
    method collect_endpoint_paths (line 655) | def collect_endpoint_paths(self) -> set[str]:
    method iter_endpoints (line 661) | def iter_endpoints(self, prefix: str) -> Iterator[tuple[str, str]]:
  function _load_openapi_paths (line 673) | def _load_openapi_paths(openapi_path: Path) -> set[str]:
  class StainlessConfig (line 687) | class StainlessConfig:
    method make (line 702) | def make(cls) -> StainlessConfig:
    method referenced_paths (line 718) | def referenced_paths(self) -> set[str]:
    method readme_endpoint_paths (line 725) | def readme_endpoint_paths(self) -> set[str]:
    method endpoint_map (line 738) | def endpoint_map(self) -> dict[str, list[str]]:
    method validate_unique_endpoints (line 745) | def validate_unique_endpoints(self) -> None:
    method validate_readme_endpoints (line 757) | def validate_readme_endpoints(self) -> None:
    method to_dict (line 766) | def to_dict(self) -> dict[str, Any]:
    method validate_against_openapi (line 775) | def validate_against_openapi(self, openapi_path: Path) -> None:
    method validate (line 785) | def validate(self, openapi_path: Path | None = None) -> None:
  function build_config (line 792) | def build_config() -> dict[str, Any]:
  function write_config (line 796) | def write_config(repo_root: Path, openapi_path: Path | None = None) -> P...
  function main (line 806) | def main() -> None:

FILE: scripts/openapi_generator/state.py
  function register_dynamic_model (line 26) | def register_dynamic_model(name: str, model: type) -> type:
  function reset_generator_state (line 37) | def reset_generator_state() -> None:

FILE: scripts/provider_codegen.py
  function get_api_docstring (line 23) | def get_api_docstring(api_name: str) -> str | None:
  class ChangedPathTracker (line 40) | class ChangedPathTracker:
    method __init__ (line 43) | def __init__(self):
    method add_paths (line 46) | def add_paths(self, *paths):
    method changed_paths (line 52) | def changed_paths(self):
  function is_pydantic_model (line 56) | def is_pydantic_model(annotation: Any) -> bool:
  function get_nested_type_fields (line 63) | def get_nested_type_fields(type_class: Any, visited: set[str] | None = N...
  function extract_type_annotation (line 141) | def extract_type_annotation(annotation: Any, expand_models: bool = False...
  function get_config_class_info (line 180) | def get_config_class_info(config_class_path: str) -> dict[str, Any]:
  function generate_provider_docs (line 315) | def generate_provider_docs(progress, provider_spec: Any, api_name: str) ...
  function generate_index_docs (line 602) | def generate_index_docs(api_name: str, api_docstring: str | None, provid...
  function process_provider_registry (line 643) | def process_provider_registry(progress, change_tracker: ChangedPathTrack...
  function check_for_changes (line 689) | def check_for_changes(change_tracker: ChangedPathTracker) -> bool:
  function main (line 714) | def main():

FILE: scripts/provider_compat_matrix.py
  function _file_to_category (line 50) | def _file_to_category(filename: str) -> str:
  function _discover_categories (line 60) | def _discover_categories() -> list[str]:
  function _test_name_to_feature (line 67) | def _test_name_to_feature(test_name: str) -> str:
  class ProviderResults (line 83) | class ProviderResults:
  function _extract_provider_from_test_id (line 106) | def _extract_provider_from_test_id(test_id: str) -> str | None:
  function _extract_test_name (line 114) | def _extract_test_name(test_id: str) -> str | None:
  function _extract_test_file (line 123) | def _extract_test_file(test_id: str) -> str | None:
  function scan_recordings (line 130) | def scan_recordings(recordings_dir: Path) -> dict[str, ProviderResults]:
  function _collect_all_categories (line 209) | def _collect_all_categories(provider_map: dict[str, ProviderResults]) ->...
  function _compute_summary (line 226) | def _compute_summary(provider_map: dict[str, ProviderResults]) -> dict:
  function _load_provider_names (line 257) | def _load_provider_names() -> dict[str, str]:
  function _pname (line 278) | def _pname(provider: str) -> str:
  function _icon (line 284) | def _icon(outcome: str | None) -> str:
  function generate_matrix_markdown (line 290) | def generate_matrix_markdown(provider_map: dict[str, ProviderResults]) -...
  function main (line 375) | def main():

FILE: scripts/responses_test_coverage.py
  function _load_spec (line 43) | def _load_spec(path: Path) -> dict[str, Any]:
  function _resolve_ref (line 50) | def _resolve_ref(ref: str, spec: dict[str, Any]) -> dict[str, Any]:
  function _collect_properties (line 58) | def _collect_properties(
  function _get_type_value (line 83) | def _get_type_value(schema: dict[str, Any], spec: dict[str, Any]) -> str...
  function _extract_oneof_types (line 99) | def _extract_oneof_types(ref: str, spec: dict[str, Any]) -> list[str]:
  function _extract_routes_from_file (line 118) | def _extract_routes_from_file(filepath: Path) -> list[tuple[str, str]]:
  function _parse_route (line 134) | def _parse_route(method: str, path: str) -> tuple[str, str, str]:
  class Feature (line 175) | class Feature:
  function build_feature_matrix (line 192) | def build_feature_matrix(
  function _get_call_chain (line 293) | def _get_call_chain(node: ast.Call) -> str | None:
  function _is_openai_call (line 308) | def _is_openai_call(chain: str) -> bool:
  function _strip_client_prefix (line 312) | def _strip_client_prefix(chain: str) -> str:
  class TestEvidence (line 318) | class TestEvidence:
  function _analyze_test_ast (line 329) | def _analyze_test_ast(func_node: ast.AST) -> TestEvidence:
  function _extract_openai_test_functions (line 371) | def _extract_openai_test_functions(filepath: Path) -> list[tuple[str, as...
  function _scan_streaming_helpers (line 390) | def _scan_streaming_helpers(test_dir: Path) -> set[str]:
  function _match_evidence (line 406) | def _match_evidence(features: list[Feature], evidence_map: dict[str, Tes...
  function run_coverage (line 457) | def run_coverage(test_dir: Path = TESTS_DIR, spec_path: Path = OPENAI_SP...
  function get_tested_property_names (line 472) | def get_tested_property_names(features: list[Feature] | None = None) -> ...
  function print_report (line 511) | def print_report(features: list[Feature], verbose: bool = False) -> float:
  function main (line 573) | def main() -> None:

FILE: src/llama_stack/cli/llama.py
  class LlamaCLIParser (line 18) | class LlamaCLIParser:
    method __init__ (line 21) | def __init__(self):
    method parse_args (line 39) | def parse_args(self) -> argparse.Namespace:
    method run (line 45) | def run(self, args: argparse.Namespace) -> None:
  function main (line 49) | def main():

FILE: src/llama_stack/cli/scripts/run.py
  function install_wheel_from_presigned (line 12) | def install_wheel_from_presigned():

FILE: src/llama_stack/cli/stack/_list_deps.py
  function format_output_deps_only (line 37) | def format_output_deps_only(
  function run_stack_list_deps_command (line 67) | def run_stack_list_deps_command(args: argparse.Namespace) -> None:
  function quote_if_needed (line 130) | def quote_if_needed(dep):
  function quote_special_dep (line 137) | def quote_special_dep(dep_string):

FILE: src/llama_stack/cli/stack/list_apis.py
  class StackListApis (line 12) | class StackListApis(Subcommand):
    method __init__ (line 13) | def __init__(self, subparsers: argparse._SubParsersAction):
    method _add_arguments (line 24) | def _add_arguments(self):
    method _run_apis_list_cmd (line 27) | def _run_apis_list_cmd(self, args: argparse.Namespace) -> None:

FILE: src/llama_stack/cli/stack/list_deps.py
  class StackListDeps (line 11) | class StackListDeps(Subcommand):
    method __init__ (line 12) | def __init__(self, subparsers: argparse._SubParsersAction):
    method _add_arguments (line 23) | def _add_arguments(self):
    method _run_stack_list_deps_command (line 46) | def _run_stack_list_deps_command(self, args: argparse.Namespace) -> None:

FILE: src/llama_stack/cli/stack/list_providers.py
  class StackListProviders (line 12) | class StackListProviders(Subcommand):
    method __init__ (line 13) | def __init__(self, subparsers: argparse._SubParsersAction):
    method providable_apis (line 25) | def providable_apis(self):
    method _add_arguments (line 30) | def _add_arguments(self):
    method _run_providers_list_cmd (line 39) | def _run_providers_list_cmd(self, args: argparse.Namespace) -> None:

FILE: src/llama_stack/cli/stack/list_stacks.py
  class StackListBuilds (line 15) | class StackListBuilds(Subcommand):
    method __init__ (line 18) | def __init__(self, subparsers: argparse._SubParsersAction):
    method _get_distribution_dirs (line 29) | def _get_distribution_dirs(self) -> dict[str, tuple[Path, str]]:
    method _list_stack_command (line 55) | def _list_stack_command(self, args: argparse.Namespace) -> None:

FILE: src/llama_stack/cli/stack/remove.py
  class StackRemove (line 18) | class StackRemove(Subcommand):
    method __init__ (line 21) | def __init__(self, subparsers: argparse._SubParsersAction):
    method _add_arguments (line 32) | def _add_arguments(self) -> None:
    method _get_distribution_dirs (line 46) | def _get_distribution_dirs(self) -> dict[str, Path]:
    method _list_stacks (line 57) | def _list_stacks(self) -> None:
    method _remove_stack_build_command (line 68) | def _remove_stack_build_command(self, args: argparse.Namespace) -> None:

FILE: src/llama_stack/cli/stack/run.py
  class StackRun (line 30) | class StackRun(Subcommand):
    method __init__ (line 31) | def __init__(self, subparsers: argparse._SubParsersAction):
    method _add_arguments (line 42) | def _add_arguments(self):
    method _run_stack_run_cmd (line 68) | def _run_stack_run_cmd(self, args: argparse.Namespace) -> None:
    method _uvicorn_run (line 123) | def _uvicorn_run(self, config_file: Path | None, args: argparse.Namesp...
    method _start_ui_development_server (line 183) | def _start_ui_development_server(self, stack_server_port: int):

FILE: src/llama_stack/cli/stack/stack.py
  class StackParser (line 21) | class StackParser(Subcommand):
    method __init__ (line 22) | def __init__(self, subparsers: argparse._SubParsersAction):

FILE: src/llama_stack/cli/stack/utils.py
  class ImageType (line 13) | class ImageType(Enum):
  function print_subcommand_description (line 18) | def print_subcommand_description(parser, subparsers):

FILE: src/llama_stack/cli/subcommand.py
  class Subcommand (line 8) | class Subcommand:
    method __init__ (line 11) | def __init__(self, *args, **kwargs):
    method create (line 15) | def create(cls, *args, **kwargs):
    method _add_arguments (line 18) | def _add_arguments(self):

FILE: src/llama_stack/cli/table.py
  function print_table (line 13) | def print_table(rows, headers=None, separate_rows: bool = False, sort_by...

FILE: src/llama_stack/cli/utils.py
  function add_config_distro_args (line 15) | def add_config_distro_args(parser: argparse.ArgumentParser):
  function get_config_from_args (line 26) | def get_config_from_args(args: argparse.Namespace) -> str | None:

FILE: src/llama_stack/core/access_control/access_control.py
  function matches_resource (line 27) | def matches_resource(resource_scope: str, actual_resource: str) -> bool:
  function matches_scope (line 40) | def matches_scope(
  function as_list (line 53) | def as_list(obj: Any) -> list[Any]:
  function matches_conditions (line 59) | def matches_conditions(
  function default_policy (line 71) | def default_policy() -> list[AccessRule]:
  function is_action_allowed (line 92) | def is_action_allowed(
  class AccessDeniedError (line 157) | class AccessDeniedError(RuntimeError):
    method __init__ (line 158) | def __init__(self, action: str | None = None, resource: ProtectedResou...
  function _build_access_denied_message (line 167) | def _build_access_denied_message(action: str | None, resource: Protected...

FILE: src/llama_stack/core/access_control/conditions.py
  class User (line 10) | class User(Protocol):
  class ProtectedResource (line 15) | class ProtectedResource(Protocol):
  class Condition (line 21) | class Condition(Protocol):
    method matches (line 22) | def matches(self, resource: ProtectedResource, user: User) -> bool: ...
  class UserInOwnersList (line 25) | class UserInOwnersList:
    method __init__ (line 26) | def __init__(self, name: str):
    method owners_values (line 29) | def owners_values(self, resource: ProtectedResource) -> list[str] | None:
    method matches (line 40) | def matches(self, resource: ProtectedResource, user: User) -> bool:
    method __repr__ (line 52) | def __repr__(self):
  class UserNotInOwnersList (line 56) | class UserNotInOwnersList(UserInOwnersList):
    method __init__ (line 57) | def __init__(self, name: str):
    method matches (line 60) | def matches(self, resource: ProtectedResource, user: User) -> bool:
    method __repr__ (line 63) | def __repr__(self):
  class UserWithValueInList (line 67) | class UserWithValueInList:
    method __init__ (line 68) | def __init__(self, name: str, value: str):
    method matches (line 72) | def matches(self, resource: ProtectedResource, user: User) -> bool:
    method __repr__ (line 78) | def __repr__(self):
  class UserWithValueNotInList (line 82) | class UserWithValueNotInList(UserWithValueInList):
    method __init__ (line 83) | def __init__(self, name: str, value: str):
    method matches (line 86) | def matches(self, resource: ProtectedResource, user: User) -> bool:
    method __repr__ (line 89) | def __repr__(self):
  class UserIsOwner (line 93) | class UserIsOwner:
    method matches (line 94) | def matches(self, resource: ProtectedResource, user: User) -> bool:
    method __repr__ (line 97) | def __repr__(self):
  class UserIsNotOwner (line 101) | class UserIsNotOwner:
    method matches (line 102) | def matches(self, resource: ProtectedResource, user: User) -> bool:
    method __repr__ (line 105) | def __repr__(self):
  class ResourceIsUnowned (line 109) | class ResourceIsUnowned:
    method matches (line 110) | def matches(self, resource: ProtectedResource, user: User) -> bool:
    method __repr__ (line 113) | def __repr__(self):
  function parse_condition (line 117) | def parse_condition(condition: str) -> Condition:
  function parse_conditions (line 138) | def parse_conditions(conditions: list[str]) -> list[Condition]:

FILE: src/llama_stack/core/access_control/datatypes.py
  class Action (line 15) | class Action(StrEnum):
  class Scope (line 22) | class Scope(BaseModel):
  class RouteScope (line 28) | class RouteScope(BaseModel):
  function _mutually_exclusive (line 43) | def _mutually_exclusive(obj, a: str, b: str):
  function _require_one_of (line 48) | def _require_one_of(obj, a: str, b: str):
  class AccessRule (line 53) | class AccessRule(BaseModel):
    method validate_rule_format (line 111) | def validate_rule_format(self) -> Self:
  class RouteAccessRule (line 126) | class RouteAccessRule(BaseModel):
    method validate_rule_format (line 176) | def validate_rule_format(self) -> Self:

FILE: src/llama_stack/core/admin.py
  class AdminImplConfig (line 41) | class AdminImplConfig(BaseModel):
  function get_provider_impl (line 45) | async def get_provider_impl(config, deps):
  class AdminImpl (line 51) | class AdminImpl(Admin):
    method __init__ (line 52) | def __init__(self, config: AdminImplConfig, deps):
    method initialize (line 56) | async def initialize(self) -> None:
    method shutdown (line 59) | async def shutdown(self) -> None:
    method list_providers (line 64) | async def list_providers(self) -> ListProvidersResponse:
    method inspect_provider (line 91) | async def inspect_provider(self, request: InspectProviderRequest) -> P...
    method get_providers_health (line 99) | async def get_providers_health(self) -> dict[str, dict[str, HealthResp...
    method list_routes (line 160) | async def list_routes(self, request: ListRoutesRequest) -> ListRoutesR...
    method health (line 263) | async def health(self) -> HealthInfo:
    method version (line 266) | async def version(self) -> VersionInfo:

FILE: src/llama_stack/core/build.py
  class ApiInput (line 33) | class ApiInput(BaseModel):
  function get_provider_dependencies (line 38) | def get_provider_dependencies(
  function print_pip_install_help (line 87) | def print_pip_install_help(config: StackConfig):

FILE: src/llama_stack/core/client.py
  function get_client_impl (line 23) | async def get_client_impl(protocol, config: RemoteProviderConfig, _deps:...
  function create_api_client_class (line 30) | def create_api_client_class(protocol) -> type:
  function extract_non_async_iterator_type (line 189) | def extract_non_async_iterator_type(type_hint):
  function extract_async_iterator_type (line 198) | def extract_async_iterator_type(type_hint):

FILE: src/llama_stack/core/configure.py
  function configure_single_provider (line 28) | def configure_single_provider(registry: dict[str, ProviderSpec], provide...
  function configure_api_providers (line 47) | def configure_api_providers(config: StackConfig, build_spec: Distributio...
  function upgrade_from_routing_table (line 120) | def upgrade_from_routing_table(
  function parse_and_maybe_upgrade_config (line 195) | def parse_and_maybe_upgrade_config(config_dict: dict[str, Any]) -> Stack...

FILE: src/llama_stack/core/connectors/connectors.py
  class ConnectorServiceConfig (line 32) | class ConnectorServiceConfig(BaseModel):
  function get_provider_impl (line 38) | async def get_provider_impl(config: ConnectorServiceConfig):
  class ConnectorServiceImpl (line 47) | class ConnectorServiceImpl(Connectors):
    method __init__ (line 50) | def __init__(self, config: ConnectorServiceConfig):
    method _get_key (line 54) | def _get_key(self, connector_id: str) -> str:
    method initialize (line 58) | async def initialize(self):
    method register_connector (line 67) | async def register_connector(
    method unregister_connector (line 104) | async def unregister_connector(self, connector_id: str):
    method get_connector (line 111) | async def get_connector(
    method list_connectors (line 129) | async def list_connectors(self) -> ListConnectorsResponse:
    method get_connector_tool (line 135) | async def get_connector_tool(self, request: GetConnectorToolRequest, a...
    method list_connector_tools (line 145) | async def list_connector_tools(
    method shutdown (line 155) | async def shutdown(self):

FILE: src/llama_stack/core/conversations/conversations.py
  class ConversationServiceConfig (line 45) | class ConversationServiceConfig(BaseModel):
  function get_provider_impl (line 56) | async def get_provider_impl(config: ConversationServiceConfig, deps: dic...
  class ConversationServiceImpl (line 63) | class ConversationServiceImpl(Conversations):
    method __init__ (line 66) | def __init__(self, config: ConversationServiceConfig, deps: dict[Any, ...
    method initialize (line 79) | async def initialize(self) -> None:
    method create_conversation (line 101) | async def create_conversation(self, request: CreateConversationRequest...
    method get_conversation (line 145) | async def get_conversation(self, request: GetConversationRequest) -> C...
    method update_conversation (line 157) | async def update_conversation(self, conversation_id: str, request: Upd...
    method openai_delete_conversation (line 172) | async def openai_delete_conversation(self, request: DeleteConversation...
    method _validate_conversation_id (line 185) | def _validate_conversation_id(self, conversation_id: str) -> None:
    method _get_or_generate_item_id (line 194) | def _get_or_generate_item_id(self, item: ConversationItem, item_dict: ...
    method _get_validated_conversation (line 206) | async def _get_validated_conversation(self, conversation_id: str) -> C...
    method add_items (line 210) | async def add_items(self, conversation_id: str, request: AddItemsReque...
    method retrieve (line 252) | async def retrieve(self, request: RetrieveItemRequest) -> Conversation...
    method list_items (line 269) | async def list_items(self, request: ListItemsRequest) -> ConversationI...
    method openai_delete_conversation_item (line 302) | async def openai_delete_conversation_item(self, request: DeleteItemReq...
    method shutdown (line 324) | async def shutdown(self) -> None:

FILE: src/llama_stack/core/datatypes.py
  class RegistryEntrySource (line 56) | class RegistryEntrySource(StrEnum):
  class User (line 61) | class User(BaseModel):
    method __init__ (line 66) | def __init__(self, principal: str, attributes: dict[str, list[str]] | ...
  class ResourceWithOwner (line 70) | class ResourceWithOwner(Resource):
  class ModelWithOwner (line 79) | class ModelWithOwner(Model, ResourceWithOwner):
  class ShieldWithOwner (line 83) | class ShieldWithOwner(Shield, ResourceWithOwner):
  class VectorStoreWithOwner (line 87) | class VectorStoreWithOwner(VectorStore, ResourceWithOwner):
  class DatasetWithOwner (line 91) | class DatasetWithOwner(Dataset, ResourceWithOwner):
  class ScoringFnWithOwner (line 95) | class ScoringFnWithOwner(ScoringFn, ResourceWithOwner):
  class BenchmarkWithOwner (line 99) | class BenchmarkWithOwner(Benchmark, ResourceWithOwner):
  class ToolGroupWithOwner (line 103) | class ToolGroupWithOwner(ToolGroup, ResourceWithOwner):
  class AutoRoutedProviderSpec (line 124) | class AutoRoutedProviderSpec(ProviderSpec):
  class RoutingTableProviderSpec (line 137) | class RoutingTableProviderSpec(ProviderSpec):
  class Provider (line 147) | class Provider(BaseModel):
  class BuildProvider (line 165) | class BuildProvider(BaseModel):
  class DistributionSpec (line 179) | class DistributionSpec(BaseModel):
  class OAuth2JWKSConfig (line 195) | class OAuth2JWKSConfig(BaseModel):
  class OAuth2IntrospectionConfig (line 202) | class OAuth2IntrospectionConfig(BaseModel):
  class AuthProviderType (line 209) | class AuthProviderType(StrEnum):
  class OAuth2TokenAuthConfig (line 218) | class OAuth2TokenAuthConfig(BaseModel):
    method validate_claims_mapping (line 244) | def validate_claims_mapping(cls, v):
    method validate_mode (line 251) | def validate_mode(self) -> Self:
  class CustomAuthConfig (line 259) | class CustomAuthConfig(BaseModel):
  class GitHubTokenAuthConfig (line 269) | class GitHubTokenAuthConfig(BaseModel):
  class KubernetesAuthProviderConfig (line 286) | class KubernetesAuthProviderConfig(BaseModel):
    method validate_api_server_url (line 306) | def validate_api_server_url(cls, v):
    method validate_claims_mapping (line 316) | def validate_claims_mapping(cls, v):
  class AuthenticationConfig (line 329) | class AuthenticationConfig(BaseModel):
  class AuthenticationRequiredError (line 346) | class AuthenticationRequiredError(Exception):
  class QualifiedModel (line 350) | class QualifiedModel(BaseModel):
  class RerankerModel (line 358) | class RerankerModel(BaseModel):
  class RewriteQueryParams (line 365) | class RewriteQueryParams(BaseModel):
    method validate_prompt (line 387) | def validate_prompt(cls, v: str) -> str:
    method validate_max_tokens (line 394) | def validate_max_tokens(cls, v: int) -> int:
    method validate_temperature (line 403) | def validate_temperature(cls, v: float) -> float:
  class FileSearchParams (line 409) | class FileSearchParams(BaseModel):
    method validate_header_template (line 423) | def validate_header_template(cls, v: str) -> str:
  class ContextPromptParams (line 433) | class ContextPromptParams(BaseModel):
    method validate_chunk_annotation_template (line 447) | def validate_chunk_annotation_template(cls, v: str) -> str:
    method validate_context_template (line 458) | def validate_context_template(cls, v: str) -> str:
  class AnnotationPromptParams (line 466) | class AnnotationPromptParams(BaseModel):
    method validate_chunk_annotation_template (line 484) | def validate_chunk_annotation_template(cls, v: str) -> str:
    method validate_annotation_instruction_template (line 497) | def validate_annotation_instruction_template(cls, v: str) -> str:
  class FileIngestionParams (line 503) | class FileIngestionParams(BaseModel):
  class ChunkRetrievalParams (line 516) | class ChunkRetrievalParams(BaseModel):
  class FileBatchParams (line 541) | class FileBatchParams(BaseModel):
  class ContextualRetrievalParams (line 558) | class ContextualRetrievalParams(BaseModel):
  class VectorStoresConfig (line 586) | class VectorStoresConfig(BaseModel):
  class SafetyConfig (line 636) | class SafetyConfig(BaseModel):
  class QuotaPeriod (line 645) | class QuotaPeriod(StrEnum):
  class QuotaConfig (line 649) | class QuotaConfig(BaseModel):
  class CORSConfig (line 658) | class CORSConfig(BaseModel):
    method validate_credentials_config (line 668) | def validate_credentials_config(self) -> Self:
  function process_cors_config (line 674) | def process_cors_config(cors_config: bool | CORSConfig | None) -> CORSCo...
  class RegisteredResources (line 693) | class RegisteredResources(BaseModel):
  class ServerConfig (line 705) | class ServerConfig(BaseModel):
  class StackConfig (line 748) | class StackConfig(BaseModel):
    method validate_external_providers_dir (line 827) | def validate_external_providers_dir(cls, v):
    method validate_distro_name_migration (line 835) | def validate_distro_name_migration(self) -> "StackConfig":
    method validate_server_stores (line 863) | def validate_server_stores(self) -> "StackConfig":

FILE: src/llama_stack/core/distribution.py
  function stack_apis (line 31) | def stack_apis() -> list[Api]:
  class AutoRoutedApiInfo (line 35) | class AutoRoutedApiInfo(BaseModel):
  function builtin_automatically_routed_apis (line 40) | def builtin_automatically_routed_apis() -> list[AutoRoutedApiInfo]:
  function providable_apis (line 73) | def providable_apis() -> list[Api]:
  function _load_remote_provider_spec (line 78) | def _load_remote_provider_spec(spec_data: dict[str, Any], api: Api) -> P...
  function _load_inline_provider_spec (line 83) | def _load_inline_provider_spec(spec_data: dict[str, Any], api: Api, prov...
  function get_provider_registry (line 88) | def get_provider_registry(
  function get_external_providers_from_dir (line 172) | def get_external_providers_from_dir(
  function get_external_providers_from_module (line 224) | def get_external_providers_from_module(

FILE: src/llama_stack/core/exceptions/mapping.py
  function translate_exception_to_http (line 44) | def translate_exception_to_http(exc: Exception) -> HTTPException | None:

FILE: src/llama_stack/core/exceptions/translation.py
  function translate_exception (line 15) | def translate_exception(exc: Exception) -> HTTPException:

FILE: src/llama_stack/core/external.py
  function load_external_apis (line 17) | def load_external_apis(config: StackConfig | None) -> dict[Api, External...

FILE: src/llama_stack/core/id_generation.py
  function generate_object_id (line 15) | def generate_object_id(kind: str, factory: IdFactory) -> str:
  function set_id_override (line 28) | def set_id_override(override: IdOverride) -> IdOverride | None:
  function reset_id_override (line 38) | def reset_id_override(previous: IdOverride | None) -> None:

FILE: src/llama_stack/core/inspect.py
  class DistributionInspectConfig (line 31) | class DistributionInspectConfig(BaseModel):
  function get_provider_impl (line 35) | async def get_provider_impl(config, deps):
  class DistributionInspectImpl (line 41) | class DistributionInspectImpl(Inspect):
    method __init__ (line 42) | def __init__(self, config: DistributionInspectConfig, deps):
    method initialize (line 46) | async def initialize(self) -> None:
    method list_routes (line 49) | async def list_routes(self, api_filter: str | None = None) -> ListRout...
    method health (line 162) | async def health(self) -> HealthInfo:
    method version (line 165) | async def version(self) -> VersionInfo:
    method shutdown (line 168) | async def shutdown(self) -> None:

FILE: src/llama_stack/core/library_client.py
  function convert_pydantic_to_json_value (line 59) | def convert_pydantic_to_json_value(value: Any) -> Any:
  function convert_to_pydantic (line 72) | def convert_to_pydantic(annotation: Any, value: Any) -> Any:
  class LibraryClientUploadFile (line 113) | class LibraryClientUploadFile:
    method __init__ (line 116) | def __init__(self, filename: str, content: bytes):
    method read (line 121) | async def read(self) -> bytes:
  class LibraryClientHttpxResponse (line 125) | class LibraryClientHttpxResponse:
    method __init__ (line 128) | def __init__(self, response):
  class LlamaStackAsLibraryClient (line 134) | class LlamaStackAsLibraryClient(LlamaStackClient):
    method __init__ (line 135) | def __init__(
    method initialize (line 158) | def initialize(self):
    method shutdown (line 164) | def shutdown(self) -> None:
    method __enter__ (line 187) | def __enter__(self) -> "LlamaStackAsLibraryClient":
    method __exit__ (line 199) | def __exit__(self, exc_type, exc_val, exc_tb) -> None:
    method request (line 203) | def request(self, *args, **kwargs):
  class AsyncLlamaStackAsLibraryClient (line 233) | class AsyncLlamaStackAsLibraryClient(AsyncLlamaStackClient):
    method __init__ (line 234) | def __init__(
    method _remove_root_logger_handlers (line 268) | def _remove_root_logger_handlers(self):
    method initialize (line 278) | async def initialize(self) -> bool:
    method shutdown (line 326) | async def shutdown(self) -> None:
    method __aenter__ (line 346) | async def __aenter__(self) -> "AsyncLlamaStackAsLibraryClient":
    method __aexit__ (line 359) | async def __aexit__(self, exc_type, exc_val, exc_tb) -> None:
    method request (line 363) | async def request(
    method _handle_file_uploads (line 396) | def _handle_file_uploads(self, options: Any, body: dict) -> tuple[dict...
    method _call_non_streaming (line 421) | async def _call_non_streaming(
    method _call_streaming (line 490) | async def _call_streaming(
    method _convert_body (line 565) | def _convert_body(self, func: Any, body: dict | None = None, exclude_p...

FILE: src/llama_stack/core/prompts/prompts.py
  class PromptServiceConfig (line 27) | class PromptServiceConfig(BaseModel):
  function get_provider_impl (line 36) | async def get_provider_impl(config: PromptServiceConfig, deps: dict[Any,...
  class PromptServiceImpl (line 43) | class PromptServiceImpl(Prompts):
    method __init__ (line 46) | def __init__(self, config: PromptServiceConfig, deps: dict[Any, Any]):
    method initialize (line 51) | async def initialize(self) -> None:
    method _get_default_key (line 58) | def _get_default_key(self, prompt_id: str) -> str:
    method _get_prompt_key (line 62) | async def _get_prompt_key(self, prompt_id: str, version: int | None = ...
    method _get_version_key (line 73) | def _get_version_key(self, prompt_id: str, version: str) -> str:
    method _get_list_key_prefix (line 77) | def _get_list_key_prefix(self) -> str:
    method _serialize_prompt (line 81) | def _serialize_prompt(self, prompt: Prompt) -> str:
    method _deserialize_prompt (line 93) | def _deserialize_prompt(self, data: str) -> Prompt:
    method list_prompts (line 104) | async def list_prompts(self) -> ListPromptsResponse:
    method get_prompt (line 127) | async def get_prompt(self, request: GetPromptRequest) -> Prompt:
    method create_prompt (line 137) | async def create_prompt(self, request: CreatePromptRequest) -> Prompt:
    method update_prompt (line 157) | async def update_prompt(self, request: UpdatePromptRequest) -> Prompt:
    method delete_prompt (line 187) | async def delete_prompt(self, request: DeletePromptRequest) -> None:
    method list_prompt_versions (line 197) | async def list_prompt_versions(self, request: ListPromptVersionsReques...
    method set_default_version (line 223) | async def set_default_version(self, request: SetDefaultVersionRequest)...
    method shutdown (line 235) | async def shutdown(self) -> None:

FILE: src/llama_stack/core/providers.py
  class ProviderImplConfig (line 28) | class ProviderImplConfig(BaseModel):
  function get_provider_impl (line 32) | async def get_provider_impl(config, deps):
  class ProviderImpl (line 38) | class ProviderImpl(Providers):
    method __init__ (line 39) | def __init__(self, config, deps):
    method initialize (line 43) | async def initialize(self) -> None:
    method shutdown (line 46) | async def shutdown(self) -> None:
    method list_providers (line 50) | async def list_providers(self) -> ListProvidersResponse:
    method inspect_provider (line 77) | async def inspect_provider(self, request: InspectProviderRequest) -> P...
    method get_providers_health (line 85) | async def get_providers_health(self) -> dict[str, dict[str, HealthResp...

FILE: src/llama_stack/core/request_headers.py
  class RequestProviderDataContext (line 23) | class RequestProviderDataContext(AbstractContextManager):
    method __init__ (line 26) | def __init__(self, provider_data: dict[str, Any] | None = None, user: ...
    method __enter__ (line 33) | def __enter__(self):
    method __exit__ (line 38) | def __exit__(self, exc_type, exc_val, exc_tb):
  class NeedsRequestProviderData (line 44) | class NeedsRequestProviderData:
    method get_request_provider_data (line 45) | def get_request_provider_data(self) -> Any:
  function parse_request_provider_data (line 68) | def parse_request_provider_data(headers: dict[str, str]) -> dict[str, An...
  function request_provider_data_context (line 90) | def request_provider_data_context(
  function get_authenticated_user (line 98) | def get_authenticated_user() -> User | None:
  function user_from_scope (line 106) | def user_from_scope(scope: dict) -> User | None:

FILE: src/llama_stack/core/resolver.py
  class InvalidProviderError (line 70) | class InvalidProviderError(Exception):
  function api_protocol_map (line 74) | def api_protocol_map(external_apis: dict[Api, ExternalApiSpec] | None = ...
  function api_protocol_map_for_compliance_check (line 123) | def api_protocol_map_for_compliance_check(config: Any) -> dict[Api, Any]:
  function additional_protocols_map (line 131) | def additional_protocols_map() -> dict[Api, Any]:
  class ProviderWithSpec (line 147) | class ProviderWithSpec(Provider):
  function resolve_impls (line 154) | async def resolve_impls(
  function specs_for_autorouted_apis (line 185) | def specs_for_autorouted_apis(apis_to_serve: list[str] | set[str]) -> di...
  function validate_and_prepare_providers (line 232) | def validate_and_prepare_providers(
  function validate_provider (line 261) | def validate_provider(provider: Provider, api: Api, provider_registry: P...
  function sort_providers_by_deps (line 276) | def sort_providers_by_deps(
  function instantiate_providers (line 290) | async def instantiate_providers(
  function topological_sort (line 341) | def topological_sort(
  function instantiate_provider (line 375) | async def instantiate_provider(
  function check_protocol_compliance (line 443) | def check_protocol_compliance(obj: Any, protocol: Any) -> None:
  function resolve_remote_stack_impls (line 490) | async def resolve_remote_stack_impls(

FILE: src/llama_stack/core/routers/__init__.py
  function get_routing_table_impl (line 19) | async def get_routing_table_impl(
  function get_auto_router_impl (line 53) | async def get_auto_router_impl(

FILE: src/llama_stack/core/routers/datasets.py
  class DatasetIORouter (line 23) | class DatasetIORouter(DatasetIO):
    method __init__ (line 24) | def __init__(
    method initialize (line 31) | async def initialize(self) -> None:
    method shutdown (line 35) | async def shutdown(self) -> None:
    method register_dataset (line 39) | async def register_dataset(
    method iterrows (line 56) | async def iterrows(self, request: IterRowsRequest) -> PaginatedResponse:
    method append_rows (line 67) | async def append_rows(self, params: AppendRowsParams) -> None:

FILE: src/llama_stack/core/routers/eval_scoring.py
  class ScoringRouter (line 35) | class ScoringRouter(Scoring):
    method __init__ (line 36) | def __init__(
    method initialize (line 43) | async def initialize(self) -> None:
    method shutdown (line 47) | async def shutdown(self) -> None:
    method score_batch (line 51) | async def score_batch(
    method score (line 75) | async def score(
  class EvalRouter (line 95) | class EvalRouter(Eval):
    method __init__ (line 96) | def __init__(
    method initialize (line 103) | async def initialize(self) -> None:
    method shutdown (line 107) | async def shutdown(self) -> None:
    method run_eval (line 111) | async def run_eval(
    method evaluate_rows (line 138) | async def evaluate_rows(
    method job_status (line 175) | async def job_status(
    method job_cancel (line 200) | async def job_cancel(
    method job_result (line 225) | async def job_result(

FILE: src/llama_stack/core/routers/inference.py
  class InferenceRouter (line 57) | class InferenceRouter(Inference):
    method __init__ (line 60) | def __init__(
    method initialize (line 69) | async def initialize(self) -> None:
    method shutdown (line 72) | async def shutdown(self) -> None:
    method register_model (line 80) | async def register_model(
    method _get_model_provider (line 100) | async def _get_model_provider(self, model_id: str, expected_model_type...
    method _get_provider_by_fallback (line 112) | async def _get_provider_by_fallback(self, model_id: str, expected_mode...
    method rerank (line 146) | async def rerank(
    method openai_completion (line 154) | async def openai_completion(
    method openai_chat_completion (line 172) | async def openai_chat_completion(
    method openai_embeddings (line 220) | async def openai_embeddings(
    method list_chat_completions (line 235) | async def list_chat_completions(
    method get_chat_completion (line 248) | async def get_chat_completion(self, request: GetChatCompletionRequest)...
    method _nonstream_openai_chat_completion (line 253) | async def _nonstream_openai_chat_completion(
    method health (line 264) | async def health(self) -> dict[str, HealthResponse]:
    method stream_tokens_and_compute_metrics_openai_chat (line 287) | async def stream_tokens_and_compute_metrics_openai_chat(

FILE: src/llama_stack/core/routers/safety.py
  class SafetyRouter (line 28) | class SafetyRouter(Safety):
    method __init__ (line 29) | def __init__(
    method initialize (line 38) | async def initialize(self) -> None:
    method shutdown (line 42) | async def shutdown(self) -> None:
    method register_shield (line 46) | async def register_shield(self, request: RegisterShieldRequest) -> Shi...
    method unregister_shield (line 50) | async def unregister_shield(self, identifier: str) -> None:
    method run_shield (line 54) | async def run_shield(self, request: RunShieldRequest) -> RunShieldResp...
    method run_moderation (line 62) | async def run_moderation(self, request: RunModerationRequest) -> Moder...

FILE: src/llama_stack/core/routers/tool_runtime.py
  class ToolRuntimeRouter (line 27) | class ToolRuntimeRouter(ToolRuntime):
    method __init__ (line 28) | def __init__(
    method initialize (line 35) | async def initialize(self) -> None:
    method shutdown (line 39) | async def shutdown(self) -> None:
    method invoke_tool (line 43) | async def invoke_tool(self, tool_name: str, kwargs: dict[str, Any], au...
    method list_runtime_tools (line 98) | async def list_runtime_tools(

FILE: src/llama_stack/core/routers/vector_io.py
  class VectorIORouter (line 68) | class VectorIORouter(VectorIO):
    method __init__ (line 71) | def __init__(
    method initialize (line 81) | async def initialize(self) -> None:
    method shutdown (line 85) | async def shutdown(self) -> None:
    method _get_provider_id (line 89) | def _get_provider_id(self, vector_store_id: str) -> str:
    method _rewrite_query_for_search (line 109) | async def _rewrite_query_for_search(self, query: str) -> str:
    method _get_embedding_model_dimension (line 149) | async def _get_embedding_model_dimension(self, embedding_model_id: str...
    method insert_chunks (line 162) | async def insert_chunks(
    method query_chunks (line 195) | async def query_chunks(
    method openai_create_vector_store (line 243) | async def openai_create_vector_store(
    method openai_list_vector_stores (line 352) | async def openai_list_vector_stores(
    method openai_retrieve_vector_store (line 404) | async def openai_retrieve_vector_store(
    method openai_update_vector_store (line 411) | async def openai_update_vector_store(
    method openai_delete_vector_store (line 429) | async def openai_delete_vector_store(
    method openai_search_vector_store (line 448) | async def openai_search_vector_store(
    method openai_attach_file_to_vector_store (line 495) | async def openai_attach_file_to_vector_store(
    method openai_list_files_in_vector_store (line 539) | async def openai_list_files_in_vector_store(
    method openai_retrieve_vector_store_file (line 558) | async def openai_retrieve_vector_store_file(
    method openai_retrieve_vector_store_file_contents (line 569) | async def openai_retrieve_vector_store_file_contents(
    method openai_update_vector_store_file (line 588) | async def openai_update_vector_store_file(
    method openai_delete_vector_store_file (line 601) | async def openai_delete_vector_store_file(
    method health (line 624) | async def health(self) -> dict[str, HealthResponse]:
    method openai_create_vector_store_file_batch (line 647) | async def openai_create_vector_store_file_batch(
    method openai_retrieve_vector_store_file_batch (line 660) | async def openai_retrieve_vector_store_file_batch(
    method openai_list_files_in_vector_store_file_batch (line 671) | async def openai_list_files_in_vector_store_file_batch(
    method openai_cancel_vector_store_file_batch (line 692) | async def openai_cancel_vector_store_file_batch(

FILE: src/llama_stack/core/routing_tables/benchmarks.py
  class BenchmarksRoutingTable (line 27) | class BenchmarksRoutingTable(CommonRoutingTableImpl, Benchmarks):
    method list_benchmarks (line 28) | async def list_benchmarks(self, request: ListBenchmarksRequest) -> Lis...
    method get_benchmark (line 31) | async def get_benchmark(self, request: GetBenchmarkRequest) -> Benchmark:
    method register_benchmark (line 37) | async def register_benchmark(
    method unregister_benchmark (line 63) | async def unregister_benchmark(self, request: UnregisterBenchmarkReque...

FILE: src/llama_stack/core/routing_tables/common.py
  function get_impl_api (line 26) | def get_impl_api(p: Any) -> Api:
  function register_object_with_provider (line 31) | async def register_object_with_provider(obj: RoutableObject, p: Any) -> ...
  function unregister_object_from_provider (line 54) | async def unregister_object_from_provider(obj: RoutableObject, p: Any) -...
  class CommonRoutingTableImpl (line 77) | class CommonRoutingTableImpl(RoutingTable):
    method __init__ (line 78) | def __init__(
    method initialize (line 88) | async def initialize(self) -> None:
    method shutdown (line 120) | async def shutdown(self) -> None:
    method refresh (line 124) | async def refresh(self) -> None:
    method get_provider_impl (line 127) | async def get_provider_impl(self, routing_key: str, provider_id: str |...
    method get_object_by_identifier (line 173) | async def get_object_by_identifier(self, type: str, identifier: str) -...
    method unregister_object (line 186) | async def unregister_object(self, obj: RoutableObjectWithProvider) -> ...
    method register_object (line 193) | async def register_object(self, obj: RoutableObjectWithProvider) -> Ro...
    method assert_action_allowed (line 231) | async def assert_action_allowed(
    method get_all_with_type (line 245) | async def get_all_with_type(self, type: str) -> list[RoutableObjectWit...
  function lookup_model (line 258) | async def lookup_model(routing_table: CommonRoutingTableImpl, model_id: ...

FILE: src/llama_stack/core/routing_tables/datasets.py
  class DatasetsRoutingTable (line 34) | class DatasetsRoutingTable(CommonRoutingTableImpl, Datasets):
    method list_datasets (line 35) | async def list_datasets(self) -> ListDatasetsResponse:
    method get_dataset (line 38) | async def get_dataset(self, request: GetDatasetRequest) -> Dataset:
    method register_dataset (line 44) | async def register_dataset(self, request: RegisterDatasetRequest) -> D...
    method unregister_dataset (line 89) | async def unregister_dataset(self, request: UnregisterDatasetRequest) ...

FILE: src/llama_stack/core/routing_tables/models.py
  class ModelsRoutingTable (line 36) | class ModelsRoutingTable(CommonRoutingTableImpl, Models):
    method refresh (line 39) | async def refresh(self) -> None:
    method _get_dynamic_models_from_provider_data (line 63) | async def _get_dynamic_models_from_provider_data(self) -> list[Model]:
    method list_models (line 139) | async def list_models(self) -> ListModelsResponse:
    method openai_list_models (line 152) | async def openai_list_models(self) -> OpenAIListModelsResponse:
    method get_model (line 182) | async def get_model(self, request_or_model_id: GetModelRequest | str) ...
    method get_provider_impl (line 190) | async def get_provider_impl(self, model_id: str) -> Any:
    method has_model (line 196) | async def has_model(self, model_id: str) -> bool:
    method register_model (line 209) | async def register_model(
    method unregister_model (line 264) | async def unregister_model(
    method update_registered_models (line 285) | async def update_registered_models(

FILE: src/llama_stack/core/routing_tables/scoring_functions.py
  class ScoringFunctionsRoutingTable (line 27) | class ScoringFunctionsRoutingTable(CommonRoutingTableImpl, ScoringFuncti...
    method list_scoring_functions (line 28) | async def list_scoring_functions(self, request: ListScoringFunctionsRe...
    method get_scoring_function (line 31) | async def get_scoring_function(self, request: GetScoringFunctionReques...
    method register_scoring_function (line 37) | async def register_scoring_function(
    method unregister_scoring_function (line 63) | async def unregister_scoring_function(self, request: UnregisterScoring...

FILE: src/llama_stack/core/routing_tables/shields.py
  class ShieldsRoutingTable (line 26) | class ShieldsRoutingTable(CommonRoutingTableImpl, Shields):
    method list_shields (line 27) | async def list_shields(self) -> ListShieldsResponse:
    method get_shield (line 30) | async def get_shield(self, request: GetShieldRequest) -> Shield:
    method register_shield (line 36) | async def register_shield(self, request: RegisterShieldRequest) -> Shi...
    method unregister_shield (line 61) | async def unregister_shield(self, request: UnregisterShieldRequest) ->...

FILE: src/llama_stack/core/routing_tables/toolgroups.py
  function parse_toolgroup_from_toolgroup_name_pair (line 26) | def parse_toolgroup_from_toolgroup_name_pair(toolgroup_name_with_maybe_t...
  class ToolGroupsRoutingTable (line 35) | class ToolGroupsRoutingTable(CommonRoutingTableImpl, ToolGroups):
    method get_provider_impl (line 40) | async def get_provider_impl(self, routing_key: str, provider_id: str |...
    method list_tools (line 52) | async def list_tools(
    method _index_tools (line 81) | async def _index_tools(self, toolgroup: ToolGroup, authorization: str ...
    method list_tool_groups (line 95) | async def list_tool_groups(self) -> ListToolGroupsResponse:
    method get_tool_group (line 98) | async def get_tool_group(self, toolgroup_id: str) -> ToolGroup:
    method get_tool (line 104) | async def get_tool(self, tool_name: str) -> ToolDef:
    method register_tool_group (line 113) | async def register_tool_group(
    method unregister_toolgroup (line 135) | async def unregister_toolgroup(self, toolgroup_id: str) -> None:
    method shutdown (line 138) | async def shutdown(self) -> None:

FILE: src/llama_stack/core/routing_tables/vector_stores.py
  class VectorStoresRoutingTable (line 45) | class VectorStoresRoutingTable(CommonRoutingTableImpl):
    method __init__ (line 52) | def __init__(
    method list_vector_stores (line 63) | async def list_vector_stores(self) -> list[VectorStoreWithOwner]:
    method register_vector_store (line 67) | async def register_vector_store(
    method insert_chunks (line 103) | async def insert_chunks(
    method query_chunks (line 111) | async def query_chunks(
    method openai_retrieve_vector_store (line 119) | async def openai_retrieve_vector_store(
    method openai_update_vector_store (line 127) | async def openai_update_vector_store(
    method openai_delete_vector_store (line 136) | async def openai_delete_vector_store(
    method unregister_vector_store (line 146) | async def unregister_vector_store(self, vector_store_id: str) -> None:
    method openai_search_vector_store (line 156) | async def openai_search_vector_store(
    method openai_attach_file_to_vector_store (line 165) | async def openai_attach_file_to_vector_store(
    method openai_list_files_in_vector_store (line 174) | async def openai_list_files_in_vector_store(
    method openai_retrieve_vector_store_file (line 194) | async def openai_retrieve_vector_store_file(
    method openai_retrieve_vector_store_file_contents (line 206) | async def openai_retrieve_vector_store_file_contents(
    method openai_update_vector_store_file (line 223) | async def openai_update_vector_store_file(
    method openai_delete_vector_store_file (line 237) | async def openai_delete_vector_store_file(
    method openai_create_vector_store_file_batch (line 249) | async def openai_create_vector_store_file_batch(
    method openai_retrieve_vector_store_file_batch (line 261) | async def openai_retrieve_vector_store_file_batch(
    method openai_list_files_in_vector_store_file_batch (line 273) | async def openai_list_files_in_vector_store_file_batch(
    method openai_cancel_vector_store_file_batch (line 295) | async def openai_cancel_vector_store_file_batch(

FILE: src/llama_stack/core/server/auth.py
  class AuthenticationMiddleware (line 24) | class AuthenticationMiddleware:
    method __init__ (line 94) | def __init__(self, app, auth_config: AuthenticationConfig, impls):
    method __call__ (line 99) | async def __call__(self, scope, receive, send):
    method _send_auth_error (line 160) | async def _send_auth_error(self, send, message, status=401):
  class RouteAuthorizationMiddleware (line 172) | class RouteAuthorizationMiddleware:
    method __init__ (line 180) | def __init__(self, app, route_policy: list[RouteAccessRule]):
    method __call__ (line 184) | async def __call__(self, scope, receive, send):
    method _is_route_allowed (line 209) | def _is_route_allowed(self, route: str, user: User | None) -> bool:
    method _rule_matches (line 247) | def _rule_matches(self, rule: RouteAccessRule, route: str, user: User ...
    method _route_matches (line 267) | def _route_matches(self, request_route: str, rule_patterns: str | list...
    method _evaluate_conditions (line 303) | def _evaluate_conditions(self, rule: RouteAccessRule, user: User | Non...
    method _send_error (line 350) | async def _send_error(self, send, message: str, status: int = 403):
  class _RouteContext (line 363) | class _RouteContext:
    method __init__ (line 371) | def __init__(self):

FILE: src/llama_stack/core/server/auth_providers.py
  class AuthResponse (line 30) | class AuthResponse(BaseModel):
  class AuthRequestContext (line 41) | class AuthRequestContext(BaseModel):
  class AuthRequest (line 49) | class AuthRequest(BaseModel):
  class AuthProvider (line 55) | class AuthProvider(ABC):
    method validate_token (line 59) | async def validate_token(self, token: str, scope: dict | None = None) ...
    method close (line 64) | async def close(self):
    method get_auth_error_message (line 68) | def get_auth_error_message(self, scope: dict | None = None) -> str:
  function get_attributes_from_claims (line 73) | def get_attributes_from_claims(claims: dict[str, str], mapping: dict[str...
  class OAuth2TokenAuthProvider (line 108) | class OAuth2TokenAuthProvider(AuthProvider):
    method __init__ (line 115) | def __init__(self, config: OAuth2TokenAuthConfig):
    method validate_token (line 119) | async def validate_token(self, token: str, scope: dict | None = None) ...
    method _get_jwks_client (line 126) | def _get_jwks_client(self) -> jwt.PyJWKClient:
    method validate_jwt_token (line 164) | async def validate_jwt_token(self, token: str, scope: dict | None = No...
    method introspect_token (line 192) | async def introspect_token(self, token: str, scope: dict | None = None...
    method close (line 248) | async def close(self):
    method get_auth_error_message (line 251) | def get_auth_error_message(self, scope: dict | None = None) -> str:
  class CustomAuthProvider (line 263) | class CustomAuthProvider(AuthProvider):
    method __init__ (line 266) | def __init__(self, config: CustomAuthConfig):
    method validate_token (line 270) | async def validate_token(self, token: str, scope: dict | None = None) ...
    method close (line 327) | async def close(self):
    method get_auth_error_message (line 333) | def get_auth_error_message(self, scope: dict | None = None) -> str:
  class GitHubTokenAuthProvider (line 342) | class GitHubTokenAuthProvider(AuthProvider):
    method __init__ (line 350) | def __init__(self, config: GitHubTokenAuthConfig):
    method validate_token (line 353) | async def validate_token(self, token: str, scope: dict | None = None) ...
    method close (line 379) | async def close(self):
    method get_auth_error_message (line 383) | def get_auth_error_message(self, scope: dict | None = None) -> str:
  function _get_github_user_info (line 388) | async def _get_github_user_info(access_token: str, github_api_base_url: ...
  class KubernetesAuthProvider (line 406) | class KubernetesAuthProvider(AuthProvider):
    method __init__ (line 413) | def __init__(self, config: KubernetesAuthProviderConfig):
    method _httpx_verify_value (line 416) | def _httpx_verify_value(self) -> bool | str:
    method validate_token (line 429) | async def validate_token(self, token: str, scope: dict | None = None) ...
    method close (line 484) | async def close(self):
  function create_auth_provider (line 489) | def create_auth_provider(config: AuthenticationConfig) -> AuthProvider:

FILE: src/llama_stack/core/server/fastapi_router_registry.py
  function has_router (line 71) | def has_router(api: "Api") -> bool:
  function build_fastapi_router (line 83) | def build_fastapi_router(api: "Api", impl: Any) -> APIRouter | None:
  function get_router_routes (line 103) | def get_router_routes(router: APIRouter) -> list[APIRoute]:

FILE: src/llama_stack/core/server/quota.py
  class QuotaMiddleware (line 21) | class QuotaMiddleware:
    method __init__ (line 33) | def __init__(
    method _get_kv (line 48) | async def _get_kv(self) -> KVStore:
    method __call__ (line 59) | async def __call__(self, scope: Scope, receive: Receive, send: Send):
    method _send_error (line 101) | async def _send_error(self, send: Send, status: int, message: str):

FILE: src/llama_stack/core/server/routes.py
  function get_all_api_routes (line 32) | def get_all_api_routes(
  function initialize_route_impls (line 79) | def initialize_route_impls(impls, external_apis: dict[Api, ExternalApiSp...
  function find_matching_route (line 168) | def find_matching_route(method: str, path: str, route_impls: RouteImpls)...

FILE: src/llama_stack/core/server/server.py
  function warn_with_traceback (line 70) | def warn_with_traceback(message, category, filename, lineno, file=None, ...
  function create_sse_event (line 80) | def create_sse_event(data: Any) -> str:
  function global_exception_handler (line 89) | async def global_exception_handler(request: Request, exc: Exception):
  class StackApp (line 105) | class StackApp(FastAPI):
    method __init__ (line 111) | def __init__(self, config: StackConfig, *args, **kwargs):
  function lifespan (line 124) | async def lifespan(app: StackApp):
  function is_streaming_request (line 135) | def is_streaming_request(func_name: str, request: Request, **kwargs):
  function maybe_await (line 150) | async def maybe_await(value):
  function sse_generator (line 156) | async def sse_generator(event_gen_coroutine):
  function log_request_pre_validation (line 171) | async def log_request_pre_validation(request: Request):
  function create_dynamic_typed_route (line 188) | def create_dynamic_typed_route(func: Any, method: str, route: str) -> Ca...
  class ClientVersionMiddleware (line 252) | class ClientVersionMiddleware:
    method __init__ (line 253) | def __init__(self, app):
    method __call__ (line 257) | async def __call__(self, scope, receive, send):
  class ProviderDataMiddleware (line 288) | class ProviderDataMiddleware:
    method __init__ (line 296) | def __init__(self, app):
    method __call__ (line 299) | async def __call__(self, scope, receive, send):
  function create_app (line 324) | def create_app() -> StackApp:
  function _log_run_config (line 493) | def _log_run_config(run_config: StackConfig):
  function extract_path_params (line 501) | def extract_path_params(route: str) -> list[str]:
  function remove_disabled_providers (line 509) | def remove_disabled_providers(obj):

FILE: src/llama_stack/core/stack.py
  class LlamaStack (line 83) | class LlamaStack(
  function is_request_model (line 145) | def is_request_model(t: Any) -> bool:
  function invoke_with_optional_request (line 158) | async def invoke_with_optional_request(method: Any) -> Any:
  function register_resources (line 220) | async def register_resources(run_config: StackConfig, impls: dict[Api, A...
  function register_connectors (line 258) | async def register_connectors(run_config: StackConfig, impls: dict[Api, ...
  function validate_vector_stores_config (line 286) | async def validate_vector_stores_config(vector_stores_config: VectorStor...
  function _validate_embedding_model (line 305) | async def _validate_embedding_model(embedding_model: QualifiedModel, imp...
  function _validate_reranker_model (line 337) | async def _validate_reranker_model(reranker_model: RerankerModel, impls:...
  function _validate_rewrite_query_model (line 359) | async def _validate_rewrite_query_model(rewrite_query_model: QualifiedMo...
  function validate_safety_config (line 383) | async def validate_safety_config(safety_config: SafetyConfig | None, imp...
  class EnvVarError (line 407) | class EnvVarError(Exception):
    method __init__ (line 408) | def __init__(self, var_name: str, path: str = ""):
  function replace_env_vars (line 419) | def replace_env_vars(config: Any, path: str = "") -> Any:
  function _convert_string_to_proper_type (line 561) | def _convert_string_to_proper_type(value: str) -> Any:
  function cast_distro_name_to_string (line 589) | def cast_distro_name_to_string(config_dict: dict[str, Any]) -> dict[str,...
  function add_internal_implementations (line 596) | def add_internal_implementations(impls: dict[Api, Any], config: StackCon...
  function _initialize_storage (line 639) | def _initialize_storage(run_config: StackConfig):
  class Stack (line 658) | class Stack:
    method __init__ (line 659) | def __init__(self, run_config: StackConfig, provider_registry: Provide...
    method initialize (line 666) | async def initialize(self):
    method create_registry_refresh_task (line 708) | def create_registry_refresh_task(self):
    method shutdown (line 727) | async def shutdown(self):
  function refresh_registry_once (line 767) | async def refresh_registry_once(impls: dict[Api, Any]):
  function refresh_registry_task (line 774) | async def refresh_registry_task(impls: dict[Api, Any]):
  function get_stack_run_config_from_distro (line 782) | def get_stack_run_config_from_distro(distro: str) -> StackConfig:
  function run_config_from_dynamic_config_spec (line 793) | def run_config_from_dynamic_config_spec(

FILE: src/llama_stack/core/storage/datatypes.py
  class StorageBackendType (line 19) | class StorageBackendType(StrEnum):
  class CommonConfig (line 28) | class CommonConfig(BaseModel):
  class RedisKVStoreConfig (line 35) | class RedisKVStoreConfig(CommonConfig):
    method url (line 41) | def url(self) -> str:
    method pip_packages (line 45) | def pip_packages(cls) -> list[str]:
    method sample_run_config (line 49) | def sample_run_config(cls):
  class SqliteKVStoreConfig (line 57) | class SqliteKVStoreConfig(CommonConfig):
    method pip_packages (line 64) | def pip_packages(cls) -> list[str]:
    method sample_run_config (line 68) | def sample_run_config(cls, __distro_dir__: str, db_name: str = "kvstor...
  class PostgresKVStoreConfig (line 75) | class PostgresKVStoreConfig(CommonConfig):
    method sample_run_config (line 87) | def sample_run_config(cls, table_name: str = "llamastack_kvstore", **k...
    method validate_table_name (line 100) | def validate_table_name(cls, v: str) -> str:
    method pip_packages (line 115) | def pip_packages(cls) -> list[str]:
  class MongoDBKVStoreConfig (line 119) | class MongoDBKVStoreConfig(CommonConfig):
    method pip_packages (line 129) | def pip_packages(cls) -> list[str]:
    method sample_run_config (line 133) | def sample_run_config(cls, collection_name: str = "llamastack_kvstore"):
  class SqlAlchemySqlStoreConfig (line 145) | class SqlAlchemySqlStoreConfig(BaseModel):
    method engine_str (line 150) | def engine_str(self) -> str: ...
    method pip_packages (line 154) | def pip_packages(cls) -> list[str]:
  class SqliteSqlStoreConfig (line 158) | class SqliteSqlStoreConfig(SqlAlchemySqlStoreConfig):
    method engine_str (line 165) | def engine_str(self) -> str:
    method sample_run_config (line 169) | def sample_run_config(cls, __distro_dir__: str, db_name: str = "sqlsto...
    method pip_packages (line 176) | def pip_packages(cls) -> list[str]:
  class PostgresSqlStoreConfig (line 180) | class PostgresSqlStoreConfig(SqlAlchemySqlStoreConfig):
    method engine_str (line 192) | def engine_str(self) -> str:
    method pip_packages (line 196) | def pip_packages(cls) -> list[str]:
    method sample_run_config (line 200) | def sample_run_config(cls, **kwargs):
  class SqlStoreReference (line 216) | class SqlStoreReference(BaseModel):
  class KVStoreReference (line 229) | class KVStoreReference(BaseModel):
  class InferenceStoreReference (line 252) | class InferenceStoreReference(SqlStoreReference):
  class ResponsesStoreReference (line 265) | class ResponsesStoreReference(InferenceStoreReference):
  class ServerStoresConfig (line 274) | class ServerStoresConfig(BaseModel):
  function _default_backends (line 310) | def _default_backends() -> dict[str, StorageBackendConfig]:
  class StorageConfig (line 322) | class StorageConfig(BaseModel):

FILE: src/llama_stack/core/storage/kvstore/config.py
  function get_pip_packages (line 24) | def get_pip_packages(store_config: dict | KVStoreConfig) -> list[str]:

FILE: src/llama_stack/core/storage/kvstore/kvstore.py
  function kvstore_dependencies (line 31) | def kvstore_dependencies():
  class InmemoryKVStoreImpl (line 42) | class InmemoryKVStoreImpl(KVStore):
    method __init__ (line 43) | def __init__(self):
    method initialize (line 46) | async def initialize(self) -> None:
    method get (line 49) | async def get(self, key: str) -> str | None:
    method set (line 52) | async def set(self, key: str, value: str, expiration: datetime | None ...
    method values_in_range (line 55) | async def values_in_range(self, start_key: str, end_key: str) -> list[...
    method keys_in_range (line 58) | async def keys_in_range(self, start_key: str, end_key: str) -> list[str]:
    method delete (line 62) | async def delete(self, key: str) -> None:
    method shutdown (line 65) | async def shutdown(self) -> None:
  function register_kvstore_backends (line 74) | def register_kvstore_backends(backends: dict[str, StorageBackendConfig])...
  function kvstore_impl (line 88) | async def kvstore_impl(reference: KVStoreReference) -> KVStore:
  function shutdown_kvstore_backends (line 134) | async def shutdown_kvstore_backends() -> None:

FILE: src/llama_stack/core/storage/kvstore/mongodb/mongodb.py
  class MongoDBKVStoreImpl (line 20) | class MongoDBKVStoreImpl(KVStore):
    method __init__ (line 21) | def __init__(self, config: MongoDBKVStoreConfig):
    method collection (line 26) | def collection(self) -> AsyncCollection:
    method initialize (line 31) | async def initialize(self) -> None:
    method _namespaced_key (line 44) | def _namespaced_key(self, key: str) -> str:
    method set (line 49) | async def set(self, key: str, value: str, expiration: datetime | None ...
    method get (line 54) | async def get(self, key: str) -> str | None:
    method delete (line 60) | async def delete(self, key: str) -> None:
    method values_in_range (line 64) | async def values_in_range(self, start_key: str, end_key: str) -> list[...
    method keys_in_range (line 76) | async def keys_in_range(self, start_key: str, end_key: str) -> list[str]:
    method shutdown (line 87) | async def shutdown(self) -> None:

FILE: src/llama_stack/core/storage/kvstore/postgres/postgres.py
  class PostgresKVStoreImpl (line 21) | class PostgresKVStoreImpl(KVStore):
    method __init__ (line 22) | def __init__(self, config: PostgresKVStoreConfig):
    method initialize (line 27) | async def initialize(self) -> None:
    method _cursor_or_raise (line 55) | def _cursor_or_raise(self) -> DictCursor:
    method _namespaced_key (line 60) | def _namespaced_key(self, key: str) -> str:
    method set (line 65) | async def set(self, key: str, value: str, expiration: datetime | None ...
    method get (line 78) | async def get(self, key: str) -> str | None:
    method delete (line 92) | async def delete(self, key: str) -> None:
    method values_in_range (line 100) | async def values_in_range(self, start_key: str, end_key: str) -> list[...
    method keys_in_range (line 116) | async def keys_in_range(self, start_key: str, end_key: str) -> list[str]:
    method shutdown (line 127) | async def shutdown(self) -> None:

FILE: src/llama_stack/core/storage/kvstore/redis/redis.py
  class RedisKVStoreImpl (line 16) | class RedisKVStoreImpl(KVStore):
    method __init__ (line 17) | def __init__(self, config: RedisKVStoreConfig):
    method initialize (line 21) | async def initialize(self) -> None:
    method _client (line 24) | def _client(self) -> Redis:
    method _namespaced_key (line 29) | def _namespaced_key(self, key: str) -> str:
    method set (line 34) | async def set(self, key: str, value: str, expiration: datetime | None ...
    method get (line 41) | async def get(self, key: str) -> str | None:
    method delete (line 54) | async def delete(self, key: str) -> None:
    method values_in_range (line 58) | async def values_in_range(self, start_key: str, end_key: str) -> list[...
    method keys_in_range (line 85) | async def keys_in_range(self, start_key: str, end_key: str) -> list[str]:
    method shutdown (line 103) | async def shutdown(self) -> None:

FILE: src/llama_stack/core/storage/kvstore/sqlite/sqlite.py
  class SqliteKVStoreImpl (line 20) | class SqliteKVStoreImpl(KVStore):
    method __init__ (line 21) | def __init__(self, config: SqliteKVStoreConfig):
    method __str__ (line 26) | def __str__(self):
    method _is_memory_db (line 29) | def _is_memory_db(self) -> bool:
    method initialize (line 33) | async def initialize(self):
    method shutdown (line 68) | async def shutdown(self):
    method set (line 74) | async def set(self, key: str, value: str, expiration: datetime | None ...
    method get (line 91) | async def get(self, key: str) -> str | None:
    method delete (line 120) | async def delete(self, key: str) -> None:
    method values_in_range (line 131) | async def values_in_range(self, start_key: str, end_key: str) -> list[...
    method keys_in_range (line 156) | async def keys_in_range(self, start_key: str, end_key: str) -> list[str]:

FILE: src/llama_stack/core/storage/sqlstore/authorized_sqlstore.py
  function _enhance_item_with_access_control (line 51) | def _enhance_item_with_access_control(item: Mapping[str, Any], current_u...
  class SqlRecord (line 68) | class SqlRecord(ProtectedResource):
    method __init__ (line 69) | def __init__(self, record_id: str, table_name: str, owner: User | None):
  class AuthorizedSqlStore (line 75) | class AuthorizedSqlStore:
    method __init__ (line 83) | def __init__(self, sql_store: SqlStore, policy: list[AccessRule]):
    method _detect_database_type (line 95) | def _detect_database_type(self) -> None:
    method _validate_sql_optimized_policy (line 104) | def _validate_sql_optimized_policy(self) -> None:
    method create_table (line 119) | async def create_table(self, table: str, schema: Mapping[str, ColumnTy...
    method insert (line 132) | async def insert(self, table: str, data: Mapping[str, Any] | Sequence[...
    method upsert (line 142) | async def upsert(
    method fetch_all (line 159) | async def fetch_all(
    method fetch_one (line 204) | async def fetch_one(
    method update (line 222) | async def update(self, table: str, data: Mapping[str, Any], where: Map...
    method delete (line 237) | async def delete(self, table: str, where: Mapping[str, Any]) -> None:
    method _build_access_control_where_clause (line 241) | def _build_access_control_where_clause(self, policy: list[AccessRule])...
    method _json_extract (line 255) | def _json_extract(self, column: str, path: str) -> str:
    method _json_extract_text (line 272) | def _json_extract_text(self, column: str, path: str) -> str:
    method _get_public_access_conditions (line 289) | def _get_public_access_conditions(self) -> list[str]:
    method _build_default_policy_where_clause (line 300) | def _build_default_policy_where_clause(self, current_user: User | None...
    method _build_conservative_where_clause (line 330) | def _build_conservative_where_clause(self) -> tuple[str, dict[str, Any]]:

FILE: src/llama_stack/core/storage/sqlstore/sqlalchemy_sqlstore.py
  function _build_where_expr (line 47) | def _build_where_expr(column: ColumnElement, value: Any) -> ColumnElement:
  class SqlAlchemySqlStoreImpl (line 71) | class SqlAlchemySqlStoreImpl(SqlStore):
    method __init__ (line 72) | def __init__(self, config: SqlAlchemySqlStoreConfig):
    method _ensure_engine (line 82) | async def _ensure_engine(self):
    method shutdown (line 104) | async def shutdown(self):
    method create_engine (line 110) | def create_engine(self) -> AsyncEngine:
    method create_table (line 148) | async def create_table(
    method insert (line 185) | async def insert(self, table: str, data: Mapping[str, Any] | Sequence[...
    method upsert (line 192) | async def upsert(
    method fetch_all (line 217) | async def fetch_all(
    method fetch_one (line 323) | async def fetch_one(
    method update (line 336) | async def update(
    method delete (line 354) | async def delete(self, table: str, where: Mapping[str, Any]) -> None:
    method add_column_if_not_exists (line 367) | async def add_column_if_not_exists(
    method _add_column_now (line 384) | async def _add_column_now(
    method _get_dialect_insert (line 432) | def _get_dialect_insert(self, table: Table):

FILE: src/llama_stack/core/storage/sqlstore/sqlstore.py
  function get_pip_packages (line 34) | def get_pip_packages(store_config: dict | SqlStoreConfig) -> list[str]:
  function sqlstore_impl (line 48) | def sqlstore_impl(reference: SqlStoreReference) -> SqlStore:
  function register_sqlstore_backends (line 78) | def register_sqlstore_backends(backends: dict[str, StorageBackendConfig]...
  function shutdown_sqlstore_backends (line 90) | async def shutdown_sqlstore_backends() -> None:

FILE: src/llama_stack/core/store/registry.py
  class DistributionRegistry (line 22) | class DistributionRegistry(Protocol):
    method get_all (line 23) | async def get_all(self) -> list[RoutableObjectWithProvider]: ...
    method initialize (line 25) | async def initialize(self) -> None: ...
    method get (line 27) | async def get(self, type: str, identifier: str) -> RoutableObjectWithP...
    method get_cached (line 29) | def get_cached(self, type: str, identifier: str) -> RoutableObjectWith...
    method update (line 31) | async def update(self, obj: RoutableObjectWithProvider) -> RoutableObj...
    method register (line 33) | async def register(self, obj: RoutableObjectWithProvider) -> bool: ...
    method delete (line 35) | async def delete(self, type: str, identifier: str) -> None: ...
  function _get_registry_key_range (line 43) | def _get_registry_key_range() -> tuple[str, str]:
  function _parse_registry_values (line 49) | def _parse_registry_values(values: list[str]) -> list[RoutableObjectWith...
  class DiskDistributionRegistry (line 63) | class DiskDistributionRegistry(DistributionRegistry):
    method __init__ (line 64) | def __init__(self, kvstore: KVStore):
    method initialize (line 67) | async def initialize(self) -> None:
    method get_cached (line 70) | def get_cached(self, type: str, identifier: str) -> RoutableObjectWith...
    method get_all (line 74) | async def get_all(self) -> list[RoutableObjectWithProvider]:
    method get (line 79) | async def get(self, type: str, identifier: str) -> RoutableObjectWithP...
    method update (line 90) | async def update(self, obj: RoutableObjectWithProvider) -> RoutableObj...
    method register (line 97) | async def register(self, obj: RoutableObjectWithProvider) -> bool:
    method delete (line 129) | async def delete(self, type: str, identifier: str) -> None:
  class CachedDiskDistributionRegistry (line 133) | class CachedDiskDistributionRegistry(DiskDistributionRegistry):
    method __init__ (line 134) | def __init__(self, kvstore: KVStore, cache_ttl_seconds: float = 5.0):
    method _locked_cache (line 144) | async def _locked_cache(self):
    method _ensure_initialized (line 149) | async def _ensure_initialized(self):
    method initialize (line 169) | async def initialize(self) -> None:
    method get_cached (line 172) | def get_cached(self, type: str, identifier: str) -> RoutableObjectWith...
    method _should_refresh_cache (line 175) | def _should_refresh_cache(self) -> bool:
    method _refresh_cache_from_db (line 180) | async def _refresh_cache_from_db(self) -> None:
    method get_all (line 196) | async def get_all(self) -> list[RoutableObjectWithProvider]:
    method get (line 207) | async def get(self, type: str, identifier: str) -> RoutableObjectWithP...
    method register (line 226) | async def register(self, obj: RoutableObjectWithProvider) -> bool:
    method update (line 237) | async def update(self, obj: RoutableObjectWithProvider) -> RoutableObj...
    method delete (line 244) | async def delete(self, type: str, identifier: str) -> None:
  function create_dist_registry (line 252) | async def create_dist_registry(

FILE: src/llama_stack/core/task.py
  class RequestContext (line 19) | class RequestContext:
  function capture_request_context (line 34) | def capture_request_context() -> RequestContext:
  function activate_request_context (line 43) | def activate_request_context(ctx: RequestContext):
  function create_detached_background_task (line 58) | def create_detached_background_task(coro: Coroutine[Any, Any, Any]) -> a...

FILE: src/llama_stack/core/testing_context.py
  function get_test_context (line 15) | def get_test_context() -> str | None:
  function set_test_context (line 19) | def set_test_context(value: str | None):
  function reset_test_context (line 23) | def reset_test_context(token) -> None:
  function sync_test_context_from_provider_data (line 27) | def sync_test_context_from_provider_data():
  function is_debug_mode (line 47) | def is_debug_mode() -> bool:

FILE: src/llama_stack/core/utils/config.py
  function redact_sensitive_fields (line 10) | def redact_sensitive_fields(data: dict[str, Any]) -> dict[str, Any]:

FILE: src/llama_stack/core/utils/config_resolution.py
  function resolve_config_or_distro (line 18) | def resolve_config_or_distro(
  function _get_distro_config_path (line 72) | def _get_distro_config_path(distro_name: str, path: str | None = None) -...
  function _format_resolution_error (line 79) | def _format_resolution_error(config_or_distro: str) -> str:
  function _get_available_distros (line 104) | def _get_available_distros() -> list[str]:
  function _format_distro_suggestions (line 116) | def _format_distro_suggestions(distros: list[str], user_input: str) -> str:

FILE: src/llama_stack/core/utils/context.py
  function preserve_contexts_async_generator (line 13) | def preserve_contexts_async_generator[T](

FILE: src/llama_stack/core/utils/dynamic.py
  function instantiate_class_type (line 10) | def instantiate_class_type(fully_qualified_name):

FILE: src/llama_stack/core/utils/exec.py
  function formulate_run_args (line 20) | def formulate_run_args(image_type: str, distro_name: str) -> list:
  function in_notebook (line 44) | def in_notebook():
  function run_command (line 58) | def run_command(command: list[str]) -> int:

FILE: src/llama_stack/core/utils/image_types.py
  class LlamaStackImageType (line 10) | class LlamaStackImageType(enum.Enum):

FILE: src/llama_stack/core/utils/model_utils.py
  function model_local_dir (line 12) | def model_local_dir(descriptor: str) -> str:

FILE: src/llama_stack/core/utils/prompt_for_config.py
  function is_list_of_primitives (line 21) | def is_list_of_primitives(field_type):
  function is_basemodel_without_fields (line 31) | def is_basemodel_without_fields(typ):
  function can_recurse (line 35) | def can_recurse(typ):
  function get_literal_values (line 39) | def get_literal_values(field):
  function is_optional (line 46) | def is_optional(field_type):
  function get_non_none_type (line 51) | def get_non_none_type(field_type):
  function manually_validate_field (line 56) | def manually_validate_field(model: type[BaseModel], field_name: str, val...
  function is_discriminated_union (line 65) | def is_discriminated_union(typ) -> bool:
  function prompt_for_discriminated_union (line 75) | def prompt_for_discriminated_union(
  function prompt_for_config (line 129) | def prompt_for_config(config_type: type[BaseModel], existing_config: Bas...

FILE: src/llama_stack/core/utils/serialize.py
  class EnumEncoder (line 12) | class EnumEncoder(json.JSONEncoder):
    method default (line 13) | def default(self, obj):

FILE: src/llama_stack/core/utils/type_inspection.py
  function is_unwrapped_body_param (line 19) | def is_unwrapped_body_param(param_type: Any) -> bool:
  function is_body_param (line 48) | def is_body_param(param_type: Any) -> bool:

FILE: src/llama_stack/distributions/ci-tests/ci_tests.py
  function get_distribution_template (line 16) | def get_distribution_template() -> DistributionTemplate:

FILE: src/llama_stack/distributions/dell/dell.py
  function get_distribution_template (line 25) | def get_distribution_template() -> DistributionTemplate:

FILE: src/llama_stack/distributions/nvidia/nvidia.py
  function get_distribution_template (line 18) | def get_distribution_template(name: str = "nvidia") -> DistributionTempl...

FILE: src/llama_stack/distributions/oci/oci.py
  function get_distribution_template (line 16) | def get_distribution_template(name: str = "oci") -> DistributionTemplate:

FILE: src/llama_stack/distributions/open-benchmark/open_benchmark.py
  function get_inference_providers (line 38) | def get_inference_providers() -> tuple[list[Provider], dict[str, list[Pr...
  function get_distribution_template (line 96) | def get_distribution_template() -> DistributionTemplate:

FILE: src/llama_stack/distributions/starter-gpu/starter_gpu.py
  function get_distribution_template (line 13) | def get_distribution_template() -> DistributionTemplate:

FILE: src/llama_stack/distributions/starter/starter.py
  function _get_config_for_provider (line 49) | def _get_config_for_provider(provider_spec: ProviderSpec) -> dict[str, A...
  function get_remote_inference_providers (line 90) | def get_remote_inference_providers() -> list[Provider]:
  function get_distribution_template (line 118) | def get_distribution_template(name: str = "starter") -> DistributionTemp...

FILE: src/llama_stack/distributions/template.py
  function filter_empty_values (line 42) | def filter_empty_values(obj: Any) -> Any:
  function get_model_registry (line 87) | def get_model_registry(
  function get_shield_registry (line 129) | def get_shield_registry(
  class DefaultModel (line 172) | class DefaultModel(BaseModel):
  class RunConfigSettings (line 177) | class RunConfigSettings(BaseModel):
    method run_config (line 191) | def run_config(
  class DistributionTemplate (line 308) | class DistributionTemplate(BaseModel):
    method generate_markdown_docs (line 329) | def generate_markdown_docs(self) -> str:
    method save_distribution (line 385) | def save_distribution(self, yaml_output_dir: Path, doc_output_dir: Pat...

FILE: src/llama_stack/distributions/watsonx/watsonx.py
  function get_distribution_template (line 14) | def get_distribution_template(name: str = "watsonx") -> DistributionTemp...

FILE: src/llama_stack/env.py
  class MissingCredentialError (line 10) | class MissingCredentialError(Exception):
  function get_env_or_fail (line 14) | def get_env_or_fail(key: str) -> str:

FILE: src/llama_stack/log.py
  class LoggingConfig (line 21) | class LoggingConfig(BaseModel):
  function config_to_category_levels (line 62) | def config_to_category_levels(category: str, level: str):
  function parse_yaml_config (line 94) | def parse_yaml_config(yaml_config: LoggingConfig) -> dict[str, int]:
  function parse_environment_config (line 111) | def parse_environment_config(env_config: str) -> dict[str, int]:
  function strip_rich_markup (line 139) | def strip_rich_markup(text):
  class CustomRichHandler (line 144) | class CustomRichHandler(RichHandler):
    method __init__ (line 145) | def __init__(self, *args, **kwargs):
    method emit (line 153) | def emit(self, record):
  class CustomFileHandler (line 166) | class CustomFileHandler(logging.FileHandler):
    method __init__ (line 167) | def __init__(self, filename, mode="a", encoding=None, delay=False):
    method emit (line 173) | def emit(self, record):
  function setup_logging (line 179) | def setup_logging(category_levels: dict[str, int] | None = None, log_fil...
  function get_logger (line 313) | def get_logger(

FILE: src/llama_stack/models/llama/sku_list.py
  function resolve_model (line 21) | def resolve_model(descriptor: str) -> Model | None:
  function all_registered_models (line 28) | def all_registered_models() -> list[Model]:
  function llama2_family (line 40) | def llama2_family() -> list[Model]:
  function llama3_family (line 47) | def llama3_family() -> list[Model]:
  function llama3_1_family (line 54) | def llama3_1_family() -> list[Model]:
  function llama3_2_family (line 61) | def llama3_2_family() -> list[Model]:
  function llama3_3_family (line 68) | def llama3_3_family() -> list[Model]:
  function llama4_family (line 74) | def llama4_family() -> list[Model]:
  function llama4_base_models (line 81) | def llama4_base_models() -> list[Model]:
  function llama4_instruct_models (line 100) | def llama4_instruct_models() -> list[Model]:
  function llama2_base_models (line 128) | def llama2_base_models() -> list[Model]:
  function llama3_base_models (line 187) | def llama3_base_models() -> list[Model]:
  function llama3_1_base_models (line 228) | def llama3_1_base_models() -> list[Model]:
  function llama3_2_base_models (line 326) | def llama3_2_base_models() -> list[Model]:
  function llama2_instruct_models (line 409) | def llama2_instruct_models() -> list[Model]:
  function llama3_instruct_models (line 468) | def llama3_instruct_models() -> list[Model]:
  function llama3_1_instruct_models (line 509) | def llama3_1_instruct_models() -> list[Model]:
  function arch_args_1b (line 607) | def arch_args_1b() -> dict:
  function arch_args_3b (line 622) | def arch_args_3b() -> dict:
  function llama3_2_quantized_models (line 637) | def llama3_2_quantized_models() -> list[Model]:
  function llama3_2_instruct_models (line 706) | def llama3_2_instruct_models() -> list[Model]:
  function llama3_3_instruct_models (line 768) | def llama3_3_instruct_models() -> list[Model]:
  function safety_models (line 792) | def safety_models() -> list[Model]:
  class LlamaDownloadInfo (line 919) | class LlamaDownloadInfo:
  function llama_meta_net_info (line 925) | def llama_meta_net_info(model: Model) -> LlamaDownloadInfo:
  function llama_meta_pth_size (line 1004) | def llama_meta_pth_size(model: Model) -> int:

FILE: src/llama_stack/models/llama/sku_types.py
  class CheckpointQuantizationFormat (line 13) | class CheckpointQuantizationFormat(Enum):
  class ModelFamily (line 25) | class ModelFamily(Enum):
  class CoreModelId (line 35) | class CoreModelId(Enum):
  function is_multimodal (line 87) | def is_multimodal(model_id) -> bool:
  function model_family (line 99) | def model_family(model_id) -> ModelFamily:
  class Model (line 159) | class Model(BaseModel):
    method model_family (line 174) | def model_family(self) -> ModelFamily:
    method descriptor (line 178) | def descriptor(self, shorten_default_variant: bool = True) -> str:
    method is_instruct_model (line 184) | def is_instruct_model(self) -> bool:
    method is_featured (line 189) | def is_featured(self) -> bool:
    method max_seq_length (line 199) | def max_seq_length(self) -> int:

FILE: src/llama_stack/providers/inline/agents/builtin/__init__.py
  function get_provider_impl (line 14) | async def get_provider_impl(

FILE: src/llama_stack/providers/inline/agents/builtin/agents.py
  class BuiltinAgentsImpl (line 42) | class BuiltinAgentsImpl(Agents):
    method __init__ (line 43) | def __init__(
    method initialize (line 71) | async def initialize(self) -> None:
    method shutdown (line 92) | async def shutdown(self) -> None:
    method get_openai_response (line 97) | async def get_openai_response(
    method create_openai_response (line 104) | async def create_openai_response(
    method list_openai_responses (line 149) | async def list_openai_responses(
    method list_openai_response_input_items (line 158) | async def list_openai_response_input_items(
    method delete_openai_response (line 172) | async def delete_openai_response(

FILE: src/llama_stack/providers/inline/agents/builtin/config.py
  class AgentPersistenceConfig (line 15) | class AgentPersistenceConfig(BaseModel):
  class BuiltinAgentsImplConfig (line 22) | class BuiltinAgentsImplConfig(BaseModel):
    method sample_run_config (line 30) | def sample_run_config(cls, __distro_dir__: str) -> dict[str, Any]:

FILE: src/llama_stack/providers/inline/agents/builtin/responses/openai_responses.py
  class _BackgroundWorkItem (line 93) | class _BackgroundWorkItem:
  class OpenAIResponsePreviousResponseWithInputItems (line 100) | class OpenAIResponsePreviousResponseWithInputItems(BaseModel):
  class OpenAIResponsesImpl (line 105) | class OpenAIResponsesImpl:
    method __init__ (line 106) | def __init__(
    method initialize (line 139) | async def initialize(self) -> None:
    method _ensure_workers_started (line 148) | async def _ensure_workers_started(self) -> None:
    method shutdown (line 155) | async def shutdown(self) -> None:
    method _background_worker (line 161) | async def _background_worker(self) -> None:
    method _prepend_previous_response (line 208) | async def _prepend_previous_response(
    method _process_input_with_previous_response (line 224) | async def _process_input_with_previous_response(
    method _prepend_prompt (line 298) | async def _prepend_prompt(
    method get_openai_response (line 370) | async def get_openai_response(
    method list_openai_responses (line 377) | async def list_openai_responses(
    method list_openai_response_input_items (line 386) | async def list_openai_response_input_items(
    method _store_response (line 407) | async def _store_response(
    method _prepare_input_items_for_storage (line 444) | def _prepare_input_items_for_storage(
    method _persist_streaming_state (line 476) | async def _persist_streaming_state(
    method create_openai_response (line 563) | async def create_openai_response(
    method _create_background_response (line 766) | async def _create_background_response(
    method _run_background_response_loop (line 878) | async def _run_background_response_loop(
    method _create_streaming_response (line 977) | async def _create_streaming_response(
    method delete_openai_response (line 1134) | async def delete_openai_response(self, response_id: str) -> OpenAIDele...
    method _sync_response_to_conversation (line 1137) | async def _sync_response_to_conversation(

FILE: src/llama_stack/providers/inline/agents/builtin/responses/streaming.py
  function extract_openai_error (line 148) | def extract_openai_error(exc: Exception) -> tuple[str, str]:
  function convert_tooldef_to_chat_tool (line 192) | def convert_tooldef_to_chat_tool(tool_def):
  class StreamingResponseOrchestrator (line 208) | class StreamingResponseOrchestrator:
    method __init__ (line 209) | def __init__(
    method _create_refusal_response (line 297) | async def _create_refusal_response(self, violation_message: str) -> Op...
    method _clone_outputs (line 330) | def _clone_outputs(self, outputs: list[OpenAIResponseOutput]) -> list[...
    method _snapshot_response (line 339) | def _snapshot_response(
    method create_response (line 382) | async def create_response(self) -> AsyncIterator[OpenAIResponseObjectS...
    method _separate_tool_calls (line 674) | def _separate_tool_calls(self, current_response, messages) -> tuple[li...
    method _accumulate_chunk_usage (line 731) | def _accumulate_chunk_usage(self, chunk: OpenAIChatCompletionChunk) ->...
    method _handle_reasoning_content_chunk (line 776) | async def _handle_reasoning_content_chunk(
    method _handle_refusal_content_chunk (line 807) | async def _handle_refusal_content_chunk(
    method _emit_reasoning_done_events (line 838) | async def _emit_reasoning_done_events(
    method _emit_refusal_done_events (line 868) | async def _emit_refusal_done_events(
    method _process_streaming_chunks (line 898) | async def _process_streaming_chunks(
    method _build_chat_completion (line 1229) | def _build_chat_completion(self, result: ChatCompletionResult) -> Open...
    method _coordinate_tool_execution (line 1255) | async def _coordinate_tool_execution(
    method _process_new_tools (line 1391) | async def _process_new_tools(
    method _process_mcp_tool (line 1442) | async def _process_mcp_tool(
    method _process_tools (line 1536) | async def _process_tools(
    method _approval_required (line 1552) | def _approval_required(self, tool_name: str) -> bool:
    method _add_mcp_approval_request (line 1567) | async def _add_mcp_approval_request(
    method _add_mcp_list_tools (line 1594) | async def _add_mcp_list_tools(
    method _reuse_mcp_list_tools (line 1627) | async def _reuse_mcp_list_tools(
  function _process_tool_choice (line 1650) | async def _process_tool_choice(
  function resolve_mcp_connector_id (line 1759) | async def resolve_mcp_connector_id(

FILE: src/llama_stack/providers/inline/agents/builtin/responses/tool_executor.py
  class ToolExecutor (line 51) | class ToolExecutor:
    method __init__ (line 52) | def __init__(
    method execute_tool_call (line 67) | async def execute_tool_call(
    method _execute_file_search_via_vector_store (line 126) | async def _execute_file_search_via_vector_store(
    method _emit_progress_events (line 247) | async def _emit_progress_events(
    method _execute_tool (line 313) | async def _execute_tool(
    method _emit_completion_events (line 380) | async def _emit_completion_events(
    method _build_result_messages (line 420) | async def _build_result_messages(

FILE: src/llama_stack/providers/inline/agents/builtin/responses/types.py
  function _json_equal (line 38) | def _json_equal(a: str, b: str) -> bool:
  class ToolExecutionResult (line 46) | class ToolExecutionResult(BaseModel):
  class ChatCompletionResult (line 57) | class ChatCompletionResult:
    method content_text (line 73) | def content_text(self) -> str:
    method has_tool_calls (line 78) | def has_tool_calls(self) -> bool:
  class ToolContext (line 83) | class ToolContext(BaseModel):
    method __init__ (line 97) | def __init__(
    method recover_tools_from_previous_response (line 108) | def recover_tools_from_previous_response(
    method available_tools (line 147) | def available_tools(self) -> list[OpenAIResponseTool]:
  class ChatCompletionContext (line 169) | class ChatCompletionContext(BaseModel):
    method __init__ (line 184) | def __init__(
    method approval_response (line 216) | def approval_response(self, tool_name: str, arguments: str) -> OpenAIR...
    method _approval_request (line 220) | def _approval_request(self, tool_name: str, arguments: str) -> OpenAIR...
    method available_tools (line 226) | def available_tools(self) -> list[OpenAIResponseTool]:

FILE: src/llama_stack/providers/inline/agents/builtin/responses/utils.py
  function extract_bytes_from_file (line 63) | async def extract_bytes_from_file(file_id: str, files_api: Files) -> bytes:
  function generate_base64_ascii_text_from_bytes (line 79) | def generate_base64_ascii_text_from_bytes(raw_bytes: bytes) -> str:
  function construct_data_url (line 89) | def construct_data_url(ascii_text: str, mime_type: str | None) -> str:
  function convert_chat_choice_to_response_message (line 103) | async def convert_chat_choice_to_response_message(
  function _build_tool_result_messages (line 129) | async def _build_tool_result_messages(
  function _file_part_to_text_part (line 167) | def _file_part_to_text_part(part: OpenAIFile) -> OpenAIChatCompletionCon...
  function convert_response_content_to_chat_content (line 185) | async def convert_response_content_to_chat_content(
  function convert_response_input_to_chat_messages (line 281) | async def convert_response_input_to_chat_messages(
  function _extract_tool_call_ids (line 396) | def _extract_tool_call_ids(messages: list[OpenAIMessageParam]) -> set[str]:
  function convert_response_text_to_chat_response_format (line 409) | async def convert_response_text_to_chat_response_format(
  function get_message_type_by_role (line 429) | async def get_message_type_by_role(role: str) -> type[OpenAIMessageParam...
  function _extract_citations_from_text (line 440) | def _extract_citations_from_text(
  function is_function_tool_call (line 487) | def is_function_tool_call(
  function run_guardrails (line 499) | async def run_guardrails(safety_api: Safety | None, messages: str, guard...
  function extract_guardrail_ids (line 546) | def extract_guardrail_ids(guardrails: list | None) -> list[str]:
  function convert_mcp_tool_choice (line 563) | def convert_mcp_tool_choice(

FILE: src/llama_stack/providers/inline/agents/builtin/safety.py
  class SafetyException (line 15) | class SafetyException(Exception):  # noqa: N818
    method __init__ (line 16) | def __init__(self, violation: SafetyViolation):
  class ShieldRunnerMixin (line 21) | class ShieldRunnerMixin:
    method __init__ (line 22) | def __init__(
    method run_multiple_shields (line 32) | async def run_multiple_shields(self, messages: list[OpenAIMessageParam...

FILE: src/llama_stack/providers/inline/batches/reference/__init__.py
  function get_provider_impl (line 19) | async def get_provider_impl(config: ReferenceBatchesImplConfig, deps: di...

FILE: src/llama_stack/providers/inline/batches/reference/batches.py
  class AsyncBytesIO (line 61) | class AsyncBytesIO:
    method __init__ (line 69) | def __init__(self, data: bytes):
    method read (line 72) | async def read(self, n=-1):
    method seek (line 75) | async def seek(self, pos, whence=0):
    method __enter__ (line 78) | def __enter__(self):
    method __exit__ (line 81) | def __exit__(self, exc_type, exc_val, exc_tb):
    method __getattr__ (line 84) | def __getattr__(self, name):
  class BatchRequest (line 88) | class BatchRequest(BaseModel):
  function convert_to_openai_message_param (line 96) | def convert_to_openai_message_param(msg: dict[str, Any]) -> OpenAIMessag...
  class ReferenceBatchesImpl (line 114) | class ReferenceBatchesImpl(Batches):
    method __init__ (line 121) | def __init__(
    method initialize (line 141) | async def initialize(self) -> None:
    method shutdown (line 145) | async def shutdown(self) -> None:
    method create_batch (line 153) | async def create_batch(
    method cancel_batch (line 258) | async def cancel_batch(self, request: CancelBatchRequest) -> BatchObject:
    method list_batches (line 276) | async def list_batches(
    method retrieve_batch (line 314) | async def retrieve_batch(self, request: RetrieveBatchRequest) -> Batch...
    method _update_batch (line 322) | async def _update_batch(self, batch_id: str, **updates) -> None:
    method _validate_input (line 345) | async def _validate_input(self, batch: BatchObject) -> tuple[list[Batc...
    method _process_batch (line 524) | async def _process_batch(self, batch_id: str) -> None:
    method _process_batch_impl (line 545) | async def _process_batch_impl(self, batch_id: str) -> None:
    method _process_single_request (line 620) | async def _process_single_request(self, batch_id: str, request: BatchR...
    method _create_output_file (line 683) | async def _create_output_file(self, batch_id: str, results: list[dict]...

FILE: src/llama_stack/providers/inline/batches/reference/config.py
  class ReferenceBatchesImplConfig (line 12) | class ReferenceBatchesImplConfig(BaseModel):
    method sample_run_config (line 34) | def sample_run_config(cls, __distro_dir__: str) -> dict:

FILE: src/llama_stack/providers/inline/datasetio/localfs/__init__.py
  function get_provider_impl (line 12) | async def get_provider_impl(

FILE: src/llama_stack/providers/inline/datasetio/localfs/config.py
  class LocalFSDatasetIOConfig (line 13) | class LocalFSDatasetIOConfig(BaseModel):
    method sample_run_config (line 17) | def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict...

FILE: src/llama_stack/providers/inline/datasetio/localfs/datasetio.py
  class PandasDataframeDataset (line 18) | class PandasDataframeDataset:
    method __init__ (line 19) | def __init__(self, dataset_def: Dataset, *args, **kwargs) -> None:
    method __len__ (line 24) | def __len__(self) -> int:
    method __getitem__ (line 28) | def __getitem__(self, idx):
    method load (line 35) | async def load(self) -> None:
  class LocalFSDatasetIOImpl (line 52) | class LocalFSDatasetIOImpl(DatasetIO, DatasetsProtocolPrivate):
    method __init__ (line 53) | def __init__(self, config: LocalFSDatasetIOConfig) -> None:
    method initialize (line 59) | async def initialize(self) -> None:
    method shutdown (line 70) | async def shutdown(self) -> None: ...
    method register_dataset (line 72) | async def register_dataset(
    method unregister_dataset (line 84) | async def unregister_dataset(self, dataset_id: str) -> None:
    method iterrows (line 89) | async def iterrows(
    method append_rows (line 102) | async def append_rows(self, dataset_id: str, rows: list[dict[str, Any]...

FILE: src/llama_stack/providers/inline/eval/builtin/__init__.py
  function get_provider_impl (line 13) | async def get_provider_impl(

FILE: src/llama_stack/providers/inline/eval/builtin/config.py
  class BuiltinEvalConfig (line 13) | class BuiltinEvalConfig(BaseModel):
    method sample_run_config (line 17) | def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict...

FILE: src/llama_stack/providers/inline/eval/builtin/eval.py
  class BuiltinEvalImpl (line 43) | class BuiltinEvalImpl(
    method __init__ (line 47) | def __init__(
    method initialize (line 68) | async def initialize(self) -> None:
    method shutdown (line 79) | async def shutdown(self) -> None: ...
    method register_benchmark (line 81) | async def register_benchmark(self, task_def: Benchmark) -> None:
    method unregister_benchmark (line 90) | async def unregister_benchmark(self, benchmark_id: str) -> None:
    method run_eval (line 97) | async def run_eval(
    method _run_model_generation (line 128) | async def _run_model_generation(
    method evaluate_rows (line 174) | async def evaluate_rows(
    method job_status (line 206) | async def job_status(self, request: JobStatusRequest) -> Job:
    method job_cancel (line 212) | async def job_cancel(self, request: JobCancelRequest) -> None:
    method job_result (line 215) | async def job_result(self, request: JobResultRequest) -> EvaluateRespo...

FILE: src/llama_stack/providers/inline/file_processor/pypdf/__init__.py
  function get_provider_impl (line 14) | async def get_provider_impl(config: PyPDFFileProcessorConfig, deps: dict...

FILE: src/llama_stack/providers/inline/file_processor/pypdf/adapter.py
  class PyPDFFileProcessorAdapter (line 16) | class PyPDFFileProcessorAdapter:
    method __init__ (line 19) | def __init__(self, config: PyPDFFileProcessorConfig, files_api=None) -...
    method process_file (line 24) | async def process_file(
    method shutdown (line 37) | async def shutdown(self) -> None:

FILE: src/llama_stack/providers/inline/file_processor/pypdf/config.py
  class PyPDFFileProcessorConfig (line 14) | class PyPDFFileProcessorConfig(BaseModel):
    method sample_run_config (line 46) | def sample_run_config(cls, **kwargs: Any) -> dict[str, Any]:

FILE: src/llama_stack/providers/inline/file_processor/pypdf/pypdf.py
  class PyPDFFileProcessor (line 33) | class PyPDFFileProcessor:
    method __init__ (line 36) | def __init__(self, config: PyPDFFileProcessorConfig, files_api=None) -...
    method process_file (line 40) | async def process_file(
    method _extract_pdf_text (line 132) | def _extract_pdf_text(self, reader: PdfReader) -> tuple[str, list[str]]:
    method _extract_pdf_metadata (line 148) | def _extract_pdf_metadata(self, reader: PdfReader) -> dict[str, Any]:
    method _clean_text (line 172) | def _clean_text(self, text: str) -> str:
    method _create_chunks (line 186) | def _create_chunks(

FILE: src/llama_stack/providers/inline/files/localfs/__init__.py
  function get_provider_impl (line 17) | async def get_provider_impl(config: LocalfsFilesImplConfig, deps: dict[A...

FILE: src/llama_stack/providers/inline/files/localfs/config.py
  class LocalfsFilesImplConfig (line 14) | class LocalfsFilesImplConfig(BaseModel):
    method sample_run_config (line 24) | def sample_run_config(cls, __distro_dir__: str) -> dict[str, Any]:

FILE: src/llama_stack/providers/inline/files/localfs/files.py
  class LocalfsFilesImpl (line 60) | class LocalfsFilesImpl(Files):
    method __init__ (line 61) | def __init__(self, config: LocalfsFilesImplConfig, policy: list[Access...
    method initialize (line 66) | async def initialize(self) -> None:
    method shutdown (line 87) | async def shutdown(self) -> None:
    method _generate_file_id (line 92) | def _generate_file_id(self) -> str:
    method _validate_file_id (line 96) | def _validate_file_id(self, file_id: str) -> None:
    method _validate_path_containment (line 103) | def _validate_path_containment(self, file_path: Path) -> Path:
    method _get_file_path (line 120) | def _get_file_path(self, file_id: str) -> Path:
    method _lookup_file_id (line 126) | async def _lookup_file_id(self, file_id: str, action: Action = Action....
    method openai_upload_file (line 142) | async def openai_upload_file(
    method openai_list_files (line 194) | async def openai_list_files(
    method openai_retrieve_file (line 241) | async def openai_retrieve_file(self, request: RetrieveFileRequest) -> ...
    method openai_delete_file (line 247) | async def openai_delete_file(self, request: DeleteFileRequest) -> Open...
    method openai_retrieve_file_content (line 264) | async def openai_retrieve_file_content(self, request: RetrieveFileCont...

FILE: src/llama_stack/providers/inline/inference/sentence_transformers/__init__.py
  function get_provider_impl (line 14) | async def get_provider_impl(

FILE: src/llama_stack/providers/inline/inference/sentence_transformers/config.py
  class SentenceTransformersInferenceConfig (line 12) | class SentenceTransformersInferenceConfig(BaseModel):
    method sample_run_config (line 14) | def sample_run_config(cls, **kwargs) -> dict[str, Any]:

FILE: src/llama_stack/providers/inline/inference/sentence_transformers/sentence_transformers.py
  class SentenceTransformersInferenceImpl (line 30) | class SentenceTransformersInferenceImpl(
    method __init__ (line 37) | def __init__(self, config: SentenceTransformersInferenceConfig) -> None:
    method initialize (line 40) | async def initialize(self) -> None:
    method shutdown (line 43) | async def shutdown(self) -> None:
    method should_refresh_models (line 46) | async def should_refresh_models(self) -> bool:
    method list_models (line 49) | async def list_models(self) -> list[Model] | None:
    method register_model (line 62) | async def register_model(self, model: Model) -> Model:
    method unregister_model (line 65) | async def unregister_model(self, model_id: str) -> None:
    method openai_completion (line 68) | async def openai_completion(
    method openai_chat_completion (line 74) | async def openai_chat_completion(

FILE: src/llama_stack/providers/inline/inference/transformers/__init__.py
  function get_provider_impl (line 14) | async def get_provider_impl(

FILE: src/llama_stack/providers/inline/inference/transformers/config.py
  class TransformersInferenceConfig (line 12) | class TransformersInferenceConfig(BaseModel):
    method sample_run_config (line 14) | def sample_run_config(cls, **kwargs) -> dict[str, Any]:

FILE: src/llama_stack/providers/inline/inference/transformers/transformers.py
  class TransformersInferenceImpl (line 48) | class TransformersInferenceImpl(
    method __init__ (line 54) | def __init__(self, config: TransformersInferenceConfig) -> None:
    method initialize (line 57) | async def initialize(self) -> None:
    method shutdown (line 60) | async def shutdown(self) -> None:
    method should_refresh_models (line 63) | async def should_refresh_models(self) -> bool:
    method list_models (line 66) | async def list_models(self) -> list[Model] | None:
    method register_model (line 76) | async def register_model(self, model: Model) -> Model:
    method unregister_model (line 79) | async def unregister_model(self, model_id: str) -> None:
    method openai_completion (line 82) | async def openai_completion(
    method openai_chat_completion (line 88) | async def openai_chat_completion(
    method openai_embeddings (line 94) | async def openai_embeddings(
    method rerank (line 100) | async def rerank(
    method load_reranker_model (line 138) | async def load_reranker_model(self, model: str) -> tuple[Any, Any]:
    method compute_reranked_scores (line 171) | def compute_reranked_scores(
    method extract_text (line 222) | def extract_text(
    method format_instruction (line 232) | def format_instruction(self, instruction: str, query: str, document: s...

FILE: src/llama_stack/providers/inline/safety/code_scanner/__init__.py
  function get_provider_impl (line 12) | async def get_provider_impl(config: CodeScannerConfig, deps: dict[str, A...

FILE: src/llama_stack/providers/inline/safety/code_scanner/code_scanner.py
  class BuiltinCodeScannerSafetyImpl (line 40) | class BuiltinCodeScannerSafetyImpl(Safety):
    method __init__ (line 41) | def __init__(self, config: CodeScannerConfig, deps) -> None:
    method initialize (line 44) | async def initialize(self) -> None:
    method shutdown (line 47) | async def shutdown(self) -> None:
    method register_shield (line 50) | async def register_shield(self, shield: Shield) -> None:
    method run_shield (line 56) | async def run_shield(self, request: RunShieldRequest) -> RunShieldResp...
    method get_moderation_object_results (line 76) | def get_moderation_object_results(self, scan_result: "CodeShieldScanRe...
    method run_moderation (line 102) | async def run_moderation(self, request: RunModerationRequest) -> Moder...

FILE: src/llama_stack/providers/inline/safety/code_scanner/config.py
  class CodeScannerConfig (line 12) | class CodeScannerConfig(BaseModel):
    method sample_run_config (line 14) | def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict...

FILE: src/llama_stack/providers/inline/safety/llama_guard/__init__.py
  function get_provider_impl (line 12) | async def get_provider_impl(config: LlamaGuardConfig, deps: dict[str, An...

FILE: src/llama_stack/providers/inline/safety/llama_guard/config.py
  class LlamaGuardConfig (line 12) | class LlamaGuardConfig(BaseModel):
    method sample_run_config (line 16) | def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict...

FILE: src/llama_stack/providers/inline/safety/llama_guard/llama_guard.py
  class LlamaGuardSafetyImpl (line 143) | class LlamaGuardSafetyImpl(Safety, ShieldsProtocolPrivate):
    method __init__ (line 144) | def __init__(self, config: LlamaGuardConfig, deps) -> None:
    method initialize (line 148) | async def initialize(self) -> None:
    method shutdown (line 151) | async def shutdown(self) -> None:
    method register_shield (line 154) | async def register_shield(self, shield: Shield) -> None:
    method unregister_shield (line 159) | async def unregister_shield(self, identifier: str) -> None:
    method run_shield (line 164) | async def run_shield(self, request: RunShieldRequest) -> RunShieldResp...
    method run_moderation (line 198) | async def run_moderation(self, request: RunModerationRequest) -> Moder...
  class LlamaGuardShield (line 230) | class LlamaGuardShield:
    method __init__ (line 231) | def __init__(
    method check_unsafe_response (line 252) | def check_unsafe_response(self, response: str) -> str | None:
    method get_safety_categories (line 264) | def get_safety_categories(self) -> list[str]:
    method validate_messages (line 280) | def validate_messages(self, messages: list[OpenAIMessageParam]) -> lis...
    method run (line 291) | async def run(self, messages: list[OpenAIMessageParam]) -> RunShieldRe...
    method build_text_shield_input (line 310) | def build_text_shield_input(self, messages: list[OpenAIMessageParam]) ...
    method build_vision_shield_input (line 313) | def build_vision_shield_input(self, messages: list[OpenAIMessageParam]...
    method build_prompt (line 347) | def build_prompt(self, messages: list[OpenAIMessageParam]) -> str:
    method get_shield_response (line 359) | def get_shield_response(self, response: str) -> RunShieldResponse:
    method run_moderation (line 380) | async def run_moderation(self, messages: list[OpenAIMessageParam]) -> ...
    method create_moderation_object (line 398) | def create_moderation_object(self, model: str, unsafe_code: str | None...
    method is_content_safe (line 467) | def is_content_safe(self, response: str, unsafe_code: str | None = Non...
    method get_moderation_object (line 479) | def get_moderation_object(self, response: str) -> ModerationObject:

FILE: src/llama_stack/providers/inline/safety/prompt_guard/__init__.py
  function get_provider_impl (line 12) | async def get_provider_impl(config: PromptGuardConfig, deps: dict[str, A...

FILE: src/llama_stack/providers/inline/safety/prompt_guard/config.py
  class PromptGuardType (line 13) | class PromptGuardType(Enum):
  class PromptGuardConfig (line 18) | class PromptGuardConfig(BaseModel):
    method validate_guard_type (line 23) | def validate_guard_type(cls, v):
    method sample_run_config (line 29) | def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict...

FILE: src/llama_stack/providers/inline/safety/prompt_guard/prompt_guard.py
  class PromptGuardSafetyImpl (line 35) | class PromptGuardSafetyImpl(ShieldToModerationMixin, Safety, ShieldsProt...
    method __init__ (line 38) | def __init__(self, config: PromptGuardConfig, _deps) -> None:
    method initialize (line 41) | async def initialize(self) -> None:
    method shutdown (line 51) | async def shutdown(self) -> None:
    method register_shield (line 54) | async def register_shield(self, shield: Shield) -> None:
    method unregister_shield (line 58) | async def unregister_shield(self, identifier: str) -> None:
    method run_shield (line 61) | async def run_shield(self, request: RunShieldRequest) -> RunShieldResp...
  class PromptGuardShield (line 69) | class PromptGuardShield:
    method __init__ (line 70) | def __init__(
    method run (line 97) | async def run(self, messages: list[OpenAIMessageParam]) -> RunShieldRe...

FILE: src/llama_stack/providers/inline/scoring/basic/__init__.py
  function get_provider_impl (line 13) | async def get_provider_impl(

FILE: src/llama_stack/providers/inline/scoring/basic/config.py
  class BasicScoringConfig (line 11) | class BasicScoringConfig(BaseModel):
    method sample_run_config (line 13) | def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict...

FILE: src/llama_stack/providers/inline/scoring/basic/scoring.py
  class BasicScoringImpl (line 41) | class BasicScoringImpl(
    method __init__ (line 45) | def __init__(
    method initialize (line 56) | async def initialize(self) -> None:
    method shutdown (line 62) | async def shutdown(self) -> None: ...
    method list_scoring_functions (line 64) | async def list_scoring_functions(self) -> list[ScoringFn]:
    method register_scoring_function (line 74) | async def register_scoring_function(self, function_def: ScoringFn) -> ...
    method score_batch (line 77) | async def score_batch(
    method score (line 96) | async def score(

FILE: src/llama_stack/providers/inline/scoring/basic/scoring_fn/docvqa_scoring_fn.py
  function normalize_answer (line 191) | def normalize_answer(s: str) -> str:
  class DocVQAScoringFn (line 215) | class DocVQAScoringFn(RegisteredBaseScoringFn):
    method __init__ (line 222) | def __init__(self, *args, **kwargs) -> None:
    method score_row (line 228) | async def score_row(

FILE: src/llama_stack/providers/inline/scoring/basic/scoring_fn/equality_scoring_fn.py
  class EqualityScoringFn (line 15) | class EqualityScoringFn(RegisteredBaseScoringFn):
    method __init__ (line 20) | def __init__(self, *args, **kwargs) -> None:
    method score_row (line 26) | async def score_row(

FILE: src/llama_stack/providers/inline/scoring/basic/scoring_fn/ifeval_scoring_fn.py
  class IfEvalScoringFn (line 17) | class IfEvalScoringFn(RegisteredBaseScoringFn):
    method __init__ (line 22) | def __init__(self, *args, **kwargs) -> None:
    method score_row (line 28) | async def score_row(

FILE: src/llama_stack/providers/inline/scoring/basic/scoring_fn/regex_parser_math_response_scoring_fn.py
  class RegexParserMathResponseScoringFn (line 17) | class RegexParserMathResponseScoringFn(RegisteredBaseScoringFn):
    method __init__ (line 22) | def __init__(self, *args, **kwargs) -> None:
    method score_row (line 28) | async def score_row(

FILE: src/llama_stack/providers/inline/scoring/basic/scoring_fn/regex_parser_scoring_fn.py
  class RegexParserScoringFn (line 17) | class RegexParserScoringFn(RegisteredBaseScoringFn):
    method __init__ (line 22) | def __init__(self, *args, **kwargs) -> None:
    method score_row (line 28) | async def score_row(

FILE: src/llama_stack/providers/inline/scoring/basic/scoring_fn/subset_of_scoring_fn.py
  class SubsetOfScoringFn (line 15) | class SubsetOfScoringFn(RegisteredBaseScoringFn):
    method __init__ (line 20) | def __init__(self, *args, **kwargs) -> None:
    method score_row (line 26) | async def score_row(

FILE: src/llama_stack/providers/inline/scoring/basic/utils/ifeval_utils.py
  function split_into_sentences (line 1609) | def split_into_sentences(text):
  function count_words (line 1660) | def count_words(text):
  function split_chinese_japanese_hindi (line 1668) | def split_chinese_japanese_hindi(lines: str) -> Iterable[str]:
  function count_words_cjk (line 1684) | def count_words_cjk(text: str) -> int:
  function _get_sentence_tokenizer (line 1711) | def _get_sentence_tokenizer():
  function count_sentences (line 1715) | def count_sentences(text):
  function get_langid (line 1722) | def get_langid(text: str, lid_path: str | None = None) -> str:
  function generate_keywords (line 1738) | def generate_keywords(num_keywords):
  class Instruction (line 1822) | class Instruction:
    method __init__ (line 1825) | def __init__(self, instruction_id):
    method build_description (line 1828) | def build_description(self, **kwargs):
    method get_instruction_args (line 1831) | def get_instruction_args(self):
    method get_instruction_args_keys (line 1834) | def get_instruction_args_keys(self):
    method check_following (line 1837) | def check_following(self, value):
  class ResponseLanguageChecker (line 1841) | class ResponseLanguageChecker(Instruction):
    method build_description (line 1844) | def build_description(self, *, language=None):
    method get_instruction_args (line 1866) | def get_instruction_args(self):
    method get_instruction_args_keys (line 1870) | def get_instruction_args_keys(self):
    method check_following (line 1874) | def check_following(self, value):
  class NumberOfSentences (line 1893) | class NumberOfSentences(Instruction):
    method build_description (line 1896) | def build_description(self, *, num_sentences=None, relation=None):
    method get_instruction_args (line 1931) | def get_instruction_args(self):
    method get_instruction_args_keys (line 1938) | def get_instruction_args_keys(self):
    method check_following (line 1942) | def check_following(self, value):
  class PlaceholderChecker (line 1969) | class PlaceholderChecker(Instruction):
    method build_description (line 1972) | def build_description(self, *, num_placeholders=None):
    method get_instruction_args (line 1991) | def get_instruction_args(self):
    method get_instruction_args_keys (line 1995) | def get_instruction_args_keys(self):
    method check_following (line 1999) | def check_following(self, value):
  class BulletListChecker (line 2014) | class BulletListChecker(Instruction):
    method build_description (line 2017) | def build_description(self, *, num_bullets=None):
    method get_instruction_args (line 2038) | def get_instruction_args(self):
    method get_instruction_args_keys (line 2042) | def get_instruction_args_keys(self):
    method check_following (line 2046) | def check_following(self, value):
  class ConstrainedResponseChecker (line 2063) | class ConstrainedResponseChecker(Instruction):
    method build_description (line 2066) | def build_description(self):
    method get_instruction_args (line 2073) | def get_instruction_args(self):
    method get_instruction_args_keys (line 2077) | def get_instruction_args_keys(self):
    method check_following (line 2081) | def check_following(self, value):
  class ConstrainedStartChecker (line 2098) | class ConstrainedStartChecker(Instruction):
    method build_description (line 2101) | def build_description(self, *, starter=None):
    method get_instruction_args (line 2119) | def get_instruction_args(self):
    method get_instruction_args_keys (line 2123) | def get_instruction_args_keys(self):
    method check_following (line 2127) | def check_following(self, value):
  class HighlightSectionChecker (line 2142) | class HighlightSectionChecker(Instruction):
    method build_description (line 2145) | def build_description(self, *, num_highlights=None):
    method get_instruction_args (line 2166) | def get_instruction_args(self):
    method get_instruction_args_keys (line 2170) | def get_instruction_args_keys(self):
    method check_following (line 2174) | def check_following(self, value):
  class SectionChecker (line 2198) | class SectionChecker(Instruction):
    method build_description (line 2201) | def build_description(self, *, section_spliter=None, num_sections=None):
    method get_instruction_args (line 2231) | def get_instruction_args(self):
    method get_instruction_args_keys (line 2238) | def get_instruction_args_keys(self):
    method check_following (line 2242) | def check_following(self, value):
  class ParagraphChecker (line 2261) | class ParagraphChecker(Instruction):
    method build_description (line 2264) | def build_description(self, *, num_paragraphs=None):
    method get_instruction_args (line 2283) | def get_instruction_args(self):
    method get_instruction_args_keys (line 2287) | def get_instruction_args_keys(self):
    method check_following (line 2291) | def check_following(self, value):
  class PostscriptChecker (line 2315) | class PostscriptChecker(Instruction):
    method build_description (line 2318) | def build_description(self, *, postscript_marker=None):
    method get_instruction_args (line 2338) | def get_instruction_args(self):
    method get_instruction_args_keys (line 2342) | def get_instruction_args_keys(self):
    method check_following (line 2346) | def check_following(self, value):
  class RephraseChecker (line 2368) | class RephraseChecker(Instruction):
    method build_description (line 2371) | def build_description(self, *, original_message):
    method get_instruction_args (line 2394) | def get_instruction_args(self):
    method get_instruction_args_keys (line 2398) | def get_instruction_args_keys(self):
    method check_following (line 2402) | def check_following(self, value):
    method is_change (line 2422) | def is_change(self, response):
    method strip_changes (line 2426) | def strip_changes(self, response):
  class KeywordChecker (line 2431) | class KeywordChecker(Instruction):
    method build_description (line 2434) | def build_description(self, *, keywords=None):
    method get_instruction_args (line 2455) | def get_instruction_args(self):
    method get_instruction_args_keys (line 2459) | def get_instruction_args_keys(self):
    method check_following (line 2463) | def check_following(self, value):
  class KeywordFrequencyChecker (line 2471) | class KeywordFrequencyChecker(Instruction):
    method build_description (line 2474) | def build_description(self, *, keyword=None, frequency=None, relation=...
    method get_instruction_args (line 2518) | def get_instruction_args(self):
    method get_instruction_args_keys (line 2526) | def get_instruction_args_keys(self):
    method check_following (line 2530) | def check_following(self, value):
  class NumberOfWords (line 2540) | class NumberOfWords(Instruction):
    method build_description (line 2543) | def build_description(self, *, num_words=None, relation=None):
    method get_instruction_args (line 2576) | def get_instruction_args(self):
    method get_instruction_args_keys (line 2580) | def get_instruction_args_keys(self):
    method check_following (line 2584) | def check_following(self, value):
  class JsonFormat (line 2600) | class JsonFormat(Instruction):
    method build_description (line 2603) | def build_description(self):
    method get_instruction_args (line 2609) | def get_instruction_args(self):
    method get_instruction_args_keys (line 2613) | def get_instruction_args_keys(self):
    method check_following (line 2617) | def check_following(self, value):
  class ParagraphFirstWordCheck (line 2634) | class ParagraphFirstWordCheck(Instruction):
    method build_description (line 2637) | def build_description(self, num_paragraphs=None, nth_paragraph=None, f...
    method get_instruction_args (line 2677) | def get_instruction_args(self):
    method get_instruction_args_keys (line 2685) | def get_instruction_args_keys(self):
    method check_following (line 2689) | def check_following(self, value):
  class KeySentenceChecker (line 2733) | class KeySentenceChecker(Instruction):
    method build_description (line 2736) | def build_description(self, key_sentences=None, num_sentences=None):
    method get_instruction_args (line 2763) | def get_instruction_args(self):
    method get_instruction_args_keys (line 2770) | def get_instruction_args_keys(self):
    method check_following (line 2774) | def check_following(self, value):
  class ForbiddenWords (line 2785) | class ForbiddenWords(Instruction):
    method build_description (line 2788) | def build_description(self, forbidden_words=None):
    method get_instruction_args (line 2808) | def get_instruction_args(self):
    method get_instruction_args_keys (line 2812) | def get_instruction_args_keys(self):
    method check_following (line 2816) | def check_following(self, value):
  class RephraseParagraph (line 2824) | class RephraseParagraph(Instruction):
    method build_description (line 2827) | def build_description(self, *, original_paragraph, low, high):
    method get_instruction_args (line 2855) | def get_instruction_args(self):
    method get_instruction_args_keys (line 2863) | def get_instruction_args_keys(self):
    method check_following (line 2867) | def check_following(self, value):
  class TwoResponsesChecker (line 2881) | class TwoResponsesChecker(Instruction):
    method build_description (line 2884) | def build_description(self):
    method get_instruction_args (line 2892) | def get_instruction_args(self):
    method get_instruction_args_keys (line 2896) | def get_instruction_args_keys(self):
    method check_following (line 2900) | def check_following(self, value):
  class RepeatPromptThenAnswer (line 2920) | class RepeatPromptThenAnswer(Instruction):
    method build_description (line 2923) | def build_description(self, *, prompt_to_repeat=None):
    method get_instruction_args (line 2944) | def get_instruction_args(self):
    method get_instruction_args_keys (line 2947) | def get_instruction_args_keys(self):
    method check_following (line 2951) | def check_following(self, value):
  class EndChecker (line 2957) | class EndChecker(Instruction):
    method build_description (line 2960) | def build_description(self, *, end_phrase=None):
    method get_instruction_args (line 2977) | def get_instruction_args(self):
    method get_instruction_args_keys (line 2980) | def get_instruction_args_keys(self):
    method check_following (line 2984) | def check_following(self, value):
  class TitleChecker (line 2991) | class TitleChecker(Instruction):
    method build_description (line 2994) | def build_description(self):
    method get_instruction_args (line 3001) | def get_instruction_args(self):
    method get_instruction_args_keys (line 3004) | def get_instruction_args_keys(self):
    method check_following (line 3008) | def check_following(self, value):
  class LetterFrequencyChecker (line 3020) | class LetterFrequencyChecker(Instruction):
    method build_description (line 3023) | def build_description(self, *, letter=None, let_frequency=None, let_re...
    method get_instruction_args (line 3068) | def get_instruction_args(self):
    method get_instruction_args_keys (line 3076) | def get_instruction_args_keys(self):
    method check_following (line 3080) | def check_following(self, value):
  class CapitalLettersEnglishChecker (line 3091) | class CapitalLettersEnglishChecker(Instruction):
    method build_description (line 3094) | def build_description(self):
    method get_instruction_args (line 3099) | def get_instruction_args(self):
    method get_instruction_args_keys (line 3102) | def get_instruction_args_keys(self):
    method check_following (line 3106) | def check_following(self, value):
  class LowercaseLettersEnglishChecker (line 3118) | class LowercaseLettersEnglishChecker(Instruction):
    method build_description (line 3121) | def build_description(self):
    method get_instruction_args (line 3128) | def get_instruction_args(self):
    method get_instruction_args_keys (line 3131) | def get_instruction_args_keys(self):
    method check_following (line 3135) | def check_following(self, value):
  class CommaChecker (line 3147) | class CommaChecker(Instruction):
    method build_description (line 3150) | def build_description(self, **kwargs):
    method get_instruction_args (line 3155) | def get_instruction_args(self):
    method get_instruction_args_keys (line 3158) | def get_instruction_args_keys(self):
    method check_following (line 3162) | def check_following(self, value):
  class CapitalWordFrequencyChecker (line 3167) | class CapitalWordFrequencyChecker(Instruction):
    method build_description (line 3170) | def build_description(
    method get_instruction_args (line 3205) | def get_instruction_args(self):
    method get_instruction_args_keys (line 3212) | def get_instruction_args_keys(self):
    method check_following (line 3216) | def check_following(self, value):
  class QuotationChecker (line 3231) | class QuotationChecker(Instruction):
    method build_description (line 3234) | def build_description(self):
    method get_instruction_args (line 3239) | def get_instruction_args(self):
    method get_instruction_args_keys (line 3243) | def get_instruction_args_keys(self):
    method check_following (line 3247) | def check_following(self, value):

FILE: src/llama_stack/providers/inline/scoring/basic/utils/math_utils.py
  function try_evaluate_frac (line 72) | def try_evaluate_frac(expression: str, fmt: str = "0.2e") -> str:
  function try_evaluate_latex (line 90) | def try_evaluate_latex(expression: str, fmt: str = ".2e") -> str:
  function first_answer (line 101) | def first_answer(text: str, markers: Sequence[str] = ("Q:", "A:")) -> str:
  function extract_result_from_boxed (line 107) | def extract_result_from_boxed(answer: str) -> str:
  function normalize_final_answer (line 142) | def normalize_final_answer(final_answer: str, regex_pattern: str, match_...
  function _normalise_result (line 189) | def _normalise_result(string: str) -> str:
  function _remove_right_units (line 254) | def _remove_right_units(string: str) -> str:
  function _fix_sqrt (line 267) | def _fix_sqrt(string: str) -> str:
  function _fix_fracs (line 284) | def _fix_fracs(string: str) -> str:
  function _fix_a_slash_b (line 318) | def _fix_a_slash_b(string: str) -> str:

FILE: src/llama_stack/providers/inline/scoring/braintrust/__init__.py
  class BraintrustProviderDataValidator (line 15) | class BraintrustProviderDataValidator(BaseModel):
  function get_provider_impl (line 19) | async def get_provider_impl(

FILE: src/llama_stack/providers/inline/scoring/braintrust/braintrust.py
  function _get_braintrust_evaluators (line 61) | def _get_braintrust_evaluators() -> dict[str, Any]:
  class BraintrustScoringImpl (line 101) | class BraintrustScoringImpl(
    method __init__ (line 106) | def __init__(
    method initialize (line 117) | async def initialize(self) -> None: ...
    method shutdown (line 119) | async def shutdown(self) -> None: ...
    method list_scoring_functions (line 121) | async def list_scoring_functions(self) -> list[ScoringFn]:
    method register_scoring_function (line 130) | async def register_scoring_function(self, scoring_fn: ScoringFn) -> None:
    method set_api_key (line 133) | async def set_api_key(self) -> None:
    method score_batch (line 145) | async def score_batch(
    method score_row (line 166) | async def score_row(self, input_row: dict[str, Any], scoring_fn_identi...
    method score (line 185) | async def score(

FILE: src/llama_stack/providers/inline/scoring/braintrust/config.py
  class BraintrustScoringConfig (line 11) | class BraintrustScoringConfig(BaseModel):
    method sample_run_config (line 18) | def sample_run_config(cls, **kwargs) -> dict[str, Any]:

FILE: src/llama_stack/providers/inline/scoring/llm_as_judge/__init__.py
  function get_provider_impl (line 13) | async def get_provider_impl(

FILE: src/llama_stack/providers/inline/scoring/llm_as_judge/config.py
  class LlmAsJudgeScoringConfig (line 11) | class LlmAsJudgeScoringConfig(BaseModel):
    method sample_run_config (line 13) | def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict...

FILE: src/llama_stack/providers/inline/scoring/llm_as_judge/scoring.py
  class LlmAsJudgeScoringImpl (line 28) | class LlmAsJudgeScoringImpl(
    method __init__ (line 32) | def __init__(
    method initialize (line 44) | async def initialize(self) -> None:
    method shutdown (line 48) | async def shutdown(self) -> None: ...
    method list_scoring_functions (line 50) | async def list_scoring_functions(self) -> list[ScoringFn]:
    method register_scoring_function (line 60) | async def register_scoring_function(self, function_def: ScoringFn) -> ...
    method unregister_scoring_function (line 63) | async def unregister_scoring_function(self, scoring_fn_id: str) -> None:
    method score_batch (line 66) | async def score_batch(
    method score (line 85) | async def score(

FILE: src/llama_stack/providers/inline/scoring/llm_as_judge/scoring_fn/llm_as_judge_scoring_fn.py
  class LlmAsJudgeScoringFn (line 16) | class LlmAsJudgeScoringFn(RegisteredBaseScoringFn):
    method __init__ (line 21) | def __init__(self, inference_api: Inference, *arg, **kwargs) -> None:
    method score_row (line 29) | async def score_row(

FILE: src/llama_stack/providers/inline/tool_runtime/file_search/__init__.py
  function get_provider_impl (line 14) | async def get_provider_impl(config: FileSearchToolRuntimeConfig, deps: d...

FILE: src/llama_stack/providers/inline/tool_runtime/file_search/config.py
  class FileSearchToolRuntimeConfig (line 14) | class FileSearchToolRuntimeConfig(BaseModel):
    method sample_run_config (line 21) | def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict...

FILE: src/llama_stack/providers/inline/tool_runtime/file_search/context_retriever.py
  function generate_rag_query (line 24) | async def generate_rag_query(
  function default_rag_query_generator (line 42) | async def default_rag_query_generator(
  function llm_rag_query_generator (line 50) | async def llm_rag_query_generator(

FILE: src/llama_stack/providers/inline/tool_runtime/file_search/file_search.py
  function raw_data_from_doc (line 52) | async def raw_data_from_doc(doc: RAGDocument) -> tuple[bytes, str]:
  class FileSearchToolRuntimeImpl (line 98) | class FileSearchToolRuntimeImpl(ToolGroupsProtocolPrivate, ToolRuntime):
    method __init__ (line 99) | def __init__(
    method initialize (line 111) | async def initialize(self):
    method shutdown (line 114) | async def shutdown(self):
    method register_toolgroup (line 117) | async def register_toolgroup(self, toolgroup: ToolGroup) -> None:
    method unregister_toolgroup (line 120) | async def unregister_toolgroup(self, toolgroup_id: str) -> None:
    method insert (line 123) | async def insert(
    method query (line 186) | async def query(
    method list_runtime_tools (line 302) | async def list_runtime_tools(
    method invoke_tool (line 334) | async def invoke_tool(

FILE: src/llama_stack/providers/inline/vector_io/chroma/__init__.py
  function get_provider_impl (line 14) | async def get_provider_impl(config: ChromaVectorIOConfig, deps: dict[Api...

FILE: src/llama_stack/providers/inline/vector_io/chroma/config.py
  class ChromaVectorIOConfig (line 16) | class ChromaVectorIOConfig(BaseModel):
    method sample_run_config (line 21) | def sample_run_config(

FILE: src/llama_stack/providers/inline/vector_io/faiss/__init__.py
  function get_provider_impl (line 14) | async def get_provider_impl(config: FaissVectorIOConfig, deps: dict[Api,...

FILE: src/llama_stack/providers/inline/vector_io/faiss/config.py
  class FaissVectorIOConfig (line 16) | class FaissVectorIOConfig(BaseModel):
    method sample_run_config (line 20) | def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict...

FILE: src/llama_stack/providers/inline/vector_io/faiss/faiss.py
  function _get_faiss (line 21) | def _get_faiss() -> Any:
  function _get_numpy (line 38) | def _get_numpy() -> Any:
  function _list_op (line 79) | def _list_op(mv: Any, fv: Any, *, negate: bool) -> bool:
  class FaissIndex (line 105) | class FaissIndex(EmbeddingIndex):
    method __init__ (line 106) | def __init__(self, dimension: int, kvstore: KVStore | None = None, ban...
    method create (line 122) | async def create(cls, dimension: int, kvstore: KVStore | None = None, ...
    method initialize (line 127) | async def initialize(self) -> None:
    method _save_index (line 158) | async def _save_index(self):
    method delete (line 173) | async def delete(self):
    method add_chunks (line 179) | async def add_chunks(self, embedded_chunks: list[EmbeddedChunk]):
    method delete_chunks (line 205) | async def delete_chunks(self, chunks_for_deletion: list[ChunkForDeleti...
    method _resolve_filter_positions (line 242) | def _resolve_filter_positions(self, filter_obj: Filter) -> set[int]:
    method query_vector (line 283) | async def query_vector(
    method query_keyword (line 321) | async def query_keyword(
    method query_hybrid (line 328) | async def query_hybrid(
  class FaissVectorIOAdapter (line 343) | class FaissVectorIOAdapter(OpenAIVectorStoreMixin, VectorIO, VectorStore...
    method __init__ (line 344) | def __init__(self, config: FaissVectorIOConfig, inference_api: Inferen...
    method initialize (line 349) | async def initialize(self) -> None:
    method shutdown (line 368) | async def shutdown(self) -> None:
    method health (line 372) | async def health(self) -> HealthResponse:
    method register_vector_store (line 388) | async def register_vector_store(self, vector_store: VectorStore) -> None:
    method list_vector_stores (line 402) | async def list_vector_stores(self) -> list[VectorStore]:
    method unregister_vector_store (line 405) | async def unregister_vector_store(self, vector_store_id: str) -> None:
    method _get_and_cache_vector_store_index (line 416) | async def _get_and_cache_vector_store_index(self, vector_store_id: str...
    method insert_chunks (line 437) | async def insert_chunks(self, request: InsertChunksRequest) -> None:
    method query_chunks (line 444) | async def query_chunks(self, request: QueryChunksRequest) -> QueryChun...
    method delete_chunks (line 451) | async def delete_chunks(self, request: DeleteChunksRequest) -> None:

FILE: src/llama_stack/providers/inline/vector_io/milvus/__init__.py
  function get_provider_impl (line 14) | async def get_provider_impl(config: MilvusVectorIOConfig, deps: dict[Api...

FILE: src/llama_stack/providers/inline/vector_io/milvus/config.py
  class MilvusVectorIOConfig (line 16) | class MilvusVectorIOConfig(BaseModel):
    method sample_run_config (line 22) | def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict...

FILE: src/llama_stack/providers/inline/vector_io/qdrant/__init__.py
  function get_provider_impl (line 14) | async def get_provider_impl(config: QdrantVectorIOConfig, deps: dict[Api...

FILE: src/llama_stack/providers/inline/vector_io/qdrant/config.py
  class QdrantVectorIOConfig (line 17) | class QdrantVectorIOConfig(BaseModel):
    method sample_run_config (line 22) | def sample_run_config(cls, __distro_dir__: str) -> dict[str, Any]:

FILE: src/llama_stack/providers/inline/vector_io/sqlite_vec/__init__.py
  function get_provider_impl (line 14) | async def get_provider_impl(config: SQLiteVectorIOConfig, deps: dict[Api...

FILE: src/llama_stack/providers/inline/vector_io/sqlite_vec/config.py
  class SQLiteVectorIOConfig (line 14) | class SQLiteVectorIOConfig(BaseModel):
    method sample_run_config (line 19) | def sample_run_config(cls, __distro_dir__: str) -> dict[str, Any]:

FILE: src/llama_stack/providers/inline/vector_io/sqlite_vec/sqlite_vec.py
  function _get_numpy (line 22) | def _get_numpy() -> Any:
  function _get_sqlite_vec (line 39) | def _get_sqlite_vec() -> Any:
  function serialize_vector (line 104) | def serialize_vector(vector: list[float]) -> bytes:
  function _create_sqlite_connection (line 109) | def _create_sqlite_connection(db_path: str):
  function _make_sql_identifier (line 118) | def _make_sql_identifier(name: str) -> str:
  class SQLiteVecIndex (line 122) | class SQLiteVecIndex(EmbeddingIndex):
    method __init__ (line 131) | def __init__(self, dimension: int, db_path: str, bank_id: str, kvstore...
    method create (line 141) | async def create(cls, dimension: int, db_path: str, bank_id: str):
    method initialize (line 146) | async def initialize(self) -> None:
    method delete (line 178) | async def delete(self) -> None:
    method add_chunks (line 193) | async def add_chunks(self, embedded_chunks: list[EmbeddedChunk], batch...
    method _translate_filters (line 256) | def _translate_filters(self, filters: Filter | None) -> tuple[str, lis...
    method _translate_single_filter (line 271) | def _translate_single_filter(self, filter_obj: Filter) -> tuple[str, l...
    method _translate_comparison_filter (line 280) | def _translate_comparison_filter(self, filter_obj: ComparisonFilter) -...
    method _translate_compound_filter (line 301) | def _translate_compound_filter(self, filter_obj: CompoundFilter) -> tu...
    method query_vector (line 321) | async def query_vector(
    method query_keyword (line 370) | async def query_keyword(
    method query_hybrid (line 419) | async def query_hybrid(
    method delete_chunks (line 487) | async def delete_chunks(self, chunks_for_deletion: list[ChunkForDeleti...
  class SQLiteVecVectorIOAdapter (line 519) | class SQLiteVecVectorIOAdapter(OpenAIVectorStoreMixin, VectorIO, VectorS...
    method __init__ (line 526) | def __init__(self, config, inference_api: Inference, files_api: Files ...
    method initialize (line 532) | async def initialize(self) -> None:
    method shutdown (line 548) | async def shutdown(self) -> None:
    method list_vector_stores (line 552) | async def list_vector_stores(self) -> list[VectorStore]:
    method register_vector_store (line 555) | async def register_vector_store(self, vector_store: VectorStore) -> None:
    method _get_and_cache_vector_store_index (line 569) | async def _get_and_cache_vector_store_index(self, vector_store_id: str...
    method unregister_vector_store (line 596) | async def unregister_vector_store(self, vector_store_id: str) -> None:
    method insert_chunks (line 602) | async def insert_chunks(self, request: InsertChunksRequest) -> None:
    method query_chunks (line 609) | async def query_chunks(self, request: QueryChunksRequest) -> QueryChun...
    method delete_chunks (line 616) | async def delete_chunks(self, request: DeleteChunksRequest) -> None:

FILE: src/llama_stack/providers/registry/agents.py
  function available_providers (line 16) | def available_providers() -> list[ProviderSpec]:

FILE: src/llama_stack/providers/registry/batches.py
  function available_providers (line 11) | def available_providers() -> list[ProviderSpec]:

FILE: src/llama_stack/providers/registry/datasetio.py
  function available_providers (line 16) | def available_providers() -> list[ProviderSpec]:

FILE: src/llama_stack/providers/registry/eval.py
  function available_providers (line 11) | def available_providers() -> list[ProviderSpec]:

FILE: src/llama_stack/providers/registry/file_processors.py
  function available_providers (line 14) | def available_providers() -> list[ProviderSpec]:

FILE: src/llama_stack/providers/registry/files.py
  function available_providers (line 11) | def available_providers() -> list[ProviderSpec]:

FILE: src/llama_stack/providers/registry/inference.py
  function available_providers (line 29) | def available_providers() -> list[ProviderSpec]:

FILE: src/llama_stack/providers/registry/safety.py
  function available_providers (line 16) | def available_providers() -> list[ProviderSpec]:

FILE: src/llama_stack/providers/registry/scoring.py
  function available_providers (line 11) | def available_providers() -> list[ProviderSpec]:

FILE: src/llama_stack/providers/registry/tool_runtime.py
  function available_providers (line 17) | def available_providers() -> list[ProviderSpec]:

FILE: src/llama_stack/providers/registry/vector_io.py
  function available_providers (line 19) | def available_providers() -> list[ProviderSpec]:

FILE: src/llama_stack/providers/remote/datasetio/huggingface/__init__.py
  function get_adapter_impl (line 10) | async def get_adapter_impl(

FILE: src/llama_stack/providers/remote/datasetio/huggingface/config.py
  class HuggingfaceDatasetIOConfig (line 13) | class HuggingfaceDatasetIOConfig(BaseModel):
    method sample_run_config (line 17) | def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict...

FILE: src/llama_stack/providers/remote/datasetio/huggingface/huggingface.py
  function parse_hf_params (line 18) | def parse_hf_params(dataset_def: Dataset):
  class HuggingfaceDatasetIOImpl (line 28) | class HuggingfaceDatasetIOImpl(DatasetIO, DatasetsProtocolPrivate):
    method __init__ (line 29) | def __init__(self, config: HuggingfaceDatasetIOConfig) -> None:
    method initialize (line 35) | async def initialize(self) -> None:
    method shutdown (line 46) | async def shutdown(self) -> None: ...
    method register_dataset (line 48) | async def register_dataset(
    method unregister_dataset (line 60) | async def unregister_dataset(self, dataset_id: str) -> None:
    method iterrows (line 65) | async def iterrows(
    method append_rows (line 80) | async def append_rows(self, dataset_id: str, rows: list[dict[str, Any]...

FILE: src/llama_stack/providers/remote/datasetio/nvidia/__init__.py
  function get_adapter_impl (line 10) | async def get_adapter_impl(

FILE: src/llama_stack/providers/remote/datasetio/nvidia/config.py
  class NvidiaDatasetIOConfig (line 14) | class NvidiaDatasetIOConfig(BaseModel):
    method __post_init__ (line 38) | def __post_init__(self):
    method sample_run_config (line 55) | def sample_run_config(cls, **kwargs) -> dict[str, Any]:

FILE: src/llama_stack/providers/remote/datasetio/nvidia/datasetio.py
  class NvidiaDatasetIOAdapter (line 16) | class NvidiaDatasetIOAdapter:
    method __init__ (line 19) | def __init__(self, config: NvidiaDatasetIOConfig):
    method _make_request (line 23) | async def _make_request(
    method register_dataset (line 50) | async def register_dataset(
    method update_dataset (line 83) | async def update_dataset(
    method unregister_dataset (line 94) | async def unregister_dataset(
    method iterrows (line 104) | async def iterrows(
    method append_rows (line 112) | async def append_rows(self, dataset_id: str, rows: list[dict[str, Any]...

FILE: src/llama_stack/providers/remote/eval/nvidia/__init__.py
  function get_adapter_impl (line 13) | async def get_adapter_impl(

FILE: src/llama_stack/providers/remote/eval/nvidia/config.py
  class NVIDIAEvalConfig (line 12) | class NVIDIAEvalConfig(BaseModel):
    method sample_run_config (line 26) | def sample_run_config(cls, **kwargs) -> dict[str, Any]:

FILE: src/llama_stack/providers/remote/eval/nvidia/eval.py
  class NVIDIAEvalImpl (line 36) | class NVIDIAEvalImpl(
    method __init__ (line 41) | def __init__(
    method client (line 61) | def client(self) -> httpx.AsyncClient:
    method initialize (line 66) | async def initialize(self) -> None:
    method shutdown (line 69) | async def shutdown(self) -> None:
    method _evaluator_get (line 73) | async def _evaluator_get(self, path: str):
    method _evaluator_post (line 79) | async def _evaluator_post(self, path: str, data: dict[str, Any]):
    method _evaluator_delete (line 85) | async def _evaluator_delete(self, path: str) -> None:
    method register_benchmark (line 90) | async def register_benchmark(self, task_def: Benchmark) -> None:
    method unregister_benchmark (line 102) | async def unregister_benchmark(self, benchmark_id: str) -> None:
    method run_eval (line 106) | async def run_eval(
    method evaluate_rows (line 128) | async def evaluate_rows(
    method job_status (line 134) | async def job_status(self, request: JobStatusRequest) -> Job:
    method job_cancel (line 155) | async def job_cancel(self, request: JobCancelRequest) -> None:
    method job_result (line 159) | async def job_result(self, request: JobResultRequest) -> EvaluateRespo...

FILE: src/llama_stack/providers/remote/files/openai/__init__.py
  function get_adapter_impl (line 14) | async def get_adapter_impl(config: OpenAIFilesImplConfig, deps: dict[Api...

FILE: src/llama_stack/providers/remote/files/openai/config.py
  class OpenAIFilesImplConfig (line 14) | class OpenAIFilesImplConfig(BaseModel):
    method sample_run_config (line 21) | def sample_run_config(cls, __distro_dir__: str) -> dict[str, Any]:

FILE: src/llama_stack/providers/remote/files/openai/files.py
  function _make_file_object (line 38) | def _make_file_object(
  class OpenAIFilesImpl (line 69) | class OpenAIFilesImpl(Files):
    method __init__ (line 72) | def __init__(self, config: OpenAIFilesImplConfig, policy: list[AccessR...
    method _now (line 78) | def _now(self) -> int:
    method _get_file (line 82) | async def _get_file(
    method _delete_file (line 92) | async def _delete_file(self, file_id: str) -> None:
    method _delete_if_expired (line 103) | async def _delete_if_expired(self, file_id: str) -> None:
    method initialize (line 109) | async def initialize(self) -> None:
    method shutdown (line 125) | async def shutdown(self) -> None:
    method client (line 129) | def client(self) -> OpenAI:
    method sql_store (line 134) | def sql_store(self) -> AuthorizedSqlStore:
    method openai_upload_file (line 138) | async def openai_upload_file(
    method openai_list_files (line 188) | async def openai_list_files(
    method openai_retrieve_file (line 221) | async def openai_retrieve_file(self, request: RetrieveFileRequest) -> ...
    method openai_delete_file (line 227) | async def openai_delete_file(self, request: DeleteFileRequest) -> Open...
    method openai_retrieve_file_content (line 234) | async def openai_retrieve_file_content(self, request: RetrieveFileCont...

FILE: src/llama_stack/providers/remote/files/s3/__init__.py
  function get_adapter_impl (line 14) | async def get_adapter_impl(config: S3FilesImplConfig, deps: dict[Api, An...

FILE: src/llama_stack/providers/remote/files/s3/config.py
  class S3FilesImplConfig (line 14) | class S3FilesImplConfig(BaseModel):
    method sample_run_config (line 32) | def sample_run_config(cls, __distro_dir__: str) -> dict[str, Any]:

FILE: src/llama_stack/providers/remote/files/s3/files.py
  function _create_s3_client (line 48) | def _create_s3_client(config: S3FilesImplConfig) -> "S3Client":
  function _create_bucket_if_not_exists (line 75) | async def _create_bucket_if_not_exists(client: "S3Client", config: S3Fil...
  function _make_file_object (line 105) | def _make_file_object(
  class S3FilesImpl (line 139) | class S3FilesImpl(Files):
    method __init__ (line 142) | def __init__(self, config: S3FilesImplConfig, policy: list[AccessRule]...
    method _now (line 148) | def _now(self) -> int:
    method _get_file (line 152) | async def _get_file(
    method _delete_file (line 162) | async def _delete_file(self, file_id: str) -> None:
    method _delete_if_expired (line 175) | async def _delete_if_expired(self, file_id: str) -> None:
    method initialize (line 181) | async def initialize(self) -> None:
    method shutdown (line 199) | async def shutdown(self) -> None:
    method client (line 203) | def client(self) -> "S3Client":
    method sql_store (line 208) | def sql_store(self) -> AuthorizedSqlStore:
    method openai_upload_file (line 212) | async def openai_upload_file(
    method openai_list_files (line 265) | async def openai_list_files(
    method openai_retrieve_file (line 300) | async def openai_retrieve_file(self, request: RetrieveFileRequest) -> ...
    method openai_delete_file (line 306) | async def openai_delete_file(self, request: DeleteFileRequest) -> Open...
    method openai_retrieve_file_content (line 313) | async def openai_retrieve_file_content(self, request: RetrieveFileCont...

FILE: src/llama_stack/providers/remote/inference/anthropic/__init__.py
  function get_adapter_impl (line 10) | async def get_adapter_impl(config: AnthropicConfig, _deps):

FILE: src/llama_stack/providers/remote/inference/anthropic/anthropic.py
  class AnthropicInferenceAdapter (line 16) | class AnthropicInferenceAdapter(OpenAIMixin):
    method get_base_url (line 32) | def get_base_url(self):
    method list_provider_model_ids (line 35) | async def list_provider_model_ids(self) -> Iterable[str]:

FILE: src/llama_stack/providers/remote/inference/anthropic/config.py
  class AnthropicProviderDataValidator (line 15) | class AnthropicProviderDataValidator(BaseModel):
  class AnthropicConfig (line 23) | class AnthropicConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 25) | def sample_run_config(cls, api_key: str = "${env.ANTHROPIC_API_KEY:=}"...

FILE: src/llama_stack/providers/remote/inference/azure/__init__.py
  function get_adapter_impl (line 10) | async def get_adapter_impl(config: AzureConfig, _deps):

FILE: src/llama_stack/providers/remote/inference/azure/azure.py
  class AzureInferenceAdapter (line 12) | class AzureInferenceAdapter(OpenAIMixin):
    method get_base_url (line 17) | def get_base_url(self) -> str:

FILE: src/llama_stack/providers/remote/inference/azure/config.py
  class AzureProviderDataValidator (line 16) | class AzureProviderDataValidator(BaseModel):
  class AzureConfig (line 34) | class AzureConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 49) | def sample_run_config(

FILE: src/llama_stack/providers/remote/inference/bedrock/__init__.py
  function get_adapter_impl (line 9) | async def get_adapter_impl(config: BedrockConfig, _deps):

FILE: src/llama_stack/providers/remote/inference/bedrock/bedrock.py
  class BedrockInferenceAdapter (line 28) | class BedrockInferenceAdapter(OpenAIMixin):
    method get_base_url (line 38) | def get_base_url(self) -> str:
    method openai_embeddings (line 42) | async def openai_embeddings(
    method openai_completion (line 52) | async def openai_completion(
    method openai_chat_completion (line 63) | async def openai_chat_completion(

FILE: src/llama_stack/providers/remote/inference/bedrock/config.py
  class BedrockProviderDataValidator (line 14) | class BedrockProviderDataValidator(BaseModel):
  class BedrockConfig (line 21) | class BedrockConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 28) | def sample_run_config(cls, **kwargs):

FILE: src/llama_stack/providers/remote/inference/cerebras/__init__.py
  function get_adapter_impl (line 10) | async def get_adapter_impl(config: CerebrasImplConfig, _deps):

FILE: src/llama_stack/providers/remote/inference/cerebras/cerebras.py
  class CerebrasInferenceAdapter (line 16) | class CerebrasInferenceAdapter(OpenAIMixin):
    method get_base_url (line 21) | def get_base_url(self) -> str:
    method openai_embeddings (line 24) | async def openai_embeddings(

FILE: src/llama_stack/providers/remote/inference/cerebras/config.py
  class CerebrasProviderDataValidator (line 18) | class CerebrasProviderDataValidator(BaseModel):
  class CerebrasImplConfig (line 26) | class CerebrasImplConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 33) | def sample_run_config(cls, api_key: str = "${env.CEREBRAS_API_KEY:=}",...

FILE: src/llama_stack/providers/remote/inference/databricks/__init__.py
  function get_adapter_impl (line 10) | async def get_adapter_impl(config: DatabricksImplConfig, _deps):

FILE: src/llama_stack/providers/remote/inference/databricks/config.py
  class DatabricksProviderDataValidator (line 15) | class DatabricksProviderDataValidator(BaseModel):
  class DatabricksImplConfig (line 23) | class DatabricksImplConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 35) | def sample_run_config(

FILE: src/llama_stack/providers/remote/inference/databricks/databricks.py
  class DatabricksInferenceAdapter (line 20) | class DatabricksInferenceAdapter(OpenAIMixin):
    method get_base_url (line 31) | def get_base_url(self) -> str:
    method list_provider_model_ids (line 34) | async def list_provider_model_ids(self) -> Iterable[str]:
    method openai_completion (line 50) | async def openai_completion(

FILE: src/llama_stack/providers/remote/inference/fireworks/__init__.py
  class FireworksProviderDataValidator (line 12) | class FireworksProviderDataValidator(BaseModel):
  function get_adapter_impl (line 16) | async def get_adapter_impl(config: FireworksImplConfig, _deps):

FILE: src/llama_stack/providers/remote/inference/fireworks/config.py
  class FireworksImplConfig (line 16) | class FireworksImplConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 23) | def sample_run_config(cls, api_key: str = "${env.FIREWORKS_API_KEY:=}"...

FILE: src/llama_stack/providers/remote/inference/fireworks/fireworks.py
  class FireworksInferenceAdapter (line 16) | class FireworksInferenceAdapter(OpenAIMixin):
    method get_base_url (line 26) | def get_base_url(self) -> str:

FILE: src/llama_stack/providers/remote/inference/gemini/__init__.py
  function get_adapter_impl (line 10) | async def get_adapter_impl(config: GeminiConfig, _deps):

FILE: src/llama_stack/providers/remote/inference/gemini/config.py
  class GeminiProviderDataValidator (line 15) | class GeminiProviderDataValidator(BaseModel):
  class GeminiConfig (line 23) | class GeminiConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 25) | def sample_run_config(cls, api_key: str = "${env.GEMINI_API_KEY:=}", *...

FILE: src/llama_stack/providers/remote/inference/gemini/gemini.py
  class GeminiInferenceAdapter (line 21) | class GeminiInferenceAdapter(OpenAIMixin):
    method get_base_url (line 30) | def get_base_url(self):
    method openai_embeddings (line 33) | async def openai_embeddings(

FILE: src/llama_stack/providers/remote/inference/groq/__init__.py
  function get_adapter_impl (line 10) | async def get_adapter_impl(config: GroqConfig, _deps):

FILE: src/llama_stack/providers/remote/inference/groq/config.py
  class GroqProviderDataValidator (line 15) | class GroqProviderDataValidator(BaseModel):
  class GroqConfig (line 23) | class GroqConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 30) | def sample_run_config(cls, api_key: str = "${env.GROQ_API_KEY:=}", **k...

FILE: src/llama_stack/providers/remote/inference/groq/groq.py
  class GroqInferenceAdapter (line 12) | class GroqInferenceAdapter(OpenAIMixin):
    method get_base_url (line 17) | def get_base_url(self) -> str:

FILE: src/llama_stack/providers/remote/inference/llama_cpp_server/__init__.py
  function get_adapter_impl (line 10) | async def get_adapter_impl(config: LlamaCppServerConfig, _deps):

FILE: src/llama_stack/providers/remote/inference/llama_cpp_server/config.py
  class LlamaCppServerConfig (line 16) | class LlamaCppServerConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 23) | def sample_run_config(cls, **kwargs) -> dict[str, Any]:

FILE: src/llama_stack/providers/remote/inference/llama_cpp_server/llama_cpp_server.py
  class LlamaCppServerInferenceAdapter (line 12) | class LlamaCppServerInferenceAdapter(OpenAIMixin):
    method get_api_key (line 15) | def get_api_key(self) -> str | None:
    method get_base_url (line 20) | def get_base_url(self) -> str:

FILE: src/llama_stack/providers/remote/inference/llama_openai_compat/__init__.py
  function get_adapter_impl (line 10) | async def get_adapter_impl(config: LlamaCompatConfig, _deps):

FILE: src/llama_stack/providers/remote/inference/llama_openai_compat/config.py
  class LlamaProviderDataValidator (line 15) | class LlamaProviderDataValidator(BaseModel):
  class LlamaCompatConfig (line 23) | class LlamaCompatConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 30) | def sample_run_config(cls, api_key: str = "${env.LLAMA_API_KEY}", **kw...

FILE: src/llama_stack/providers/remote/inference/llama_openai_compat/llama.py
  class LlamaCompatInferenceAdapter (line 22) | class LlamaCompatInferenceAdapter(OpenAIMixin):
    method get_base_url (line 30) | def get_base_url(self) -> str:
    method openai_completion (line 38) | async def openai_completion(
    method openai_embeddings (line 44) | async def openai_embeddings(

FILE: src/llama_stack/providers/remote/inference/nvidia/__init__.py
  function get_adapter_impl (line 12) | async def get_adapter_impl(config: NVIDIAConfig, _deps) -> Inference:

FILE: src/llama_stack/providers/remote/inference/nvidia/config.py
  class NVIDIAProviderDataValidator (line 16) | class NVIDIAProviderDataValidator(BaseModel):
  class NVIDIAConfig (line 24) | class NVIDIAConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 65) | def sample_run_config(

FILE: src/llama_stack/providers/remote/inference/nvidia/nvidia.py
  class NVIDIAInferenceAdapter (line 29) | class NVIDIAInferenceAdapter(OpenAIMixin):
    method initialize (line 46) | async def initialize(self) -> None:
    method get_api_key (line 55) | def get_api_key(self) -> str:
    method get_base_url (line 69) | def get_base_url(self) -> str:
    method list_provider_model_ids (line 77) | async def list_provider_model_ids(self) -> Iterable[str]:
    method construct_model_from_identifier (line 91) | def construct_model_from_identifier(self, identifier: str) -> Model:
    method rerank (line 104) | async def rerank(

FILE: src/llama_stack/providers/remote/inference/nvidia/utils.py
  function _is_nvidia_hosted (line 10) | def _is_nvidia_hosted(config: NVIDIAConfig) -> bool:

FILE: src/llama_stack/providers/remote/inference/oci/__init__.py
  function get_adapter_impl (line 12) | async def get_adapter_impl(config: OCIConfig, _deps) -> InferenceProvider:

FILE: src/llama_stack/providers/remote/inference/oci/auth.py
  class HttpxOciAuth (line 18) | class HttpxOciAuth(httpx.Auth):
    method __init__ (line 30) | def __init__(self, signer: OciAuthSigner):
    method auth_flow (line 34) | def auth_flow(self, request: httpx.Request) -> Generator[httpx.Request...
  class OciInstancePrincipalAuth (line 59) | class OciInstancePrincipalAuth(HttpxOciAuth):
    method __init__ (line 60) | def __init__(self, **kwargs: Mapping[str, Any]):
  class OciUserPrincipalAuth (line 64) | class OciUserPrincipalAuth(HttpxOciAuth):
    method __init__ (line 65) | def __init__(self, config_file: str = DEFAULT_LOCATION, profile_name: ...

FILE: src/llama_stack/providers/remote/inference/oci/config.py
  class OCIProviderDataValidator (line 16) | class OCIProviderDataValidator(BaseModel):
  class OCIConfig (line 37) | class OCIConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 60) | def sample_run_config(

FILE: src/llama_stack/providers/remote/inference/oci/oci.py
  class OCIInferenceAdapter (line 33) | class OCIInferenceAdapter(OpenAIMixin):
    method initialize (line 38) | async def initialize(self) -> None:
    method get_base_url (line 49) | def get_base_url(self) -> str:
    method get_api_key (line 53) | def get_api_key(self) -> str | None:
    method get_extra_client_params (line 57) | def get_extra_client_params(self) -> dict[str, Any]:
    method _get_oci_signer (line 74) | def _get_oci_signer(self) -> oci.signer.AbstractBaseSigner | None:
    method _get_oci_config (line 79) | def _get_oci_config(self) -> dict:
    method _get_auth (line 91) | def _get_auth(self) -> httpx.Auth:
    method list_provider_model_ids (line 101) | async def list_provider_model_ids(self) -> Iterable[str]:
    method construct_model_from_identifier (line 142) | def construct_model_from_identifier(self, identifier: str) -> Model:

FILE: src/llama_stack/providers/remote/inference/ollama/__init__.py
  function get_adapter_impl (line 10) | async def get_adapter_impl(config: OllamaImplConfig, _deps):

FILE: src/llama_stack/providers/remote/inference/ollama/config.py
  class OllamaImplConfig (line 16) | class OllamaImplConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 22) | def sample_run_config(

FILE: src/llama_stack/providers/remote/inference/ollama/ollama.py
  class OllamaInferenceAdapter (line 25) | class OllamaInferenceAdapter(OpenAIMixin):
    method ollama_client (line 54) | def ollama_client(self) -> AsyncOllamaClient:
    method get_api_key (line 67) | def get_api_key(self):
    method get_base_url (line 70) | def get_base_url(self):
    method initialize (line 73) | async def initialize(self) -> None:
    method health (line 81) | async def health(self) -> HealthResponse:
    method shutdown (line 95) | async def shutdown(self) -> None:
    method register_model (line 98) | async def register_model(self, model: Model) -> Model:

FILE: src/llama_stack/providers/remote/inference/openai/__init__.py
  function get_adapter_impl (line 10) | async def get_adapter_impl(config: OpenAIConfig, _deps):

FILE: src/llama_stack/providers/remote/inference/openai/config.py
  class OpenAIProviderDataValidator (line 15) | class OpenAIProviderDataValidator(BaseModel):
  class OpenAIConfig (line 23) | class OpenAIConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 30) | def sample_run_config(

FILE: src/llama_stack/providers/remote/inference/openai/openai.py
  class OpenAIInferenceAdapter (line 18) | class OpenAIInferenceAdapter(OpenAIMixin):
    method get_base_url (line 34) | def get_base_url(self) -> str:

FILE: src/llama_stack/providers/remote/inference/passthrough/__init__.py
  class PassthroughProviderDataValidator (line 12) | class PassthroughProviderDataValidator(BaseModel):
  function get_adapter_impl (line 17) | async def get_adapter_impl(config: PassthroughImplConfig, _deps):

FILE: src/llama_stack/providers/remote/inference/passthrough/config.py
  class PassthroughImplConfig (line 16) | class PassthroughImplConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 23) | def sample_run_config(

FILE: src/llama_stack/providers/remote/inference/passthrough/passthrough.py
  class PassthroughInferenceAdapter (line 28) | class PassthroughInferenceAdapter(NeedsRequestProviderData, Inference):
    method __init__ (line 29) | def __init__(self, config: PassthroughImplConfig) -> None:
    method initialize (line 32) | async def initialize(self) -> None:
    method shutdown (line 35) | async def shutdown(self) -> None:
    method unregister_model (line 38) | async def unregister_model(self, model_id: str) -> None:
    method register_model (line 41) | async def register_model(self, model: Model) -> Model:
    method list_models (line 44) | async def list_models(self) -> list[Model]:
    method should_refresh_models (line 70) | async def should_refresh_models(self) -> bool:
    method _get_openai_client (line 74) | def _get_openai_client(self) -> AsyncOpenAI:
    method _get_passthrough_url (line 84) | def _get_passthrough_url(self) -> str:
    method _get_passthrough_api_key (line 96) | def _get_passthrough_api_key(self) -> str:
    method openai_completion (line 108) | async def openai_completion(
    method openai_chat_completion (line 122) | async def openai_chat_completion(
    method openai_embeddings (line 136) | async def openai_embeddings(

FILE: src/llama_stack/providers/remote/inference/runpod/__init__.py
  function get_adapter_impl (line 10) | async def get_adapter_impl(config: RunpodImplConfig, _deps):

FILE: src/llama_stack/providers/remote/inference/runpod/config.py
  class RunpodProviderDataValidator (line 15) | class RunpodProviderDataValidator(BaseModel):
  class RunpodImplConfig (line 23) | class RunpodImplConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 35) | def sample_run_config(cls, **kwargs: Any) -> dict[str, Any]:

FILE: src/llama_stack/providers/remote/inference/runpod/runpod.py
  class RunpodInferenceAdapter (line 12) | class RunpodInferenceAdapter(OpenAIMixin):
    method get_base_url (line 22) | def get_base_url(self) -> str:

FILE: src/llama_stack/providers/remote/inference/sambanova/__init__.py
  function get_adapter_impl (line 10) | async def get_adapter_impl(config: SambaNovaImplConfig, _deps):

FILE: src/llama_stack/providers/remote/inference/sambanova/config.py
  class SambaNovaProviderDataValidator (line 15) | class SambaNovaProviderDataValidator(BaseModel):
  class SambaNovaImplConfig (line 23) | class SambaNovaImplConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 30) | def sample_run_config(cls, api_key: str = "${env.SAMBANOVA_API_KEY:=}"...

FILE: src/llama_stack/providers/remote/inference/sambanova/sambanova.py
  class SambaNovaInferenceAdapter (line 13) | class SambaNovaInferenceAdapter(OpenAIMixin):
    method get_base_url (line 22) | def get_base_url(self) -> str:

FILE: src/llama_stack/providers/remote/inference/tgi/__init__.py
  function get_adapter_impl (line 10) | async def get_adapter_impl(

FILE: src/llama_stack/providers/remote/inference/tgi/config.py
  class TGIImplConfig (line 15) | class TGIImplConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 24) | def sample_run_config(
  class InferenceEndpointImplConfig (line 35) | class InferenceEndpointImplConfig(BaseModel):
    method sample_run_config (line 45) | def sample_run_config(
  class InferenceAPIImplConfig (line 58) | class InferenceAPIImplConfig(BaseModel):
    method sample_run_config (line 68) | def sample_run_config(

FILE: src/llama_stack/providers/remote/inference/tgi/tgi.py
  class _HfAdapter (line 25) | class _HfAdapter(OpenAIMixin):
    method get_api_key (line 35) | def get_api_key(self):
    method get_base_url (line 38) | def get_base_url(self):
    method list_provider_model_ids (line 41) | async def list_provider_model_ids(self) -> Iterable[str]:
    method openai_embeddings (line 44) | async def openai_embeddings(
  class TGIAdapter (line 51) | class TGIAdapter(_HfAdapter):
    method initialize (line 52) | async def initialize(self, config: TGIImplConfig) -> None:
  class InferenceAPIAdapter (line 72) | class InferenceAPIAdapter(_HfAdapter):
    method initialize (line 73) | async def initialize(self, config: InferenceAPIImplConfig) -> None:
  class InferenceEndpointAdapter (line 81) | class InferenceEndpointAdapter(_HfAdapter):
    method initialize (line 82) | async def initialize(self, config: InferenceEndpointImplConfig) -> None:

FILE: src/llama_stack/providers/remote/inference/together/__init__.py
  class TogetherProviderDataValidator (line 12) | class TogetherProviderDataValidator(BaseModel):
  function get_adapter_impl (line 16) | async def get_adapter_impl(config: TogetherImplConfig, _deps):

FILE: src/llama_stack/providers/remote/inference/together/config.py
  class TogetherImplConfig (line 16) | class TogetherImplConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 23) | def sample_run_config(cls, **kwargs) -> dict[str, Any]:

FILE: src/llama_stack/providers/remote/inference/together/together.py
  class TogetherInferenceAdapter (line 29) | class TogetherInferenceAdapter(OpenAIMixin, NeedsRequestProviderData):
    method get_base_url (line 44) | def get_base_url(self):
    method _get_client (line 47) | def _get_client(self) -> AsyncTogether:
    method list_provider_model_ids (line 61) | async def list_provider_model_ids(self) -> Iterable[str]:
    method openai_embeddings (line 65) | async def openai_embeddings(

FILE: src/llama_stack/providers/remote/inference/vertexai/__init__.py
  function get_adapter_impl (line 10) | async def get_adapter_impl(config: VertexAIConfig, _deps):

FILE: src/llama_stack/providers/remote/inference/vertexai/config.py
  class VertexAIProviderDataValidator (line 15) | class VertexAIProviderDataValidator(BaseModel):
  class VertexAIConfig (line 28) | class VertexAIConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 55) | def sample_run_config(

FILE: src/llama_stack/providers/remote/inference/vertexai/converters.py
  function _to_dict (line 53) | def _to_dict(obj: Any) -> dict[str, Any]:
  class _GeminiLogprobCandidate (line 63) | class _GeminiLogprobCandidate(BaseModel):
    method _coerce_none_token (line 74) | def _coerce_none_token(cls, v: Any) -> str:
  class _GeminiTopCandidatesEntry (line 78) | class _GeminiTopCandidatesEntry(BaseModel):
    method _coerce_none_list (line 87) | def _coerce_none_list(cls, v: Any) -> list:
  class _GeminiLogprobsResult (line 91) | class _GeminiLogprobsResult(BaseModel):
    method _coerce_none_list (line 101) | def _coerce_none_list(cls, v: Any) -> list:
  class _CandidateData (line 106) | class _CandidateData:
  function convert_finish_reason (line 132) | def convert_finish_reason(
  function convert_response_format (line 142) | def convert_response_format(
  function _extract_text_content (line 169) | def _extract_text_content(content: str | list[dict[str, Any]] | None) ->...
  function _convert_image_url_part (line 188) | def _convert_image_url_part(part: dict[str, Any]) -> dict[str, Any] | None:
  function _convert_user_message (line 210) | def _convert_user_message(msg: dict[str, Any]) -> dict[str, Any]:
  function _parse_tool_call_arguments (line 237) | def _parse_tool_call_arguments(arguments: str | dict[str, Any]) -> dict[...
  function _convert_assistant_message (line 248) | def _convert_assistant_message(msg: dict[str, Any]) -> dict[str, Any] | ...
  function _convert_tool_message (line 274) | def _convert_tool_message(msg: dict[str, Any], all_messages: list[Any]) ...
  function convert_openai_messages_to_gemini (line 292) | def convert_openai_messages_to_gemini(
  function _find_function_name_for_tool_call_id (line 328) | def _find_function_name_for_tool_call_id(messages: list[Any], tool_call_...
  function convert_openai_tools_to_gemini (line 343) | def convert_openai_tools_to_gemini(
  function convert_deprecated_functions_to_tools (line 381) | def convert_deprecated_functions_to_tools(
  function convert_deprecated_function_call_to_tool_choice (line 392) | def convert_deprecated_function_call_to_tool_choice(
  function generate_completion_id (line 409) | def generate_completion_id() -> str:
  function _extract_candidate_parts (line 414) | def _extract_candidate_parts(candidate: Any) -> tuple[list[str], list[st...
  function _build_top_logprobs (line 455) | def _build_top_logprobs(
  function _build_token_logprob (line 476) | def _build_token_logprob(
  function _extract_logprobs (line 498) | def _extract_logprobs(candidate: Any) -> OpenAIChoiceLogprobs | None:
  function _process_candidates (line 522) | def _process_candidates(response_or_chunk: Any) -> list[_CandidateData]:
  function _resolve_finish_reason_common (line 540) | def _resolve_finish_reason_common(
  function _resolve_finish_reason (line 553) | def _resolve_finish_reason(
  function convert_gemini_response_to_openai (line 562) | def convert_gemini_response_to_openai(
  function extract_usage (line 613) | def extract_usage(
  function _resolve_stream_finish_reason (line 644) | def _resolve_stream_finish_reason(
  function convert_gemini_stream_chunk_to_openai (line 656) | def convert_gemini_stream_chunk_to_openai(
  function convert_gemini_stream_chunk_to_openai_completion (line 699) | def convert_gemini_stream_chunk_to_openai_completion(
  function convert_completion_prompt_to_contents (line 739) | def convert_completion_prompt_to_contents(prompt: str) -> list[dict[str,...
  function convert_gemini_response_to_openai_completion (line 744) | def convert_gemini_response_to_openai_completion(

FILE: src/llama_stack/providers/remote/inference/vertexai/utils.py
  function resolve_timeout_ms (line 23) | def resolve_timeout_ms(timeout: TimeoutConfig | float | None) -> int | N...
  function resolve_ssl_verify (line 36) | def resolve_ssl_verify(tls: TLSConfig) -> ssl.SSLContext | Path | str | ...
  function build_httpx_kwargs (line 51) | def build_httpx_kwargs(network_config: NetworkConfig) -> tuple[dict[str,...
  function build_http_options (line 70) | def build_http_options(network_config: NetworkConfig | None) -> genai_ty...

FILE: src/llama_stack/providers/remote/inference/vertexai/vertexai.py
  class GeminiSamplingParams (line 51) | class GeminiSamplingParams(BaseModel):
    method from_openai_params (line 74) | def from_openai_params(cls, params: OpenAIChatCompletionRequestWithExt...
  class GeminiCompletionSamplingParams (line 101) | class GeminiCompletionSamplingParams(BaseModel):
    method from_openai_params (line 120) | def from_openai_params(cls, params: OpenAICompletionRequestWithExtraBo...
  class VertexAIInferenceAdapter (line 139) | class VertexAIInferenceAdapter(NeedsRequestProviderData, BaseModel):
    method _close_managed_httpx_client (line 155) | async def _close_managed_httpx_client(self) -> None:
    method initialize (line 163) | async def initialize(self) -> None:
    method shutdown (line 185) | async def shutdown(self) -> None:
    method register_model (line 190) | async def register_model(self, model: Model) -> Model:
    method unregister_model (line 198) | async def unregister_model(self, model_id: str) -> None:
    method _create_adc_client (line 201) | def _create_adc_client(self, project: str, location: str) -> Client:
    method _create_client (line 208) | def _create_client(self, project: str, location: str, *, access_token:...
    method check_model_availability (line 225) | async def check_model_availability(self, model: str) -> bool:
    method _is_usable_model (line 246) | def _is_usable_model(model: Any) -> bool:
    method _get_request_provider_overrides (line 256) | def _get_request_provider_overrides(self) -> VertexAIProviderDataValid...
    method _get_client (line 273) | def _get_client(self) -> Client:
    method _get_provider_model_id (line 295) | async def _get_provider_model_id(self, model: str) -> str:
    method list_provider_model_ids (line 305) | async def list_provider_model_ids(self) -> list[str]:
    method list_models (line 329) | async def list_models(self) -> list[Model] | None:
    method should_refresh_models (line 372) | async def should_refresh_models(self) -> bool:
    method _filter_fields (line 375) | def _filter_fields(self, **kwargs):
    method model_dump (line 386) | def model_dump(self, **kwargs):
    method model_dump_json (line 391) | def model_dump_json(self, **kwargs):
    method _build_tool_config (line 396) | def _build_tool_config(self, tool_choice: str | dict[str, Any] | None)...
    method _make_tool_config (line 414) | def _make_tool_config(
    method _collect_sampling_params (line 426) | def _collect_sampling_params(params: OpenAIChatCompletionRequestWithEx...
    method _build_thinking_config (line 442) | def _build_thinking_config(reasoning_effort: str | None) -> genai_type...
    method _build_generation_config (line 475) | def _build_generation_config(
    method _stream_chat_completion (line 510) | async def _stream_chat_completion(
    method _stream_completion (line 554) | async def _stream_completion(
    method _is_model_allowed (line 590) | def _is_model_allowed(self, provider_model_id: str) -> bool:
    method _validate_model_allowed (line 595) | def _validate_model_allowed(self, provider_model_id: str) -> None:
    method _localize_image_url (line 603) | async def _localize_image_url(message: OpenAIMessageParam) -> OpenAIMe...
    method _warn_unsupported_chat_params (line 622) | def _warn_unsupported_chat_params(params: OpenAIChatCompletionRequestW...
    method _resolve_deprecated_tools (line 637) | def _resolve_deprecated_tools(
    method openai_chat_completion (line 654) | async def openai_chat_completion(
    method _validate_completion_prompt (line 697) | def _validate_completion_prompt(prompt: Any) -> list[str]:
    method _warn_unsupported_completion_params (line 709) | def _warn_unsupported_completion_params(params: OpenAICompletionReques...
    method _build_completion_config (line 720) | def _build_completion_config(
    method openai_completion (line 735) | async def openai_completion(
    method openai_embeddings (line 818) | async def openai_embeddings(
    method rerank (line 867) | async def rerank(

FILE: src/llama_stack/providers/remote/inference/vllm/__init__.py
  class VLLMProviderDataValidator (line 12) | class VLLMProviderDataValidator(BaseModel):
  function get_adapter_impl (line 16) | async def get_adapter_impl(config: VLLMInferenceAdapterConfig, _deps):

FILE: src/llama_stack/providers/remote/inference/vllm/config.py
  class VLLMInferenceAdapterConfig (line 21) | class VLLMInferenceAdapterConfig(RemoteInferenceProviderConfig):
    method migrate_tls_verify_to_network (line 43) | def migrate_tls_verify_to_network(self) -> "VLLMInferenceAdapterConfig":
    method sample_run_config (line 64) | def sample_run_config(

FILE: src/llama_stack/providers/remote/inference/vllm/vllm.py
  class VLLMInferenceAdapter (line 37) | class VLLMInferenceAdapter(OpenAIMixin):
    method get_api_key (line 44) | def get_api_key(self) -> str | None:
    method get_base_url (line 49) | def get_base_url(self) -> str:
    method initialize (line 55) | async def initialize(self) -> None:
    method health (line 70) | async def health(self) -> HealthResponse:
    method check_model_availability (line 91) | async def check_model_availability(self, model: str) -> bool:
    method openai_chat_completion (line 105) | async def openai_chat_completion(
    method construct_model_from_identifier (line 117) | def construct_model_from_identifier(self, identifier: str) -> Model:
    method rerank (line 136) | async def rerank(

FILE: src/llama_stack/providers/remote/inference/watsonx/__init__.py
  function get_adapter_impl (line 10) | async def get_adapter_impl(config: WatsonXConfig, _deps):

FILE: src/llama_stack/providers/remote/inference/watsonx/config.py
  class WatsonXProviderDataValidator (line 16) | class WatsonXProviderDataValidator(BaseModel):
  class WatsonXConfig (line 25) | class WatsonXConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 40) | def sample_run_config(cls, **kwargs) -> dict[str, Any]:

FILE: src/llama_stack/providers/remote/inference/watsonx/watsonx.py
  function _get_iam_token (line 40) | def _get_iam_token(api_key: str) -> str:
  class WatsonXInferenceAdapter (line 72) | class WatsonXInferenceAdapter(OpenAIMixin):
    method __init__ (line 80) | def __init__(self, config: WatsonXConfig):
    method get_base_url (line 83) | def get_base_url(self) -> str:
    method get_extra_client_params (line 86) | def get_extra_client_params(self) -> dict[str, Any]:
    method client (line 93) | def client(self) -> AsyncOpenAI:
    method initialize (line 114) | async def initialize(self) -> None:
    method shutdown (line 117) | async def shutdown(self) -> None:
    method list_provider_model_ids (line 120) | async def list_provider_model_ids(self) -> Iterable[str]:
    method construct_model_from_identifier (line 127) | def construct_model_from_identifier(self, identifier: str) -> Model:
    method _inject_project_id (line 151) | def _inject_project_id(self, params: Any) -> Any:
    method openai_chat_completion (line 160) | async def openai_chat_completion(
    method openai_completion (line 171) | async def openai_completion(
    method openai_embeddings (line 179) | async def openai_embeddings(
    method _get_model_specs (line 187) | def _get_model_specs(self) -> list[dict[str, Any]]:

FILE: src/llama_stack/providers/remote/safety/bedrock/__init__.py
  function get_adapter_impl (line 13) | async def get_adapter_impl(config: BedrockSafetyConfig, _deps) -> Any:

FILE: src/llama_stack/providers/remote/safety/bedrock/bedrock.py
  class BedrockSafetyAdapter (line 28) | class BedrockSafetyAdapter(ShieldToModerationMixin, Safety, ShieldsProto...
    method __init__ (line 29) | def __init__(self, config: BedrockSafetyConfig) -> None:
    method initialize (line 33) | async def initialize(self) -> None:
    method shutdown (line 40) | async def shutdown(self) -> None:
    method register_shield (line 43) | async def register_shield(self, shield: Shield) -> None:
    method unregister_shield (line 56) | async def unregister_shield(self, identifier: str) -> None:
    method run_shield (line 59) | async def run_shield(self, request: RunShieldRequest) -> RunShieldResp...

FILE: src/llama_stack/providers/remote/safety/bedrock/config.py
  class BedrockSafetyConfig (line 13) | class BedrockSafetyConfig(BedrockBaseConfig):

FILE: src/llama_stack/providers/remote/safety/nvidia/__init__.py
  function get_adapter_impl (line 13) | async def get_adapter_impl(config: NVIDIASafetyConfig, _deps) -> Any:

FILE: src/llama_stack/providers/remote/safety/nvidia/config.py
  class NVIDIASafetyConfig (line 15) | class NVIDIASafetyConfig(BaseModel):
    method sample_run_config (line 36) | def sample_run_config(cls, **kwargs) -> dict[str, Any]:

FILE: src/llama_stack/providers/remote/safety/nvidia/nvidia.py
  class NVIDIASafetyAdapter (line 30) | class NVIDIASafetyAdapter(ShieldToModerationMixin, Safety, ShieldsProtoc...
    method __init__ (line 31) | def __init__(self, config: NVIDIASafetyConfig) -> None:
    method initialize (line 40) | async def initialize(self) -> None:
    method shutdown (line 43) | async def shutdown(self) -> None:
    method register_shield (line 46) | async def register_shield(self, shield: Shield) -> None:
    method unregister_shield (line 50) | async def unregister_shield(self, identifier: str) -> None:
    method run_shield (line 53) | async def run_shield(self, request: RunShieldRequest) -> RunShieldResp...
  class NeMoGuardrails (line 63) | class NeMoGuardrails:
    method __init__ (line 71) | def __init__(
    method _guardrails_post (line 101) | async def _guardrails_post(self, path: str, data: Any | None):
    method run (line 111) | async def run(self, messages: list[OpenAIMessageParam]) -> RunShieldRe...

FILE: src/llama_stack/providers/remote/safety/passthrough/__init__.py
  function get_adapter_impl (line 12) | async def get_adapter_impl(config: PassthroughSafetyConfig, _deps: Any) ...

FILE: src/llama_stack/providers/remote/safety/passthrough/config.py
  class PassthroughProviderDataValidator (line 29) | class PassthroughProviderDataValidator(BaseModel):
  class PassthroughSafetyConfig (line 40) | class PassthroughSafetyConfig(BaseModel):
    method validate_forward_headers (line 60) | def validate_forward_headers(cls, v: dict[str, str]) -> dict[str, str]:
    method sample_run_config (line 72) | def sample_run_config(

FILE: src/llama_stack/providers/remote/safety/passthrough/passthrough.py
  class PassthroughSafetyAdapter (line 32) | class PassthroughSafetyAdapter(
    method __init__ (line 41) | def __init__(self, config: PassthroughSafetyConfig) -> None:
    method initialize (line 45) | async def initialize(self) -> None:
    method shutdown (line 48) | async def shutdown(self) -> None:
    method register_shield (line 52) | async def register_shield(self, shield: Shield) -> None:
    method unregister_shield (line 55) | async def unregister_shield(self, identifier: str) -> None:
    method _get_api_key (line 58) | def _get_api_key(self) -> str | None:
    method _build_forward_headers (line 70) | def _build_forward_headers(self) -> dict[str, str]:
    method _build_request_headers (line 92) | def _build_request_headers(self) -> dict[str, str]:
    method run_shield (line 105) | async def run_shield(self, request: RunShieldRequest) -> RunShieldResp...
    method run_moderation (line 142) | async def run_moderation(self, request: RunModerationRequest) -> Moder...
    method _post_moderation (line 190) | async def _post_moderation(self, url: str, payload: dict[str, Any], he...
    method _parse_moderation_response (line 219) | def _parse_moderation_response(self, data: dict[str, Any]) -> RunShiel...

FILE: src/llama_stack/providers/remote/safety/sambanova/__init__.py
  function get_adapter_impl (line 13) | async def get_adapter_impl(config: SambaNovaSafetyConfig, _deps) -> Any:

FILE: src/llama_stack/providers/remote/safety/sambanova/config.py
  class SambaNovaProviderDataValidator (line 14) | class SambaNovaProviderDataValidator(BaseModel):
  class SambaNovaSafetyConfig (line 22) | class SambaNovaSafetyConfig(BaseModel):
    method sample_run_config (line 33) | def sample_run_config(cls, api_key: str = "${env.SAMBANOVA_API_KEY:=}"...

FILE: src/llama_stack/providers/remote/safety/sambanova/sambanova.py
  class SambaNovaSafetyAdapter (line 31) | class SambaNovaSafetyAdapter(ShieldToModerationMixin, Safety, ShieldsPro...
    method __init__ (line 32) | def __init__(self, config: SambaNovaSafetyConfig) -> None:
    method initialize (line 36) | async def initialize(self) -> None:
    method shutdown (line 39) | async def shutdown(self) -> None:
    method _get_api_key (line 42) | def _get_api_key(self) -> str:
    method register_shield (line 54) | async def register_shield(self, shield: Shield) -> None:
    method unregister_shield (line 70) | async def unregister_shield(self, identifier: str) -> None:
    method run_shield (line 73) | async def run_shield(self, request: RunShieldRequest) -> RunShieldResp...

FILE: src/llama_stack/providers/remote/tool_runtime/bing_search/__init__.py
  class BingSearchToolProviderDataValidator (line 14) | class BingSearchToolProviderDataValidator(BaseModel):
  function get_adapter_impl (line 18) | async def get_adapter_impl(config: BingSearchToolConfig, _deps):

FILE: src/llama_stack/providers/remote/tool_runtime/bing_search/bing_search.py
  class BingSearchToolRuntimeImpl (line 26) | class BingSearchToolRuntimeImpl(ToolGroupsProtocolPrivate, ToolRuntime, ...
    method __init__ (line 27) | def __init__(self, config: BingSearchToolConfig):
    method initialize (line 31) | async def initialize(self):
    method register_toolgroup (line 34) | async def register_toolgroup(self, toolgroup: ToolGroup) -> None:
    method unregister_toolgroup (line 37) | async def unregister_toolgroup(self, toolgroup_id: str) -> None:
    method _get_api_key (line 40) | def _get_api_key(self) -> str:
    method list_runtime_tools (line 51) | async def list_runtime_tools(
    method invoke_tool (line 76) | async def invoke_tool(
    method _clean_response (line 100) | def _clean_response(self, search_response):

FILE: src/llama_stack/providers/remote/tool_runtime/bing_search/config.py
  class BingSearchToolConfig (line 12) | class BingSearchToolConfig(BaseModel):
    method sample_run_config (line 19) | def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict...

FILE: src/llama_stack/providers/remote/tool_runtime/brave_search/__init__.py
  class BraveSearchToolProviderDataValidator (line 13) | class BraveSearchToolProviderDataValidator(BaseModel):
  function get_adapter_impl (line 17) | async def get_adapter_impl(config: BraveSearchToolConfig, _deps):

FILE: src/llama_stack/providers/remote/tool_runtime/brave_search/brave_search.py
  class BraveSearchToolRuntimeImpl (line 25) | class BraveSearchToolRuntimeImpl(ToolGroupsProtocolPrivate, ToolRuntime,...
    method __init__ (line 26) | def __init__(self, config: BraveSearchToolConfig):
    method initialize (line 29) | async def initialize(self):
    method register_toolgroup (line 32) | async def register_toolgroup(self, toolgroup: ToolGroup) -> None:
    method unregister_toolgroup (line 35) | async def unregister_toolgroup(self, toolgroup_id: str) -> None:
    method _get_api_key (line 38) | def _get_api_key(self) -> str:
    method list_runtime_tools (line 49) | async def list_runtime_tools(
    method invoke_tool (line 74) | async def invoke_tool(
    method _clean_brave_response (line 98) | def _clean_brave_response(self, search_response):
    method _clean_result_by_type (line 110) | def _clean_result_by_type(self, r_type, results, idx=None):

FILE: src/llama_stack/providers/remote/tool_runtime/brave_search/config.py
  class BraveSearchToolConfig (line 12) | class BraveSearchToolConfig(BaseModel):
    method sample_run_config (line 23) | def sample_run_config(cls, __distro_dir__: str) -> dict[str, Any]:

FILE: src/llama_stack/providers/remote/tool_runtime/model_context_protocol/__init__.py
  function get_adapter_impl (line 10) | async def get_adapter_impl(config: MCPProviderConfig, _deps):

FILE: src/llama_stack/providers/remote/tool_runtime/model_context_protocol/config.py
  class MCPProviderDataValidator (line 12) | class MCPProviderDataValidator(BaseModel):
  class MCPProviderConfig (line 23) | class MCPProviderConfig(BaseModel):
    method sample_run_config (line 25) | def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict...

FILE: src/llama_stack/providers/remote/tool_runtime/model_context_protocol/model_context_protocol.py
  class ModelContextProtocolToolRuntimeImpl (line 28) | class ModelContextProtocolToolRuntimeImpl(ToolGroupsProtocolPrivate, Too...
    method __init__ (line 29) | def __init__(self, config: MCPProviderConfig, _deps: dict[Api, Any]):
    method initialize (line 32) | async def initialize(self):
    method register_toolgroup (line 35) | async def register_toolgroup(self, toolgroup: ToolGroup) -> None:
    method unregister_toolgroup (line 38) | async def unregister_toolgroup(self, toolgroup_id: str) -> None:
    method list_runtime_tools (line 41) | async def list_runtime_tools(
    method invoke_tool (line 56) | async def invoke_tool(
    method get_headers_from_request (line 77) | async def get_headers_from_request(self, mcp_endpoint_uri: str) -> dic...

FILE: src/llama_stack/providers/remote/tool_runtime/tavily_search/__init__.py
  class TavilySearchToolProviderDataValidator (line 13) | class TavilySearchToolProviderDataValidator(BaseModel):
  function get_adapter_impl (line 17) | async def get_adapter_impl(config: TavilySearchToolConfig, _deps):

FILE: src/llama_stack/providers/remote/tool_runtime/tavily_search/config.py
  class TavilySearchToolConfig (line 12) | class TavilySearchToolConfig(BaseModel):
    method sample_run_config (line 23) | def sample_run_config(cls, __distro_dir__: str) -> dict[str, Any]:

FILE: src/llama_stack/providers/remote/tool_runtime/tavily_search/tavily_search.py
  class TavilySearchToolRuntimeImpl (line 26) | class TavilySearchToolRuntimeImpl(ToolGroupsProtocolPrivate, ToolRuntime...
    method __init__ (line 27) | def __init__(self, config: TavilySearchToolConfig):
    method initialize (line 30) | async def initialize(self):
    method register_toolgroup (line 33) | async def register_toolgroup(self, toolgroup: ToolGroup) -> None:
    method unregister_toolgroup (line 36) | async def unregister_toolgroup(self, toolgroup_id: str) -> None:
    method _get_api_key (line 39) | def _get_api_key(self) -> str:
    method list_runtime_tools (line 50) | async def list_runtime_tools(
    method invoke_tool (line 75) | async def invoke_tool(
    method _clean_tavily_response (line 88) | def _clean_tavily_response(self, search_response, top_k=3):

FILE: src/llama_stack/providers/remote/tool_runtime/wolfram_alpha/__init__.py
  class WolframAlphaToolProviderDataValidator (line 15) | class WolframAlphaToolProviderDataValidator(BaseModel):
  function get_adapter_impl (line 19) | async def get_adapter_impl(config: WolframAlphaToolConfig, _deps):

FILE: src/llama_stack/providers/remote/tool_runtime/wolfram_alpha/config.py
  class WolframAlphaToolConfig (line 12) | class WolframAlphaToolConfig(BaseModel):
    method sample_run_config (line 18) | def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict...

FILE: src/llama_stack/providers/remote/tool_runtime/wolfram_alpha/wolfram_alpha.py
  class WolframAlphaToolRuntimeImpl (line 26) | class WolframAlphaToolRuntimeImpl(ToolGroupsProtocolPrivate, ToolRuntime...
    method __init__ (line 27) | def __init__(self, config: WolframAlphaToolConfig):
    method initialize (line 31) | async def initialize(self):
    method register_toolgroup (line 34) | async def register_toolgroup(self, toolgroup: ToolGroup) -> None:
    method unregister_toolgroup (line 37) | async def unregister_toolgroup(self, toolgroup_id: str) -> None:
    method _get_api_key (line 40) | def _get_api_key(self) -> str:
    method list_runtime_tools (line 51) | async def list_runtime_tools(
    method invoke_tool (line 76) | async def invoke_tool(
    method _clean_wolfram_alpha_response (line 91) | def _clean_wolfram_alpha_response(self, wa_response):

FILE: src/llama_stack/providers/remote/vector_io/chroma/__init__.py
  function get_adapter_impl (line 12) | async def get_adapter_impl(config: ChromaVectorIOConfig, deps: dict[Api,...

FILE: src/llama_stack/providers/remote/vector_io/chroma/chroma.py
  function maybe_await (line 51) | async def maybe_await(result):
  class ChromaIndex (line 57) | class ChromaIndex(EmbeddingIndex):
    method __init__ (line 58) | def __init__(self, client: ChromaClientType, collection, kvstore: KVSt...
    method initialize (line 63) | async def initialize(self):
    method add_chunks (line 66) | async def add_chunks(self, chunks: list[EmbeddedChunk]):
    method query_vector (line 78) | async def query_vector(
    method delete (line 112) | async def delete(self):
    method query_keyword (line 115) | async def query_keyword(
    method delete_chunks (line 165) | async def delete_chunks(self, chunks_for_deletion: list[ChunkForDeleti...
    method query_hybrid (line 170) | async def query_hybrid(
  class ChromaVectorIOAdapter (line 232) | class ChromaVectorIOAdapter(OpenAIVectorStoreMixin, VectorIO, VectorStor...
    method __init__ (line 233) | def __init__(
    method initialize (line 246) | async def initialize(self) -> None:
    method shutdown (line 263) | async def shutdown(self) -> None:
    method register_vector_store (line 267) | async def register_vector_store(self, vector_store: VectorStore) -> None:
    method unregister_vector_store (line 277) | async def unregister_vector_store(self, vector_store_id: str) -> None:
    method insert_chunks (line 285) | async def insert_chunks(self, request: InsertChunksRequest) -> None:
    method query_chunks (line 292) | async def query_chunks(self, request: QueryChunksRequest) -> QueryChun...
    method _get_and_cache_vector_store_index (line 300) | async def _get_and_cache_vector_store_index(self, vector_store_id: str...
    method delete_chunks (line 321) | async def delete_chunks(self, request: DeleteChunksRequest) -> None:

FILE: src/llama_stack/providers/remote/vector_io/chroma/config.py
  class ChromaVectorIOConfig (line 16) | class ChromaVectorIOConfig(BaseModel):
    method sample_run_config (line 21) | def sample_run_config(cls, __distro_dir__: str, url: str = "${env.CHRO...

FILE: src/llama_stack/providers/remote/vector_io/elasticsearch/__init__.py
  function get_adapter_impl (line 12) | async def get_adapter_impl(config: ElasticsearchVectorIOConfig, deps: di...

FILE: src/llama_stack/providers/remote/vector_io/elasticsearch/config.py
  class ElasticsearchVectorIOConfig (line 16) | class ElasticsearchVectorIOConfig(BaseModel):
    method sample_run_config (line 24) | def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict...

FILE: src/llama_stack/providers/remote/vector_io/elasticsearch/elasticsearch.py
  class ElasticsearchIndex (line 45) | class ElasticsearchIndex(EmbeddingIndex):
    method __init__ (line 46) | def __init__(self, client: AsyncElasticsearch, collection_name: str):
    method _is_rerank_linear_param_valid (line 57) | async def _is_rerank_linear_param_valid(self, value: dict) -> bool:
    method _convert_to_linear_params (line 70) | def _convert_to_linear_params(self, reranker_params: dict[str, Any]) -...
    method initialize (line 105) | async def initialize(self) -> None:
    method add_chunks (line 110) | async def add_chunks(self, chunks: list[EmbeddedChunk]):
    method delete_chunks (line 173) | async def delete_chunks(self, chunks_for_deletion: list[ChunkForDeleti...
    method _results_to_chunks (line 194) | async def _results_to_chunks(self, results: dict) -> QueryChunksResponse:
    method query_vector (line 219) | async def query_vector(
    method query_keyword (line 241) | async def query_keyword(
    method query_hybrid (line 263) | async def query_hybrid(
    method delete (line 368) | async def delete(self):
  class ElasticsearchVectorIOAdapter (line 378) | class ElasticsearchVectorIOAdapter(OpenAIVectorStoreMixin, VectorIO, Vec...
    method __init__ (line 379) | def __init__(
    method initialize (line 392) | async def initialize(self) -> None:
    method shutdown (line 408) | async def shutdown(self) -> None:
    method register_vector_store (line 413) | async def register_vector_store(self, vector_store: VectorStore) -> None:
    method unregister_vector_store (line 426) | async def unregister_vector_store(self, vector_store_id: str) -> None:
    method _get_and_cache_vector_store_index (line 434) | async def _get_and_cache_vector_store_index(self, vector_store_id: str...
    method insert_chunks (line 453) | async def insert_chunks(self, request: InsertChunksRequest) -> None:
    method query_chunks (line 460) | async def query_chunks(self, request: QueryChunksRequest) -> QueryChun...
    method delete_chunks (line 467) | async def delete_chunks(self, request: DeleteChunksRequest) -> None:

FILE: src/llama_stack/providers/remote/vector_io/milvus/__init__.py
  function get_adapter_impl (line 12) | async def get_adapter_impl(config: MilvusVectorIOConfig, deps: dict[Api,...

FILE: src/llama_stack/providers/remote/vector_io/milvus/config.py
  class MilvusVectorIOConfig (line 16) | class MilvusVectorIOConfig(BaseModel):
    method sample_run_config (line 27) | def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict...

FILE: src/llama_stack/providers/remote/vector_io/milvus/milvus.py
  function _fmt (line 51) | def _fmt(v: Any) -> str:
  class MilvusIndex (line 70) | class MilvusIndex(EmbeddingIndex):
    method __init__ (line 71) | def __init__(
    method initialize (line 79) | async def initialize(self):
    method delete (line 84) | async def delete(self):
    method add_chunks (line 88) | async def add_chunks(self, chunks: list[EmbeddedChunk]):
    method _translate_filters (line 148) | def _translate_filters(self, filters: Filter | None) -> str | None:
    method _translate_single_filter (line 162) | def _translate_single_filter(self, filter_obj: Filter) -> str:
    method _translate_comparison_filter (line 171) | def _translate_comparison_filter(self, filter_obj: ComparisonFilter) -...
    method _translate_compound_filter (line 194) | def _translate_compound_filter(self, filter_obj: CompoundFilter) -> str:
    method query_vector (line 211) | async def query_vector(
    method query_keyword (line 238) | async def query_keyword(
    method _fallback_keyword_search (line 285) | async def _fallback_keyword_search(self, query_string: str, k: int, sc...
    method query_hybrid (line 302) | async def query_hybrid(
    method delete_chunks (line 367) | async def delete_chunks(self, chunks_for_deletion: list[ChunkForDeleti...
  class MilvusVectorIOAdapter (line 381) | class MilvusVectorIOAdapter(OpenAIVectorStoreMixin, VectorIO, VectorStor...
    method __init__ (line 382) | def __init__(
    method initialize (line 395) | async def initialize(self) -> None:
    method shutdown (line 425) | async def shutdown(self) -> None:
    method register_vector_store (line 430) | async def register_vector_store(self, vector_store: VectorStore) -> None:
    method _get_and_cache_vector_store_index (line 443) | async def _get_and_cache_vector_store_index(self, vector_store_id: str...
    method unregister_vector_store (line 465) | async def unregister_vector_store(self, vector_store_id: str) -> None:
    method insert_chunks (line 470) | async def insert_chunks(self, request: InsertChunksRequest) -> None:
    method query_chunks (line 477) | async def query_chunks(self, request: QueryChunksRequest) -> QueryChun...
    method delete_chunks (line 483) | async def delete_chunks(self, request: DeleteChunksRequest) -> None:

FILE: src/llama_stack/providers/remote/vector_io/oci/__init__.py
  function get_adapter_impl (line 11) | async def get_adapter_impl(config: OCI26aiVectorIOConfig, deps: dict[Api...

FILE: src/llama_stack/providers/remote/vector_io/oci/config.py
  class OCI26aiVectorIOConfig (line 16) | class OCI26aiVectorIOConfig(BaseModel):
    method sample_run_config (line 28) | def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict...

FILE: src/llama_stack/providers/remote/vector_io/oci/oci26ai.py
  function normalize_embedding (line 56) | def normalize_embedding(embedding: np.typing.NDArray) -> np.typing.NDArray:
  class OCI26aiIndex (line 74) | class OCI26aiIndex(EmbeddingIndex):
    method __init__ (line 75) | def __init__(
    method initialize (line 91) | async def initialize(self) -> None:
    method index_exists (line 113) | async def index_exists(self, index_name: str) -> bool:
    method create_indexes (line 129) | async def create_indexes(self):
    method add_chunks (line 164) | async def add_chunks(self, embedded_chunks: list[EmbeddedChunk]):
    method query_vector (line 216) | async def query_vector(
    method query_keyword (line 293) | async def query_keyword(
    method query_hybrid (line 358) | async def query_hybrid(
    method delete (line 422) | async def delete(self):
    method delete_chunks (line 431) | async def delete_chunks(self, chunks_for_deletion: list[ChunkForDeleti...
  class OCI26aiVectorIOAdapter (line 450) | class OCI26aiVectorIOAdapter(OpenAIVectorStoreMixin, VectorIO, VectorSto...
    method __init__ (line 451) | def __init__(
    method initialize (line 464) | async def initialize(self) -> None:
    method shutdown (line 504) | async def shutdown(self) -> None:
    method register_vector_store (line 511) | async def register_vector_store(self, vector_store: VectorStore) -> None:
    method _get_and_cache_vector_store_index (line 537) | async def _get_and_cache_vector_store_index(self, vector_store_id: str...
    method unregister_vector_store (line 564) | async def unregister_vector_store(self, vector_store_id: str) -> None:
    method insert_chunks (line 575) | async def insert_chunks(self, request: InsertChunksRequest) -> None:
    method query_chunks (line 582) | async def query_chunks(self, request: QueryChunksRequest) -> QueryChun...
    method delete_chunks (line 596) | async def delete_chunks(self, request: DeleteChunksRequest) -> None:

FILE: src/llama_stack/providers/remote/vector_io/pgvector/__init__.py
  function get_adapter_impl (line 12) | async def get_adapter_impl(config: PGVectorVectorIOConfig, deps: dict[Ap...

FILE: src/llama_stack/providers/remote/vector_io/pgvector/config.py
  class PGVectorIndexType (line 16) | class PGVectorIndexType(StrEnum):
  class PGVectorHNSWVectorIndex (line 23) | class PGVectorHNSWVectorIndex(BaseModel):
  class PGVectorIVFFlatVectorIndex (line 46) | class PGVectorIVFFlatVectorIndex(BaseModel):
    method validate_probes (line 62) | def validate_probes(self) -> Self:
  class PGVectorVectorIOConfig (line 77) | class PGVectorVectorIOConfig(BaseModel):
    method sample_run_config (line 95) | def sample_run_config(

FILE: src/llama_stack/providers/remote/vector_io/pgvector/pgvector.py
  function check_extension_version (line 69) | def check_extension_version(cur):
  function create_vector_extension (line 75) | def create_vector_extension(cur: cursor) -> None:
  function upsert_models (line 86) | def upsert_models(conn, keys_models: list[tuple[str, BaseModel]]):
  function remove_vector_store_metadata (line 101) | def remove_vector_store_metadata(conn: psycopg2.extensions.connection, v...
  function load_models (line 121) | def load_models(cur, cls):
  class PGVectorIndex (line 127) | class PGVectorIndex(EmbeddingIndex):
    method __init__ (line 150) | def __init__(
    method initialize (line 168) | async def initialize(self) -> None:
    method add_chunks (line 220) | async def add_chunks(self, chunks: list[EmbeddedChunk]):
    method query_vector (line 258) | async def query_vector(
    method query_keyword (line 321) | async def query_keyword(
    method query_hybrid (line 369) | async def query_hybrid(
    method _translate_filters (line 433) | def _translate_filters(self, filters: ComparisonFilter | CompoundFilte...
    method _translate_single_filter (line 448) | def _translate_single_filter(self, filter_obj: ComparisonFilter | Comp...
    method _translate_comparison_filter (line 457) | def _translate_comparison_filter(self, filter_obj: ComparisonFilter) -...
    method _translate_compound_filter (line 490) | def _translate_compound_filter(self, filter_obj: CompoundFilter) -> tu...
    method delete (line 510) | async def delete(self):
    method delete_chunks (line 514) | async def delete_chunks(self, chunks_for_deletion: list[ChunkForDeleti...
    method get_pgvector_index_operator_class (line 521) | def get_pgvector_index_operator_class(self) -> str:
    method get_pgvector_search_function (line 529) | def get_pgvector_search_function(self) -> str:
    method check_distance_metric_availability (line 532) | def check_distance_metric_availability(self, distance_metric: str) -> ...
    method create_hnsw_vector_index (line 548) | async def create_hnsw_vector_index(self, cur: cursor) -> None:
    method create_ivfflat_vector_index (line 590) | async def create_ivfflat_vector_index(self, cur: cursor) -> None:
    method check_conflicting_vector_index_exists (line 640) | async def check_conflicting_vector_index_exists(self, cur: cursor) -> ...
    method fetch_number_of_records (line 685) | async def fetch_number_of_records(self, cur: cursor) -> int:
    method create_gin_index (line 719) | async def create_gin_index(self, cur: cursor) -> None:
  class PGVectorVectorIOAdapter (line 749) | class PGVectorVectorIOAdapter(OpenAIVectorStoreMixin, VectorIO, VectorSt...
    method __init__ (line 750) | def __init__(
    method initialize (line 760) | async def initialize(self) -> None:
    method shutdown (line 813) | async def shutdown(self) -> None:
    method register_vector_store (line 820) | async def register_vector_store(self, vector_store: VectorStore) -> None:
    method unregister_vector_store (line 845) | async def unregister_vector_store(self, vector_store_id: str) -> None:
    method insert_chunks (line 859) | async def insert_chunks(self, request: InsertChunksRequest) -> None:
    method query_chunks (line 863) | async def query_chunks(self, request: QueryChunksRequest) -> QueryChun...
    method _get_and_cache_vector_store_index (line 867) | async def _get_and_cache_vector_store_index(self, vector_store_id: str...
    method delete_chunks (line 892) | async def delete_chunks(self, request: DeleteChunksRequest) -> None:

FILE: src/llama_stack/providers/remote/vector_io/qdrant/__init__.py
  function get_adapter_impl (line 12) | async def get_adapter_impl(config: QdrantVectorIOConfig, deps: dict[Api,...

FILE: src/llama_stack/providers/remote/vector_io/qdrant/config.py
  class QdrantVectorIOConfig (line 16) | class QdrantVectorIOConfig(BaseModel):
    method sample_run_config (line 30) | def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict...

FILE: src/llama_stack/providers/remote/vector_io/qdrant/qdrant.py
  function convert_id (line 51) | def convert_id(_id: str) -> str:
  class QdrantIndex (line 65) | class QdrantIndex(EmbeddingIndex):
    method __init__ (line 66) | def __init__(self, client: AsyncQdrantClient, collection_name: str):
    method initialize (line 70) | async def initialize(self) -> None:
    method add_chunks (line 75) | async def add_chunks(self, chunks: list[EmbeddedChunk]):
    method delete_chunks (line 115) | async def delete_chunks(self, chunks_for_deletion: list[ChunkForDeleti...
    method query_vector (line 126) | async def query_vector(
    method query_keyword (line 159) | async def query_keyword(
    method query_hybrid (line 233) | async def query_hybrid(
    method delete (line 321) | async def delete(self):
  class QdrantVectorIOAdapter (line 325) | class QdrantVectorIOAdapter(OpenAIVectorStoreMixin, VectorIO, VectorStor...
    method __init__ (line 326) | def __init__(
    method initialize (line 339) | async def initialize(self) -> None:
    method shutdown (line 356) | async def shutdown(self) -> None:
    method register_vector_store (line 361) | async def register_vector_store(self, vector_store: VectorStore) -> None:
    method unregister_vector_store (line 375) | async def unregister_vector_store(self, vector_store_id: str) -> None:
    method _get_and_cache_vector_store_index (line 384) | async def _get_and_cache_vector_store_index(self, vector_store_id: str...
    method insert_chunks (line 406) | async def insert_chunks(self, request: InsertChunksRequest) -> None:
    method query_chunks (line 413) | async def query_chunks(self, request: QueryChunksRequest) -> QueryChun...
    method openai_attach_file_to_vector_store (line 420) | async def openai_attach_file_to_vector_store(
    method delete_chunks (line 429) | async def delete_chunks(self, request: DeleteChunksRequest) -> None:

FILE: src/llama_stack/providers/remote/vector_io/weaviate/__init__.py
  function get_adapter_impl (line 12) | async def get_adapter_impl(config: WeaviateVectorIOConfig, deps: dict[Ap...

FILE: src/llama_stack/providers/remote/vector_io/weaviate/config.py
  class WeaviateVectorIOConfig (line 16) | class WeaviateVectorIOConfig(BaseModel):
    method sample_run_config (line 24) | def sample_run_config(cls, __distro_dir__: str, **kwargs: Any) -> dict...

FILE: src/llama_stack/providers/remote/vector_io/weaviate/weaviate.py
  class WeaviateIndex (line 55) | class WeaviateIndex(EmbeddingIndex):
    method __init__ (line 56) | def __init__(self, client: weaviate.WeaviateClient, collection_name: s...
    method initialize (line 61) | async def initialize(self):
    method add_chunks (line 64) | async def add_chunks(self, chunks: list[EmbeddedChunk]):
    method delete_chunks (line 86) | async def delete_chunks(self, chunks_for_deletion: list[ChunkForDeleti...
    method query_vector (line 92) | async def query_vector(
    method delete (line 147) | async def delete(self, chunk_ids: list[str] | None = None) -> None:
    method query_keyword (line 160) | async def query_keyword(
    method query_hybrid (line 214) | async def query_hybrid(
  class WeaviateVectorIOAdapter (line 289) | class WeaviateVectorIOAdapter(OpenAIVectorStoreMixin, VectorIO, VectorSt...
    method __init__ (line 290) | def __init__(self, config: WeaviateVectorIOConfig, inference_api: Infe...
    method _get_client (line 298) | def _get_client(self) -> weaviate.WeaviateClient:
    method initialize (line 316) | async def initialize(self) -> None:
    method shutdown (line 341) | async def shutdown(self) -> None:
    method register_vector_store (line 347) | async def register_vector_store(self, vector_store: VectorStore) -> None:
    method unregister_vector_store (line 364) | async def unregister_vector_store(self, vector_store_id: str) -> None:
    method _get_and_cache_vector_store_index (line 373) | async def _get_and_cache_vector_store_index(self, vector_store_id: str...
    method insert_chunks (line 400) | async def insert_chunks(self, request: InsertChunksRequest) -> None:
    method query_chunks (line 407) | async def query_chunks(self, request: QueryChunksRequest) -> QueryChun...
    method delete_chunks (line 414) | async def delete_chunks(self, request: DeleteChunksRequest) -> None:

FILE: src/llama_stack/providers/utils/bedrock/client.py
  function create_bedrock_client (line 18) | def create_bedrock_client(config: BedrockBaseConfig, service_name: str =...

FILE: src/llama_stack/providers/utils/bedrock/config.py
  class BedrockBaseConfig (line 14) | class BedrockBaseConfig(RemoteInferenceProviderConfig):
    method sample_run_config (line 63) | def sample_run_config(cls, **kwargs):

FILE: src/llama_stack/providers/utils/bedrock/refreshable_boto_session.py
  class RefreshableBotoSession (line 16) | class RefreshableBotoSession:
    method __init__ (line 27) | def __init__(
    method __get_session_credentials (line 63) | def __get_session_credentials(self):
    method refreshable_session (line 95) | def refreshable_session(self) -> Session:

FILE: src/llama_stack/providers/utils/common/data_schema_validator.py
  class ColumnName (line 14) | class ColumnName(Enum):
  function get_valid_schemas (line 74) | def get_valid_schemas(api_str: str):
  function validate_dataset_schema (line 83) | def validate_dataset_schema(
  function validate_row_schema (line 91) | def validate_row_schema(

FILE: src/llama_stack/providers/utils/common/data_url.py
  function parse_data_url (line 10) | def parse_data_url(data_url: str):

FILE: src/llama_stack/providers/utils/datasetio/url_utils.py
  function get_dataframe_from_uri (line 15) | async def get_dataframe_from_uri(uri: str):

FILE: src/llama_stack/providers/utils/files/form_data.py
  function parse_pydantic_from_form (line 15) | async def parse_pydantic_from_form[T: BaseModel](request: Request, field...
  function parse_expires_after (line 63) | async def parse_expires_after(request: Request) -> ExpiresAfter | None:

FILE: src/llama_stack/providers/utils/files/sanitize.py
  function sanitize_content_disposition_filename (line 12) | def sanitize_content_disposition_filename(filename: str) -> str:

FILE: src/llama_stack/providers/utils/inference/embedding_mixin.py
  class SentenceTransformerEmbeddingMixin (line 36) | class SentenceTransformerEmbeddingMixin:
    method openai_embeddings (line 39) | async def openai_embeddings(
    method _load_sentence_transformer_model (line 81) | async def _load_sentence_transformer_model(self, model: str) -> "Sente...

FILE: src/llama_stack/providers/utils/inference/http_client.py
  function _build_ssl_context (line 25) | def _build_ssl_context(tls_config: TLSConfig) -> ssl.SSLContext | bool |...
  function _build_proxy_mounts (line 64) | def _build_proxy_mounts(proxy_config: ProxyConfig) -> dict[str, httpx.As...
  function _build_network_client_kwargs (line 93) | def _build_network_client_kwargs(network_config: NetworkConfig | None) -...
  function _extract_client_config (line 140) | def _extract_client_config(existing_client: httpx.AsyncClient | DefaultA...
  function _merge_network_config_into_client (line 169) | def _merge_network_config_into_client(
  function build_http_client (line 221) | def build_http_client(network_config: NetworkConfig | None) -> dict[str,...

FILE: src/llama_stack/providers/utils/inference/inference_store.py
  class _WriteItem (line 34) | class _WriteItem(NamedTuple):
  class InferenceStore (line 40) | class InferenceStore:
    method __init__ (line 41) | def __init__(
    method initialize (line 57) | async def initialize(self):
    method shutdown (line 80) | async def shutdown(self) -> None:
    method flush (line 95) | async def flush(self) -> None:
    method _ensure_workers_started (line 100) | async def _ensure_workers_started(self) -> None:
    method store_chat_completion (line 117) | async def store_chat_completion(
    method _worker_loop (line 135) | async def _worker_loop(self) -> None:
    method _write_chat_completion (line 150) | async def _write_chat_completion(
    method _is_unique_constraint_error (line 184) | def _is_unique_constraint_error(self, error_message: str) -> bool:
    method list_chat_completions (line 197) | async def list_chat_completions(
    method get_chat_completion (line 247) | async def get_chat_completion(self, completion_id: str) -> OpenAICompl...

FILE: src/llama_stack/providers/utils/inference/model_registry.py
  class TLSConfig (line 21) | class TLSConfig(BaseModel):
    method validate_verify (line 47) | def validate_verify(cls, v: bool | str | Path) -> bool | Path:
    method validate_cert_paths (line 62) | def validate_cert_paths(cls, v: str | Path | None) -> Path | None:
    method validate_mtls_pair (line 76) | def validate_mtls_pair(self) -> "TLSConfig":
  class ProxyConfig (line 82) | class ProxyConfig(BaseModel):
    method validate_cacert (line 108) | def validate_cacert(cls, v: str | Path | None) -> Path | None:
    method validate_proxy_config (line 122) | def validate_proxy_config(self) -> "ProxyConfig":
  class TimeoutConfig (line 128) | class TimeoutConfig(BaseModel):
  class NetworkConfig (line 141) | class NetworkConfig(BaseModel):
  class RemoteInferenceProviderConfig (line 162) | class RemoteInferenceProviderConfig(BaseModel):
  class ProviderModelEntry (line 184) | class ProviderModelEntry(BaseModel):
  function build_hf_repo_model_entry (line 192) | def build_hf_repo_model_entry(
  class ModelRegistryHelper (line 211) | class ModelRegistryHelper(ModelsProtocolPrivate):
    method __init__ (line 214) | def __init__(
    method list_models (line 235) | async def list_models(self) -> list[Model] | None:
    method should_refresh_models (line 253) | async def should_refresh_models(self) -> bool:
    method get_provider_model_id (line 256) | def get_provider_model_id(self, identifier: str) -> str | None:
    method get_llama_model (line 260) | def get_llama_model(self, provider_model_id: str) -> str | None:
    method check_model_availability (line 263) | async def check_model_availability(self, model: str) -> bool:
    method register_model (line 283) | async def register_model(self, model: Model) -> Model:
    method unregister_model (line 330) | async def unregister_model(self, model_id: str) -> None:

FILE: src/llama_stack/providers/utils/inference/openai_compat.py
  function convert_tooldef_to_openai_tool (line 17) | def convert_tooldef_to_openai_tool(
  function prepare_openai_completion_params (line 60) | async def prepare_openai_completion_params(**params):
  function get_stream_options_for_telemetry (line 78) | def get_stream_options_for_telemetry(

FILE: src/llama_stack/providers/utils/inference/openai_mixin.py
  class OpenAIMixin (line 49) | class OpenAIMixin(NeedsRequestProviderData, ABC, BaseModel):
    method get_api_key (line 114) | def get_api_key(self) -> str | None:
    method get_base_url (line 125) | def get_base_url(self) -> str:
    method get_extra_client_params (line 136) | def get_extra_client_params(self) -> dict[str, Any]:
    method construct_model_from_identifier (line 150) | def construct_model_from_identifier(self, identifier: str) -> Model:
    method list_provider_model_ids (line 174) | async def list_provider_model_ids(self) -> Iterable[str]:
    method initialize (line 189) | async def initialize(self) -> None:
    method shutdown (line 199) | async def shutdown(self) -> None:
    method client (line 210) | def client(self) -> AsyncOpenAI:
    method _get_api_key_from_config_or_provider_data (line 258) | def _get_api_key_from_config_or_provider_data(self) -> str | None:
    method _validate_model_allowed (line 269) | def _validate_model_allowed(self, provider_model_id: str) -> None:
    method _get_provider_model_id (line 282) | async def _get_provider_model_id(self, model: str) -> str:
    method _maybe_overwrite_id (line 303) | async def _maybe_overwrite_id(self, resp: Any, stream: bool | None) ->...
    method openai_completion (line 319) | async def openai_completion(
    method openai_chat_completion (line 360) | async def openai_chat_completion(
    method openai_embeddings (line 431) | async def openai_embeddings(
    method register_model (line 492) | async def register_model(self, model: Model) -> Model:
    method unregister_model (line 504) | async def unregister_model(self, model_id: str) -> None:
    method list_models (line 507) | async def list_models(self) -> list[Model] | None:
    method check_model_availability (line 547) | async def check_model_availability(self, model: str) -> bool:
    method should_refresh_models (line 565) | async def should_refresh_models(self) -> bool:
    method _filter_fields (line 573) | def _filter_fields(self, **kwargs):
    method model_dump (line 584) | def model_dump(self, **kwargs):
    method model_dump_json (line 589) | def model_dump_json(self, **kwargs):

FILE: src/llama_stack/providers/utils/inference/prompt_adapter.py
  function interleaved_content_as_str (line 25) | def interleaved_content_as_str(
  function localize_image_content (line 50) | async def localize_image_content(uri: str) -> tuple[bytes, str] | None:

FILE: src/llama_stack/providers/utils/inference/stream_utils.py
  function wrap_async_stream (line 14) | async def wrap_async_stream[T](stream: AsyncIterator[T]) -> AsyncIterato...

FILE: src/llama_stack/providers/utils/memory/file_utils.py
  function data_url_from_file (line 14) | def data_url_from_file(file_path: str) -> URL:

FILE: src/llama_stack/providers/utils/memory/openai_vector_store_mixin.py
  function _is_retriable_error (line 109) | def _is_retriable_error(exc: Exception) -> bool:
  class _ChunkContextResult (line 116) | class _ChunkContextResult(StrEnum):
  class OpenAIVectorStoreMixin (line 124) | class OpenAIVectorStoreMixin(ABC):
    method __init__ (line 133) | def __init__(
    method _get_vector_store_lock (line 155) | def _get_vector_store_lock(self, vector_store_id: str) -> asyncio.Lock:
    method _save_openai_vector_store (line 161) | async def _save_openai_vector_store(self, store_id: str, store_info: d...
    method _ensure_openai_metadata_exists (line 169) | async def _ensure_openai_metadata_exists(self, vector_store: VectorSto...
    method _load_openai_vector_stores (line 202) | async def _load_openai_vector_stores(self) -> dict[str, dict[str, Any]]:
    method _update_openai_vector_store (line 215) | async def _update_openai_vector_store(self, store_id: str, store_info:...
    method _delete_openai_vector_store_from_storage (line 223) | async def _delete_openai_vector_store_from_storage(self, store_id: str...
    method _save_openai_vector_store_file (line 231) | async def _save_openai_vector_store_file(
    method _load_openai_vector_store_file (line 246) | async def _load_openai_vector_store_file(self, store_id: str, file_id:...
    method _load_openai_vector_store_file_contents (line 253) | async def _load_openai_vector_store_file_contents(self, store_id: str,...
    method _update_openai_vector_store_file (line 261) | async def _update_openai_vector_store_file(self, store_id: str, file_i...
    method _delete_openai_vector_store_file_from_storage (line 267) | async def _delete_openai_vector_store_file_from_storage(self, store_id...
    method _save_openai_vector_store_file_batch (line 282) | async def _save_openai_vector_store_file_batch(self, batch_id: str, ba...
    method _load_openai_vector_store_file_batches (line 290) | async def _load_openai_vector_store_file_batches(self) -> dict[str, di...
    method _delete_openai_vector_store_file_batch (line 303) | async def _delete_openai_vector_store_file_batch(self, batch_id: str) ...
    method _cleanup_expired_file_batches (line 311) | async def _cleanup_expired_file_batches(self) -> None:
    method _get_completed_files_in_batch (line 334) | async def _get_completed_files_in_batch(self, vector_store_id: str, fi...
    method _analyze_batch_completion_on_resume (line 343) | async def _analyze_batch_completion_on_resume(self, batch_id: str, bat...
    method _resume_incomplete_batches (line 379) | async def _resume_incomplete_batches(self) -> None:
    method initialize_openai_vector_stores (line 397) | async def initialize_openai_vector_stores(self) -> None:
    method shutdown (line 411) | async def shutdown(self) -> None:
    method delete_chunks (line 424) | async def delete_chunks(self, request: DeleteChunksRequest) -> None:
    method register_vector_store (line 429) | async def register_vector_store(self, vector_store: VectorStore) -> None:
    method unregister_vector_store (line 434) | async def unregister_vector_store(self, vector_store_id: str) -> None:
    method insert_chunks (line 439) | async def insert_chunks(
    method query_chunks (line 447) | async def query_chunks(
    method openai_create_vector_store (line 454) | async def openai_create_vector_store(
    method openai_list_vector_stores (line 569) | async def openai_list_vector_stores(
    method openai_retrieve_vector_store (line 617) | async def openai_retrieve_vector_store(
    method openai_update_vector_store (line 628) | async def openai_update_vector_store(
    method openai_delete_vector_store (line 658) | async def openai_delete_vector_store(
    method openai_search_vector_store (line 683) | async def openai_search_vector_store(
    method _build_reranker_params (line 784) | def _build_reranker_params(
    method _chunk_to_vector_store_content (line 840) | def _chunk_to_vector_store_content(
    method openai_attach_file_to_vector_store (line 871) | async def openai_attach_file_to_vector_store(
    method _legacy_chunk_file (line 1073) | async def _legacy_chunk_file(
    method openai_list_files_in_vector_store (line 1102) | async def openai_list_files_in_vector_store(
    method openai_retrieve_vector_store_file (line 1160) | async def openai_retrieve_vector_store_file(
    method openai_retrieve_vector_store_file_contents (line 1176) | async def openai_retrieve_vector_store_file_contents(
    method openai_update_vector_store_file (line 1203) | async def openai_update_vector_store_file(
    method openai_delete_vector_store_file (line 1222) | async def openai_delete_vector_store_file(
    method openai_create_vector_store_file_batch (line 1273) | async def openai_create_vector_store_file_batch(
    method _process_files_with_concurrency (line 1332) | async def _process_files_with_concurrency(
    method _process_file_batch_async (line 1383) | async def _process_file_batch_async(
    method openai_retrieve_vector_store_file_batch (line 1434) | async def openai_retrieve_vector_store_file_batch(
    method openai_list_files_in_vector_store_file_batch (line 1462) | async def openai_list_files_in_vector_store_file_batch(
    method openai_cancel_vector_store_file_batch (line 1538) | async def openai_cancel_vector_store_file_batch(
    method _execute_contextual_chunk_transformation (line 1573) | async def _execute_contextual_chunk_transformation(

FILE: src/llama_stack/providers/utils/memory/vector_store.py
  function _get_numpy (line 27) | def _get_numpy() -> Any:
  function _get_encoding (line 67) | def _get_encoding(name: str) -> tiktoken.Encoding:
  function parse_pdf (line 77) | def parse_pdf(data: bytes) -> str:
  function content_from_data_and_mime_type (line 84) | def content_from_data_and_mime_type(data: bytes | str, mime_type: str | ...
  function make_overlapped_chunks (line 114) | def make_overlapped_chunks(
  function _validate_embedding (line 184) | def _validate_embedding(embedding: EmbeddingSequence, index: int, expect...
  class EmbeddingIndex (line 201) | class EmbeddingIndex(ABC):
    method add_chunks (line 203) | async def add_chunks(self, embedded_chunks: list[EmbeddedChunk]):
    method delete_chunks (line 207) | async def delete_chunks(self, chunks_for_deletion: list[ChunkForDeleti...
    method query_vector (line 211) | async def query_vector(
    method query_keyword (line 217) | async def query_keyword(
    method query_hybrid (line 223) | async def query_hybrid(
    method delete (line 236) | async def delete(self):
  class VectorStoreWithIndex (line 241) | class VectorStoreWithIndex:
    method insert_chunks (line 248) | async def insert_chunks(
    method query_chunks (line 258) | async def query_chunks(
    method apply_neural_rerank (line 359) | async def apply_neural_rerank(

FILE: src/llama_stack/providers/utils/pagination.py
  function paginate_records (line 12) | def paginate_records(

FILE: src/llama_stack/providers/utils/responses/responses_store.py
  class _OpenAIResponseObjectWithInputAndMessages (line 30) | class _OpenAIResponseObjectWithInputAndMessages(OpenAIResponseObjectWith...
  class ResponsesStore (line 43) | class ResponsesStore:
    method __init__ (line 44) | def __init__(
    method initialize (line 56) | async def initialize(self):
    method shutdown (line 79) | async def shutdown(self) -> None:
    method flush (line 82) | async def flush(self) -> None:
    method store_response_object (line 86) | async def store_response_object(
    method upsert_response_object (line 94) | async def upsert_response_object(
    method _write_response_object (line 126) | async def _write_response_object(
    method list_responses (line 146) | async def list_responses(
    method get_response_object (line 185) | async def get_response_object(self, response_id: str) -> _OpenAIRespon...
    method delete_response_object (line 202) | async def delete_response_object(self, response_id: str) -> OpenAIDele...
    method update_response_object (line 209) | async def update_response_object(
    method list_response_input_items (line 251) | async def list_response_input_items(
    method store_conversation_messages (line 310) | async def store_conversation_messages(self, conversation_id: str, mess...
    method get_conversation_messages (line 329) | async def get_conversation_messages(self, conversation_id: str) -> lis...

FILE: src/llama_stack/providers/utils/safety.py
  class _RunShieldProtocol (line 21) | class _RunShieldProtocol:
    method run_shield (line 22) | async def run_shield(self, request: RunShieldRequest) -> RunShieldResp...
  class ShieldToModerationMixin (line 25) | class ShieldToModerationMixin:
    method run_shield (line 35) | async def run_shield(self, request: RunShieldRequest) -> RunShieldResp...
    method run_moderation (line 37) | async def run_moderation(self, request: RunModerationRequest) -> Moder...
    method _shield_response_to_moderation_result (line 76) | def _shield_response_to_moderation_result(

FILE: src/llama_stack/providers/utils/scheduler.py
  class JobStatus (line 25) | class JobStatus(Enum):
  class JobArtifact (line 37) | class JobArtifact(BaseModel):
  class Job (line 56) | class Job:
    method __init__ (line 57) | def __init__(self, job_type: JobType, job_id: JobID, handler: JobHandl...
    method handler (line 67) | def handler(self) -> JobHandler:
    method status (line 71) | def status(self) -> JobStatus:
    method status (line 75) | def status(self, status: JobStatus):
    method artifacts (line 83) | def artifacts(self) -> list[JobArtifact]:
    method register_artifact (line 86) | def register_artifact(self, artifact: JobArtifact) -> None:
    method _find_state_transition_date (line 89) | def _find_state_transition_date(self, status: Iterable[JobStatus]) -> ...
    method scheduled_at (line 96) | def scheduled_at(self) -> datetime | None:
    method started_at (line 100) | def started_at(self) -> datetime | None:
    method completed_at (line 104) | def completed_at(self) -> datetime | None:
    method logs (line 108) | def logs(self) -> list[LogMessage]:
    method append_log (line 111) | def append_log(self, message: LogMessage) -> None:
    method cancel (line 115) | def cancel(self) -> None:
  class _SchedulerBackend (line 119) | class _SchedulerBackend(abc.ABC):
    method on_log_message_cb (line 121) | def on_log_message_cb(self, job: Job, message: LogMessage) -> None:
    method on_status_change_cb (line 125) | def on_status_change_cb(self, job: Job, status: JobStatus) -> None:
    method on_artifact_collected_cb (line 129) | def on_artifact_collected_cb(self, job: Job, artifact: JobArtifact) ->...
    method shutdown (line 133) | async def shutdown(self) -> None:
    method schedule (line 137) | def schedule(
  class _NaiveSchedulerBackend (line 147) | class _NaiveSchedulerBackend(_SchedulerBackend):
    method __init__ (line 148) | def __init__(self, timeout: int = 5):
    method _run_loop (line 156) | def _run_loop(self) -> None:
    method shutdown (line 170) | async def shutdown(self) -> None:
    method schedule (line 175) | def schedule(
    method on_log_message_cb (line 193) | def on_log_message_cb(self, job: Job, message: LogMessage) -> None:
    method on_status_change_cb (line 196) | def on_status_change_cb(self, job: Job, status: JobStatus) -> None:
    method on_artifact_collected_cb (line 199) | def on_artifact_collected_cb(self, job: Job, artifact: JobArtifact) ->...
  function _get_backend_impl (line 208) | def _get_backend_impl(backend: str) -> _SchedulerBackend:
  class Scheduler (line 215) | class Scheduler:
    method __init__ (line 216) | def __init__(self, backend: str = "naive"):
    method _on_log_message_cb (line 221) | def _on_log_message_cb(self, job: Job, message: str) -> None:
    method _on_status_change_cb (line 229) | def _on_status_change_cb(self, job: Job, status: JobStatus) -> None:
    method _on_artifact_collected_cb (line 233) | def _on_artifact_collected_cb(self, job: Job, artifact: JobArtifact) -...
    method schedule (line 237) | def schedule(self, type_: JobType, job_id: JobID, handler: JobHandler)...
    method cancel (line 253) | def cancel(self, job_id: JobID) -> None:
    method get_job (line 256) | def get_job(self, job_id: JobID) -> Job:
    method get_jobs (line 262) | def get_jobs(self, type_: JobType | None = None) -> list[Job]:
    method shutdown (line 268) | async def shutdown(self):

FILE: src/llama_stack/providers/utils/scoring/aggregation_utils.py
  function aggregate_accuracy (line 12) | def aggregate_accuracy(scoring_results: list[ScoringResultRow]) -> dict[...
  function aggregate_average (line 23) | def aggregate_average(scoring_results: list[ScoringResultRow]) -> dict[s...
  function aggregate_weighted_average (line 30) | def aggregate_weighted_average(scoring_results: list[ScoringResultRow]) ...
  function aggregate_categorical_count (line 41) | def aggregate_categorical_count(
  function aggregate_median (line 49) | def aggregate_median(scoring_results: list[ScoringResultRow]) -> dict[st...
  function aggregate_metrics (line 65) | def aggregate_metrics(

FILE: src/llama_stack/providers/utils/scoring/base_scoring_fn.py
  class BaseScoringFn (line 13) | class BaseScoringFn(ABC):
    method __init__ (line 21) | def __init__(self, *args, **kwargs) -> None:
    method __str__ (line 24) | def __str__(self) -> str:
    method score_row (line 28) | async def score_row(
    method aggregate (line 37) | async def aggregate(
    method score (line 46) | async def score(
  class RegisteredBaseScoringFn (line 55) | class RegisteredBaseScoringFn(BaseScoringFn):
    method __init__ (line 60) | def __init__(self, *args, **kwargs) -> None:
    method __str__ (line 64) | def __str__(self) -> str:
    method get_supported_scoring_fn_defs (line 67) | def get_supported_scoring_fn_defs(self) -> list[ScoringFn]:
    method register_scoring_fn_def (line 70) | def register_scoring_fn_def(self, scoring_fn: ScoringFn) -> None:
    method unregister_scoring_fn_def (line 75) | def unregister_scoring_fn_def(self, scoring_fn_id: str) -> None:
    method score_row (line 81) | async def score_row(
    method aggregate (line 89) | async def aggregate(
    method score (line 107) | async def score(

FILE: src/llama_stack/providers/utils/scoring/basic_scoring_utils.py
  class TimeoutError (line 12) | class TimeoutError(Exception):
  function time_limit (line 17) | def time_limit(seconds: float) -> Iterator[None]:

FILE: src/llama_stack/providers/utils/tools/mcp.py
  function prepare_mcp_headers (line 37) | def prepare_mcp_headers(base_headers: dict[str, str] | None, authorizati...
  class MCPProtol (line 73) | class MCPProtol(Enum):
  class MCPSessionManager (line 79) | class MCPSessionManager:
    method __init__ (line 96) | def __init__(self):
    method _make_key (line 104) | def _make_key(self, endpoint: str, headers: dict[str, str]) -> str:
    method _get_lock (line 111) | async def _get_lock(self, key: str) -> asyncio.Lock:
    method get_session (line 118) | async def get_session(self, endpoint: str, headers: dict[str, str]) ->...
    method _create_session (line 149) | async def _create_session(self, endpoint: str, headers: dict[str, str]...
    method __aenter__ (line 245) | async def __aenter__(self):
    method __aexit__ (line 249) | async def __aexit__(self, exc_type, exc_val, exc_tb):
  function client_wrapper (line 285) | async def client_wrapper(endpoint: str, headers: dict[str, str]) -> Asyn...
  function list_mcp_tools (line 358) | async def list_mcp_tools(
  function _parse_mcp_result (line 414) | def _parse_mcp_result(result) -> ToolInvocationResult:
  function invoke_mcp_tool (line 439) | async def invoke_mcp_tool(
  class MCPServerInfo (line 481) | class MCPServerInfo:
  function get_mcp_server_info (line 490) | async def get_mcp_server_info(

FILE: src/llama_stack/providers/utils/tools/ttl_dict.py
  class TTLDict (line 12) | class TTLDict(dict):
    method __init__ (line 17) | def __init__(self, ttl_seconds: float, *args, **kwargs):
    method __delitem__ (line 27) | def __delitem__(self, key):
    method __setitem__ (line 32) | def __setitem__(self, key, value):
    method _is_expired (line 37) | def _is_expired(self, key):
    method __getitem__ (line 42) | def __getitem__(self, key):
    method get (line 51) | def get(self, key, default=None):
    method __contains__ (line 57) | def __contains__(self, key):
    method __repr__ (line 64) | def __repr__(self):

FILE: src/llama_stack/providers/utils/vector_io/filters.py
  function parse_filter (line 21) | def parse_filter(filter_data: Any) -> ComparisonFilter | CompoundFilter:

FILE: src/llama_stack/providers/utils/vector_io/vector_utils.py
  function generate_chunk_id (line 15) | def generate_chunk_id(document_id: str, chunk_text: str, chunk_window: s...
  function proper_case (line 26) | def proper_case(s: str) -> str:
  function sanitize_collection_name (line 31) | def sanitize_collection_name(name: str, weaviate_format=False) -> str:
  class WeightedInMemoryAggregator (line 43) | class WeightedInMemoryAggregator:
    method _normalize_scores (line 45) | def _normalize_scores(scores: dict[str, float]) -> dict[str, float]:
    method weighted_rerank (line 64) | def weighted_rerank(
    method rrf_rerank (line 94) | def rrf_rerank(
    method combine_search_results (line 132) | def combine_search_results(
  function load_embedded_chunk_with_backward_compat (line 162) | def load_embedded_chunk_with_backward_compat(

FILE: src/llama_stack/telemetry/__init__.py
  function setup_telemetry (line 20) | def setup_telemetry():

FILE: src/llama_stack/telemetry/helpers.py
  function safety_span_name (line 23) | def safety_span_name(shield_id: str) -> str:
  function safety_request_span_attributes (line 29) | def safety_request_span_attributes(

FILE: src/llama_stack/telemetry/tool_runtime_metrics.py
  function create_tool_metric_attributes (line 48) | def create_tool_metric_attributes(

FILE: src/llama_stack/telemetry/vector_io_metrics.py
  function create_vector_metric_attributes (line 91) | def create_vector_metric_attributes(

FILE: src/llama_stack/testing/api_recorder.py
  class APIRecordingMode (line 53) | class APIRecordingMode(StrEnum):
  function _normalize_numeric_literal_strings (line 71) | def _normalize_numeric_literal_strings(value: str) -> str:
  function _normalize_body_for_hash (line 81) | def _normalize_body_for_hash(value: Any, exclude_stream_options: bool = ...
  function _allocate_test_scoped_id (line 109) | def _allocate_test_scoped_id(kind: str) -> str | None:
  function _deterministic_id_override (line 140) | def _deterministic_id_override(kind: str, factory: Callable[[], str]) ->...
  function normalize_inference_request (line 147) | def normalize_inference_request(method: str, url: str, headers: dict[str...
  function normalize_tool_request (line 192) | def normalize_tool_request(provider_name: str, tool_name: str, kwargs: d...
  function normalize_http_request (line 205) | def normalize_http_request(url: str, method: str, payload: dict[str, Any...
  function patch_httpx_for_test_id (line 245) | def patch_httpx_for_test_id():
  function unpatch_httpx_for_test_id (line 296) | def unpatch_httpx_for_test_id():
  function get_api_recording_mode (line 309) | def get_api_recording_mode() -> APIRecordingMode:
  function setup_api_recording (line 313) | def setup_api_recording():
  function _normalize_response (line 340) | def _normalize_response(data: dict[str, Any], request_hash: str) -> dict...
  function _serialize_response (line 372) | def _serialize_response(response: Any, request_hash: str = "") -> Any:
  function _deserialize_response (line 387) | def _deserialize_response(data: dict[str, Any]) -> Any:
  class ResponseStorage (line 411) | class ResponseStorage:
    method __init__ (line 414) | def __init__(self, base_dir: Path):
    method _get_test_dir (line 418) | def _get_test_dir(self) -> Path:
    method _ensure_directory (line 462) | def _ensure_directory(self):
    method store_recording (line 468) | def store_recording(self, request_hash: str, request: dict[str, Any], ...
    method find_recording (line 517) | def find_recording(self, request_hash: str) -> dict[str, Any] | None:
    method _model_list_responses (line 559) | def _model_list_responses(self, request_hash: str) -> list[dict[str, A...
  function _recording_from_file (line 580) | def _recording_from_file(response_path) -> dict[str, Any]:
  function _model_identifiers_digest (line 604) | def _model_identifiers_digest(endpoint: str, response: dict[str, Any]) -...
  function _combine_model_list_responses (line 629) | def _combine_model_list_responses(endpoint: str, records: list[dict[str,...
  function _patched_tool_invoke_method (line 667) | async def _patched_tool_invoke_method(
  function _patched_aiohttp_post (line 711) | def _patched_aiohttp_post(original_post, session_self, url: str, **kwargs):
  function _extract_provider_metadata (line 817) | def _extract_provider_metadata(client: Any, client_type: str, base_url: ...
  function _patched_inference_method (line 874) | async def _patched_inference_method(original_method, self, client_type, ...
  function patch_inference_clients (line 1046) | def patch_inference_clients():
  function unpatch_inference_clients (line 1172) | def unpatch_inference_clients():
  function api_recording (line 1215) | def api_recording(mode: str, storage_dir: str | Path | None = None) -> G...

FILE: src/llama_stack/testing/exception_utils.py
  class ProviderSDKException (line 33) | class ProviderSDKException(Protocol):
  class GenericLlamaStackError (line 40) | class GenericLlamaStackError(LlamaStackError):
    method __init__ (line 43) | def __init__(self, status_code_value: int, message: str = ""):
  function is_provider_sdk_exception (line 49) | def is_provider_sdk_exception(exc: Exception) -> TypeGuard[ProviderSDKEx...
  function serialize_exception (line 59) | def serialize_exception(exc: Exception) -> dict[str, Any]:
  function deserialize_exception (line 103) | def deserialize_exception(data: dict[str, Any]) -> Exception:

FILE: src/llama_stack/testing/providers/__init__.py
  function build_providers (line 26) | def build_providers(*configs: ProviderConfig) -> dict[str, ProviderConfig]:
  class GenericProviderError (line 37) | class GenericProviderError(Exception):
    method __init__ (line 40) | def __init__(self, status_code: int, body: dict | None = None, message...
  function detect_provider (line 52) | def detect_provider(exc: object) -> str:
  function create_provider_error (line 61) | def create_provider_error(provider: str, status_code: int, body: dict | ...

FILE: src/llama_stack/testing/providers/_config.py
  class ProviderConfig (line 15) | class ProviderConfig:
    method _module_prefix (line 31) | def _module_prefix(self) -> str:
  function _validate_provider (line 36) | def _validate_provider(config: ProviderConfig) -> None:

FILE: src/llama_stack/testing/providers/ollama.py
  function create_error (line 20) | def create_error(status_code: int, body: dict | None, message: str) -> R...

FILE: src/llama_stack/testing/providers/openai.py
  function create_error (line 37) | def create_error(status_code: int, body: dict | None, message: str) -> A...

FILE: src/llama_stack_api/admin/api.py
  class Admin (line 21) | class Admin(Protocol):
    method list_providers (line 27) | async def list_providers(self) -> ListProvidersResponse:
    method inspect_provider (line 36) | async def inspect_provider(self, request: InspectProviderRequest) -> P...
    method list_routes (line 46) | async def list_routes(self, request: ListRoutesRequest) -> ListRoutesR...
    method health (line 56) | async def health(self) -> HealthInfo:
    method version (line 65) | async def version(self) -> VersionInfo:

FILE: src/llama_stack_api/admin/fastapi_routes.py
  function create_router (line 37) | def create_router(impl: Admin) -> APIRouter:

FILE: src/llama_stack_api/admin/models.py
  class RouteInfo (line 21) | class RouteInfo(BaseModel):
  class HealthInfo (line 35) | class HealthInfo(BaseModel):
  class VersionInfo (line 45) | class VersionInfo(BaseModel):
  class ListRoutesResponse (line 55) | class ListRoutesResponse(BaseModel):
  class ProviderInfo (line 65) | class ProviderInfo(BaseModel):
  class ListProvidersResponse (line 83) | class ListProvidersResponse(BaseModel):
  class ListRoutesRequest (line 94) | class ListRoutesRequest(BaseModel):
  class InspectProviderRequest (line 107) | class InspectProviderRequest(BaseModel):

FILE: src/llama_stack_api/agents/api.py
  class Agents (line 28) | class Agents(Protocol):
    method get_openai_response (line 29) | async def get_openai_response(
    method create_openai_response (line 34) | async def create_openai_response(
    method list_openai_responses (line 39) | async def list_openai_responses(
    method list_openai_response_input_items (line 44) | async def list_openai_response_input_items(
    method delete_openai_response (line 49) | async def delete_openai_response(

FILE: src/llama_stack_api/agents/fastapi_routes.py
  function create_sse_event (line 54) | def create_sse_event(data: Any) -> str:
  function sse_generator (line 63) | async def sse_generator(event_gen):
  function get_list_response_input_items_request (line 91) | async def get_list_response_input_items_request(
  function _preserve_context_for_sse (line 123) | def _preserve_context_for_sse(event_gen):
  function create_router (line 145) | def create_router(impl: Agents) -> APIRouter:

FILE: src/llama_stack_api/agents/models.py
  class ResponseItemInclude (line 30) | class ResponseItemInclude(StrEnum):
  class ResponseTruncation (line 42) | class ResponseTruncation(StrEnum):
  class ResponseGuardrailSpec (line 49) | class ResponseGuardrailSpec(BaseModel):
  class ResponseStreamOptions (line 61) | class ResponseStreamOptions(BaseModel):
  class CreateResponseRequest (line 73) | class CreateResponseRequest(BaseModel):
  class RetrieveResponseRequest (line 206) | class RetrieveResponseRequest(BaseModel):
  class ListResponsesRequest (line 214) | class ListResponsesRequest(BaseModel):
  class ListResponseInputItemsRequest (line 228) | class ListResponseInputItemsRequest(BaseModel):
  class DeleteResponseRequest (line 248) | class DeleteResponseRequest(BaseModel):

FILE: src/llama_stack_api/batches/api.py
  class Batches (line 21) | class Batches(Protocol):
    method create_batch (line 35) | async def create_batch(
    method retrieve_batch (line 40) | async def retrieve_batch(
    method cancel_batch (line 45) | async def cancel_batch(
    method list_batches (line 50) | async def list_batches(

FILE: src/llama_stack_api/batches/fastapi_routes.py
  function create_router (line 41) | def create_router(impl: Batches) -> APIRouter:

FILE: src/llama_stack_api/batches/models.py
  class CreateBatchRequest (line 22) | class CreateBatchRequest(BaseModel):
  class ListBatchesRequest (line 37) | class ListBatchesRequest(BaseModel):
  class RetrieveBatchRequest (line 47) | class RetrieveBatchRequest(BaseModel):
  class CancelBatchRequest (line 54) | class CancelBatchRequest(BaseModel):
  class ListBatchesResponse (line 61) | class ListBatchesResponse(BaseModel):

FILE: src/llama_stack_api/benchmarks/api.py
  class Benchmarks (line 20) | class Benchmarks(Protocol):
    method list_benchmarks (line 21) | async def list_benchmarks(
    method get_benchmark (line 26) | async def get_benchmark(
    method register_benchmark (line 31) | async def register_benchmark(
    method unregister_benchmark (line 36) | async def unregister_benchmark(

FILE: src/llama_stack_api/benchmarks/fastapi_routes.py
  function create_router (line 38) | def create_router(impl: Benchmarks) -> APIRouter:

FILE: src/llama_stack_api/benchmarks/models.py
  class ListBenchmarksRequest (line 22) | class ListBenchmarksRequest(BaseModel):
  class GetBenchmarkRequest (line 29) | class GetBenchmarkRequest(BaseModel):
  class RegisterBenchmarkRequest (line 36) | class RegisterBenchmarkRequest(BaseModel):
  class UnregisterBenchmarkRequest (line 50) | class UnregisterBenchmarkRequest(BaseModel):
  class CommonBenchmarkFields (line 56) | class CommonBenchmarkFields(BaseModel):
  class Benchmark (line 68) | class Benchmark(CommonBenchmarkFields, Resource):
    method benchmark_id (line 77) | def benchmark_id(self) -> str:
    method provider_benchmark_id (line 81) | def provider_benchmark_id(self) -> str | None:
  class BenchmarkInput (line 85) | class BenchmarkInput(CommonBenchmarkFields, BaseModel):
  class ListBenchmarksResponse (line 94) | class ListBenchmarksResponse(BaseModel):

FILE: src/llama_stack_api/common/content_types.py
  class URL (line 15) | class URL(BaseModel):
  class _URLOrData (line 24) | class _URLOrData(BaseModel):
    method validator (line 38) | def validator(cls, values):
  class ImageContentItem (line 45) | class ImageContentItem(BaseModel):
  class TextContentItem (line 57) | class TextContentItem(BaseModel):
  class TextDelta (line 81) | class TextDelta(BaseModel):
  class ImageDelta (line 93) | class ImageDelta(BaseModel):

FILE: src/llama_stack_api/common/errors.py
  class OpenAIErrorDetail (line 16) | class OpenAIErrorDetail(BaseModel):
  class OpenAIErrorResponse (line 28) | class OpenAIErrorResponse(BaseModel):
    method from_message (line 41) | def from_message(
    method to_dict (line 47) | def to_dict(self) -> dict:
    method to_bytes (line 51) | def to_bytes(self) -> bytes:
  class LlamaStackError (line 56) | class LlamaStackError(Exception):
    method __init__ (line 61) | def __init__(self, message: str):
  class ClientListCommand (line 65) | class ClientListCommand:
    method __init__ (line 77) | def __init__(
    method __str__ (line 87) | def __str__(self) -> str:
  class ResourceNotFoundError (line 101) | class ResourceNotFoundError(LlamaStackError):
    method __init__ (line 123) | def __init__(
  class ModelNotFoundError (line 144) | class ModelNotFoundError(ResourceNotFoundError):
    method __init__ (line 147) | def __init__(self, model_name: str) -> None:
  class VectorStoreNotFoundError (line 151) | class VectorStoreNotFoundError(ResourceNotFoundError):
    method __init__ (line 154) | def __init__(self, vector_store_name: str) -> None:
  class DatasetNotFoundError (line 158) | class DatasetNotFoundError(ResourceNotFoundError):
    method __init__ (line 161) | def __init__(self, dataset_name: str) -> None:
  class ToolGroupNotFoundError (line 165) | class ToolGroupNotFoundError(ResourceNotFoundError):
    method __init__ (line 168) | def __init__(self, toolgroup_name: str) -> None:
  class ConversationNotFoundError (line 172) | class ConversationNotFoundError(ResourceNotFoundError):
    method __init__ (line 175) | def __init__(self, conversation_id: str) -> None:
  class ConversationItemNotFoundError (line 179) | class ConversationItemNotFoundError(ResourceNotFoundError):
    method __init__ (line 182) | def __init__(self, item_id: str, conversation_id: str) -> None:
  class ConnectorNotFoundError (line 193) | class ConnectorNotFoundError(ResourceNotFoundError):
    method __init__ (line 196) | def __init__(self, connector_id: str) -> None:
  class ConnectorToolNotFoundError (line 200) | class ConnectorToolNotFoundError(ResourceNotFoundError):
    method __init__ (line 203) | def __init__(self, connector_id: str, tool_name: str) -> None:
  class OpenAIFileObjectNotFoundError (line 212) | class OpenAIFileObjectNotFoundError(ResourceNotFoundError):
    method __init__ (line 215) | def __init__(self, file_id: str) -> None:
  class BatchNotFoundError (line 219) | class BatchNotFoundError(ResourceNotFoundError):
    method __init__ (line 222) | def __init__(self, batch_id: str) -> None:
  class UnsupportedModelError (line 226) | class UnsupportedModelError(LlamaStackError):
    method __init__ (line 231) | def __init__(self, model_name: str, supported_models_list: list[str]):
  class ModelTypeError (line 236) | class ModelTypeError(LlamaStackError):
    method __init__ (line 241) | def __init__(self, model_name: str, model_type: str, expected_model_ty...
  class ConflictError (line 248) | class ConflictError(LlamaStackError):
    method __init__ (line 253) | def __init__(self, message: str) -> None:
  class TokenValidationError (line 257) | class TokenValidationError(LlamaStackError):
    method __init__ (line 262) | def __init__(self, message: str) -> None:
  class InvalidParameterError (line 266) | class InvalidParameterError(ValueError, LlamaStackError):
    method __init__ (line 279) | def __init__(self, param_name: str, value: object, constraint: str) ->...
  class ServiceNotEnabledError (line 284) | class ServiceNotEnabledError(LlamaStackError, ValueError):
    method __init__ (line 298) | def __init__(self, service_name: str, *, provider_specific_message: st...
  class InternalServerError (line 305) | class InternalServerError(LlamaStackError):
    method __init__ (line 314) | def __init__(self, detail: str | None = None) -> None:
  class ResponseNotFoundError (line 319) | class ResponseNotFoundError(ResourceNotFoundError):
    method __init__ (line 322) | def __init__(self, response_id: str) -> None:
  class ResponseInputItemNotFoundError (line 326) | class ResponseInputItemNotFoundError(ResourceNotFoundError):
    method __init__ (line 329) | def __init__(self, item_id: str, response_id: str) -> None:
  class FileTooLargeError (line 340) | class FileTooLargeError(LlamaStackError):
    method __init__ (line 345) | def __init__(self, file_size: int, max_size: int) -> None:

FILE: src/llama_stack_api/common/job_types.py
  class JobStatus (line 13) | class JobStatus(Enum):
  class Job (line 30) | class Job(BaseModel):

FILE: src/llama_stack_api/common/responses.py
  class Order (line 15) | class Order(Enum):
  class PaginatedResponse (line 26) | class PaginatedResponse(BaseModel):
  class MetricInResponse (line 60) | class MetricInResponse(BaseModel):
  class MetricResponseMixin (line 72) | class MetricResponseMixin(BaseModel):

FILE: src/llama_stack_api/common/type_system.py
  class StringType (line 15) | class StringType(BaseModel):
  class NumberType (line 25) | class NumberType(BaseModel):
  class BooleanType (line 35) | class BooleanType(BaseModel):
  class ArrayType (line 45) | class ArrayType(BaseModel):
  class ObjectType (line 55) | class ObjectType(BaseModel):
  class JsonType (line 65) | class JsonType(BaseModel):
  class UnionType (line 75) | class UnionType(BaseModel):
  class ChatCompletionInputType (line 85) | class ChatCompletionInputType(BaseModel):
  class CompletionInputType (line 96) | class CompletionInputType(BaseModel):
  class DialogType (line 107) | class DialogType(BaseModel):

FILE: src/llama_stack_api/common/upload_safety.py
  function read_upload_with_size_limit (line 28) | async def read_upload_with_size_limit(
  class PreReadUploadFile (line 81) | class PreReadUploadFile(UploadFile):
    method __init__ (line 90) | def __init__(self, content: bytes, *, filename: str | None = None, con...

FILE: src/llama_stack_api/connectors/api.py
  class Connectors (line 29) | class Connectors(Protocol):
    method get_connector (line 32) | async def get_connector(
    method list_connectors (line 38) | async def list_connectors(self) -> ListConnectorsResponse: ...
    method list_connector_tools (line 40) | async def list_connector_tools(
    method get_connector_tool (line 46) | async def get_connector_tool(

FILE: src/llama_stack_api/connectors/fastapi_routes.py
  function create_router (line 36) | def create_router(impl: Connectors) -> APIRouter:

FILE: src/llama_stack_api/connectors/models.py
  class ConnectorType (line 21) | class ConnectorType(StrEnum):
  class CommonConnectorFields (line 27) | class CommonConnectorFields(BaseModel):
  class Connector (line 37) | class Connector(CommonConnectorFields):
  class ConnectorInput (line 47) | class ConnectorInput(CommonConnectorFields):
  class GetConnectorRequest (line 55) | class GetConnectorRequest(BaseModel):
  class ListConnectorToolsRequest (line 62) | class ListConnectorToolsRequest(BaseModel):
  class GetConnectorToolRequest (line 69) | class GetConnectorToolRequest(BaseModel):
  class ListConnectorsResponse (line 80) | class ListConnectorsResponse(BaseModel):
  class ListToolsResponse (line 87) | class ListToolsResponse(BaseModel):

FILE: src/llama_stack_api/conversations/api.py
  class Conversations (line 27) | class Conversations(Protocol):
    method create_conversation (line 30) | async def create_conversation(self, request: CreateConversationRequest...
    method get_conversation (line 32) | async def get_conversation(self, request: GetConversationRequest) -> C...
    method update_conversation (line 34) | async def update_conversation(self, conversation_id: str, request: Upd...
    method openai_delete_conversation (line 36) | async def openai_delete_conversation(self, request: DeleteConversation...
    method add_items (line 38) | async def add_items(self, conversation_id: str, request: AddItemsReque...
    method retrieve (line 40) | async def retrieve(self, request: RetrieveItemRequest) -> Conversation...
    method list_items (line 42) | async def list_items(self, request: ListItemsRequest) -> ConversationI...
    method openai_delete_conversation_item (line 44) | async def openai_delete_conversation_item(self, request: DeleteItemReq...

FILE: src/llama_stack_api/conversations/fastapi_routes.py
  class _ListItemsQueryParams (line 45) | class _ListItemsQueryParams(BaseModel):
  function create_router (line 64) | def create_router(impl: Conversations) -> APIRouter:

FILE: src/llama_stack_api/conversations/models.py
  class Conversation (line 35) | class Conversation(BaseModel):
  class ConversationMessage (line 56) | class ConversationMessage(BaseModel):
  function _ensure_item_type (line 84) | def _ensure_item_type(item: Any) -> Any:
  class ConversationDeletedResource (line 91) | class ConversationDeletedResource(BaseModel):
  class ConversationItemCreateRequest (line 100) | class ConversationItemCreateRequest(BaseModel):
    method default_message_type (line 111) | def default_message_type(cls, value: Any) -> Any:
  class ConversationItemInclude (line 117) | class ConversationItemInclude(StrEnum):
  class ConversationItemList (line 130) | class ConversationItemList(BaseModel):
  class ConversationItemDeletedResource (line 141) | class ConversationItemDeletedResource(BaseModel):
  class CreateConversationRequest (line 153) | class CreateConversationRequest(BaseModel):
    method default_message_type (line 167) | def default_message_type(cls, value: Any) -> Any:
  class GetConversationRequest (line 174) | class GetConversationRequest(BaseModel):
  class UpdateConversationRequest (line 181) | class UpdateConversationRequest(BaseModel):
  class DeleteConversationRequest (line 191) | class DeleteConversationRequest(BaseModel):
  class AddItemsRequest (line 198) | class AddItemsRequest(BaseModel):
    method default_message_type (line 209) | def default_message_type(cls, value: Any) -> Any:
  class RetrieveItemRequest (line 216) | class RetrieveItemRequest(BaseModel):
  class ListItemsRequest (line 224) | class ListItemsRequest(BaseModel):
  class DeleteItemRequest (line 247) | class DeleteItemRequest(BaseModel):

FILE: src/llama_stack_api/datasetio/api.py
  class DatasetStore (line 25) | class DatasetStore(Protocol):
    method get_dataset (line 26) | def get_dataset(self, dataset_id: str) -> Dataset: ...
  class DatasetIO (line 30) | class DatasetIO(Protocol):
    method iterrows (line 40) | async def iterrows(self, request: IterRowsRequest) -> PaginatedRespons...
    method append_rows (line 42) | async def append_rows(self, params: AppendRowsParams) -> None: ...

FILE: src/llama_stack_api/datasetio/fastapi_routes.py
  function create_router (line 29) | def create_router(impl: DatasetIO) -> APIRouter:

FILE: src/llama_stack_api/datasetio/models.py
  class IterRowsRequest (line 22) | class IterRowsRequest(BaseModel):
  class AppendRowsRequest (line 37) | class AppendRowsRequest(BaseModel):
  class AppendRowsParams (line 43) | class AppendRowsParams(BaseModel):

FILE: src/llama_stack_api/datasets/__init__.py
  class DatasetInput (line 37) | class DatasetInput(CommonDatasetFields):

FILE: src/llama_stack_api/datasets/api.py
  class Datasets (line 26) | class Datasets(Protocol):
    method register_dataset (line 29) | async def register_dataset(self, request: RegisterDatasetRequest) -> D...
    method get_dataset (line 31) | async def get_dataset(self, request: GetDatasetRequest) -> Dataset: ...
    method list_datasets (line 33) | async def list_datasets(self) -> ListDatasetsResponse: ...
    method unregister_dataset (line 35) | async def unregister_dataset(self, request: UnregisterDatasetRequest) ...

FILE: src/llama_stack_api/datasets/fastapi_routes.py
  function create_router (line 34) | def create_router(impl: Datasets) -> APIRouter:

FILE: src/llama_stack_api/datasets/models.py
  class DatasetPurpose (line 22) | class DatasetPurpose(StrEnum):
  class DatasetType (line 31) | class DatasetType(Enum):
  class URIDataSource (line 41) | class URIDataSource(BaseModel):
  class RowsDataSource (line 52) | class RowsDataSource(BaseModel):
  class CommonDatasetFields (line 69) | class CommonDatasetFields(BaseModel):
  class Dataset (line 81) | class Dataset(CommonDatasetFields, Resource):
    method dataset_id (line 90) | def dataset_id(self) -> str:
    method provider_dataset_id (line 94) | def provider_dataset_id(self) -> str | None:
  class ListDatasetsResponse (line 99) | class ListDatasetsResponse(BaseModel):
  class RegisterDatasetRequest (line 109) | class RegisterDatasetRequest(BaseModel):
  class GetDatasetRequest (line 125) | class GetDatasetRequest(BaseModel):
  class UnregisterDatasetRequest (line 132) | class UnregisterDatasetRequest(BaseModel):

FILE: src/llama_stack_api/datatypes.py
  class DynamicApiMeta (line 23) | class DynamicApiMeta(EnumMeta):
    method __new__ (line 24) | def __new__(cls, name, bases, namespace):
    method __call__ (line 38) | def __call__(cls, value):
    method __iter__ (line 58) | def __iter__(cls):
    method add (line 64) | def add(cls, value):
  class Api (line 92) | class Api(Enum, metaclass=DynamicApiMeta):
  class Error (line 148) | class Error(BaseModel):
  class ExternalApiSpec (line 164) | class ExternalApiSpec(BaseModel):
  class ModelsProtocolPrivate (line 178) | class ModelsProtocolPrivate(Protocol):
    method register_model (line 203) | async def register_model(self, model: Model) -> Model: ...
    method unregister_model (line 205) | async def unregister_model(self, model_id: str) -> None: ...
    method list_models (line 214) | async def list_models(self) -> list[Model] | None: ...
    method should_refresh_models (line 216) | async def should_refresh_models(self) -> bool: ...
  class ShieldsProtocolPrivate (line 219) | class ShieldsProtocolPrivate(Protocol):
    method register_shield (line 220) | async def register_shield(self, shield: Shield) -> None: ...
    method unregister_shield (line 222) | async def unregister_shield(self, identifier: str) -> None: ...
  class VectorStoresProtocolPrivate (line 225) | class VectorStoresProtocolPrivate(Protocol):
    method register_vector_store (line 226) | async def register_vector_store(self, vector_store: VectorStore) -> No...
    method unregister_vector_store (line 228) | async def unregister_vector_store(self, vector_store_id: str) -> None:...
  class DatasetsProtocolPrivate (line 231) | class DatasetsProtocolPrivate(Protocol):
    method register_dataset (line 232) | async def register_dataset(self, dataset: Dataset) -> None: ...
    method unregister_dataset (line 234) | async def unregister_dataset(self, dataset_id: str) -> None: ...
  class ScoringFunctionsProtocolPrivate (line 237) | class ScoringFunctionsProtocolPrivate(Protocol):
    method list_scoring_functions (line 238) | async def list_scoring_functions(self) -> list[ScoringFn]: ...
    method register_scoring_function (line 240) | async def register_scoring_function(self, scoring_fn: ScoringFn) -> No...
  class BenchmarksProtocolPrivate (line 243) | class BenchmarksProtocolPrivate(Protocol):
    method register_benchmark (line 244) | async def register_benchmark(self, benchmark: Benchmark) -> None: ...
  class ToolGroupsProtocolPrivate (line 247) | class ToolGroupsProtocolPrivate(Protocol):
    method register_toolgroup (line 248) | async def register_toolgroup(self, toolgroup: ToolGroup) -> None: ...
    method unregister_toolgroup (line 250) | async def unregister_toolgroup(self, toolgroup_id: str) -> None: ...
  class ProviderSpec (line 254) | class ProviderSpec(BaseModel):
    method is_sample (line 303) | def is_sample(self) -> bool:
  class RoutingTable (line 307) | class RoutingTable(Protocol):
    method get_provider_impl (line 308) | async def get_provider_impl(self, routing_key: str) -> Any: ...
  class InlineProviderSpec (line 312) | class InlineProviderSpec(ProviderSpec):
  class RemoteProviderConfig (line 328) | class RemoteProviderConfig(BaseModel):
    method url (line 334) | def url(self) -> str:
    method from_url (line 340) | def from_url(cls, url: str) -> "RemoteProviderConfig":
  class RemoteProviderSpec (line 347) | class RemoteProviderSpec(ProviderSpec):
    method container_image (line 361) | def container_image(self) -> str | None:
  class HealthStatus (line 365) | class HealthStatus(StrEnum):

FILE: src/llama_stack_api/eval/api.py
  class Eval (line 22) | class Eval(Protocol):
    method run_eval (line 27) | async def run_eval(
    method evaluate_rows (line 34) | async def evaluate_rows(
    method job_status (line 41) | async def job_status(self, request: JobStatusRequest) -> Job:
    method job_cancel (line 45) | async def job_cancel(self, request: JobCancelRequest) -> None:
    method job_result (line 49) | async def job_result(self, request: JobResultRequest) -> EvaluateRespo...

FILE: src/llama_stack_api/eval/compat.py
  function _emit_deprecation_warning (line 40) | def _emit_deprecation_warning(method_name: str, request_class: str) -> N...
  function _format_missing_params (line 49) | def _format_missing_params(required: list[str], provided: dict[str, Any]...
  function _validate_not_empty (line 63) | def _validate_not_empty(value: Any, name: str) -> None:
  function resolve_run_eval_request (line 69) | def resolve_run_eval_request(
  function resolve_evaluate_rows_request (line 113) | def resolve_evaluate_rows_request(
  function resolve_job_status_request (line 171) | def resolve_job_status_request(
  function resolve_job_cancel_request (line 215) | def resolve_job_cancel_request(
  function resolve_job_result_request (line 259) | def resolve_job_result_request(

FILE: src/llama_stack_api/eval/fastapi_routes.py
  function create_router (line 31) | def create_router(impl: Eval) -> APIRouter:

FILE: src/llama_stack_api/eval/models.py
  class ModelCandidate (line 18) | class ModelCandidate(BaseModel):
  class BenchmarkConfig (line 33) | class BenchmarkConfig(BaseModel):
  class EvaluateResponse (line 50) | class EvaluateResponse(BaseModel):
  class BenchmarkIdRequest (line 60) | class BenchmarkIdRequest(BaseModel):
  class RunEvalRequest (line 67) | class RunEvalRequest(BaseModel):
  class RunEvalBodyRequest (line 75) | class RunEvalBodyRequest(BaseModel):
  class EvaluateRowsRequest (line 82) | class EvaluateRowsRequest(BaseModel):
  class EvaluateRowsBodyRequest (line 94) | class EvaluateRowsBodyRequest(BaseModel):
  class JobStatusRequest (line 105) | class JobStatusRequest(BaseModel):
  class JobCancelRequest (line 113) | class JobCancelRequest(BaseModel):
  class JobResultRequest (line 121) | class JobResultRequest(BaseModel):

FILE: src/llama_stack_api/file_processors/api.py
  class FileProcessors (line 15) | class FileProcessors(Protocol):
    method process_file (line 37) | async def process_file(

FILE: src/llama_stack_api/file_processors/fastapi_routes.py
  function create_router (line 37) | def create_router(impl: FileProcessors, max_upload_size_bytes: int = DEF...

FILE: src/llama_stack_api/file_processors/models.py
  class ProcessFileResponse (line 26) | class ProcessFileResponse(BaseModel):
  class ProcessFileRequest (line 42) | class ProcessFileRequest(BaseModel):

FILE: src/llama_stack_api/files/api.py
  class Files (line 24) | class Files(Protocol):
    method openai_upload_file (line 27) | async def openai_upload_file(
    method openai_list_files (line 33) | async def openai_list_files(
    method openai_retrieve_file (line 38) | async def openai_retrieve_file(
    method openai_delete_file (line 43) | async def openai_delete_file(
    method openai_retrieve_file_content (line 48) | async def openai_retrieve_file_content(

FILE: src/llama_stack_api/files/fastapi_routes.py
  function create_router (line 43) | def create_router(impl: Files, max_upload_size_bytes: int = DEFAULT_MAX_...

FILE: src/llama_stack_api/files/models.py
  class OpenAIFilePurpose (line 16) | class OpenAIFilePurpose(StrEnum):
  class OpenAIFileObject (line 26) | class OpenAIFileObject(BaseModel):
  class ExpiresAfter (line 41) | class ExpiresAfter(BaseModel):
  class ListOpenAIFileResponse (line 54) | class ListOpenAIFileResponse(BaseModel):
  class OpenAIFileDeleteResponse (line 65) | class OpenAIFileDeleteResponse(BaseModel):
  class ListFilesRequest (line 74) | class ListFilesRequest(BaseModel):
  class RetrieveFileRequest (line 84) | class RetrieveFileRequest(BaseModel):
  class DeleteFileRequest (line 91) | class DeleteFileRequest(BaseModel):
  class RetrieveFileContentRequest (line 98) | class RetrieveFileContentRequest(BaseModel):
  class UploadFileRequest (line 105) | class UploadFileRequest(BaseModel):

FILE: src/llama_stack_api/filters.py
  class ComparisonFilter (line 30) | class ComparisonFilter(BaseModel):
  class CompoundFilter (line 44) | class CompoundFilter(BaseModel):

FILE: src/llama_stack_api/inference/api.py
  class ModelStore (line 29) | class ModelStore(Protocol):
    method get_model (line 30) | async def get_model(self, identifier: str) -> Model: ...
  class InferenceProvider (line 34) | class InferenceProvider(Protocol):
    method rerank (line 43) | async def rerank(
    method openai_completion (line 51) | async def openai_completion(
    method openai_chat_completion (line 58) | async def openai_chat_completion(
    method openai_embeddings (line 65) | async def openai_embeddings(
  class Inference (line 73) | class Inference(InferenceProvider):
    method list_chat_completions (line 84) | async def list_chat_completions(
    method get_chat_completion (line 91) | async def get_chat_completion(self, request: GetChatCompletionRequest)...

FILE: src/llama_stack_api/inference/fastapi_routes.py
  function _create_sse_event (line 48) | def _create_sse_event(data: Any) -> str:
  function _sse_generator (line 57) | async def _sse_generator(event_gen: AsyncIterator[Any], context: str = "...
  function _http_exception_from_value_error (line 72) | def _http_exception_from_value_error(exc: ValueError) -> HTTPException:
  function _http_exception_from_sse_error (line 78) | def _http_exception_from_sse_error(exc: Exception) -> HTTPException:
  function _preserve_context_for_sse (line 90) | def _preserve_context_for_sse(event_gen):
  function create_router (line 121) | def create_router(impl: Inference) -> APIRouter:

FILE: src/llama_stack_api/inference/models.py
  class GreedySamplingStrategy (line 31) | class GreedySamplingStrategy(BaseModel):
  class TopPSamplingStrategy (line 40) | class TopPSamplingStrategy(BaseModel):
  class TopKSamplingStrategy (line 53) | class TopKSamplingStrategy(BaseModel):
  class SamplingParams (line 68) | class SamplingParams(BaseModel):
  class LogProbConfig (line 92) | class LogProbConfig(BaseModel):
  class QuantizationType (line 100) | class QuantizationType(Enum):
  class Fp8QuantizationConfig (line 109) | class Fp8QuantizationConfig(BaseModel):
  class Bf16QuantizationConfig (line 118) | class Bf16QuantizationConfig(BaseModel):
  class Int4QuantizationConfig (line 125) | class Int4QuantizationConfig(BaseModel):
  class UserMessage (line 142) | class UserMessage(BaseModel):
  class SystemMessage (line 156) | class SystemMessage(BaseModel):
  class ToolResponseMessage (line 169) | class ToolResponseMessage(BaseModel):
  class ToolChoice (line 177) | class ToolChoice(Enum):
  class TokenLogProbs (line 186) | class TokenLogProbs(BaseModel):
  class ChatCompletionResponseEventType (line 194) | class ChatCompletionResponseEventType(Enum):
  class ResponseFormatType (line 202) | class ResponseFormatType(StrEnum):
  class JsonSchemaResponseFormat (line 210) | class JsonSchemaResponseFormat(BaseModel):
  class GrammarResponseFormat (line 220) | class GrammarResponseFormat(BaseModel):
  class CompletionRequest (line 237) | class CompletionRequest(BaseModel):
  class SystemMessageBehavior (line 246) | class SystemMessageBehavior(Enum):
  class EmbeddingsResponse (line 254) | class EmbeddingsResponse(BaseModel):
  class RerankData (line 264) | class RerankData(BaseModel):
  class RerankResponse (line 274) | class RerankResponse(BaseModel):
  class OpenAIChatCompletionContentPartTextParam (line 284) | class OpenAIChatCompletionContentPartTextParam(BaseModel):
  class OpenAIImageURL (line 292) | class OpenAIImageURL(BaseModel):
  class OpenAIChatCompletionContentPartImageParam (line 302) | class OpenAIChatCompletionContentPartImageParam(BaseModel):
  class OpenAIFileFile (line 312) | class OpenAIFileFile(BaseModel):
    method validate_file_reference (line 320) | def validate_file_reference(self) -> Self:
  class OpenAIFile (line 328) | class OpenAIFile(BaseModel):
  class OpenAIUserMessageParam (line 346) | class OpenAIUserMessageParam(BaseModel):
  class OpenAISystemMessageParam (line 357) | class OpenAISystemMessageParam(BaseModel):
  class OpenAIChatCompletionToolCallFunction (line 371) | class OpenAIChatCompletionToolCallFunction(BaseModel):
  class OpenAIChatCompletionToolCall (line 379) | class OpenAIChatCompletionToolCall(BaseModel):
  class OpenAIAssistantMessageParam (line 391) | class OpenAIAssistantMessageParam(BaseModel):
  class OpenAIToolMessageParam (line 409) | class OpenAIToolMessageParam(BaseModel):
  class OpenAIDeveloperMessageParam (line 418) | class OpenAIDeveloperMessageParam(BaseModel):
  class OpenAIResponseFormatText (line 442) | class OpenAIResponseFormatText(BaseModel):
  class OpenAIJSONSchema (line 449) | class OpenAIJSONSchema(TypedDict, total=False):
  class OpenAIResponseFormatJSONSchema (line 464) | class OpenAIResponseFormatJSONSchema(BaseModel):
  class OpenAIResponseFormatJSONObject (line 474) | class OpenAIResponseFormatJSONObject(BaseModel):
  class FunctionToolConfig (line 490) | class FunctionToolConfig(BaseModel):
  class OpenAIChatCompletionToolChoiceFunctionTool (line 495) | class OpenAIChatCompletionToolChoiceFunctionTool(BaseModel):
    method __init__ (line 503) | def __init__(self, name: str):
  class CustomToolConfig (line 508) | class CustomToolConfig(BaseModel):
  class OpenAIChatCompletionToolChoiceCustomTool (line 515) | class OpenAIChatCompletionToolChoiceCustomTool(BaseModel):
    method __init__ (line 521) | def __init__(self, name: str):
  class AllowedToolsConfig (line 526) | class AllowedToolsConfig(BaseModel):
  class OpenAIChatCompletionToolChoiceAllowedTools (line 532) | class OpenAIChatCompletionToolChoiceAllowedTools(BaseModel):
    method __init__ (line 540) | def __init__(self, tools: list[dict[str, Any]], mode: Literal["auto", ...
  class OpenAITopLogProb (line 556) | class OpenAITopLogProb(BaseModel):
  class OpenAITokenLogProb (line 565) | class OpenAITokenLogProb(BaseModel):
  class OpenAIChoiceLogprobs (line 577) | class OpenAIChoiceLogprobs(BaseModel):
  class OpenAIChoiceDelta (line 593) | class OpenAIChoiceDelta(BaseModel):
  class OpenAIChunkChoice (line 619) | class OpenAIChunkChoice(BaseModel):
  class OpenAIChatCompletionResponseMessage (line 635) | class OpenAIChatCompletionResponseMessage(BaseModel):
  class OpenAIChoice (line 662) | class OpenAIChoice(BaseModel):
  class OpenAIChatCompletionUsageCompletionTokensDetails (line 675) | class OpenAIChatCompletionUsageCompletionTokensDetails(BaseModel):
  class OpenAIChatCompletionUsagePromptTokensDetails (line 681) | class OpenAIChatCompletionUsagePromptTokensDetails(BaseModel):
  class OpenAIChatCompletionUsage (line 688) | class OpenAIChatCompletionUsage(BaseModel):
  class OpenAIChatCompletion (line 703) | class OpenAIChatCompletion(BaseModel):
  class OpenAIChatCompletionChunk (line 723) | class OpenAIChatCompletionChunk(BaseModel):
  class OpenAICompletionLogprobs (line 745) | class OpenAICompletionLogprobs(BaseModel):
  class OpenAICompletionChoice (line 757) | class OpenAICompletionChoice(BaseModel):
  class OpenAICompletion (line 769) | class OpenAICompletion(BaseModel):
  class OpenAIEmbeddingData (line 780) | class OpenAIEmbeddingData(BaseModel):
  class OpenAIEmbeddingUsage (line 793) | class OpenAIEmbeddingUsage(BaseModel):
  class OpenAIEmbeddingsResponse (line 801) | class OpenAIEmbeddingsResponse(BaseModel):
  class TextTruncation (line 810) | class TextTruncation(Enum):
  class EmbeddingTaskType (line 818) | class EmbeddingTaskType(Enum):
  class ServiceTier (line 825) | class ServiceTier(StrEnum):
  class OpenAICompletionWithInputMessages (line 834) | class OpenAICompletionWithInputMessages(OpenAIChatCompletion):
  class ListOpenAIChatCompletionResponse (line 841) | class ListOpenAIChatCompletionResponse(BaseModel):
  class OpenAICompletionRequestWithExtraBody (line 855) | class OpenAICompletionRequestWithExtraBody(BaseModel, extra="allow"):
  class OpenAIChatCompletionRequestWithExtraBody (line 887) | class OpenAIChatCompletionRequestWithExtraBody(BaseModel, extra="allow"):
  class OpenAIEmbeddingsRequestWithExtraBody (line 939) | class OpenAIEmbeddingsRequestWithExtraBody(BaseModel, extra="allow"):
    method _reject_explicit_null (line 969) | def _reject_explicit_null(cls, v: Any, info: Any) -> Any:
  class ListChatCompletionsRequest (line 978) | class ListChatCompletionsRequest(BaseModel):
  class GetChatCompletionRequest (line 991) | class GetChatCompletionRequest(BaseModel):
  class RerankRequest (line 998) | class RerankRequest(BaseModel):

FILE: src/llama_stack_api/inspect_api/api.py
  class Inspect (line 18) | class Inspect(Protocol):
    method list_routes (line 21) | async def list_routes(self, api_filter: ApiFilter | None = None) -> Li...
    method health (line 23) | async def health(self) -> HealthInfo: ...
    method version (line 25) | async def version(self) -> VersionInfo: ...

FILE: src/llama_stack_api/inspect_api/fastapi_routes.py
  function create_router (line 29) | def create_router(impl: Inspect) -> APIRouter:

FILE: src/llama_stack_api/internal/kvstore.py
  class KVStore (line 11) | class KVStore(Protocol):
    method set (line 15) | async def set(self, key: str, value: str, expiration: datetime | None ...
    method get (line 17) | async def get(self, key: str) -> str | None: ...
    method delete (line 19) | async def delete(self, key: str) -> None: ...
    method values_in_range (line 21) | async def values_in_range(self, start_key: str, end_key: str) -> list[...
    method keys_in_range (line 23) | async def keys_in_range(self, start_key: str, end_key: str) -> list[st...
    method shutdown (line 25) | async def shutdown(self) -> None: ...

FILE: src/llama_stack_api/internal/sqlstore.py
  class ColumnType (line 16) | class ColumnType(Enum):
  class ColumnDefinition (line 26) | class ColumnDefinition(BaseModel):
  class SqlStore (line 33) | class SqlStore(Protocol):
    method create_table (line 36) | async def create_table(self, table: str, schema: Mapping[str, ColumnTy...
    method insert (line 38) | async def insert(self, table: str, data: Mapping[str, Any] | Sequence[...
    method upsert (line 40) | async def upsert(
    method fetch_all (line 48) | async def fetch_all(
    method fetch_one (line 59) | async def fetch_one(
    method update (line 68) | async def update(self, table: str, data: Mapping[str, Any], where: Map...
    method delete (line 70) | async def delete(self, table: str, where: Mapping[str, Any]) -> None: ...
    method add_column_if_not_exists (line 72) | async def add_column_if_not_exists(
    method shutdown (line 80) | async def shutdown(self) -> None: ...

FILE: src/llama_stack_api/models/api.py
  class Models (line 27) | class Models(Protocol):
    method list_models (line 30) | async def list_models(self) -> ListModelsResponse: ...
    method openai_list_models (line 32) | async def openai_list_models(self) -> OpenAIListModelsResponse: ...
    method get_model (line 34) | async def get_model(self, request: GetModelRequest) -> Model: ...
    method register_model (line 36) | async def register_model(self, request: RegisterModelRequest) -> Model...
    method unregister_model (line 38) | async def unregister_model(self, request: UnregisterModelRequest) -> N...

FILE: src/llama_stack_api/models/fastapi_routes.py
  function create_router (line 34) | def create_router(impl: Models) -> APIRouter:

FILE: src/llama_stack_api/models/models.py
  class ModelType (line 23) | class ModelType(StrEnum):
  class CommonModelFields (line 36) | class CommonModelFields(BaseModel):
  class Model (line 44) | class Model(CommonModelFields, Resource):
    method model_id (line 58) | def model_id(self) -> str:
    method provider_model_id (line 62) | def provider_model_id(self) -> str:
    method validate_provider_resource_id (line 76) | def validate_provider_resource_id(cls, v):
  class ModelInput (line 82) | class ModelInput(CommonModelFields):
  class ListModelsResponse (line 91) | class ListModelsResponse(BaseModel):
  class OpenAIModel (line 98) | class OpenAIModel(BaseModel):
  class OpenAIListModelsResponse (line 116) | class OpenAIListModelsResponse(BaseModel):
  class GetModelRequest (line 126) | class GetModelRequest(BaseModel):
  class RegisterModelRequest (line 133) | class RegisterModelRequest(BaseModel):
  class UnregisterModelRequest (line 148) | class UnregisterModelRequest(BaseModel):

FILE: src/llama_stack_api/openai_responses.py
  class OpenAIResponseError (line 23) | class OpenAIResponseError(BaseModel):
  class OpenAIResponseInputMessageContentText (line 35) | class OpenAIResponseInputMessageContentText(BaseModel):
  class OpenAIResponseInputMessageContentImage (line 47) | class OpenAIResponseInputMessageContentImage(BaseModel):
  class OpenAIResponseInputMessageContentFile (line 63) | class OpenAIResponseInputMessageContentFile(BaseModel):
    method validate_file_source (line 80) | def validate_file_source(self) -> "OpenAIResponseInputMessageContentFi...
  class OpenAIResponsePrompt (line 98) | class OpenAIResponsePrompt(BaseModel):
  class OpenAIResponseAnnotationFileCitation (line 113) | class OpenAIResponseAnnotationFileCitation(BaseModel):
  class OpenAIResponseAnnotationCitation (line 129) | class OpenAIResponseAnnotationCitation(BaseModel):
  class OpenAIResponseAnnotationContainerFileCitation (line 147) | class OpenAIResponseAnnotationContainerFileCitation(BaseModel):
  class OpenAIResponseAnnotationFilePath (line 157) | class OpenAIResponseAnnotationFilePath(BaseModel):
  class OpenAIResponseOutputMessageContentOutputText (line 174) | class OpenAIResponseOutputMessageContentOutputText(BaseModel):
  class OpenAIResponseContentPartRefusal (line 182) | class OpenAIResponseContentPartRefusal(BaseModel):
  class OpenAIResponseMessage (line 201) | class OpenAIResponseMessage(BaseModel):
  class OpenAIResponseOutputMessageWebSearchToolCall (line 219) | class OpenAIResponseOutputMessageWebSearchToolCall(BaseModel):
  class OpenAIResponseOutputMessageFileSearchToolCallResults (line 232) | class OpenAIResponseOutputMessageFileSearchToolCallResults(BaseModel):
  class OpenAIResponseOutputMessageFileSearchToolCall (line 250) | class OpenAIResponseOutputMessageFileSearchToolCall(BaseModel):
  class OpenAIResponseOutputMessageFunctionToolCall (line 268) | class OpenAIResponseOutputMessageFunctionToolCall(BaseModel):
  class OpenAIResponseOutputMessageMCPCall (line 288) | class OpenAIResponseOutputMessageMCPCall(BaseModel):
  class MCPListToolsTool (line 309) | class MCPListToolsTool(BaseModel):
  class OpenAIResponseOutputMessageMCPListTools (line 323) | class OpenAIResponseOutputMessageMCPListTools(BaseModel):
  class OpenAIResponseMCPApprovalRequest (line 339) | class OpenAIResponseMCPApprovalRequest(BaseModel):
  class OpenAIResponseMCPApprovalResponse (line 352) | class OpenAIResponseMCPApprovalResponse(BaseModel):
  class OpenAIResponseTextFormat (line 381) | class OpenAIResponseTextFormat(TypedDict, total=False):
  class OpenAIResponseText (line 399) | class OpenAIResponseText(BaseModel):
  class OpenAIResponseReasoning (line 409) | class OpenAIResponseReasoning(BaseModel):
  class OpenAIResponseInputToolWebSearch (line 426) | class OpenAIResponseInputToolWebSearch(BaseModel):
  class OpenAIResponseInputToolFunction (line 446) | class OpenAIResponseInputToolFunction(BaseModel):
  class OpenAIResponseInputToolFileSearch (line 464) | class OpenAIResponseInputToolFileSearch(BaseModel):
  class ApprovalFilter (line 481) | class ApprovalFilter(BaseModel):
  class AllowedToolsFilter (line 492) | class AllowedToolsFilter(BaseModel):
  class OpenAIResponseInputToolMCP (line 502) | class OpenAIResponseInputToolMCP(BaseModel):
    method validate_server_or_connector (line 526) | def validate_server_or_connector(self) -> "OpenAIResponseInputToolMCP":
  class OpenAIResponseToolMCP (line 543) | class OpenAIResponseToolMCP(BaseModel):
  class OpenAIResponseInputToolChoiceAllowedTools (line 567) | class OpenAIResponseInputToolChoiceAllowedTools(BaseModel):
  class OpenAIResponseInputToolChoiceFileSearch (line 581) | class OpenAIResponseInputToolChoiceFileSearch(BaseModel):
  class OpenAIResponseInputToolChoiceWebSearch (line 591) | class OpenAIResponseInputToolChoiceWebSearch(BaseModel):
  class OpenAIResponseInputToolChoiceFunctionTool (line 606) | class OpenAIResponseInputToolChoiceFunctionTool(BaseModel):
  class OpenAIResponseInputToolChoiceMCPTool (line 618) | class OpenAIResponseInputToolChoiceMCPTool(BaseModel):
  class OpenAIResponseInputToolChoiceCustomTool (line 632) | class OpenAIResponseInputToolChoiceCustomTool(BaseModel):
  class OpenAIResponseInputToolChoiceMode (line 643) | class OpenAIResponseInputToolChoiceMode(str, Enum):
  class OpenAIResponseUsageOutputTokensDetails (line 665) | class OpenAIResponseUsageOutputTokensDetails(BaseModel):
  class OpenAIResponseUsageInputTokensDetails (line 674) | class OpenAIResponseUsageInputTokensDetails(BaseModel):
  class OpenAIResponseUsage (line 684) | class OpenAIResponseUsage(BaseModel):
  class OpenAIResponseIncompleteDetails (line 702) | class OpenAIResponseIncompleteDetails(BaseModel):
  class OpenAIResponseObject (line 712) | class OpenAIResponseObject(BaseModel):
  class OpenAIDeleteResponseObject (line 781) | class OpenAIDeleteResponseObject(BaseModel):
  class OpenAIResponseObjectStreamResponseCreated (line 795) | class OpenAIResponseObjectStreamResponseCreated(BaseModel):
  class OpenAIResponseObjectStreamResponseInProgress (line 809) | class OpenAIResponseObjectStreamResponseInProgress(BaseModel):
  class OpenAIResponseObjectStreamResponseCompleted (line 823) | class OpenAIResponseObjectStreamResponseCompleted(BaseModel):
  class OpenAIResponseObjectStreamResponseIncomplete (line 837) | class OpenAIResponseObjectStreamResponseIncomplete(BaseModel):
  class OpenAIResponseObjectStreamResponseFailed (line 851) | class OpenAIResponseObjectStreamResponseFailed(BaseModel):
  class OpenAIResponseObjectStreamResponseOutputItemAdded (line 865) | class OpenAIResponseObjectStreamResponseOutputItemAdded(BaseModel):
  class OpenAIResponseObjectStreamResponseOutputItemDone (line 883) | class OpenAIResponseObjectStreamResponseOutputItemDone(BaseModel):
  class OpenAIResponseObjectStreamResponseOutputTextDelta (line 901) | class OpenAIResponseObjectStreamResponseOutputTextDelta(BaseModel):
  class OpenAIResponseObjectStreamResponseOutputTextDone (line 923) | class OpenAIResponseObjectStreamResponseOutputTextDone(BaseModel):
  class OpenAIResponseObjectStreamResponseFunctionCallArgumentsDelta (line 943) | class OpenAIResponseObjectStreamResponseFunctionCallArgumentsDelta(BaseM...
  class OpenAIResponseObjectStreamResponseFunctionCallArgumentsDone (line 961) | class OpenAIResponseObjectStreamResponseFunctionCallArgumentsDone(BaseMo...
  class OpenAIResponseObjectStreamResponseWebSearchCallInProgress (line 979) | class OpenAIResponseObjectStreamResponseWebSearchCallInProgress(BaseModel):
  class OpenAIResponseObjectStreamResponseWebSearchCallSearching (line 995) | class OpenAIResponseObjectStreamResponseWebSearchCallSearching(BaseModel):
  class OpenAIResponseObjectStreamResponseWebSearchCallCompleted (line 1003) | class OpenAIResponseObjectStreamResponseWebSearchCallCompleted(BaseModel):
  class OpenAIResponseObjectStreamResponseMcpListToolsInProgress (line 1019) | class OpenAIResponseObjectStreamResponseMcpListToolsInProgress(BaseModel):
  class OpenAIResponseObjectStreamResponseMcpListToolsFailed (line 1025) | class OpenAIResponseObjectStreamResponseMcpListToolsFailed(BaseModel):
  class OpenAIResponseObjectStreamResponseMcpListToolsCompleted (line 1031) | class OpenAIResponseObjectStreamResponseMcpListToolsCompleted(BaseModel):
  class OpenAIResponseObjectStreamResponseMcpCallArgumentsDelta (line 1037) | class OpenAIResponseObjectStreamResponseMcpCallArgumentsDelta(BaseModel):
  class OpenAIResponseObjectStreamResponseMcpCallArgumentsDone (line 1046) | class OpenAIResponseObjectStreamResponseMcpCallArgumentsDone(BaseModel):
  class OpenAIResponseObjectStreamResponseMcpCallInProgress (line 1055) | class OpenAIResponseObjectStreamResponseMcpCallInProgress(BaseModel):
  class OpenAIResponseObjectStreamResponseMcpCallFailed (line 1071) | class OpenAIResponseObjectStreamResponseMcpCallFailed(BaseModel):
  class OpenAIResponseObjectStreamResponseMcpCallCompleted (line 1083) | class OpenAIResponseObjectStreamResponseMcpCallCompleted(BaseModel):
  class OpenAIResponseContentPartOutputText (line 1095) | class OpenAIResponseContentPartOutputText(BaseModel):
  class OpenAIResponseContentPartReasoningText (line 1111) | class OpenAIResponseContentPartReasoningText(BaseModel):
  class OpenAIResponseObjectStreamResponseContentPartAdded (line 1130) | class OpenAIResponseObjectStreamResponseContentPartAdded(BaseModel):
  class OpenAIResponseObjectStreamResponseContentPartDone (line 1152) | class OpenAIResponseObjectStreamResponseContentPartDone(BaseModel):
  class OpenAIResponseObjectStreamResponseReasoningTextDelta (line 1174) | class OpenAIResponseObjectStreamResponseReasoningTextDelta(BaseModel):
  class OpenAIResponseObjectStreamResponseReasoningTextDone (line 1194) | class OpenAIResponseObjectStreamResponseReasoningTextDone(BaseModel):
  class OpenAIResponseContentPartReasoningSummary (line 1214) | class OpenAIResponseContentPartReasoningSummary(BaseModel):
  class OpenAIResponseObjectStreamResponseReasoningSummaryPartAdded (line 1226) | class OpenAIResponseObjectStreamResponseReasoningSummaryPartAdded(BaseMo...
  class OpenAIResponseObjectStreamResponseReasoningSummaryPartDone (line 1246) | class OpenAIResponseObjectStreamResponseReasoningSummaryPartDone(BaseMod...
  class OpenAIResponseObjectStreamResponseReasoningSummaryTextDelta (line 1266) | class OpenAIResponseObjectStreamResponseReasoningSummaryTextDelta(BaseMo...
  class OpenAIResponseObjectStreamResponseReasoningSummaryTextDone (line 1286) | class OpenAIResponseObjectStreamResponseReasoningSummaryTextDone(BaseMod...
  class OpenAIResponseObjectStreamResponseRefusalDelta (line 1306) | class OpenAIResponseObjectStreamResponseRefusalDelta(BaseModel):
  class OpenAIResponseObjectStreamResponseRefusalDone (line 1326) | class OpenAIResponseObjectStreamResponseRefusalDone(BaseModel):
  class OpenAIResponseObjectStreamResponseOutputTextAnnotationAdded (line 1346) | class OpenAIResponseObjectStreamResponseOutputTextAnnotationAdded(BaseMo...
  class OpenAIResponseObjectStreamResponseFileSearchCallInProgress (line 1368) | class OpenAIResponseObjectStreamResponseFileSearchCallInProgress(BaseMod...
  class OpenAIResponseObjectStreamResponseFileSearchCallSearching (line 1384) | class OpenAIResponseObjectStreamResponseFileSearchCallSearching(BaseModel):
  class OpenAIResponseObjectStreamResponseFileSearchCallCompleted (line 1400) | class OpenAIResponseObjectStreamResponseFileSearchCallCompleted(BaseModel):
  class OpenAIResponseInputFunctionToolCallOutput (line 1458) | class OpenAIResponseInputFunctionToolCallOutput(BaseModel):
  class ListOpenAIResponseInputItem (line 1482) | class ListOpenAIResponseInputItem(BaseModel):
  class OpenAIResponseObjectWithInput (line 1494) | class OpenAIResponseObjectWithInput(OpenAIResponseObject):
    method to_response_object (line 1502) | def to_response_object(self) -> OpenAIResponseObject:
  class ListOpenAIResponseObject (line 1508) | class ListOpenAIResponseObject(BaseModel):

FILE: src/llama_stack_api/prompts/api.py
  class Prompts (line 29) | class Prompts(Protocol):
    method list_prompts (line 32) | async def list_prompts(self) -> ListPromptsResponse: ...
    method list_prompt_versions (line 34) | async def list_prompt_versions(self, request: ListPromptVersionsReques...
    method get_prompt (line 36) | async def get_prompt(self, request: GetPromptRequest) -> Prompt: ...
    method create_prompt (line 38) | async def create_prompt(self, request: CreatePromptRequest) -> Prompt:...
    method update_prompt (line 40) | async def update_prompt(self, request: UpdatePromptRequest) -> Prompt:...
    method delete_prompt (line 42) | async def delete_prompt(self, request: DeletePromptRequest) -> None: ...
    method set_default_version (line 44) | async def set_default_version(self, request: SetDefaultVersionRequest)...

FILE: src/llama_stack_api/prompts/fastapi_routes.py
  function create_router (line 39) | def create_router(impl: Prompts) -> APIRouter:

FILE: src/llama_stack_api/prompts/models.py
  class Prompt (line 22) | class Prompt(BaseModel):
    method validate_prompt_id (line 37) | def validate_prompt_id(cls, prompt_id: str) -> str:
    method validate_version (line 56) | def validate_version(cls, prompt_version: int) -> int:
    method validate_prompt_variables (line 62) | def validate_prompt_variables(self):
    method generate_prompt_id (line 77) | def generate_prompt_id(cls) -> str:
  class ListPromptsResponse (line 85) | class ListPromptsResponse(BaseModel):
  class ListPromptVersionsRequest (line 95) | class ListPromptVersionsRequest(BaseModel):
  class GetPromptRequest (line 102) | class GetPromptRequest(BaseModel):
  class CreatePromptRequest (line 110) | class CreatePromptRequest(BaseModel):
  class UpdatePromptBodyRequest (line 120) | class UpdatePromptBodyRequest(BaseModel):
  class UpdatePromptRequest (line 132) | class UpdatePromptRequest(BaseModel):
  class DeletePromptRequest (line 145) | class DeletePromptRequest(BaseModel):
  class SetDefaultVersionBodyRequest (line 152) | class SetDefaultVersionBodyRequest(BaseModel):
  class SetDefaultVersionRequest (line 159) | class SetDefaultVersionRequest(BaseModel):

FILE: src/llama_stack_api/providers/api.py
  class Providers (line 13) | class Providers(Protocol):
    method list_providers (line 14) | async def list_providers(self) -> ListProvidersResponse: ...
    method inspect_provider (line 16) | async def inspect_provider(self, request: InspectProviderRequest) -> P...

FILE: src/llama_stack_api/providers/fastapi_routes.py
  function create_router (line 27) | def create_router(impl: Providers) -> APIRouter:

FILE: src/llama_stack_api/rag_tool.py
  class RRFRanker (line 15) | class RRFRanker(BaseModel):
  class WeightedRanker (line 28) | class WeightedRanker(BaseModel):
  class RAGDocument (line 54) | class RAGDocument(BaseModel):
  class RAGQueryResult (line 70) | class RAGQueryResult(BaseModel):
  class RAGQueryGenerator (line 81) | class RAGQueryGenerator(Enum):
  class RAGSearchMode (line 94) | class RAGSearchMode(StrEnum):
  class DefaultRAGQueryGeneratorConfig (line 107) | class DefaultRAGQueryGeneratorConfig(BaseModel):
  class LLMRAGQueryGeneratorConfig (line 118) | class LLMRAGQueryGeneratorConfig(BaseModel):
  class RAGQueryConfig (line 137) | class RAGQueryConfig(BaseModel):
    method validate_chunk_template (line 161) | def validate_chunk_template(cls, v: str) -> str:

FILE: src/llama_stack_api/resource.py
  class ResourceType (line 12) | class ResourceType(StrEnum):
  class Resource (line 24) | class Resource(BaseModel):

FILE: src/llama_stack_api/router_utils.py
  function create_query_dependency (line 42) | def create_query_dependency[T: BaseModel](model_class: type[T]) -> Calla...
  function create_path_dependency (line 107) | def create_path_dependency[T: BaseModel](model_class: type[T]) -> Callab...
  function try_translate_to_http_exception (line 169) | def try_translate_to_http_exception(exc: Exception) -> HTTPException | N...
  class ExceptionTranslatingRoute (line 191) | class ExceptionTranslatingRoute(APIRoute):
    method get_route_handler (line 201) | def get_route_handler(self):

FILE: src/llama_stack_api/safety/api.py
  class Safety (line 15) | class Safety(Protocol):
    method run_shield (line 23) | async def run_shield(self, request: RunShieldRequest) -> RunShieldResp...
    method run_moderation (line 27) | async def run_moderation(self, request: RunModerationRequest) -> Moder...

FILE: src/llama_stack_api/safety/datatypes.py
  class ModerationObjectResults (line 17) | class ModerationObjectResults(BaseModel):
  class ModerationObject (line 35) | class ModerationObject(BaseModel):
  class ViolationLevel (line 44) | class ViolationLevel(Enum):
  class SafetyViolation (line 53) | class SafetyViolation(BaseModel):
  class RunShieldResponse (line 64) | class RunShieldResponse(BaseModel):
  class ShieldStore (line 70) | class ShieldStore(Protocol):
    method get_shield (line 73) | async def get_shield(self, request: GetShieldRequest) -> Shield: ...

FILE: src/llama_stack_api/safety/fastapi_routes.py
  function create_router (line 19) | def create_router(impl: Safety) -> APIRouter:

FILE: src/llama_stack_api/safety/models.py
  class RunShieldRequest (line 14) | class RunShieldRequest(BaseModel):
  class RunModerationRequest (line 22) | class RunModerationRequest(BaseModel):

FILE: src/llama_stack_api/schema_utils.py
  class ExtraBodyField (line 12) | class ExtraBodyField[T]:
    method __init__ (line 45) | def __init__(self, description: str | None = None):
  class SchemaInfo (line 53) | class SchemaInfo:
  function json_schema_type (line 64) | def json_schema_type(cls):
  function register_schema (line 86) | def register_schema(schema_type, name: str | None = None):
  function get_registered_schema_info (line 109) | def get_registered_schema_info(schema_type: Any) -> SchemaInfo | None:
  function iter_registered_schema_types (line 114) | def iter_registered_schema_types() -> Iterable[SchemaInfo]:
  function iter_json_schema_types (line 119) | def iter_json_schema_types() -> Iterable[type]:
  function iter_dynamic_schema_types (line 124) | def iter_dynamic_schema_types() -> Iterable[type]:
  function register_dynamic_schema_type (line 129) | def register_dynamic_schema_type(schema_type: type, name: str | None = N...
  function clear_dynamic_schema_types (line 136) | def clear_dynamic_schema_types() -> None:
  class WebMethod (line 142) | class WebMethod:
  function webmethod (line 159) | def webmethod(
  function remove_null_from_anyof (line 207) | def remove_null_from_anyof(schema: dict, *, add_nullable: bool = False) ...
  function nullable_openai_style (line 246) | def nullable_openai_style(schema: dict) -> None:

FILE: src/llama_stack_api/scoring/api.py
  class ScoringFunctionStore (line 21) | class ScoringFunctionStore(Protocol):
    method get_scoring_function (line 24) | def get_scoring_function(self, scoring_fn_id: str) -> ScoringFn: ...
  class Scoring (line 28) | class Scoring(Protocol):
    method score_batch (line 33) | async def score_batch(self, request: ScoreBatchRequest) -> ScoreBatchR...
    method score (line 35) | async def score(self, request: ScoreRequest) -> ScoreResponse: ...

FILE: src/llama_stack_api/scoring/fastapi_routes.py
  function create_router (line 24) | def create_router(impl: Scoring) -> APIRouter:

FILE: src/llama_stack_api/scoring/models.py
  class ScoringResult (line 25) | class ScoringResult(BaseModel):
  class ScoreBatchResponse (line 37) | class ScoreBatchResponse(BaseModel):
  class ScoreResponse (line 45) | class ScoreResponse(BaseModel):
  class ScoreRequest (line 54) | class ScoreRequest(BaseModel):
  class ScoreBatchRequest (line 64) | class ScoreBatchRequest(BaseModel):

FILE: src/llama_stack_api/scoring_functions/api.py
  class ScoringFunctions (line 20) | class ScoringFunctions(Protocol):
    method list_scoring_functions (line 21) | async def list_scoring_functions(
    method get_scoring_function (line 26) | async def get_scoring_function(
    method register_scoring_function (line 31) | async def register_scoring_function(
    method unregister_scoring_function (line 36) | async def unregister_scoring_function(

FILE: src/llama_stack_api/scoring_functions/fastapi_routes.py
  function create_router (line 37) | def create_router(impl: ScoringFunctions) -> APIRouter:

FILE: src/llama_stack_api/scoring_functions/models.py
  class ScoringFnParamsType (line 24) | class ScoringFnParamsType(StrEnum):
  class AggregationFunctionType (line 37) | class AggregationFunctionType(StrEnum):
  class LLMAsJudgeScoringFnParams (line 54) | class LLMAsJudgeScoringFnParams(BaseModel):
  class RegexParserScoringFnParams (line 77) | class RegexParserScoringFnParams(BaseModel):
  class BasicScoringFnParams (line 96) | class BasicScoringFnParams(BaseModel):
  class ListScoringFunctionsRequest (line 117) | class ListScoringFunctionsRequest(BaseModel):
  class GetScoringFunctionRequest (line 124) | class GetScoringFunctionRequest(BaseModel):
  class RegisterScoringFunctionRequest (line 131) | class RegisterScoringFunctionRequest(BaseModel):
  class UnregisterScoringFunctionRequest (line 148) | class UnregisterScoringFunctionRequest(BaseModel):
  class CommonScoringFnFields (line 154) | class CommonScoringFnFields(BaseModel):
  class ScoringFn (line 170) | class ScoringFn(CommonScoringFnFields, Resource):
    method scoring_fn_id (line 178) | def scoring_fn_id(self) -> str:
    method provider_scoring_fn_id (line 182) | def provider_scoring_fn_id(self) -> str | None:
  class ScoringFnInput (line 186) | class ScoringFnInput(CommonScoringFnFields, BaseModel):
  class ListScoringFunctionsResponse (line 193) | class ListScoringFunctionsResponse(BaseModel):

FILE: src/llama_stack_api/shields/api.py
  class Shields (line 24) | class Shields(Protocol):
    method list_shields (line 25) | async def list_shields(self) -> ListShieldsResponse:
    method get_shield (line 29) | async def get_shield(self, request: GetShieldRequest) -> Shield:
    method register_shield (line 33) | async def register_shield(self, request: RegisterShieldRequest) -> Shi...
    method unregister_shield (line 37) | async def unregister_shield(self, request: UnregisterShieldRequest) ->...

FILE: src/llama_stack_api/shields/fastapi_routes.py
  function create_router (line 34) | def create_router(impl: Shields) -> APIRouter:

FILE: src/llama_stack_api/shields/models.py
  class CommonShieldFields (line 21) | class CommonShieldFields(BaseModel):
  class Shield (line 26) | class Shield(CommonShieldFields, Resource):
    method shield_id (line 32) | def shield_id(self) -> str:
    method provider_shield_id (line 36) | def provider_shield_id(self) -> str | None:
  class ShieldInput (line 40) | class ShieldInput(CommonShieldFields):
  class ListShieldsResponse (line 47) | class ListShieldsResponse(BaseModel):
  class GetShieldRequest (line 54) | class GetShieldRequest(BaseModel):
  class RegisterShieldRequest (line 61) | class RegisterShieldRequest(BaseModel):
  class UnregisterShieldRequest (line 71) | class UnregisterShieldRequest(BaseModel):

FILE: src/llama_stack_api/tools.py
  class ToolDef (line 20) | class ToolDef(BaseModel):
  class ToolGroupInput (line 40) | class ToolGroupInput(BaseModel):
  class ToolGroup (line 56) | class ToolGroup(Resource):
  class ToolInvocationResult (line 70) | class ToolInvocationResult(BaseModel):
  class ToolStore (line 85) | class ToolStore(Protocol):
    method get_tool (line 86) | async def get_tool(self, tool_name: str) -> ToolDef: ...
    method get_tool_group (line 87) | async def get_tool_group(self, toolgroup_id: str) -> ToolGroup: ...
  class ListToolGroupsResponse (line 91) | class ListToolGroupsResponse(BaseModel):
  class ListToolDefsResponse (line 101) | class ListToolDefsResponse(BaseModel):
  class ToolGroups (line 111) | class ToolGroups(Protocol):
    method register_tool_group (line 113) | async def register_tool_group(
    method get_tool_group (line 130) | async def get_tool_group(
    method list_tool_groups (line 142) | async def list_tool_groups(self) -> ListToolGroupsResponse:
    method list_tools (line 150) | async def list_tools(self, toolgroup_id: str | None = None) -> ListToo...
    method get_tool (line 159) | async def get_tool(
    method unregister_toolgroup (line 171) | async def unregister_toolgroup(
  class SpecialToolGroup (line 182) | class SpecialToolGroup(Enum):
  class ToolRuntime (line 192) | class ToolRuntime(Protocol):
    method list_runtime_tools (line 197) | async def list_runtime_tools(
    method invoke_tool (line 213) | async def invoke_tool(

FILE: src/llama_stack_api/validators.py
  function validate_embeddings_input_is_text (line 16) | def validate_embeddings_input_is_text(

FILE: src/llama_stack_api/vector_io/api.py
  class VectorStoreTable (line 44) | class VectorStoreTable(Protocol):
    method get_vector_store (line 45) | def get_vector_store(self, vector_store_id: str) -> VectorStore | None...
  class VectorIO (line 49) | class VectorIO(Protocol):
    method insert_chunks (line 54) | async def insert_chunks(
    method query_chunks (line 61) | async def query_chunks(
    method openai_create_vector_store (line 69) | async def openai_create_vector_store(
    method openai_list_vector_stores (line 79) | async def openai_list_vector_stores(
    method openai_retrieve_vector_store (line 89) | async def openai_retrieve_vector_store(
    method openai_update_vector_store (line 96) | async def openai_update_vector_store(
    method openai_delete_vector_store (line 104) | async def openai_delete_vector_store(
    method openai_search_vector_store (line 111) | async def openai_search_vector_store(
    method openai_attach_file_to_vector_store (line 122) | async def openai_attach_file_to_vector_store(
    method openai_list_files_in_vector_store (line 130) | async def openai_list_files_in_vector_store(
    method openai_retrieve_vector_store_file (line 142) | async def openai_retrieve_vector_store_file(
    method openai_retrieve_vector_store_file_contents (line 150) | async def openai_retrieve_vector_store_file_contents(
    method openai_update_vector_store_file (line 160) | async def openai_update_vector_store_file(
    method openai_delete_vector_store_file (line 169) | async def openai_delete_vector_store_file(
    method openai_create_vector_store_file_batch (line 177) | async def openai_create_vector_store_file_batch(
    method openai_retrieve_vector_store_file_batch (line 188) | async def openai_retrieve_vector_store_file_batch(
    method openai_list_files_in_vector_store_file_batch (line 196) | async def openai_list_files_in_vector_store_file_batch(
    method openai_cancel_vector_store_file_batch (line 209) | async def openai_cancel_vector_store_file_batch(

FILE: src/llama_stack_api/vector_io/fastapi_routes.py
  function create_router (line 49) | def create_router(impl: VectorIO) -> APIRouter:
  function _raise_or_http_400_for_value_error (line 378) | def _raise_or_http_400_for_value_error(request: Request, exc: ValueError...

FILE: src/llama_stack_api/vector_io/models.py
  class ChunkMetadata (line 30) | class ChunkMetadata(BaseModel):
  class Chunk (line 59) | class Chunk(BaseModel):
    method document_id (line 75) | def document_id(self) -> str | None:
  class EmbeddedChunk (line 92) | class EmbeddedChunk(Chunk):
  class QueryChunksResponse (line 107) | class QueryChunksResponse(BaseModel):
  class VectorStoreFileCounts (line 119) | class VectorStoreFileCounts(BaseModel):
  class VectorStoreObject (line 137) | class VectorStoreObject(BaseModel):
  class VectorStoreCreateRequest (line 167) | class VectorStoreCreateRequest(BaseModel):
  class VectorStoreModifyRequest (line 185) | class VectorStoreModifyRequest(BaseModel):
  class VectorStoreListResponse (line 199) | class VectorStoreListResponse(BaseModel):
  class VectorStoreSearchRequest (line 217) | class VectorStoreSearchRequest(BaseModel):
  class VectorStoreContent (line 235) | class VectorStoreContent(BaseModel):
  class VectorStoreSearchResponse (line 253) | class VectorStoreSearchResponse(BaseModel):
  class VectorStoreSearchResponsePage (line 271) | class VectorStoreSearchResponsePage(BaseModel):
  class VectorStoreDeleteResponse (line 289) | class VectorStoreDeleteResponse(BaseModel):
  class VectorStoreFileContentResponse (line 303) | class VectorStoreFileContentResponse(BaseModel):
  class VectorStoreChunkingStrategyAuto (line 319) | class VectorStoreChunkingStrategyAuto(BaseModel):
  class VectorStoreChunkingStrategyStaticConfig (line 329) | class VectorStoreChunkingStrategyStaticConfig(BaseModel):
  class VectorStoreChunkingStrategyStatic (line 341) | class VectorStoreChunkingStrategyStatic(BaseModel):
  function _strip_context_prompt_default (line 362) | def _strip_context_prompt_default(schema: dict) -> None:
  class VectorStoreChunkingStrategyContextualConfig (line 371) | class VectorStoreChunkingStrategyContextualConfig(BaseModel):
    method validate_config (line 406) | def validate_config(self) -> "VectorStoreChunkingStrategyContextualCon...
  class VectorStoreChunkingStrategyContextual (line 423) | class VectorStoreChunkingStrategyContextual(BaseModel):
  class SearchRankingOptions (line 439) | class SearchRankingOptions(BaseModel):
    method validate_weights (line 498) | def validate_weights(cls, v: dict[str, float] | None) -> dict[str, flo...
  class VectorStoreFileLastError (line 510) | class VectorStoreFileLastError(BaseModel):
  function _sanitize_vector_store_attributes (line 545) | def _sanitize_vector_store_attributes(metadata: dict[str, Any] | None) -...
  class VectorStoreFileObject (line 585) | class VectorStoreFileObject(BaseModel):
    method _validate_attributes (line 611) | def _validate_attributes(cls, v: dict[str, Any] | None) -> dict[str, s...
  class VectorStoreListFilesResponse (line 617) | class VectorStoreListFilesResponse(BaseModel):
  class VectorStoreFileDeleteResponse (line 635) | class VectorStoreFileDeleteResponse(BaseModel):
  class VectorStoreFileBatchObject (line 649) | class VectorStoreFileBatchObject(BaseModel):
  class VectorStoreFilesListInBatchResponse (line 669) | class VectorStoreFilesListInBatchResponse(BaseModel):
  class OpenAICreateVectorStoreRequestWithExtraBody (line 688) | class OpenAICreateVectorStoreRequestWithExtraBody(BaseModel, extra="allo...
  class OpenAICreateVectorStoreFileBatchRequestWithExtraBody (line 707) | class OpenAICreateVectorStoreFileBatchRequestWithExtraBody(BaseModel, ex...
  class ChunkForDeletion (line 721) | class ChunkForDeletion(BaseModel):
  class InsertChunksRequest (line 733) | class InsertChunksRequest(BaseModel):
  class DeleteChunksRequest (line 742) | class DeleteChunksRequest(BaseModel):
  class QueryChunksRequest (line 750) | class QueryChunksRequest(BaseModel):
  class OpenAIUpdateVectorStoreRequest (line 759) | class OpenAIUpdateVectorStoreRequest(BaseModel):
  class OpenAISearchVectorStoreRequest (line 768) | class OpenAISearchVectorStoreRequest(BaseModel):
  class OpenAIAttachFileRequest (line 780) | class OpenAIAttachFileRequest(BaseModel):
  class OpenAIUpdateVectorStoreFileRequest (line 791) | class OpenAIUpdateVectorStoreFileRequest(BaseModel):

FILE: src/llama_stack_api/vector_stores.py
  class VectorStore (line 15) | class VectorStore(Resource):
    method vector_store_id (line 30) | def vector_store_id(self) -> str:
    method provider_vector_store_id (line 34) | def provider_vector_store_id(self) -> str | None:
  class VectorStoreInput (line 38) | class VectorStoreInput(BaseModel):

FILE: src/llama_stack_ui/app/api/v1/[...path]/route.ts
  constant BACKEND_URL (line 4) | const BACKEND_URL =
  function proxyRequest (line 8) | async function proxyRequest(request: NextRequest, method: string) {
  function GET (line 115) | async function GET(request: NextRequest) {
  function POST (line 119) | async function POST(request: NextRequest) {
  function PUT (line 123) | async function PUT(request: NextRequest) {
  function DELETE (line 127) | async function DELETE(request: NextRequest) {
  function PATCH (line 131) | async function PATCH(request: NextRequest) {
  function OPTIONS (line 135) | async function OPTIONS(request: NextRequest) {

FILE: src/llama_stack_ui/app/auth/signin/page.tsx
  function SignInPage (line 16) | function SignInPage() {

FILE: src/llama_stack_ui/app/chat-playground/chunk-processor.test.tsx
  function createProcessChunk (line 4) | function createProcessChunk() {

FILE: src/llama_stack_ui/app/chat-playground/page.tsx
  type ModelWithMetadata (line 24) | type ModelWithMetadata = Model & {
  function ChatPlaygroundPage (line 39) | function ChatPlaygroundPage() {

FILE: src/llama_stack_ui/app/layout.tsx
  function Layout (line 30) | function Layout({ children }: { children: React.ReactNode }) {

FILE: src/llama_stack_ui/app/logs/chat-completions/[id]/page.tsx
  function ChatCompletionDetailPage (line 9) | function ChatCompletionDetailPage() {

FILE: src/llama_stack_ui/app/logs/chat-completions/layout.tsx
  function ChatCompletionsLayout (line 6) | function ChatCompletionsLayout({

FILE: src/llama_stack_ui/app/logs/chat-completions/page.tsx
  function ChatCompletionsPage (line 5) | function ChatCompletionsPage() {

FILE: src/llama_stack_ui/app/logs/files/[id]/page.tsx
  function FileDetailPage (line 6) | function FileDetailPage() {

FILE: src/llama_stack_ui/app/logs/files/layout.tsx
  function FileDetailLayout (line 9) | function FileDetailLayout({

FILE: src/llama_stack_ui/app/logs/files/page.tsx
  function FilesPage (line 6) | function FilesPage() {

FILE: src/llama_stack_ui/app/logs/responses/[id]/page.tsx
  function ResponseDetailPage (line 10) | function ResponseDetailPage() {

FILE: src/llama_stack_ui/app/logs/responses/layout.tsx
  function ResponsesLayout (line 6) | function ResponsesLayout({

FILE: src/llama_stack_ui/app/logs/responses/page.tsx
  function ResponsesPage (line 5) | function ResponsesPage() {

FILE: src/llama_stack_ui/app/logs/vector-stores/[id]/files/[fileId]/contents/[contentId]/page.tsx
  function ContentDetailPage (line 26) | function ContentDetailPage() {

FILE: src/llama_stack_ui/app/logs/vector-stores/[id]/files/[fileId]/contents/page.tsx
  function ContentsListPage (line 35) | function ContentsListPage() {

FILE: src/llama_stack_ui/app/logs/vector-stores/[id]/files/[fileId]/page.tsx
  function FileDetailPage (line 28) | function FileDetailPage() {

FILE: src/llama_stack_ui/app/logs/vector-stores/[id]/page.tsx
  function VectorStoreDetailPage (line 10) | function VectorStoreDetailPage() {

FILE: src/llama_stack_ui/app/logs/vector-stores/layout.tsx
  function VectorStoreDetailLayout (line 9) | function VectorStoreDetailLayout({

FILE: src/llama_stack_ui/app/logs/vector-stores/page.tsx
  function VectorStoresPage (line 29) | function VectorStoresPage() {

FILE: src/llama_stack_ui/app/page.tsx
  function Home (line 1) | function Home() {

FILE: src/llama_stack_ui/app/prompts/page.tsx
  function PromptsPage (line 3) | function PromptsPage() {

FILE: src/llama_stack_ui/components/chat-completions/chat-completion-detail.tsx
  type ChatCompletionDetailViewProps (line 15) | interface ChatCompletionDetailViewProps {
  function ChatCompletionDetailView (line 22) | function ChatCompletionDetailView({

FILE: src/llama_stack_ui/components/chat-completions/chat-completions-table.tsx
  type ChatCompletionsTableProps (line 16) | interface ChatCompletionsTableProps {
  function formatChatCompletionToRow (line 21) | function formatChatCompletionToRow(completion: ChatCompletion): LogTable...
  function ChatCompletionsTable (line 32) | function ChatCompletionsTable({

FILE: src/llama_stack_ui/components/chat-completions/chat-messasge-item.tsx
  type ChatMessageItemProps (line 12) | interface ChatMessageItemProps {
  function ChatMessageItem (line 15) | function ChatMessageItem({ message }: ChatMessageItemProps) {

FILE: src/llama_stack_ui/components/chat-playground/chat-message.tsx
  type Animation (line 57) | type Animation = VariantProps<typeof chatBubbleVariants>["animation"];
  type Attachment (line 59) | interface Attachment {
  type PartialToolCall (line 65) | interface PartialToolCall {
  type ToolCall (line 70) | interface ToolCall {
  type ToolResult (line 75) | interface ToolResult {
  type ToolInvocation (line 84) | type ToolInvocation = PartialToolCall | ToolCall | ToolResult;
  type ReasoningPart (line 86) | interface ReasoningPart {
  type ToolInvocationPart (line 91) | interface ToolInvocationPart {
  type TextPart (line 96) | interface TextPart {
  type SourcePart (line 102) | interface SourcePart {
  type FilePart (line 107) | interface FilePart {
  type StepStartPart (line 113) | interface StepStartPart {
  type MessagePart (line 117) | type MessagePart =
  type Message (line 125) | interface Message {
  type ChatMessageProps (line 135) | interface ChatMessageProps extends Message {
  function dataUrlToUint8Array (line 280) | function dataUrlToUint8Array(data: string) {
  function ToolCall (line 327) | function ToolCall({

FILE: src/llama_stack_ui/components/chat-playground/chat.tsx
  type ChatPropsBase (line 21) | interface ChatPropsBase {
  type ChatPropsWithoutSuggestions (line 41) | interface ChatPropsWithoutSuggestions extends ChatPropsBase {
  type ChatPropsWithSuggestions (line 46) | interface ChatPropsWithSuggestions extends ChatPropsBase {
  type ChatProps (line 51) | type ChatProps = ChatPropsWithoutSuggestions | ChatPropsWithSuggestions;
  function Chat (line 53) | function Chat({
  function ChatMessages (line 252) | function ChatMessages({
  type ChatFormProps (line 309) | interface ChatFormProps {
  function createFileList (line 351) | function createFileList(files: File[] | FileList): FileList {

FILE: src/llama_stack_ui/components/chat-playground/conversations.test.tsx
  type ChatSession (line 7) | interface ChatSession {

FILE: src/llama_stack_ui/components/chat-playground/conversations.tsx
  type ChatSession (line 23) | interface ChatSession {
  type SessionManagerProps (line 35) | interface SessionManagerProps {
  constant CURRENT_SESSION_KEY (line 42) | const CURRENT_SESSION_KEY = "chat-playground-current-session";
  function Conversations (line 77) | function Conversations({

FILE: src/llama_stack_ui/components/chat-playground/interrupt-prompt.tsx
  type InterruptPromptProps (line 6) | interface InterruptPromptProps {
  function InterruptPrompt (line 11) | function InterruptPrompt({ isOpen, close }: InterruptPromptProps) {

FILE: src/llama_stack_ui/components/chat-playground/markdown-renderer.tsx
  type MarkdownRendererProps (line 8) | interface MarkdownRendererProps {
  function MarkdownRenderer (line 12) | function MarkdownRenderer({ children }: MarkdownRendererProps) {
  type HighlightedPre (line 22) | interface HighlightedPre extends React.HTMLAttributes<HTMLPreElement> {
  type CodeBlockProps (line 115) | interface CodeBlockProps extends React.HTMLAttributes<HTMLPreElement> {
  function childrenTakeAllStringContents (line 158) | function childrenTakeAllStringContents(element: unknown): string {
  constant COMPONENTS (line 178) | const COMPONENTS = {
  function withClass (line 232) | function withClass(Tag: keyof JSX.IntrinsicElements, classes: string) {

FILE: src/llama_stack_ui/components/chat-playground/message-components.tsx
  type MessageBlockProps (line 3) | interface MessageBlockProps {
  type ToolCallBlockProps (line 35) | interface ToolCallBlockProps {

FILE: src/llama_stack_ui/components/chat-playground/message-input.tsx
  type MessageInputBaseProps (line 16) | interface MessageInputBaseProps
  type MessageInputWithoutAttachmentProps (line 27) | interface MessageInputWithoutAttachmentProps extends MessageInputBasePro...
  type MessageInputWithAttachmentsProps (line 31) | interface MessageInputWithAttachmentsProps extends MessageInputBaseProps {
  type MessageInputProps (line 37) | type MessageInputProps =
  function MessageInput (line 41) | function MessageInput({
  type FileUploadOverlayProps (line 330) | interface FileUploadOverlayProps {
  function FileUploadOverlay (line 334) | function FileUploadOverlay({ isDragging }: FileUploadOverlayProps) {
  function TranscribingOverlay (line 354) | function TranscribingOverlay() {
  type RecordingPromptProps (line 384) | interface RecordingPromptProps {
  function RecordingPrompt (line 389) | function RecordingPrompt({ isVisible, onStopRecording }: RecordingPrompt...
  type RecordingControlsProps (line 417) | interface RecordingControlsProps {
  function RecordingControls (line 425) | function RecordingControls({

FILE: src/llama_stack_ui/components/chat-playground/message-list.tsx
  type AdditionalMessageOptions (line 8) | type AdditionalMessageOptions = Omit<ChatMessageProps, keyof Message>;
  type MessageListProps (line 10) | interface MessageListProps {
  function MessageList (line 19) | function MessageList({

FILE: src/llama_stack_ui/components/chat-playground/prompt-suggestions.tsx
  type PromptSuggestionsProps (line 1) | interface PromptSuggestionsProps {
  function PromptSuggestions (line 7) | function PromptSuggestions({

FILE: src/llama_stack_ui/components/chat-playground/typing-indicator.tsx
  function TypingIndicator (line 3) | function TypingIndicator() {

FILE: src/llama_stack_ui/components/chat-playground/vector-db-creator.tsx
  type VectorDBCreatorProps (line 17) | interface VectorDBCreatorProps {
  type VectorDBProvider (line 23) | interface VectorDBProvider {
  function VectorDBCreator (line 29) | function VectorDBCreator({

FILE: src/llama_stack_ui/components/files/csv-viewer.tsx
  type CSVViewerProps (line 15) | interface CSVViewerProps {
  type ParsedCSV (line 20) | interface ParsedCSV {
  constant MAX_CSV_SIZE (line 27) | const MAX_CSV_SIZE = 10 * 1024 * 1024;
  constant WARN_CSV_SIZE (line 28) | const WARN_CSV_SIZE = 5 * 1024 * 1024;
  function CSVViewer (line 30) | function CSVViewer({ content }: CSVViewerProps) {

FILE: src/llama_stack_ui/components/files/file-detail.tsx
  constant MAX_TEXT_PREVIEW_SIZE (line 30) | const MAX_TEXT_PREVIEW_SIZE = 50 * 1024 * 1024;
  constant WARN_TEXT_PREVIEW_SIZE (line 31) | const WARN_TEXT_PREVIEW_SIZE = 10 * 1024 * 1024;
  function FileDetail (line 33) | function FileDetail() {

FILE: src/llama_stack_ui/components/files/file-editor.tsx
  type FileEditorProps (line 24) | interface FileEditorProps {
  function FileEditor (line 31) | function FileEditor({

FILE: src/llama_stack_ui/components/files/file-upload-zone.tsx
  type FileUploadZoneProps (line 15) | interface FileUploadZoneProps {
  function FileUploadZone (line 24) | function FileUploadZone({

FILE: src/llama_stack_ui/components/files/files-management.tsx
  function FilesManagement (line 30) | function FilesManagement() {

FILE: src/llama_stack_ui/components/files/json-viewer.tsx
  type JsonViewerProps (line 8) | interface JsonViewerProps {
  type JsonValue (line 13) | type JsonValue =
  type JsonNode (line 21) | interface JsonNode {
  function JsonViewer (line 28) | function JsonViewer({ content }: JsonViewerProps) {

FILE: src/llama_stack_ui/components/layout/app-sidebar.tsx
  type SidebarItem (line 86) | interface SidebarItem {
  function AppSidebar (line 92) | function AppSidebar() {

FILE: src/llama_stack_ui/components/layout/detail-layout.tsx
  function DetailLoadingView (line 5) | function DetailLoadingView() {
  function DetailErrorView (line 43) | function DetailErrorView({
  function DetailNotFoundView (line 62) | function DetailNotFoundView({
  type PropertyItemProps (line 77) | interface PropertyItemProps {
  function PropertyItem (line 84) | function PropertyItem({
  type PropertiesCardProps (line 106) | interface PropertiesCardProps {
  function PropertiesCard (line 110) | function PropertiesCard({ children }: PropertiesCardProps) {
  type DetailLayoutProps (line 125) | interface DetailLayoutProps {
  function DetailLayout (line 131) | function DetailLayout({

FILE: src/llama_stack_ui/components/layout/logs-layout.tsx
  type LogsLayoutProps (line 11) | interface LogsLayoutProps {
  function LogsLayout (line 17) | function LogsLayout({

FILE: src/llama_stack_ui/components/layout/page-breadcrumb.tsx
  type BreadcrumbSegment (line 14) | interface BreadcrumbSegment {
  type PageBreadcrumbProps (line 19) | interface PageBreadcrumbProps {
  function PageBreadcrumb (line 24) | function PageBreadcrumb({ segments, className }: PageBreadcrumbProps) {

FILE: src/llama_stack_ui/components/logs/logs-table.tsx
  type LogTableRow (line 20) | interface LogTableRow {
  type LogsTableProps (line 29) | interface LogsTableProps {
  function LogsTable (line 46) | function LogsTable({

FILE: src/llama_stack_ui/components/prompts/prompt-editor.tsx
  type PromptEditorProps (line 20) | interface PromptEditorProps {
  function PromptEditor (line 28) | function PromptEditor({

FILE: src/llama_stack_ui/components/prompts/prompt-list.tsx
  type PromptListProps (line 18) | interface PromptListProps {
  function PromptList (line 24) | function PromptList({ prompts, onEdit, onDelete }: PromptListProps) {

FILE: src/llama_stack_ui/components/prompts/prompt-management.tsx
  function PromptManagement (line 11) | function PromptManagement() {

FILE: src/llama_stack_ui/components/prompts/types.ts
  type Prompt (line 1) | interface Prompt {
  type PromptFormData (line 9) | interface PromptFormData {
  type PromptFilters (line 14) | interface PromptFilters {

FILE: src/llama_stack_ui/components/providers/session-provider.tsx
  function SessionProvider (line 5) | function SessionProvider({ children }: { children: React.ReactNode }) {

FILE: src/llama_stack_ui/components/responses/grouping/grouped-items-display.tsx
  type GroupedItemsDisplayProps (line 10) | interface GroupedItemsDisplayProps {
  function GroupedItemsDisplay (line 16) | function GroupedItemsDisplay({

FILE: src/llama_stack_ui/components/responses/hooks/function-call-grouping.ts
  type GroupedItem (line 8) | interface GroupedItem {
  function useFunctionCallGrouping (line 20) | function useFunctionCallGrouping(

FILE: src/llama_stack_ui/components/responses/items/function-call-item.tsx
  type FunctionCallItemProps (line 7) | interface FunctionCallItemProps {
  function FunctionCallItemComponent (line 13) | function FunctionCallItemComponent({

FILE: src/llama_stack_ui/components/responses/items/generic-item.tsx
  type GenericItemProps (line 7) | interface GenericItemProps {
  function GenericItemComponent (line 13) | function GenericItemComponent({

FILE: src/llama_stack_ui/components/responses/items/grouped-function-call-item.tsx
  type GroupedFunctionCallItemProps (line 7) | interface GroupedFunctionCallItemProps {
  function GroupedFunctionCallItemComponent (line 14) | function GroupedFunctionCallItemComponent({

FILE: src/llama_stack_ui/components/responses/items/item-renderer.tsx
  type ItemRendererProps (line 12) | interface ItemRendererProps {
  function ItemRenderer (line 19) | function ItemRenderer({

FILE: src/llama_stack_ui/components/responses/items/message-item.tsx
  type MessageItemProps (line 4) | interface MessageItemProps {
  function MessageItemComponent (line 11) | function MessageItemComponent({

FILE: src/llama_stack_ui/components/responses/items/web-search-item.tsx
  type WebSearchItemProps (line 7) | interface WebSearchItemProps {
  function WebSearchItemComponent (line 13) | function WebSearchItemComponent({

FILE: src/llama_stack_ui/components/responses/responses-detail.tsx
  type ResponseDetailViewProps (line 16) | interface ResponseDetailViewProps {
  function ResponseDetailView (line 26) | function ResponseDetailView({

FILE: src/llama_stack_ui/components/responses/responses-table.tsx
  type ResponsesTableProps (line 21) | interface ResponsesTableProps {
  function getInputText (line 49) | function getInputText(response: OpenAIResponse): string {
  function getOutputText (line 57) | function getOutputText(response: OpenAIResponse): string {
  function extractContentFromItem (line 85) | function extractContentFromItem(item: {
  function formatFunctionCall (line 104) | function formatFunctionCall(functionCall: FunctionCallItem): string {
  function formatWebSearchCall (line 110) | function formatWebSearchCall(webSearchCall: WebSearchCallItem): string {
  function formatResponseToRow (line 114) | function formatResponseToRow(response: OpenAIResponse): LogTableRow {
  function ResponsesTable (line 125) | function ResponsesTable({ paginationOptions }: ResponsesTableProps) {

FILE: src/llama_stack_ui/components/responses/utils/item-types.ts
  type BaseItem (line 12) | interface BaseItem {
  type MessageItem (line 17) | type MessageItem = ResponseMessage;
  type FunctionCallItem (line 18) | type FunctionCallItem = ResponseToolCall & { type: "function_call" };
  type WebSearchCallItem (line 19) | type WebSearchCallItem = ResponseToolCall & { type: "web_search_call" };
  type FunctionCallOutputItem (line 20) | type FunctionCallOutputItem = BaseItem & {
  type AnyResponseItem (line 26) | type AnyResponseItem =
  function isMessageInput (line 31) | function isMessageInput(
  function isMessageItem (line 37) | function isMessageItem(item: AnyResponseItem): item is MessageItem {
  function isFunctionCallItem (line 41) | function isFunctionCallItem(
  function isWebSearchCallItem (line 47) | function isWebSearchCallItem(
  function isFunctionCallOutputItem (line 53) | function isFunctionCallOutputItem(

FILE: src/llama_stack_ui/components/ui/audio-visualizer.tsx
  constant AUDIO_CONFIG (line 6) | const AUDIO_CONFIG = {
  type AudioVisualizerProps (line 19) | interface AudioVisualizerProps {
  function AudioVisualizer (line 25) | function AudioVisualizer({

FILE: src/llama_stack_ui/components/ui/badge.tsx
  type BadgeProps (line 26) | interface BadgeProps
  function Badge (line 30) | function Badge({ className, variant, ...props }: BadgeProps) {

FILE: src/llama_stack_ui/components/ui/breadcrumb.tsx
  function Breadcrumb (line 7) | function Breadcrumb({ ...props }: React.ComponentProps<"nav">) {
  function BreadcrumbList (line 11) | function BreadcrumbList({ className, ...props }: React.ComponentProps<"o...
  function BreadcrumbItem (line 24) | function BreadcrumbItem({ className, ...props }: React.ComponentProps<"l...
  function BreadcrumbLink (line 34) | function BreadcrumbLink({
  function BreadcrumbPage (line 52) | function BreadcrumbPage({ className, ...props }: React.ComponentProps<"s...
  function BreadcrumbSeparator (line 65) | function BreadcrumbSeparator({
  function BreadcrumbEllipsis (line 83) | function BreadcrumbEllipsis({

FILE: src/llama_stack_ui/components/ui/button.tsx
  function Button (line 38) | function Button({

FILE: src/llama_stack_ui/components/ui/card.tsx
  function Card (line 5) | function Card({ className, ...props }: React.ComponentProps<"div">) {
  function CardHeader (line 18) | function CardHeader({ className, ...props }: React.ComponentProps<"div">) {
  function CardTitle (line 31) | function CardTitle({ className, ...props }: React.ComponentProps<"div">) {
  function CardDescription (line 41) | function CardDescription({ className, ...props }: React.ComponentProps<"...
  function CardAction (line 51) | function CardAction({ className, ...props }: React.ComponentProps<"div">) {
  function CardContent (line 64) | function CardContent({ className, ...props }: React.ComponentProps<"div"...
  function CardFooter (line 74) | function CardFooter({ className, ...props }: React.ComponentProps<"div">) {

FILE: src/llama_stack_ui/components/ui/collapsible.tsx
  function Collapsible (line 5) | function Collapsible({
  function CollapsibleTrigger (line 11) | function CollapsibleTrigger({
  function CollapsibleContent (line 22) | function CollapsibleContent({

FILE: src/llama_stack_ui/components/ui/copy-button.tsx
  type CopyButtonProps (line 9) | type CopyButtonProps = {
  function CopyButton (line 14) | function CopyButton({ content, copyMessage }: CopyButtonProps) {

FILE: src/llama_stack_ui/components/ui/dropdown-menu.tsx
  function DropdownMenu (line 9) | function DropdownMenu({
  function DropdownMenuPortal (line 15) | function DropdownMenuPortal({
  function DropdownMenuTrigger (line 23) | function DropdownMenuTrigger({
  function DropdownMenuContent (line 34) | function DropdownMenuContent({
  function DropdownMenuGroup (line 54) | function DropdownMenuGroup({
  function DropdownMenuItem (line 62) | function DropdownMenuItem({
  function DropdownMenuCheckboxItem (line 85) | function DropdownMenuCheckboxItem({
  function DropdownMenuRadioGroup (line 111) | function DropdownMenuRadioGroup({
  function DropdownMenuRadioItem (line 122) | function DropdownMenuRadioItem({
  function DropdownMenuLabel (line 146) | function DropdownMenuLabel({
  function DropdownMenuSeparator (line 166) | function DropdownMenuSeparator({
  function DropdownMenuShortcut (line 179) | function DropdownMenuShortcut({
  function DropdownMenuSub (line 195) | function DropdownMenuSub({
  function DropdownMenuSubTrigger (line 201) | function DropdownMenuSubTrigger({
  function DropdownMenuSubContent (line 225) | function DropdownMenuSubContent({

FILE: src/llama_stack_ui/components/ui/file-preview.tsx
  type FilePreviewProps (line 7) | interface FilePreviewProps {

FILE: src/llama_stack_ui/components/ui/input.tsx
  function Input (line 5) | function Input({ className, type, ...props }: React.ComponentProps<"inpu...

FILE: src/llama_stack_ui/components/ui/mode-toggle.tsx
  function ModeToggle (line 15) | function ModeToggle() {

FILE: src/llama_stack_ui/components/ui/select.tsx
  function Select (line 9) | function Select({
  function SelectGroup (line 15) | function SelectGroup({
  function SelectValue (line 21) | function SelectValue({
  function SelectTrigger (line 27) | function SelectTrigger({
  function SelectContent (line 53) | function SelectContent({
  function SelectLabel (line 88) | function SelectLabel({
  function SelectItem (line 101) | function SelectItem({
  function SelectSeparator (line 125) | function SelectSeparator({
  function SelectScrollUpButton (line 138) | function SelectScrollUpButton({
  function SelectScrollDownButton (line 156) | function SelectScrollDownButton({

FILE: src/llama_stack_ui/components/ui/separator.tsx
  function Separator (line 8) | function Separator({

FILE: src/llama_stack_ui/components/ui/sheet.tsx
  function Sheet (line 9) | function Sheet({ ...props }: React.ComponentProps<typeof SheetPrimitive....
  function SheetTrigger (line 13) | function SheetTrigger({
  function SheetClose (line 19) | function SheetClose({
  function SheetPortal (line 25) | function SheetPortal({
  function SheetOverlay (line 31) | function SheetOverlay({
  function SheetContent (line 47) | function SheetContent({
  function SheetHeader (line 84) | function SheetHeader({ className, ...props }: React.ComponentProps<"div"...
  function SheetFooter (line 94) | function SheetFooter({ className, ...props }: React.ComponentProps<"div"...
  function SheetTitle (line 104) | function SheetTitle({
  function SheetDescription (line 117) | function SheetDescription({

FILE: src/llama_stack_ui/components/ui/sidebar.tsx
  constant SIDEBAR_COOKIE_NAME (line 28) | const SIDEBAR_COOKIE_NAME = "sidebar_state";
  constant SIDEBAR_COOKIE_MAX_AGE (line 29) | const SIDEBAR_COOKIE_MAX_AGE = 60 * 60 * 24 * 7;
  constant SIDEBAR_WIDTH (line 30) | const SIDEBAR_WIDTH = "16rem";
  constant SIDEBAR_WIDTH_MOBILE (line 31) | const SIDEBAR_WIDTH_MOBILE = "18rem";
  constant SIDEBAR_WIDTH_ICON (line 32) | const SIDEBAR_WIDTH_ICON = "3rem";
  constant SIDEBAR_KEYBOARD_SHORTCUT (line 33) | const SIDEBAR_KEYBOARD_SHORTCUT = "b";
  type SidebarContextProps (line 35) | type SidebarContextProps = {
  function useSidebar (line 47) | function useSidebar() {
  function SidebarProvider (line 56) | function SidebarProvider({
  function Sidebar (line 154) | function Sidebar({
  function SidebarTrigger (line 256) | function SidebarTrigger({
  function SidebarRail (line 282) | function SidebarRail({ className, ...props }: React.ComponentProps<"butt...
  function SidebarInset (line 307) | function SidebarInset({ className, ...props }: React.ComponentProps<"mai...
  function SidebarInput (line 321) | function SidebarInput({
  function SidebarHeader (line 335) | function SidebarHeader({ className, ...props }: React.ComponentProps<"di...
  function SidebarFooter (line 346) | function SidebarFooter({ className, ...props }: React.ComponentProps<"di...
  function SidebarSeparator (line 357) | function SidebarSeparator({
  function SidebarContent (line 371) | function SidebarContent({ className, ...props }: React.ComponentProps<"d...
  function SidebarGroup (line 385) | function SidebarGroup({ className, ...props }: React.ComponentProps<"div...
  function SidebarGroupLabel (line 396) | function SidebarGroupLabel({
  function SidebarGroupAction (line 417) | function SidebarGroupAction({
  function SidebarGroupContent (line 440) | function SidebarGroupContent({
  function SidebarMenu (line 454) | function SidebarMenu({ className, ...props }: React.ComponentProps<"ul">) {
  function SidebarMenuItem (line 465) | function SidebarMenuItem({ className, ...props }: React.ComponentProps<"...
  function SidebarMenuButton (line 498) | function SidebarMenuButton({
  function SidebarMenuAction (line 548) | function SidebarMenuAction({
  function SidebarMenuBadge (line 580) | function SidebarMenuBadge({
  function SidebarMenuSkeleton (line 602) | function SidebarMenuSkeleton({
  function SidebarMenuSub (line 640) | function SidebarMenuSub({ className, ...props }: React.ComponentProps<"u...
  function SidebarMenuSubItem (line 655) | function SidebarMenuSubItem({
  function SidebarMenuSubButton (line 669) | function SidebarMenuSubButton({

FILE: src/llama_stack_ui/components/ui/sign-in-button.tsx
  function SignInButton (line 8) | function SignInButton() {

FILE: src/llama_stack_ui/components/ui/skeleton.tsx
  function Skeleton (line 3) | function Skeleton({ className, ...props }: React.ComponentProps<"div">) {

FILE: src/llama_stack_ui/components/ui/table.tsx
  function Table (line 7) | function Table({ className, ...props }: React.ComponentProps<"table">) {
  function TableHeader (line 22) | function TableHeader({ className, ...props }: React.ComponentProps<"thea...
  function TableBody (line 32) | function TableBody({ className, ...props }: React.ComponentProps<"tbody"...
  function TableFooter (line 42) | function TableFooter({ className, ...props }: React.ComponentProps<"tfoo...
  function TableRow (line 55) | function TableRow({ className, ...props }: React.ComponentProps<"tr">) {
  function TableHead (line 68) | function TableHead({ className, ...props }: React.ComponentProps<"th">) {
  function TableCell (line 81) | function TableCell({ className, ...props }: React.ComponentProps<"td">) {
  function TableCaption (line 94) | function TableCaption({

FILE: src/llama_stack_ui/components/ui/textarea.tsx
  type TextareaProps (line 5) | type TextareaProps = React.TextareaHTMLAttributes<HTMLTextAreaElement>;

FILE: src/llama_stack_ui/components/ui/theme-provider.tsx
  function ThemeProvider (line 6) | function ThemeProvider({

FILE: src/llama_stack_ui/components/ui/tooltip.tsx
  function TooltipProvider (line 8) | function TooltipProvider({
  function Tooltip (line 21) | function Tooltip({
  function TooltipTrigger (line 31) | function TooltipTrigger({
  function TooltipContent (line 37) | function TooltipContent({

FILE: src/llama_stack_ui/components/vector-stores/vector-store-detail.tsx
  type VectorStoreDetailViewProps (line 30) | interface VectorStoreDetailViewProps {
  function VectorStoreDetailView (line 40) | function VectorStoreDetailView({

FILE: src/llama_stack_ui/components/vector-stores/vector-store-editor.tsx
  type VectorStoreFormData (line 18) | interface VectorStoreFormData {
  type VectorStoreEditorProps (line 25) | interface VectorStoreEditorProps {
  function VectorStoreEditor (line 34) | function VectorStoreEditor({

FILE: src/llama_stack_ui/hooks/use-audio-recording.ts
  type UseAudioRecordingOptions (line 5) | interface UseAudioRecordingOptions {
  function useAudioRecording (line 10) | function useAudioRecording({

FILE: src/llama_stack_ui/hooks/use-auth-client.ts
  function useAuthClient (line 5) | function useAuthClient() {

FILE: src/llama_stack_ui/hooks/use-auto-scroll.ts
  constant ACTIVATION_THRESHOLD (line 4) | const ACTIVATION_THRESHOLD = 50;
  constant MIN_SCROLL_UP_THRESHOLD (line 6) | const MIN_SCROLL_UP_THRESHOLD = 10;
  function useAutoScroll (line 8) | function useAutoScroll(dependencies: React.DependencyList) {

FILE: src/llama_stack_ui/hooks/use-autosize-textarea.ts
  type UseAutosizeTextAreaProps (line 3) | interface UseAutosizeTextAreaProps {
  function useAutosizeTextArea (line 10) | function useAutosizeTextArea({

FILE: src/llama_stack_ui/hooks/use-copy-to-clipboard.ts
  type UseCopyToClipboardProps (line 4) | type UseCopyToClipboardProps = {
  function useCopyToClipboard (line 9) | function useCopyToClipboard({

FILE: src/llama_stack_ui/hooks/use-infinite-scroll.ts
  type UseInfiniteScrollOptions (line 5) | interface UseInfiniteScrollOptions {
  function useInfiniteScroll (line 21) | function useInfiniteScroll(

FILE: src/llama_stack_ui/hooks/use-mobile.ts
  constant MOBILE_BREAKPOINT (line 3) | const MOBILE_BREAKPOINT = 768;
  function useIsMobile (line 5) | function useIsMobile() {

FILE: src/llama_stack_ui/hooks/use-pagination.ts
  type PaginationState (line 9) | interface PaginationState<T> {
  type PaginationResponse (line 17) | interface PaginationResponse<T> {
  type PaginationReturn (line 25) | interface PaginationReturn<T> {
  type UsePaginationParams (line 34) | interface UsePaginationParams<T> extends UsePaginationOptions {
  function usePagination (line 49) | function usePagination<T>({

FILE: src/llama_stack_ui/instrumentation.ts
  function register (line 9) | async function register() {

FILE: src/llama_stack_ui/jest.setup.ts
  method observe (line 9) | observe() {}
  method unobserve (line 10) | unobserve() {}
  method disconnect (line 11) | disconnect() {}
  method constructor (line 16) | constructor(callback: IntersectionObserverCallback) {}
  method observe (line 17) | observe() {}
  method unobserve (line 18) | unobserve() {}
  method disconnect (line 19) | disconnect() {}
  method takeRecords (line 20) | takeRecords() {

FILE: src/llama_stack_ui/lib/audio-utils.ts
  type RecordAudioType (line 1) | type RecordAudioType = {

FILE: src/llama_stack_ui/lib/auth.ts
  method jwt (line 18) | async jwt({ token, account }) {
  method session (line 25) | async session({ session, token }) {

FILE: src/llama_stack_ui/lib/config-validator.ts
  function validateServerConfig (line 5) | function validateServerConfig() {

FILE: src/llama_stack_ui/lib/contents-api.ts
  type VectorStoreContent (line 4) | type VectorStoreContent = FileContentResponse.Content;
  type VectorStoreContentsResponse (line 5) | type VectorStoreContentsResponse = FileContentResponse;
  type VectorStoreContentItem (line 7) | interface VectorStoreContentItem {
  type VectorStoreContentDeleteResponse (line 18) | interface VectorStoreContentDeleteResponse {
  type VectorStoreListContentsResponse (line 24) | interface VectorStoreListContentsResponse {
  class ContentsAPI (line 32) | class ContentsAPI {
    method constructor (line 33) | constructor(private client: LlamaStackClient) {}
    method getFileContents (line 35) | async getFileContents(
    method getContent (line 68) | async getContent(
    method updateContent (line 83) | async updateContent(): Promise<VectorStoreContentItem> {
    method deleteContent (line 87) | async deleteContent(): Promise<VectorStoreContentDeleteResponse> {
    method listContents (line 91) | async listContents(

FILE: src/llama_stack_ui/lib/file-utils.ts
  function formatFileSize (line 11) | function formatFileSize(bytes: number): string {
  function getFileTypeIcon (line 27) | function getFileTypeIcon(
  function validateFile (line 127) | function validateFile(file: File): {
  function getFileExtension (line 162) | function getFileExtension(filename: string): string {
  function formatPurpose (line 170) | function formatPurpose(
  function getPurposeDescription (line 200) | function getPurposeDescription(
  function formatTimestamp (line 230) | function formatTimestamp(timestamp: number): string {
  function isTextFile (line 237) | function isTextFile(mimeType: string): boolean {
  function createDownloadUrl (line 248) | function createDownloadUrl(content: string | Blob): string {
  function downloadFile (line 263) | function downloadFile(url: string, filename: string): void {
  function truncateFilename (line 276) | function truncateFilename(

FILE: src/llama_stack_ui/lib/file-validation.ts
  type FileValidationResult (line 8) | interface FileValidationResult {
  function validateFileForUpload (line 17) | function validateFileForUpload(file: File): FileValidationResult {
  function validateUploadParams (line 102) | function validateUploadParams(
  function formatFileSize (line 217) | function formatFileSize(bytes: number): string {
  function detectPotentialCorruption (line 233) | function detectPotentialCorruption(file: File): string[] {
  function formatValidationErrors (line 259) | function formatValidationErrors(errors: FileValidationError[]): string {
  function formatValidationWarnings (line 272) | function formatValidationWarnings(warnings: string[]): string {

FILE: src/llama_stack_ui/lib/format-message-content.ts
  function extractTextFromContentPart (line 4) | function extractTextFromContentPart(
  function extractDisplayableText (line 39) | function extractDisplayableText(

FILE: src/llama_stack_ui/lib/format-tool-call.tsx
  function formatToolCallToString (line 8) | function formatToolCallToString(toolCall: {

FILE: src/llama_stack_ui/lib/truncate-text.ts
  function truncateText (line 1) | function truncateText(

FILE: src/llama_stack_ui/lib/types.ts
  type TextContentBlock (line 1) | interface TextContentBlock {
  type ImageUrlDetail (line 6) | interface ImageUrlDetail {
  type ImageUrlContentBlock (line 11) | interface ImageUrlContentBlock {
  type ChatMessageContentPart (line 18) | type ChatMessageContentPart =
  type ChatMessage (line 23) | interface ChatMessage {
  type Choice (line 30) | interface Choice {
  type ChatCompletion (line 37) | interface ChatCompletion {
  type ListChatCompletionsResponse (line 46) | interface ListChatCompletionsResponse {
  type PaginationStatus (line 54) | type PaginationStatus = "idle" | "loading" | "loading-more" | "error";
  type PaginationState (line 56) | interface PaginationState {
  type UsePaginationOptions (line 64) | interface UsePaginationOptions {
  type ResponseInputMessageContent (line 74) | interface ResponseInputMessageContent {
  type ResponseMessage (line 81) | interface ResponseMessage {
  type ResponseToolCall (line 89) | interface ResponseToolCall {
  type ResponseOutput (line 98) | type ResponseOutput = ResponseMessage | ResponseToolCall;
  type ResponseInput (line 100) | interface ResponseInput {
  type OpenAIResponse (line 107) | interface OpenAIResponse {
  type InputItemListResponse (line 127) | interface InputItemListResponse {
  type FileResource (line 133) | interface FileResource {
  type ListFilesResponse (line 149) | interface ListFilesResponse {
  type DeleteFileResponse (line 157) | interface DeleteFileResponse {
  type FileCreateParams (line 163) | interface FileCreateParams {
  type FileUploadFormData (line 178) | interface FileUploadFormData {
  type FileValidationError (line 190) | interface FileValidationError {
  constant SUPPORTED_FILE_TYPES (line 196) | const SUPPORTED_FILE_TYPES = [
  constant FILE_TYPE_EXTENSIONS (line 207) | const FILE_TYPE_EXTENSIONS = {
  type SupportedFileType (line 219) | type SupportedFileType = (typeof SUPPORTED_FILE_TYPES)[number];
  constant MAX_FILE_SIZE (line 222) | const MAX_FILE_SIZE = 100 * 1024 * 1024;
  constant DEFAULT_EXPIRES_AFTER (line 223) | const DEFAULT_EXPIRES_AFTER = 24 * 60 * 60;

FILE: src/llama_stack_ui/lib/utils.tsx
  function cn (line 4) | function cn(...inputs: ClassValue[]) {

FILE: src/llama_stack_ui/scripts/postbuild.js
  function copyDir (line 22) | function copyDir(src, dest) {

FILE: src/llama_stack_ui/types/next-auth.d.ts
  type Session (line 4) | interface Session {

FILE: tests/backward_compat/test_eval_compat.py
  function sample_benchmark_config (line 37) | def sample_benchmark_config():
  class TestResolveRunEvalRequest (line 46) | class TestResolveRunEvalRequest:
    method test_new_style_with_request_object (line 49) | def test_new_style_with_request_object(self, sample_benchmark_config):
    method test_old_style_with_individual_params (line 64) | def test_old_style_with_individual_params(self, sample_benchmark_config):
    method test_request_object_takes_precedence_over_individual_params (line 82) | def test_request_object_takes_precedence_over_individual_params(self, ...
    method test_missing_parameters_raises_error (line 101) | def test_missing_parameters_raises_error(self, sample_benchmark_config):
  class TestResolveEvaluateRowsRequest (line 119) | class TestResolveEvaluateRowsRequest:
    method test_new_style_with_request_object (line 122) | def test_new_style_with_request_object(self, sample_benchmark_config):
    method test_old_style_with_individual_params (line 142) | def test_old_style_with_individual_params(self, sample_benchmark_config):
    method test_request_object_takes_precedence_over_individual_params (line 162) | def test_request_object_takes_precedence_over_individual_params(self, ...
    method test_missing_parameters_raises_error (line 187) | def test_missing_parameters_raises_error(self, sample_benchmark_config):
  class TestResolveJobStatusRequest (line 202) | class TestResolveJobStatusRequest:
    method test_new_style_with_request_object (line 205) | def test_new_style_with_request_object(self):
    method test_old_style_with_individual_params (line 219) | def test_old_style_with_individual_params(self):
    method test_request_object_takes_precedence_over_individual_params (line 233) | def test_request_object_takes_precedence_over_individual_params(self):
    method test_missing_parameters_raises_error (line 251) | def test_missing_parameters_raises_error(self):
  class TestResolveJobCancelRequest (line 268) | class TestResolveJobCancelRequest:
    method test_new_style_with_request_object (line 271) | def test_new_style_with_request_object(self):
    method test_old_style_with_individual_params (line 285) | def test_old_style_with_individual_params(self):
    method test_request_object_takes_precedence_over_individual_params (line 299) | def test_request_object_takes_precedence_over_individual_params(self):
    method test_missing_parameters_raises_error (line 317) | def test_missing_parameters_raises_error(self):
  class TestResolveJobResultRequest (line 334) | class TestResolveJobResultRequest:
    method test_new_style_with_request_object (line 337) | def test_new_style_with_request_object(self):
    method test_old_style_with_individual_params (line 351) | def test_old_style_with_individual_params(self):
    method test_request_object_takes_precedence_over_individual_params (line 365) | def test_request_object_takes_precedence_over_individual_params(self):
    method test_missing_parameters_raises_error (line 383) | def test_missing_parameters_raises_error(self):
  class TestEmptyValueValidation (line 400) | class TestEmptyValueValidation:
    method test_empty_benchmark_id_old_style (line 403) | def test_empty_benchmark_id_old_style(self, sample_benchmark_config):
    method test_empty_benchmark_id_in_request_object (line 410) | def test_empty_benchmark_id_in_request_object(self, sample_benchmark_c...
    method test_none_benchmark_id_in_request_object (line 421) | def test_none_benchmark_id_in_request_object(self, sample_benchmark_co...
    method test_empty_job_id_old_style (line 440) | def test_empty_job_id_old_style(self, resolver, request_class):
    method test_empty_job_id_in_request_object (line 455) | def test_empty_job_id_in_request_object(self, resolver, request_class):
    method test_none_job_id_in_request_object (line 474) | def test_none_job_id_in_request_object(self, resolver, request_class):
    method test_empty_input_rows_old_style (line 485) | def test_empty_input_rows_old_style(self, sample_benchmark_config):
    method test_empty_scoring_functions_old_style (line 497) | def test_empty_scoring_functions_old_style(self, sample_benchmark_conf...
    method test_empty_input_rows_in_request_object (line 509) | def test_empty_input_rows_in_request_object(self, sample_benchmark_con...
    method test_none_input_rows_in_request_object (line 522) | def test_none_input_rows_in_request_object(self, sample_benchmark_conf...

FILE: tests/backward_compat/test_run_config.py
  function get_test_configs (line 24) | def get_test_configs():
  function test_load_run_config (line 49) | def test_load_run_config(config_file):

FILE: tests/common/mcp.py
  function default_tools (line 17) | def default_tools():
  function dependency_tools (line 43) | def dependency_tools():
  function make_mcp_server (line 152) | def make_mcp_server(

FILE: tests/external/llama-stack-api-weather/src/llama_stack_api_weather/weather.py
  function available_providers (line 12) | def available_providers() -> list[ProviderSpec]:
  class WeatherProvider (line 25) | class WeatherProvider(Protocol):
    method get_available_locations (line 31) | async def get_available_locations() -> dict[str, list[str]]:

FILE: tests/external/llama-stack-provider-kaze/src/llama_stack_provider_kaze/__init__.py
  function get_adapter_impl (line 15) | async def get_adapter_impl(config: KazeProviderConfig, _deps):

FILE: tests/external/llama-stack-provider-kaze/src/llama_stack_provider_kaze/config.py
  class KazeProviderConfig (line 10) | class KazeProviderConfig(BaseModel):

FILE: tests/external/llama-stack-provider-kaze/src/llama_stack_provider_kaze/kaze.py
  class WeatherKazeAdapter (line 12) | class WeatherKazeAdapter(WeatherProvider):
    method __init__ (line 15) | def __init__(
    method initialize (line 21) | async def initialize(self) -> None:
    method get_available_locations (line 24) | async def get_available_locations(self) -> dict[str, list[str]]:

FILE: tests/integration/admin/test_admin.py
  class TestAdmin (line 12) | class TestAdmin:
    method test_admin_providers_list (line 13) | def test_admin_providers_list(self, llama_stack_client: LlamaStackAsLi...
    method test_health (line 23) | def test_health(self, llama_stack_client: LlamaStackAsLibraryClient | ...
    method test_version (line 28) | def test_version(self, llama_stack_client: LlamaStackAsLibraryClient |...
    method test_list_routes_default (line 33) | def test_list_routes_default(self, llama_stack_client: LlamaStackAsLib...
    method test_list_routes_filter_by_deprecated (line 49) | def test_list_routes_filter_by_deprecated(self, llama_stack_client: Ll...
    method test_list_routes_filter_by_v1 (line 58) | def test_list_routes_filter_by_v1(self, llama_stack_client: LlamaStack...

FILE: tests/integration/agents/test_openai_responses.py
  function test_responses_store (line 38) | def test_responses_store(compat_client, text_model_id, stream, tools):
  function test_list_response_input_items (line 90) | def test_list_response_input_items(compat_client, text_model_id):
  function test_list_response_input_items_with_limit_and_order (line 124) | def test_list_response_input_items_with_limit_and_order(openai_client, c...
  function test_function_call_output_response (line 217) | def test_function_call_output_response(openai_client, client_with_models...
  function test_function_call_output_response_with_none_arguments (line 269) | def test_function_call_output_response_with_none_arguments(openai_client...
  function test_input_safety_guardrails (line 303) | def test_input_safety_guardrails(compat_client, text_model_id, stream):
  function test_output_safety_guardrails_unsafe_content (line 349) | def test_output_safety_guardrails_unsafe_content(compat_client, text_mod...
  function test_output_safety_guardrails_safe_content (line 391) | def test_output_safety_guardrails_safe_content(compat_client, text_model...
  function test_guardrails_with_tools (line 432) | def test_guardrails_with_tools(compat_client, text_model_id):
  function test_response_with_instructions (line 471) | def test_response_with_instructions(openai_client, client_with_models, t...

FILE: tests/integration/batches/conftest.py
  class BatchHelper (line 20) | class BatchHelper:
    method __init__ (line 23) | def __init__(self, client):
    method create_file (line 28) | def create_file(self, content: str | list[dict], filename_prefix="batc...
    method wait_for (line 63) | def wait_for(
  function batch_helper (line 134) | def batch_helper(openai_client):

FILE: tests/integration/batches/test_batches.py
  class TestBatchesIntegration (line 35) | class TestBatchesIntegration:
    method test_batch_creation_and_retrieval (line 38) | def test_batch_creation_and_retrieval(self, openai_client, batch_helpe...
    method test_batch_listing (line 82) | def test_batch_listing(self, openai_client, batch_helper, text_model_id):
    method test_batch_immediate_cancellation (line 123) | def test_batch_immediate_cancellation(self, openai_client, batch_helpe...
    method test_batch_e2e_chat_completions (line 164) | def test_batch_e2e_chat_completions(self, openai_client, batch_helper,...
    method test_batch_e2e_completions (line 272) | def test_batch_e2e_completions(self, openai_client, batch_helper, text...
    method test_batch_e2e_embeddings (line 327) | def test_batch_e2e_embeddings(self, openai_client, batch_helper, embed...

FILE: tests/integration/batches/test_batches_errors.py
  class TestBatchesErrorHandling (line 52) | class TestBatchesErrorHandling:
    method test_batch_nonexistent_file_id (line 55) | def test_batch_nonexistent_file_id(self, openai_client, batch_helper):
    method test_batch_invalid_endpoint (line 86) | def test_batch_invalid_endpoint(self, openai_client, batch_helper, tex...
    method test_batch_malformed_jsonl (line 126) | def test_batch_malformed_jsonl(self, openai_client, batch_helper):
    method test_file_malformed_batch_file (line 169) | def test_file_malformed_batch_file(self, openai_client, batch_helper, ...
    method test_batch_retrieve_nonexistent (line 182) | def test_batch_retrieve_nonexistent(self, openai_client):
    method test_batch_cancel_nonexistent (line 191) | def test_batch_cancel_nonexistent(self, openai_client):
    method test_batch_cancel_completed (line 200) | def test_batch_cancel_completed(self, openai_client, batch_helper, tex...
    method test_batch_missing_required_fields (line 251) | def test_batch_missing_required_fields(self, openai_client, batch_help...
    method test_batch_invalid_completion_window (line 356) | def test_batch_invalid_completion_window(self, openai_client, batch_he...
    method test_batch_streaming_not_supported (line 384) | def test_batch_streaming_not_supported(self, openai_client, batch_help...
    method test_batch_mixed_streaming_requests (line 432) | def test_batch_mixed_streaming_requests(self, openai_client, batch_hel...
    method test_batch_endpoint_mismatch (line 497) | def test_batch_endpoint_mismatch(self, openai_client, batch_helper, te...
    method test_batch_error_handling_invalid_model (line 543) | def test_batch_error_handling_invalid_model(self, openai_client, batch...
    method test_batch_missing_required_body_fields (line 589) | def test_batch_missing_required_body_fields(self, openai_client, batch...
    method test_batch_invalid_metadata_types (line 656) | def test_batch_invalid_metadata_types(self, openai_client, batch_helpe...

FILE: tests/integration/batches/test_batches_idempotency.py
  class TestBatchesIdempotencyIntegration (line 30) | class TestBatchesIdempotencyIntegration:
    method test_idempotent_batch_creation_successful (line 33) | def test_idempotent_batch_creation_successful(self, openai_client):
    method test_idempotency_conflict_with_different_params (line 67) | def test_idempotency_conflict_with_different_params(self, openai_client):

FILE: tests/integration/client-typescript/setup.ts
  function loadTestConfig (line 18) | function loadTestConfig() {
  constant TEST_CONFIG (line 40) | const TEST_CONFIG = loadTestConfig();
  function createTestClient (line 65) | function createTestClient(testId?: string): LlamaStackClient {
  function skipIfNoModel (line 86) | function skipIfNoModel(modelType: 'text' | 'embedding'): typeof test {
  function requireTextModel (line 102) | function requireTextModel(): string {
  function requireEmbeddingModel (line 116) | function requireEmbeddingModel(): string {
  function getResponseOutputText (line 132) | function getResponseOutputText(response: any): string {

FILE: tests/integration/conftest.py
  function pytest_runtest_makereport (line 28) | def pytest_runtest_makereport(item, call):
  function pytest_sessionstart (line 36) | def pytest_sessionstart(session):
  function suppress_httpx_logs (line 64) | def suppress_httpx_logs(caplog):
  function _track_test_context (line 70) | def _track_test_context(request):
  function pytest_runtest_teardown (line 85) | def pytest_runtest_teardown(item):
  function pytest_configure (line 101) | def pytest_configure(config):
  function pytest_addoption (line 157) | def pytest_addoption(parser):
  function get_short_id (line 243) | def get_short_id(value):
  function pytest_generate_tests (line 247) | def pytest_generate_tests(metafunc):
  function pytest_ignore_collect (line 328) | def pytest_ignore_collect(path: str, config: pytest.Config) -> bool:
  function pytest_collection_modifyitems (line 363) | def pytest_collection_modifyitems(config: pytest.Config, items: list[pyt...
  function get_vector_io_provider_ids (line 414) | def get_vector_io_provider_ids(client):
  function vector_provider_wrapper (line 420) | def vector_provider_wrapper(func):
  function vector_io_provider_id (line 452) | def vector_io_provider_id(request, client_with_models):

FILE: tests/integration/conversations/test_openai_conversations.py
  function get_auth_token (line 13) | def get_auth_token(env_var: str, default: str) -> str:
  class TestOpenAIConversations (line 19) | class TestOpenAIConversations:
    method test_conversation_create (line 21) | def test_conversation_create(self, openai_client):
    method test_conversation_retrieve (line 31) | def test_conversation_retrieve(self, openai_client):
    method test_conversation_update (line 41) | def test_conversation_update(self, openai_client):
    method test_conversation_delete (line 50) | def test_conversation_delete(self, openai_client):
    method test_conversation_items_create (line 59) | def test_conversation_items_create(self, openai_client):
    method test_conversation_items_list (line 78) | def test_conversation_items_list(self, openai_client):
    method test_conversation_item_retrieve (line 96) | def test_conversation_item_retrieve(self, openai_client):
    method test_conversation_item_delete (line 112) | def test_conversation_item_delete(self, openai_client):
    method test_full_workflow (line 127) | def test_full_workflow(self, openai_client):
  function auth_enabled (line 146) | def auth_enabled() -> bool:
  class TestConversationAccessControl (line 154) | class TestConversationAccessControl:
    method alice_client (line 167) | def alice_client(self, openai_client):
    method bob_client (line 178) | def bob_client(self, openai_client):
    method test_user_cannot_retrieve_other_users_conversation (line 188) | def test_user_cannot_retrieve_other_users_conversation(self, alice_cli...
    method test_user_cannot_update_other_users_conversation (line 213) | def test_user_cannot_update_other_users_conversation(self, alice_clien...
    method test_user_cannot_delete_other_users_conversation (line 237) | def test_user_cannot_delete_other_users_conversation(self, alice_clien...
    method test_user_cannot_access_other_users_conversation_items (line 260) | def test_user_cannot_access_other_users_conversation_items(self, alice...
    method test_user_cannot_add_items_to_other_users_conversation (line 296) | def test_user_cannot_add_items_to_other_users_conversation(self, alice...
    method test_users_have_isolated_conversations (line 325) | def test_users_have_isolated_conversations(self, alice_client, bob_cli...
    method test_user_can_access_own_resources_after_denial (line 356) | def test_user_can_access_own_resources_after_denial(self, alice_client...

FILE: tests/integration/datasets/test_datasets.py
  function data_url_from_file (line 19) | def data_url_from_file(file_path: str) -> str:
  function test_register_and_iterrows (line 80) | def test_register_and_iterrows(llama_stack_client, purpose, source, prov...

FILE: tests/integration/eval/test_eval.py
  function test_evaluate_rows (line 19) | def test_evaluate_rows(llama_stack_client, text_model_id, scoring_fn_id):
  function test_evaluate_benchmark (line 69) | def test_evaluate_benchmark(llama_stack_client, text_model_id, scoring_f...

FILE: tests/integration/file_processors/test_pypdf_processor.py
  class TestPyPDFFileProcessor (line 111) | class TestPyPDFFileProcessor:
    method config (line 115) | def config(self) -> PyPDFFileProcessorConfig:
    method processor (line 122) | def processor(self, config: PyPDFFileProcessorConfig) -> PyPDFFileProc...
    method test_pdf_path (line 127) | def test_pdf_path(self) -> Path:
    method test_pdf_content (line 138) | def test_pdf_content(self, test_pdf_path: Path) -> bytes:
    method upload_file (line 144) | def upload_file(self, test_pdf_content: bytes) -> UploadFile:
    method test_process_file_basic (line 149) | async def test_process_file_basic(self, processor: PyPDFFileProcessor,...
    method test_process_file_with_auto_chunking (line 179) | async def test_process_file_with_auto_chunking(self, processor: PyPDFF...
    method test_process_file_with_static_chunking (line 211) | async def test_process_file_with_static_chunking(self, processor: PyPD...
    method test_metadata_extraction (line 239) | async def test_metadata_extraction(self, processor: PyPDFFileProcessor...
    method test_text_cleaning (line 256) | async def test_text_cleaning(self):
    method test_no_text_cleaning (line 268) | async def test_no_text_cleaning(self, upload_file: UploadFile):
    method test_no_metadata_extraction (line 281) | async def test_no_metadata_extraction(self, upload_file: UploadFile):
    method test_input_validation (line 299) | async def test_input_validation(self, processor: PyPDFFileProcessor):
    method test_file_id_without_files_api (line 310) | async def test_file_id_without_files_api(self, processor: PyPDFFilePro...
    method test_minimal_pdf_processing (line 315) | async def test_minimal_pdf_processing(self, processor: PyPDFFileProces...
    method test_options_parameter_ignored (line 326) | async def test_options_parameter_ignored(self, processor: PyPDFFilePro...
    method test_invalid_pdf_content (line 337) | async def test_invalid_pdf_content(self, processor: PyPDFFileProcessor):
    method test_empty_pdf_handling (line 348) | async def test_empty_pdf_handling(self, processor: PyPDFFileProcessor):
    method test_document_id_generation (line 359) | async def test_document_id_generation(self, processor: PyPDFFileProces...
    method test_chunk_id_uniqueness (line 370) | async def test_chunk_id_uniqueness(self, processor: PyPDFFileProcessor...
    method test_tokenizer_consistency (line 386) | async def test_tokenizer_consistency(self, processor: PyPDFFileProcess...
    method test_processing_time_metadata (line 396) | async def test_processing_time_metadata(self, processor: PyPDFFileProc...
  class TestPyPDFFileProcessorConfig (line 408) | class TestPyPDFFileProcessorConfig:
    method test_default_config_values (line 411) | def test_default_config_values(self):
    method test_config_validation (line 420) | def test_config_validation(self):

FILE: tests/integration/files/test_files.py
  function provider_type_is_openai (line 20) | def provider_type_is_openai(llama_stack_client):
  function skip_if_no_files_provider (line 28) | def skip_if_no_files_provider(llama_stack_client):
  function test_openai_client_basic_operations (line 33) | def test_openai_client_basic_operations(openai_client, provider_type_is_...
  function test_expires_after (line 96) | def test_expires_after(openai_client):
  function test_expires_after_requests (line 129) | def test_expires_after_requests(openai_client):
  function test_files_authentication_isolation (line 179) | def test_files_authentication_isolation(mock_get_authenticated_user, lla...
  function test_files_authentication_shared_attributes (line 279) | def test_files_authentication_shared_attributes(
  function test_files_authentication_anonymous_access (line 339) | def test_files_authentication_anonymous_access(

FILE: tests/integration/fixtures/common.py
  function is_port_available (line 37) | def is_port_available(port: int, host: str = "localhost") -> bool:
  function start_llama_stack_server (line 47) | def start_llama_stack_server(config_name: str, *, log_stderr: bool | Non...
  function wait_for_server_ready (line 70) | def wait_for_server_ready(base_url: str, timeout: int = 30, process: sub...
  function stop_server_on_port (line 102) | def stop_server_on_port(port: int, timeout: float = 10.0) -> None:
  function get_provider_data (line 131) | def get_provider_data():
  function inference_provider_type (line 153) | def inference_provider_type(llama_stack_client):
  function client_with_models (line 161) | def client_with_models(
  function available_shields (line 192) | def available_shields(llama_stack_client):
  function model_providers (line 197) | def model_providers(llama_stack_client):
  function skip_if_no_model (line 202) | def skip_if_no_model(request):
  function llama_stack_client (line 221) | def llama_stack_client(request):
  function parse_vector_io_provider (line 238) | def parse_vector_io_provider(config_string: str) -> str:
  function extract_model (line 250) | def extract_model(model: str | None, default: str) -> str:
  function instantiate_llama_stack_client (line 256) | def instantiate_llama_stack_client(session):
  function require_server (line 375) | def require_server(llama_stack_client):
  function openai_client (line 388) | def openai_client(llama_stack_client, require_server):
  function compat_client (line 400) | def compat_client(request, client_with_models):
  function cleanup_server_process (line 414) | def cleanup_server_process(request):

FILE: tests/integration/inference/test_openai_completion.py
  function _normalize_text (line 17) | def _normalize_text(text: str) -> str:
  function provider_from_model (line 29) | def provider_from_model(client_with_models, model_id):
  function skip_if_model_doesnt_support_openai_completion (line 39) | def skip_if_model_doesnt_support_openai_completion(client_with_models, m...
  function skip_if_doesnt_support_completions_logprobs (line 70) | def skip_if_doesnt_support_completions_logprobs(client_with_models, mode...
  function skip_if_model_doesnt_support_suffix (line 79) | def skip_if_model_doesnt_support_suffix(client_with_models, model_id):
  function skip_if_doesnt_support_n (line 96) | def skip_if_doesnt_support_n(client_with_models, model_id):
  function skip_if_model_doesnt_support_openai_chat_completion (line 122) | def skip_if_model_doesnt_support_openai_chat_completion(client_with_mode...
  function skip_if_provider_isnt_vllm (line 134) | def skip_if_provider_isnt_vllm(client_with_models, model_id):
  function skip_if_provider_isnt_openai (line 140) | def skip_if_provider_isnt_openai(client_with_models, model_id):
  function skip_if_provider_doesnt_support_tool_calling (line 148) | def skip_if_provider_doesnt_support_tool_calling(client_with_models, mod...
  function test_openai_completion_non_streaming (line 163) | def test_openai_completion_non_streaming(llama_stack_client, client_with...
  function test_openai_completion_non_streaming_suffix (line 185) | def test_openai_completion_non_streaming_suffix(llama_stack_client, clie...
  function test_openai_completion_streaming (line 212) | def test_openai_completion_streaming(llama_stack_client, client_with_mod...
  function test_openai_completion_guided_choice (line 229) | def test_openai_completion_guided_choice(llama_stack_client, client_with...
  function test_openai_chat_completion_non_streaming (line 254) | def test_openai_chat_completion_non_streaming(compat_client, client_with...
  function test_openai_chat_completion_streaming (line 284) | def test_openai_chat_completion_streaming(compat_client, client_with_mod...
  function test_openai_chat_completion_streaming_with_n (line 314) | def test_openai_chat_completion_streaming_with_n(compat_client, client_w...
  function test_inference_store (line 352) | def test_inference_store(compat_client, client_with_models, text_model_i...
  function test_inference_store_tool_calls (line 408) | def test_inference_store_tool_calls(compat_client, client_with_models, t...
  function test_openai_chat_completion_non_streaming_with_file (line 490) | def test_openai_chat_completion_non_streaming_with_file(openai_client, c...
  function skip_if_model_doesnt_support_reasoning (line 523) | def skip_if_model_doesnt_support_reasoning(model_id):
  function test_openai_chat_completion_reasoning_passthrough (line 529) | def test_openai_chat_completion_reasoning_passthrough(openai_client, cli...
  function skip_if_doesnt_support_completions_stop_sequence (line 571) | def skip_if_doesnt_support_completions_stop_sequence(client_with_models,...
  function test_openai_completion_stop_sequence (line 583) | def test_openai_completion_stop_sequence(client_with_models, openai_clie...
  function test_openai_completion_logprobs (line 616) | def test_openai_completion_logprobs(client_with_models, openai_client, t...
  function test_openai_completion_logprobs_streaming (line 646) | def test_openai_completion_logprobs_streaming(client_with_models, openai...
  function test_openai_chat_completion_with_tools (line 681) | def test_openai_chat_completion_with_tools(openai_client, text_model_id,...
  function test_openai_chat_completion_with_tools_and_streaming (line 705) | def test_openai_chat_completion_with_tools_and_streaming(openai_client, ...
  function test_openai_chat_completion_with_tool_choice_none (line 739) | def test_openai_chat_completion_with_tool_choice_none(openai_client, tex...
  function test_openai_chat_completion_structured_output (line 760) | def test_openai_chat_completion_structured_output(openai_client, text_mo...

FILE: tests/integration/inference/test_openai_embeddings.py
  function decode_base64_to_floats (line 25) | def decode_base64_to_floats(base64_string: str) -> list[float]:
  function provider_from_model (line 33) | def provider_from_model(client_with_models, model_id):
  function is_asymmetric_model (line 43) | def is_asymmetric_model(client_with_models, model_id):
  function get_extra_body_for_model (line 53) | def get_extra_body_for_model(client_with_models, model_id, input_type="q...
  function skip_if_model_doesnt_support_user_param (line 65) | def skip_if_model_doesnt_support_user_param(client, model_id):
  function skip_if_model_doesnt_support_encoding_format_base64 (line 74) | def skip_if_model_doesnt_support_encoding_format_base64(client, model_id):
  function skip_if_model_doesnt_support_variable_dimensions (line 95) | def skip_if_model_doesnt_support_variable_dimensions(client_with_models,...
  function compat_client (line 126) | def compat_client(request, client_with_models):
  function skip_if_model_doesnt_support_openai_embeddings (line 132) | def skip_if_model_doesnt_support_openai_embeddings(client, model_id):
  function openai_client (line 146) | def openai_client(client_with_models):
  function test_openai_embeddings_single_string (line 151) | def test_openai_embeddings_single_string(compat_client, client_with_mode...
  function test_openai_embeddings_multiple_strings (line 175) | def test_openai_embeddings_multiple_strings(compat_client, client_with_m...
  function test_openai_embeddings_with_encoding_format_float (line 201) | def test_openai_embeddings_with_encoding_format_float(compat_client, cli...
  function test_openai_embeddings_with_dimensions (line 220) | def test_openai_embeddings_with_dimensions(compat_client, client_with_mo...
  function test_openai_embeddings_with_user_parameter (line 242) | def test_openai_embeddings_with_user_parameter(compat_client, client_wit...
  function test_openai_embeddings_empty_list_error (line 263) | def test_openai_embeddings_empty_list_error(compat_client, client_with_m...
  function test_openai_embeddings_invalid_model_error (line 275) | def test_openai_embeddings_invalid_model_error(compat_client, client_wit...
  function test_openai_embeddings_different_inputs_different_outputs (line 287) | def test_openai_embeddings_different_inputs_different_outputs(compat_cli...
  function test_openai_embeddings_with_encoding_format_base64 (line 317) | def test_openai_embeddings_with_encoding_format_base64(compat_client, cl...
  function test_openai_embeddings_base64_batch_processing (line 352) | def test_openai_embeddings_base64_batch_processing(compat_client, client...

FILE: tests/integration/inference/test_openai_vision_inference.py
  function image_path (line 15) | def image_path():
  function base64_image_data (line 20) | def base64_image_data(image_path):
  function test_openai_chat_completion_image_url (line 24) | async def test_openai_chat_completion_image_url(openai_client, vision_mo...
  function test_openai_chat_completion_image_data (line 52) | async def test_openai_chat_completion_image_data(openai_client, vision_m...

FILE: tests/integration/inference/test_provider_data_routing.py
  function test_unregistered_model_routing_with_provider_data (line 29) | def test_unregistered_model_routing_with_provider_data(client_with_models):

FILE: tests/integration/inference/test_provider_error_recording.py
  class TestProviderErrorRecording (line 29) | class TestProviderErrorRecording:
    method test_provider_rejects_invalid_base64_image (line 37) | def test_provider_rejects_invalid_base64_image(self, openai_client, te...

FILE: tests/integration/inference/test_rerank.py
  function skip_if_provider_doesnt_support_rerank (line 32) | def skip_if_provider_doesnt_support_rerank(inference_provider_type):
  function _validate_rerank_response (line 38) | def _validate_rerank_response(response: InferenceRerankResponse, items: ...
  function _validate_semantic_ranking (line 60) | def _validate_semantic_ranking(response: InferenceRerankResponse, items:...
  function test_rerank_text (line 97) | def test_rerank_text(client_with_models, rerank_model_id, query, items, ...
  function test_rerank_image (line 124) | def test_rerank_image(client_with_models, rerank_model_id, query, items,...
  function test_rerank_max_results (line 141) | def test_rerank_max_results(client_with_models, rerank_model_id, inferen...
  function test_rerank_max_results_larger_than_items (line 159) | def test_rerank_max_results_larger_than_items(client_with_models, rerank...
  function test_rerank_semantic_correctness (line 206) | def test_rerank_semantic_correctness(

FILE: tests/integration/inference/test_tools_with_schemas.py
  class TestChatCompletionWithTools (line 20) | class TestChatCompletionWithTools:
    method test_simple_tool_call (line 23) | def test_simple_tool_call(self, llama_stack_client, text_model_id):
    method test_tool_with_complex_schema (line 48) | def test_tool_with_complex_schema(self, llama_stack_client, text_model...
  class TestOpenAICompatibility (line 93) | class TestOpenAICompatibility:
    method test_openai_chat_completion_with_tools (line 96) | def test_openai_chat_completion_with_tools(self, compat_client, text_m...
    method test_openai_format_preserves_complex_schemas (line 125) | def test_openai_format_preserves_complex_schemas(self, compat_client, ...
  class TestMCPToolsInChatCompletion (line 159) | class TestMCPToolsInChatCompletion:
    method mcp_with_schemas (line 163) | def mcp_with_schemas(self):
    method test_mcp_tools_in_inference (line 174) | def test_mcp_tools_in_inference(self, llama_stack_client, text_model_i...
  class TestStreamingWithTools (line 225) | class TestStreamingWithTools:
    method test_streaming_tool_calls (line 228) | def test_streaming_tool_calls(self, llama_stack_client, text_model_id):
  class TestEdgeCases (line 257) | class TestEdgeCases:
    method test_tool_without_schema (line 260) | def test_tool_without_schema(self, llama_stack_client, text_model_id):
    method test_multiple_tools_with_different_schemas (line 281) | def test_multiple_tools_with_different_schemas(self, llama_stack_clien...

FILE: tests/integration/inference/test_vision_inference.py
  function image_path (line 18) | def image_path():
  function base64_image_data (line 23) | def base64_image_data(image_path):
  function base64_image_url (line 29) | def base64_image_url(base64_image_data):
  function test_image_chat_completion_non_streaming (line 33) | def test_image_chat_completion_non_streaming(client_with_models, vision_...
  function multi_image_data (line 60) | def multi_image_data():
  function multi_image_url (line 75) | def multi_image_url(multi_image_data):
  function test_image_chat_completion_multiple_images (line 80) | def test_image_chat_completion_multiple_images(client_with_models, visio...
  function test_image_chat_completion_streaming (line 160) | def test_image_chat_completion_streaming(client_with_models, vision_mode...
  function test_image_chat_completion_base64 (line 188) | def test_image_chat_completion_base64(client_with_models, vision_model_i...

FILE: tests/integration/inspect/test_inspect.py
  class TestInspect (line 13) | class TestInspect:
    method test_health (line 15) | def test_health(self, llama_stack_client: LlamaStackAsLibraryClient | ...
    method test_version (line 21) | def test_version(self, llama_stack_client: LlamaStackAsLibraryClient |...
    method test_list_routes_default (line 27) | def test_list_routes_default(self, llama_stack_client: LlamaStackAsLib...
    method test_list_routes_filter_by_deprecated (line 46) | def test_list_routes_filter_by_deprecated(self, llama_stack_client: Ll...
    method test_list_routes_filter_by_v1 (line 61) | def test_list_routes_filter_by_v1(self, llama_stack_client: LlamaStack...

FILE: tests/integration/providers/nvidia/test_datastore.py
  function skip_if_no_nvidia_provider (line 18) | def skip_if_no_nvidia_provider(llama_stack_client):
  function test_register_and_unregister (line 32) | def test_register_and_unregister(llama_stack_client, provider_id):

FILE: tests/integration/providers/test_providers.py
  class TestProviders (line 12) | class TestProviders:
    method test_providers (line 13) | def test_providers(self, llama_stack_client: LlamaStackAsLibraryClient...

FILE: tests/integration/providers/utils/sqlstore/test_authorized_sqlstore.py
  function get_postgres_config (line 26) | def get_postgres_config():
  function get_sqlite_config (line 37) | def get_sqlite_config():
  function authorized_store (line 59) | def authorized_store(backend_config):
  function create_test_table (line 78) | async def create_test_table(authorized_store, table_name):
  function cleanup_records (line 89) | async def cleanup_records(sql_store, table_name, record_ids):
  function test_authorized_store_attributes (line 100) | async def test_authorized_store_attributes(mock_get_authenticated_user, ...
  function test_user_ownership_policy (line 204) | async def test_user_ownership_policy(mock_get_authenticated_user, author...
  function test_sqlrecord_created_with_no_owner (line 264) | async def test_sqlrecord_created_with_no_owner(mock_get_authenticated_us...

FILE: tests/integration/responses/conftest.py
  function pytest_configure (line 20) | def pytest_configure(config):
  function responses_client (line 34) | def responses_client(compat_client):

FILE: tests/integration/responses/fixtures/fixtures.py
  function _load_all_verification_configs (line 19) | def _load_all_verification_configs():
  function verification_config (line 54) | def verification_config():
  function provider (line 63) | def provider(request, verification_config):
  function base_url (line 82) | def base_url(request, provider, verification_config):
  function api_key (line 89) | def api_key(request, provider, verification_config):
  function model_mapping (line 101) | def model_mapping(provider, providers_model_mapping):
  function openai_client (line 106) | def openai_client(base_url, api_key, provider):

FILE: tests/integration/responses/fixtures/test_cases.py
  class ResponsesTestCase (line 13) | class ResponsesTestCase(BaseModel):

FILE: tests/integration/responses/helpers.py
  function new_vector_store (line 10) | def new_vector_store(openai_client, name, embedding_model, embedding_dim...
  function upload_file (line 27) | def upload_file(openai_client, name, file_path):
  function wait_for_file_attachment (line 39) | def wait_for_file_attachment(compat_client, vector_store_id, file_id):
  function setup_mcp_tools (line 58) | def setup_mcp_tools(tools, mcp_server_info):

FILE: tests/integration/responses/streaming_assertions.py
  class StreamingValidator (line 12) | class StreamingValidator:
    method __init__ (line 15) | def __init__(self, chunks: list[Any]):
    method assert_basic_event_sequence (line 19) | def assert_basic_event_sequence(self):
    method assert_response_consistency (line 35) | def assert_response_consistency(self):
    method assert_has_incremental_content (line 46) | def assert_has_incremental_content(self):
    method assert_content_quality (line 68) | def assert_content_quality(self, expected_content: str):
    method assert_has_tool_calls (line 76) | def assert_has_tool_calls(self):
    method assert_has_mcp_events (line 104) | def assert_has_mcp_events(self):
    method assert_rich_streaming (line 132) | def assert_rich_streaming(self, min_chunks: int = 10):
    method validate_event_structure (line 138) | def validate_event_structure(self):

FILE: tests/integration/responses/test_basic_responses.py
  function provider_from_model (line 15) | def provider_from_model(client_with_models, text_model_id):
  function skip_if_provider_isnt_vllm (line 25) | def skip_if_provider_isnt_vllm(client_with_models, text_model_id):
  function skip_if_chat_completions_logprobs_not_supported (line 33) | def skip_if_chat_completions_logprobs_not_supported(client_with_models, ...
  function test_response_non_streaming_basic (line 40) | def test_response_non_streaming_basic(responses_client, text_model_id, c...
  function test_response_streaming_basic (line 71) | def test_response_streaming_basic(responses_client, text_model_id, case):
  function test_response_streaming_incremental_content (line 131) | def test_response_streaming_incremental_content(responses_client, text_m...
  function test_response_non_streaming_multi_turn (line 197) | def test_response_non_streaming_multi_turn(responses_client, text_model_...
  function test_response_non_streaming_image (line 211) | def test_response_non_streaming_image(responses_client, text_model_id, c...
  function test_response_non_streaming_multi_turn_image (line 224) | def test_response_non_streaming_multi_turn_image(responses_client, text_...
  function test_include_logprobs_non_streaming (line 239) | def test_include_logprobs_non_streaming(client_with_models, text_model_id):
  function test_include_logprobs_streaming (line 277) | def test_include_logprobs_streaming(client_with_models, text_model_id):
  function test_include_logprobs_with_web_search (line 314) | def test_include_logprobs_with_web_search(client_with_models, text_model...
  function test_include_logprobs_with_function_tools (line 347) | def test_include_logprobs_with_function_tools(client_with_models, text_m...
  function test_response_extra_body_guided_choice (line 389) | def test_response_extra_body_guided_choice(client_with_models, text_mode...

FILE: tests/integration/responses/test_conversation_responses.py
  class TestConversationResponses (line 11) | class TestConversationResponses:
    method test_conversation_basic_workflow (line 14) | def test_conversation_basic_workflow(self, openai_client, text_model_id):
    method test_conversation_multi_turn_and_streaming (line 35) | def test_conversation_multi_turn_and_streaming(self, openai_client, te...
    method test_conversation_context_loading (line 68) | def test_conversation_context_loading(self, openai_client, text_model_...
    method test_conversation_error_handling (line 86) | def test_conversation_error_handling(self, openai_client, text_model_id):
    method test_conversation_backward_compatibility (line 119) | def test_conversation_backward_compatibility(self, openai_client, text...

FILE: tests/integration/responses/test_file_search.py
  function test_response_text_format (line 29) | def test_response_text_format(responses_client, text_model_id, text_form...
  function vector_store_with_filtered_files (line 47) | def vector_store_with_filtered_files(
  function test_response_file_search_filter_by_region (line 140) | def test_response_file_search_filter_by_region(responses_client, text_mo...
  function test_response_file_search_filter_by_category (line 173) | def test_response_file_search_filter_by_category(responses_client, text_...
  function test_response_file_search_filter_by_date_range (line 205) | def test_response_file_search_filter_by_date_range(responses_client, tex...
  function test_response_file_search_filter_compound_and (line 250) | def test_response_file_search_filter_compound_and(responses_client, text...
  function test_response_file_search_filter_compound_or (line 288) | def test_response_file_search_filter_compound_or(responses_client, text_...
  function test_response_file_search_streaming_events (line 333) | def test_response_file_search_streaming_events(responses_client, text_mo...

FILE: tests/integration/responses/test_mcp_authentication.py
  function test_mcp_authorization_bearer (line 20) | def test_mcp_authorization_bearer(responses_client, text_model_id):
  function test_mcp_authorization_error_when_header_provided (line 56) | def test_mcp_authorization_error_when_header_provided(responses_client, ...
  function test_mcp_authorization_backward_compatibility (line 84) | def test_mcp_authorization_backward_compatibility(responses_client, text...

FILE: tests/integration/responses/test_openai_responses.py
  class TestOpenAIResponses (line 15) | class TestOpenAIResponses:
    method _invalid_base64_image_input (line 18) | def _invalid_base64_image_input(self):
    method test_openai_response_with_max_output_tokens (line 32) | def test_openai_response_with_max_output_tokens(self, openai_client, t...
    method test_openai_response_with_small_max_output_tokens (line 46) | def test_openai_response_with_small_max_output_tokens(self, openai_cli...
    method test_openai_response_max_output_tokens_below_minimum (line 73) | def test_openai_response_max_output_tokens_below_minimum(self, openai_...
    method test_openai_response_streaming_failed_error_code_is_spec_compliant (line 86) | def test_openai_response_streaming_failed_error_code_is_spec_compliant...
    method test_openai_response_streaming_invalid_base64_image_failure_code_is_spec_compliant (line 104) | def test_openai_response_streaming_invalid_base64_image_failure_code_i...
    method test_openai_response_with_prompt_cache_key (line 140) | def test_openai_response_with_prompt_cache_key(self, openai_client, te...
    method test_openai_response_with_prompt_cache_key_streaming (line 153) | def test_openai_response_with_prompt_cache_key_streaming(self, openai_...
    method test_openai_response_with_prompt_cache_key_and_previous_response (line 178) | def test_openai_response_with_prompt_cache_key_and_previous_response(s...
    method test_openai_response_with_safety_identifier (line 204) | def test_openai_response_with_safety_identifier(self, openai_client, t...
    method test_openai_response_with_safety_identifier_streaming (line 217) | def test_openai_response_with_safety_identifier_streaming(self, openai...
    method test_openai_response_with_safety_identifier_and_previous_response (line 242) | def test_openai_response_with_safety_identifier_and_previous_response(...
    method test_openai_response_with_truncation_disabled (line 268) | def test_openai_response_with_truncation_disabled(self, openai_client,...
    method test_openai_response_with_truncation_disabled_streaming (line 280) | def test_openai_response_with_truncation_disabled_streaming(self, open...
    method test_openai_response_with_truncation_and_previous_response (line 304) | def test_openai_response_with_truncation_and_previous_response(self, o...
    method test_openai_response_with_truncation_auto_error (line 328) | def test_openai_response_with_truncation_auto_error(self, openai_clien...
    method test_openai_response_with_top_p (line 340) | def test_openai_response_with_top_p(self, openai_client, text_model_id):
    method test_openai_response_with_top_p_streaming (line 352) | def test_openai_response_with_top_p_streaming(self, openai_client, tex...
    method test_openai_response_with_top_p_and_previous_response (line 376) | def test_openai_response_with_top_p_and_previous_response(self, openai...
    method test_openai_response_with_top_logprobs (line 400) | def test_openai_response_with_top_logprobs(self, openai_client, text_m...
    method test_openai_response_with_top_logprobs_streaming (line 412) | def test_openai_response_with_top_logprobs_streaming(self, openai_clie...
    method test_openai_response_with_top_logprobs_and_previous_response (line 436) | def test_openai_response_with_top_logprobs_and_previous_response(self,...
    method _function_tools (line 460) | def _function_tools(self):
    method test_openai_response_with_parallel_tool_calls_enabled (line 493) | def test_openai_response_with_parallel_tool_calls_enabled(self, openai...
    method test_openai_response_with_parallel_tool_calls_disabled (line 515) | def test_openai_response_with_parallel_tool_calls_disabled(self, opena...
    method test_openai_response_with_parallel_tool_calls_disabled_streaming (line 533) | def test_openai_response_with_parallel_tool_calls_disabled_streaming(s...
    method test_openai_response_with_parallel_tool_calls_and_previous_response (line 560) | def test_openai_response_with_parallel_tool_calls_and_previous_respons...
    method test_openai_response_background_returns_queued (line 583) | def test_openai_response_background_returns_queued(self, openai_client...
    method test_openai_response_background_completes (line 598) | def test_openai_response_background_completes(self, openai_client, tex...
    method test_openai_response_background_and_stream_mutually_exclusive (line 636) | def test_openai_response_background_and_stream_mutually_exclusive(self...
    method test_openai_response_background_false_is_synchronous (line 649) | def test_openai_response_background_false_is_synchronous(self, openai_...
    method _skip_service_tier_for_unsupported (line 661) | def _skip_service_tier_for_unsupported(self, text_model_id):
    method test_openai_response_with_service_tier_auto (line 667) | def test_openai_response_with_service_tier_auto(self, openai_client, t...
    method test_openai_response_with_service_tier (line 686) | def test_openai_response_with_service_tier(self, openai_client, text_m...
    method test_openai_response_with_service_tier_flex (line 700) | def test_openai_response_with_service_tier_flex(self, openai_client, t...
    method test_openai_response_with_service_tier_auto_streaming (line 721) | def test_openai_response_with_service_tier_auto_streaming(self, openai...
    method test_openai_response_with_service_tier_streaming (line 748) | def test_openai_response_with_service_tier_streaming(self, openai_clie...
    method test_openai_response_with_service_tier_flex_streaming (line 774) | def test_openai_response_with_service_tier_flex_streaming(self, openai...
    method test_openai_response_with_service_tier_auto_and_previous_response (line 802) | def test_openai_response_with_service_tier_auto_and_previous_response(...
    method test_openai_response_with_service_tier_and_previous_response (line 827) | def test_openai_response_with_service_tier_and_previous_response(self,...
    method test_openai_response_streaming_includes_usage (line 851) | def test_openai_response_streaming_includes_usage(self, openai_client,...
    method test_openai_response_with_stream_options_includes_usage (line 877) | def test_openai_response_with_stream_options_includes_usage(self, open...
    method test_openai_response_with_stream_options_non_streaming (line 900) | def test_openai_response_with_stream_options_non_streaming(self, opena...
    method test_openai_response_with_stream_options_and_previous_response (line 915) | def test_openai_response_with_stream_options_and_previous_response(sel...
    method test_openai_response_incomplete_details_null_when_completed (line 946) | def test_openai_response_incomplete_details_null_when_completed(self, ...
    method test_openai_response_incomplete_details_length (line 957) | def test_openai_response_incomplete_details_length(self, openai_client...
    method test_openai_response_incomplete_details_length_streaming (line 979) | def test_openai_response_incomplete_details_length_streaming(self, ope...
    method test_openai_response_incomplete_details_max_iterations_exceeded (line 1003) | def test_openai_response_incomplete_details_max_iterations_exceeded(se...
    method test_openai_response_incomplete_details_max_iterations_exceeded_streaming (line 1024) | def test_openai_response_incomplete_details_max_iterations_exceeded_st...
    method _is_reasoning_model (line 1046) | def _is_reasoning_model(model_id: str) -> bool:
    method _skip_reasoning_effort_for_unsupported (line 1054) | def _skip_reasoning_effort_for_unsupported(self, text_model_id):
    method test_openai_response_reasoning_effort (line 1059) | def test_openai_response_reasoning_effort(self, openai_client, text_mo...
    method test_openai_response_reasoning_effort_streaming (line 1075) | def test_openai_response_reasoning_effort_streaming(self, openai_clien...

FILE: tests/integration/responses/test_prompt_templates.py
  function responses_client_with_prompts (line 11) | def responses_client_with_prompts(responses_client):
  function text_var (line 20) | def text_var(value: str) -> dict:
  function test_basic_prompt_template (line 25) | def test_basic_prompt_template(responses_client_with_prompts, text_model...
  function test_multi_variable_prompt_template (line 47) | def test_multi_variable_prompt_template(responses_client_with_prompts, t...
  function test_prompt_template_no_variables (line 76) | def test_prompt_template_no_variables(responses_client_with_prompts, tex...
  function test_multi_version_prompt_template (line 92) | def test_multi_version_prompt_template(responses_client_with_prompts, te...
  function test_prompt_template_with_streaming (line 143) | def test_prompt_template_with_streaming(responses_client_with_prompts, t...
  function test_prompt_template_with_multi_turn (line 174) | def test_prompt_template_with_multi_turn(responses_client_with_prompts, ...
  function test_nonexistent_prompt_id (line 202) | def test_nonexistent_prompt_id(responses_client_with_prompts, text_model...
  function test_invalid_prompt_version (line 217) | def test_invalid_prompt_version(responses_client_with_prompts, text_mode...

FILE: tests/integration/responses/test_reasoning.py
  function provider_from_model (line 12) | def provider_from_model(client_with_models, text_model_id):
  function skip_if_reasoning_content_not_provided (line 22) | def skip_if_reasoning_content_not_provided(client_with_models, text_mode...
  function test_reasoning_basic_streaming (line 28) | def test_reasoning_basic_streaming(client_with_models, text_model_id):

FILE: tests/integration/responses/test_responses_access_control.py
  function get_auth_token (line 22) | def get_auth_token(env_var: str, default: str) -> str:
  function auth_enabled (line 27) | def auth_enabled() -> bool:
  class TestResponsesAccessControl (line 35) | class TestResponsesAccessControl:
    method alice_client (line 48) | def alice_client(self, openai_client, request):
    method bob_client (line 64) | def bob_client(self, openai_client, request):
    method _create_stored_response (line 78) | def _create_stored_response(self, client, text_model_id: str, input_te...
    method test_user_cannot_retrieve_other_users_response (line 87) | def test_user_cannot_retrieve_other_users_response(self, alice_client,...
    method test_user_cannot_delete_other_users_response (line 115) | def test_user_cannot_delete_other_users_response(self, alice_client, b...
    method test_users_have_isolated_responses (line 142) | def test_users_have_isolated_responses(self, alice_client, bob_client,...
    method test_user_can_access_own_resources_after_denial (line 179) | def test_user_can_access_own_resources_after_denial(self, alice_client...
    method test_user_cannot_access_other_users_response_input_items (line 211) | def test_user_cannot_access_other_users_response_input_items(
    method test_previous_response_id_access_control (line 239) | def test_previous_response_id_access_control(self, alice_client, bob_c...

FILE: tests/integration/responses/test_responses_errors.py
  class TestResponsesAPIErrors (line 65) | class TestResponsesAPIErrors:
    method test_invalid_model_raises_not_found_error (line 73) | def test_invalid_model_raises_not_found_error(self, openai_client):
    method test_invalid_previous_response_id_raises_not_found_error (line 92) | def test_invalid_previous_response_id_raises_not_found_error(self, ope...
    method test_invalid_max_tool_calls_raises_bad_request (line 111) | def test_invalid_max_tool_calls_raises_bad_request(self, openai_client...
    method test_invalid_temperature_raises_bad_request (line 127) | def test_invalid_temperature_raises_bad_request(self, openai_client, t...
    method test_invalid_tool_choice_raises_bad_request (line 144) | def test_invalid_tool_choice_raises_bad_request(self, openai_client, t...
    method test_retrieve_nonexistent_response_raises_not_found_error (line 162) | def test_retrieve_nonexistent_response_raises_not_found_error(self, op...
    method test_delete_nonexistent_response_raises_not_found_error (line 171) | def test_delete_nonexistent_response_raises_not_found_error(self, open...
    method test_conflicting_previous_response_and_conversation_raises_bad_request (line 180) | def test_conflicting_previous_response_and_conversation_raises_bad_req...
    method test_malformed_request_returns_sdk_compatible_error (line 194) | def test_malformed_request_returns_sdk_compatible_error(self, openai_c...
    method test_guardrails_without_safety_api_raises_service_unavailable (line 205) | def test_guardrails_without_safety_api_raises_service_unavailable(
  class TestConversationsAPIErrors (line 223) | class TestConversationsAPIErrors:
    method test_invalid_conversation_id_format_raises_bad_request (line 230) | def test_invalid_conversation_id_format_raises_bad_request(self, opena...
    method test_nonexistent_conversation_raises_not_found_error (line 248) | def test_nonexistent_conversation_raises_not_found_error(self, openai_...
  class TestResponsesAPIStreamingErrors (line 266) | class TestResponsesAPIStreamingErrors:
    method _consume_stream (line 290) | def _consume_stream(self, stream):
    method _get_terminal_event (line 294) | def _get_terminal_event(self, chunks):
    method _assert_valid_error_code (line 301) | def _assert_valid_error_code(self, error_code: str):
    method _assert_failed_event_structure (line 307) | def _assert_failed_event_structure(self, failed_event):
    method test_completed_response_has_no_error (line 317) | def test_completed_response_has_no_error(self, openai_client, text_mod...
    method test_incomplete_response_has_no_error (line 351) | def test_incomplete_response_has_no_error(self, openai_client, text_mo...
    method test_invalid_image_url_returns_image_error (line 383) | def test_invalid_image_url_returns_image_error(self, openai_client, vi...
    method test_invalid_base64_image_returns_image_error (line 435) | def test_invalid_base64_image_returns_image_error(self, openai_client,...
    method test_non_vision_model_returns_error_for_image_input (line 491) | def test_non_vision_model_returns_error_for_image_input(self, openai_c...
    method test_non_vision_model_with_base64_image_returns_server_error (line 556) | def test_non_vision_model_with_base64_image_returns_server_error(self,...

FILE: tests/integration/responses/test_structured_output.py
  function test_json_schema_string_types (line 12) | def test_json_schema_string_types(responses_client, text_model_id):
  function test_json_schema_integer_types (line 45) | def test_json_schema_integer_types(responses_client, text_model_id):
  function test_json_schema_boolean_types (line 76) | def test_json_schema_boolean_types(responses_client, text_model_id):
  function test_json_schema_float_types (line 109) | def test_json_schema_float_types(responses_client, text_model_id):
  function test_json_schema_array_of_strings (line 140) | def test_json_schema_array_of_strings(responses_client, text_model_id):
  function test_json_schema_array_of_integers (line 176) | def test_json_schema_array_of_integers(responses_client, text_model_id):
  function test_json_schema_array_of_objects (line 210) | def test_json_schema_array_of_objects(responses_client, text_model_id):
  function test_json_schema_nested_objects (line 259) | def test_json_schema_nested_objects(responses_client, text_model_id):
  function test_json_schema_mixed_types_structures (line 315) | def test_json_schema_mixed_types_structures(responses_client, text_model...

FILE: tests/integration/responses/test_tool_responses.py
  function _skip_tool_tests_for_watsonx (line 33) | def _skip_tool_tests_for_watsonx(request):
  function test_response_non_streaming_web_search (line 43) | def test_response_non_streaming_web_search(responses_client, text_model_...
  function test_response_non_streaming_file_search (line 61) | def test_response_non_streaming_file_search(
  function test_response_non_streaming_file_search_empty_vector_store (line 115) | def test_response_non_streaming_file_search_empty_vector_store(
  function test_response_sequential_file_search (line 140) | def test_response_sequential_file_search(
  function test_response_non_streaming_mcp_tool (line 204) | def test_response_non_streaming_mcp_tool(responses_client, text_model_id...
  function test_response_sequential_mcp_tool (line 274) | def test_response_sequential_mcp_tool(responses_client, text_model_id, c...
  function test_response_connector_resolution_mcp_tool (line 324) | def test_response_connector_resolution_mcp_tool(responses_client, text_m...
  function test_response_mcp_tool_approval (line 375) | def test_response_mcp_tool_approval(responses_client, text_model_id, cas...
  function test_response_non_streaming_custom_tool (line 447) | def test_response_non_streaming_custom_tool(responses_client, text_model...
  function test_response_function_call_ordering_1 (line 461) | def test_response_function_call_ordering_1(responses_client, text_model_...
  function test_response_function_call_ordering_2 (line 492) | def test_response_function_call_ordering_2(responses_client, text_model_...
  function test_function_call_output_list_text (line 548) | def test_function_call_output_list_text(responses_client, text_model_id):
  function test_function_call_output_list_text_multi_block (line 597) | def test_function_call_output_list_text_multi_block(responses_client, te...
  function test_function_call_output_list_image (line 649) | def test_function_call_output_list_image(responses_client, vision_model_...
  function test_function_call_output_list_file (line 712) | def test_function_call_output_list_file(responses_client, text_model_id,...
  function test_response_non_streaming_multi_turn_tool_execution (line 767) | def test_response_non_streaming_multi_turn_tool_execution(responses_clie...
  function test_response_streaming_multi_turn_tool_execution (line 814) | def test_response_streaming_multi_turn_tool_execution(responses_client, ...
  function test_max_tool_calls_with_function_tools (line 886) | def test_max_tool_calls_with_function_tools(responses_client, text_model...
  function test_max_tool_calls_invalid (line 942) | def test_max_tool_calls_invalid(responses_client, text_model_id):
  function test_max_tool_calls_with_mcp_tools (line 969) | def test_max_tool_calls_with_mcp_tools(responses_client, text_model_id):
  function test_parallel_tool_calls_with_function_tools (line 1039) | def test_parallel_tool_calls_with_function_tools(responses_client, text_...
  function test_parallel_tool_calls_with_mcp_tools (line 1113) | def test_parallel_tool_calls_with_mcp_tools(responses_client, text_model...
  function test_response_streaming_web_search (line 1166) | def test_response_streaming_web_search(responses_client, text_model_id, ...
  function test_response_multi_turn_streaming_web_search (line 1200) | def test_response_multi_turn_streaming_web_search(responses_client, text...

FILE: tests/integration/safety/test_llama_guard.py
  function data_url_from_image (line 23) | def data_url_from_image(file_path):
  function text_model (line 37) | def text_model(request, client_with_models):
  function text_shield_id (line 53) | def text_shield_id(client_with_models, safety_provider, text_model) -> G...
  function vision_model (line 71) | def vision_model(request, client_with_models):
  function vision_shield_id (line 87) | def vision_shield_id(client_with_models, safety_provider, vision_model) ...
  function safety_provider (line 110) | def safety_provider(client_with_models) -> str:
  function test_unsafe_text_examples (line 139) | def test_unsafe_text_examples(client_with_models, text_shield_id, example):
  function test_safe_text_examples (line 164) | def test_safe_text_examples(client_with_models, text_shield_id, example):
  function test_llama_guard_for_code_interpreter_abuse (line 193) | def test_llama_guard_for_code_interpreter_abuse(client_with_models, text...
  function test_llama_guard_with_conversation_history (line 214) | def test_llama_guard_with_conversation_history(client_with_models, text_...
  function safe_image (line 242) | def safe_image():
  function unsafe_image (line 257) | def unsafe_image():
  function test_vision_safety_with_safe_image (line 271) | def test_vision_safety_with_safe_image(client_with_models, vision_shield...
  function test_vision_safety_with_unsafe_image (line 293) | def test_vision_safety_with_unsafe_image(client_with_models, vision_shie...

FILE: tests/integration/safety/test_passthrough.py
  class _ModerationHandler (line 39) | class _ModerationHandler(BaseHTTPRequestHandler):
    method do_POST (line 46) | def do_POST(self):  # noqa: N802 - required by BaseHTTPRequestHandler
    method log_message (line 74) | def log_message(self, format, *args):
  function mock_server (line 79) | def mock_server():
  function _make_adapter (line 89) | async def _make_adapter(base_url: str, **kwargs) -> PassthroughSafetyAda...
  function test_run_shield_safe_content (line 112) | async def test_run_shield_safe_content(mock_server):
  function test_run_moderation_safe_content (line 123) | async def test_run_moderation_safe_content(mock_server):
  function test_run_shield_flagged_content (line 132) | async def test_run_shield_flagged_content(mock_server):
  function test_api_key_sent_as_bearer (line 152) | async def test_api_key_sent_as_bearer(mock_server):
  function test_forward_headers_sent_downstream (line 161) | async def test_forward_headers_sent_downstream(mock_server):
  function test_sensitive_headers_rejected_at_config_time (line 178) | async def test_sensitive_headers_rejected_at_config_time(mock_server):
  function test_multiple_messages_all_checked (line 189) | async def test_multiple_messages_all_checked(mock_server):
  function test_connection_error_wrapped (line 215) | async def test_connection_error_wrapped(mock_server):

FILE: tests/integration/safety/test_safety.py
  function data_url_from_image (line 16) | def data_url_from_image(file_path):
  function code_scanner_shield_id (line 29) | def code_scanner_shield_id(available_shields):
  function test_unsafe_examples (line 36) | def test_unsafe_examples(client_with_models, shield_id):
  function test_safe_examples_with_run_moderation (line 58) | def test_safe_examples_with_run_moderation(client_with_models, shield_id):
  function test_unsafe_examples_with_run_moderation (line 81) | def test_unsafe_examples_with_run_moderation(client_with_models, shield_...
  function test_safe_examples (line 101) | def test_safe_examples(client_with_models, shield_id):
  function test_safety_with_code_scanner (line 126) | def test_safety_with_code_scanner(client_with_models, code_scanner_shiel...
  function test_unsafe_input_with_code_scanner_run_moderation (line 161) | def test_unsafe_input_with_code_scanner_run_moderation(client_with_model...
  function test_safe_input_with_code_scanner_run_moderation (line 188) | def test_safe_input_with_code_scanner_run_moderation(client_with_models,...
  function test_safety_with_code_interpreter_abuse (line 217) | def test_safety_with_code_interpreter_abuse(client_with_models, shield_id):
  function test_safety_prompt_injection (line 248) | def test_safety_prompt_injection():

FILE: tests/integration/safety/test_vision_safety.py
  function data_url_from_image (line 18) | def data_url_from_image(file_path):
  function test_safety_with_image (line 30) | def test_safety_with_image(llama_stack_client, model_providers):

FILE: tests/integration/scoring/test_scoring.py
  function sample_judge_prompt_template (line 16) | def sample_judge_prompt_template():
  function sample_scoring_fn_id (line 21) | def sample_scoring_fn_id():
  function register_scoring_function (line 25) | def register_scoring_function(
  function unregister_scoring_function (line 47) | def unregister_scoring_function(llama_stack_client, scoring_fn_id):
  function test_scoring_functions_list (line 57) | def test_scoring_functions_list(llama_stack_client):
  function test_scoring_functions_register (line 63) | def test_scoring_functions_register(
  function test_scoring_functions_unregister (line 92) | def test_scoring_functions_unregister(
  function test_scoring_score (line 138) | def test_scoring_score(llama_stack_client, scoring_fn_id):
  function test_scoring_score_with_params_llm_as_judge (line 157) | def test_scoring_score_with_params_llm_as_judge(
  function test_scoring_score_with_aggregation_functions (line 196) | def test_scoring_score_with_aggregation_functions(

FILE: tests/integration/suites.py
  class Suite (line 31) | class Suite(BaseModel):
  class Setup (line 37) | class Setup(BaseModel):

FILE: tests/integration/telemetry/collectors/base.py
  class MetricStub (line 17) | class MetricStub:
  class SpanStub (line 26) | class SpanStub:
    method context (line 37) | def context(self):
    method get_trace_id (line 43) | def get_trace_id(self) -> str | None:
    method has_message (line 53) | def has_message(self, text: str) -> bool:
    method is_root_span (line 62) | def is_root_span(self) -> bool:
    method is_autotraced (line 68) | def is_autotraced(self) -> bool:
    method get_span_type (line 74) | def get_span_type(self) -> str | None:
    method get_class_method (line 80) | def get_class_method(self) -> tuple[str | None, str | None]:
    method get_location (line 86) | def get_location(self) -> str | None:
  function _value_to_python (line 93) | def _value_to_python(value: Any) -> Any:
  function attributes_to_dict (line 112) | def attributes_to_dict(key_values: Iterable[Any]) -> dict[str, Any]:
  function events_to_list (line 116) | def events_to_list(events: Iterable[Any]) -> list[dict[str, Any]]:
  class BaseTelemetryCollector (line 127) | class BaseTelemetryCollector:
    method __init__ (line 137) | def __init__(self):
    method _get_baseline_stabilization_delay (line 141) | def _get_baseline_stabilization_delay(cls) -> float:
    method _get_metric_key (line 160) | def _get_metric_key(self, metric: MetricStub) -> tuple[str, str]:
    method _compute_metric_delta (line 166) | def _compute_metric_delta(self, metric: MetricStub) -> int | float | N...
    method get_spans (line 181) | def get_spans(
    method get_metrics (line 214) | def get_metrics(
    method _accumulate_metric (line 260) | def _accumulate_metric(
    method _has_enough_metrics (line 293) | def _has_enough_metrics(
    method _convert_attributes_to_dict (line 308) | def _convert_attributes_to_dict(attrs: Any) -> dict[str, Any]:
    method _extract_trace_span_ids (line 325) | def _extract_trace_span_ids(span: Any) -> tuple[str | None, str | None]:
    method _create_span_stub_from_opentelemetry (line 344) | def _create_span_stub_from_opentelemetry(span: Any) -> SpanStub:
    method _create_span_stub_from_protobuf (line 360) | def _create_span_stub_from_protobuf(span: Any, resource_attrs: dict[st...
    method _extract_metric_from_opentelemetry (line 380) | def _extract_metric_from_opentelemetry(metric: Any) -> MetricStub | None:
    method _create_metric_stubs_from_protobuf (line 417) | def _create_metric_stubs_from_protobuf(metric: Any) -> list[MetricStub]:
    method _extract_data_point_value (line 465) | def _extract_data_point_value(data_point: Any, metric_type: str) -> fl...
    method clear (line 482) | def clear(self) -> None:
    method _snapshot_spans (line 496) | def _snapshot_spans(self) -> tuple[SpanStub, ...]:  # pragma: no cover...
    method _snapshot_metrics (line 499) | def _snapshot_metrics(self) -> tuple[MetricStub, ...] | None:  # pragm...
    method _clear_impl (line 502) | def _clear_impl(self) -> None:  # pragma: no cover - interface hook
    method shutdown (line 505) | def shutdown(self) -> None:

FILE: tests/integration/telemetry/collectors/in_memory.py
  class InMemoryTelemetryCollector (line 22) | class InMemoryTelemetryCollector(BaseTelemetryCollector):
    method __init__ (line 29) | def __init__(self, span_exporter: InMemorySpanExporter, metric_reader:...
    method _snapshot_spans (line 34) | def _snapshot_spans(self) -> tuple[SpanStub, ...]:
    method _snapshot_metrics (line 40) | def _snapshot_metrics(self) -> tuple[MetricStub, ...] | None:
    method _clear_impl (line 56) | def _clear_impl(self) -> None:
  class InMemoryTelemetryManager (line 61) | class InMemoryTelemetryManager:
    method __init__ (line 62) | def __init__(self) -> None:
    method shutdown (line 81) | def shutdown(self) -> None:

FILE: tests/integration/telemetry/collectors/otlp.py
  class OtlpHttpTestCollector (line 22) | class OtlpHttpTestCollector(BaseTelemetryCollector):
    method __init__ (line 23) | def __init__(self) -> None:
    method _handle_traces (line 43) | def _handle_traces(self, request: ExportTraceServiceRequest) -> None:
    method _handle_metrics (line 59) | def _handle_metrics(self, request: ExportMetricsServiceRequest) -> None:
    method _snapshot_spans (line 74) | def _snapshot_spans(self) -> tuple[SpanStub, ...]:
    method _snapshot_metrics (line 78) | def _snapshot_metrics(self) -> tuple[MetricStub, ...] | None:
    method _clear_impl (line 82) | def _clear_impl(self) -> None:
    method shutdown (line 116) | def shutdown(self) -> None:
  class _CollectorHandler (line 122) | class _CollectorHandler(BaseHTTPRequestHandler):
    method do_POST (line 123) | def do_POST(self) -> None:  # noqa: N802 Function name `do_POST` shoul...
    method _respond_ok (line 144) | def _respond_ok(self) -> None:

FILE: tests/integration/telemetry/conftest.py
  function telemetry_test_collector (line 20) | def telemetry_test_collector():
  function llama_stack_client (line 54) | def llama_stack_client(telemetry_test_collector, request):
  function mock_otlp_collector (line 62) | def mock_otlp_collector(telemetry_test_collector):

FILE: tests/integration/telemetry/test_completions.py
  function test_streaming_chunk_count (line 18) | def test_streaming_chunk_count(mock_otlp_collector, llama_stack_client, ...
  function test_telemetry_format_completeness (line 54) | def test_telemetry_format_completeness(mock_otlp_collector, llama_stack_...

FILE: tests/integration/telemetry/test_tool_runtime_metrics.py
  function test_tool_metrics_with_websearch_tool (line 16) | async def test_tool_metrics_with_websearch_tool(mock_otlp_collector, lla...
  function test_tool_error_metrics_basic (line 55) | async def test_tool_error_metrics_basic(mock_otlp_collector, llama_stack...
  function test_metrics_accumulate (line 86) | async def test_metrics_accumulate(mock_otlp_collector, llama_stack_client):

FILE: tests/integration/test_cases/test_case.py
  class TestCase (line 11) | class TestCase:
    method __init__ (line 18) | def __init__(self, name):
    method __getitem__ (line 38) | def __getitem__(self, key):

FILE: tests/integration/test_persistence_integration.py
  function test_starter_distribution_config_loads_and_resolves (line 18) | def test_starter_distribution_config_loads_and_resolves():
  function test_postgres_demo_distribution_config_loads (line 48) | def test_postgres_demo_distribution_config_loads():

FILE: tests/integration/tool_runtime/test_builtin_tools.py
  function sample_search_query (line 14) | def sample_search_query():
  function sample_wolfram_alpha_query (line 19) | def sample_wolfram_alpha_query():
  function test_web_search_tool (line 23) | def test_web_search_tool(llama_stack_client, sample_search_query):
  function test_wolfram_alpha_tool (line 49) | def test_wolfram_alpha_tool(llama_stack_client, sample_wolfram_alpha_que...

FILE: tests/integration/tool_runtime/test_mcp.py
  function mcp_server (line 17) | def mcp_server():
  function test_mcp_invocation (line 22) | def test_mcp_invocation(llama_stack_client, text_model_id, mcp_server):

FILE: tests/integration/tool_runtime/test_mcp_json_schema.py
  function mcp_server_with_complex_schemas (line 20) | def mcp_server_with_complex_schemas():
  function mcp_server_with_output_schemas (line 73) | def mcp_server_with_output_schemas():
  class TestMCPSchemaPreservation (line 99) | class TestMCPSchemaPreservation:
    method test_mcp_tools_list_with_schemas (line 102) | def test_mcp_tools_list_with_schemas(self, llama_stack_client, mcp_ser...
    method test_mcp_schema_with_refs_preserved (line 143) | def test_mcp_schema_with_refs_preserved(self, llama_stack_client, mcp_...
    method test_mcp_output_schema_preserved (line 190) | def test_mcp_output_schema_preserved(self, llama_stack_client, mcp_ser...
  class TestMCPToolInvocation (line 230) | class TestMCPToolInvocation:
    method test_invoke_mcp_tool_with_nested_data (line 233) | def test_invoke_mcp_tool_with_nested_data(self, llama_stack_client, mc...
    method test_invoke_with_flexible_schema (line 274) | def test_invoke_with_flexible_schema(self, llama_stack_client, mcp_ser...
  class TestAgentWithMCPTools (line 318) | class TestAgentWithMCPTools:
    method test_agent_with_complex_mcp_tool (line 322) | def test_agent_with_complex_mcp_tool(self, llama_stack_client, text_mo...

FILE: tests/integration/tool_runtime/test_registration.py
  function test_register_and_unregister_toolgroup (line 16) | def test_register_and_unregister_toolgroup(llama_stack_client):

FILE: tests/integration/tools/test_tools.py
  function test_toolsgroups_unregister (line 8) | def test_toolsgroups_unregister(llama_stack_client):

FILE: tests/integration/vector_io/test_openai_vector_stores.py
  function skip_if_provider_doesnt_support_openai_vector_stores (line 24) | def skip_if_provider_doesnt_support_openai_vector_stores(client_with_mod...
  function skip_if_provider_doesnt_support_native_filtering (line 54) | def skip_if_provider_doesnt_support_native_filtering(vector_io_provider_...
  function skip_if_provider_doesnt_support_openai_vector_stores_search (line 62) | def skip_if_provider_doesnt_support_openai_vector_stores_search(
  function sample_chunks (line 145) | def sample_chunks():
  function compat_client_with_empty_stores (line 3298) | def compat_client_with_empty_stores(compat_client):
  function test_openai_create_vector_store (line 3331) | def test_openai_create_vector_store(
  function test_openai_create_vector_store_default (line 3359) | def test_openai_create_vector_store_default(compat_client_with_empty_sto...
  function test_openai_list_vector_stores (line 3368) | def test_openai_list_vector_stores(
  function test_openai_retrieve_vector_store (line 3412) | def test_openai_retrieve_vector_store(
  function test_openai_update_vector_store (line 3441) | def test_openai_update_vector_store(
  function test_openai_delete_vector_store (line 3474) | def test_openai_delete_vector_store(
  function test_openai_vector_store_search_empty (line 3506) | def test_openai_vector_store_search_empty(
  function test_openai_vector_store_with_chunks (line 3537) | def test_openai_vector_store_with_chunks(
  function test_openai_vector_store_search_relevance (line 3605) | def test_openai_vector_store_search_relevance(
  function test_openai_vector_store_search_with_ranking_options (line 3658) | def test_openai_vector_store_search_with_ranking_options(
  function test_openai_vector_store_search_with_high_score_filter (line 3718) | def test_openai_vector_store_search_with_high_score_filter(
  function test_openai_vector_store_search_with_weighted_ranker (line 3784) | def test_openai_vector_store_search_with_weighted_ranker(
  function test_openai_vector_store_search_with_rrf_ranker (line 3882) | def test_openai_vector_store_search_with_rrf_ranker(
  function test_openai_vector_store_search_with_ranker_defaults (line 3935) | def test_openai_vector_store_search_with_ranker_defaults(
  function test_openai_vector_store_search_neural_ranker_validation (line 3997) | def test_openai_vector_store_search_neural_ranker_validation(
  function test_openai_vector_store_search_with_ranking_options_combined (line 4062) | def test_openai_vector_store_search_with_ranking_options_combined(
  function test_openai_vector_store_search_with_max_num_results (line 4129) | def test_openai_vector_store_search_with_max_num_results(
  function test_openai_vector_store_attach_file (line 4171) | def test_openai_vector_store_attach_file(
  function test_openai_vector_store_search_with_typed_filters (line 4233) | def test_openai_vector_store_search_with_typed_filters(
  function test_openai_vector_store_attach_files_on_creation (line 4353) | def test_openai_vector_store_attach_files_on_creation(
  function test_openai_vector_store_list_files (line 4418) | def test_openai_vector_store_list_files(
  function test_openai_vector_store_list_files_invalid_vector_store (line 4493) | def test_openai_vector_store_list_files_invalid_vector_store(
  function test_openai_vector_store_retrieve_file_contents (line 4510) | def test_openai_vector_store_retrieve_file_contents(
  function test_openai_vector_store_delete_file (line 4571) | def test_openai_vector_store_delete_file(
  function test_openai_vector_store_delete_file_removes_from_vector_store (line 4637) | def test_openai_vector_store_delete_file_removes_from_vector_store(
  function test_openai_vector_store_update_file (line 4689) | def test_openai_vector_store_update_file(
  function test_create_vector_store_files_duplicate_vector_store_name (line 4746) | def test_create_vector_store_files_duplicate_vector_store_name(
  function test_openai_vector_store_search_modes (line 4818) | def test_openai_vector_store_search_modes(
  function test_openai_vector_store_file_batch_create_and_retrieve (line 4857) | def test_openai_vector_store_file_batch_create_and_retrieve(
  function test_openai_vector_store_file_batch_list_files (line 4922) | def test_openai_vector_store_file_batch_list_files(
  function test_openai_vector_store_file_batch_cancel (line 5017) | def test_openai_vector_store_file_batch_cancel(
  function test_openai_vector_store_file_batch_retrieve_contents (line 5074) | def test_openai_vector_store_file_batch_retrieve_contents(
  function test_openai_vector_store_file_batch_error_handling (line 5147) | def test_openai_vector_store_file_batch_error_handling(
  function test_openai_vector_store_embedding_config_from_metadata (line 5206) | def test_openai_vector_store_embedding_config_from_metadata(
  function test_openai_vector_store_file_contents_with_extra_query (line 5260) | def test_openai_vector_store_file_contents_with_extra_query(
  function test_openai_vector_store_search_with_rewrite_query (line 5354) | def test_openai_vector_store_search_with_rewrite_query(
  function test_openai_vector_store_contextual_chunking (line 5401) | def test_openai_vector_store_contextual_chunking(
  function test_openai_vector_store_contextual_chunking_error_without_model (line 5501) | def test_openai_vector_store_contextual_chunking_error_without_model(
  function test_openai_vector_store_contextual_vs_static_chunks (line 5549) | def test_openai_vector_store_contextual_vs_static_chunks(

FILE: tests/integration/vector_io/test_vector_io.py
  function sample_chunks (line 17) | def sample_chunks(embedding_dimension):
  function client_with_empty_registry (line 69) | def client_with_empty_registry(client_with_models):
  function test_vector_store_retrieve (line 84) | def test_vector_store_retrieve(
  function test_vector_store_register (line 107) | def test_vector_store_register(
  function test_insert_chunks (line 146) | def test_insert_chunks(
  function test_insert_chunks_with_precomputed_embeddings (line 187) | def test_insert_chunks_with_precomputed_embeddings(
  function test_query_returns_valid_object_when_identical_to_embedding_in_vdb (line 248) | def test_query_returns_valid_object_when_identical_to_embedding_in_vdb(
  function test_auto_extract_embedding_dimension (line 310) | def test_auto_extract_embedding_dimension(
  function test_provider_auto_selection_single_provider (line 322) | def test_provider_auto_selection_single_provider(
  function test_provider_id_override (line 335) | def test_provider_id_override(

FILE: tests/integration/vector_io/test_vector_io_filters.py
  function skip_if_provider_doesnt_support_filters (line 31) | def skip_if_provider_doesnt_support_filters(vector_io_provider_id: str):
  function filter_test_chunks (line 43) | def filter_test_chunks(embedding_dimension):
  function client_with_empty_registry (line 137) | def client_with_empty_registry(client_with_models):
  function test_filter_eq_string (line 156) | def test_filter_eq_string(
  function test_filter_ne_string (line 198) | def test_filter_ne_string(
  function test_filter_gt_numeric (line 240) | def test_filter_gt_numeric(
  function test_filter_gte_numeric (line 282) | def test_filter_gte_numeric(
  function test_filter_lt_numeric (line 324) | def test_filter_lt_numeric(
  function test_filter_lte_numeric (line 366) | def test_filter_lte_numeric(
  function test_filter_in_list (line 408) | def test_filter_in_list(
  function test_filter_nin_list (line 450) | def test_filter_nin_list(
  function test_filter_eq_boolean (line 492) | def test_filter_eq_boolean(
  function test_filter_and_compound (line 539) | def test_filter_and_compound(
  function test_filter_or_compound (line 588) | def test_filter_or_compound(
  function test_filter_nested_compound (line 638) | def test_filter_nested_compound(
  function test_filter_no_matches (line 699) | def test_filter_no_matches(
  function test_filter_null_returns_all (line 737) | def test_filter_null_returns_all(
  function test_filter_multiple_and_conditions (line 773) | def test_filter_multiple_and_conditions(
  function test_openai_search_with_comparison_filter (line 829) | def test_openai_search_with_comparison_filter(
  function test_openai_search_with_compound_filter (line 870) | def test_openai_search_with_compound_filter(

FILE: tests/unit/cli/test_stack_config.py
  function config_with_distro_name_int (line 19) | def config_with_distro_name_int():
  function up_to_date_config (line 74) | def up_to_date_config():
  function old_config (line 126) | def old_config():
  function invalid_config (line 163) | def invalid_config():
  function test_parse_and_maybe_upgrade_config_up_to_date (line 172) | def test_parse_and_maybe_upgrade_config_up_to_date(up_to_date_config):
  function test_parse_and_maybe_upgrade_config_old_format (line 178) | def test_parse_and_maybe_upgrade_config_old_format(old_config):
  function test_parse_and_maybe_upgrade_config_invalid (line 198) | def test_parse_and_maybe_upgrade_config_invalid(invalid_config):
  function test_parse_and_maybe_upgrade_config_distro_name_int (line 203) | def test_parse_and_maybe_upgrade_config_distro_name_int(config_with_dist...
  function test_parse_and_maybe_upgrade_config_sets_external_providers_dir (line 208) | def test_parse_and_maybe_upgrade_config_sets_external_providers_dir(up_t...
  function test_parse_and_maybe_upgrade_config_preserves_custom_external_providers_dir (line 220) | def test_parse_and_maybe_upgrade_config_preserves_custom_external_provid...
  function test_generate_run_config_from_providers (line 231) | def test_generate_run_config_from_providers():
  function test_providers_flag_generates_config_with_api_keys (line 257) | def test_providers_flag_generates_config_with_api_keys():
  function config_with_image_name (line 305) | def config_with_image_name():
  function test_parse_config_with_deprecated_image_name (line 346) | def test_parse_config_with_deprecated_image_name(config_with_image_name):
  function test_parse_config_with_both_names_prefers_distro_name (line 368) | def test_parse_config_with_both_names_prefers_distro_name():

FILE: tests/unit/cli/test_stack_list_deps.py
  function stack_list_deps (line 26) | def stack_list_deps() -> StackListDeps:
  class TestArguments (line 31) | class TestArguments:
    method test_providers_flag_registered (line 32) | def test_providers_flag_registered(self, stack_list_deps: StackListDeps):
    method test_providers_default_is_none (line 36) | def test_providers_default_is_none(self, stack_list_deps: StackListDeps):
    method test_providers_accepts_multiple_pairs (line 40) | def test_providers_accepts_multiple_pairs(self, stack_list_deps: Stack...
    method test_config_and_providers_are_independent (line 44) | def test_config_and_providers_are_independent(self, stack_list_deps: S...
  class TestDelegation (line 51) | class TestDelegation:
    method test_providers_calls_dynamic_config_spec (line 52) | def test_providers_calls_dynamic_config_spec(self, stack_list_deps: St...
    method test_providers_passes_semicolon_spec_unchanged (line 72) | def test_providers_passes_semicolon_spec_unchanged(self, stack_list_de...
  class TestErrorPropagation (line 95) | class TestErrorPropagation:
    method test_value_error_causes_exit_1 (line 96) | def test_value_error_causes_exit_1(self, stack_list_deps: StackListDeps):
    method test_value_error_message_printed_to_stderr (line 109) | def test_value_error_message_printed_to_stderr(self, stack_list_deps: ...
  class TestOutput (line 124) | class TestOutput:
    method test_normal_deps_printed (line 125) | def test_normal_deps_printed(self, stack_list_deps: StackListDeps, cap...
    method test_server_dependencies_always_included (line 146) | def test_server_dependencies_always_included(self, stack_list_deps: St...

FILE: tests/unit/cli/test_stack_run.py
  function stack_run (line 26) | def stack_run() -> StackRun:
  class TestArguments (line 31) | class TestArguments:
    method test_providers_flag_registered (line 32) | def test_providers_flag_registered(self, stack_run: StackRun):
    method test_providers_default_is_none (line 36) | def test_providers_default_is_none(self, stack_run: StackRun):
    method test_providers_accepts_multiple_pairs (line 40) | def test_providers_accepts_multiple_pairs(self, stack_run: StackRun):
  class TestDelegation (line 45) | class TestDelegation:
    method test_providers_calls_dynamic_config_spec (line 46) | def test_providers_calls_dynamic_config_spec(self, stack_run: StackRun...
    method test_providers_writes_config_yaml (line 68) | def test_providers_writes_config_yaml(self, stack_run: StackRun, tmp_p...
  class TestErrorPropagation (line 88) | class TestErrorPropagation:
    method test_value_error_causes_exit (line 89) | def test_value_error_causes_exit(self, stack_run: StackRun, tmp_path: ...

FILE: tests/unit/conftest.py
  function pytest_sessionstart (line 14) | def pytest_sessionstart(session) -> None:
  function suppress_httpx_logs (line 24) | def suppress_httpx_logs(caplog):

FILE: tests/unit/conversations/test_api_models.py
  function test_conversation_model_defaults (line 11) | def test_conversation_model_defaults():
  function test_openai_client_compatibility (line 23) | def test_openai_client_compatibility():
  function test_conversation_item_list (line 43) | def test_conversation_item_list():

FILE: tests/unit/conversations/test_conversations.py
  function service (line 52) | async def service():
  function test_conversation_lifecycle (line 76) | async def test_conversation_lifecycle(service):
  function test_conversation_items (line 89) | async def test_conversation_items(service):
  function test_invalid_conversation_id (line 110) | async def test_invalid_conversation_id(service):
  function test_invalid_conversation_id_on_retrieve (line 115) | async def test_invalid_conversation_id_on_retrieve(service):
  function test_invalid_conversation_id_on_update (line 120) | async def test_invalid_conversation_id_on_update(service):
  function test_invalid_conversation_id_on_delete (line 127) | async def test_invalid_conversation_id_on_delete(service):
  function test_nonexistent_conversation_raises_conversation_not_found (line 132) | async def test_nonexistent_conversation_raises_conversation_not_found(se...
  function test_retrieve_nonexistent_item_raises_conversation_item_not_found (line 139) | async def test_retrieve_nonexistent_item_raises_conversation_item_not_fo...
  function test_openai_type_compatibility (line 149) | async def test_openai_type_compatibility(service):
  function test_items_not_returned_on_creation_or_retrieval (line 181) | async def test_items_not_returned_on_creation_or_retrieval(service):
  function test_policy_configuration (line 202) | async def test_policy_configuration():
  function test_add_items_defaults_message_type (line 236) | async def test_add_items_defaults_message_type(service):
  function test_create_conversation_defaults_message_type (line 249) | async def test_create_conversation_defaults_message_type(service):

FILE: tests/unit/core/connectors/test_connectors.py
  function mock_kvstore (line 30) | def mock_kvstore():
  function connector_service (line 61) | async def connector_service(mock_kvstore):
  function sample_tool_def (line 76) | def sample_tool_def():
  function mock_connectors_api (line 87) | def mock_connectors_api():
  function sample_connector (line 94) | def sample_connector():
  class TestRegisterConnector (line 108) | class TestRegisterConnector:
    method test_register_new_connector (line 111) | async def test_register_new_connector(self, connector_service, mock_kv...
    method test_register_connector_different_config_updates (line 129) | async def test_register_connector_different_config_updates(self, conne...
  class TestGetConnector (line 156) | class TestGetConnector:
    method test_get_connector_not_found (line 159) | async def test_get_connector_not_found(self, connector_service):
    method test_get_connector_returns_with_server_info (line 166) | async def test_get_connector_returns_with_server_info(self, connector_...
    method test_get_connector_with_authorization (line 191) | async def test_get_connector_with_authorization(self, connector_service):
  class TestListConnectors (line 220) | class TestListConnectors:
    method test_list_connectors_empty (line 223) | async def test_list_connectors_empty(self, connector_service):
    method test_list_connectors_returns_all (line 229) | async def test_list_connectors_returns_all(self, connector_service):
    method test_list_connectors_after_unregister (line 256) | async def test_list_connectors_after_unregister(self, connector_service):
  class TestUnregisterConnector (line 282) | class TestUnregisterConnector:
    method test_unregister_existing_connector (line 285) | async def test_unregister_existing_connector(self, connector_service, ...
    method test_unregister_nonexistent_connector_does_not_raise (line 303) | async def test_unregister_nonexistent_connector_does_not_raise(self, c...
  class TestKeyPrefix (line 312) | class TestKeyPrefix:
    method test_connectors_use_namespaced_keys (line 315) | async def test_connectors_use_namespaced_keys(self, connector_service,...
  class TestMCPToolValidation (line 332) | class TestMCPToolValidation:
    method test_mcp_tool_requires_server_url_or_connector_id (line 335) | def test_mcp_tool_requires_server_url_or_connector_id(self):
    method test_mcp_tool_accepts_server_url_only (line 344) | def test_mcp_tool_accepts_server_url_only(self):
    method test_mcp_tool_accepts_connector_id_only (line 354) | def test_mcp_tool_accepts_connector_id_only(self):
    method test_mcp_tool_accepts_both_server_url_and_connector_id (line 364) | def test_mcp_tool_accepts_both_server_url_and_connector_id(self):
  class TestConnectorIdResolution (line 379) | class TestConnectorIdResolution:
    method test_connector_id_resolved_to_server_url (line 382) | async def test_connector_id_resolved_to_server_url(self, mock_connecto...
    method test_server_url_not_overwritten_when_provided (line 402) | async def test_server_url_not_overwritten_when_provided(self, mock_con...
    method test_connector_id_resolution_propagates_not_found_error (line 423) | async def test_connector_id_resolution_propagates_not_found_error(self...

FILE: tests/unit/core/routers/test_agents_router.py
  function test_openapi_create_response_advertises_json_and_sse_200 (line 30) | def test_openapi_create_response_advertises_json_and_sse_200():
  function test_create_response_returns_sse_streaming_response_when_impl_streams (line 59) | async def test_create_response_returns_sse_streaming_response_when_impl_...
  function test_create_response_maps_value_error_to_400 (line 84) | def test_create_response_maps_value_error_to_400():
  function test_create_response_returns_json_for_non_streaming (line 102) | async def test_create_response_returns_json_for_non_streaming():
  function test_sse_format_is_correct (line 136) | async def test_sse_format_is_correct():
  function test_sse_stream_keeps_provider_context (line 172) | async def test_sse_stream_keeps_provider_context():
  function test_sse_stream_reports_value_error_as_http_exception (line 211) | async def test_sse_stream_reports_value_error_as_http_exception():
  function test_get_response_returns_response_object (line 244) | async def test_get_response_returns_response_object():
  function test_get_response_maps_value_error_to_400 (line 278) | def test_get_response_maps_value_error_to_400():
  function test_list_responses_returns_list (line 296) | async def test_list_responses_returns_list():
  function test_list_input_items_returns_items (line 328) | async def test_list_input_items_returns_items():
  function test_delete_response_returns_confirmation (line 354) | async def test_delete_response_returns_confirmation():
  function test_delete_response_maps_value_error_to_400 (line 384) | def test_delete_response_maps_value_error_to_400():
  function test_request_validation_error_passes_through_route_class (line 402) | def test_request_validation_error_passes_through_route_class():
  function test_exception_translating_route_converts_value_error_to_400 (line 430) | def test_exception_translating_route_converts_value_error_to_400():
  function test_unknown_exception_propagates_to_global_handler (line 453) | def test_unknown_exception_propagates_to_global_handler():
  function test_consecutive_value_errors_keep_connection_alive (line 477) | def test_consecutive_value_errors_keep_connection_alive():

FILE: tests/unit/core/routers/test_connectors_router.py
  function anyio_backend (line 17) | def anyio_backend():
  function _create_mock_connector (line 38) | def _create_mock_connector():
  function _create_mock_tool (line 50) | def _create_mock_tool():
  function _get_endpoint (line 58) | def _get_endpoint(router, path: str, method: str = "GET"):
  function test_list_connectors_returns_empty_list (line 68) | async def test_list_connectors_returns_empty_list():
  function test_list_connectors_returns_connectors (line 84) | async def test_list_connectors_returns_connectors():
  function test_get_connector_returns_connector (line 106) | async def test_get_connector_returns_connector():
  function test_get_connector_with_authorization (line 125) | async def test_get_connector_with_authorization():
  function test_get_connector_not_found_raises_error (line 145) | async def test_get_connector_not_found_raises_error():
  function test_list_connector_tools_returns_tools (line 164) | async def test_list_connector_tools_returns_tools():
  function test_list_connector_tools_empty (line 183) | async def test_list_connector_tools_empty():
  function test_list_connector_tools_with_authorization (line 199) | async def test_list_connector_tools_with_authorization():
  function test_get_connector_tool_returns_tool (line 220) | async def test_get_connector_tool_returns_tool():
  function test_get_connector_tool_with_authorization (line 242) | async def test_get_connector_tool_with_authorization():
  function test_get_connector_tool_not_found_raises_error (line 263) | async def test_get_connector_tool_not_found_raises_error():
  function test_openapi_schema_has_connectors_endpoints (line 285) | def test_openapi_schema_has_connectors_endpoints():
  function test_openapi_schema_list_connectors_is_get (line 301) | def test_openapi_schema_list_connectors_is_get():
  function test_openapi_schema_get_connector_has_path_param (line 315) | def test_openapi_schema_get_connector_has_path_param():
  function test_openapi_schema_has_authorization_query_param (line 331) | def test_openapi_schema_has_authorization_query_param():

FILE: tests/unit/core/routers/test_conversations_router.py
  function test_consecutive_errors_keep_connection_alive (line 17) | def test_consecutive_errors_keep_connection_alive():

FILE: tests/unit/core/routers/test_inference_router.py
  function mock_routing_table (line 34) | def mock_routing_table():
  function test_rerank_calls_provider_correctly (line 52) | async def test_rerank_calls_provider_correctly(mock_routing_table):

FILE: tests/unit/core/routers/test_safety_router.py
  function test_run_moderation_uses_default_shield_when_model_missing (line 20) | async def test_run_moderation_uses_default_shield_when_model_missing():

FILE: tests/unit/core/routers/test_vector_io.py
  function _mock_routing_table (line 22) | def _mock_routing_table(**kwargs):
  function test_single_provider_auto_selection (line 29) | async def test_single_provider_auto_selection():
  function test_create_vector_stores_multiple_providers_missing_provider_id_error (line 54) | async def test_create_vector_stores_multiple_providers_missing_provider_...
  function test_update_vector_store_provider_id_change_fails (line 76) | async def test_update_vector_store_provider_id_change_fails():
  function test_update_vector_store_same_provider_id_succeeds (line 109) | async def test_update_vector_store_same_provider_id_succeeds():
  function test_register_vector_store_only_once (line 134) | async def test_register_vector_store_only_once():
  function test_create_vector_store_with_unknown_embedding_model_raises_error (line 160) | async def test_create_vector_store_with_unknown_embedding_model_raises_e...
  function test_create_vector_store_with_wrong_model_type_raises_error (line 175) | async def test_create_vector_store_with_wrong_model_type_raises_error():
  function test_query_rewrite_functionality (line 189) | async def test_query_rewrite_functionality():
  function test_query_rewrite_error_when_not_configured (line 251) | async def test_query_rewrite_error_when_not_configured():
  function test_query_rewrite_with_custom_prompt (line 271) | async def test_query_rewrite_with_custom_prompt():
  function test_search_without_rewrite (line 315) | async def test_search_without_rewrite():

FILE: tests/unit/core/routers/test_vector_stores_abac.py
  class MockDistRegistry (line 45) | class MockDistRegistry:
    method __init__ (line 48) | def __init__(self):
    method get_cached (line 51) | def get_cached(self, type_name: str, identifier: str):
  function mock_provider (line 56) | def mock_provider():
  function router_with_real_routing_table (line 154) | def router_with_real_routing_table(mock_provider):
  function test_operation_enforces_correct_abac_permission (line 324) | async def test_operation_enforces_correct_abac_permission(
  function test_operations_fail_before_provider_when_unauthorized (line 347) | async def test_operations_fail_before_provider_when_unauthorized(router_...

FILE: tests/unit/core/test_dynamic_config.py
  function _set_required_env_vars (line 30) | def _set_required_env_vars(monkeypatch: pytest.MonkeyPatch):
  class _SampleConfig (line 37) | class _SampleConfig(BaseModel):
    method sample_run_config (line 44) | def sample_run_config(cls, __distro_dir__: str = "", **kwargs: Any) ->...
  function _make_registry (line 53) | def _make_registry(provider_type: str) -> dict[Api, dict[str, ProviderSp...
  class TestParsing (line 62) | class TestParsing:
    method test_missing_equals_raises (line 63) | def test_missing_equals_raises(self):
    method test_semicolon_separator_accepted (line 67) | def test_semicolon_separator_accepted(self, tmp_path: Path):
    method test_comma_separator_accepted (line 77) | def test_comma_separator_accepted(self, tmp_path: Path):
  class TestValidation (line 87) | class TestValidation:
    method test_invalid_api_name_raises (line 88) | def test_invalid_api_name_raises(self):
    method test_api_absent_from_registry_raises (line 92) | def test_api_absent_from_registry_raises(self):
    method test_unknown_provider_raises (line 96) | def test_unknown_provider_raises(self):
  class TestProviderResolution (line 110) | class TestProviderResolution:
    method test_exact_match (line 111) | def test_exact_match(self, tmp_path: Path):
    method test_inline_prefix_fallback (line 120) | def test_inline_prefix_fallback(self, tmp_path: Path):
    method test_remote_prefix_fallback (line 129) | def test_remote_prefix_fallback(self, tmp_path: Path):
  class TestMultiProvider (line 139) | class TestMultiProvider:
    method test_same_api_twice_accumulates (line 140) | def test_same_api_twice_accumulates(self, tmp_path: Path):
  class TestEnvVarSubstitution (line 161) | class TestEnvVarSubstitution:
    method test_default_value_used_when_env_not_set (line 162) | def test_default_value_used_when_env_not_set(self, tmp_path: Path, mon...
    method test_env_value_overrides_default (line 176) | def test_env_value_overrides_default(self, tmp_path: Path, monkeypatch...
    method test_missing_required_env_var_raises (line 189) | def test_missing_required_env_var_raises(self, tmp_path: Path, monkeyp...
  class TestConfigShape (line 201) | class TestConfigShape:
    method test_apis_list_matches_providers (line 202) | def test_apis_list_matches_providers(self, tmp_path: Path):
    method test_storage_paths_use_distro_dir (line 212) | def test_storage_paths_use_distro_dir(self, tmp_path: Path):
    method test_sqlite_store_dir_honoured (line 224) | def test_sqlite_store_dir_honoured(self, tmp_path: Path, monkeypatch: ...
    method test_provider_id_set_to_spec_string (line 243) | def test_provider_id_set_to_spec_string(self, tmp_path: Path):

FILE: tests/unit/core/test_errors.py
  class TestClientListCommand (line 35) | class TestClientListCommand:
    method test_basic_command (line 36) | def test_basic_command(self):
    method test_with_argument (line 40) | def test_with_argument(self):
    method test_with_multiple_arguments (line 44) | def test_with_multiple_arguments(self):
    method test_with_resource_plural (line 48) | def test_with_resource_plural(self):
  class TestTranslateExceptionToHttp (line 53) | class TestTranslateExceptionToHttp:
    method test_value_error (line 59) | def test_value_error(self):
    method test_permission_error (line 66) | def test_permission_error(self):
    method test_connection_error (line 73) | def test_connection_error(self):
    method test_timeout_error (line 79) | def test_timeout_error(self):
    method test_asyncio_timeout_error (line 86) | def test_asyncio_timeout_error(self):
    method test_not_implemented_error (line 92) | def test_not_implemented_error(self):
    method test_subclass_one_level_deep (line 101) | def test_subclass_one_level_deep(self):
    method test_subclass_two_levels_deep (line 113) | def test_subclass_two_levels_deep(self):
    method test_llama_stack_error_not_in_map (line 127) | def test_llama_stack_error_not_in_map(self):
    method test_multiple_mapped_parents_first_wins (line 139) | def test_multiple_mapped_parents_first_wins(self):
    method test_multiple_mapped_parents_reversed_order (line 152) | def test_multiple_mapped_parents_reversed_order(self):
    method test_unmapped_ancestors_before_mapped (line 166) | def test_unmapped_ancestors_before_mapped(self):
    method test_unmapped_runtime_error (line 182) | def test_unmapped_runtime_error(self):
    method test_unmapped_key_error (line 186) | def test_unmapped_key_error(self):
    method test_bare_exception (line 190) | def test_bare_exception(self):
    method test_llama_stack_error_base_not_in_map (line 194) | def test_llama_stack_error_base_not_in_map(self):
    method test_detail_uses_exception_message (line 207) | def test_detail_uses_exception_message(self):
    method test_empty_message_uses_fallback (line 214) | def test_empty_message_uses_fallback(self):
    method test_connection_error_uses_message (line 222) | def test_connection_error_uses_message(self):
    method test_connection_error_empty_uses_fallback (line 228) | def test_connection_error_empty_uses_fallback(self):
    method test_permission_error_uses_message (line 234) | def test_permission_error_uses_message(self):
    method test_timeout_uses_message (line 240) | def test_timeout_uses_message(self):
    method test_timeout_empty_uses_fallback (line 246) | def test_timeout_empty_uses_fallback(self):
  class TestTranslateException (line 253) | class TestTranslateException:
    method test_always_returns_http_exception (line 260) | def test_always_returns_http_exception(self):
    method test_resource_not_found_error_uses_404 (line 281) | def test_resource_not_found_error_uses_404(self):
    method test_model_not_found_error_uses_404 (line 287) | def test_model_not_found_error_uses_404(self):
    method test_batch_not_found_error_uses_404 (line 294) | def test_batch_not_found_error_uses_404(self):
    method test_conflict_error_uses_409 (line 299) | def test_conflict_error_uses_409(self):
    method test_token_validation_error_uses_401 (line 305) | def test_token_validation_error_uses_401(self):
    method test_model_type_error_uses_400 (line 311) | def test_model_type_error_uses_400(self):
    method test_unsupported_model_error_uses_400 (line 317) | def test_unsupported_model_error_uses_400(self):
    method test_invalid_parameter_error_uses_400 (line 322) | def test_invalid_parameter_error_uses_400(self):
    method test_service_not_enabled_error_uses_503 (line 327) | def test_service_not_enabled_error_uses_503(self):
    method test_internal_server_error_uses_500 (line 332) | def test_internal_server_error_uses_500(self):
    method test_response_not_found_error_uses_404 (line 337) | def test_response_not_found_error_uses_404(self):
    method test_response_input_item_not_found_error_uses_404 (line 342) | def test_response_input_item_not_found_error_uses_404(self):
    method test_conversation_not_found_error_uses_404 (line 347) | def test_conversation_not_found_error_uses_404(self):
    method test_conversation_item_not_found_error_uses_404 (line 352) | def test_conversation_item_not_found_error_uses_404(self):
    method test_llama_stack_error_preserves_message (line 357) | def test_llama_stack_error_preserves_message(self):
    method test_plain_value_error_maps_to_400 (line 365) | def test_plain_value_error_maps_to_400(self):
    method test_permission_error_maps_to_403 (line 370) | def test_permission_error_maps_to_403(self):
    method test_not_implemented_error_maps_to_501 (line 375) | def test_not_implemented_error_maps_to_501(self):
    method test_timeout_error_maps_to_504 (line 380) | def test_timeout_error_maps_to_504(self):
    method test_connection_error_maps_to_502 (line 385) | def test_connection_error_maps_to_502(self):
    method test_provider_sdk_exception_preserves_status_code (line 392) | def test_provider_sdk_exception_preserves_status_code(self):
    method test_provider_sdk_exception_with_401 (line 405) | def test_provider_sdk_exception_with_401(self):
    method test_runtime_error_falls_to_500 (line 417) | def test_runtime_error_falls_to_500(self):
    method test_bare_exception_falls_to_500 (line 422) | def test_bare_exception_falls_to_500(self):
    method test_key_error_falls_to_500 (line 427) | def test_key_error_falls_to_500(self):
    method test_unknown_exception_detail_is_generic (line 432) | def test_unknown_exception_detail_is_generic(self):
    method test_request_validation_error (line 441) | def test_request_validation_error(self):
    method test_pydantic_validation_error (line 448) | def test_pydantic_validation_error(self):

FILE: tests/unit/core/test_provider_data_context.py
  function request_provider_data_context (line 19) | def request_provider_data_context(headers):
  function create_sse_event (line 29) | def create_sse_event(data):
  function sse_generator (line 33) | async def sse_generator(event_gen_coroutine):
  function async_event_gen (line 40) | async def async_event_gen():
  function test_provider_data_context_cleared_between_sse_requests (line 47) | async def test_provider_data_context_cleared_between_sse_requests():

FILE: tests/unit/core/test_stack_validation.py
  class TestVectorStoresValidation (line 37) | class TestVectorStoresValidation:
    method test_validate_missing_model (line 38) | async def test_validate_missing_model(self):
    method test_validate_success (line 70) | async def test_validate_success(self):
    method test_validate_rewrite_query_prompt_missing_placeholder (line 117) | async def test_validate_rewrite_query_prompt_missing_placeholder(self):
  class TestSafetyConfigValidation (line 127) | class TestSafetyConfigValidation:
    method test_validate_success (line 128) | async def test_validate_success(self):
    method test_validate_wrong_shield_id (line 143) | async def test_validate_wrong_shield_id(self):
  class TestRegisterConnectors (line 161) | class TestRegisterConnectors:
    method _make_stack_config (line 164) | def _make_stack_config(self, connectors: list[ConnectorInput]) -> Stac...
    method test_register_connectors_creates_new (line 181) | async def test_register_connectors_creates_new(self):
    method test_register_connectors_removes_orphans (line 218) | async def test_register_connectors_removes_orphans(self):
    method test_register_connectors_updates_existing (line 260) | async def test_register_connectors_updates_existing(self):
    method test_register_connectors_full_cud_flow (line 299) | async def test_register_connectors_full_cud_flow(self):
    method test_register_connectors_empty_config_removes_all (line 351) | async def test_register_connectors_empty_config_removes_all(self):
    method test_register_connectors_skipped_if_api_not_available (line 380) | async def test_register_connectors_skipped_if_api_not_available(self):

FILE: tests/unit/core/test_storage_references.py
  function _base_run_config (line 29) | def _base_run_config(**overrides):
  function test_references_require_known_backend (line 66) | def test_references_require_known_backend():
  function test_references_must_match_backend_family (line 71) | def test_references_must_match_backend_family():
  function test_valid_configuration_passes_validation (line 81) | def test_valid_configuration_passes_validation():
  function test_default_backends_resolve_env_vars (line 90) | def test_default_backends_resolve_env_vars(backend_key, monkeypatch):
  function test_default_backends_respect_sqlite_store_dir (line 104) | def test_default_backends_respect_sqlite_store_dir(monkeypatch):
  function test_default_backends_fallback_to_distribs_base_dir (line 112) | def test_default_backends_fallback_to_distribs_base_dir(monkeypatch):
  function test_default_backends_expand_user_home (line 122) | def test_default_backends_expand_user_home(monkeypatch):

FILE: tests/unit/core/test_storage_shutdown.py
  class TestKVStoreShutdown (line 32) | class TestKVStoreShutdown:
    method test_sqlite_kvstore_shutdown_memory (line 35) | async def test_sqlite_kvstore_shutdown_memory(self):
    method test_sqlite_kvstore_shutdown_file (line 55) | async def test_sqlite_kvstore_shutdown_file(self):
    method test_inmemory_kvstore_shutdown (line 73) | async def test_inmemory_kvstore_shutdown(self):
    method test_shutdown_kvstore_backends (line 89) | async def test_shutdown_kvstore_backends(self):
  class TestSqlStoreShutdown (line 114) | class TestSqlStoreShutdown:
    method test_sqlalchemy_sqlstore_shutdown (line 117) | async def test_sqlalchemy_sqlstore_shutdown(self):
    method test_shutdown_sqlstore_backends (line 141) | async def test_shutdown_sqlstore_backends(self):
  class TestKVStoreProtocolShutdown (line 164) | class TestKVStoreProtocolShutdown:
    method test_sqlite_kvstore_has_shutdown (line 167) | async def test_sqlite_kvstore_has_shutdown(self):
    method test_inmemory_kvstore_has_shutdown (line 174) | async def test_inmemory_kvstore_has_shutdown(self):
  class TestSqlStoreProtocolShutdown (line 181) | class TestSqlStoreProtocolShutdown:
    method test_sqlalchemy_sqlstore_has_shutdown (line 184) | async def test_sqlalchemy_sqlstore_has_shutdown(self):
  class TestShutdownIdempotency (line 195) | class TestShutdownIdempotency:
    method test_sqlite_kvstore_shutdown_idempotent (line 198) | async def test_sqlite_kvstore_shutdown_idempotent(self):
    method test_sqlalchemy_sqlstore_shutdown_idempotent (line 209) | async def test_sqlalchemy_sqlstore_shutdown_idempotent(self):
    method test_inmemory_kvstore_shutdown_idempotent (line 220) | async def test_inmemory_kvstore_shutdown_idempotent(self):

FILE: tests/unit/core/test_task.py
  class _CollectingExporter (line 23) | class _CollectingExporter(SpanExporter):
    method __init__ (line 26) | def __init__(self):
    method export (line 29) | def export(self, spans):
  function test_detached_task_runs_coroutine (line 34) | async def test_detached_task_runs_coroutine():
  function test_detached_task_clears_otel_context (line 46) | async def test_detached_task_clears_otel_context():
  function test_detached_task_clears_provider_data (line 68) | async def test_detached_task_clears_provider_data():
  function test_detached_task_restores_caller_context (line 86) | async def test_detached_task_restores_caller_context():
  function test_detached_task_produces_independent_trace (line 105) | async def test_detached_task_produces_independent_trace():
  function test_normal_child_task_shares_trace (line 130) | async def test_normal_child_task_shares_trace():
  function test_context_through_queue_pattern (line 156) | async def test_context_through_queue_pattern():
  function test_capture_and_activate_request_context (line 202) | async def test_capture_and_activate_request_context():
  function test_activate_restores_on_exit (line 229) | async def test_activate_restores_on_exit():
  function test_context_through_queue_no_cross_contamination (line 252) | async def test_context_through_queue_no_cross_contamination():

FILE: tests/unit/core/test_type_inspection.py
  class SampleModel (line 15) | class SampleModel(BaseModel):
  function test_is_body_param_with_body_annotation (line 19) | def test_is_body_param_with_body_annotation() -> None:
  function test_is_body_param_with_query_annotation (line 23) | def test_is_body_param_with_query_annotation() -> None:
  function test_is_body_param_with_plain_type (line 27) | def test_is_body_param_with_plain_type() -> None:
  function test_is_unwrapped_body_param_with_default_embed (line 32) | def test_is_unwrapped_body_param_with_default_embed() -> None:
  function test_is_unwrapped_body_param_with_embed_false (line 37) | def test_is_unwrapped_body_param_with_embed_false() -> None:
  function test_is_unwrapped_body_param_with_embed_true (line 41) | def test_is_unwrapped_body_param_with_embed_true() -> None:

FILE: tests/unit/core/test_vector_store_config_registration.py
  class TestVectorStoreRegistration (line 21) | class TestVectorStoreRegistration:
    method test_basic_registration (line 24) | async def test_basic_registration(self):
    method test_empty_config (line 154) | async def test_empty_config(self):
    method test_registration_with_optional_fields (line 178) | async def test_registration_with_optional_fields(self):
  class TestOpenAIMetadataCreation (line 266) | class TestOpenAIMetadataCreation:
    method test_config_registration_creates_openai_metadata (line 269) | async def test_config_registration_creates_openai_metadata(self, disk_...

FILE: tests/unit/core/test_vector_stores_config.py
  class TestVectorStoresConfigValidation (line 13) | class TestVectorStoresConfigValidation:
    method test_default_config_is_valid (line 16) | def test_default_config_is_valid(self):
    method test_template_validation_errors (line 31) | def test_template_validation_errors(self):
    method test_rewrite_query_params_validation (line 55) | def test_rewrite_query_params_validation(self):
    method test_custom_configuration (line 75) | def test_custom_configuration(self):
  class TestOptionalArchitecture (line 102) | class TestOptionalArchitecture:
    method test_guaranteed_defaults_behavior (line 105) | def test_guaranteed_defaults_behavior(self):
    method test_guaranteed_defaults_have_expected_values (line 114) | def test_guaranteed_defaults_have_expected_values(self):
    method test_end_to_end_template_usage (line 139) | def test_end_to_end_template_usage(self):

FILE: tests/unit/distribution/routers/test_routing_tables.py
  class Impl (line 48) | class Impl:
    method __init__ (line 49) | def __init__(self, api: Api):
    method __provider_spec__ (line 53) | def __provider_spec__(self):
  class InferenceImpl (line 59) | class InferenceImpl(Impl):
    method __init__ (line 60) | def __init__(self):
    method register_model (line 63) | async def register_model(self, model: Model):
    method unregister_model (line 66) | async def unregister_model(self, model_id: str):
    method should_refresh_models (line 69) | async def should_refresh_models(self):
    method list_models (line 72) | async def list_models(self):
    method shutdown (line 90) | async def shutdown(self):
  class SafetyImpl (line 94) | class SafetyImpl(Impl):
    method __init__ (line 95) | def __init__(self):
    method register_shield (line 98) | async def register_shield(self, shield: Shield):
    method unregister_shield (line 101) | async def unregister_shield(self, shield_id: str):
  class DatasetsImpl (line 105) | class DatasetsImpl(Impl):
    method __init__ (line 106) | def __init__(self):
    method register_dataset (line 109) | async def register_dataset(self, dataset: Dataset):
    method unregister_dataset (line 112) | async def unregister_dataset(self, dataset_id: str):
  class ScoringFunctionsImpl (line 116) | class ScoringFunctionsImpl(Impl):
    method __init__ (line 117) | def __init__(self):
    method list_scoring_functions (line 120) | async def list_scoring_functions(self):
    method register_scoring_function (line 123) | async def register_scoring_function(self, scoring_fn):
    method unregister_scoring_function (line 126) | async def unregister_scoring_function(self, scoring_fn_id: str):
  class BenchmarksImpl (line 130) | class BenchmarksImpl(Impl):
    method __init__ (line 131) | def __init__(self):
    method register_benchmark (line 134) | async def register_benchmark(self, benchmark):
    method unregister_benchmark (line 137) | async def unregister_benchmark(self, benchmark_id: str):
  class ToolGroupsImpl (line 141) | class ToolGroupsImpl(Impl):
    method __init__ (line 142) | def __init__(self):
    method register_toolgroup (line 145) | async def register_toolgroup(self, toolgroup: ToolGroup):
    method unregister_toolgroup (line 148) | async def unregister_toolgroup(self, toolgroup_id: str):
    method list_runtime_tools (line 151) | async def list_runtime_tools(self, toolgroup_id, mcp_endpoint, authori...
  function test_models_routing_table (line 166) | async def test_models_routing_table(cached_disk_dist_registry):
  function test_shields_routing_table (line 221) | async def test_shields_routing_table(cached_disk_dist_registry):
  function test_datasets_routing_table (line 266) | async def test_datasets_routing_table(cached_disk_dist_registry):
  function test_scoring_functions_routing_table (line 299) | async def test_scoring_functions_routing_table(cached_disk_dist_registry):
  function test_double_registration_models_positive (line 343) | async def test_double_registration_models_positive(cached_disk_dist_regi...
  function test_double_registration_models_negative (line 360) | async def test_double_registration_models_negative(cached_disk_dist_regi...
  function test_double_registration_scoring_functions_positive (line 375) | async def test_double_registration_scoring_functions_positive(cached_dis...
  function test_double_registration_scoring_functions_negative (line 408) | async def test_double_registration_scoring_functions_negative(cached_dis...
  function test_double_registration_different_providers (line 437) | async def test_double_registration_different_providers(cached_disk_dist_...
  function test_benchmarks_routing_table (line 456) | async def test_benchmarks_routing_table(cached_disk_dist_registry):
  function test_benchmarks_routing_table_stores_dataset_id (line 484) | async def test_benchmarks_routing_table_stores_dataset_id(cached_disk_di...
  function test_tool_groups_routing_table (line 510) | async def test_tool_groups_routing_table(cached_disk_dist_registry):
  function test_models_alias_registration_and_lookup (line 530) | async def test_models_alias_registration_and_lookup(cached_disk_dist_reg...
  function test_models_multi_provider_disambiguation (line 557) | async def test_models_multi_provider_disambiguation(cached_disk_dist_reg...
  function test_models_fallback_lookup_behavior (line 588) | async def test_models_fallback_lookup_behavior(cached_disk_dist_registry):
  function test_models_source_tracking_default (line 616) | async def test_models_source_tracking_default(cached_disk_dist_registry):
  function test_models_source_tracking_provider (line 634) | async def test_models_source_tracking_provider(cached_disk_dist_registry):
  function test_models_source_interaction_preserves_default (line 670) | async def test_models_source_interaction_preserves_default(cached_disk_d...
  function test_models_source_interaction_cleanup_provider_models (line 727) | async def test_models_source_interaction_cleanup_provider_models(cached_...
  function test_tool_groups_routing_table_exception_handling (line 783) | async def test_tool_groups_routing_table_exception_handling(cached_disk_...

FILE: tests/unit/distribution/test_api_recordings.py
  function temp_storage_dir (line 35) | def temp_storage_dir():
  function real_openai_chat_response (line 42) | def real_openai_chat_response():
  function real_embeddings_response (line 61) | def real_embeddings_response():
  class TestInferenceRecording (line 74) | class TestInferenceRecording:
    method test_request_normalization (line 77) | def test_request_normalization(self):
    method test_request_normalization_edge_cases (line 111) | def test_request_normalization_edge_cases(self):
    method test_response_storage (line 175) | def test_response_storage(self, temp_storage_dir):
    method test_recording_mode (line 198) | async def test_recording_mode(self, temp_storage_dir, real_openai_chat...
    method test_replay_mode (line 223) | async def test_replay_mode(self, temp_storage_dir, real_openai_chat_re...
    method test_replay_missing_recording (line 260) | async def test_replay_missing_recording(self, temp_storage_dir):
    method test_embeddings_recording (line 272) | async def test_embeddings_recording(self, temp_storage_dir, real_embed...
    method test_live_mode (line 306) | async def test_live_mode(self, real_openai_chat_response):
  class TestExceptionRecordingReplay (line 324) | class TestExceptionRecordingReplay:
    method test_openai_error_is_recorded_and_replayed (line 334) | async def test_openai_error_is_recorded_and_replayed(self, temp_storag...
    method test_replay_legacy_exception_format_raises_generic (line 371) | async def test_replay_legacy_exception_format_raises_generic(self, tem...

FILE: tests/unit/distribution/test_context.py
  function test_preserve_contexts_with_exception (line 16) | async def test_preserve_contexts_with_exception():
  function test_preserve_contexts_empty_generator (line 43) | async def test_preserve_contexts_empty_generator():
  function test_preserve_contexts_across_event_loops (line 67) | async def test_preserve_contexts_across_event_loops():

FILE: tests/unit/distribution/test_distribution.py
  class SampleConfig (line 28) | class SampleConfig(BaseModel):
    method sample_run_config (line 35) | def sample_run_config(cls, **kwargs: Any) -> dict[str, Any]:
  function _default_storage (line 41) | def _default_storage() -> StorageConfig:
  function make_stack_config (line 56) | def make_stack_config(**overrides) -> StackConfig:
  function mock_providers (line 69) | def mock_providers():
  function base_config (line 84) | def base_config(tmp_path):
  function provider_spec_yaml (line 102) | def provider_spec_yaml():
  function inline_provider_spec_yaml (line 114) | def inline_provider_spec_yaml():
  function api_directories (line 131) | def api_directories(tmp_path):
  function make_import_module_side_effect (line 144) | def make_import_module_side_effect(
  class TestProviderRegistry (line 178) | class TestProviderRegistry:
    method test_builtin_providers (line 181) | def test_builtin_providers(self, mock_providers):
    method test_internal_apis_excluded (line 190) | def test_internal_apis_excluded(self):
    method test_external_remote_providers (line 208) | def test_external_remote_providers(self, api_directories, mock_provide...
    method test_external_inline_providers (line 225) | def test_external_inline_providers(self, api_directories, mock_provide...
    method test_invalid_yaml (line 246) | def test_invalid_yaml(self, api_directories, mock_providers, base_conf...
    method test_missing_directory (line 257) | def test_missing_directory(self, mock_providers):
    method test_empty_api_directory (line 275) | def test_empty_api_directory(self, api_directories, mock_providers, ba...
    method test_malformed_remote_provider_spec (line 280) | def test_malformed_remote_provider_spec(self, api_directories, mock_pr...
    method test_malformed_inline_provider_spec (line 295) | def test_malformed_inline_provider_spec(self, api_directories, mock_pr...
    method test_external_provider_from_module_success (line 311) | def test_external_provider_from_module_success(self, mock_providers):
    method test_external_provider_from_module_not_found (line 351) | def test_external_provider_from_module_not_found(self, mock_providers):
    method test_external_provider_from_module_missing_get_provider_spec (line 374) | def test_external_provider_from_module_missing_get_provider_spec(self,...
    method test_external_provider_from_module_listing (line 396) | def test_external_provider_from_module_listing(self, mock_providers):
  class TestGetExternalProvidersFromModule (line 426) | class TestGetExternalProvidersFromModule:
    method test_stackrunconfig_provider_without_module (line 429) | def test_stackrunconfig_provider_without_module(self, mock_providers):
    method test_stackrunconfig_with_version_spec (line 453) | def test_stackrunconfig_with_version_spec(self, mock_providers):
    method test_buildconfig_does_not_import_module (line 492) | def test_buildconfig_does_not_import_module(self, mock_providers):
    method test_buildconfig_multiple_providers (line 528) | def test_buildconfig_multiple_providers(self, mock_providers):
    method test_distributionspec_does_not_import_module (line 552) | def test_distributionspec_does_not_import_module(self, mock_providers):
    method test_list_return_from_get_provider_spec (line 584) | def test_list_return_from_get_provider_spec(self, mock_providers):
    method test_list_return_filters_by_provider_type (line 632) | def test_list_return_filters_by_provider_type(self, mock_providers):
    method test_list_return_adds_multiple_provider_types (line 680) | def test_list_return_adds_multiple_provider_types(self, mock_providers):
    method test_module_not_found_raises_value_error (line 737) | def test_module_not_found_raises_value_error(self, mock_providers):
    method test_generic_exception_is_raised (line 767) | def test_generic_exception_is_raised(self, mock_providers):
    method test_empty_provider_list (line 804) | def test_empty_provider_list(self, mock_providers):
    method test_multiple_apis_with_providers (line 819) | def test_multiple_apis_with_providers(self, mock_providers):

FILE: tests/unit/distribution/test_library_client_initialization.py
  class TestLlamaStackAsLibraryClientAutoInitialization (line 23) | class TestLlamaStackAsLibraryClientAutoInitialization:
    method test_sync_client_auto_initialization (line 26) | def test_sync_client_auto_initialization(self, monkeypatch):
    method test_async_client_auto_initialization (line 49) | async def test_async_client_auto_initialization(self, monkeypatch):
    method test_initialize_method_backward_compatibility (line 75) | def test_initialize_method_backward_compatibility(self, monkeypatch):
    method test_async_initialize_method_idempotent (line 102) | async def test_async_initialize_method_idempotent(self, monkeypatch):
    method test_route_impls_automatically_set (line 128) | def test_route_impls_automatically_set(self, monkeypatch):
  class TestLlamaStackAsLibraryClientShutdown (line 150) | class TestLlamaStackAsLibraryClientShutdown:
    method test_async_client_shutdown (line 153) | async def test_async_client_shutdown(self, monkeypatch):
    method test_async_client_shutdown_idempotent (line 190) | async def test_async_client_shutdown_idempotent(self, monkeypatch):
    method test_async_client_shutdown_before_initialize (line 223) | async def test_async_client_shutdown_before_initialize(self, monkeypat...
    method test_sync_client_shutdown (line 249) | def test_sync_client_shutdown(self, monkeypatch):
    method test_sync_client_shutdown_idempotent (line 279) | def test_sync_client_shutdown_idempotent(self, monkeypatch):
    method test_async_client_has_shutdown_method (line 307) | def test_async_client_has_shutdown_method(self, monkeypatch):
    method test_sync_client_has_shutdown_method (line 332) | def test_sync_client_has_shutdown_method(self, monkeypatch):
  class TestLlamaStackAsLibraryClientContextManager (line 358) | class TestLlamaStackAsLibraryClientContextManager:
    method test_async_client_context_manager (line 361) | async def test_async_client_context_manager(self, monkeypatch):
    method test_async_client_context_manager_with_exception (line 390) | async def test_async_client_context_manager_with_exception(self, monke...
    method test_sync_client_context_manager (line 419) | def test_sync_client_context_manager(self, monkeypatch):
    method test_sync_client_context_manager_with_exception (line 448) | def test_sync_client_context_manager_with_exception(self, monkeypatch):
    method test_async_client_has_context_manager_methods (line 477) | def test_async_client_has_context_manager_methods(self, monkeypatch):
    method test_sync_client_has_context_manager_methods (line 502) | def test_sync_client_has_context_manager_methods(self, monkeypatch):

FILE: tests/unit/distribution/test_list_deps_output.py
  function test_stack_list_deps_basic (line 17) | def test_stack_list_deps_basic():
  function test_stack_list_deps_with_distro_uv (line 39) | def test_stack_list_deps_with_distro_uv():
  function test_list_deps_formatting_quotes_only_for_uv (line 54) | def test_list_deps_formatting_quotes_only_for_uv():

FILE: tests/unit/distribution/test_stack_list.py
  function list_stacks_command (line 18) | def list_stacks_command():
  function mock_distribs_base_dir (line 26) | def mock_distribs_base_dir(tmp_path):
  function mock_distro_dir (line 40) | def mock_distro_dir(tmp_path):
  function create_path_mock (line 54) | def create_path_mock(builtin_dist_dir):
  class TestStackList (line 67) | class TestStackList:
    method test_builtin_distros_shown_without_running (line 70) | def test_builtin_distros_shown_without_running(self, list_stacks_comma...
    method test_custom_distribution_overrides_builtin (line 92) | def test_custom_distribution_overrides_builtin(self, list_stacks_comma...
    method test_hidden_directories_ignored (line 108) | def test_hidden_directories_ignored(self, list_stacks_command, mock_di...

FILE: tests/unit/files/test_files.py
  class MockUploadFile (line 28) | class MockUploadFile:
    method __init__ (line 31) | def __init__(self, content: bytes, filename: str, content_type: str = ...
    method read (line 36) | async def read(self):
  function files_provider (line 41) | async def files_provider(tmp_path):
  function sample_text_file (line 59) | def sample_text_file():
  function sample_json_file (line 66) | def sample_json_file():
  function large_file (line 73) | def large_file():
  class TestOpenAIFilesAPI (line 79) | class TestOpenAIFilesAPI:
    method test_upload_file_success (line 82) | async def test_upload_file_success(self, files_provider, sample_text_f...
    method test_upload_different_purposes (line 97) | async def test_upload_different_purposes(self, files_provider, sample_...
    method test_upload_different_file_types (line 109) | async def test_upload_different_file_types(self, files_provider, sampl...
    method test_list_files_empty (line 124) | async def test_list_files_empty(self, files_provider):
    method test_list_files_with_content (line 133) | async def test_list_files_with_content(self, files_provider, sample_te...
    method test_list_files_with_purpose_filter (line 151) | async def test_list_files_with_purpose_filter(self, files_provider, sa...
    method test_list_files_with_limit (line 164) | async def test_list_files_with_limit(self, files_provider, sample_text...
    method test_list_files_with_order (line 176) | async def test_list_files_with_order(self, files_provider, sample_text...
    method test_retrieve_file_success (line 198) | async def test_retrieve_file_success(self, files_provider, sample_text...
    method test_retrieve_file_not_found (line 218) | async def test_retrieve_file_not_found(self, files_provider):
    method test_retrieve_file_content_success (line 223) | async def test_retrieve_file_content_success(self, files_provider, sam...
    method test_retrieve_file_content_not_found (line 238) | async def test_retrieve_file_content_not_found(self, files_provider):
    method test_delete_file_success (line 245) | async def test_delete_file_success(self, files_provider, sample_text_f...
    method test_delete_file_not_found (line 266) | async def test_delete_file_not_found(self, files_provider):
    method test_file_persistence_across_operations (line 271) | async def test_file_persistence_across_operations(self, files_provider...
    method test_multiple_files_operations (line 302) | async def test_multiple_files_operations(self, files_provider, sample_...
    method test_file_id_uniqueness (line 330) | async def test_file_id_uniqueness(self, files_provider, sample_text_fi...
    method test_file_no_filename_handling (line 343) | async def test_file_no_filename_handling(self, files_provider):
    method test_openaifileobject_schema_has_no_type_object (line 353) | async def test_openaifileobject_schema_has_no_type_object(self):
    method test_file_content_response_schema_is_string (line 378) | async def test_file_content_response_schema_is_string(self):
    method test_expires_at_can_be_none (line 400) | async def test_expires_at_can_be_none(self):
    method test_after_pagination_works (line 412) | async def test_after_pagination_works(self, files_provider, sample_tex...

FILE: tests/unit/files/test_path_security.py
  class MockUploadFile (line 47) | class MockUploadFile:
    method __init__ (line 48) | def __init__(self, content: bytes, filename: str):
    method read (line 52) | async def read(self):
  function files_provider (line 57) | async def files_provider(tmp_path):
  function conversation_service (line 74) | async def conversation_service():
  class TestFileIdValidation (line 99) | class TestFileIdValidation:
    method test_valid_file_ids_accepted (line 112) | def test_valid_file_ids_accepted(self, files_provider, good_id):
    method test_malicious_file_ids_rejected (line 131) | def test_malicious_file_ids_rejected(self, files_provider, bad_id):
  class TestPathContainment (line 141) | class TestPathContainment:
    method test_valid_path_accepted (line 144) | def test_valid_path_accepted(self, files_provider, tmp_path):
    method test_traversal_via_dotdot_rejected (line 150) | def test_traversal_via_dotdot_rejected(self, files_provider):
    method test_absolute_path_outside_storage_rejected (line 156) | def test_absolute_path_outside_storage_rejected(self, files_provider):
    method test_symlink_escape_rejected (line 160) | def test_symlink_escape_rejected(self, files_provider, tmp_path):
    method test_poisoned_db_path_rejected (line 170) | async def test_poisoned_db_path_rejected(self, files_provider):
  class TestContentDispositionSanitization (line 200) | class TestContentDispositionSanitization:
    method test_normal_filename_unchanged (line 203) | def test_normal_filename_unchanged(self):
    method test_traversal_sequences_removed (line 206) | def test_traversal_sequences_removed(self):
    method test_null_bytes_stripped (line 211) | def test_null_bytes_stripped(self):
    method test_hidden_file_gets_prefix (line 215) | def test_hidden_file_gets_prefix(self):
    method test_empty_filename_returns_download (line 219) | def test_empty_filename_returns_download(self):
    method test_backslashes_replaced (line 222) | def test_backslashes_replaced(self):
    method test_slashes_replaced (line 226) | def test_slashes_replaced(self):
    method test_upload_sanitizes_stored_filename (line 230) | async def test_upload_sanitizes_stored_filename(self, files_provider):
    method test_content_disposition_header_uses_sanitized_name (line 243) | async def test_content_disposition_header_uses_sanitized_name(self, fi...
  class TestConversationIdValidation (line 267) | class TestConversationIdValidation:
    method test_valid_conversation_id_accepted (line 272) | def test_valid_conversation_id_accepted(self, conversation_service):
    method test_malicious_conversation_ids_rejected (line 288) | def test_malicious_conversation_ids_rejected(self, conversation_servic...
    method test_add_items_rejects_bad_conversation_id (line 292) | async def test_add_items_rejects_bad_conversation_id(self, conversatio...

FILE: tests/unit/files/test_upload_safety.py
  function _make_upload_file (line 19) | def _make_upload_file(content: bytes, *, filename: str = "test.bin", siz...
  class _NoSizeUploadFile (line 26) | class _NoSizeUploadFile:
    method __init__ (line 33) | def __init__(self, content: bytes):
    method read (line 37) | async def read(self, n: int = -1) -> bytes:
  class _UnchunkedUploadFile (line 41) | class _UnchunkedUploadFile:
    method __init__ (line 48) | def __init__(self, content: bytes):
    method read (line 52) | async def read(self) -> bytes:
  function _make_no_size_upload (line 56) | def _make_no_size_upload(content: bytes) -> _NoSizeUploadFile:
  function _make_unchunked_upload (line 60) | def _make_unchunked_upload(content: bytes) -> _UnchunkedUploadFile:
  class TestReadUploadWithSizeLimit (line 64) | class TestReadUploadWithSizeLimit:
    method test_file_under_limit_succeeds (line 67) | async def test_file_under_limit_succeeds(self):
    method test_file_exactly_at_limit_succeeds (line 72) | async def test_file_exactly_at_limit_succeeds(self):
    method test_file_over_limit_raises (line 77) | async def test_file_over_limit_raises(self):
    method test_file_over_limit_error_has_413_status (line 82) | async def test_file_over_limit_error_has_413_status(self):
    method test_precheck_rejects_when_size_known (line 88) | async def test_precheck_rejects_when_size_known(self):
    method test_precheck_not_triggered_when_size_none (line 95) | async def test_precheck_not_triggered_when_size_none(self):
    method test_large_file_chunked_read (line 102) | async def test_large_file_chunked_read(self):
    method test_empty_file_succeeds (line 110) | async def test_empty_file_succeeds(self):
    method test_error_message_includes_sizes (line 114) | async def test_error_message_includes_sizes(self):
    method test_no_size_attr_under_limit (line 121) | async def test_no_size_attr_under_limit(self):
    method test_no_size_attr_over_limit (line 127) | async def test_no_size_attr_over_limit(self):
    method test_no_size_attr_large_file_chunked_read (line 133) | async def test_no_size_attr_large_file_chunked_read(self):
    method test_no_chunked_read_support_under_limit (line 142) | async def test_no_chunked_read_support_under_limit(self):
    method test_no_chunked_read_support_over_limit (line 148) | async def test_no_chunked_read_support_over_limit(self):
  class TestPreReadUploadFile (line 155) | class TestPreReadUploadFile:
    method test_read_returns_content (line 158) | async def test_read_returns_content(self):
    method test_attributes_preserved (line 162) | async def test_attributes_preserved(self):
    method test_is_upload_file_subclass (line 168) | async def test_is_upload_file_subclass(self):

FILE: tests/unit/fixtures.py
  function sqlite_kvstore (line 15) | async def sqlite_kvstore(tmp_path):
  function disk_dist_registry (line 24) | async def disk_dist_registry(sqlite_kvstore):
  function cached_disk_dist_registry (line 31) | async def cached_disk_dist_registry(sqlite_kvstore):

FILE: tests/unit/prompts/prompts/conftest.py
  function temp_prompt_store (line 25) | async def temp_prompt_store(tmp_path_factory):

FILE: tests/unit/prompts/prompts/test_prompts.py
  class TestPrompts (line 20) | class TestPrompts:
    method test_create_and_get_prompt (line 21) | async def test_create_and_get_prompt(self, temp_prompt_store):
    method test_update_prompt (line 32) | async def test_update_prompt(self, temp_prompt_store):
    method test_update_prompt_with_version (line 40) | async def test_update_prompt_with_version(self, temp_prompt_store):
    method test_delete_prompt (line 66) | async def test_delete_prompt(self, temp_prompt_store):
    method test_list_prompts (line 72) | async def test_list_prompts(self, temp_prompt_store):
    method test_version (line 82) | async def test_version(self, temp_prompt_store):
    method test_set_default_version (line 92) | async def test_set_default_version(self, temp_prompt_store):
    method test_prompt_id_generation_and_validation (line 110) | async def test_prompt_id_generation_and_validation(self, temp_prompt_s...
    method test_list_shows_default_versions (line 118) | async def test_list_shows_default_versions(self, temp_prompt_store):
    method test_get_all_prompt_versions (line 136) | async def test_get_all_prompt_versions(self, temp_prompt_store):
    method test_prompt_variable_validation (line 157) | async def test_prompt_variable_validation(self, temp_prompt_store):
    method test_update_prompt_set_as_default_behavior (line 171) | async def test_update_prompt_set_as_default_behavior(self, temp_prompt...

FILE: tests/unit/providers/agents/builtin/fixtures/__init__.py
  function load_chat_completion_fixture (line 18) | def load_chat_completion_fixture(filename: str) -> OpenAIChatCompletion:

FILE: tests/unit/providers/agents/builtin/test_openai_responses.py
  function mock_inference_api (line 90) | def mock_inference_api():
  function mock_tool_groups_api (line 96) | def mock_tool_groups_api():
  function mock_tool_runtime_api (line 102) | def mock_tool_runtime_api():
  function mock_responses_store (line 108) | def mock_responses_store():
  function mock_vector_io_api (line 114) | def mock_vector_io_api():
  function mock_conversations_api (line 120) | def mock_conversations_api():
  function mock_safety_api (line 127) | def mock_safety_api():
  function mock_prompts_api (line 133) | def mock_prompts_api():
  function mock_files_api (line 139) | def mock_files_api():
  function mock_connectors_api (line 146) | def mock_connectors_api():
  function openai_responses_impl (line 152) | def openai_responses_impl(
  function fake_stream (line 178) | async def fake_stream(fixture: str = "simple_chat_completion.yaml"):
  function test_create_openai_response_with_string_input (line 209) | async def test_create_openai_response_with_string_input(openai_responses...
  function test_failed_stream_persists_non_system_messages (line 276) | async def test_failed_stream_persists_non_system_messages(openai_respons...
  function test_failed_stream_raises_internal_server_error_in_non_streaming_mode (line 331) | async def test_failed_stream_raises_internal_server_error_in_non_streami...
  function test_create_openai_response_with_string_input_with_tools (line 384) | async def test_create_openai_response_with_string_input_with_tools(opena...
  function test_create_openai_response_with_tool_call_type_none (line 455) | async def test_create_openai_response_with_tool_call_type_none(openai_re...
  function test_create_openai_response_with_tool_call_function_arguments_none (line 540) | async def test_create_openai_response_with_tool_call_function_arguments_...
  function test_create_openai_response_with_multiple_messages (line 661) | async def test_create_openai_response_with_multiple_messages(openai_resp...
  function test_prepend_previous_response_basic (line 708) | async def test_prepend_previous_response_basic(openai_responses_impl, mo...
  function test_prepend_previous_response_web_search (line 749) | async def test_prepend_previous_response_web_search(openai_responses_imp...
  function test_prepend_previous_response_mcp_tool_call (line 796) | async def test_prepend_previous_response_mcp_tool_call(openai_responses_...
  function test_create_openai_response_with_instructions (line 845) | async def test_create_openai_response_with_instructions(openai_responses...
  function test_create_openai_response_with_instructions_and_multiple_messages (line 874) | async def test_create_openai_response_with_instructions_and_multiple_mes...
  function test_create_openai_response_with_instructions_and_previous_response (line 919) | async def test_create_openai_response_with_instructions_and_previous_res...
  function test_create_openai_response_with_previous_response_instructions (line 981) | async def test_create_openai_response_with_previous_response_instructions(
  function test_list_openai_response_input_items_delegation (line 1045) | async def test_list_openai_response_input_items_delegation(openai_respon...
  function test_responses_store_list_input_items_logic (line 1080) | async def test_responses_store_list_input_items_logic():
  function test_store_response_uses_rehydrated_input_with_previous_response (line 1164) | async def test_store_response_uses_rehydrated_input_with_previous_response(
  function test_reuse_mcp_tool_list (line 1236) | async def test_reuse_mcp_tool_list(
  function test_create_openai_response_with_text_format (line 1302) | async def test_create_openai_response_with_text_format(
  function test_create_openai_response_with_invalid_text_format (line 1326) | async def test_create_openai_response_with_invalid_text_format(openai_re...
  function test_create_openai_response_with_output_types_as_input (line 1341) | async def test_create_openai_response_with_output_types_as_input(
  function test_create_openai_response_with_prompt (line 1414) | async def test_create_openai_response_with_prompt(openai_responses_impl,...
  function test_prepend_prompt_successful_without_variables (line 1470) | async def test_prepend_prompt_successful_without_variables(openai_respon...
  function test_prepend_prompt_invalid_variable (line 1503) | async def test_prepend_prompt_invalid_variable(openai_responses_impl, mo...
  function test_prepend_prompt_not_found (line 1540) | async def test_prepend_prompt_not_found(openai_responses_impl, mock_prom...
  function test_prepend_prompt_variable_substitution (line 1565) | async def test_prepend_prompt_variable_substitution(openai_responses_imp...
  function test_prepend_prompt_with_image_variable (line 1604) | async def test_prepend_prompt_with_image_variable(openai_responses_impl,...
  function test_prepend_prompt_with_file_variable (line 1668) | async def test_prepend_prompt_with_file_variable(openai_responses_impl, ...
  function test_prepend_prompt_with_mixed_variables (line 1733) | async def test_prepend_prompt_with_mixed_variables(openai_responses_impl...
  function test_prepend_prompt_with_image_using_image_url (line 1828) | async def test_prepend_prompt_with_image_using_image_url(openai_response...
  function test_prepend_prompt_image_variable_missing_required_fields (line 1878) | async def test_prepend_prompt_image_variable_missing_required_fields(ope...
  function test_mcp_tool_connector_id_resolved_to_server_url (line 1905) | async def test_mcp_tool_connector_id_resolved_to_server_url(
  function test_file_search_results_include_chunk_metadata_attributes (line 1951) | async def test_file_search_results_include_chunk_metadata_attributes(moc...
  function test_create_openai_response_with_max_output_tokens_non_streaming (line 2036) | async def test_create_openai_response_with_max_output_tokens_non_streaming(
  function test_create_openai_response_with_max_output_tokens_streaming (line 2073) | async def test_create_openai_response_with_max_output_tokens_streaming(
  function test_create_openai_response_with_max_output_tokens_boundary_value (line 2118) | async def test_create_openai_response_with_max_output_tokens_boundary_va...
  function test_create_openai_response_with_max_output_tokens_and_tools (line 2144) | async def test_create_openai_response_with_max_output_tokens_and_tools(o...
  function test_params_passed_through_full_chain_to_backend_service (line 2231) | async def test_params_passed_through_full_chain_to_backend_service(
  function test_function_tool_strict_field_excluded_when_none (line 2338) | async def test_function_tool_strict_field_excluded_when_none(openai_resp...
  function test_function_tool_strict_field_included_when_set (line 2383) | async def test_function_tool_strict_field_included_when_set(openai_respo...
  function test_function_tool_strict_false_included (line 2425) | async def test_function_tool_strict_false_included(openai_responses_impl...
  function test_create_openai_response_with_truncation_disabled_streaming (line 2459) | async def test_create_openai_response_with_truncation_disabled_streaming(
  function test_create_openai_response_with_truncation_auto_streaming (line 2499) | async def test_create_openai_response_with_truncation_auto_streaming(
  function test_create_openai_response_with_prompt_cache_key_and_previous_response (line 2544) | async def test_create_openai_response_with_prompt_cache_key_and_previous...
  function test_create_openai_response_with_service_tier (line 2589) | async def test_create_openai_response_with_service_tier(openai_responses...
  function test_create_openai_response_service_tier_auto_transformation (line 2617) | async def test_create_openai_response_service_tier_auto_transformation(o...
  function test_create_openai_response_service_tier_propagation_streaming (line 2661) | async def test_create_openai_response_service_tier_propagation_streaming...
  function test_response_object_incomplete_details_null_when_completed (line 2721) | def test_response_object_incomplete_details_null_when_completed():
  function test_response_object_incomplete_details_with_max_output_tokens_reason (line 2742) | def test_response_object_incomplete_details_with_max_output_tokens_reaso...
  function test_response_object_incomplete_details_with_length_reason (line 2765) | def test_response_object_incomplete_details_with_length_reason():
  function test_response_object_incomplete_details_with_max_iterations_exceeded_reason (line 2788) | def test_response_object_incomplete_details_with_max_iterations_exceeded...
  function test_agent_loop_incomplete_due_to_max_output_tokens (line 2811) | async def test_agent_loop_incomplete_due_to_max_output_tokens(
  function test_agent_loop_incomplete_due_to_max_iterations (line 2893) | async def test_agent_loop_incomplete_due_to_max_iterations(
  function test_agent_loop_incomplete_due_to_length_finish_reason (line 2966) | async def test_agent_loop_incomplete_due_to_length_finish_reason(openai_...
  function test_create_openai_response_with_top_logprobs_boundary_values (line 3024) | async def test_create_openai_response_with_top_logprobs_boundary_values(
  function test_create_openai_response_with_frequency_penalty_default (line 3054) | async def test_create_openai_response_with_frequency_penalty_default(ope...
  function test_create_openai_response_with_presence_penalty_default (line 3078) | async def test_create_openai_response_with_presence_penalty_default(open...
  function test_hallucinated_tool_call_does_not_cause_500 (line 3103) | async def test_hallucinated_tool_call_does_not_cause_500(openai_response...
  function test_create_openai_response_with_stream_options_merges_with_default (line 3170) | async def test_create_openai_response_with_stream_options_merges_with_de...
  function test_create_openai_response_with_empty_stream_options (line 3201) | async def test_create_openai_response_with_empty_stream_options(openai_r...

FILE: tests/unit/providers/agents/builtin/test_openai_responses_conversations.py
  function responses_impl_with_conversations (line 31) | def responses_impl_with_conversations(
  class TestConversationValidation (line 58) | class TestConversationValidation:
    method test_nonexistent_conversation_raises_error (line 61) | async def test_nonexistent_conversation_raises_error(
  class TestMessageSyncing (line 76) | class TestMessageSyncing:
    method test_sync_response_to_conversation_simple (line 79) | async def test_sync_response_to_conversation_simple(
    method test_sync_response_to_conversation_api_error (line 122) | async def test_sync_response_to_conversation_api_error(
    method test_sync_with_list_input (line 134) | async def test_sync_with_list_input(self, responses_impl_with_conversa...
  class TestIntegrationWorkflow (line 161) | class TestIntegrationWorkflow:
    method test_create_response_with_valid_conversation (line 164) | async def test_create_response_with_valid_conversation(
    method test_create_response_with_invalid_conversation_id (line 225) | async def test_create_response_with_invalid_conversation_id(self, resp...
    method test_create_response_with_nonexistent_conversation (line 234) | async def test_create_response_with_nonexistent_conversation(
    method test_conversation_and_previous_response_id (line 248) | async def test_conversation_and_previous_response_id(

FILE: tests/unit/providers/agents/builtin/test_response_conversion_utils.py
  function mock_files_api (line 55) | def mock_files_api():
  class TestConvertChatChoiceToResponseMessage (line 60) | class TestConvertChatChoiceToResponseMessage:
    method test_convert_string_content (line 61) | async def test_convert_string_content(self):
    method test_convert_none_content (line 76) | async def test_convert_none_content(self):
  class TestConvertResponseContentToChatContent (line 94) | class TestConvertResponseContentToChatContent:
    method test_convert_string_content (line 95) | async def test_convert_string_content(self, mock_files_api):
    method test_convert_text_content_parts (line 99) | async def test_convert_text_content_parts(self, mock_files_api):
    method test_convert_image_content (line 113) | async def test_convert_image_content(self, mock_files_api):
    method test_convert_image_content_with_file_id_calls_retrieve_with_request_objects (line 123) | async def test_convert_image_content_with_file_id_calls_retrieve_with_...
    method test_convert_file_content_with_file_id_calls_retrieve_with_request_objects (line 135) | async def test_convert_file_content_with_file_id_calls_retrieve_with_r...
  class TestConvertResponseInputToChatMessages (line 148) | class TestConvertResponseInputToChatMessages:
    method test_convert_string_input (line 149) | async def test_convert_string_input(self):
    method test_convert_function_tool_call_output (line 156) | async def test_convert_function_tool_call_output(self):
    method test_convert_function_tool_call_output_with_list_content (line 180) | async def test_convert_function_tool_call_output_with_list_content(self):
    method test_convert_function_tool_call_output_with_multi_block_list_content (line 202) | async def test_convert_function_tool_call_output_with_multi_block_list...
    method test_convert_function_tool_call_output_with_image_content (line 225) | async def test_convert_function_tool_call_output_with_image_content(se...
    method test_convert_function_tool_call_output_with_mixed_text_and_image (line 254) | async def test_convert_function_tool_call_output_with_mixed_text_and_i...
    method test_convert_function_tool_call (line 279) | async def test_convert_function_tool_call(self):
    method test_convert_function_call_ordering (line 297) | async def test_convert_function_call_ordering(self):
    method test_convert_response_message (line 338) | async def test_convert_response_message(self):
  class TestConvertResponseTextToChatResponseFormat (line 355) | class TestConvertResponseTextToChatResponseFormat:
    method test_convert_text_format (line 356) | async def test_convert_text_format(self):
    method test_convert_json_object_format (line 363) | async def test_convert_json_object_format(self):
    method test_convert_json_schema_format (line 369) | async def test_convert_json_schema_format(self):
    method test_default_text_format (line 384) | async def test_default_text_format(self):
  class TestGetMessageTypeByRole (line 392) | class TestGetMessageTypeByRole:
    method test_user_role (line 393) | async def test_user_role(self):
    method test_system_role (line 397) | async def test_system_role(self):
    method test_assistant_role (line 401) | async def test_assistant_role(self):
    method test_developer_role (line 405) | async def test_developer_role(self):
    method test_unknown_role (line 409) | async def test_unknown_role(self):
  class TestIsFunctionToolCall (line 414) | class TestIsFunctionToolCall:
    method test_is_function_tool_call_true (line 415) | def test_is_function_tool_call_true(self):
    method test_is_function_tool_call_false_different_name (line 434) | def test_is_function_tool_call_false_different_name(self):
    method test_is_function_tool_call_false_no_function (line 452) | def test_is_function_tool_call_false_no_function(self):
    method test_is_function_tool_call_false_wrong_type (line 467) | def test_is_function_tool_call_false_wrong_type(self):
  class TestExtractCitationsFromText (line 484) | class TestExtractCitationsFromText:
    method test_extract_citations_and_annotations (line 485) | def test_extract_citations_and_annotations(self):

FILE: tests/unit/providers/agents/builtin/test_response_tool_context.py
  class TestToolContext (line 34) | class TestToolContext:
    method test_no_tools (line 35) | def test_no_tools(self):
    method test_no_previous_tools (line 47) | def test_no_previous_tools(self):
    method test_reusable_server (line 62) | def test_reusable_server(self):
    method test_multiple_reusable_servers (line 91) | def test_multiple_reusable_servers(self):
    method test_multiple_servers_only_one_reusable (line 134) | def test_multiple_servers_only_one_reusable(self):
    method test_mismatched_allowed_tools (line 170) | def test_mismatched_allowed_tools(self):
  class TestProcessToolChoice (line 211) | class TestProcessToolChoice:
    method setup_method (line 214) | def setup_method(self):
    method test_mode_auto (line 227) | async def test_mode_auto(self):
    method test_mode_none (line 233) | async def test_mode_none(self):
    method test_mode_required_with_tools (line 239) | async def test_mode_required_with_tools(self):
    method test_mode_required_without_tools (line 253) | async def test_mode_required_without_tools(self):
    method test_allowed_tools_function (line 259) | async def test_allowed_tools_function(self):
    method test_allowed_tools_custom (line 276) | async def test_allowed_tools_custom(self):
    method test_allowed_tools_file_search (line 291) | async def test_allowed_tools_file_search(self):
    method test_allowed_tools_web_search (line 303) | async def test_allowed_tools_web_search(self):
    method test_allowed_tools_mcp_server_label (line 324) | async def test_allowed_tools_mcp_server_label(self):
    method test_allowed_tools_mixed_types (line 342) | async def test_allowed_tools_mixed_types(self):
    method test_allowed_tools_invalid_type (line 365) | async def test_allowed_tools_invalid_type(self):
    method test_specific_custom_tool_valid (line 381) | async def test_specific_custom_tool_valid(self):
    method test_specific_custom_tool_invalid (line 390) | async def test_specific_custom_tool_invalid(self):
    method test_specific_function_tool_valid (line 396) | async def test_specific_function_tool_valid(self):
    method test_specific_function_tool_invalid (line 404) | async def test_specific_function_tool_invalid(self):
    method test_specific_file_search_valid (line 410) | async def test_specific_file_search_valid(self):
    method test_specific_file_search_invalid (line 418) | async def test_specific_file_search_invalid(self):
    method test_specific_web_search_valid (line 425) | async def test_specific_web_search_valid(self):
    method test_specific_web_search_invalid (line 433) | async def test_specific_web_search_invalid(self):
    method test_specific_mcp_tool_with_name (line 440) | async def test_specific_mcp_tool_with_name(self):
    method test_specific_mcp_tool_with_name_not_in_chat_tools (line 452) | async def test_specific_mcp_tool_with_name_not_in_chat_tools(self):
    method test_specific_mcp_tool_server_label_only (line 462) | async def test_specific_mcp_tool_server_label_only(self):
    method test_specific_mcp_tool_unknown_server (line 478) | async def test_specific_mcp_tool_unknown_server(self):
    method test_empty_chat_tools (line 488) | async def test_empty_chat_tools(self):
    method test_empty_server_label_to_tools (line 494) | async def test_empty_server_label_to_tools(self):
    method test_allowed_tools_empty_list (line 501) | async def test_allowed_tools_empty_list(self):
    method test_mcp_tool_multiple_servers (line 509) | async def test_mcp_tool_multiple_servers(self):
    method test_allowed_tools_filters_effective_tools_correctly (line 533) | async def test_allowed_tools_filters_effective_tools_correctly(self):

FILE: tests/unit/providers/agents/builtin/test_responses_background.py
  class TestBackgroundFieldInResponseObject (line 27) | class TestBackgroundFieldInResponseObject:
    method test_background_field_default_is_none (line 30) | def test_background_field_default_is_none(self):
    method test_background_field_can_be_true (line 42) | def test_background_field_can_be_true(self):
    method test_background_field_can_be_false (line 55) | def test_background_field_can_be_false(self):
  class TestResponseStatus (line 69) | class TestResponseStatus:
    method test_valid_status_values (line 76) | def test_valid_status_values(self, status):
    method test_queued_status_with_background (line 89) | def test_queued_status_with_background(self):
  class TestResponseObjectSerialization (line 104) | class TestResponseObjectSerialization:
    method test_model_dump_includes_background (line 107) | def test_model_dump_includes_background(self):
    method test_model_dump_json_includes_background (line 122) | def test_model_dump_json_includes_background(self):
  class TestResponseErrorForBackground (line 137) | class TestResponseErrorForBackground:
    method test_error_response_with_background (line 140) | def test_error_response_with_background(self):
  function _make_responses_impl (line 162) | def _make_responses_impl():
  class _CollectingExporter (line 178) | class _CollectingExporter(SpanExporter):
    method __init__ (line 181) | def __init__(self):
    method export (line 184) | def export(self, spans):
  class TestResponsesOtelContextPropagation (line 189) | class TestResponsesOtelContextPropagation:
    method test_worker_attributes_work_to_correct_request_trace (line 192) | async def test_worker_attributes_work_to_correct_request_trace(self):
    method test_worker_does_not_leak_context_between_items (line 237) | async def test_worker_does_not_leak_context_between_items(self):
    method test_error_handling_runs_under_request_context (line 285) | async def test_error_handling_runs_under_request_context(self):
  function _set_authenticated_user (line 340) | def _set_authenticated_user(user: User | None):
  class TestResponsesProviderDataPropagation (line 348) | class TestResponsesProviderDataPropagation:
    method test_worker_runs_under_correct_user_identity (line 356) | async def test_worker_runs_under_correct_user_identity(self):
    method test_worker_does_not_leak_identity_between_items (line 395) | async def test_worker_does_not_leak_identity_between_items(self):
    method test_error_handler_runs_under_correct_identity (line 432) | async def test_error_handler_runs_under_correct_identity(self):

FILE: tests/unit/providers/agents/builtin/test_responses_safety_utils.py
  function mock_apis (line 23) | def mock_apis():
  function responses_impl (line 40) | def responses_impl(mock_apis):
  function test_extract_guardrail_ids_from_strings (line 45) | def test_extract_guardrail_ids_from_strings(responses_impl):
  function test_extract_guardrail_ids_from_objects (line 52) | def test_extract_guardrail_ids_from_objects(responses_impl):
  function test_extract_guardrail_ids_mixed_formats (line 62) | def test_extract_guardrail_ids_mixed_formats(responses_impl):
  function test_extract_guardrail_ids_none_input (line 73) | def test_extract_guardrail_ids_none_input(responses_impl):
  function test_extract_guardrail_ids_empty_list (line 79) | def test_extract_guardrail_ids_empty_list(responses_impl):
  function test_extract_guardrail_ids_unknown_format (line 85) | def test_extract_guardrail_ids_unknown_format(responses_impl):
  function mock_safety_api (line 95) | def mock_safety_api():
  function test_run_guardrails_no_violation (line 107) | async def test_run_guardrails_no_violation(mock_safety_api):
  function test_run_guardrails_with_violation (line 128) | async def test_run_guardrails_with_violation(mock_safety_api):
  function test_run_guardrails_empty_inputs (line 148) | async def test_run_guardrails_empty_inputs(mock_safety_api):

FILE: tests/unit/providers/agents/builtin/test_safety_optional.py
  function mock_persistence_config (line 32) | def mock_persistence_config():
  function mock_deps (line 47) | def mock_deps():
  class TestProviderInitialization (line 71) | class TestProviderInitialization:
    method test_initialization_with_safety_api_present (line 74) | async def test_initialization_with_safety_api_present(self, mock_persi...
    method test_initialization_without_safety_api (line 91) | async def test_initialization_without_safety_api(self, mock_persistenc...
  class TestGuardrailsFunctionality (line 107) | class TestGuardrailsFunctionality:
    method test_run_guardrails_with_none_safety_api (line 110) | async def test_run_guardrails_with_none_safety_api(self):
    method test_run_guardrails_with_empty_messages (line 115) | async def test_run_guardrails_with_empty_messages(self):
    method test_run_guardrails_with_none_safety_api_ignores_guardrails (line 126) | async def test_run_guardrails_with_none_safety_api_ignores_guardrails(...
    method test_create_response_rejects_guardrails_without_safety_api (line 136) | async def test_create_response_rejects_guardrails_without_safety_api(s...
    method test_create_response_succeeds_without_guardrails_and_no_safety_api (line 178) | async def test_create_response_succeeds_without_guardrails_and_no_safe...

FILE: tests/unit/providers/agents/builtin/test_streaming_errors.py
  function _make_mock_exc (line 16) | def _make_mock_exc(body):
  class TestExtractOpenaiError (line 24) | class TestExtractOpenaiError:
    method test_nested_format_extracts_correctly (line 35) | def test_nested_format_extracts_correctly(self):
    method test_direct_format_extracts_correctly (line 43) | def test_direct_format_extracts_correctly(self):
    method test_maps_invalid_base64_to_responses_api_code (line 51) | def test_maps_invalid_base64_to_responses_api_code(self):
    method test_direct_format_maps_codes (line 59) | def test_direct_format_maps_codes(self):
    method test_missing_message_uses_str_exc (line 67) | def test_missing_message_uses_str_exc(self):
    method test_nested_error_not_dict_falls_back_to_direct (line 75) | def test_nested_error_not_dict_falls_back_to_direct(self):
    method test_missing_code_returns_server_error_preserves_message (line 83) | def test_missing_code_returns_server_error_preserves_message(self):
    method test_none_body_returns_server_error (line 91) | def test_none_body_returns_server_error(self):
    method test_non_dict_body_returns_server_error (line 98) | def test_non_dict_body_returns_server_error(self):
    method test_no_code_preserves_message (line 104) | def test_no_code_preserves_message(self):
    method test_empty_string_code_returns_server_error (line 112) | def test_empty_string_code_returns_server_error(self):
    method test_unknown_provider_code_falls_back_to_server_error (line 120) | def test_unknown_provider_code_falls_back_to_server_error(self):
    method test_valid_codes_pass_through (line 128) | def test_valid_codes_pass_through(self):
    method test_all_spec_codes_are_valid (line 137) | def test_all_spec_codes_are_valid(self):

FILE: tests/unit/providers/batches/conftest.py
  function provider (line 22) | async def provider():
  function sample_batch_data (line 49) | def sample_batch_data():

FILE: tests/unit/providers/batches/test_reference.py
  class TestReferenceBatchesImpl (line 72) | class TestReferenceBatchesImpl:
    method _validate_batch_type (line 75) | def _validate_batch_type(self, batch, expected_metadata=None):
    method test_create_and_retrieve_batch_success (line 177) | async def test_create_and_retrieve_batch_success(self, provider, sampl...
    method test_create_batch_without_metadata (line 204) | async def test_create_batch_without_metadata(self, provider):
    method test_create_batch_completion_window (line 212) | async def test_create_batch_completion_window(self, provider):
    method test_create_batch_invalid_endpoints (line 224) | async def test_create_batch_invalid_endpoints(self, provider, endpoint):
    method test_create_batch_invalid_metadata (line 231) | async def test_create_batch_invalid_metadata(self, provider):
    method test_retrieve_batch_not_found (line 253) | async def test_retrieve_batch_not_found(self, provider):
    method test_cancel_batch_success (line 258) | async def test_cancel_batch_success(self, provider, sample_batch_data):
    method test_cancel_batch_invalid_statuses (line 271) | async def test_cancel_batch_invalid_statuses(self, provider, sample_ba...
    method test_cancel_batch_not_found (line 282) | async def test_cancel_batch_not_found(self, provider):
    method test_list_batches_empty (line 287) | async def test_list_batches_empty(self, provider):
    method test_list_batches_single_batch (line 297) | async def test_list_batches_single_batch(self, provider, sample_batch_...
    method test_list_batches_multiple_batches (line 310) | async def test_list_batches_multiple_batches(self, provider):
    method test_list_batches_with_limit (line 331) | async def test_list_batches_with_limit(self, provider):
    method test_list_batches_with_pagination (line 350) | async def test_list_batches_with_pagination(self, provider):
    method test_list_batches_invalid_after (line 372) | async def test_list_batches_invalid_after(self, provider, sample_batch...
    method test_kvstore_persistence (line 381) | async def test_kvstore_persistence(self, provider, sample_batch_data):
    method test_validate_input_file_not_found (line 392) | async def test_validate_input_file_not_found(self, provider):
    method test_validate_input_file_exists_empty_content (line 415) | async def test_validate_input_file_exists_empty_content(self, provider):
    method test_validate_input_file_mixed_valid_invalid_json (line 437) | async def test_validate_input_file_mixed_valid_invalid_json(self, prov...
    method test_validate_input_invalid_model (line 471) | async def test_validate_input_invalid_model(self, provider):
    method test_validate_input_missing_parameters_chat_completions (line 511) | async def test_validate_input_missing_parameters_chat_completions(
    method test_validate_input_missing_parameters_completions (line 566) | async def test_validate_input_missing_parameters_completions(
    method test_validate_input_url_mismatch (line 610) | async def test_validate_input_url_mismatch(self, provider):
    method test_validate_input_multiple_errors_per_request (line 637) | async def test_validate_input_multiple_errors_per_request(self, provid...
    method test_validate_input_invalid_request_format (line 669) | async def test_validate_input_invalid_request_format(self, provider):
    method test_validate_input_invalid_parameter_types (line 706) | async def test_validate_input_invalid_parameter_types(
    method test_max_concurrent_batches (line 750) | async def test_max_concurrent_batches(self, provider):
    method test_create_batch_embeddings_endpoint (line 778) | async def test_create_batch_embeddings_endpoint(self, provider):

FILE: tests/unit/providers/batches/test_reference_idempotency.py
  class TestReferenceBatchesIdempotency (line 51) | class TestReferenceBatchesIdempotency:
    method test_idempotent_batch_creation_same_params (line 54) | async def test_idempotent_batch_creation_same_params(self, provider, s...
    method test_different_idempotency_keys_create_different_batches (line 83) | async def test_different_idempotency_keys_create_different_batches(sel...
    method test_non_idempotent_behavior_without_key (line 91) | async def test_non_idempotent_behavior_without_key(self, provider, sam...
    method test_same_idempotency_key_different_params_conflict (line 111) | async def test_same_idempotency_key_different_params_conflict(

FILE: tests/unit/providers/files/conftest.py
  class MockUploadFile (line 16) | class MockUploadFile:
    method __init__ (line 17) | def __init__(self, content: bytes, filename: str, content_type: str = ...
    method read (line 22) | async def read(self):
  function sample_text_file (line 27) | def sample_text_file():
  function sample_text_file2 (line 33) | def sample_text_file2():
  function s3_config (line 39) | def s3_config(tmp_path):
  function s3_client (line 53) | def s3_client():
  function s3_provider (line 62) | async def s3_provider(s3_config, s3_client):  # s3_client provides the m...

FILE: tests/unit/providers/files/test_s3_files.py
  class TestS3FilesImpl (line 23) | class TestS3FilesImpl:
    method test_upload_file (line 26) | async def test_upload_file(self, s3_provider, sample_text_file, s3_cli...
    method test_list_files_empty (line 43) | async def test_list_files_empty(self, s3_provider):
    method test_retrieve_file (line 52) | async def test_retrieve_file(self, s3_provider, sample_text_file):
    method test_retrieve_file_content (line 67) | async def test_retrieve_file_content(self, s3_provider, sample_text_fi...
    method test_delete_file (line 82) | async def test_delete_file(self, s3_provider, sample_text_file, s3_con...
    method test_list_files (line 103) | async def test_list_files(self, s3_provider, sample_text_file, sample_...
    method test_list_files_with_purpose_filter (line 124) | async def test_list_files_with_purpose_filter(self, s3_provider, sampl...
    method test_nonexistent_file_retrieval (line 144) | async def test_nonexistent_file_retrieval(self, s3_provider):
    method test_nonexistent_file_content_retrieval (line 149) | async def test_nonexistent_file_content_retrieval(self, s3_provider):
    method test_nonexistent_file_deletion (line 156) | async def test_nonexistent_file_deletion(self, s3_provider):
    method test_upload_file_without_filename (line 161) | async def test_upload_file_without_filename(self, s3_provider, sample_...
    method test_file_operations_when_s3_object_deleted (line 175) | async def test_file_operations_when_s3_object_deleted(self, s3_provide...
    method test_upload_file_s3_put_object_failure (line 193) | async def test_upload_file_s3_put_object_failure(self, s3_provider, sa...
    method test_default_no_expiration (line 213) | async def test_default_no_expiration(self, s3_provider, sample_text_fi...
    method test_default_batch_expiration (line 222) | async def test_default_batch_expiration(self, s3_provider, sample_text...
    method test_expired_file_is_unavailable (line 235) | async def test_expired_file_is_unavailable(self, s3_provider, sample_t...
    method test_unsupported_expires_after_anchor (line 274) | async def test_unsupported_expires_after_anchor(self, s3_provider, sam...
    method test_nonint_expires_after_seconds (line 287) | async def test_nonint_expires_after_seconds(self, s3_provider, sample_...
    method test_expires_after_seconds_out_of_bounds (line 300) | async def test_expires_after_seconds_out_of_bounds(self, s3_provider, ...

FILE: tests/unit/providers/files/test_s3_files_auth.py
  function test_listing_hides_other_users_file (line 24) | async def test_listing_hides_other_users_file(s3_provider, sample_text_f...
  function _make_request_for_op (line 42) | def _make_request_for_op(op_name: str, file_id: str):
  function test_cannot_access_other_user_file (line 62) | async def test_cannot_access_other_user_file(s3_provider, sample_text_fi...
  function test_shared_role_allows_listing (line 83) | async def test_shared_role_allows_listing(s3_provider, sample_text_file):
  function test_shared_role_allows_access (line 110) | async def test_shared_role_allows_access(s3_provider, sample_text_file, ...

FILE: tests/unit/providers/inference/bedrock/test_config.py
  class TestBedrockBaseConfig (line 13) | class TestBedrockBaseConfig:
    method test_defaults_work_without_env_vars (line 14) | def test_defaults_work_without_env_vars(self):
    method test_env_vars_get_picked_up (line 28) | def test_env_vars_get_picked_up(self):
    method test_partial_env_setup (line 48) | def test_partial_env_setup(self):
    method test_bad_max_attempts_breaks (line 57) | def test_bad_max_attempts_breaks(self):

FILE: tests/unit/providers/inference/test_bedrock_adapter.py
  function test_adapter_initialization (line 19) | def test_adapter_initialization():
  function test_client_url_construction (line 27) | def test_client_url_construction():
  function test_api_key_from_config (line 34) | def test_api_key_from_config():
  function test_api_key_from_header_overrides_config (line 40) | def test_api_key_from_header_overrides_config():
  function test_authentication_error_handling (line 53) | async def test_authentication_error_handling():

FILE: tests/unit/providers/inference/test_bedrock_config.py
  function test_bedrock_config_defaults_no_env (line 10) | def test_bedrock_config_defaults_no_env(monkeypatch):
  function test_bedrock_config_reads_from_env (line 19) | def test_bedrock_config_reads_from_env(monkeypatch):
  function test_bedrock_config_with_values (line 26) | def test_bedrock_config_with_values():
  function test_bedrock_config_sample (line 33) | def test_bedrock_config_sample():

FILE: tests/unit/providers/inference/test_inference_client_caching.py
  function test_openai_provider_data_used (line 128) | def test_openai_provider_data_used(config_cls, adapter_cls, provider_dat...
  function test_watsonx_provider_data_used (line 152) | def test_watsonx_provider_data_used(config_cls, adapter_cls, provider_da...

FILE: tests/unit/providers/inference/test_openai_base_url_config.py
  class TestOpenAIBaseURLConfig (line 15) | class TestOpenAIBaseURLConfig:
    method test_default_base_url_without_env_var (line 18) | def test_default_base_url_without_env_var(self):
    method test_custom_base_url_from_config (line 26) | def test_custom_base_url_from_config(self):
    method test_base_url_from_environment_variable (line 36) | def test_base_url_from_environment_variable(self):
    method test_config_overrides_environment_variable (line 48) | def test_config_overrides_environment_variable(self):
    method test_client_uses_configured_base_url (line 59) | def test_client_uses_configured_base_url(self, mock_openai_class):
    method test_check_model_availability_uses_configured_url (line 80) | async def test_check_model_availability_uses_configured_url(self, mock...
    method test_environment_variable_affects_model_availability_check (line 121) | async def test_environment_variable_affects_model_availability_check(s...

FILE: tests/unit/providers/inference/test_remote_vllm.py
  function vllm_inference_adapter (line 41) | async def vllm_inference_adapter():
  function test_health_status_success (line 49) | async def test_health_status_success(vllm_inference_adapter):
  function test_health_status_failure (line 78) | async def test_health_status_failure(vllm_inference_adapter):
  function test_health_status_no_static_api_key (line 99) | async def test_health_status_no_static_api_key(vllm_inference_adapter):
  function test_openai_chat_completion_is_async (line 123) | async def test_openai_chat_completion_is_async(vllm_inference_adapter):
  function test_vllm_completion_extra_body (line 171) | async def test_vllm_completion_extra_body():
  function test_vllm_chat_completion_extra_body (line 244) | async def test_vllm_chat_completion_extra_body():

FILE: tests/unit/providers/inference/vertexai/conftest.py
  function _install_google_shims (line 12) | def _install_google_shims() -> None:

FILE: tests/unit/providers/inference/vertexai/test_adapter.py
  function _async_pager (line 19) | async def _async_pager(items):
  function _make_fake_streaming_chunk (line 25) | def _make_fake_streaming_chunk(text: str = "chunk") -> SimpleNamespace:
  function vertex_config (line 46) | def vertex_config() -> VertexAIConfig:
  function adapter (line 52) | def adapter(vertex_config: VertexAIConfig) -> VertexAIInferenceAdapter:
  function patch_chat_completion_dependencies (line 60) | def patch_chat_completion_dependencies(monkeypatch):
  class TestVertexAIAdapterInit (line 145) | class TestVertexAIAdapterInit:
    method test_init_sets_config_and_default_client (line 146) | def test_init_sets_config_and_default_client(
    method test_initialize_sets_default_client (line 153) | async def test_initialize_sets_default_client(self, monkeypatch, adapt...
    method test_initialize_failure_keeps_default_client_unset (line 163) | async def test_initialize_failure_keeps_default_client_unset(self, mon...
  class TestVertexAIClientManagement (line 177) | class TestVertexAIClientManagement:
    method test_create_client_with_access_token_uses_credentials (line 178) | def test_create_client_with_access_token_uses_credentials(self, monkey...
    method test_get_client_uses_default_client (line 197) | def test_get_client_uses_default_client(self, monkeypatch):
    method test_get_client_uses_provider_override_with_token (line 206) | def test_get_client_uses_provider_override_with_token(self, monkeypatch):
    method test_get_client_project_override_reuses_configured_token (line 228) | def test_get_client_project_override_reuses_configured_token(self, mon...
    method test_get_client_raises_when_no_client_available (line 252) | def test_get_client_raises_when_no_client_available(self, monkeypatch):
  class TestVertexAIModelListing (line 261) | class TestVertexAIModelListing:
    method test_list_provider_model_ids_filters_and_deduplicates (line 262) | async def test_list_provider_model_ids_filters_and_deduplicates(self, ...
    method test_list_models_returns_correct_attributes (line 301) | async def test_list_models_returns_correct_attributes(
    method test_list_models_respects_allowed_models (line 322) | async def test_list_models_respects_allowed_models(self, monkeypatch, ...
    method test_list_models_propagates_errors (line 337) | async def test_list_models_propagates_errors(self, monkeypatch):
    method test_should_refresh_models_returns_config_value (line 349) | async def test_should_refresh_models_returns_config_value(self):
  class TestVertexAIModelAvailability (line 360) | class TestVertexAIModelAvailability:
    method test_check_model_availability (line 369) | async def test_check_model_availability(self, monkeypatch, model, avai...
    method test_cache_hit_avoids_api_call (line 394) | async def test_cache_hit_avoids_api_call(self, monkeypatch, adapter: V...
    method test_cached_unknown_model_returns_false_without_api_call (line 409) | async def test_cached_unknown_model_returns_false_without_api_call(
    method test_list_models_populates_model_cache (line 424) | async def test_list_models_populates_model_cache(self, monkeypatch, ad...
    method test_list_models_clears_cache_on_refresh (line 438) | async def test_list_models_clears_cache_on_refresh(self, monkeypatch, ...
  class TestVertexAIAllowedModelsValidation (line 456) | class TestVertexAIAllowedModelsValidation:
    method test_validate_allowed_model_passes (line 457) | def test_validate_allowed_model_passes(self):
    method test_validate_disallowed_model_raises (line 464) | def test_validate_disallowed_model_raises(self):
    method test_validate_no_allowed_models_passes_anything (line 472) | def test_validate_no_allowed_models_passes_anything(self):
  class TestVertexAIUnsupportedOps (line 478) | class TestVertexAIUnsupportedOps:
    method test_unsupported_operations_raise (line 485) | async def test_unsupported_operations_raise(self, method_name, call_kw...
  class TestOpenAIChatCompletionImagePreDownload (line 493) | class TestOpenAIChatCompletionImagePreDownload:
    method test_http_image_url_is_downloaded (line 494) | async def test_http_image_url_is_downloaded(self, monkeypatch, patch_c...
    method test_data_uri_not_re_downloaded (line 526) | async def test_data_uri_not_re_downloaded(self, monkeypatch, patch_cha...
    method test_failed_download_raises_value_error (line 557) | async def test_failed_download_raises_value_error(self, monkeypatch, p...
  class TestCollectSamplingParams (line 585) | class TestCollectSamplingParams:
    method test_single_sampling_param_forwarded (line 596) | def test_single_sampling_param_forwarded(self, param_name, param_value...
    method test_all_three_params_combined (line 607) | def test_all_three_params_combined(self):
    method test_none_values_are_excluded (line 621) | def test_none_values_are_excluded(self):
    method test_zero_values_are_forwarded (line 632) | def test_zero_values_are_forwarded(self):
    method test_existing_params_unaffected (line 649) | def test_existing_params_unaffected(self):
    method test_logprobs_mapping (line 674) | def test_logprobs_mapping(self, input_param, input_value, expected_key...
    method test_logprobs_none_excluded (line 685) | def test_logprobs_none_excluded(self):
  class TestVertexAINetworkConfig (line 696) | class TestVertexAINetworkConfig:
    method test_build_http_options_none_returns_none (line 699) | def test_build_http_options_none_returns_none(self):
    method test_build_http_options_empty_config_returns_none (line 703) | def test_build_http_options_empty_config_returns_none(self):
    method test_build_http_options_headers_only (line 708) | def test_build_http_options_headers_only(self):
    method test_build_http_options_float_timeout_converts_to_ms (line 714) | def test_build_http_options_float_timeout_converts_to_ms(self):
    method test_build_http_options_timeout_config_read_wins (line 720) | def test_build_http_options_timeout_config_read_wins(self):
    method test_build_http_options_timeout_config_connect_fallback (line 726) | def test_build_http_options_timeout_config_connect_fallback(self):
    method test_build_http_options_timeout_config_both_uses_read (line 733) | def test_build_http_options_timeout_config_both_uses_read(self):
    method test_build_http_options_timeout_config_both_none_returns_none (line 740) | def test_build_http_options_timeout_config_both_none_returns_none(self):
    method test_build_http_options_tls_verify_false_produces_ssl_context (line 745) | def test_build_http_options_tls_verify_false_produces_ssl_context(self):
    method test_build_http_options_tls_verify_true_still_sets_httpx_client (line 759) | def test_build_http_options_tls_verify_true_still_sets_httpx_client(se...
    method test_build_http_options_tls_verify_false_httpx_client_has_follow_redirects (line 767) | def test_build_http_options_tls_verify_false_httpx_client_has_follow_r...
    method test_build_http_options_proxy_url_produces_httpx_client (line 775) | def test_build_http_options_proxy_url_produces_httpx_client(self):
    method test_build_http_options_no_proxy_logs_warning (line 784) | def test_build_http_options_no_proxy_logs_warning(self, caplog):
    method test_build_http_options_combined_all_fields (line 790) | def test_build_http_options_combined_all_fields(self):
  class TestVertexAIClientWithNetworkConfig (line 807) | class TestVertexAIClientWithNetworkConfig:
    method test_create_client_with_token_passes_http_options (line 810) | def test_create_client_with_token_passes_http_options(self, monkeypatch):
    method test_create_adc_client_passes_http_options (line 824) | def test_create_adc_client_passes_http_options(self, monkeypatch):
    method test_create_client_no_network_config_no_http_options (line 838) | def test_create_client_no_network_config_no_http_options(self, monkeyp...
    method test_initialize_builds_http_options_from_config (line 853) | async def test_initialize_builds_http_options_from_config(self, monkey...
    method test_shutdown_closes_managed_httpx_async_client (line 871) | async def test_shutdown_closes_managed_httpx_async_client(self, monkey...
    method test_initialize_closes_existing_managed_httpx_async_client (line 893) | async def test_initialize_closes_existing_managed_httpx_async_client(s...
  class TestBuildThinkingConfig (line 917) | class TestBuildThinkingConfig:
    method test_none_returns_none (line 918) | def test_none_returns_none(self):
    method test_none_string_uses_budget_zero (line 923) | def test_none_string_uses_budget_zero(self):
    method test_effort_maps_to_thinking_level (line 939) | def test_effort_maps_to_thinking_level(self, effort: str, expected_lev...
    method test_model_extra_overrides_thinking_config (line 945) | def test_model_extra_overrides_thinking_config(self):
  class TestStreamChatCompletion (line 971) | class TestStreamChatCompletion:
    method _make_fake_chunk (line 974) | def _make_fake_chunk(
    method test_stream_with_include_usage_emits_final_chunk (line 991) | async def test_stream_with_include_usage_emits_final_chunk(self, monke...
    method test_stream_without_stream_options_no_extra_chunk (line 1029) | async def test_stream_without_stream_options_no_extra_chunk(self, monk...
    method test_stream_with_include_usage_false_no_extra_chunk (line 1060) | async def test_stream_with_include_usage_false_no_extra_chunk(self, mo...
  function make_adapter_with_mock_embed (line 1091) | def make_adapter_with_mock_embed(monkeypatch):
  class TestVertexAIEmbeddings (line 1124) | class TestVertexAIEmbeddings:
    method test_single_string_returns_one_embedding (line 1127) | async def test_single_string_returns_one_embedding(self, make_adapter_...
    method test_batch_strings_returns_multiple_embeddings (line 1140) | async def test_batch_strings_returns_multiple_embeddings(self, make_ad...
    method test_dimensions_forwarded_as_output_dimensionality (line 1154) | async def test_dimensions_forwarded_as_output_dimensionality(self, mak...
    method test_user_forwarded_as_labels (line 1165) | async def test_user_forwarded_as_labels(self, make_adapter_with_mock_e...
    method test_base64_encoding_format (line 1176) | async def test_base64_encoding_format(self, make_adapter_with_mock_emb...
    method test_token_array_input_raises_value_error (line 1196) | async def test_token_array_input_raises_value_error(self, make_adapter...
    method test_usage_returns_zeros (line 1207) | async def test_usage_returns_zeros(self, make_adapter_with_mock_embed):
    method test_no_config_when_no_options (line 1217) | async def test_no_config_when_no_options(self, make_adapter_with_mock_...
    method test_embedding_usage_with_real_tokens (line 1227) | async def test_embedding_usage_with_real_tokens(self, make_adapter_wit...
    method test_embedding_usage_fallback_when_no_metadata (line 1238) | async def test_embedding_usage_fallback_when_no_metadata(self, make_ad...
    method test_embedding_usage_fallback_when_metadata_missing_fields (line 1248) | async def test_embedding_usage_fallback_when_metadata_missing_fields(s...
  class TestDroppedParameterWarnings (line 1260) | class TestDroppedParameterWarnings:
    method test_unsupported_param_logged (line 1279) | async def test_unsupported_param_logged(
    method test_parallel_tool_calls_warning (line 1310) | async def test_parallel_tool_calls_warning(
  class TestTelemetryStreamOptions (line 1330) | class TestTelemetryStreamOptions:
    method _patch_stream_chat_completion (line 1333) | def _patch_stream_chat_completion(self, monkeypatch, adapter: VertexAI...
    method test_stream_options_injected_when_telemetry_active (line 1365) | async def test_stream_options_injected_when_telemetry_active(self, mon...
    method test_stream_options_not_injected_when_telemetry_inactive (line 1392) | async def test_stream_options_not_injected_when_telemetry_inactive(sel...
  class TestEmbeddingsModelExtra (line 1419) | class TestEmbeddingsModelExtra:
    method test_model_extra_debug_log (line 1422) | async def test_model_extra_debug_log(self, make_adapter_with_mock_embe...
    method test_no_model_extra_no_debug_log (line 1445) | async def test_no_model_extra_no_debug_log(self, make_adapter_with_moc...
  class TestDeprecatedFunctionCalling (line 1465) | class TestDeprecatedFunctionCalling:
    method test_functions_converted_to_tools_when_tools_absent (line 1466) | async def test_functions_converted_to_tools_when_tools_absent(
    method test_tools_takes_priority_over_functions (line 1496) | async def test_tools_takes_priority_over_functions(self, monkeypatch, ...
    method test_function_call_converted_to_tool_choice (line 1525) | async def test_function_call_converted_to_tool_choice(
    method test_tool_choice_takes_priority_over_function_call (line 1551) | async def test_tool_choice_takes_priority_over_function_call(

FILE: tests/unit/providers/inference/vertexai/test_adapter_completion.py
  function _async_pager (line 22) | async def _async_pager(items):
  function _make_fake_gemini_response (line 28) | def _make_fake_gemini_response(text: str = "The answer") -> SimpleNamesp...
  function _make_fake_streaming_chunk (line 36) | def _make_fake_streaming_chunk(text: str = "chunk") -> SimpleNamespace:
  function make_completion_adapter (line 45) | def make_completion_adapter(monkeypatch):
  class TestOpenAICompletion (line 63) | class TestOpenAICompletion:
    method test_string_prompt_returns_completion (line 64) | async def test_string_prompt_returns_completion(self, make_completion_...
    method test_list_string_prompt_accepted (line 72) | async def test_list_string_prompt_accepted(self, make_completion_adapt...
    method test_token_array_prompt_raises_value_error (line 80) | async def test_token_array_prompt_raises_value_error(self, make_comple...
    method test_nested_token_array_prompt_raises_value_error (line 87) | async def test_nested_token_array_prompt_raises_value_error(self, make...
    method test_stream_list_prompt_raises_value_error (line 94) | async def test_stream_list_prompt_raises_value_error(self, monkeypatch):
    method test_unmappable_params_logged (line 107) | async def test_unmappable_params_logged(self, make_completion_adapter,...
    method test_user_param_debug_log (line 126) | async def test_user_param_debug_log(self, make_completion_adapter, cap...
    method test_sampling_params_mapped (line 140) | async def test_sampling_params_mapped(self, make_completion_adapter):
    method test_logprobs_param_sets_response_logprobs (line 161) | async def test_logprobs_param_sets_response_logprobs(self, make_comple...
    method test_logprobs_false_sets_response_logprobs_none (line 174) | async def test_logprobs_false_sets_response_logprobs_none(self, make_c...
    method test_stream_raises_not_implemented_removed (line 187) | async def test_stream_raises_not_implemented_removed(self, monkeypatch):
    method test_stream_completion_with_string_prompt (line 208) | async def test_stream_completion_with_string_prompt(self, monkeypatch):
    method test_stream_completion_rejects_list_prompt (line 237) | async def test_stream_completion_rejects_list_prompt(self, monkeypatch):
    method _patch_streaming_dependencies (line 250) | def _patch_streaming_dependencies(self, monkeypatch, adapter, fake_cli...
    method test_stream_multi_prompt_shared_completion_id (line 256) | async def test_stream_multi_prompt_shared_completion_id(self, monkeypa...
    method test_stream_multi_prompt_indices (line 272) | async def test_stream_multi_prompt_indices(self, monkeypatch):
    method test_stream_single_item_list_prompt (line 289) | async def test_stream_single_item_list_prompt(self, monkeypatch):
    method test_echo_non_streaming_prepends_prompt (line 302) | async def test_echo_non_streaming_prepends_prompt(self, make_completio...
    method test_echo_false_does_not_prepend (line 309) | async def test_echo_false_does_not_prepend(self, make_completion_adapt...
    method test_echo_multi_prompt_non_streaming (line 316) | async def test_echo_multi_prompt_non_streaming(self, make_completion_a...
    method test_non_streaming_n_two_returns_two_choices (line 324) | async def test_non_streaming_n_two_returns_two_choices(self, make_comp...
    method test_echo_streaming_emits_prompt_chunk (line 353) | async def test_echo_streaming_emits_prompt_chunk(self, monkeypatch):
  class TestCompletionModelExtra (line 368) | class TestCompletionModelExtra:
    method test_non_streaming_model_extra_forwarded (line 369) | async def test_non_streaming_model_extra_forwarded(self, make_completi...
    method test_streaming_model_extra_forwarded (line 384) | async def test_streaming_model_extra_forwarded(self, monkeypatch):
    method test_no_model_extra_does_not_add_extra_fields (line 415) | async def test_no_model_extra_does_not_add_extra_fields(self, make_com...
  class TestCompletionStreamOptions (line 428) | class TestCompletionStreamOptions:
    method _patch_stream_completion (line 429) | def _patch_stream_completion(self, monkeypatch, adapter: VertexAIInfer...
    method test_stream_options_injected_when_telemetry_active (line 456) | async def test_stream_options_injected_when_telemetry_active(self, mon...
    method test_stream_options_not_injected_when_telemetry_inactive (line 480) | async def test_stream_options_not_injected_when_telemetry_inactive(sel...
  class TestVertexAIModelSerialization (line 504) | class TestVertexAIModelSerialization:
    method test_model_dump_excludes_injected_extra_fields (line 505) | def test_model_dump_excludes_injected_extra_fields(self):
    method test_model_dump_json_excludes_injected_extra_fields (line 513) | def test_model_dump_json_excludes_injected_extra_fields(self):
    method test_model_dump_without_extra_fields_works_normally (line 522) | def test_model_dump_without_extra_fields_works_normally(self):

FILE: tests/unit/providers/inference/vertexai/test_config.py
  function project_name (line 20) | def project_name() -> str:
  class TestVertexAIConfig (line 25) | class TestVertexAIConfig:
    method test_config_field_population (line 43) | def test_config_field_population(
    method test_config_explicit_overrides_env (line 57) | def test_config_explicit_overrides_env(self):
    method test_access_token_alias_accepts_plain_string (line 71) | def test_access_token_alias_accepts_plain_string(self):
    method test_auth_credential_field_name_accepts_plain_string (line 82) | def test_auth_credential_field_name_accepts_plain_string(self):
    method test_sample_run_config (line 104) | def test_sample_run_config(self, sample_kwargs, expected_project, expe...
    method test_config_missing_required_project (line 110) | def test_config_missing_required_project(self):
    method test_auth_credential_excluded_from_serialization (line 115) | def test_auth_credential_excluded_from_serialization(self, project_name):
  class TestVertexAIProviderDataValidator (line 126) | class TestVertexAIProviderDataValidator:
    method test_validator_field_population (line 146) | def test_validator_field_population(
  class TestVertexAIConfigBackwardCompatibility (line 162) | class TestVertexAIConfigBackwardCompatibility:
    method test_access_token_alias_populates_auth_credential (line 163) | def test_access_token_alias_populates_auth_credential(self, project_na...
    method test_auth_credential_field_name_still_supported (line 173) | def test_auth_credential_field_name_still_supported(self, project_name):

FILE: tests/unit/providers/inference/vertexai/test_converters.py
  function weather_tool_call (line 50) | def weather_tool_call() -> dict[str, Any]:
  function _make_text_part (line 62) | def _make_text_part(text: str) -> Any:
  function _make_thought_part (line 67) | def _make_thought_part(thought_text: str) -> Any:
  function _make_function_call_part (line 72) | def _make_function_call_part(name: str, args: dict) -> Any:
  function _make_candidate (line 80) | def _make_candidate(
  function _make_response (line 96) | def _make_response(
  function _make_function_call_response (line 111) | def _make_function_call_response() -> Any:
  function _make_logprob_candidate (line 123) | def _make_logprob_candidate(
  function _make_top_candidates_entry (line 132) | def _make_top_candidates_entry(candidates: list) -> Any:
  function _make_logprobs_result (line 137) | def _make_logprobs_result(
  class TestConvertFinishReason (line 148) | class TestConvertFinishReason:
    method test_standard_mappings (line 164) | def test_standard_mappings(self, input_reason, expected):
    method test_none (line 168) | def test_none(self):
    method test_unknown_value (line 172) | def test_unknown_value(self):
    method test_case_insensitive (line 177) | def test_case_insensitive(self, input_reason):
  class TestConvertResponseFormat (line 183) | class TestConvertResponseFormat:
    method test_convert_response_format (line 207) | def test_convert_response_format(self, response_format, expected):
  class TestExtractTextContent (line 212) | class TestExtractTextContent:
    method test_extract_text_content (line 235) | def test_extract_text_content(self, input_content, expected):
  class TestConvertImageUrlPart (line 240) | class TestConvertImageUrlPart:
    method _convert_part (line 241) | def _convert_part(self, part: dict[str, Any]) -> dict[str, Any] | None:
    method test_data_uri_to_inline_data (line 255) | def test_data_uri_to_inline_data(self, fmt):
    method test_image_detail_parameter_ignored (line 261) | def test_image_detail_parameter_ignored(self):
    method test_unsupported_url_scheme_returns_none (line 280) | def test_unsupported_url_scheme_returns_none(self, url):
  class TestConvertUserMessageWithImages (line 286) | class TestConvertUserMessageWithImages:
    method test_user_message_conversion (line 369) | def test_user_message_conversion(self, message, expected_parts):
    method test_image_only_no_text (line 373) | def test_image_only_no_text(self):
    method test_unsupported_url_scheme_skipped (line 389) | def test_unsupported_url_scheme_skipped(self):
    method test_user_message_with_image_in_full_conversion (line 402) | def test_user_message_with_image_in_full_conversion(self):
  class TestConvertOpenAIMessagesToGemini (line 429) | class TestConvertOpenAIMessagesToGemini:
    method test_system_and_user_message_conversion (line 458) | def test_system_and_user_message_conversion(self, messages, expected_s...
    method test_assistant_message (line 465) | def test_assistant_message(self):
    method test_assistant_with_tool_calls (line 476) | def test_assistant_with_tool_calls(self, weather_tool_call):
    method test_tool_response_message (line 495) | def test_tool_response_message(self, weather_tool_call):
    method test_tool_response_non_json (line 524) | def test_tool_response_non_json(self):
    method test_tool_response_json_array_wrapped_in_dict (line 550) | def test_tool_response_json_array_wrapped_in_dict(self):
    method test_empty_messages (line 576) | def test_empty_messages(self):
    method test_assistant_with_text_and_tool_calls (line 582) | def test_assistant_with_text_and_tool_calls(self):
    method test_tool_call_id_not_found (line 605) | def test_tool_call_id_not_found(self):
  class TestConvertOpenAIToolsToGemini (line 619) | class TestConvertOpenAIToolsToGemini:
    method test_single_function_tool (line 620) | def test_single_function_tool(self):
    method test_multiple_tools (line 645) | def test_multiple_tools(self):
    method test_no_convertible_tools_returns_none (line 659) | def test_no_convertible_tools_returns_none(self, tools):
    method test_tool_without_parameters (line 663) | def test_tool_without_parameters(self):
  class TestConvertDeprecatedFunctions (line 672) | class TestConvertDeprecatedFunctions:
    method test_single_function_converts_to_tool (line 673) | def test_single_function_converts_to_tool(self):
    method test_multiple_functions (line 684) | def test_multiple_functions(self):
    method test_empty_functions_returns_empty (line 695) | def test_empty_functions_returns_empty(self):
  class TestConvertDeprecatedFunctionCall (line 700) | class TestConvertDeprecatedFunctionCall:
    method test_auto_passthrough (line 701) | def test_auto_passthrough(self):
    method test_none_passthrough (line 705) | def test_none_passthrough(self):
    method test_named_function_converts_to_tool_choice (line 709) | def test_named_function_converts_to_tool_choice(self):
    method test_unknown_string_passthrough (line 714) | def test_unknown_string_passthrough(self):
    method test_dict_without_name_passthrough (line 718) | def test_dict_without_name_passthrough(self):
  class TestConvertGeminiResponseToOpenAI (line 724) | class TestConvertGeminiResponseToOpenAI:
    method test_simple_text_response (line 725) | def test_simple_text_response(self):
    method test_function_call_response_sets_finish_reason_and_type (line 739) | def test_function_call_response_sets_finish_reason_and_type(self):
    method test_function_call_response_sets_function_payload (line 751) | def test_function_call_response_sets_function_payload(self):
    method test_multi_part_response (line 766) | def test_multi_part_response(self):
    method test_text_and_function_call_response (line 774) | def test_text_and_function_call_response(self):
    method test_no_candidates_safety_filtered (line 793) | def test_no_candidates_safety_filtered(self):
    method test_none_candidates (line 801) | def test_none_candidates(self):
    method test_usage_metadata (line 808) | def test_usage_metadata(self):
    method test_no_usage_metadata (line 822) | def test_no_usage_metadata(self):
    method test_empty_parts (line 831) | def test_empty_parts(self):
    method test_candidate_with_none_content (line 838) | def test_candidate_with_none_content(self):
    method test_multiple_function_calls (line 845) | def test_multiple_function_calls(self):
  class TestExtractUsage (line 869) | class TestExtractUsage:
    method test_extract_usage_with_cached_tokens (line 870) | def test_extract_usage_with_cached_tokens(self):
    method test_extract_usage_with_reasoning_tokens (line 886) | def test_extract_usage_with_reasoning_tokens(self):
    method test_extract_usage_no_details_when_fields_absent (line 902) | def test_extract_usage_no_details_when_fields_absent(self):
    method test_extract_usage_with_both_details (line 916) | def test_extract_usage_with_both_details(self):
  class TestConvertGeminiStreamChunkToOpenAI (line 935) | class TestConvertGeminiStreamChunkToOpenAI:
    method test_first_chunk_has_role (line 936) | def test_first_chunk_has_role(self):
    method test_subsequent_chunk_no_role (line 948) | def test_subsequent_chunk_no_role(self):
    method test_final_chunk_with_finish_reason (line 955) | def test_final_chunk_with_finish_reason(self):
    method test_chunk_with_tool_call (line 961) | def test_chunk_with_tool_call(self):
    method test_empty_chunk_no_candidates (line 976) | def test_empty_chunk_no_candidates(self):
    method test_stream_usage (line 985) | def test_stream_usage(self):
    method test_safety_filtered_chunk (line 998) | def test_safety_filtered_chunk(self):
  class TestGenerateCompletionId (line 1005) | class TestGenerateCompletionId:
    method test_format (line 1006) | def test_format(self):
    method test_uniqueness (line 1014) | def test_uniqueness(self):
  class TestExtractLogprobs (line 1020) | class TestExtractLogprobs:
    method test_logprobs_none_when_not_present (line 1021) | def test_logprobs_none_when_not_present(self):
    method test_logprobs_extracted_from_response (line 1027) | def test_logprobs_extracted_from_response(self):
    method test_logprobs_bytes_is_utf8_encoding (line 1043) | def test_logprobs_bytes_is_utf8_encoding(self):
    method test_top_logprobs_extracted (line 1055) | def test_top_logprobs_extracted(self):
    method test_logprobs_in_streaming_chunk (line 1077) | def test_logprobs_in_streaming_chunk(self):
  class TestReasoningContent (line 1102) | class TestReasoningContent:
    method test_thought_parts_separated_from_text (line 1103) | def test_thought_parts_separated_from_text(self):
    method test_multiple_thought_parts_concatenated (line 1114) | def test_multiple_thought_parts_concatenated(self):
    method test_no_thought_parts_reasoning_content_none (line 1131) | def test_no_thought_parts_reasoning_content_none(self):
    method test_only_thought_parts_text_none (line 1138) | def test_only_thought_parts_text_none(self):
    method test_thought_parts_in_streaming_chunk (line 1145) | def test_thought_parts_in_streaming_chunk(self):
    method test_thought_parts_in_non_streaming_response (line 1159) | def test_thought_parts_in_non_streaming_response(self):
    method test_thought_and_tool_calls (line 1170) | def test_thought_and_tool_calls(self):
  class TestConvertCompletionPromptToContents (line 1189) | class TestConvertCompletionPromptToContents:
    method test_string_prompt_to_user_message (line 1192) | def test_string_prompt_to_user_message(self):
    method test_empty_prompt (line 1197) | def test_empty_prompt(self):
  class TestConvertGeminiResponseToCompletion (line 1203) | class TestConvertGeminiResponseToCompletion:
    method test_simple_text_completion (line 1206) | def test_simple_text_completion(self):
    method test_no_candidates_returns_empty_text (line 1216) | def test_no_candidates_returns_empty_text(self):
    method test_finish_reason_stop (line 1223) | def test_finish_reason_stop(self):
    method test_completion_logprobs_extracted_when_present (line 1231) | def test_completion_logprobs_extracted_when_present(self):
    method test_completion_logprobs_none_when_absent (line 1252) | def test_completion_logprobs_none_when_absent(self):
    method test_multiple_candidates_return_multiple_choices (line 1260) | def test_multiple_candidates_return_multiple_choices(self):
  class TestConvertGeminiStreamChunkToOpenAICompletion (line 1276) | class TestConvertGeminiStreamChunkToOpenAICompletion:
    method test_stream_completion_chunk_with_text (line 1279) | def test_stream_completion_chunk_with_text(self):
    method test_stream_completion_chunk_mid_stream_uses_stop_default (line 1297) | def test_stream_completion_chunk_mid_stream_uses_stop_default(self):
    method test_stream_completion_chunk_with_logprobs (line 1309) | def test_stream_completion_chunk_with_logprobs(self):
    method test_stream_completion_chunk_empty_candidates (line 1332) | def test_stream_completion_chunk_empty_candidates(self):
    method test_stream_completion_chunk_none_candidates (line 1343) | def test_stream_completion_chunk_none_candidates(self):
    method test_stream_completion_chunk_multiple_candidates (line 1353) | def test_stream_completion_chunk_multiple_candidates(self):

FILE: tests/unit/providers/inline/agents/builtin/responses/test_streaming.py
  function mock_safety_api (line 19) | def mock_safety_api():
  function mock_inference_api (line 33) | def mock_inference_api():
  function mock_context (line 39) | def mock_context():
  function test_convert_tooldef_to_chat_tool_preserves_items_field (line 48) | def test_convert_tooldef_to_chat_tool_preserves_items_field():

FILE: tests/unit/providers/nvidia/conftest.py
  function patch_aiohttp_session (line 20) | def patch_aiohttp_session():
  function event_loop (line 26) | def event_loop():
  function run_async (line 35) | def run_async():

FILE: tests/unit/providers/nvidia/test_datastore.py
  function nvidia_adapter (line 18) | def nvidia_adapter():
  function _assert_request (line 33) | def _assert_request(mock_call, expected_method, expected_path, expected_...
  function test_register_dataset (line 45) | def test_register_dataset(nvidia_adapter, run_async):
  function test_unregister_dataset (line 81) | def test_unregister_dataset(nvidia_adapter, run_async):
  function test_register_dataset_with_custom_namespace_project (line 96) | def test_register_dataset_with_custom_namespace_project(run_async):

FILE: tests/unit/providers/nvidia/test_eval.py
  function nvidia_eval_setup (line 37) | def nvidia_eval_setup():
  function _assert_request_body (line 78) | def _assert_request_body(mock_evaluator_post, expected_json):
  function test_register_benchmark (line 95) | async def test_register_benchmark(nvidia_eval_setup):
  function test_run_eval (line 135) | async def test_run_eval(nvidia_eval_setup):
  function test_job_status (line 172) | async def test_job_status(nvidia_eval_setup):
  function test_job_cancel (line 192) | async def test_job_cancel(nvidia_eval_setup):
  function test_job_result (line 207) | async def test_job_result(nvidia_eval_setup):

FILE: tests/unit/providers/nvidia/test_rerank_inference.py
  class MockResponse (line 19) | class MockResponse:
    method __init__ (line 20) | def __init__(self, status=200, json_data=None, text_data="OK"):
    method json (line 25) | async def json(self):
    method text (line 28) | async def text(self):
  class MockSession (line 32) | class MockSession:
    method __init__ (line 33) | def __init__(self, response):
    method __aenter__ (line 37) | async def __aenter__(self):
    method __aexit__ (line 40) | async def __aexit__(self, exc_type, exc_val, exc_tb):
    method post (line 43) | def post(self, url, **kwargs):
  function create_adapter (line 59) | def create_adapter(config=None, rerank_endpoints=None):
  function test_rerank_basic_functionality (line 78) | async def test_rerank_basic_functionality():
  function test_missing_rankings_key (line 98) | async def test_missing_rankings_key():
  function test_hosted_with_endpoint (line 109) | async def test_hosted_with_endpoint():
  function test_hosted_without_endpoint (line 123) | async def test_hosted_without_endpoint():
  function test_hosted_model_not_in_endpoint_mapping (line 138) | async def test_hosted_model_not_in_endpoint_mapping():
  function test_self_hosted_ignores_endpoint (line 153) | async def test_self_hosted_ignores_endpoint():
  function test_max_num_results (line 169) | async def test_max_num_results():
  function test_http_error (line 183) | async def test_http_error():
  function test_client_error (line 193) | async def test_client_error():
  function test_list_models_includes_configured_rerank_models (line 204) | async def test_list_models_includes_configured_rerank_models():
  function test_list_provider_model_ids_has_no_duplicates (line 233) | async def test_list_provider_model_ids_has_no_duplicates():
  function test_list_provider_model_ids_uses_configured_on_dynamic_failure (line 252) | async def test_list_provider_model_ids_uses_configured_on_dynamic_failur...

FILE: tests/unit/providers/nvidia/test_safety.py
  class FakeNVIDIASafetyAdapter (line 27) | class FakeNVIDIASafetyAdapter(NVIDIASafetyAdapter):
    method __init__ (line 30) | def __init__(self, config: NVIDIASafetyConfig, shield_store):
  function nvidia_adapter (line 36) | def nvidia_adapter():
  function mock_guardrails_post (line 55) | def mock_guardrails_post():
  function _assert_request (line 62) | def _assert_request(
  function test_register_shield_with_valid_id (line 97) | async def test_register_shield_with_valid_id(nvidia_adapter):
  function test_register_shield_without_id (line 111) | async def test_register_shield_without_id(nvidia_adapter):
  function test_run_shield_allowed (line 126) | async def test_run_shield_allowed(nvidia_adapter, mock_guardrails_post):
  function test_run_shield_blocked (line 182) | async def test_run_shield_blocked(nvidia_adapter, mock_guardrails_post):
  function test_run_shield_not_found (line 240) | async def test_run_shield_not_found(nvidia_adapter, mock_guardrails_post):
  function test_run_shield_http_error (line 262) | async def test_run_shield_http_error(nvidia_adapter, mock_guardrails_post):
  function test_init_nemo_guardrails (line 317) | def test_init_nemo_guardrails():
  function test_init_nemo_guardrails_invalid_temperature (line 349) | def test_init_nemo_guardrails_invalid_temperature():

FILE: tests/unit/providers/safety/passthrough/conftest.py
  class FakePassthroughSafetyAdapter (line 19) | class FakePassthroughSafetyAdapter(PassthroughSafetyAdapter):
    method __init__ (line 22) | def __init__(self, config: PassthroughSafetyConfig, shield_store: Asyn...
  function _stub_provider_spec (line 27) | def _stub_provider_spec(adapter: PassthroughSafetyAdapter) -> None:
  function mock_httpx_response (line 34) | def mock_httpx_response(json_data: dict, status_code: int = 200) -> Magi...
  function shield_store (line 43) | def shield_store() -> AsyncMock:
  function config (line 50) | def config() -> PassthroughSafetyConfig:
  function config_with_api_key (line 55) | def config_with_api_key() -> PassthroughSafetyConfig:
  function config_with_forward_headers (line 63) | def config_with_forward_headers() -> PassthroughSafetyConfig:
  function adapter (line 71) | def adapter(config: PassthroughSafetyConfig, shield_store: AsyncMock) ->...
  function adapter_with_api_key (line 79) | def adapter_with_api_key(
  function adapter_with_forward_headers (line 89) | def adapter_with_forward_headers(
  function provider_data_ctx (line 98) | def provider_data_ctx(data: dict) -> AbstractContextManager:

FILE: tests/unit/providers/safety/passthrough/test_config.py
  function test_config_requires_base_url (line 16) | def test_config_requires_base_url():
  function test_config_api_key_stored_as_secret (line 21) | def test_config_api_key_stored_as_secret():
  function test_config_forward_headers_accepts_mapping (line 30) | def test_config_forward_headers_accepts_mapping():
  function test_config_sample_run_config (line 38) | def test_config_sample_run_config():
  function test_provider_data_validator_allows_extra_keys (line 44) | def test_provider_data_validator_allows_extra_keys():

FILE: tests/unit/providers/safety/passthrough/test_headers.py
  function _make_adapter_with_headers (line 36) | def _make_adapter_with_headers(
  function test_forward_headers_maps_keys (line 50) | async def test_forward_headers_maps_keys(adapter_with_forward_headers: F...
  function test_forward_headers_skips_missing_keys (line 57) | async def test_forward_headers_skips_missing_keys(
  function test_forward_headers_sent_in_run_shield (line 67) | async def test_forward_headers_sent_in_run_shield(shield_store: AsyncMoc...
  function test_forward_headers_sent_in_run_moderation (line 88) | async def test_forward_headers_sent_in_run_moderation(shield_store: Asyn...
  function test_forward_headers_does_not_leak_unlisted_keys (line 101) | async def test_forward_headers_does_not_leak_unlisted_keys(shield_store:...
  function test_url_trailing_slash_stripped (line 113) | async def test_url_trailing_slash_stripped(shield_store: AsyncMock) -> N...
  function test_forward_headers_coerces_non_string_values (line 140) | async def test_forward_headers_coerces_non_string_values(shield_store: A...
  function test_forward_headers_crlf_stripped (line 153) | async def test_forward_headers_crlf_stripped(shield_store: AsyncMock) ->...
  function test_forward_headers_rejects_dunder_keys (line 163) | async def test_forward_headers_rejects_dunder_keys() -> None:
  function test_config_api_key_cannot_be_overridden_by_forward_headers (line 171) | async def test_config_api_key_cannot_be_overridden_by_forward_headers(sh...
  function test_forward_headers_rejects_blocked_headers (line 191) | async def test_forward_headers_rejects_blocked_headers(header_name: str)...
  function test_forward_headers_rejects_multiple_violations (line 199) | async def test_forward_headers_rejects_multiple_violations() -> None:

FILE: tests/unit/providers/safety/passthrough/test_passthrough.py
  function _make_shield (line 32) | def _make_shield(shield_id: str = "test-shield", provider_resource_id: s...
  function test_run_shield_safe_content (line 44) | async def test_run_shield_safe_content(adapter: FakePassthroughSafetyAda...
  function test_run_shield_flagged_content (line 62) | async def test_run_shield_flagged_content(adapter: FakePassthroughSafety...
  function test_run_shield_not_found (line 86) | async def test_run_shield_not_found(adapter: FakePassthroughSafetyAdapte...
  function test_run_shield_http_5xx (line 94) | async def test_run_shield_http_5xx(adapter: FakePassthroughSafetyAdapter...
  function test_run_shield_http_4xx_raises_value_error (line 110) | async def test_run_shield_http_4xx_raises_value_error(adapter: FakePasst...
  function test_run_shield_uses_provider_resource_id (line 127) | async def test_run_shield_uses_provider_resource_id(adapter: FakePassthr...
  function test_run_shield_url_construction (line 144) | async def test_run_shield_url_construction(adapter: FakePassthroughSafet...
  function test_run_shield_multiple_messages (line 165) | async def test_run_shield_multiple_messages(adapter: FakePassthroughSafe...
  function test_run_shield_empty_messages (line 188) | async def test_run_shield_empty_messages(adapter: FakePassthroughSafetyA...
  function test_run_shield_empty_results_raises (line 194) | async def test_run_shield_empty_results_raises(adapter: FakePassthroughS...
  function test_run_moderation_safe (line 208) | async def test_run_moderation_safe(adapter: FakePassthroughSafetyAdapter...
  function test_run_moderation_flagged (line 223) | async def test_run_moderation_flagged(adapter: FakePassthroughSafetyAdap...
  function test_run_moderation_multiple_inputs (line 237) | async def test_run_moderation_multiple_inputs(adapter: FakePassthroughSa...
  function test_run_moderation_no_model (line 257) | async def test_run_moderation_no_model(adapter: FakePassthroughSafetyAda...
  function test_run_moderation_http_error (line 272) | async def test_run_moderation_http_error(adapter: FakePassthroughSafetyA...
  function test_api_key_from_config (line 288) | async def test_api_key_from_config(adapter_with_api_key: FakePassthrough...
  function test_api_key_from_provider_data (line 292) | async def test_api_key_from_provider_data(adapter: FakePassthroughSafety...
  function test_api_key_config_wins_over_provider_data (line 297) | async def test_api_key_config_wins_over_provider_data(adapter_with_api_k...
  function test_api_key_none_when_neither_set (line 302) | async def test_api_key_none_when_neither_set(adapter: FakePassthroughSaf...
  function test_authorization_header_sent (line 306) | async def test_authorization_header_sent(adapter_with_api_key: FakePasst...
  function test_empty_api_key_falls_through_to_provider_data (line 311) | async def test_empty_api_key_falls_through_to_provider_data(adapter: Fak...
  function test_run_shield_null_results_raises (line 323) | async def test_run_shield_null_results_raises(adapter: FakePassthroughSa...
  function test_run_shield_null_categories (line 334) | async def test_run_shield_null_categories(adapter: FakePassthroughSafety...
  function test_run_moderation_null_results_raises (line 351) | async def test_run_moderation_null_results_raises(adapter: FakePassthrou...
  function test_run_moderation_null_categories (line 359) | async def test_run_moderation_null_categories(adapter: FakePassthroughSa...
  function test_run_shield_flags_later_results (line 377) | async def test_run_shield_flags_later_results(adapter: FakePassthroughSa...
  function test_run_shield_timeout_wrapped (line 407) | async def test_run_shield_timeout_wrapped(adapter: FakePassthroughSafety...
  function test_run_shield_connect_error_wrapped (line 417) | async def test_run_shield_connect_error_wrapped(adapter: FakePassthrough...
  function test_run_moderation_non_json_response (line 427) | async def test_run_moderation_non_json_response(adapter: FakePassthrough...
  function test_run_moderation_non_dict_json_response (line 436) | async def test_run_moderation_non_dict_json_response(adapter: FakePassth...

FILE: tests/unit/providers/safety/passthrough/test_registry.py
  function test_passthrough_in_registry (line 10) | def test_passthrough_in_registry():
  function test_passthrough_registry_has_provider_data_validator (line 16) | def test_passthrough_registry_has_provider_data_validator():
  function test_passthrough_registry_module_path (line 23) | def test_passthrough_registry_module_path():
  function test_registry_alphabetical_order (line 29) | def test_registry_alphabetical_order():

FILE: tests/unit/providers/test_bedrock.py
  function test_can_create_adapter (line 15) | def test_can_create_adapter():
  function test_different_aws_regions (line 24) | def test_different_aws_regions():
  function test_basic_chat_completion (line 35) | async def test_basic_chat_completion():

FILE: tests/unit/providers/test_configs.py
  class TestProviderConfigurations (line 16) | class TestProviderConfigurations:
    method test_api_providers (line 20) | def test_api_providers(self, api):
    method _verify_provider_config (line 34) | def _verify_provider_config(self, provider_type, provider_spec):
    method test_remote_inference_url_standardization (line 47) | def test_remote_inference_url_standardization(self):

FILE: tests/unit/providers/test_lazy_imports.py
  function _check_module_import_isolation (line 19) | def _check_module_import_isolation(module_path: str, forbidden_modules: ...
  class TestPromptGuardLazyImports (line 63) | class TestPromptGuardLazyImports:
    method test_no_torch_transformers_on_import (line 66) | def test_no_torch_transformers_on_import(self):
  class TestBraintrustLazyImports (line 80) | class TestBraintrustLazyImports:
    method test_braintrust_import_no_autoevals (line 83) | def test_braintrust_import_no_autoevals(self):
  function _check_no_forbidden_imports (line 97) | def _check_no_forbidden_imports(module_path: str, forbidden: list[str]) ...
  class TestEmbeddingMixinLazyImports (line 116) | class TestEmbeddingMixinLazyImports:
    method test_no_torch_on_import (line 119) | def test_no_torch_on_import(self):
  class TestFaissLazyImports (line 127) | class TestFaissLazyImports:
    method test_no_faiss_numpy_on_import (line 130) | def test_no_faiss_numpy_on_import(self):
  class TestSqliteVecLazyImports (line 138) | class TestSqliteVecLazyImports:
    method test_no_numpy_sqlite_vec_on_import (line 141) | def test_no_numpy_sqlite_vec_on_import(self):
  class TestVectorStoreLazyImports (line 149) | class TestVectorStoreLazyImports:
    method test_no_numpy_on_import (line 152) | def test_no_numpy_on_import(self):

FILE: tests/unit/providers/test_provider_data_secretstr.py
  function _is_secretstr (line 22) | def _is_secretstr(annotation) -> bool:
  function _looks_secret (line 31) | def _looks_secret(name: str) -> bool:
  class TestProviderDataSecretStr (line 36) | class TestProviderDataSecretStr:
    method test_provider_data_fields_are_secretstr (line 38) | def test_provider_data_fields_are_secretstr(self, api):

FILE: tests/unit/providers/utils/common/test_data_url.py
  class TestParseDataUrlValid (line 20) | class TestParseDataUrlValid:
    method test_plain_text (line 21) | def test_plain_text(self):
    method test_base64_flag (line 28) | def test_base64_flag(self):
    method test_charset_only (line 35) | def test_charset_only(self):
    method test_charset_and_base64 (line 43) | def test_charset_and_base64(self):
    method test_image_mime_type (line 50) | def test_image_mime_type(self):
    method test_mime_type_with_plus (line 56) | def test_mime_type_with_plus(self):
    method test_empty_data_portion (line 62) | def test_empty_data_portion(self):
    method test_multiline_data (line 68) | def test_multiline_data(self):
    method test_hyphenated_charset (line 73) | def test_hyphenated_charset(self):
    method test_mime_type_with_dot (line 77) | def test_mime_type_with_dot(self):
  class TestParseDataUrlInvalid (line 82) | class TestParseDataUrlInvalid:
    method test_http_url_raises (line 83) | def test_http_url_raises(self):
    method test_missing_comma_raises (line 87) | def test_missing_comma_raises(self):
    method test_empty_string_raises (line 91) | def test_empty_string_raises(self):
    method test_missing_mime_type_raises (line 95) | def test_missing_mime_type_raises(self):
    method test_plain_string_raises (line 100) | def test_plain_string_raises(self):

FILE: tests/unit/providers/utils/inference/test_embedding_validation.py
  class TestEmbeddingValidation (line 12) | class TestEmbeddingValidation:
    method test_valid_string_input (line 15) | def test_valid_string_input(self):
    method test_valid_list_of_strings_input (line 21) | def test_valid_list_of_strings_input(self):
    method test_invalid_list_of_ints_input (line 27) | def test_invalid_list_of_ints_input(self):
    method test_invalid_list_of_list_of_ints_input (line 37) | def test_invalid_list_of_list_of_ints_input(self):
    method test_error_message_includes_model_name (line 47) | def test_error_message_includes_model_name(self):

FILE: tests/unit/providers/utils/inference/test_network_config.py
  class TestTLSConfig (line 28) | class TestTLSConfig:
    method test_default_values (line 31) | def test_default_values(self):
    method test_verify_boolean (line 40) | def test_verify_boolean(self):
    method test_verify_valid_path (line 45) | def test_verify_valid_path(self):
    method test_verify_invalid_path (line 57) | def test_verify_invalid_path(self):
    method test_min_version_valid (line 62) | def test_min_version_valid(self):
    method test_ciphers_list (line 70) | def test_ciphers_list(self):
    method test_mtls_requires_both_cert_and_key (line 76) | def test_mtls_requires_both_cert_and_key(self):
    method test_mtls_valid_paths (line 88) | def test_mtls_valid_paths(self):
  class TestProxyConfig (line 107) | class TestProxyConfig:
    method test_default_values (line 110) | def test_default_values(self):
    method test_single_proxy_url (line 118) | def test_single_proxy_url(self):
    method test_granular_proxy_settings (line 123) | def test_granular_proxy_settings(self):
    method test_no_proxy_list (line 129) | def test_no_proxy_list(self):
    method test_cacert_valid_path (line 134) | def test_cacert_valid_path(self):
    method test_cacert_invalid_path (line 146) | def test_cacert_invalid_path(self):
    method test_cacert_with_proxy_url (line 151) | def test_cacert_with_proxy_url(self):
    method test_cacert_with_granular_proxies (line 164) | def test_cacert_with_granular_proxies(self):
    method test_url_and_granular_conflict (line 178) | def test_url_and_granular_conflict(self):
  class TestTimeoutConfig (line 184) | class TestTimeoutConfig:
    method test_default_values (line 187) | def test_default_values(self):
    method test_with_connect_only (line 193) | def test_with_connect_only(self):
    method test_with_read_only (line 199) | def test_with_read_only(self):
    method test_with_both (line 205) | def test_with_both(self):
  class TestNetworkConfig (line 212) | class TestNetworkConfig:
    method test_default_values (line 215) | def test_default_values(self):
    method test_with_tls_config (line 222) | def test_with_tls_config(self):
    method test_with_proxy_config (line 228) | def test_with_proxy_config(self):
    method test_with_timeout_float (line 234) | def test_with_timeout_float(self):
    method test_with_timeout_config (line 239) | def test_with_timeout_config(self):
    method test_full_config (line 247) | def test_full_config(self):
  class TestBuildSSLContext (line 260) | class TestBuildSSLContext:
    method test_simple_verify_true (line 263) | def test_simple_verify_true(self):
    method test_simple_verify_false (line 269) | def test_simple_verify_false(self):
    method test_verify_with_path (line 275) | def test_verify_with_path(self):
    method test_with_min_version_returns_ssl_context (line 288) | def test_with_min_version_returns_ssl_context(self):
    method test_with_min_version_tls13 (line 295) | def test_with_min_version_tls13(self):
    method test_with_ciphers_returns_ssl_context (line 302) | def test_with_ciphers_returns_ssl_context(self):
    method test_verify_false_with_advanced_options (line 308) | def test_verify_false_with_advanced_options(self):
  class TestBuildProxyMounts (line 317) | class TestBuildProxyMounts:
    method test_none_config (line 320) | def test_none_config(self):
    method test_single_url (line 326) | def test_single_url(self):
    method test_granular_proxies (line 334) | def test_granular_proxies(self):
    method test_proxy_with_cacert (line 342) | def test_proxy_with_cacert(self):
    method test_granular_proxies_with_cacert (line 377) | def test_granular_proxies_with_cacert(self):
  class TestBuildHttpClient (line 413) | class TestBuildHttpClient:
    method test_none_config (line 416) | def test_none_config(self):
    method test_empty_network_config (line 421) | def test_empty_network_config(self):
    method test_with_tls_config (line 427) | def test_with_tls_config(self):
    method test_with_timeout_float (line 434) | def test_with_timeout_float(self):
    method test_with_timeout_config (line 445) | def test_with_timeout_config(self):
    method test_with_timeout_config_connect_only (line 460) | def test_with_timeout_config_connect_only(self):
    method test_with_timeout_config_read_only (line 470) | def test_with_timeout_config_read_only(self):
    method test_with_proxy (line 480) | def test_with_proxy(self):
  class TestVLLMBackwardCompatibility (line 488) | class TestVLLMBackwardCompatibility:
    method test_legacy_tls_verify_true (line 491) | def test_legacy_tls_verify_true(self):
    method test_legacy_tls_verify_false (line 502) | def test_legacy_tls_verify_false(self):
    method test_legacy_tls_verify_path (line 513) | def test_legacy_tls_verify_path(self):
    method test_new_network_config_style (line 531) | def test_new_network_config_style(self):
    method test_network_not_overwritten_by_tls_verify (line 546) | def test_network_not_overwritten_by_tls_verify(self):

FILE: tests/unit/providers/utils/inference/test_openai_compat.py
  class TestGetStreamOptionsForTelemetry (line 14) | class TestGetStreamOptionsForTelemetry:
    method test_returns_original_when_not_streaming (line 15) | def test_returns_original_when_not_streaming(self):
    method test_streaming_without_active_span_returns_original (line 22) | def test_streaming_without_active_span_returns_original(self):
    method test_streaming_with_inactive_span_returns_original (line 30) | def test_streaming_with_inactive_span_returns_original(self):
    method test_streaming_with_active_span_adds_usage_when_missing (line 40) | def test_streaming_with_active_span_adds_usage_when_missing(self):
    method test_streaming_with_active_span_merges_existing_options (line 49) | def test_streaming_with_active_span_merges_existing_options(self):
    method test_streaming_with_active_span_overrides_include_usage_false (line 60) | def test_streaming_with_active_span_overrides_include_usage_false(self):

FILE: tests/unit/providers/utils/inference/test_openai_mixin.py
  class OpenAIMixinImpl (line 28) | class OpenAIMixinImpl(OpenAIMixin):
    method get_api_key (line 31) | def get_api_key(self) -> str:
    method get_base_url (line 34) | def get_base_url(self) -> str:
  class OpenAIMixinWithEmbeddingsImpl (line 38) | class OpenAIMixinWithEmbeddingsImpl(OpenAIMixinImpl):
  class OpenAIMixinWithCustomModelConstruction (line 47) | class OpenAIMixinWithCustomModelConstruction(OpenAIMixinImpl):
    method construct_model_from_identifier (line 58) | def construct_model_from_identifier(self, identifier: str) -> Model:
  function mixin (line 70) | def mixin():
  function mixin_with_embeddings (line 87) | def mixin_with_embeddings():
  function mixin_with_custom_model_construction (line 94) | def mixin_with_custom_model_construction():
  function mock_models (line 101) | def mock_models():
  function mock_client_with_models (line 108) | def mock_client_with_models(mock_models):
  function mock_client_with_empty_models (line 121) | def mock_client_with_empty_models():
  function mock_client_with_exception (line 134) | def mock_client_with_exception():
  function mock_client_context (line 142) | def mock_client_context():
  function _assert_models_match_expected (line 151) | def _assert_models_match_expected(actual_models, expected_models):
  class TestOpenAIMixinListModels (line 164) | class TestOpenAIMixinListModels:
    method test_list_models_success (line 167) | async def test_list_models_success(self, mixin, mock_client_with_model...
    method test_list_models_empty_response (line 194) | async def test_list_models_empty_response(self, mixin, mock_client_wit...
  class TestOpenAIMixinCheckModelAvailability (line 204) | class TestOpenAIMixinCheckModelAvailability:
    method test_check_model_availability_with_cache (line 207) | async def test_check_model_availability_with_cache(self, mixin, mock_c...
    method test_check_model_availability_without_cache (line 220) | async def test_check_model_availability_without_cache(self, mixin, moc...
    method test_check_model_availability_model_not_found (line 232) | async def test_check_model_availability_model_not_found(self, mixin, m...
    method test_check_model_availability_with_pre_registered_model (line 241) | async def test_check_model_availability_with_pre_registered_model(
    method test_check_model_availability_fallback_to_provider_when_not_in_store (line 258) | async def test_check_model_availability_fallback_to_provider_when_not_...
  class TestOpenAIMixinCacheBehavior (line 276) | class TestOpenAIMixinCacheBehavior:
    method test_cache_overwrites_on_list_models_call (line 279) | async def test_cache_overwrites_on_list_models_call(self, mixin, mock_...
  class TestOpenAIMixinImagePreprocessing (line 299) | class TestOpenAIMixinImagePreprocessing:
    method test_openai_chat_completion_with_image_preprocessing_enabled (line 302) | async def test_openai_chat_completion_with_image_preprocessing_enabled...
    method test_openai_chat_completion_with_image_preprocessing_disabled (line 337) | async def test_openai_chat_completion_with_image_preprocessing_disable...
  class TestOpenAIMixinEmbeddingModelMetadata (line 369) | class TestOpenAIMixinEmbeddingModelMetadata:
    method test_embedding_model_identified_and_augmented (line 372) | async def test_embedding_model_identified_and_augmented(self, mixin_wi...
  class TestOpenAIMixinCustomModelConstruction (line 411) | class TestOpenAIMixinCustomModelConstruction:
    method test_mixed_model_types_identification (line 414) | async def test_mixed_model_types_identification(self, mixin_with_custo...
  class TestOpenAIMixinAllowedModels (line 460) | class TestOpenAIMixinAllowedModels:
    method test_list_models_with_allowed_models_filter (line 463) | async def test_list_models_with_allowed_models_filter(self, mixin, moc...
    method test_list_models_with_empty_allowed_models (line 478) | async def test_list_models_with_empty_allowed_models(self, mixin, mock...
    method test_list_models_with_omitted_allowed_models (line 488) | async def test_list_models_with_omitted_allowed_models(self, mixin, mo...
    method test_check_model_availability_with_allowed_models (line 503) | async def test_check_model_availability_with_allowed_models(
  class TestOpenAIMixinModelRegistration (line 515) | class TestOpenAIMixinModelRegistration:
    method test_register_model_success (line 518) | async def test_register_model_success(self, mock_client_with_models, m...
    method test_register_model_not_available (line 542) | async def test_register_model_not_available(self, mock_client_with_mod...
    method test_register_model_with_allowed_models_filter (line 563) | async def test_register_model_with_allowed_models_filter(self, mock_cl...
    method test_register_embedding_model (line 595) | async def test_register_embedding_model(self, mixin_with_embeddings, m...
    method test_unregister_model (line 620) | async def test_unregister_model(self, mixin):
    method test_should_refresh_models (line 626) | async def test_should_refresh_models(self, mixin):
    method test_register_model_error_propagation (line 638) | async def test_register_model_error_propagation(self, mock_client_with...
    method test_register_model_default_behavior_no_validation (line 657) | async def test_register_model_default_behavior_no_validation(self, moc...
    method test_register_model_with_validation_enabled (line 678) | async def test_register_model_with_validation_enabled(self, mock_clien...
    method test_register_model_with_validation_explicitly_disabled (line 700) | async def test_register_model_with_validation_explicitly_disabled(
  class ProviderDataValidator (line 726) | class ProviderDataValidator(BaseModel):
  class OpenAIMixinWithProviderData (line 732) | class OpenAIMixinWithProviderData(OpenAIMixinImpl):
    method get_api_key (line 737) | def get_api_key(self) -> str:
    method get_base_url (line 740) | def get_base_url(self):
  class CustomListProviderModelIdsImplementation (line 744) | class CustomListProviderModelIdsImplementation(OpenAIMixinImpl):
    method list_provider_model_ids (line 749) | async def list_provider_model_ids(self) -> Iterable[str]:
  class TestOpenAIMixinCustomListProviderModelIds (line 754) | class TestOpenAIMixinCustomListProviderModelIds:
    method custom_model_ids_list (line 758) | def custom_model_ids_list(self):
    method config (line 763) | def config(self):
    method adapter (line 768) | def adapter(self, custom_model_ids_list, config):
    method test_is_used (line 774) | async def test_is_used(self, adapter, custom_model_ids_list):
    method test_populates_cache (line 783) | async def test_populates_cache(self, adapter, custom_model_ids_list):
    method test_respects_allowed_models (line 791) | async def test_respects_allowed_models(self, config):
    method test_with_empty_list (line 804) | async def test_with_empty_list(self, config):
    method test_wrong_type_raises_error (line 814) | async def test_wrong_type_raises_error(self, config):
    method test_non_iterable_raises_error (line 836) | async def test_non_iterable_raises_error(self, config):
    method test_accepts_various_iterables (line 846) | async def test_accepts_various_iterables(self, config):
  class TestOpenAIMixinProviderDataApiKey (line 875) | class TestOpenAIMixinProviderDataApiKey:
    method mixin_with_provider_data_field (line 879) | def mixin_with_provider_data_field(self):
    method mixin_with_provider_data_field_and_none_api_key (line 895) | def mixin_with_provider_data_field_and_none_api_key(self, mixin_with_p...
    method test_no_provider_data (line 899) | def test_no_provider_data(self, mixin_with_provider_data_field):
    method test_with_provider_data (line 903) | def test_with_provider_data(self, mixin_with_provider_data_field):
    method test_with_wrong_key (line 910) | def test_with_wrong_key(self, mixin_with_provider_data_field):
    method test_error_when_no_config_and_provider_data_has_wrong_key (line 915) | def test_error_when_no_config_and_provider_data_has_wrong_key(
    method test_error_message_includes_correct_field_names (line 923) | def test_error_message_includes_correct_field_names(self, mixin_with_p...
  class TestOpenAIMixinAllowedModelsInference (line 933) | class TestOpenAIMixinAllowedModelsInference:
    method test_inference_with_allowed_models (line 936) | async def test_inference_with_allowed_models(self, mixin, mock_client_...
    method test_inference_with_disallowed_models (line 969) | async def test_inference_with_disallowed_models(self, mixin, mock_clie...
    method test_inference_with_no_restrictions (line 1000) | async def test_inference_with_no_restrictions(self, mixin, mock_client...
  class TestOpenAIMixinStreamOptionsInjection (line 1026) | class TestOpenAIMixinStreamOptionsInjection:
    method test_chat_completion_injects_stream_options_when_telemetry_active (line 1029) | async def test_chat_completion_injects_stream_options_when_telemetry_a...
    method test_chat_completion_preserves_existing_stream_options (line 1050) | async def test_chat_completion_preserves_existing_stream_options(self,...
    method test_chat_completion_no_injection_when_telemetry_inactive (line 1072) | async def test_chat_completion_no_injection_when_telemetry_inactive(se...
    method test_chat_completion_no_injection_when_not_streaming (line 1092) | async def test_chat_completion_no_injection_when_not_streaming(self, m...
    method test_completion_injects_stream_options_when_telemetry_active (line 1111) | async def test_completion_injects_stream_options_when_telemetry_active...
    method test_completion_no_injection_when_telemetry_inactive (line 1129) | async def test_completion_no_injection_when_telemetry_inactive(self, m...
    method test_params_not_mutated (line 1146) | async def test_params_not_mutated(self, mixin, mock_client_context):
    method test_chat_completion_overrides_include_usage_false (line 1165) | async def test_chat_completion_overrides_include_usage_false(self, mix...
    method test_no_injection_when_provider_doesnt_support_stream_options (line 1188) | async def test_no_injection_when_provider_doesnt_support_stream_option...
    method test_completion_no_injection_when_provider_doesnt_support_stream_options (line 1212) | async def test_completion_no_injection_when_provider_doesnt_support_st...
  class TestOpenAIMixinSafetyIdentifierPassing (line 1237) | class TestOpenAIMixinSafetyIdentifierPassing:
    method test_chat_completion_passes_safety_identifier (line 1240) | async def test_chat_completion_passes_safety_identifier(self, mixin, m...
    method test_chat_completion_with_top_p (line 1258) | async def test_chat_completion_with_top_p(self, mixin, mock_client_con...
  class TestOpenAIMixinPromptCacheKey (line 1279) | class TestOpenAIMixinPromptCacheKey:
    method test_chat_completion_with_prompt_cache_key (line 1282) | async def test_chat_completion_with_prompt_cache_key(self, mixin, mock...
  class TestOpenAIMixinServiceTier (line 1303) | class TestOpenAIMixinServiceTier:
    method test_chat_completion_passes_service_tier_to_openai (line 1306) | async def test_chat_completion_passes_service_tier_to_openai(self, mix...
  class TestOpenAIMixinTopLogprobs (line 1327) | class TestOpenAIMixinTopLogprobs:
    method test_chat_completion_with_top_logprobs_value_5 (line 1330) | async def test_chat_completion_with_top_logprobs_value_5(self, mixin, ...
    method test_chat_completion_with_top_logprobs_boundary_min (line 1348) | async def test_chat_completion_with_top_logprobs_boundary_min(self, mi...
    method test_chat_completion_with_top_logprobs_boundary_max (line 1366) | async def test_chat_completion_with_top_logprobs_boundary_max(self, mi...
  class TestOpenAIMixinUserProvidedStreamOptions (line 1385) | class TestOpenAIMixinUserProvidedStreamOptions:
    method test_user_stream_options_passed_through_when_telemetry_inactive (line 1388) | async def test_user_stream_options_passed_through_when_telemetry_inact...
    method test_user_stream_options_include_usage_false_overridden_by_telemetry (line 1415) | async def test_user_stream_options_include_usage_false_overridden_by_t...

FILE: tests/unit/providers/utils/inference/test_remote_inference_provider_config.py
  class TestRemoteInferenceProviderConfig (line 31) | class TestRemoteInferenceProviderConfig:
    method test_provider_config_auth_credentials (line 56) | def test_provider_config_auth_credentials(self, monkeypatch, config_cl...

FILE: tests/unit/providers/utils/inference/test_transformers_inference.py
  class ConcreteRerankerImpl (line 23) | class ConcreteRerankerImpl(TransformersInferenceImpl):
    method __init__ (line 24) | def __init__(self):
  class TestExtractText (line 28) | class TestExtractText:
    method test_extract_text_from_string (line 29) | def test_extract_text_from_string(self):
    method test_extract_text_from_text_content_part (line 33) | def test_extract_text_from_text_content_part(self):
    method test_extract_text_from_image_raises (line 38) | def test_extract_text_from_image_raises(self):
  class TestFormatInstruction (line 45) | class TestFormatInstruction:
    method test_format_instruction_basic (line 46) | def test_format_instruction_basic(self):
    method test_format_instruction_default_instruction (line 53) | def test_format_instruction_default_instruction(self):
  class TestRerank (line 59) | class TestRerank:
    method test_rerank_empty_items_rejected_by_validation (line 60) | def test_rerank_empty_items_rejected_by_validation(self):
    method test_rerank_invalid_max_num_results_rejected_by_validation (line 65) | def test_rerank_invalid_max_num_results_rejected_by_validation(self, m...
    method test_rerank_returns_sorted_results (line 71) | async def test_rerank_returns_sorted_results(self, mock_compute, mock_...
    method test_rerank_respects_max_num_results (line 102) | async def test_rerank_respects_max_num_results(self, mock_compute, moc...
    method test_rerank_builds_correct_pairs (line 126) | async def test_rerank_builds_correct_pairs(self, mock_compute, mock_lo...

FILE: tests/unit/providers/utils/memory/test_reranking.py
  class TestNormalizeScores (line 12) | class TestNormalizeScores:
    method test_normalize_scores_basic (line 15) | def test_normalize_scores_basic(self):
    method test_normalize_scores_identical (line 25) | def test_normalize_scores_identical(self):
    method test_normalize_scores_empty (line 33) | def test_normalize_scores_empty(self):
    method test_normalize_scores_single (line 40) | def test_normalize_scores_single(self):
  class TestWeightedRerank (line 48) | class TestWeightedRerank:
    method test_weighted_rerank_basic (line 51) | def test_weighted_rerank_basic(self):
    method test_weighted_rerank_alpha_zero (line 68) | def test_weighted_rerank_alpha_zero(self):
    method test_weighted_rerank_alpha_one (line 79) | def test_weighted_rerank_alpha_one(self):
    method test_weighted_rerank_no_overlap (line 90) | def test_weighted_rerank_no_overlap(self):
  class TestRRFRerank (line 105) | class TestRRFRerank:
    method test_rrf_rerank_basic (line 108) | def test_rrf_rerank_basic(self):
    method test_rrf_rerank_rank_calculation (line 127) | def test_rrf_rerank_rank_calculation(self):
    method test_rrf_rerank_impact_factor (line 144) | def test_rrf_rerank_impact_factor(self):
    method test_rrf_rerank_missing_documents (line 157) | def test_rrf_rerank_missing_documents(self):
  class TestCombineSearchResults (line 172) | class TestCombineSearchResults:
    method test_combine_search_results_rrf_default (line 175) | def test_combine_search_results_rrf_default(self):
    method test_combine_search_results_rrf_explicit (line 186) | def test_combine_search_results_rrf_explicit(self):
    method test_combine_search_results_weighted (line 198) | def test_combine_search_results_weighted(self):
    method test_combine_search_results_unknown_type (line 210) | def test_combine_search_results_unknown_type(self):
    method test_combine_search_results_empty_params (line 223) | def test_combine_search_results_empty_params(self):
    method test_combine_search_results_empty_scores (line 234) | def test_combine_search_results_empty_scores(self):

FILE: tests/unit/providers/utils/memory/test_vector_store.py
  function test_content_from_data_and_mime_type_success_utf8 (line 15) | def test_content_from_data_and_mime_type_success_utf8():
  function test_content_from_data_and_mime_type_error_win1252 (line 29) | def test_content_from_data_and_mime_type_error_win1252():
  function test_content_from_data_and_mime_type_both_encodings_fail (line 43) | def test_content_from_data_and_mime_type_both_encodings_fail():
  function test_memory_tool_error_handling (line 57) | async def test_memory_tool_error_handling():

FILE: tests/unit/providers/utils/test_form_data.py
  class _TestModel (line 18) | class _TestModel(BaseModel):
  function test_parse_pydantic_from_form_bracket_notation (line 25) | async def test_parse_pydantic_from_form_bracket_notation():
  function test_parse_pydantic_from_form_json_string (line 42) | async def test_parse_pydantic_from_form_json_string():
  function test_parse_pydantic_from_form_bracket_takes_precedence (line 59) | async def test_parse_pydantic_from_form_bracket_takes_precedence():
  function test_parse_pydantic_from_form_missing_field (line 78) | async def test_parse_pydantic_from_form_missing_field():
  function test_parse_pydantic_from_form_invalid_json (line 90) | async def test_parse_pydantic_from_form_invalid_json():
  function test_parse_pydantic_from_form_invalid_data (line 104) | async def test_parse_pydantic_from_form_invalid_data():
  function test_parse_expires_after_bracket_notation (line 118) | async def test_parse_expires_after_bracket_notation():
  function test_parse_expires_after_json_string (line 135) | async def test_parse_expires_after_json_string():
  function test_parse_expires_after_missing (line 152) | async def test_parse_expires_after_missing():
  function test_parse_pydantic_from_form_type_conversion (line 164) | async def test_parse_pydantic_from_form_type_conversion():

FILE: tests/unit/providers/utils/test_model_registry.py
  function known_model (line 43) | def known_model() -> Model:
  function known_model2 (line 52) | def known_model2() -> Model:
  function known_provider_model (line 61) | def known_provider_model(known_model: Model) -> ProviderModelEntry:
  function known_provider_model2 (line 69) | def known_provider_model2(known_model2: Model) -> ProviderModelEntry:
  function unknown_model (line 77) | def unknown_model() -> Model:
  function helper (line 86) | def helper(known_provider_model: ProviderModelEntry, known_provider_mode...
  class MockModelRegistryHelperWithDynamicModels (line 90) | class MockModelRegistryHelperWithDynamicModels(ModelRegistryHelper):
    method __init__ (line 93) | def __init__(self, model_entries: list[ProviderModelEntry], available_...
    method check_model_availability (line 97) | async def check_model_availability(self, model: str) -> bool:
  function dynamic_model (line 102) | def dynamic_model() -> Model:
  function helper_with_dynamic_models (line 112) | def helper_with_dynamic_models(
  function test_lookup_unknown_model (line 121) | async def test_lookup_unknown_model(helper: ModelRegistryHelper, unknown...
  function test_register_unknown_provider_model (line 125) | async def test_register_unknown_provider_model(helper: ModelRegistryHelp...
  function test_register_model (line 130) | async def test_register_model(helper: ModelRegistryHelper, known_model: ...
  function test_register_model_from_alias (line 141) | async def test_register_model_from_alias(helper: ModelRegistryHelper, kn...
  function test_register_model_existing (line 152) | async def test_register_model_existing(helper: ModelRegistryHelper, know...
  function test_register_model_existing_different (line 157) | async def test_register_model_existing_different(
  function test_register_model_during_init (line 179) | async def test_register_model_during_init(helper: ModelRegistryHelper, k...
  function test_register_model_from_check_model_availability (line 190) | async def test_register_model_from_check_model_availability(
  function test_register_model_not_in_static_or_dynamic (line 211) | async def test_register_model_not_in_static_or_dynamic(
  function test_register_alias_for_dynamic_model (line 231) | async def test_register_alias_for_dynamic_model(

FILE: tests/unit/providers/utils/test_openai_compat_conversion.py
  class TestSimpleSchemaConversion (line 15) | class TestSimpleSchemaConversion:
    method test_simple_tool_conversion (line 18) | def test_simple_tool_conversion(self):
    method test_tool_without_description (line 46) | def test_tool_without_description(self):
  class TestComplexSchemaConversion (line 58) | class TestComplexSchemaConversion:
    method test_schema_with_refs_and_defs (line 61) | def test_schema_with_refs_and_defs(self):
    method test_anyof_schema_conversion (line 118) | def test_anyof_schema_conversion(self):
    method test_nested_objects_conversion (line 142) | def test_nested_objects_conversion(self):
    method test_array_schemas_with_constraints (line 176) | def test_array_schemas_with_constraints(self):
  class TestOutputSchemaHandling (line 206) | class TestOutputSchemaHandling:
    method test_output_schema_is_dropped (line 209) | def test_output_schema_is_dropped(self):
    method test_only_output_schema_no_input (line 228) | def test_only_output_schema_no_input(self):
  class TestEdgeCases (line 240) | class TestEdgeCases:
    method test_tool_with_no_schemas (line 243) | def test_tool_with_no_schemas(self):
    method test_empty_input_schema (line 254) | def test_empty_input_schema(self):
    method test_schema_with_additional_properties (line 264) | def test_schema_with_additional_properties(self):
    method test_schema_with_pattern_properties (line 277) | def test_schema_with_pattern_properties(self):
    method test_schema_identity (line 286) | def test_schema_identity(self):
  class TestConversionConsistency (line 310) | class TestConversionConsistency:
    method test_multiple_tools_with_shared_defs (line 313) | def test_multiple_tools_with_shared_defs(self):
    method test_conversion_is_pure (line 337) | def test_conversion_is_pure(self):

FILE: tests/unit/providers/utils/test_openai_mixin_streaming.py
  class MockAsyncStream (line 25) | class MockAsyncStream:
    method __init__ (line 28) | def __init__(self, chunks):
    method __aiter__ (line 32) | def __aiter__(self):
    method __anext__ (line 35) | async def __anext__(self):
    method close (line 41) | async def close(self):
  class MockChunk (line 45) | class MockChunk:
    method __init__ (line 46) | def __init__(self, chunk_id: str, content: str = "test"):
  class OpenAIMixinTestImpl (line 51) | class OpenAIMixinTestImpl(OpenAIMixin):
    method get_api_key (line 54) | def get_api_key(self) -> str:
    method get_base_url (line 57) | def get_base_url(self) -> str:
  function mixin (line 62) | def mixin():
  class TestIssue3185Regression (line 69) | class TestIssue3185Regression:
    method test_streaming_result_has_aclose (line 70) | async def test_streaming_result_has_aclose(self, mixin):
    method test_streaming_yields_all_chunks (line 81) | async def test_streaming_yields_all_chunks(self, mixin):
    method test_non_streaming_returns_directly (line 92) | async def test_non_streaming_returns_directly(self, mixin):
  class TestIdOverwriting (line 102) | class TestIdOverwriting:
    method test_ids_overwritten_when_enabled (line 103) | async def test_ids_overwritten_when_enabled(self):
    method test_ids_preserved_when_disabled (line 115) | async def test_ids_preserved_when_disabled(self):

FILE: tests/unit/providers/utils/test_safety_mixin.py
  function test_providers_use_mixin (line 41) | def test_providers_use_mixin(provider_class):
  function test_safe_content (line 50) | async def test_safe_content():
  function test_unsafe_content (line 64) | async def test_unsafe_content():
  function test_missing_violation_type_defaults_to_unsafe (line 87) | async def test_missing_violation_type_defaults_to_unsafe():
  function test_non_string_violation_type_defaults_to_unsafe (line 108) | async def test_non_string_violation_type_defaults_to_unsafe():
  function test_multiple_inputs (line 129) | async def test_multiple_inputs():
  function test_run_shield_receives_correct_params (line 156) | async def test_run_shield_receives_correct_params():
  function test_model_none_raises_error (line 176) | async def test_model_none_raises_error():

FILE: tests/unit/providers/utils/test_scheduler.py
  function test_scheduler_unknown_backend (line 14) | async def test_scheduler_unknown_backend():
  function wait_for_job_completed (line 19) | async def wait_for_job_completed(sched: Scheduler, job_id: str) -> None:
  function test_scheduler_naive (line 28) | async def test_scheduler_naive():
  function test_scheduler_naive_handler_raises (line 88) | async def test_scheduler_naive_handler_raises():

FILE: tests/unit/providers/vector_io/conftest.py
  function vector_provider (line 31) | def vector_provider(request):
  function vector_store_id (line 36) | def vector_store_id() -> str:
  function embedding_dimension (line 41) | def embedding_dimension() -> int:
  function sample_chunks (line 46) | def sample_chunks():
  function sample_chunks_with_metadata (line 92) | def sample_chunks_with_metadata():
  function sample_embeddings (line 113) | def sample_embeddings(sample_chunks):
  function sample_embeddings_with_metadata (line 119) | def sample_embeddings_with_metadata(sample_chunks_with_metadata):
  function mock_inference_api (line 125) | def mock_inference_api(embedding_dimension):
  function unique_kvstore_config (line 134) | async def unique_kvstore_config(tmp_path_factory):
  function sqlite_vec_db_path (line 145) | def sqlite_vec_db_path(tmp_path_factory):
  function sqlite_vec_vec_index (line 151) | async def sqlite_vec_vec_index(embedding_dimension, tmp_path_factory):
  function sqlite_vec_adapter (line 163) | async def sqlite_vec_adapter(sqlite_vec_db_path, unique_kvstore_config, ...
  function faiss_vec_db_path (line 189) | def faiss_vec_db_path(tmp_path_factory):
  function faiss_vec_index (line 195) | async def faiss_vec_index(embedding_dimension):
  function faiss_vec_adapter (line 202) | async def faiss_vec_adapter(unique_kvstore_config, mock_inference_api, e...
  function mock_psycopg2_connection (line 225) | def mock_psycopg2_connection():
  function pgvector_vec_index (line 238) | async def pgvector_vec_index(embedding_dimension, mock_psycopg2_connecti...
  function pgvector_vec_adapter (line 281) | async def pgvector_vec_adapter(unique_kvstore_config, mock_inference_api...
  function qdrant_vec_index (line 348) | async def qdrant_vec_index(embedding_dimension):
  function qdrant_vec_adapter (line 394) | async def qdrant_vec_adapter(unique_kvstore_config, mock_inference_api, ...
  function vector_io_adapter (line 452) | def vector_io_adapter(vector_provider, request):
  function vector_index (line 463) | def vector_index(vector_provider, request):

FILE: tests/unit/providers/vector_io/test_contextual_retrieval.py
  class MockInferenceAPI (line 35) | class MockInferenceAPI:
    method __init__ (line 36) | def __init__(self):
  class MockVectorStoreProvider (line 40) | class MockVectorStoreProvider(OpenAIVectorStoreMixin):
    method __init__ (line 41) | def __init__(self, inference_api):
    method register_vector_store (line 45) | async def register_vector_store(self, vector_store) -> None:
    method unregister_vector_store (line 48) | async def unregister_vector_store(self, vector_store_id) -> None:
    method insert_chunks (line 51) | async def insert_chunks(self, request: InsertChunksRequest) -> None:
    method query_chunks (line 54) | async def query_chunks(self, request: QueryChunksRequest) -> QueryChun...
    method delete_chunks (line 57) | async def delete_chunks(self, request: DeleteChunksRequest) -> None:
  function inference_api (line 62) | def inference_api():
  function provider (line 68) | def provider(inference_api):
  function strategy_config (line 74) | def strategy_config():
  function provider_with_model (line 82) | def provider_with_model(inference_api):
  function fast_retry (line 94) | def fast_retry(monkeypatch):
  function create_mock_response (line 102) | def create_mock_response(content):
  class _RetriableError (line 113) | class _RetriableError(Exception):
    method __init__ (line 116) | def __init__(self, message: str, status_code: int):
  function create_chunk (line 121) | def create_chunk(content, chunk_id="chunk_1", document_id="doc_1"):
  function test_execute_contextual_chunk_transformation_success (line 128) | async def test_execute_contextual_chunk_transformation_success(inference...
  function test_execute_contextual_chunk_transformation_list_content (line 140) | async def test_execute_contextual_chunk_transformation_list_content(infe...
  function test_execute_contextual_chunk_transformation_partial_failure (line 165) | async def test_execute_contextual_chunk_transformation_partial_failure(i...
  function test_execute_contextual_chunk_transformation_total_failure (line 183) | async def test_execute_contextual_chunk_transformation_total_failure(inf...
  function test_execute_contextual_chunk_transformation_custom_prompt (line 193) | async def test_execute_contextual_chunk_transformation_custom_prompt(inf...
  function test_execute_contextual_chunk_transformation_empty_response (line 215) | async def test_execute_contextual_chunk_transformation_empty_response(in...
  function test_execute_contextual_chunk_transformation_timeout (line 227) | async def test_execute_contextual_chunk_transformation_timeout(inference...
  function test_execute_contextual_chunk_transformation_all_empty_context_raises (line 242) | async def test_execute_contextual_chunk_transformation_all_empty_context...
  function test_execute_contextual_chunk_transformation_partial_empty_context (line 255) | async def test_execute_contextual_chunk_transformation_partial_empty_con...
  function test_contextual_config_validation_model_id_required (line 274) | def test_contextual_config_validation_model_id_required():
  function test_contextual_config_validation_overlap_less_than_size (line 280) | def test_contextual_config_validation_overlap_less_than_size():
  function test_contextual_config_validation_prompt_placeholders (line 290) | def test_contextual_config_validation_prompt_placeholders():
  function test_contextual_config_validation_prompt_ordering (line 299) | def test_contextual_config_validation_prompt_ordering():
  function test_execute_contextual_chunk_transformation_document_too_large (line 308) | async def test_execute_contextual_chunk_transformation_document_too_larg...
  function test_execute_contextual_chunk_transformation_model_fallback (line 318) | async def test_execute_contextual_chunk_transformation_model_fallback(in...
  function test_execute_contextual_chunk_transformation_retries_on_retriable_error (line 330) | async def test_execute_contextual_chunk_transformation_retries_on_retria...
  function test_execute_contextual_chunk_transformation_no_retry_on_non_retriable (line 346) | async def test_execute_contextual_chunk_transformation_no_retry_on_non_r...
  function test_execute_contextual_chunk_transformation_retry_exhaustion (line 359) | async def test_execute_contextual_chunk_transformation_retry_exhaustion(...

FILE: tests/unit/providers/vector_io/test_faiss.py
  function loop (line 33) | def loop():
  function embedding_dimension (line 38) | def embedding_dimension():
  function vector_store_id (line 43) | def vector_store_id():
  function sample_chunks (line 48) | def sample_chunks():
  function sample_embeddings (line 85) | def sample_embeddings(embedding_dimension):
  function mock_vector_store (line 90) | def mock_vector_store(vector_store_id, embedding_dimension) -> MagicMock:
  function mock_files_api (line 99) | def mock_files_api():
  function faiss_config (line 105) | def faiss_config():
  function faiss_index (line 112) | async def faiss_index(embedding_dimension):
  function test_faiss_query_vector_returns_infinity_when_query_and_embedding_are_identical (line 117) | async def test_faiss_query_vector_returns_infinity_when_query_and_embedd...
  function test_meta_index_populated_on_add_chunks (line 152) | async def test_meta_index_populated_on_add_chunks(faiss_index, sample_ch...
  function test_meta_index_updated_on_delete_chunks (line 176) | async def test_meta_index_updated_on_delete_chunks(faiss_index, sample_c...
  function test_resolve_filter_positions_eq (line 202) | async def test_resolve_filter_positions_eq(faiss_index, sample_chunks, s...
  function test_resolve_filter_positions_in_nin (line 227) | async def test_resolve_filter_positions_in_nin(faiss_index, sample_chunk...
  function test_resolve_filter_positions_compound (line 256) | async def test_resolve_filter_positions_compound(faiss_index, sample_chu...
  function test_query_vector_with_filter_returns_correct_chunks (line 299) | async def test_query_vector_with_filter_returns_correct_chunks(embedding...
  function test_health_success (line 343) | async def test_health_success():
  function test_health_failure (line 368) | async def test_health_failure():

FILE: tests/unit/providers/vector_io/test_sqlite_vec.py
  function loop (line 33) | def loop():
  function sqlite_vec_index (line 38) | async def sqlite_vec_index(embedding_dimension, tmp_path_factory):
  function test_query_chunk_metadata (line 46) | async def test_query_chunk_metadata(sqlite_vec_index, sample_chunks_with...
  function test_query_chunks_full_text_search (line 65) | async def test_query_chunks_full_text_search(sqlite_vec_index, sample_ch...
  function test_query_chunks_hybrid (line 95) | async def test_query_chunks_hybrid(sqlite_vec_index, sample_chunks, samp...
  function test_query_chunks_full_text_search_k_greater_than_results (line 129) | async def test_query_chunks_full_text_search_k_greater_than_results(sqli...
  function test_chunk_id_conflict (line 155) | async def test_chunk_id_conflict(sqlite_vec_index, sample_chunks, embedd...
  function sqlite_vec_adapter (line 190) | async def sqlite_vec_adapter(sqlite_connection):
  function test_query_chunks_hybrid_no_keyword_matches (line 198) | async def test_query_chunks_hybrid_no_keyword_matches(sqlite_vec_index, ...
  function test_query_chunks_hybrid_score_threshold (line 239) | async def test_query_chunks_hybrid_score_threshold(sqlite_vec_index, sam...
  function test_query_chunks_hybrid_different_embedding (line 273) | async def test_query_chunks_hybrid_different_embedding(
  function test_query_chunks_hybrid_rrf_ranking (line 311) | async def test_query_chunks_hybrid_rrf_ranking(sqlite_vec_index, sample_...
  function test_query_chunks_hybrid_score_selection (line 348) | async def test_query_chunks_hybrid_score_selection(sqlite_vec_index, sam...
  function test_query_chunks_hybrid_mixed_results (line 408) | async def test_query_chunks_hybrid_mixed_results(sqlite_vec_index, sampl...
  function test_query_chunks_hybrid_weighted_reranker_parametrization (line 452) | async def test_query_chunks_hybrid_weighted_reranker_parametrization(
  function test_query_chunks_hybrid_rrf_impact_factor (line 539) | async def test_query_chunks_hybrid_rrf_impact_factor(sqlite_vec_index, s...
  function test_query_chunks_hybrid_edge_cases (line 583) | async def test_query_chunks_hybrid_edge_cases(sqlite_vec_index, sample_c...
  function test_query_chunks_hybrid_tie_breaking (line 639) | async def test_query_chunks_hybrid_tie_breaking(

FILE: tests/unit/providers/vector_io/test_vector_io_openai_vector_stores.py
  function mock_resume_file_batches (line 41) | def mock_resume_file_batches(request):
  function test_initialize_index (line 50) | async def test_initialize_index(vector_index):
  function test_add_chunks_query_vector (line 54) | async def test_add_chunks_query_vector(vector_index, sample_chunks, samp...
  function test_chunk_id_conflict (line 76) | async def test_chunk_id_conflict(vector_index, sample_chunks, embedding_...
  function test_initialize_adapter_with_existing_kvstore (line 102) | async def test_initialize_adapter_with_existing_kvstore(vector_io_adapter):
  function test_persistence_across_adapter_restarts (line 112) | async def test_persistence_across_adapter_restarts(vector_io_adapter):
  function test_vector_store_lazy_loading_from_kvstore (line 125) | async def test_vector_store_lazy_loading_from_kvstore(vector_io_adapter):
  function test_vector_store_preloading_on_initialization (line 158) | async def test_vector_store_preloading_on_initialization(vector_io_adapt...
  function test_kvstore_none_raises_runtime_error (line 194) | async def test_kvstore_none_raises_runtime_error(vector_io_adapter):
  function test_register_and_unregister_vector_store (line 218) | async def test_register_and_unregister_vector_store(vector_io_adapter):
  function test_query_unregistered_raises (line 230) | async def test_query_unregistered_raises(vector_io_adapter, vector_provi...
  function test_insert_chunks_calls_underlying_index (line 236) | async def test_insert_chunks_calls_underlying_index(vector_io_adapter, s...
  function test_insert_chunks_missing_db_raises (line 260) | async def test_insert_chunks_missing_db_raises(vector_io_adapter):
  function test_insert_chunks_with_missing_document_id (line 268) | async def test_insert_chunks_with_missing_document_id(vector_io_adapter):
  function test_document_id_with_invalid_type_raises_error (line 347) | async def test_document_id_with_invalid_type_raises_error():
  function test_query_chunks_calls_underlying_index_and_returns (line 375) | async def test_query_chunks_calls_underlying_index_and_returns(vector_io...
  function test_query_chunks_missing_db_raises (line 414) | async def test_query_chunks_missing_db_raises(vector_io_adapter):
  function test_save_openai_vector_store (line 422) | async def test_save_openai_vector_store(vector_io_adapter):
  function test_update_openai_vector_store (line 438) | async def test_update_openai_vector_store(vector_io_adapter):
  function test_delete_openai_vector_store (line 454) | async def test_delete_openai_vector_store(vector_io_adapter):
  function test_load_openai_vector_stores (line 469) | async def test_load_openai_vector_stores(vector_io_adapter):
  function test_save_openai_vector_store_file (line 484) | async def test_save_openai_vector_store_file(vector_io_adapter, tmp_path...
  function test_update_openai_vector_store_file (line 505) | async def test_update_openai_vector_store_file(vector_io_adapter, tmp_pa...
  function test_load_openai_vector_store_file_contents (line 538) | async def test_load_openai_vector_store_file_contents(vector_io_adapter,...
  function test_delete_openai_vector_store_file_from_storage (line 561) | async def test_delete_openai_vector_store_file_from_storage(vector_io_ad...
  function test_create_vector_store_file_batch (line 587) | async def test_create_vector_store_file_batch(vector_io_adapter):
  function test_retrieve_vector_store_file_batch (line 615) | async def test_retrieve_vector_store_file_batch(vector_io_adapter):
  function test_cancel_vector_store_file_batch (line 646) | async def test_cancel_vector_store_file_batch(vector_io_adapter):
  function test_list_files_in_vector_store_file_batch (line 677) | async def test_list_files_in_vector_store_file_batch(vector_io_adapter):
  function test_file_batch_validation_errors (line 724) | async def test_file_batch_validation_errors(vector_io_adapter):
  function test_file_batch_pagination (line 761) | async def test_file_batch_pagination(vector_io_adapter):
  function test_file_batch_status_filtering (line 832) | async def test_file_batch_status_filtering(vector_io_adapter):
  function test_cancel_completed_batch_fails (line 891) | async def test_cancel_completed_batch_fails(vector_io_adapter):
  function test_file_batch_persistence_across_restarts (line 923) | async def test_file_batch_persistence_across_restarts(vector_io_adapter):
  function test_cancelled_batch_persists_in_storage (line 978) | async def test_cancelled_batch_persists_in_storage(vector_io_adapter):
  function test_only_in_progress_batches_resumed (line 1026) | async def test_only_in_progress_batches_resumed(vector_io_adapter):
  function test_cleanup_expired_file_batches (line 1085) | async def test_cleanup_expired_file_batches(vector_io_adapter):
  function test_expired_batch_access_error (line 1156) | async def test_expired_batch_access_error(vector_io_adapter):
  function test_max_concurrent_files_per_batch (line 1190) | async def test_max_concurrent_files_per_batch(vector_io_adapter):
  function test_embedding_config_from_metadata (line 1239) | async def test_embedding_config_from_metadata(vector_io_adapter):
  function test_embedding_config_from_extra_body (line 1263) | async def test_embedding_config_from_extra_body(vector_io_adapter):
  function test_embedding_config_consistency_check_passes (line 1287) | async def test_embedding_config_consistency_check_passes(vector_io_adapt...
  function test_embedding_config_defaults_when_missing (line 1314) | async def test_embedding_config_defaults_when_missing(vector_io_adapter):
  function test_embedding_config_required_model_missing (line 1337) | async def test_embedding_config_required_model_missing(vector_io_adapter):
  function test_search_vector_store_ignores_rewrite_query (line 1352) | async def test_search_vector_store_ignores_rewrite_query(vector_io_adapt...
  function test_create_gin_index_executes_correct_sql (line 1394) | async def test_create_gin_index_executes_correct_sql():
  function test_create_gin_index_raises_runtime_error_on_db_error (line 1434) | async def test_create_gin_index_raises_runtime_error_on_db_error():
  function test_gin_index_creation_in_initialize_call (line 1471) | async def test_gin_index_creation_in_initialize_call():
  function test_set_ef_search_called_before_select_in_query_vector (line 1506) | async def test_set_ef_search_called_before_select_in_query_vector(mock_p...
  function test_apply_default_ef_search_for_query_vector (line 1542) | async def test_apply_default_ef_search_for_query_vector(mock_psycopg2_co...

FILE: tests/unit/providers/vector_io/test_vector_utils.py
  function test_generate_chunk_id (line 25) | def test_generate_chunk_id():
  function test_generate_chunk_id_with_window (line 39) | def test_generate_chunk_id_with_window():
  function test_chunk_creation_with_explicit_id (line 62) | def test_chunk_creation_with_explicit_id():
  function test_chunk_with_metadata (line 81) | def test_chunk_with_metadata():
  function test_chunk_serialization (line 100) | def test_chunk_serialization():
  function test_vector_store_file_object_attributes_validation (line 119) | def test_vector_store_file_object_attributes_validation():
  function test_vector_store_file_object_attributes_constraints (line 151) | def test_vector_store_file_object_attributes_constraints():
  function test_load_embedded_chunk_backward_compatibility (line 193) | def test_load_embedded_chunk_backward_compatibility():
  function test_load_embedded_chunk_fallbacks (line 238) | def test_load_embedded_chunk_fallbacks():

FILE: tests/unit/rag/test_rag_query.py
  class TestRagQuery (line 16) | class TestRagQuery:
    method test_query_raises_on_empty_vector_store_ids (line 17) | async def test_query_raises_on_empty_vector_store_ids(self):
    method test_query_chunk_metadata_handling (line 25) | async def test_query_chunk_metadata_handling(self):
    method test_query_raises_incorrect_mode (line 76) | async def test_query_raises_incorrect_mode(self):
    method test_query_accepts_valid_modes (line 80) | async def test_query_accepts_valid_modes(self):
    method test_query_adds_vector_store_id_to_chunk_metadata (line 94) | async def test_query_adds_vector_store_id_to_chunk_metadata(self):

FILE: tests/unit/rag/test_vector_store.py
  function read_file (line 40) | def read_file(file_path: str) -> bytes:
  function data_url_from_file (line 45) | def data_url_from_file(file_path: str) -> str:
  class TestChunk (line 57) | class TestChunk:
    method test_chunk (line 58) | def test_chunk(self):
    method test_embedded_chunk (line 75) | def test_embedded_chunk(self):
  class TestValidateEmbedding (line 106) | class TestValidateEmbedding:
    method test_valid_list_embeddings (line 107) | def test_valid_list_embeddings(self):
    method test_valid_numpy_embeddings (line 112) | def test_valid_numpy_embeddings(self):
    method test_invalid_embedding_type (line 118) | def test_invalid_embedding_type(self):
    method test_non_numeric_values (line 130) | def test_non_numeric_values(self):
    method test_wrong_dimension (line 142) | def test_wrong_dimension(self):
  class TestVectorStore (line 153) | class TestVectorStore:
    method test_make_overlapped_chunks (line 161) | def test_make_overlapped_chunks(self, window_len, overlap_len, expecte...
    method test_raise_overlapped_chunks_metadata_serialization_error (line 188) | def test_raise_overlapped_chunks_metadata_serialization_error(self):
  class TestVectorStoreWithIndex (line 208) | class TestVectorStoreWithIndex:
    method test_insert_chunks_with_embedded_chunks (line 209) | async def test_insert_chunks_with_embedded_chunks(self):
    method test_insert_chunks_with_multiple_embedded_chunks (line 258) | async def test_insert_chunks_with_multiple_embedded_chunks(self):
  class TestNeuralRerank (line 331) | class TestNeuralRerank:
    method _make_vector_store_with_index (line 332) | def _make_vector_store_with_index(self, vector_stores_config=None):
    method _make_query_chunks_response (line 351) | def _make_query_chunks_response(self, chunks_data):
    method test_neural_rerank_reorders_chunks (line 376) | async def test_neural_rerank_reorders_chunks(self):
    method test_neural_rerank_uses_default_model_from_config (line 429) | async def test_neural_rerank_uses_default_model_from_config(self):
    method test_neural_rerank_uses_model_from_params (line 465) | async def test_neural_rerank_uses_model_from_params(self):
    method test_neural_rerank_respects_max_chunks_limit (line 501) | async def test_neural_rerank_respects_max_chunks_limit(self):

FILE: tests/unit/registry/test_registry.py
  function sample_vector_store (line 22) | def sample_vector_store():
  function sample_model (line 33) | def sample_model():
  function test_registry_initialization (line 41) | async def test_registry_initialization(disk_dist_registry):
  function test_basic_registration (line 47) | async def test_basic_registration(disk_dist_registry, sample_vector_stor...
  function test_cached_registry_initialization (line 65) | async def test_cached_registry_initialization(sqlite_kvstore, sample_vec...
  function test_cached_registry_updates (line 90) | async def test_cached_registry_updates(cached_disk_dist_registry):
  function test_duplicate_provider_registration_conflict (line 120) | async def test_duplicate_provider_registration_conflict(cached_disk_dist...
  function test_get_all_objects (line 147) | async def test_get_all_objects(cached_disk_dist_registry):
  function test_parse_registry_values_error_handling (line 179) | async def test_parse_registry_values_error_handling(sqlite_kvstore):
  function test_cached_registry_error_handling (line 217) | async def test_cached_registry_error_handling(sqlite_kvstore):
  function test_double_registration_identical_objects (line 247) | async def test_double_registration_identical_objects(disk_dist_registry):
  function test_double_registration_conflicting_objects (line 272) | async def test_double_registration_conflicting_objects(disk_dist_registry):
  function test_restart_registration_with_owner_mismatch (line 304) | async def test_restart_registration_with_owner_mismatch(disk_dist_regist...
  function test_double_registration_with_cache_conflict (line 345) | async def test_double_registration_with_cache_conflict(cached_disk_dist_...
  function test_multi_worker_cache_synchronization (line 383) | async def test_multi_worker_cache_synchronization(sqlite_kvstore, sample...
  function test_multi_worker_get_all_synchronization (line 427) | async def test_multi_worker_get_all_synchronization(sqlite_kvstore, samp...

FILE: tests/unit/registry/test_registry_acl.py
  function test_registry_cache_with_acl (line 13) | async def test_registry_cache_with_acl(cached_disk_dist_registry):
  function test_registry_empty_acl (line 48) | async def test_registry_empty_acl(cached_disk_dist_registry):
  function test_registry_serialization (line 84) | async def test_registry_serialization(cached_disk_dist_registry):

FILE: tests/unit/server/test_access_control.py
  class AsyncMock (line 20) | class AsyncMock(MagicMock):
    method __call__ (line 21) | async def __call__(self, *args, **kwargs):
  function _return_model (line 25) | def _return_model(model):
  function test_setup (line 30) | async def test_setup(cached_disk_dist_registry):
  function test_access_control_with_cache (line 44) | async def test_access_control_with_cache(mock_get_authenticated_user, te...
  function test_access_control_and_updates (line 108) | async def test_access_control_and_updates(mock_get_authenticated_user, t...
  function test_access_control_empty_attributes (line 146) | async def test_access_control_empty_attributes(mock_get_authenticated_us...
  function test_no_user_attributes (line 170) | async def test_no_user_attributes(mock_get_authenticated_user, test_setup):
  function test_automatic_access_attributes (line 200) | async def test_automatic_access_attributes(mock_get_authenticated_user, ...
  function test_setup_with_access_policy (line 244) | async def test_setup_with_access_policy(cached_disk_dist_registry):
  function test_access_policy (line 279) | async def test_access_policy(mock_get_authenticated_user, test_setup_wit...
  function test_permit_when (line 346) | def test_permit_when():
  function test_permit_unless (line 365) | def test_permit_unless():
  function test_forbid_when (line 388) | def test_forbid_when():
  function test_forbid_unless (line 410) | def test_forbid_unless():
  function test_user_has_attribute (line 432) | def test_user_has_attribute():
  function test_user_does_not_have_attribute (line 450) | def test_user_does_not_have_attribute():
  function test_is_owner (line 468) | def test_is_owner():
  function test_is_not_owner (line 487) | def test_is_not_owner():
  function test_invalid_rule_permit_and_forbid_both_specified (line 506) | def test_invalid_rule_permit_and_forbid_both_specified():
  function test_invalid_rule_neither_permit_or_forbid_specified (line 517) | def test_invalid_rule_neither_permit_or_forbid_specified():
  function test_invalid_rule_when_and_unless_both_specified (line 526) | def test_invalid_rule_when_and_unless_both_specified():
  function test_invalid_condition (line 537) | def test_invalid_condition():
  function test_condition_reprs (line 558) | def test_condition_reprs(condition):
  function test_regex_resource_matching (line 564) | def test_regex_resource_matching():
  function test_access_policy_with_regex (line 585) | def test_access_policy_with_regex():
  function restricted_user (line 625) | def restricted_user():
  function admin_user (line 631) | def admin_user():
  function rbac_policy (line 637) | def rbac_policy():
  class TestInferenceRouterRBACBypass (line 655) | class TestInferenceRouterRBACBypass:
    method mock_routing_table (line 659) | def mock_routing_table(self):
    method test_registry_path_and_fallback_path_consistent (line 667) | async def test_registry_path_and_fallback_path_consistent(
  class TestModelListingRBACBypass (line 718) | class TestModelListingRBACBypass:
    method test_dynamic_models_respect_rbac (line 725) | async def test_dynamic_models_respect_rbac(
  function test_invalid_regex_pattern_in_access_policy_logs_warning (line 799) | def test_invalid_regex_pattern_in_access_policy_logs_warning(caplog):

FILE: tests/unit/server/test_auth.py
  function suppress_auth_errors (line 31) | def suppress_auth_errors(caplog):
  class MockResponse (line 37) | class MockResponse:
    method __init__ (line 38) | def __init__(self, status_code, json_data):
    method json (line 42) | def json(self):
    method raise_for_status (line 45) | def raise_for_status(self):
  function mock_auth_endpoint (line 51) | def mock_auth_endpoint():
  function valid_api_key (line 56) | def valid_api_key():
  function invalid_api_key (line 61) | def invalid_api_key():
  function valid_token (line 66) | def valid_token():
  function invalid_token (line 71) | def invalid_token():
  function http_app (line 76) | def http_app(mock_auth_endpoint):
  function http_client (line 95) | def http_client(http_app):
  function mock_scope (line 100) | def mock_scope():
  function mock_http_middleware (line 114) | def mock_http_middleware(mock_auth_endpoint):
  function mock_impls (line 127) | def mock_impls():
  function middleware_with_mocks (line 133) | def middleware_with_mocks(mock_auth_endpoint):
  function mock_post_success (line 170) | async def mock_post_success(*args, **kwargs):
  function mock_post_failure (line 186) | async def mock_post_failure(*args, **kwargs):
  function mock_post_exception (line 190) | async def mock_post_exception(*args, **kwargs):
  function test_missing_auth_header (line 195) | def test_missing_auth_header(http_client):
  function test_invalid_auth_header_format (line 202) | def test_invalid_auth_header_format(http_client):
  function test_valid_http_authentication (line 209) | def test_valid_http_authentication(http_client, valid_api_key):
  function test_invalid_http_authentication (line 216) | def test_invalid_http_authentication(http_client, invalid_api_key, suppr...
  function test_http_auth_service_error (line 223) | def test_http_auth_service_error(http_client, valid_api_key, suppress_au...
  function test_http_auth_request_payload (line 229) | def test_http_auth_request_payload(http_client, valid_api_key, mock_auth...
  function test_http_middleware_with_access_attributes (line 258) | async def test_http_middleware_with_access_attributes(mock_http_middlewa...
  function oauth2_app (line 296) | def oauth2_app():
  function oauth2_client (line 318) | def oauth2_client(oauth2_app):
  function test_missing_auth_header_oauth2 (line 322) | def test_missing_auth_header_oauth2(oauth2_client):
  function test_invalid_auth_header_format_oauth2 (line 329) | def test_invalid_auth_header_format_oauth2(oauth2_client):
  function jwt_token_valid (line 336) | def jwt_token_valid():
  function mock_jwks_urlopen (line 353) | def mock_jwks_urlopen():
  function mock_jwks_urlopen_with_auth_required (line 376) | def mock_jwks_urlopen_with_auth_required():
  function test_valid_oauth2_authentication (line 417) | def test_valid_oauth2_authentication(oauth2_client, jwt_token_valid, moc...
  function test_invalid_oauth2_authentication (line 423) | def test_invalid_oauth2_authentication(oauth2_client, invalid_token, moc...
  function oauth2_app_with_jwks_token (line 430) | def oauth2_app_with_jwks_token():
  function oauth2_client_with_jwks_token (line 454) | def oauth2_client_with_jwks_token(oauth2_app_with_jwks_token):
  function test_oauth2_with_jwks_token_expected (line 458) | def test_oauth2_with_jwks_token_expected(
  function test_oauth2_with_jwks_token_configured (line 465) | def test_oauth2_with_jwks_token_configured(oauth2_client_with_jwks_token...
  function test_get_attributes_from_claims (line 471) | def test_get_attributes_from_claims():
  function mock_introspection_endpoint (line 593) | def mock_introspection_endpoint():
  function introspection_app (line 598) | def introspection_app(mock_introspection_endpoint):
  function introspection_app_with_custom_mapping (line 621) | def introspection_app_with_custom_mapping(mock_introspection_endpoint):
  function introspection_client (line 651) | def introspection_client(introspection_app):
  function introspection_client_with_custom_mapping (line 656) | def introspection_client_with_custom_mapping(introspection_app_with_cust...
  function test_missing_auth_header_introspection (line 660) | def test_missing_auth_header_introspection(introspection_client):
  function test_invalid_auth_header_format_introspection (line 667) | def test_invalid_auth_header_format_introspection(introspection_client):
  function mock_introspection_active (line 673) | async def mock_introspection_active(*args, **kwargs):
  function mock_introspection_inactive (line 686) | async def mock_introspection_inactive(*args, **kwargs):
  function mock_introspection_invalid (line 695) | async def mock_introspection_invalid(*args, **kwargs):
  function mock_introspection_failed (line 706) | async def mock_introspection_failed(*args, **kwargs):
  function test_valid_introspection_authentication (line 714) | def test_valid_introspection_authentication(introspection_client, valid_...
  function test_inactive_introspection_authentication (line 721) | def test_inactive_introspection_authentication(introspection_client, inv...
  function test_invalid_introspection_authentication (line 728) | def test_invalid_introspection_authentication(introspection_client, inva...
  function test_failed_introspection_authentication (line 735) | def test_failed_introspection_authentication(introspection_client, inval...
  function test_valid_introspection_with_custom_mapping_authentication (line 742) | def test_valid_introspection_with_custom_mapping_authentication(
  function mock_kubernetes_api_server (line 753) | def mock_kubernetes_api_server():
  function kubernetes_auth_app (line 758) | def kubernetes_auth_app(mock_kubernetes_api_server):
  function kubernetes_auth_client (line 782) | def kubernetes_auth_client(kubernetes_auth_app):
  function test_missing_auth_header_kubernetes_auth (line 786) | def test_missing_auth_header_kubernetes_auth(kubernetes_auth_client):
  function test_invalid_auth_header_format_kubernetes_auth (line 792) | def test_invalid_auth_header_format_kubernetes_auth(kubernetes_auth_clie...
  function mock_kubernetes_selfsubjectreview_success (line 798) | async def mock_kubernetes_selfsubjectreview_success(*args, **kwargs):
  function mock_kubernetes_selfsubjectreview_failure (line 817) | async def mock_kubernetes_selfsubjectreview_failure(*args, **kwargs):
  function mock_kubernetes_selfsubjectreview_http_error (line 821) | async def mock_kubernetes_selfsubjectreview_http_error(*args, **kwargs):
  function test_valid_kubernetes_auth_authentication (line 826) | def test_valid_kubernetes_auth_authentication(kubernetes_auth_client, va...
  function test_invalid_kubernetes_auth_authentication (line 833) | def test_invalid_kubernetes_auth_authentication(kubernetes_auth_client, ...
  function test_kubernetes_auth_http_error (line 840) | def test_kubernetes_auth_http_error(kubernetes_auth_client, valid_token,...
  function test_kubernetes_auth_request_payload (line 846) | def test_kubernetes_auth_request_payload(
  function test_unauthenticated_endpoint_access_health (line 887) | async def test_unauthenticated_endpoint_access_health(middleware_with_mo...
  function test_unauthenticated_endpoint_denied_for_other_paths (line 906) | async def test_unauthenticated_endpoint_denied_for_other_paths(middlewar...

FILE: tests/unit/server/test_auth_github.py
  function suppress_auth_errors (line 20) | def suppress_auth_errors(caplog):
  class MockResponse (line 26) | class MockResponse:
    method __init__ (line 27) | def __init__(self, status_code, json_data):
    method json (line 31) | def json(self):
    method raise_for_status (line 34) | def raise_for_status(self):
  function github_token_app (line 42) | def github_token_app():
  function github_token_client (line 70) | def github_token_client(github_token_app):
  function test_authenticated_endpoint_without_token (line 74) | def test_authenticated_endpoint_without_token(github_token_client):
  function test_authenticated_endpoint_with_invalid_bearer_format (line 82) | def test_authenticated_endpoint_with_invalid_bearer_format(github_token_...
  function test_authenticated_endpoint_with_valid_github_token (line 90) | def test_authenticated_endpoint_with_valid_github_token(mock_client_clas...
  function test_authenticated_endpoint_with_invalid_github_token (line 130) | def test_authenticated_endpoint_with_invalid_github_token(mock_client_cl...
  function test_github_enterprise_support (line 147) | def test_github_enterprise_support(mock_client_class):
  function test_github_token_auth_error_message_format (line 199) | def test_github_token_auth_error_message_format(github_token_client):

FILE: tests/unit/server/test_cors.py
  function test_cors_config_defaults (line 12) | def test_cors_config_defaults():
  function test_cors_config_explicit_config (line 24) | def test_cors_config_explicit_config():
  function test_cors_config_regex (line 35) | def test_cors_config_regex():
  function test_cors_config_wildcard_credentials_error (line 42) | def test_cors_config_wildcard_credentials_error():
  function test_process_cors_config_false (line 50) | def test_process_cors_config_false():
  function test_process_cors_config_true (line 55) | def test_process_cors_config_true():
  function test_process_cors_config_passthrough (line 67) | def test_process_cors_config_passthrough():
  function test_process_cors_config_invalid_type (line 74) | def test_process_cors_config_invalid_type():
  function test_cors_config_model_dump (line 79) | def test_cors_config_model_dump():

FILE: tests/unit/server/test_quota.py
  function suppress_quota_warnings (line 22) | def suppress_quota_warnings(caplog):
  class InjectClientIDMiddleware (line 27) | class InjectClientIDMiddleware(BaseHTTPMiddleware):
    method __init__ (line 32) | def __init__(self, app, client_id="client1"):
    method dispatch (line 36) | async def dispatch(self, request: Request, call_next):
  function build_quota_config (line 41) | def build_quota_config(db_path) -> QuotaConfig:
  function auth_app (line 53) | def auth_app(tmp_path, request):
  function test_authenticated_quota_allows_up_to_limit (line 80) | def test_authenticated_quota_allows_up_to_limit(auth_app, suppress_quota...
  function test_authenticated_quota_blocks_after_limit (line 86) | def test_authenticated_quota_blocks_after_limit(auth_app, suppress_quota...
  function test_anonymous_quota_allows_up_to_limit (line 95) | def test_anonymous_quota_allows_up_to_limit(tmp_path, request, suppress_...
  function test_anonymous_quota_blocks_after_limit (line 117) | def test_anonymous_quota_blocks_after_limit(tmp_path, request, suppress_...

FILE: tests/unit/server/test_replace_env_vars.py
  function setup_env_vars (line 15) | def setup_env_vars():
  function test_simple_replacement (line 34) | def test_simple_replacement(setup_env_vars):
  function test_simple_replacement_raises_when_not_set (line 38) | def test_simple_replacement_raises_when_not_set(setup_env_vars):
  function test_default_value_when_not_set (line 45) | def test_default_value_when_not_set(setup_env_vars):
  function test_default_value_when_set (line 49) | def test_default_value_when_set(setup_env_vars):
  function test_default_value_when_empty (line 53) | def test_default_value_when_empty(setup_env_vars):
  function test_none_value_when_empty (line 57) | def test_none_value_when_empty(setup_env_vars):
  function test_value_when_set (line 61) | def test_value_when_set(setup_env_vars):
  function test_empty_var_no_default (line 65) | def test_empty_var_no_default(setup_env_vars):
  function test_conditional_value_when_set (line 69) | def test_conditional_value_when_set(setup_env_vars):
  function test_conditional_value_when_not_set (line 73) | def test_conditional_value_when_not_set(setup_env_vars):
  function test_conditional_value_when_empty (line 77) | def test_conditional_value_when_empty(setup_env_vars):
  function test_conditional_value_with_zero (line 81) | def test_conditional_value_with_zero(setup_env_vars):
  function test_mixed_syntax (line 85) | def test_mixed_syntax(setup_env_vars):
  function test_nested_structures (line 90) | def test_nested_structures(setup_env_vars):
  function test_explicit_strings_preserved (line 100) | def test_explicit_strings_preserved(setup_env_vars):
  function test_resource_with_empty_benchmark_id_skipped (line 107) | def test_resource_with_empty_benchmark_id_skipped(setup_env_vars):
  function test_resource_with_set_benchmark_id_not_skipped (line 121) | def test_resource_with_set_benchmark_id_not_skipped(setup_env_vars):
  function test_resource_with_empty_model_id_skipped (line 139) | def test_resource_with_empty_model_id_skipped(setup_env_vars):
  function test_resource_with_empty_shield_id_skipped (line 153) | def test_resource_with_empty_shield_id_skipped(setup_env_vars):
  function test_multiple_resources_with_conditional_ids (line 167) | def test_multiple_resources_with_conditional_ids(setup_env_vars):
  function test_auth_provider_disabled_when_type_not_set (line 190) | def test_auth_provider_disabled_when_type_not_set(setup_env_vars):
  function test_auth_provider_enabled_when_type_is_set (line 211) | def test_auth_provider_enabled_when_type_is_set(setup_env_vars):
  function test_auth_provider_disabled_when_type_is_empty (line 237) | def test_auth_provider_disabled_when_type_is_empty(setup_env_vars):
  function test_auth_provider_with_hardcoded_type (line 255) | def test_auth_provider_with_hardcoded_type(setup_env_vars):
  function test_auth_provider_with_complex_config (line 276) | def test_auth_provider_with_complex_config(setup_env_vars):

FILE: tests/unit/server/test_resolver.py
  function add_protocol_methods (line 32) | def add_protocol_methods(cls: type, protocol: type[Protocol]) -> None:
  class SampleConfig (line 49) | class SampleConfig(BaseModel):
    method sample_run_config (line 56) | def sample_run_config(cls, **kwargs: Any) -> dict[str, Any]:
  class SampleImpl (line 62) | class SampleImpl:
    method __init__ (line 63) | def __init__(self, config: SampleConfig, deps: dict[Api, Any], provide...
    method initialize (line 70) | async def initialize(self):
  function make_run_config (line 74) | def make_run_config(**overrides) -> StackConfig:
  function test_resolve_impls_basic (line 103) | async def test_resolve_impls_basic():

FILE: tests/unit/server/test_route_auth.py
  function admin_user (line 26) | def admin_user():
  function developer_user (line 37) | def developer_user():
  function regular_user (line 48) | def regular_user():
  function create_mock_auth_provider (line 58) | def create_mock_auth_provider(user: User):
  function create_app_with_route_policy (line 65) | def create_app_with_route_policy(route_policy: list[RouteAccessRule], us...
  function test_no_route_policy_allows_all (line 111) | async def test_no_route_policy_allows_all(regular_user):
  function test_exact_path_match (line 149) | async def test_exact_path_match(developer_user):
  function test_wildcard_prefix_match (line 168) | async def test_wildcard_prefix_match():
  function test_full_wildcard_match (line 184) | async def test_full_wildcard_match():
  function test_multiple_paths_in_rule (line 196) | async def test_multiple_paths_in_rule(regular_user):
  function test_condition_evaluation_with_roles (line 215) | async def test_condition_evaluation_with_roles(developer_user, regular_u...
  function test_admin_full_wildcard_access (line 233) | async def test_admin_full_wildcard_access(admin_user, developer_user):
  function test_forbid_rule (line 258) | async def test_forbid_rule(admin_user, developer_user):
  function test_no_matching_rule_denies_access (line 289) | async def test_no_matching_rule_denies_access(regular_user):
  function test_multiple_conditions (line 307) | async def test_multiple_conditions(admin_user):
  function test_rule_order_matters (line 322) | async def test_rule_order_matters(developer_user):
  function test_websocket_passthrough (line 346) | async def test_websocket_passthrough():
  function test_route_blocking_without_auth (line 384) | async def test_route_blocking_without_auth():
  function test_forbid_rule_without_auth (line 415) | async def test_forbid_rule_without_auth():
  function test_rule_with_condition_requires_user (line 442) | async def test_rule_with_condition_requires_user():
  function test_mixed_rules_with_and_without_conditions (line 460) | async def test_mixed_rules_with_and_without_conditions(admin_user, regul...
  function test_regex_path_matching (line 498) | async def test_regex_path_matching():
  function test_route_policy_with_regex (line 524) | async def test_route_policy_with_regex(developer_user, regular_user):
  function test_invalid_regex_pattern_logs_warning (line 559) | async def test_invalid_regex_pattern_logs_warning(caplog):

FILE: tests/unit/server/test_schema_registry.py
  function test_json_schema_registry_contains_known_model (line 20) | def test_json_schema_registry_contains_known_model() -> None:
  function test_registered_schema_registry_contains_sampling_strategy (line 24) | def test_registered_schema_registry_contains_sampling_strategy() -> None:
  function test_dynamic_schema_registration_round_trip (line 33) | def test_dynamic_schema_registration_round_trip() -> None:

FILE: tests/unit/server/test_server.py
  class TestTranslateException (line 18) | class TestTranslateException:
    method test_translate_access_denied_error (line 21) | def test_translate_access_denied_error(self):
    method test_translate_access_denied_error_with_context (line 30) | def test_translate_access_denied_error_with_context(self):
    method test_translate_permission_error (line 57) | def test_translate_permission_error(self):
    method test_translate_value_error (line 66) | def test_translate_value_error(self):
    method test_translate_bad_request_error (line 75) | def test_translate_bad_request_error(self):
    method test_translate_authentication_required_error (line 89) | def test_translate_authentication_required_error(self):
    method test_translate_timeout_error (line 98) | def test_translate_timeout_error(self):
    method test_translate_asyncio_timeout_error (line 107) | def test_translate_asyncio_timeout_error(self):
    method test_translate_connection_error (line 116) | def test_translate_connection_error(self):
    method test_translate_not_implemented_error (line 125) | def test_translate_not_implemented_error(self):
    method test_translate_validation_error (line 134) | def test_translate_validation_error(self):
    method test_translate_generic_exception (line 158) | def test_translate_generic_exception(self):
    method test_translate_runtime_error (line 167) | def test_translate_runtime_error(self):
    method test_multiple_access_denied_scenarios (line 176) | def test_multiple_access_denied_scenarios(self):
  class TestRemoveDisabledProviders (line 199) | class TestRemoveDisabledProviders:
    method test_remove_explicitly_disabled_provider (line 202) | def test_remove_explicitly_disabled_provider(self):
    method test_remove_empty_provider_id (line 216) | def test_remove_empty_provider_id(self):
    method test_keep_models_with_none_provider_model_id (line 230) | def test_keep_models_with_none_provider_model_id(self):

FILE: tests/unit/server/test_sse.py
  function suppress_sse_errors (line 18) | def suppress_sse_errors(caplog):
  function test_sse_generator_basic (line 23) | async def test_sse_generator_basic():
  function test_sse_generator_client_disconnected (line 44) | async def test_sse_generator_client_disconnected():
  function test_sse_generator_client_disconnected_before_response_starts (line 66) | async def test_sse_generator_client_disconnected_before_response_starts():
  function test_sse_generator_error_before_response_starts (line 82) | async def test_sse_generator_error_before_response_starts(suppress_sse_e...
  function test_paginated_response_url_setting (line 99) | async def test_paginated_response_url_setting():

FILE: tests/unit/server/test_test_context_middleware.py
  function app_with_middleware (line 19) | def app_with_middleware():
  function test_mode_env (line 36) | def test_mode_env(monkeypatch):
  function test_middleware_returns_none_without_header (line 42) | def test_middleware_returns_none_without_header(app_with_middleware, tes...
  function test_middleware_extracts_test_id_from_header (line 51) | def test_middleware_extracts_test_id_from_header(app_with_middleware, te...
  function test_middleware_handles_empty_provider_data (line 65) | def test_middleware_handles_empty_provider_data(app_with_middleware, tes...
  function test_middleware_handles_invalid_json (line 78) | def test_middleware_handles_invalid_json(app_with_middleware, test_mode_...
  function test_middleware_noop_without_test_mode (line 91) | def test_middleware_noop_without_test_mode(app_with_middleware):

FILE: tests/unit/server/test_vector_io_fastapi_router.py
  function test_vector_io_router_list_vector_stores (line 17) | def test_vector_io_router_list_vector_stores() -> None:
  function test_vector_io_router_search_vector_store_passes_body_fields (line 36) | def test_vector_io_router_search_vector_store_passes_body_fields() -> None:

FILE: tests/unit/telemetry/test_tool_runtime_metrics.py
  class TestToolMetricAttributes (line 22) | class TestToolMetricAttributes:
    method test_create_tool_metric_attributes_all_fields (line 25) | def test_create_tool_metric_attributes_all_fields(self):
    method test_create_tool_metric_attributes_partial_fields (line 40) | def test_create_tool_metric_attributes_partial_fields(self):
    method test_create_tool_metric_attributes_minimal (line 53) | def test_create_tool_metric_attributes_minimal(self):
    method test_create_tool_metric_attributes_empty (line 62) | def test_create_tool_metric_attributes_empty(self):
  class TestToolMetricInstruments (line 68) | class TestToolMetricInstruments:
    method test_tool_invocations_total_exists (line 71) | def test_tool_invocations_total_exists(self):
    method test_tool_duration_exists (line 76) | def test_tool_duration_exists(self):
    method test_tool_invocations_total_can_record (line 81) | def test_tool_invocations_total_can_record(self):
    method test_tool_duration_can_record (line 92) | def test_tool_duration_can_record(self):
  class TestToolMetricsConstants (line 103) | class TestToolMetricsConstants:
    method test_metric_names_follow_convention (line 106) | def test_metric_names_follow_convention(self):
  class TestToolRuntimeIntegration (line 126) | class TestToolRuntimeIntegration:
    method test_tool_runtime_metrics_success (line 129) | async def test_tool_runtime_metrics_success(self):
    method test_tool_runtime_metrics_error (line 164) | async def test_tool_runtime_metrics_error(self):

FILE: tests/unit/telemetry/test_vector_io_metrics.py
  class TestVectorMetricAttributes (line 27) | class TestVectorMetricAttributes:
    method test_create_vector_metric_attributes_all_fields (line 30) | def test_create_vector_metric_attributes_all_fields(self):
    method test_create_vector_metric_attributes_partial_fields (line 44) | def test_create_vector_metric_attributes_partial_fields(self):
    method test_create_vector_metric_attributes_empty (line 56) | def test_create_vector_metric_attributes_empty(self):
  class TestVectorMetricInstruments (line 61) | class TestVectorMetricInstruments:
    method test_counters_exist (line 64) | def test_counters_exist(self):
    method test_histograms_exist (line 76) | def test_histograms_exist(self):
    method test_counters_can_record (line 81) | def test_counters_can_record(self):
    method test_histograms_can_record (line 96) | def test_histograms_can_record(self):
  class TestVectorMetricsConstants (line 106) | class TestVectorMetricsConstants:
    method test_metric_names_follow_convention (line 109) | def test_metric_names_follow_convention(self):
  class TestVectorIORouterMetricsIntegration (line 139) | class TestVectorIORouterMetricsIntegration:
    method _create_mock_router (line 142) | def _create_mock_router(self):
    method test_insert_chunks_records_metrics (line 147) | async def test_insert_chunks_records_metrics(self):
    method test_insert_chunks_records_error_metrics (line 173) | async def test_insert_chunks_records_error_metrics(self):
    method test_query_chunks_records_metrics (line 193) | async def test_query_chunks_records_metrics(self):
    method test_search_vector_store_records_metrics (line 218) | async def test_search_vector_store_records_metrics(self):
    method test_delete_vector_store_records_metrics (line 244) | async def test_delete_vector_store_records_metrics(self):

FILE: tests/unit/test_eval_models.py
  function test_model_candidate_valid (line 21) | def test_model_candidate_valid():
  function test_benchmark_config_valid (line 30) | def test_benchmark_config_valid():
  function test_evaluate_response_valid (line 40) | def test_evaluate_response_valid():
  function test_run_eval_request_valid (line 54) | def test_run_eval_request_valid():
  function test_evaluate_rows_request_empty_arrays_fail (line 64) | def test_evaluate_rows_request_empty_arrays_fail():

FILE: tests/unit/testing/providers/test_ollama.py
  class TestOllamaCreateError (line 14) | class TestOllamaCreateError:
    method test_reconstructs_response_error_with_status_and_message (line 17) | def test_reconstructs_response_error_with_status_and_message(self):
    method test_error_attribute_matches_input (line 24) | def test_error_attribute_matches_input(self):
    method test_body_parameter_ignored_ollama_has_no_body_attr (line 29) | def test_body_parameter_ignored_ollama_has_no_body_attr(self):

FILE: tests/unit/testing/providers/test_openai.py
  class TestOpenAICreateError (line 25) | class TestOpenAICreateError:
    method test_status_code_maps_to_correct_class (line 51) | def test_status_code_maps_to_correct_class(self, status, expected_class):
    method test_body_preserved_with_parsed_attributes (line 57) | def test_body_preserved_with_parsed_attributes(self):
    method test_unmapped_status_uses_api_status_error_base (line 66) | def test_unmapped_status_uses_api_status_error_base(self):

FILE: tests/unit/testing/providers/test_providers_init.py
  class TestDetectProvider (line 27) | class TestDetectProvider:
    method test_openai_exception_detected (line 30) | def test_openai_exception_detected(self):
    method test_ollama_exception_detected (line 36) | def test_ollama_exception_detected(self):
    method test_unknown_exception_returns_unknown (line 40) | def test_unknown_exception_returns_unknown(self):
  class TestCreateProviderError (line 45) | class TestCreateProviderError:
    method test_openai_reconstructs_specific_type_by_status (line 48) | def test_openai_reconstructs_specific_type_by_status(self):
    method test_openai_unknown_status_falls_back_to_api_status_error (line 55) | def test_openai_unknown_status_falls_back_to_api_status_error(self):
    method test_ollama_reconstructs_response_error (line 61) | def test_ollama_reconstructs_response_error(self):
    method test_unknown_provider_returns_generic_with_status_and_body (line 67) | def test_unknown_provider_returns_generic_with_status_and_body(self):
  class TestProviderRegistration (line 76) | class TestProviderRegistration:
    method _fake_sdk_module (line 80) | def _fake_sdk_module(name: str = "example_sdk") -> types.ModuleType:
    method test_detect_provider_uses_registered_providers (line 84) | def test_detect_provider_uses_registered_providers(self):
    method test_create_provider_error_uses_registered_providers (line 97) | def test_create_provider_error_uses_registered_providers(self):
  class TestProviderValidation (line 122) | class TestProviderValidation:
    method _fake_sdk_module (line 126) | def _fake_sdk_module(name: str = "example_sdk") -> types.ModuleType:
    method test_valid_config_passes (line 129) | def test_valid_config_passes(self):
    method test_empty_name_raises (line 138) | def test_empty_name_raises(self):
    method test_invalid_name_type_raises (line 144) | def test_invalid_name_type_raises(self):
    method test_sdk_module_must_be_a_module (line 150) | def test_sdk_module_must_be_a_module(self):
    method test_non_callable_create_error_raises (line 156) | def test_non_callable_create_error_raises(self):

FILE: tests/unit/testing/test_exception_utils.py
  function _openai_error (line 39) | def _openai_error(cls, status_code, body, message):
  class TestSerializeException (line 46) | class TestSerializeException:
    method test_llama_stack_error_serializes_as_llama_stack_category (line 49) | def test_llama_stack_error_serializes_as_llama_stack_category(self):
    method test_provider_sdk_exception_serializes_with_provider_and_body (line 57) | def test_provider_sdk_exception_serializes_with_provider_and_body(self):
    method test_provider_sdk_uses_error_attr_over_str (line 66) | def test_provider_sdk_uses_error_attr_over_str(self):
    method test_builtin_exception_serializes_by_type_name (line 73) | def test_builtin_exception_serializes_by_type_name(self):
    method test_unknown_exception_serializes_with_type_and_message (line 80) | def test_unknown_exception_serializes_with_type_and_message(self):
  class TestDeserializeException (line 88) | class TestDeserializeException:
    method test_llama_stack_roundtrip_preserves_status_and_message (line 91) | def test_llama_stack_roundtrip_preserves_status_and_message(self):
    method test_builtin_roundtrip_reconstructs_exact_type (line 99) | def test_builtin_roundtrip_reconstructs_exact_type(self):
    method test_unknown_roundtrip_falls_back_to_generic_exception (line 106) | def test_unknown_roundtrip_falls_back_to_generic_exception(self):
    method test_deserialize_missing_category_defaults_to_unknown (line 113) | def test_deserialize_missing_category_defaults_to_unknown(self):
  class TestProviderSDKRoundTrip (line 120) | class TestProviderSDKRoundTrip:
    method test_openai_roundtrip_preserves_type_and_status (line 150) | def test_openai_roundtrip_preserves_type_and_status(self, cls, status):
    method test_openai_roundtrip_preserves_parsed_attributes (line 156) | def test_openai_roundtrip_preserves_parsed_attributes(self):
    method test_ollama_roundtrip_preserves_type_status_and_error (line 174) | def test_ollama_roundtrip_preserves_type_status_and_error(self, status...
  class TestReconstructedExceptionInterface (line 182) | class TestReconstructedExceptionInterface:
    method test_llama_stack_reconstructed_translates_to_http (line 185) | def test_llama_stack_reconstructed_translates_to_http(self):
    method test_provider_sdk_reconstructed_translates_to_http (line 192) | def test_provider_sdk_reconstructed_translates_to_http(self):
    method test_unknown_provider_uses_generic_but_preserves_status (line 204) | def test_unknown_provider_uses_generic_but_preserves_status(self):
  class TestIsProviderSdkException (line 219) | class TestIsProviderSdkException:
    method test_openai_exception_detected (line 222) | def test_openai_exception_detected(self):
    method test_ollama_exception_detected (line 227) | def test_ollama_exception_detected(self):
    method test_llama_stack_not_detected_as_provider_sdk (line 230) | def test_llama_stack_not_detected_as_provider_sdk(self):
    method test_plain_exception_not_detected (line 236) | def test_plain_exception_not_detected(self):

FILE: tests/unit/tools/test_tools_json_schema.py
  class TestToolDefValidation (line 17) | class TestToolDefValidation:
    method test_simple_input_schema (line 20) | def test_simple_input_schema(self):
    method test_input_and_output_schema (line 37) | def test_input_and_output_schema(self):
    method test_schema_with_refs_and_defs (line 54) | def test_schema_with_refs_and_defs(self):
    method test_output_schema_with_refs (line 87) | def test_output_schema_with_refs(self):
    method test_complex_json_schema_features (line 108) | def test_complex_json_schema_features(self):
    method test_invalid_json_schema_raises_error (line 163) | def test_invalid_json_schema_raises_error(self):
  class TestSchemaEquivalence (line 182) | class TestSchemaEquivalence:
    method test_schema_roundtrip (line 185) | def test_schema_roundtrip(self):
    method test_json_serialization (line 205) | def test_json_serialization(self):
  class TestBackwardsCompatibility (line 229) | class TestBackwardsCompatibility:
    method test_none_schemas (line 232) | def test_none_schemas(self):
    method test_metadata_preserved (line 239) | def test_metadata_preserved(self):

FILE: tests/unit/utils/inference/test_inference_store.py
  class _CollectingExporter (line 27) | class _CollectingExporter(SpanExporter):
    method __init__ (line 30) | def __init__(self):
    method export (line 33) | def export(self, spans):
  function setup_backends (line 39) | def setup_backends(tmp_path):
  function create_test_chat_completion (line 45) | def create_test_chat_completion(
  function test_inference_store_pagination_basic (line 67) | async def test_inference_store_pagination_basic():
  function test_inference_store_pagination_ascending (line 114) | async def test_inference_store_pagination_ascending():
  function test_inference_store_pagination_with_model_filter (line 150) | async def test_inference_store_pagination_with_model_filter():
  function test_inference_store_pagination_invalid_after (line 189) | async def test_inference_store_pagination_invalid_after():
  function test_inference_store_pagination_no_limit (line 200) | async def test_inference_store_pagination_no_limit():
  function test_inference_store_custom_table_name (line 230) | async def test_inference_store_custom_table_name():
  function test_otel_traces_not_leaked_across_requests (line 259) | async def test_otel_traces_not_leaked_across_requests():
  function test_otel_worker_does_not_inherit_first_request_trace (line 333) | async def test_otel_worker_does_not_inherit_first_request_trace():

FILE: tests/unit/utils/inference/test_provider_data_leak.py
  function setup_backends (line 39) | def setup_backends(tmp_path):
  function _set_authenticated_user (line 44) | def _set_authenticated_user(user: User | None):
  function _make_completion (line 52) | def _make_completion(completion_id: str, created: int) -> OpenAIChatComp...
  function test_provider_data_leak_through_write_queue (line 71) | async def test_provider_data_leak_through_write_queue():

FILE: tests/unit/utils/kvstore/test_sqlite_memory.py
  function test_memory_kvstore_persistence_behavior (line 12) | async def test_memory_kvstore_persistence_behavior():

FILE: tests/unit/utils/responses/test_responses_store.py
  function build_store (line 28) | def build_store(db_path: str, policy: list | None = None) -> ResponsesSt...
  function create_test_response_object (line 37) | def create_test_response_object(
  function create_test_response_input (line 52) | def create_test_response_input(content: str, input_id: str) -> OpenAIRes...
  function create_test_messages (line 64) | def create_test_messages(content: str) -> list[OpenAIMessageParam]:
  function test_responses_store_pagination_basic (line 69) | async def test_responses_store_pagination_basic():
  function test_responses_store_pagination_ascending (line 118) | async def test_responses_store_pagination_ascending():
  function test_responses_store_pagination_with_model_filter (line 156) | async def test_responses_store_pagination_with_model_filter():
  function test_responses_store_pagination_invalid_after (line 197) | async def test_responses_store_pagination_invalid_after():
  function test_responses_store_pagination_no_limit (line 209) | async def test_responses_store_pagination_no_limit():
  function test_responses_store_get_response_object (line 241) | async def test_responses_store_get_response_object():
  function test_responses_store_input_items_pagination (line 269) | async def test_responses_store_input_items_pagination():
  function test_responses_store_input_items_before_pagination (line 350) | async def test_responses_store_input_items_before_pagination():
  function test_responses_store_safety_identifier (line 409) | async def test_responses_store_safety_identifier():

FILE: tests/unit/utils/sqlstore/test_sqlstore.py
  function test_sqlstore_shutdown_disposes_engine (line 22) | async def test_sqlstore_shutdown_disposes_engine():
  function test_sqlite_sqlstore (line 53) | async def test_sqlite_sqlstore():
  function test_sqlstore_upsert_support (line 104) | async def test_sqlstore_upsert_support():
  function test_sqlstore_pagination_basic (line 136) | async def test_sqlstore_pagination_basic():
  function test_sqlstore_pagination_with_filter (line 200) | async def test_sqlstore_pagination_with_filter():
  function test_sqlstore_pagination_ascending_order (line 252) | async def test_sqlstore_pagination_ascending_order():
  function test_sqlstore_pagination_multi_column_ordering_error (line 300) | async def test_sqlstore_pagination_multi_column_ordering_error():
  function test_sqlstore_pagination_cursor_requires_order_by (line 337) | async def test_sqlstore_pagination_cursor_requires_order_by():
  function test_sqlstore_pagination_error_handling (line 354) | async def test_sqlstore_pagination_error_handling():
  function test_where_operator_gt_and_update_delete (line 403) | async def test_where_operator_gt_and_update_delete():
  function test_batch_insert (line 439) | async def test_batch_insert():
  function test_where_operator_edge_cases (line 465) | async def test_where_operator_edge_cases():
  function test_sqlstore_pagination_custom_key_column (line 486) | async def test_sqlstore_pagination_custom_key_column():
  function test_pool_pre_ping_propagates_to_engine (line 537) | async def test_pool_pre_ping_propagates_to_engine(pre_ping):
  function test_pool_pre_ping_defaults_to_true (line 548) | async def test_pool_pre_ping_defaults_to_true():
  function test_postgres_pool_config_defaults (line 557) | async def test_postgres_pool_config_defaults():
  function test_postgres_pool_kwargs_propagate_to_engine (line 565) | async def test_postgres_pool_kwargs_propagate_to_engine():
  function test_postgres_pool_recycle_omitted_when_disabled (line 583) | async def test_postgres_pool_recycle_omitted_when_disabled():
  function test_pool_recycle_is_configurable (line 597) | async def test_pool_recycle_is_configurable():

FILE: tests/unit/utils/test_authorized_sqlstore.py
  function test_authorized_fetch_with_where_sql_access_control (line 20) | async def test_authorized_fetch_with_where_sql_access_control(mock_get_a...
  function test_sql_policy_consistency (line 82) | async def test_sql_policy_consistency(mock_get_authenticated_user):
  function test_authorized_store_user_attribute_capture (line 184) | async def test_authorized_store_user_attribute_capture(mock_get_authenti...