R2R
R2R copied to clipboard
Shreyas/tests
: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
andWrappedKGEnrichementResponse
inr2r/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 returnKGEnrichementResponse
. - 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 includeRestructureRouter
. - 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