feat: implement knowledge update functionality for bots #547
Conversation
- Added a new hook `useKnowledgeUpdate` to manage knowledge updates. - Integrated knowledge update feature in `BotCard` component with loading state. - Created `updateKnowledge` API in `BotsController` to handle knowledge updates. - Updated FastAPI router to include endpoint for knowledge updates, validating bot existence. - Refactored retrieval logic to check and update knowledge based on document changes.
|
The latest updates on your projects. Learn more about Vercel for Git ↗︎
|
WalkthroughThis PR introduces a knowledge update functionality for bots. It includes a new hook Changes
|
![petercat-assistant[bot]](https://avatars.githubusercontent.com/in/875736?s=60&v=4){kind=small}
| class UpdateKnowledgeRequest(BaseModel): | ||
| bot_id: str | ||
|
|
||
| @router.post("/rag/update_knowledge", dependencies=[Depends(verify_rate_limit)]) |
There was a problem hiding this comment.
Consider adding authentication checks to ensure that only authorized users can update the knowledge of a bot. This will prevent unauthorized access and potential misuse of the update functionality.
| existing_docs = ( | ||
| client.table(TABLE_NAME) | ||
| .select("id, file_sha") | ||
| .eq("repo_name", config.repo_name) | ||
| .eq("file_path", config.file_path) | ||
| .execute() | ||
| ) |
There was a problem hiding this comment.
我本地跑这里总超时, 不知道为啥, 还没能解决
There was a problem hiding this comment.
'canceling statement due to statement timeout'
| export function useKnowledgeUpdate() { | ||
| return useMutation({ | ||
| mutationKey: ['updateKnowledge'], | ||
| mutationFn: updateKnowledge |
There was a problem hiding this comment.
在完成更新后，可以将 useGetBotRagTask client 重刷一遍，更新下 taskInfo，这样可以让 BotCard 组件下的 任务状态 ICON 转起来
useKnowledgeUpdateto manage knowledge updates.BotCardcomponent with loading state.updateKnowledgeAPI inBotsControllerto handle knowledge updates.close #547