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

changes to cognee adapter

Browse source
This commit is contained in:
vasilije 2025-07-12 10:33:53 +02:00
parent af4b61cc4d
commit 362f2f5d7b
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/openai/adapter.py
View file

@ -63,6 +63,9 @@ class OpenAIAdapter(LLMInterface):
transcription_model: str,
max_tokens: int,
streaming: bool = False,
fallback_api_key: str = "",
fallback_endpoint: str = "",
fallback_model: str = "",
):
self.aclient = instructor.from_litellm(litellm.acompletion)
self.client = instructor.from_litellm(litellm.completion)
@ -73,6 +76,9 @@ class OpenAIAdapter(LLMInterface):
self.api_version = api_version
self.max_tokens = max_tokens
self.streaming = streaming
self.fallback_api_key = fallback_api_key
self.fallback_endpoint = fallback_endpoint
self.fallback_model = fallback_model
@observe(as_type="generation")
@sleep_and_retry_async()