gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update lightrag_server.py

Browse Source
This commit is contained in:
Saifeddine ALOUI
2025-03-03 12:24:49 +01:00
committed by GitHub
parent 7b3e394730
commit 5680e9ef11
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
lightrag/api/lightrag_server.py
View File

@@ -20,7 +20,7 @@ from ascii_colors import ASCIIColors
from fastapi.middleware.cors import CORSMiddleware from fastapi.middleware.cors import CORSMiddleware
from contextlib import asynccontextmanager from contextlib import asynccontextmanager
from dotenv import load_dotenv from dotenv import load_dotenv
from .utils_api import ( from lightrag.api.utils_api import (
get_api_key_dependency, get_api_key_dependency,
parse_args, parse_args,
get_default_host, get_default_host,
@@ -30,14 +30,14 @@ from lightrag import LightRAG
from lightrag.types import GPTKeywordExtractionFormat from lightrag.types import GPTKeywordExtractionFormat
from lightrag.api import __api_version__ from lightrag.api import __api_version__
from lightrag.utils import EmbeddingFunc from lightrag.utils import EmbeddingFunc
from .routers.document_routes import ( from lightrag.api.routers.document_routes import (
DocumentManager, DocumentManager,
create_document_routes, create_document_routes,
run_scanning_process, run_scanning_process,
) )
from .routers.query_routes import create_query_routes from lightrag.api.routers.query_routes import create_query_routes
from .routers.graph_routes import create_graph_routes from lightrag.api.routers.graph_routes import create_graph_routes
from .routers.ollama_api import OllamaAPI from lightrag.api.routers.ollama_api import OllamaAPI
from lightrag.utils import logger, set_verbose_debug from lightrag.utils import logger, set_verbose_debug
from lightrag.kg.shared_storage import ( from lightrag.kg.shared_storage import (
@@ -48,7 +48,9 @@ from lightrag.kg.shared_storage import (
) )
# Load environment variables # Load environment variables
load_dotenv(override=True) # Updated to use the .env that is inside the current folder
# This update allows the user to put a different.env file for each lightrag folder
load_dotenv(".env", override=True)
# Initialize config parser # Initialize config parser
config = configparser.ConfigParser() config = configparser.ConfigParser()