diff --git a/services/rag/llamaindex/server.py b/services/rag/llamaindex/server.py index e4b0588..f65b1ab 100644 --- a/services/rag/llamaindex/server.py +++ b/services/rag/llamaindex/server.py @@ -3,13 +3,13 @@ HTTP API server for querying the vector storage via the existing retrieval pipeline. """ -from pathlib import Path import sys +from pathlib import Path from dotenv import load_dotenv from fastapi import FastAPI, HTTPException -from pydantic import BaseModel, Field from loguru import logger +from pydantic import BaseModel, Field from retrieval import initialize_retriever @@ -91,9 +91,7 @@ def test_query(payload: TestQueryRequest) -> TestQueryResponse: ) response_text = str(result) - logger.info( - f"/api/test-query completed successfully (sources={len(sources)})" - ) + logger.info(f"/api/test-query completed successfully (sources={len(sources)})") return TestQueryResponse(query=query, response=response_text, sources=sources) except HTTPException: @@ -106,4 +104,4 @@ def test_query(payload: TestQueryRequest) -> TestQueryResponse: if __name__ == "__main__": import uvicorn - uvicorn.run("server:app", host="0.0.0.0", port=8000, reload=False) + uvicorn.run("server:app", host="0.0.0.0", port=8334, reload=False)