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

implement settings

Browse source
This commit is contained in:
Vasilije 2024-05-25 19:00:36 +02:00
parent f56049f747
commit 95b56a1fb8
3 changed files with 13 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
cognee/base_config.py
View file

@ -3,10 +3,16 @@ from pydantic_settings import BaseSettings, SettingsConfigDict
from cognee.root_dir import get_absolute_path from cognee.root_dir import get_absolute_path
from cognee.shared.data_models import MonitoringTool
# Monitoring tool
class BaseConfig(BaseSettings): class BaseConfig(BaseSettings):
system_root_directory: str = get_absolute_path(".cognee_system") system_root_directory: str = get_absolute_path(".cognee_system")
data_root_directory: str = get_absolute_path(".data") data_root_directory: str = get_absolute_path(".data")
monitoring_tool: object = MonitoringTool.LANGFUSE
@ -15,7 +21,8 @@ class BaseConfig(BaseSettings):
def to_dict(self) -> dict: def to_dict(self) -> dict:
return { return {
"system_root_directory": self.system_root_directory, "system_root_directory": self.system_root_directory,
"data_root_directory": self.data_root_directory "data_root_directory": self.data_root_directory,
"monitoring_tool": self.monitoring_tool,
} }
@lru_cache @lru_cache

8
cognee/config.py
View file

@ -64,10 +64,10 @@ class Config:
# model_endpoint: str = "openai" # model_endpoint: str = "openai"
# llm_api_key: Optional[str] = os.getenv("OPENAI_API_KEY") # llm_api_key: Optional[str] = os.getenv("OPENAI_API_KEY")
openai_temperature: float = float(os.getenv("OPENAI_TEMPERATURE", 0.0)) openai_temperature: float = float(os.getenv("OPENAI_TEMPERATURE", 0.0))
openai_embedding_model = "text-embedding-3-large" # openai_embedding_model = "text-embedding-3-large"
openai_embedding_dimensions = 3072 # openai_embedding_dimensions = 3072
litellm_embedding_model = "text-embedding-3-large" # litellm_embedding_model = "text-embedding-3-large"
litellm_embedding_dimensions = 3072 # litellm_embedding_dimensions = 3072
graphistry_username = os.getenv("GRAPHISTRY_USERNAME") graphistry_username = os.getenv("GRAPHISTRY_USERNAME")
graphistry_password = os.getenv("GRAPHISTRY_PASSWORD") graphistry_password = os.getenv("GRAPHISTRY_PASSWORD")

2
cognee/infrastructure/databases/vector/embeddings/config.py
View file

@ -5,7 +5,7 @@ class EmbeddingConfig(BaseSettings):
openai_embedding_model: str = "text-embedding-3-large" openai_embedding_model: str = "text-embedding-3-large"
openai_embedding_dimensions: int = 3072 openai_embedding_dimensions: int = 3072
litellm_embedding_model: str = "text-embedding-3-large" litellm_embedding_model: str = "text-embedding-3-large"
litellm_embedding_dimensions: str = 3072 litellm_embedding_dimensions: int = 3072
model_config = SettingsConfigDict(env_file = ".env", extra = "allow") model_config = SettingsConfigDict(env_file = ".env", extra = "allow")