writer-framework
writer-framework copied to clipboard
Added WriterQuestionToKG and tests block for querying knowledge graphs
Add WriterQuestionToKG Block
This PR adds a new Workflow block that enables querying multiple knowledge graphs. Does not include an SVG logo at this time.
Features
- Query one or multiple knowledge graphs using comma-separated UUIDs
- Support for subqueries option
- Full error handling for missing or invalid inputs
- Direct integration with Writer AI SDK for graph queries
Implementation Details
- Uses
WriterAIManager.acquire_client()for direct SDK access - Handles multiple graph IDs via comma-separated string parsing
- Returns full Question response object with answer and subqueries
- Configurable subqueries option (yes/no)
Testing
Added comprehensive test suite covering:
- Single graph query
- Multiple graph queries
- Subqueries enabled/disabled
- Error handling for:
- Missing graph IDs
- Empty graph IDs
- Missing questions
- Empty questions
- Missing required fields
Let's reconvene when @mmikita95 makes the necessary changes so that you can avoid using the SDK directly
We've built dedicated KG blocks with more traceability