From 2c11e605743f9881b16ffa26aa3ecc5a2731eb22 Mon Sep 17 00:00:00 2001 From: Your Name Date: Wed, 18 Sep 2024 19:03:37 +0200 Subject: [PATCH] wip --- .../tests/unit/test_server.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/gemini/sample-apps/conversational-genai-app-template/tests/unit/test_server.py b/gemini/sample-apps/conversational-genai-app-template/tests/unit/test_server.py index 83045ba7736..3cbbdd96111 100644 --- a/gemini/sample-apps/conversational-genai-app-template/tests/unit/test_server.py +++ b/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.