Commit 49f43a45 authored by Vũ Hoàng Anh's avatar Vũ Hoàng Anh

Fix mock API routing and retriever alias

parent 566ee233
......@@ -66,7 +66,7 @@ MOCK_AI_RESPONSES = [
from agent.mock_controller import mock_chat_controller
@router.post("/mock/agent/chat", summary="Mock Agent Chat (Real Tools + Fake LLM)")
@router.post("/api/mock/agent/chat", summary="Mock Agent Chat (Real Tools + Fake LLM)")
async def mock_chat(req: MockQueryRequest, background_tasks: BackgroundTasks):
"""
Mock Agent Chat using mock_chat_controller:
......@@ -97,9 +97,7 @@ async def mock_chat(req: MockQueryRequest, background_tasks: BackgroundTasks):
raise HTTPException(status_code=500, detail=f"Mock Agent Chat Error: {e!s}")
@router.post("/mock/db/search", summary="Real Data Retrieval Tool (Agent Tool)")
@router.post("/api/mock/db/search", summary="Real Data Retrieval Tool (Agent Tool)")
async def mock_db_search(req: MockDBRequest):
"""
Dùng `data_retrieval_tool` THẬT từ Agent:
......@@ -148,7 +146,8 @@ async def mock_db_search(req: MockDBRequest):
raise HTTPException(status_code=500, detail=f"DB Search Error: {e!s}")
@router.post("/mock/retriverdb", summary="Real Embedding + Real DB Vector Search")
@router.post("/api/mock/retrieverdb", summary="Real Embedding + Real DB Vector Search")
@router.post("/api/mock/retriverdb", summary="Real Embedding + Real DB Vector Search (Legacy)")
async def mock_retriever_db(req: MockRetrieverRequest):
"""
API thực tế để test Retriever + DB Search (dùng agent tool):
......
......@@ -35,6 +35,7 @@ PUBLIC_PATHS = {
PUBLIC_PATH_PREFIXES = [
"/static",
"/mock",
"/api/mock",
]
......
......@@ -72,7 +72,7 @@ class RateLimitService:
"/openapi.json",
"/redoc",
}
self.exempt_prefixes = ["/static", "/mock"]
self.exempt_prefixes = ["/static", "/mock", "/api/mock"]
# In-memory blocklist (có thể chuyển sang Redis)
self._blocklist: dict[str, datetime] = {}
......
......@@ -73,7 +73,7 @@ try:
from api.mock_api_route import router as mock_router
app.include_router(mock_router)
print("✅ Mock API Router mounted at /mock")
print("✅ Mock API Router mounted at /api/mock")
except ImportError:
print("⚠️ Mock Router not found, skipping...")
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment