linting

HKUDS · Jan 28, 2025 · b5d0972 · b5d0972
1 parent 4ab1dea
commit b5d0972
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py
@@ -47,6 +47,7 @@ def estimate_tokens(text: str) -> int:
 
     return int(tokens)
 
+
 class OllamaServerInfos:
     # Constants for emulated Ollama model information
     LIGHTRAG_NAME = "lightrag"
@@ -55,12 +56,13 @@ class OllamaServerInfos:
     LIGHTRAG_SIZE = 7365960935  # it's a dummy value
     LIGHTRAG_CREATED_AT = "2024-01-15T00:00:00Z"
     LIGHTRAG_DIGEST = "sha256:lightrag"
-    
+
     KV_STORAGE = "JsonKVStorage"
     DOC_STATUS_STORAGE = "JsonDocStatusStorage"
     GRAPH_STORAGE = "NetworkXStorage"
     VECTOR_STORAGE = "NanoVectorDBStorage"
 
+
 # Add infos
 ollama_server_infos = OllamaServerInfos()
 
@@ -490,7 +492,9 @@ def timeout_type(value):
     parser.add_argument(
         "--simulated-model-name",
         type=str,
-        default=get_env_value("SIMULATED_MODEL_NAME", ollama_server_infos.LIGHTRAG_MODEL),
+        default=get_env_value(
+            "SIMULATED_MODEL_NAME", ollama_server_infos.LIGHTRAG_MODEL
+        ),
         help="Number of conversation history turns to include (default: from env or 3)",
     )