R2R icon indicating copy to clipboard operation
R2R copied to clipboard

Shreyas/tests

Open shreyaspimpalgaonkar opened this issue 6 months ago • 0 comments

:rocket: This description was created by Ellipsis for commit 8f892d26f99ae91b8e0606bb9b85afc36ce4ab26

Summary:

This PR restructures the knowledge graph enrichment process by introducing new response models, updating routes, modifying service logic, and adding tests.

Key points:

  • Added KGEnrichementResponse and WrappedKGEnrichementResponse in r2r/base/api/models/restructure/responses.py.
  • Updated r2r/base/api/models/__init__.py to include new response models.
  • Renamed and moved files related to knowledge graph routes to r2r/base/api/models/restructure.
  • Modified r2r/main/api/client/restructure.py to return KGEnrichementResponse.
  • Updated r2r/main/api/routes/restructure/base.py to handle graph enrichment requests.
  • Removed old knowledge graph routes and requests.
  • Updated r2r/main/app.py to include RestructureRouter.
  • Modified r2r/main/services/restructure_service.py to use async enrichment pipeline.
  • Removed print statements from r2r/providers/kg/neo4j/provider.py.
  • Added tests in tests/test_kg.py for knowledge graph extraction and clustering.

Generated with :heart: by ellipsis.dev

shreyaspimpalgaonkar avatar Aug 19 '24 23:08 shreyaspimpalgaonkar