From 908a3dc0b688e68419a5ca136d83445f7a9af4a6 Mon Sep 17 00:00:00 2001 From: Pavel Tisnovsky Date: Sun, 1 Jun 2025 12:10:49 +0200 Subject: [PATCH] Added missing type hint --- src/app/endpoints/query.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/app/endpoints/query.py b/src/app/endpoints/query.py index bea2162d..d55cc5d0 100644 --- a/src/app/endpoints/query.py +++ b/src/app/endpoints/query.py @@ -6,6 +6,7 @@ from llama_stack_client.lib.agents.agent import Agent # type: ignore from llama_stack_client import LlamaStackClient # type: ignore from llama_stack_client.types import UserMessage # type: ignore +from llama_stack_client.types.model_list_response import ModelListResponse from fastapi import APIRouter, Request, HTTPException, status @@ -43,7 +44,7 @@ def query_endpoint_handler( def select_model_id(client: LlamaStackClient, query_request: QueryRequest) -> str: """Select the model ID based on the request or available models.""" - models = client.models.list() + models: ModelListResponse = client.models.list() model_id = query_request.model provider_id = query_request.provider