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 = [ ...@@ -66,7 +66,7 @@ MOCK_AI_RESPONSES = [
from agent.mock_controller import mock_chat_controller 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): async def mock_chat(req: MockQueryRequest, background_tasks: BackgroundTasks):
""" """
Mock Agent Chat using mock_chat_controller: Mock Agent Chat using mock_chat_controller:
...@@ -97,9 +97,7 @@ async def mock_chat(req: MockQueryRequest, background_tasks: BackgroundTasks): ...@@ -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}") raise HTTPException(status_code=500, detail=f"Mock Agent Chat Error: {e!s}")
@router.post("/api/mock/db/search", summary="Real Data Retrieval Tool (Agent Tool)")
@router.post("/mock/db/search", summary="Real Data Retrieval Tool (Agent Tool)")
async def mock_db_search(req: MockDBRequest): async def mock_db_search(req: MockDBRequest):
""" """
Dùng `data_retrieval_tool` THẬT từ Agent: Dùng `data_retrieval_tool` THẬT từ Agent:
...@@ -148,7 +146,8 @@ async def mock_db_search(req: MockDBRequest): ...@@ -148,7 +146,8 @@ async def mock_db_search(req: MockDBRequest):
raise HTTPException(status_code=500, detail=f"DB Search Error: {e!s}") 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): async def mock_retriever_db(req: MockRetrieverRequest):
""" """
API thực tế để test Retriever + DB Search (dùng agent tool): API thực tế để test Retriever + DB Search (dùng agent tool):
......
...@@ -35,6 +35,7 @@ PUBLIC_PATHS = { ...@@ -35,6 +35,7 @@ PUBLIC_PATHS = {
PUBLIC_PATH_PREFIXES = [ PUBLIC_PATH_PREFIXES = [
"/static", "/static",
"/mock", "/mock",
"/api/mock",
] ]
......
...@@ -72,7 +72,7 @@ class RateLimitService: ...@@ -72,7 +72,7 @@ class RateLimitService:
"/openapi.json", "/openapi.json",
"/redoc", "/redoc",
} }
self.exempt_prefixes = ["/static", "/mock"] self.exempt_prefixes = ["/static", "/mock", "/api/mock"]
# In-memory blocklist (có thể chuyển sang Redis) # In-memory blocklist (có thể chuyển sang Redis)
self._blocklist: dict[str, datetime] = {} self._blocklist: dict[str, datetime] = {}
......
...@@ -73,7 +73,7 @@ try: ...@@ -73,7 +73,7 @@ try:
from api.mock_api_route import router as mock_router from api.mock_api_route import router as mock_router
app.include_router(mock_router) app.include_router(mock_router)
print("✅ Mock API Router mounted at /mock") print("✅ Mock API Router mounted at /api/mock")
except ImportError: except ImportError:
print("⚠️ Mock Router not found, skipping...") 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