update: integrate API

TabbyML · Nov 1, 2024 · 02bd4fd · 02bd4fd
1 parent 9a6b0a3
commit 02bd4fd
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 11 deletions.
diff --git a/crates/llama-cpp-server/llama.cpp b/crates/llama-cpp-server/llama.cpp
diff --git a/ee/tabby-ui/app/search/components/assistant-message-section.tsx b/ee/tabby-ui/app/search/components/assistant-message-section.tsx
@@ -89,7 +89,6 @@ export function AssistantMessageSection({
   } = useContext(SearchContext)
 
   const [isEditing, setIsEditing] = useState(false)
-
   const [showMoreSource, setShowMoreSource] = useState(false)
   const [relevantCodeHighlightIndex, setRelevantCodeHighlightIndex] = useState<
     number | undefined
@@ -213,7 +212,6 @@ export function AssistantMessageSection({
   const handleUpdateAssistantMessage = async (message: ConversationMessage) => {
     const errorMessage = await onUpdateMessage(message)
     if (errorMessage) {
-      // todo error handling
       return errorMessage
     } else {
       setIsEditing(false)

diff --git a/ee/tabby-ui/app/search/components/search.tsx b/ee/tabby-ui/app/search/components/search.tsx
@@ -167,18 +167,32 @@ export function Search() {
     return activePathname.match(regex)?.[1]?.split('-').pop()
   }, [activePathname])
 
+  const updateThreadMessage = useMutation(updateThreadMessageMutation)
+
   const onUpdateMessage = async (message: ConversationMessage) => {
     const messageIndex = messages.findIndex(o => o.id === message.id)
-    if (messageIndex > -1) {
-      // todo 1. call API
-      // 2. set messages
-      await setMessages(prev => {
-        const newMessages = [...prev]
-        newMessages[messageIndex] = message
-        return newMessages
+    if (messageIndex > -1 && threadId) {
+      const result = await updateThreadMessage({
+        input: {
+          threadId,
+          id: message.id,
+          content: message.content
+        }
       })
+      console.log(result)
+      console.log(result?.data?.updateThreadMessage)
+      if (result?.data?.updateThreadMessage) {
+        // 2. set messages
+        await setMessages(prev => {
+          const newMessages = [...prev]
+          newMessages[messageIndex] = message
+          return newMessages
+        })
+      } else {
+        // FIXME error handling
+        return result?.error?.message || 'Failed to save'
+      }
     } else {
-      // FIXME error handling
       return 'Failed to save'
     }
   }
@@ -891,6 +905,12 @@ const setThreadPersistedMutation = graphql(/* GraphQL */ `
   }
 `)
 
+const updateThreadMessageMutation = graphql(/* GraphQL */ `
+  mutation UpdateThreadMessage($input: UpdateMessageInput!) {
+    updateThreadMessage(input: $input)
+  }
+`)
+
 type HeaderProps = {
   threadIdFromURL?: string
   streamingDone?: boolean