cognee/cognee/tests
hajdul88 6d85165189
Feature/cog 539 implementing additional retriever approaches (#262)
* fix: refactor get_graph_from_model to return nodes and edges correctly

* fix: add missing params

* fix: remove complex zip usage

* fix: add edges to data_point properties

* fix: handle rate limit error coming from llm model

* fix: fixes lost edges and nodes in get_graph_from_model

* fix: fixes database pruning issue in pgvector

* fix: fixes database pruning issue in pgvector (#261)

* feat: adds code summary embeddings to vector DB

* fix: cognee_demo notebook pipeline is not saving summaries

* feat: implements first version of codegraph retriever

* chore: implements minor changes mostly to make the code production ready

* fix: turns off raising duplicated edges unit test as we have these in our current codegraph generation

* feat: implements unit tests for description to codepart search

* fix: fixes edge property inconsistent access in codepart retriever

* chore: implements more precise typing for get_attribute method for cogneegraph

* chore: adds spacing to tests and changes the cogneegraph getter names

---------

Co-authored-by: Boris Arzentar <borisarzentar@gmail.com>
2024-12-10 11:07:06 +01:00
..
integration test: Update typo in unstructured test 2024-12-09 16:37:46 +01:00
tasks/summarization feat: code graph swe integration 2024-11-27 09:32:29 +01:00
test_data test: Add tests for different document types 2024-12-09 15:20:50 +01:00
unit Feature/cog 539 implementing additional retriever approaches (#262) 2024-12-10 11:07:06 +01:00
__init__.py Cog 174 (#84) 2024-04-26 00:16:03 +02:00
get_graph_url.py feat: split document reader from chunker (#131) 2024-08-19 14:36:10 +02:00
test_code_generation.py Rename remaining 'query' keyword args in cognee.search to 'query_text' 2024-11-18 14:00:14 +01:00
test_deduplication.py test: Fix test for multimedia deduplication 2024-12-05 20:09:29 +01:00
test_falkordb.py fix: falkordb adapter errors 2024-11-28 09:12:37 +01:00
test_library.py test: Add checking of Networkx database deletion 2024-11-21 16:09:50 +01:00
test_milvus.py fix: Resolve Milvus connection issue, add config to milvus test, add milvus gh action 2024-12-03 11:13:54 +01:00
test_neo4j.py chore: changes query text in tests 2024-11-26 17:29:44 +01:00
test_pgvector.py chore: changes query text in tests 2024-11-26 17:29:44 +01:00
test_qdrant.py chore: changes query text in tests 2024-11-26 17:29:44 +01:00
test_weaviate.py chore: changes query text in tests 2024-11-26 17:29:44 +01:00