Add tags to OllamaAPI router

2025-02-20 12:33:43 +08:00 · 2025-02-20 12:33:43 +08:00 · 57cdab2b2b
commit 57cdab2b2b
parent 5d884f6d3e
1 changed files with 1 additions and 1 deletions
--- a/lightrag/api/routers/ollama_api.py
+++ b/lightrag/api/routers/ollama_api.py
@ -126,7 +126,7 @@ class OllamaAPI:
        self.rag = rag
        self.ollama_server_infos = ollama_server_infos
        self.top_k = top_k
-        self.router = APIRouter()
+        self.router = APIRouter(tags=["Ollama"])
        self.setup_routes()

    def setup_routes(self):