diff --git a/env.ollama-binding-options.example b/env.ollama-binding-options.example index 9c35ce21..f3c8181e 100644 --- a/env.ollama-binding-options.example +++ b/env.ollama-binding-options.example @@ -193,4 +193,3 @@ # # End of .env entries for LightRAG binding options ################################################################################ - diff --git a/lightrag/api/config.py b/lightrag/api/config.py index 6812ff42..cf8c0492 100644 --- a/lightrag/api/config.py +++ b/lightrag/api/config.py @@ -138,8 +138,7 @@ def parse_args() -> argparse.Namespace: parser.add_argument( "--timeout", - default=get_env_value("TIMEOUT", DEFAULT_TIMEOUT, - int, special_none=True), + default=get_env_value("TIMEOUT", DEFAULT_TIMEOUT, int, special_none=True), type=int, help="Timeout in seconds (useful when using slow AI). Use None for infinite timeout", ) @@ -240,8 +239,7 @@ def parse_args() -> argparse.Namespace: "--llm-binding", type=str, default=get_env_value("LLM_BINDING", "ollama"), - choices=["lollms", "ollama", "openai", - "openai-ollama", "azure_openai"], + choices=["lollms", "ollama", "openai", "openai-ollama", "azure_openai"], help="LLM binding type (default: from env or ollama)", ) parser.add_argument( @@ -255,24 +253,24 @@ def parse_args() -> argparse.Namespace: # Conditionally add binding options defined in binding_options module # This will add command line arguments for all binding options (e.g., --ollama-embedding-num_ctx) # and corresponding environment variables (e.g., OLLAMA_EMBEDDING_NUM_CTX) - if '--llm-binding' in sys.argv: + if "--llm-binding" in sys.argv: try: - idx = sys.argv.index('--llm-binding') - if idx + 1 < len(sys.argv) and sys.argv[idx + 1] == 'ollama': + idx = sys.argv.index("--llm-binding") + if idx + 1 < len(sys.argv) and sys.argv[idx + 1] == "ollama": OllamaLLMOptions.add_args(parser) except IndexError: pass - elif os.environ.get('LLM_BINDING') == 'ollama': + elif os.environ.get("LLM_BINDING") == "ollama": OllamaLLMOptions.add_args(parser) - if '--embedding-binding' in sys.argv: + if "--embedding-binding" in sys.argv: try: - idx = sys.argv.index('--embedding-binding') - if idx + 1 < len(sys.argv) and sys.argv[idx + 1] == 'ollama': + idx = sys.argv.index("--embedding-binding") + if idx + 1 < len(sys.argv) and sys.argv[idx + 1] == "ollama": OllamaEmbeddingOptions.add_args(parser) except IndexError: pass - elif os.environ.get('EMBEDDING_BINDING') == 'ollama': + elif os.environ.get("EMBEDDING_BINDING") == "ollama": OllamaEmbeddingOptions.add_args(parser) args = parser.parse_args() @@ -316,8 +314,7 @@ def parse_args() -> argparse.Namespace: "EMBEDDING_BINDING_HOST", get_default_host(args.embedding_binding) ) args.llm_binding_api_key = get_env_value("LLM_BINDING_API_KEY", None) - args.embedding_binding_api_key = get_env_value( - "EMBEDDING_BINDING_API_KEY", "") + args.embedding_binding_api_key = get_env_value("EMBEDDING_BINDING_API_KEY", "") # Inject model configuration args.llm_model = get_env_value("LLM_MODEL", "mistral-nemo:latest") @@ -339,8 +336,7 @@ def parse_args() -> argparse.Namespace: args.temperature = get_env_value("TEMPERATURE", 0.5, float) # Select Document loading tool (DOCLING, DEFAULT) - args.document_loading_engine = get_env_value( - "DOCUMENT_LOADING_ENGINE", "DEFAULT") + args.document_loading_engine = get_env_value("DOCUMENT_LOADING_ENGINE", "DEFAULT") # Add environment variables that were previously read directly args.cors_origins = get_env_value("CORS_ORIGINS", "*") @@ -349,16 +345,13 @@ def parse_args() -> argparse.Namespace: # For JWT Auth args.auth_accounts = get_env_value("AUTH_ACCOUNTS", "") - args.token_secret = get_env_value( - "TOKEN_SECRET", "lightrag-jwt-default-secret") + args.token_secret = get_env_value("TOKEN_SECRET", "lightrag-jwt-default-secret") args.token_expire_hours = get_env_value("TOKEN_EXPIRE_HOURS", 48, int) - args.guest_token_expire_hours = get_env_value( - "GUEST_TOKEN_EXPIRE_HOURS", 24, int) + args.guest_token_expire_hours = get_env_value("GUEST_TOKEN_EXPIRE_HOURS", 24, int) args.jwt_algorithm = get_env_value("JWT_ALGORITHM", "HS256") # Rerank model configuration - args.rerank_model = get_env_value( - "RERANK_MODEL", "BAAI/bge-reranker-v2-m3") + args.rerank_model = get_env_value("RERANK_MODEL", "BAAI/bge-reranker-v2-m3") args.rerank_binding_host = get_env_value("RERANK_BINDING_HOST", None) args.rerank_binding_api_key = get_env_value("RERANK_BINDING_API_KEY", None) @@ -368,8 +361,7 @@ def parse_args() -> argparse.Namespace: ) # Query configuration - args.history_turns = get_env_value( - "HISTORY_TURNS", DEFAULT_HISTORY_TURNS, int) + args.history_turns = get_env_value("HISTORY_TURNS", DEFAULT_HISTORY_TURNS, int) args.top_k = get_env_value("TOP_K", DEFAULT_TOP_K, int) args.chunk_top_k = get_env_value("CHUNK_TOP_K", DEFAULT_CHUNK_TOP_K, int) args.max_entity_tokens = get_env_value( diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py index fe092557..8845e06c 100644 --- a/lightrag/api/lightrag_server.py +++ b/lightrag/api/lightrag_server.py @@ -286,7 +286,7 @@ def create_app(args): embed_model=args.embedding_model, host=args.embedding_binding_host, api_key=args.embedding_binding_api_key, - options=OllamaEmbeddingOptions.options_dict(args) + options=OllamaEmbeddingOptions.options_dict(args), ) if args.embedding_binding == "ollama" else azure_openai_embed(