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

added files

Browse source
This commit is contained in:
vasilije 2025-07-29 09:49:01 +02:00
parent 190c7eea88
commit 514a37d5f5
3 changed files with 15 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
cognee/infrastructure/llm/generic_llm_api/adapter.py
View file

@ -10,8 +10,13 @@ from litellm.exceptions import ContentPolicyViolationError
from instructor.exceptions import InstructorRetryException from instructor.exceptions import InstructorRetryException
from cognee.infrastructure.llm.exceptions import ContentPolicyFilterError from cognee.infrastructure.llm.exceptions import ContentPolicyFilterError
from cognee.infrastructure.llm.llm_interface import LLMInterface from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.llm_interface import (
from cognee.infrastructure.llm.rate_limiter import rate_limit_async, sleep_and_retry_async LLMInterface,
)
from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.rate_limiter import (
rate_limit_async,
sleep_and_retry_async,
)
class GenericAPIAdapter(LLMInterface): class GenericAPIAdapter(LLMInterface):

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

@ -4,8 +4,10 @@ from typing import Type
from openai import OpenAI from openai import OpenAI
from pydantic import BaseModel from pydantic import BaseModel
from cognee.infrastructure.llm.llm_interface import LLMInterface from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.llm_interface import (
from cognee.infrastructure.llm.rate_limiter import ( LLMInterface,
)
from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.rate_limiter import (
rate_limit_async, rate_limit_async,
sleep_and_retry_async, sleep_and_retry_async,
) )

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

@ -9,10 +9,12 @@ from instructor.exceptions import InstructorRetryException
from cognee.exceptions import InvalidValueError from cognee.exceptions import InvalidValueError
from cognee.infrastructure.llm.prompts import read_query_prompt from cognee.infrastructure.llm.prompts import read_query_prompt
from cognee.infrastructure.llm.llm_interface import LLMInterface from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.llm_interface import (
LLMInterface,
)
from cognee.infrastructure.llm.exceptions import ContentPolicyFilterError from cognee.infrastructure.llm.exceptions import ContentPolicyFilterError
from cognee.infrastructure.files.utils.open_data_file import open_data_file from cognee.infrastructure.files.utils.open_data_file import open_data_file
from cognee.infrastructure.llm.rate_limiter import ( from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.rate_limiter import (
rate_limit_async, rate_limit_async,
rate_limit_sync, rate_limit_sync,
sleep_and_retry_async, sleep_and_retry_async,