From ebf2aaaa5c4f049bf5e9af893912d7053f175a9c Mon Sep 17 00:00:00 2001
From: HectorSin <kkang15634@ajou.ac.kr>
Date: Sat, 10 Jan 2026 20:58:08 +0900
Subject: [PATCH] refactor: add type hint for handler_class

Signed-off-by: HectorSin <kkang15634@ajou.ac.kr>
---
 cognee/shared/utils.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/cognee/shared/utils.py b/cognee/shared/utils.py
index 17501a512..460a10bfb 100644
--- a/cognee/shared/utils.py
+++ b/cognee/shared/utils.py
@@ -141,7 +141,9 @@ def embed_logo(p, layout_scale, logo_alpha, position):
 def start_visualization_server(
     host: str = "0.0.0.0",
     port: int = 8001,
-    handler_class=http.server.SimpleHTTPRequestHandler,
+    handler_class: type[
+        http.server.SimpleHTTPRequestHandler
+    ] = http.server.SimpleHTTPRequestHandler,
 ):
     """
     Spin up a simple HTTP server in a background thread to serve files.