From aeb79761d601dc703078253058afadd9de7fc4ae Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Wed, 6 Aug 2025 10:01:56 +0200 Subject: [PATCH] fix: Resolve path issue --- .../litellm_instructor/llm/get_llm_client.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py index c52cf20ae..22d101077 100644 --- a/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py +++ b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py @@ -82,7 +82,7 @@ def get_llm_client(): if llm_config.llm_api_key is None: raise InvalidValueError(message="LLM API key is not set.") - from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.generic_llm_api import ( + from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.generic_llm_api.adapter import ( GenericAPIAdapter, ) @@ -95,7 +95,7 @@ def get_llm_client(): ) elif provider == LLMProvider.ANTHROPIC: - from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.anthropic import ( + from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.anthropic.adapter import ( AnthropicAdapter, ) @@ -105,7 +105,7 @@ def get_llm_client(): if llm_config.llm_api_key is None: raise InvalidValueError(message="LLM API key is not set.") - from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.generic_llm_api import ( + from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.generic_llm_api.adapter import ( GenericAPIAdapter, ) @@ -124,7 +124,7 @@ def get_llm_client(): if llm_config.llm_api_key is None: raise InvalidValueError(message="LLM API key is not set.") - from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.gemini import ( + from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.gemini.adapter import ( GeminiAdapter, )