R2R icon indicating copy to clipboard operation
R2R copied to clipboard

Fix Create Graph

Open shreyaspimpalgaonkar opened this issue 4 months ago • 1 comments

[!IMPORTANT] Enhance knowledge graph functionality by adding entity_table_name parameter to key functions and updating integration tests accordingly.

  • Behavior:
    • Add entity_table_name parameter to get_entities() and get_entity_count() in kg.py, kg_service.py, and postgres.py.
    • Default entity_table_name to entity_embedding.
  • Tests:
    • Update test_kg_create_graph_sample_file_cli() and test_kg_create_graph_sample_file_sdk() to check for at least one entity in the response.
    • Modify test_kg_enrich_graph_sample_file_cli() and test_kg_enrich_graph_sample_file_sdk() to validate community creation.
  • Misc:
    • Fix minor issues in runner_cli.py and runner_sdk.py related to entity and community assertions.

This description was created by Ellipsis for 5b7397c02de81810b1d58ab6ee632ed384c5948d. It will automatically update as commits are pushed.

shreyaspimpalgaonkar avatar Oct 11 '24 01:10 shreyaspimpalgaonkar