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

fix: Silence model not found log

Browse source
This commit is contained in:
Andrej Milicevic 2025-09-25 18:18:08 +02:00
parent 39f170ec5b
commit 8c38d8a146
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/infrastructure/llm/utils.py
View file

@ -62,7 +62,7 @@ def get_model_max_completion_tokens(model_name: str):
max_completion_tokens = litellm.model_cost[model_name]["max_tokens"] max_completion_tokens = litellm.model_cost[model_name]["max_tokens"]
logger.debug(f"Max input tokens for {model_name}: {max_completion_tokens}") logger.debug(f"Max input tokens for {model_name}: {max_completion_tokens}")
else: else:
logger.info("Model not found in LiteLLM's model_cost.") logger.debug("Model not found in LiteLLM's model_cost.")
return max_completion_tokens return max_completion_tokens