gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix (token_tracker): removed all duplications

Browse source
This commit is contained in:
GGrassia 2025-11-28 16:06:49 +01:00
parent 5074b4e8ad
commit dc699b67ef
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
lightrag/api/lightrag_server.py
View file

@ -328,6 +328,10 @@ def create_app(args):
if config_cache.openai_llm_options: if config_cache.openai_llm_options:
kwargs.update(config_cache.openai_llm_options) kwargs.update(config_cache.openai_llm_options)
# Remove token_tracker from kwargs if it exists to avoid duplicate argument error
# (we pass it explicitly below)
kwargs.pop("token_tracker", None)
return await openai_complete_if_cache( return await openai_complete_if_cache(
args.llm_model, args.llm_model,
prompt, prompt,
@ -370,6 +374,10 @@ def create_app(args):
kwargs["timeout"] = llm_timeout kwargs["timeout"] = llm_timeout
if config_cache.azure_openai_llm_options: if config_cache.azure_openai_llm_options:
kwargs.update(config_cache.azure_openai_llm_options) kwargs.update(config_cache.azure_openai_llm_options)
# Remove token_tracker from kwargs if it exists to avoid duplicate argument error
# (we pass it explicitly below)
kwargs.pop("token_tracker", None)
return await azure_openai_complete_if_cache( return await azure_openai_complete_if_cache(
args.llm_model, args.llm_model,