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

feat: adds new errors to litellm client

Browse source
This commit is contained in:
hajdul88 2025-08-13 11:53:29 +02:00
parent b40dbf3c68
commit 623148ec6a
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py
View file

@ -83,7 +83,7 @@ def get_llm_client():
elif provider == LLMProvider.OLLAMA:
if llm_config.llm_api_key is None:
raise InvalidValueError(message="LLM API key is not set.")
raise LLMAPIKeyNotSetError()
from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.generic_llm_api.adapter import (
GenericAPIAdapter,
@ -106,7 +106,7 @@ def get_llm_client():
elif provider == LLMProvider.CUSTOM:
if llm_config.llm_api_key is None:
raise InvalidValueError(message="LLM API key is not set.")
raise LLMAPIKeyNotSetError()
from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.generic_llm_api.adapter import (
GenericAPIAdapter,
@ -125,7 +125,7 @@ def get_llm_client():
elif provider == LLMProvider.GEMINI:
if llm_config.llm_api_key is None:
raise InvalidValueError(message="LLM API key is not set.")
raise LLMAPIKeyNotSetError()
from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.gemini.adapter import (
GeminiAdapter,
@ -141,4 +141,4 @@ def get_llm_client():
)
else:
raise InvalidValueError(message=f"Unsupported LLM provider: {provider}")
raise UnsupportedLLMProviderError(provider)