Merge remote-tracking branch 'origin/dev' into feature/cog-3014-refactor-delete-feature
This commit is contained in:
commit
468205fed1
1 changed files with 0 additions and 15 deletions
|
|
@ -162,20 +162,5 @@ def get_llm_client(raise_api_key_error: bool = True):
|
||||||
endpoint=llm_config.llm_endpoint,
|
endpoint=llm_config.llm_endpoint,
|
||||||
)
|
)
|
||||||
|
|
||||||
elif provider == LLMProvider.MISTRAL:
|
|
||||||
if llm_config.llm_api_key is None:
|
|
||||||
raise LLMAPIKeyNotSetError()
|
|
||||||
|
|
||||||
from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.mistral.adapter import (
|
|
||||||
MistralAdapter,
|
|
||||||
)
|
|
||||||
|
|
||||||
return MistralAdapter(
|
|
||||||
api_key=llm_config.llm_api_key,
|
|
||||||
model=llm_config.llm_model,
|
|
||||||
max_completion_tokens=max_completion_tokens,
|
|
||||||
endpoint=llm_config.llm_endpoint,
|
|
||||||
)
|
|
||||||
|
|
||||||
else:
|
else:
|
||||||
raise UnsupportedLLMProviderError(provider)
|
raise UnsupportedLLMProviderError(provider)
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue