From f3064ce3fa5eacf642633e066cdc4b1b0e956b26 Mon Sep 17 00:00:00 2001 From: Pavel Tisnovsky Date: Tue, 14 Jan 2025 17:12:58 +0100 Subject: [PATCH] Fixed type hints Signed-off-by: Pavel Tisnovsky --- ols/app/endpoints/ols.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/ols/app/endpoints/ols.py b/ols/app/endpoints/ols.py index 98c62a4b..67a20697 100644 --- a/ols/app/endpoints/ols.py +++ b/ols/app/endpoints/ols.py @@ -89,6 +89,8 @@ def conversation_request( timestamps, ) = process_request(auth, llm_request) + summarizer_response: SummarizerResponse | Generator + if not valid: summarizer_response = SummarizerResponse( prompts.INVALID_QUERY_RESP, @@ -98,7 +100,7 @@ def conversation_request( else: summarizer_response = generate_response( conversation_id, llm_request, previous_input - ) # type: ignore[assignment] + ) timestamps["generate response"] = time.time() @@ -362,13 +364,13 @@ def generate_response( except Exception as summarizer_error: logger.error("Error while obtaining answer for user question") logger.exception(summarizer_error) - status_code, response, cause = errors_parsing.parse_generic_llm_error( # type: ignore[assignment] + status_code, response_text, cause = errors_parsing.parse_generic_llm_error( summarizer_error ) raise HTTPException( status_code=status_code, detail={ - "response": response, + "response": response_text, "cause": cause, }, )