Merge remote-tracking branch 'origin/pr-1449' into pr-1449
# Conflicts: # poetry.lock # uv.lock
This commit is contained in:
commit
ca88200b7c
3 changed files with 5395 additions and 5083 deletions
|
|
@ -162,5 +162,20 @@ 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)
|
||||||
|
|
|
||||||
2202
poetry.lock
generated
2202
poetry.lock
generated
File diff suppressed because it is too large
Load diff
Loading…
Add table
Reference in a new issue