diff --git a/README.md b/README.md index b47a6a4b6..a939aaca6 100644 --- a/README.md +++ b/README.md @@ -95,10 +95,9 @@ cognee.add([text], "example_dataset") # Add a new piece of information 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(result_text) +print(search_results) ``` Add alternative data types: diff --git a/cognee/api/v1/cognify/cognify.py b/cognee/api/v1/cognify/cognify.py index 2373c4e6e..408b92e76 100644 --- a/cognee/api/v1/cognify/cognify.py +++ b/cognee/api/v1/cognify/cognify.py @@ -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"]) 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( *[process_text( diff --git a/cognee/api/v1/config/config.py b/cognee/api/v1/config/config.py index 20e2ab6cf..6a1478d40 100644 --- a/cognee/api/v1/config/config.py +++ b/cognee/api/v1/config/config.py @@ -105,3 +105,21 @@ class config(): chunk_config = get_chunk_config() 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 + +