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-06-12 20:52:58 +02:00
parent 409d3c7d8b
commit a197177676
3 changed files with 21 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
README.md
View file

@ -95,10 +95,9 @@ cognee.add([text], "example_dataset") # Add a new piece of information
cognee.cognify() # Use LLMs and cognee to create knowledge cognee.cognify() # Use LLMs and cognee to create knowledge
search_results = cognee.search("SIMILARITY", "computer science") # Query cognee for the knowledge search_results = cognee.search("SIMILARITY", {'query': 'Tell me about NLP'}) # Query cognee for the knowledge
for result_text in search_results[0]: print(search_results)
print(result_text)
``` ```
Add alternative data types: Add alternative data types:

2
cognee/api/v1/cognify/cognify.py
View file

@ -130,7 +130,7 @@ async def cognify(datasets: Union[str, List[str]] = None):
logger.warning("File (%s) has an unknown file type. We are skipping it.", file_metadata["id"]) logger.warning("File (%s) has an unknown file type. We are skipping it.", file_metadata["id"])
added_chunks = await add_data_chunks(data_chunks) added_chunks = await add_data_chunks(data_chunks)
await add_data_chunks_basic_rag(data_chunks) # await add_data_chunks_basic_rag(data_chunks)
await asyncio.gather( await asyncio.gather(
*[process_text( *[process_text(

18
cognee/api/v1/config/config.py
View file

@ -105,3 +105,21 @@ class config():
chunk_config = get_chunk_config() chunk_config = get_chunk_config()
chunk_config.chunk_size = chunk_size chunk_config.chunk_size = chunk_size
@staticmethod
def set_vector_engine_provider(vector_engine_provider: str):
vector_db_config = get_vectordb_config()
vector_db_config.vector_engine_provider = vector_engine_provider
@staticmethod
def set_vector_db_key(db_key: str):
vector_db_config = get_vectordb_config()
vector_db_config.vector_db_key = db_key
@staticmethod
def set_vector_db_url(db_url: str):
vector_db_config = get_vectordb_config()
vector_db_config.vector_db_url = db_url