ParisNeo commited on
Commit
a31af7c
·
unverified ·
1 Parent(s): 62bdee6

Update lightrag_server.py

Browse files
Files changed (1) hide show
  1. lightrag/api/lightrag_server.py +8 -6
lightrag/api/lightrag_server.py CHANGED
@@ -20,7 +20,7 @@ from ascii_colors import ASCIIColors
20
  from fastapi.middleware.cors import CORSMiddleware
21
  from contextlib import asynccontextmanager
22
  from dotenv import load_dotenv
23
- from .utils_api import (
24
  get_api_key_dependency,
25
  parse_args,
26
  get_default_host,
@@ -30,14 +30,14 @@ from lightrag import LightRAG
30
  from lightrag.types import GPTKeywordExtractionFormat
31
  from lightrag.api import __api_version__
32
  from lightrag.utils import EmbeddingFunc
33
- from .routers.document_routes import (
34
  DocumentManager,
35
  create_document_routes,
36
  run_scanning_process,
37
  )
38
- from .routers.query_routes import create_query_routes
39
- from .routers.graph_routes import create_graph_routes
40
- from .routers.ollama_api import OllamaAPI
41
 
42
  from lightrag.utils import logger, set_verbose_debug
43
  from lightrag.kg.shared_storage import (
@@ -48,7 +48,9 @@ from lightrag.kg.shared_storage import (
48
  )
49
 
50
  # Load environment variables
51
- load_dotenv(override=True)
 
 
52
 
53
  # Initialize config parser
54
  config = configparser.ConfigParser()
 
20
  from fastapi.middleware.cors import CORSMiddleware
21
  from contextlib import asynccontextmanager
22
  from dotenv import load_dotenv
23
+ from lightrag.api.utils_api import (
24
  get_api_key_dependency,
25
  parse_args,
26
  get_default_host,
 
30
  from lightrag.types import GPTKeywordExtractionFormat
31
  from lightrag.api import __api_version__
32
  from lightrag.utils import EmbeddingFunc
33
+ from lightrag.api.routers.document_routes import (
34
  DocumentManager,
35
  create_document_routes,
36
  run_scanning_process,
37
  )
38
+ from lightrag.api.routers.query_routes import create_query_routes
39
+ from lightrag.api.routers.graph_routes import create_graph_routes
40
+ from lightrag.api.routers.ollama_api import OllamaAPI
41
 
42
  from lightrag.utils import logger, set_verbose_debug
43
  from lightrag.kg.shared_storage import (
 
48
  )
49
 
50
  # Load environment variables
51
+ # Updated to use the .env that is inside the current folder
52
+ # This update allows the user to put a different.env file for each lightrag folder
53
+ load_dotenv(".env", override=True)
54
 
55
  # Initialize config parser
56
  config = configparser.ConfigParser()