diff --git a/src/app/endpoints/streaming_query.py b/src/app/endpoints/streaming_query.py index b977754a6..d445dd212 100644 --- a/src/app/endpoints/streaming_query.py +++ b/src/app/endpoints/streaming_query.py @@ -49,7 +49,6 @@ from configuration import configuration from constants import ( INTERRUPTED_RESPONSE_MESSAGE, - TOPIC_SUMMARY_INTERRUPT_TIMEOUT_SECONDS, LLM_TOKEN_EVENT, LLM_TOOL_CALL_EVENT, LLM_TOOL_RESULT_EVENT, @@ -57,6 +56,7 @@ MEDIA_TYPE_EVENT_STREAM, MEDIA_TYPE_JSON, MEDIA_TYPE_TEXT, + TOPIC_SUMMARY_INTERRUPT_TIMEOUT_SECONDS, ) from log import get_logger from models.config import Action diff --git a/src/app/main.py b/src/app/main.py index 235b2694c..2049630c7 100644 --- a/src/app/main.py +++ b/src/app/main.py @@ -15,8 +15,8 @@ import version from a2a_storage import A2AStorageFactory from app import routers -from app.endpoints.streaming_query import shutdown_background_topic_summary_tasks from app.database import create_tables, initialize_database +from app.endpoints.streaming_query import shutdown_background_topic_summary_tasks from authorization.azure_token_manager import AzureEntraIDManager from client import AsyncLlamaStackClientHolder from configuration import configuration