From 57cdab2b2bd8e5c39202d2618ff67e54556dfa1a Mon Sep 17 00:00:00 2001
From: yangdx <yangdx@znipower.com>
Date: Thu, 20 Feb 2025 12:33:43 +0800
Subject: [PATCH] Add tags to OllamaAPI router

---
 lightrag/api/routers/ollama_api.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lightrag/api/routers/ollama_api.py b/lightrag/api/routers/ollama_api.py
index 9be8067e..ea9662ad 100644
--- 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):