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

made sure base tests pass

Browse source
This commit is contained in:
vasilije 2025-07-29 09:59:34 +02:00
parent 514a37d5f5
commit 119a6532b0
3 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
cognee/base_config.py
View file

@ -8,18 +8,24 @@ from pydantic_settings import BaseSettings, SettingsConfigDict
class BaseConfig(BaseSettings): class BaseConfig(BaseSettings):
data_root_directory: str = get_absolute_path(".data_storage") data_root_directory: str = get_absolute_path(".data_storage")
system_root_directory: str = get_absolute_path(".data_storage")
monitoring_tool: object = Observer.LANGFUSE monitoring_tool: object = Observer.LANGFUSE
langfuse_public_key: Optional[str] = os.getenv("LANGFUSE_PUBLIC_KEY") langfuse_public_key: Optional[str] = os.getenv("LANGFUSE_PUBLIC_KEY")
langfuse_secret_key: Optional[str] = os.getenv("LANGFUSE_SECRET_KEY") langfuse_secret_key: Optional[str] = os.getenv("LANGFUSE_SECRET_KEY")
langfuse_host: Optional[str] = os.getenv("LANGFUSE_HOST") langfuse_host: Optional[str] = os.getenv("LANGFUSE_HOST")
default_user_email: Optional[str] = os.getenv("DEFAULT_USER_EMAIL") default_user_email: Optional[str] = os.getenv("DEFAULT_USER_EMAIL")
default_user_password: Optional[str] = os.getenv("DEFAULT_USER_PASSWORD") default_user_password: Optional[str] = os.getenv("DEFAULT_USER_PASSWORD")
structured_output_framework: str = os.getenv(
"STRUCTURED_OUTPUT_FRAMEWORK", "llitellm_instructor"
)
model_config = SettingsConfigDict(env_file=".env", extra="allow") model_config = SettingsConfigDict(env_file=".env", extra="allow")
def to_dict(self) -> dict: def to_dict(self) -> dict:
return { return {
"data_root_directory": self.data_root_directory, "data_root_directory": self.data_root_directory,
"system_root_directory": self.system_root_directory,
"monitoring_tool": self.monitoring_tool, "monitoring_tool": self.monitoring_tool,
"structured_output_framework": self.structured_output_framework,
} }

2
cognee/infrastructure/llm/structured_output_framework/baml_src/extraction/knowledge_graph/extract_content_graph.py
View file

@ -42,7 +42,7 @@ async def extract_content_graph(
# return graph # return graph
# else: # else:
graph = await b.ExtractContentGraph( graph = await b.ExtractContentGraphGeneric(
content, mode=mode, baml_options={"client_registry": baml_registry} content, mode=mode, baml_options={"client_registry": baml_registry}
) )

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

@ -8,7 +8,9 @@ from litellm.exceptions import ContentPolicyViolationError
from instructor.exceptions import InstructorRetryException 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.structured_output_framework.llitellm_instructor.llm.prompts import (
read_query_prompt,
)
from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.llm_interface import ( from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.llm_interface import (
LLMInterface, LLMInterface,
) )