From 362f2f5d7bf43d87bcb7631a4a533fdbcbb26793 Mon Sep 17 00:00:00 2001 From: vasilije Date: Sat, 12 Jul 2025 10:33:53 +0200 Subject: [PATCH] changes to cognee adapter --- .../llitellm_instructor/llm/openai/adapter.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/openai/adapter.py b/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/openai/adapter.py index c1406d571..c7912eb4c 100644 --- a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/openai/adapter.py +++ b/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/openai/adapter.py @@ -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()