Skip to content

Commit 7ff3620

Browse files
docs(samples): Provide an end to end notebook for graphrag using custom retrievers (#159)
* Add Spanner Graph QA Chain * Formatted notebook. Added copyright message to prompts file. * Add missing imports for random graph name * Make input table name randomized in integration tests to avoid name collision for tests running parallely from different python environments * Provide timeout to graph cleanup * Make default timeout of 300 secs for ddl application * Increase timeout of integration test * Change integration test timeout * Minor formatting fixes * Make the ddl operations test fixture scoped for the module * Addressed review comments * Addressed a few other review comments. * Remove unused function * fix type check errors * Addressed review comments * Addressed review comments * Clear default project id from notebook * Add import statement for SpanerGraphQAChain to notebook * Add retrievers for Spanner Graph RAG * Add licence headers * Fix DATABASE name key * Fix lint error on import ordering * Fix lint errors * Few minor changes to the SpannerGraphNodeVectorRetriever * Fix lint error * Add an option to expand context graph by hops * Fix lint error * Addressed review comments * Remove expansion query options * Add backticks to property names * Change copyright year * Address review comments * Rename the retrievers. Merge the semantic retriever with the gql retriever. * Fixed lint errors * Change vertex ai versionto latest * Fix lint errors * Add documentation. Fixes the case where expands_by_hops is 0 * Add unit test for expand_by_hops=0 * Fix formatting for documentation * Addressed review comments * minor change * Minor change * Unpin langchain-google-spanner library * Addressed review comments --------- Co-authored-by: Amarnath Mullick <amullick@google.com> Co-authored-by: Averi Kitsch <akitsch@google.com>
1 parent aee9c92 commit 7ff3620

1 file changed

Lines changed: 7716 additions & 0 deletions

File tree

0 commit comments

Comments
 (0)