feat: add simple query endpoint (#458)

.changeset/five-goats-yawn.md

@@ -0,0 +1,5 @@
+---
+"create-llama": patch
+---
+
+Add query endpoint

templates/types/streaming/fastapi/app/api/routers/__init__.py

@@ -3,11 +3,13 @@
 from .chat import chat_router  # noqa: F401
 from .chat_config import config_router  # noqa: F401
 from .upload import file_upload_router  # noqa: F401
+from .query import query_router  # noqa: F401
 
 api_router = APIRouter()
 api_router.include_router(chat_router, prefix="/chat")
 api_router.include_router(config_router, prefix="/chat/config")
 api_router.include_router(file_upload_router, prefix="/chat/upload")
+api_router.include_router(query_router, prefix="/query")
 
 # Dynamically adding additional routers if they exist
 try:

templates/types/streaming/fastapi/app/api/routers/query.py

@@ -0,0 +1,25 @@
+import logging
+
+from fastapi import APIRouter
+from app.engine.index import IndexConfig, get_index
+from llama_index.core.base.base_query_engine import BaseQueryEngine
+
+
+query_router = r = APIRouter()
+
+logger = logging.getLogger("uvicorn")
+
+
+def get_query_engine() -> BaseQueryEngine:
+    index_config = IndexConfig(**{})
+    index = get_index(index_config)
+    return index.as_query_engine()
+
+
+@r.get("/")
+async def query_request(
+    query: str,
+) -> str:
+    query_engine = get_query_engine()
+    response = await query_engine.aquery(query)
+    return response.response