From 3db734a01e424fffe863d66f22e60791c37a31d9 Mon Sep 17 00:00:00 2001
From: "shiweisong.ssw" <shiweisong.ssw@digital-engine.com>
Date: Tue, 27 Jun 2023 17:39:05 +0800
Subject: [PATCH] feat: sync documents

---
 datacenter/app/datastores/documents/page.tsx | 48 ++++++++++++++++----
 1 file changed, 38 insertions(+), 10 deletions(-)

diff --git a/datacenter/app/datastores/documents/page.tsx b/datacenter/app/datastores/documents/page.tsx
index 55d8baa26..668313c99 100644
--- a/datacenter/app/datastores/documents/page.tsx
+++ b/datacenter/app/datastores/documents/page.tsx
@@ -4,6 +4,7 @@ import { useRouter, useSearchParams } from 'next/navigation'
 import React, { useState, useEffect } from 'react'
 import { Button, Table } from '@/lib/mui'
 import moment from 'moment'
+import { message } from 'antd'
 
 const Documents = () => {
   const router = useRouter()
@@ -51,16 +52,43 @@ const Documents = () => {
               <td>{row.status}</td>
               <td>
                 {
-                  <Button
-                    variant="outlined"
-                    onClick={() => {
-                      router.push(
-                        `/datastores/documents/chunklist?spacename=${spaceName}&documentid=${row.id}`
-                      )
-                    }}
-                  >
-                    Detail of Chunks
-                  </Button>
+                  <>
+                    <Button
+                      variant="outlined"
+                      onClick={async () => {
+                        const res = await fetch(
+                          `http://localhost:8000/knowledge/${spaceName}/document/sync`,
+                          {
+                            method: 'POST',
+                            headers: {
+                              'Content-Type': 'application/json'
+                            },
+                            body: JSON.stringify({
+                              doc_ids: [row.id]
+                            })
+                          }
+                        )
+                        const data = await res.json()
+                        if (data.success) {
+                          message.success('success');
+                        } else {
+                          message.error(data.err_msg || 'failed');
+                        }
+                      }}
+                    >
+                      Synch
+                    </Button>
+                    <Button
+                      variant="outlined"
+                      onClick={() => {
+                        router.push(
+                          `/datastores/documents/chunklist?spacename=${spaceName}&documentid=${row.id}`
+                        )
+                      }}
+                    >
+                      Detail of Chunks
+                    </Button>
+                  </>
                 }
               </td>
             </tr>