Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

refactor: refine chat flow and retrievers #589

Merged
merged 12 commits into from
Jan 24, 2025
Merged

refactor: refine chat flow and retrievers #589

merged 12 commits into from
Jan 24, 2025

Conversation

Mini256
Copy link
Member

@Mini256 Mini256 commented Jan 13, 2025

close #565

  • Add VectorSearchRetrieverr (support single KB pure retrieve)
    • metadata filters
    • similarity_threshold
  • Add VectorSearchFusionRetriever
    • support multiple KB route
    • support query decompose
  • Add KnowledgeGraphRetriever (support single KB pure retrieve)
    • Add KnowledgeGraphNode (llamaindex node)
    • metadata filters
    • similarity_threshold
  • Add KnowledgeGraphFusionRetriever
    • support multiple KB route
    • support query decompose (old IntentAnalyzer)
  • Add retrieve APIs
    • /api/v1/retrieve/chunks
    • /api/v1/retrieve/knowledge_graph
    • /api/v1/admin/knowledge_bases/{kb_id}/chunks/retrieve
    • /api/v1/admin/knowledge_bases/{kb_id}/graph/retrieve
  • Integration with ChatService
  • tests and fix bug

Feature PRs:

  • merge TiDBGraphEditor to TiDBGraphStore, TiDBGraphStore will only retain storage layer operactions, so it shouldn't depend on DsPy or LLM.
  • TiDBKnowledgeGraphIndex will handle the merge entities parts.
  • KnowledgeGraphRetriever will handle query compose parts.

Copy link

vercel bot commented Jan 13, 2025

The latest updates on your projects. Learn more about Vercel for Git ↗︎

Name Status Preview Comments Updated (UTC)
tidb-ai-preview ✅ Ready (Inspect) Visit Preview 💬 Add feedback Jan 24, 2025 2:23am
tidb-ai-storybook ✅ Ready (Inspect) Visit Preview 💬 Add feedback Jan 24, 2025 2:23am

@Mini256 Mini256 changed the title refactor: refine chat flow and retrieve flow(WIP) refactor: refine chat flow and retrievers Jan 24, 2025
Copy link

@Mini256 Mini256 merged commit 5562ed7 into main Jan 24, 2025
14 checks passed
@Mini256 Mini256 deleted the refactor-retrieve branch January 24, 2025 02:56
Mini256 added a commit to jrj5423/autoflow that referenced this pull request Jan 24, 2025
close pingcap#565

- [x] Add VectorSearchRetrieverr (support single KB pure retrieve)
   - [x] metadata filters
   - [ ] similarity_threshold
- [x] Add VectorSearchFusionRetriever
   - [ ] support multiple KB route
   - [x] support query decompose
- [x] Add KnowledgeGraphRetriever (support single KB pure retrieve)
   - [x] Add `KnowledgeGraphNode` (llamaindex node)
   - [x] metadata filters
   - [ ] similarity_threshold
- [x] Add KnowledgeGraphFusionRetriever 
   - [ ] support multiple KB route
   - [x] support query decompose (old IntentAnalyzer)
- [x] Add retrieve APIs
  - [x]  `/api/v1/retrieve/chunks`
  - [x]  `/api/v1/retrieve/knowledge_graph`
  - [x]  `/api/v1/admin/knowledge_bases/{kb_id}/chunks/retrieve`
  - [x]  `/api/v1/admin/knowledge_bases/{kb_id}/graph/retrieve` 
- [x] Integration with ChatService
- [ ] tests and fix bug

Feature PRs:
- merge `TiDBGraphEditor` to `TiDBGraphStore`, TiDBGraphStore will only
retain storage layer operactions, so it shouldn't depend on DsPy or LLM.
- `TiDBKnowledgeGraphIndex` will handle the merge entities parts.
- `KnowledgeGraphRetriever` will handle query compose parts.
NG85 pushed a commit to NG85/autoflow that referenced this pull request Feb 7, 2025
close pingcap#565

- [x] Add VectorSearchRetrieverr (support single KB pure retrieve)
   - [x] metadata filters
   - [ ] similarity_threshold
- [x] Add VectorSearchFusionRetriever
   - [ ] support multiple KB route
   - [x] support query decompose
- [x] Add KnowledgeGraphRetriever (support single KB pure retrieve)
   - [x] Add `KnowledgeGraphNode` (llamaindex node)
   - [x] metadata filters
   - [ ] similarity_threshold
- [x] Add KnowledgeGraphFusionRetriever
   - [ ] support multiple KB route
   - [x] support query decompose (old IntentAnalyzer)
- [x] Add retrieve APIs
  - [x]  `/api/v1/retrieve/chunks`
  - [x]  `/api/v1/retrieve/knowledge_graph`
  - [x]  `/api/v1/admin/knowledge_bases/{kb_id}/chunks/retrieve`
  - [x]  `/api/v1/admin/knowledge_bases/{kb_id}/graph/retrieve`
- [x] Integration with ChatService
- [ ] tests and fix bug

Feature PRs:
- merge `TiDBGraphEditor` to `TiDBGraphStore`, TiDBGraphStore will only
retain storage layer operactions, so it shouldn't depend on DsPy or LLM.
- `TiDBKnowledgeGraphIndex` will handle the merge entities parts.
- `KnowledgeGraphRetriever` will handle query compose parts.
Mini256 added a commit to sszgwdk/autoflow that referenced this pull request Feb 11, 2025
close pingcap#565

- [x] Add VectorSearchRetrieverr (support single KB pure retrieve)
   - [x] metadata filters
   - [ ] similarity_threshold
- [x] Add VectorSearchFusionRetriever
   - [ ] support multiple KB route
   - [x] support query decompose
- [x] Add KnowledgeGraphRetriever (support single KB pure retrieve)
   - [x] Add `KnowledgeGraphNode` (llamaindex node)
   - [x] metadata filters
   - [ ] similarity_threshold
- [x] Add KnowledgeGraphFusionRetriever 
   - [ ] support multiple KB route
   - [x] support query decompose (old IntentAnalyzer)
- [x] Add retrieve APIs
  - [x]  `/api/v1/retrieve/chunks`
  - [x]  `/api/v1/retrieve/knowledge_graph`
  - [x]  `/api/v1/admin/knowledge_bases/{kb_id}/chunks/retrieve`
  - [x]  `/api/v1/admin/knowledge_bases/{kb_id}/graph/retrieve` 
- [x] Integration with ChatService
- [ ] tests and fix bug

Feature PRs:
- merge `TiDBGraphEditor` to `TiDBGraphStore`, TiDBGraphStore will only
retain storage layer operactions, so it shouldn't depend on DsPy or LLM.
- `TiDBKnowledgeGraphIndex` will handle the merge entities parts.
- `KnowledgeGraphRetriever` will handle query compose parts.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Refactor Retrieve API
1 participant