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

fix (tracker_): removed passing as named arg

Browse source
This commit is contained in:
GGrassia 2025-11-28 15:54:38 +01:00
parent d971e372c9
commit 5074b4e8ad
1 changed files with 8 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
lightrag/utils.py
View file

@ -1694,9 +1694,10 @@ async def use_llm_func_with_cache(
if max_tokens is not None: if max_tokens is not None:
kwargs["max_tokens"] = max_tokens kwargs["max_tokens"] = max_tokens
# Only add token_tracker if not already in kwargs (to avoid duplicate argument error) # Note: token_tracker is NOT passed here because the LLM wrapper functions
if token_tracker is not None and "token_tracker" not in kwargs: # (e.g., optimized_openai_alike_model_complete in lightrag_server.py)
kwargs["token_tracker"] = token_tracker # already handle token_tracker directly. Passing it here would cause
# "got multiple values for keyword argument 'token_tracker'" error.
res: str = await use_llm_func( res: str = await use_llm_func(
safe_user_prompt, safe_user_prompt,
@ -1734,9 +1735,10 @@ async def use_llm_func_with_cache(
if max_tokens is not None: if max_tokens is not None:
kwargs["max_tokens"] = max_tokens kwargs["max_tokens"] = max_tokens
# Only add token_tracker if not already in kwargs (to avoid duplicate argument error) # Note: token_tracker is NOT passed here because the LLM wrapper functions
if token_tracker is not None and "token_tracker" not in kwargs: # (e.g., optimized_openai_alike_model_complete in lightrag_server.py)
kwargs["token_tracker"] = token_tracker # already handle token_tracker directly. Passing it here would cause
# "got multiple values for keyword argument 'token_tracker'" error.
try: try:
res = await use_llm_func( res = await use_llm_func(