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

Updates to the configs

Browse source
This commit is contained in:
Vasilije 2024-05-26 10:41:38 +02:00
parent dad26b3904
commit 1dcfcd39cf
2 changed files with 1 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1,14 +1,12 @@
import os import os
from functools import lru_cache from functools import lru_cache
from pydantic_settings import BaseSettings, SettingsConfigDict from pydantic_settings import BaseSettings, SettingsConfigDict
from cognee.base_config import get_base_config
from cognee.infrastructure.databases.relational.config import get_relationaldb_config from cognee.infrastructure.databases.relational.config import get_relationaldb_config
from cognee.infrastructure.databases.vector.lancedb.LanceDBAdapter import LanceDBAdapter from cognee.infrastructure.databases.vector.lancedb.LanceDBAdapter import LanceDBAdapter
from cognee.infrastructure.databases.vector.embeddings.config import get_embedding_config from cognee.infrastructure.databases.vector.embeddings.config import get_embedding_config
from cognee.infrastructure.files.storage import LocalStorage from cognee.infrastructure.files.storage import LocalStorage
embeddings_config = get_embedding_config() embeddings_config = get_embedding_config()
base_config = get_base_config()
relational_config = get_relationaldb_config() relational_config = get_relationaldb_config()
class VectorConfig(BaseSettings): class VectorConfig(BaseSettings):

2
cognee/modules/cognify/config.py
View file

@ -16,7 +16,7 @@ class CognifyConfig(BaseSettings):
summarization_model: object = SummarizedContent summarization_model: object = SummarizedContent
labeling_model: object = LabeledContent labeling_model: object = LabeledContent
cognitive_layer_model: object = DefaultCognitiveLayer cognitive_layer_model: object = DefaultCognitiveLayer
intra_layer_score_treshold: int = 0.98 intra_layer_score_treshold: float = 0.98
connect_documents: bool = False connect_documents: bool = False