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

Enable LiteLLM proxy with embedding_binding_host

Browse Source
This commit is contained in:
Enrique Catalá
2025-02-13 22:45:03 +01:00
parent 5730987349
commit 991fa33986
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lightrag/api/lightrag_server.py
View File

@@ -1021,7 +1021,8 @@ def create_app(args):
if args.embedding_binding == "azure_openai" if args.embedding_binding == "azure_openai"
else openai_embed( else openai_embed(
texts, texts,
model=args.embedding_model, # no host is used for openai, model=args.embedding_model,
base_url=args.embedding_binding_host, # If you decide to use litellm as a proxy for azure openai, this is relevant
api_key=args.embedding_binding_api_key, api_key=args.embedding_binding_api_key,
), ),
) )