wip

gemini/sample-apps/conversational-genai-app-template/tests/unit/test_server.py

@@ -25,9 +25,6 @@
 from httpx import AsyncClient
 from langchain_core.messages import HumanMessage
 
-from app.server import app
-from app.utils.input_types import InputChat
-
 # Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -67,6 +64,9 @@ def sample_input_chat() -> InputChat:
         messages=[HumanMessage(content="What is the meaning of life?")],
     )
 
+from app.server import app
+from app.utils.input_types import InputChat
+
 def test_redirect_root_to_docs() -> None:
     """
     Test that the root endpoint (/) redirects to the Swagger UI documentation.