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

fixed linting

Browse Source
This commit is contained in:
Saifeddine ALOUI
2025-01-20 00:26:28 +01:00
parent 2dfbbec407
commit 70425b0357
3 changed files with 39 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
lightrag/api/lightrag_server.py
View File

@@ -297,15 +297,13 @@ def parse_args() -> argparse.Namespace:
default=default_llm_host,
help=f"llm server host URL (default: from env or {default_llm_host})",
)
default_llm_api_key = get_env_value(
"LLM_BINDING_API_KEY", None
)
default_llm_api_key = get_env_value("LLM_BINDING_API_KEY", None)
parser.add_argument(
"--llm-binding-api-key",
default=default_llm_api_key,
help=f"llm server API key (default: from env or empty string)",
help="llm server API key (default: from env or empty string)",
)
parser.add_argument(
@@ -323,14 +321,12 @@ def parse_args() -> argparse.Namespace:
default=default_embedding_host,
help=f"embedding server host URL (default: from env or {default_embedding_host})",
)
default_embedding_api_key = get_env_value(
"EMBEDDING_BINDING_API_KEY", ""
)
default_embedding_api_key = get_env_value("EMBEDDING_BINDING_API_KEY", "")
parser.add_argument(
"--embedding-binding-api-key",
default=default_embedding_api_key,
help=f"embedding server API key (default: from env or empty string)",
help="embedding server API key (default: from env or empty string)",
)
parser.add_argument(
@@ -649,26 +645,26 @@ def create_app(args):
texts,
embed_model=args.embedding_model,
host=args.embedding_binding_host,
api_key = args.embedding_binding_api_key
api_key=args.embedding_binding_api_key,
)
if args.embedding_binding == "lollms"
else ollama_embed(
texts,
embed_model=args.embedding_model,
host=args.embedding_binding_host,
api_key = args.embedding_binding_api_key
api_key=args.embedding_binding_api_key,
)
if args.embedding_binding == "ollama"
else azure_openai_embedding(
texts,
model=args.embedding_model, # no host is used for openai,
api_key = args.embedding_binding_api_key
api_key=args.embedding_binding_api_key,
)
if args.embedding_binding == "azure_openai"
else openai_embedding(
texts,
model=args.embedding_model, # no host is used for openai,
api_key = args.embedding_binding_api_key
api_key=args.embedding_binding_api_key,
),
)
@@ -686,7 +682,7 @@ def create_app(args):
"host": args.llm_binding_host,
"timeout": args.timeout,
"options": {"num_ctx": args.max_tokens},
"api_key": args.llm_binding_api_key
"api_key": args.llm_binding_api_key,
},
embedding_func=embedding_func,
)