Update lightrag_server.py
Browse files
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 |
-
|
|
|
|
|
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()
|