@budibase/frontend-core 3.24.2 → 3.24.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@budibase/frontend-core",
-  "version": "3.24.2",
+  "version": "3.24.4",
   "description": "Budibase frontend core libraries used in builder and client",
   "author": "Budibase",
   "license": "MPL-2.0",
@@ -18,5 +18,5 @@
     "shortid": "2.2.15",
     "socket.io-client": "^4.7.5"
   },
-  "gitHead": "237b6a0e4013f770eb766e79596e6f52bfcb6ea7"
+  "gitHead": "6d1812ff9a22270318eaf1f61f4e9af51bbac383"
 }

package/src/api/agents.ts

@@ -1,6 +1,8 @@
 import {
+  AgentFileUploadResponse,
   CreateAgentRequest,
   CreateAgentResponse,
+  FetchAgentFilesResponse,
   FetchAgentsResponse,
   ToolMetadata,
   UpdateAgentRequest,
@@ -15,9 +17,26 @@ export interface AgentEndpoints {
   createAgent: (agent: CreateAgentRequest) => Promise<CreateAgentResponse>
   updateAgent: (agent: UpdateAgentRequest) => Promise<UpdateAgentResponse>
   deleteAgent: (agentId: string) => Promise<{ deleted: true }>
+  fetchAgentFiles: (agentId: string) => Promise<FetchAgentFilesResponse>
+  uploadAgentFile: (
+    agentId: string,
+    file: File
+  ) => Promise<AgentFileUploadResponse>
+  deleteAgentFile: (
+    agentId: string,
+    fileId: string
+  ) => Promise<{ deleted: true }>
 }
 
 export const buildAgentEndpoints = (API: BaseAPIClient): AgentEndpoints => ({
+  fetchTools: async (aiconfigId?: string) => {
+    const query = aiconfigId
+      ? `?aiconfigId=${encodeURIComponent(aiconfigId)}`
+      : ""
+    return await API.get({
+      url: `/api/agent/tools${query}`,
+    })
+  },
   fetchAgents: async () => {
     return await API.get({
       url: "/api/agent",
@@ -44,12 +63,25 @@ export const buildAgentEndpoints = (API: BaseAPIClient): AgentEndpoints => ({
     })
   },
 
-  fetchTools: async (aiconfigId?: string) => {
-    const query = aiconfigId
-      ? `?aiconfigId=${encodeURIComponent(aiconfigId)}`
-      : ""
+  fetchAgentFiles: async (agentId: string) => {
     return await API.get({
-      url: `/api/agent/tools${query}`,
+      url: `/api/agent/${agentId}/files`,
+    })
+  },
+
+  uploadAgentFile: async (agentId: string, file: File) => {
+    const formData = new FormData()
+    formData.append("file", file)
+    return await API.post<FormData, AgentFileUploadResponse>({
+      url: `/api/agent/${agentId}/files`,
+      body: formData,
+      json: false,
+    })
+  },
+
+  deleteAgentFile: async (agentId: string, fileId: string) => {
+    return await API.delete({
+      url: `/api/agent/${agentId}/files/${fileId}`,
     })
   },
 })

package/src/api/chatApps.ts

@@ -6,6 +6,7 @@ import {
   ChatApp,
   FetchAgentHistoryResponse,
   UpdateChatAppRequest,
+  AgentMessageMetadata,
 } from "@budibase/types"
 import { Header } from "@budibase/shared-core"
 import { BaseAPIClient } from "./types"
@@ -16,7 +17,7 @@ export interface ChatAppEndpoints {
   streamChatConversation: (
     chat: ChatConversationRequest,
     workspaceId: string
-  ) => Promise<AsyncIterable<UIMessage>>
+  ) => Promise<AsyncIterable<UIMessage<AgentMessageMetadata>>>
   deleteChatConversation: (
     chatConversationId: string,
     chatAppId: string

package/src/api/configs.ts

@@ -6,6 +6,7 @@ import {
   FindConfigResponse,
   GetPublicOIDCConfigResponse,
   GetPublicSettingsResponse,
+  GetPublicTranslationsResponse,
   OIDCLogosConfig,
   SaveConfigRequest,
   SaveConfigResponse,
@@ -17,6 +18,9 @@ export interface ConfigEndpoints {
   getConfig: (type: ConfigType) => Promise<FindConfigResponse>
   getTenantConfig: (tentantId: string) => Promise<GetPublicSettingsResponse>
   getOIDCConfigs: (tenantId: string) => Promise<GetPublicOIDCConfigResponse>
+  getPublicTranslations: (
+    tenantId?: string
+  ) => Promise<GetPublicTranslationsResponse>
   getOIDCLogos: () => Promise<Config<OIDCLogosConfig>>
   saveConfig: (config: SaveConfigRequest) => Promise<SaveConfigResponse>
   deleteConfig: (id: string, rev: string) => Promise<DeleteConfigResponse>
@@ -79,6 +83,13 @@ export const buildConfigEndpoints = (API: BaseAPIClient): ConfigEndpoints => ({
     })
   },
 
+  getPublicTranslations: async tenantId => {
+    const query = tenantId ? `?tenantId=${tenantId}` : ""
+    return await API.get({
+      url: `/api/global/configs/public/translations${query}`,
+    })
+  },
+
   /**
    * Gets the checklist for a specific tenant.
    * @param tenantId the tenant ID to get the checklist for

package/src/api/index.ts

@@ -56,6 +56,8 @@ import { buildDeploymentEndpoints } from "./deploy"
 import { buildWorkspaceFavouriteEndpoints } from "./workspaceFavourites"
 import { buildRecaptchaEndpoints } from "./recaptcha"
 import { buildAIConfigEndpoints } from "./aiConfig"
+import { buildRagConfigEndpoints } from "./ragConfigs"
+import { buildVectorDbEndpoints } from "./vectorDbs"
 
 export type { APIClient } from "./types"
 
@@ -326,5 +328,7 @@ export const createAPIClient = (config: APIClientConfig = {}): APIClient => {
     resource: buildResourceEndpoints(API),
     recaptcha: buildRecaptchaEndpoints(API),
     aiConfig: buildAIConfigEndpoints(API),
+    ragConfig: buildRagConfigEndpoints(API),
+    vectorDb: buildVectorDbEndpoints(API),
   }
 }

package/src/api/ragConfigs.ts

@@ -0,0 +1,44 @@
+import {
+  CreateRagConfigRequest,
+  RagConfig,
+  RagConfigListResponse,
+  UpdateRagConfigRequest,
+} from "@budibase/types"
+import { BaseAPIClient } from "./types"
+
+export interface RagConfigEndpoints {
+  fetch: () => Promise<RagConfigListResponse>
+  create: (config: CreateRagConfigRequest) => Promise<RagConfig>
+  update: (config: UpdateRagConfigRequest) => Promise<RagConfig>
+  delete: (id: string) => Promise<{ deleted: true }>
+}
+
+export const buildRagConfigEndpoints = (
+  API: BaseAPIClient
+): RagConfigEndpoints => ({
+  fetch: async () => {
+    return await API.get({
+      url: "/api/ragconfig",
+    })
+  },
+
+  create: async config => {
+    return await API.post({
+      url: "/api/ragconfig",
+      body: config,
+    })
+  },
+
+  update: async config => {
+    return await API.put({
+      url: "/api/ragconfig",
+      body: config,
+    })
+  },
+
+  delete: async id => {
+    return await API.delete({
+      url: `/api/ragconfig/${id}`,
+    })
+  },
+})

package/src/api/types.ts

@@ -43,6 +43,8 @@ import { DeploymentEndpoints } from "./deploy"
 import { WorkspaceFavouriteEndpoints } from "./workspaceFavourites"
 import { RecaptchaEndpoints } from "./recaptcha"
 import { AIConfigEndpoints } from "./aiConfig"
+import { RagConfigEndpoints } from "./ragConfigs"
+import { VectorDbEndpoints } from "./vectorDbs"
 
 export enum HTTPMethod {
   POST = "POST",
@@ -157,4 +159,6 @@ export type APIClient = BaseAPIClient &
     deployment: DeploymentEndpoints
     recaptcha: RecaptchaEndpoints
     aiConfig: AIConfigEndpoints
+    ragConfig: RagConfigEndpoints
+    vectorDb: VectorDbEndpoints
   }

package/src/api/vectorDbs.ts

@@ -0,0 +1,44 @@
+import {
+  CreateVectorDbRequest,
+  UpdateVectorDbRequest,
+  VectorDb,
+  VectorDbListResponse,
+} from "@budibase/types"
+import { BaseAPIClient } from "./types"
+
+export interface VectorDbEndpoints {
+  fetch: () => Promise<VectorDbListResponse>
+  create: (config: CreateVectorDbRequest) => Promise<VectorDb>
+  update: (config: UpdateVectorDbRequest) => Promise<VectorDb>
+  delete: (id: string) => Promise<{ deleted: true }>
+}
+
+export const buildVectorDbEndpoints = (
+  API: BaseAPIClient
+): VectorDbEndpoints => ({
+  fetch: async () => {
+    return await API.get({
+      url: "/api/vectordb",
+    })
+  },
+
+  create: async config => {
+    return await API.post({
+      url: "/api/vectordb",
+      body: config,
+    })
+  },
+
+  update: async config => {
+    return await API.put({
+      url: "/api/vectordb",
+      body: config,
+    })
+  },
+
+  delete: async id => {
+    return await API.delete({
+      url: `/api/vectordb/${id}`,
+    })
+  },
+})

package/src/components/Chatbox/index.svelte

@@ -1,6 +1,7 @@
 <script lang="ts">
   import { MarkdownViewer, notifications } from "@budibase/bbui"
   import type {
+    AgentMessageMetadata,
     ChatConversation,
     ChatConversationRequest,
   } from "@budibase/types"
@@ -11,8 +12,8 @@
   import { onMount } from "svelte"
   import { createEventDispatcher } from "svelte"
   import { createAPIClient } from "@budibase/frontend-core"
-  import type { UIMessage } from "ai"
   import { v4 as uuidv4 } from "uuid"
+  import type { UIMessage } from "ai"
 
   export let workspaceId: string
   export let API = createAPIClient({
@@ -114,7 +115,7 @@
       }
     }
 
-    const userMessage: UIMessage = {
+    const userMessage: UIMessage<AgentMessageMetadata> = {
       id: uuidv4(),
       role: "user",
       parts: [{ type: "text", text: inputValue }],
@@ -138,10 +139,35 @@
         workspaceId
       )
 
-      let streamedMessages: UIMessage[] = [...updatedChat.messages]
+      let streamedMessages = [...updatedChat.messages]
 
       for await (const message of messageStream) {
-        streamedMessages = [...streamedMessages, message]
+        if (message?.id) {
+          const existingIndex = streamedMessages.findIndex(
+            existing => existing.id === message.id
+          )
+          if (existingIndex !== -1) {
+            streamedMessages = streamedMessages.map((existing, index) =>
+              index === existingIndex ? message : existing
+            )
+          } else {
+            streamedMessages = [...streamedMessages, message]
+          }
+        } else if (message?.role === "assistant") {
+          const lastIndex = [...streamedMessages]
+            .reverse()
+            .findIndex(existing => existing.role === "assistant")
+          if (lastIndex !== -1) {
+            const targetIndex = streamedMessages.length - 1 - lastIndex
+            streamedMessages = streamedMessages.map((existing, index) =>
+              index === targetIndex ? message : existing
+            )
+          } else {
+            streamedMessages = [...streamedMessages, message]
+          }
+        } else {
+          streamedMessages = [...streamedMessages, message]
+        }
         chat = {
           ...updatedChat,
           messages: streamedMessages,
@@ -285,6 +311,27 @@
               </div>
             {/if}
           {/each}
+          {#if message.metadata?.ragSources?.length}
+            <div class="sources">
+              <div class="sources-title">Sources</div>
+              <ul>
+                {#each message.metadata.ragSources as source (source.sourceId)}
+                  <li class="source-item">
+                    <span class="source-name"
+                      >{source.filename || source.sourceId}</span
+                    >
+                    {#if source.chunkCount > 0}
+                      <span class="source-count"
+                        >({source.chunkCount} chunk{source.chunkCount === 1
+                          ? ""
+                          : "s"})</span
+                      >
+                    {/if}
+                  </li>
+                {/each}
+              </ul>
+            </div>
+          {/if}
         </div>
       {/if}
     {/each}
@@ -480,4 +527,42 @@
     color: var(--spectrum-global-color-gray-800);
     font-style: italic;
   }
+
+  .sources {
+    margin-top: var(--spacing-m);
+    padding-top: var(--spacing-s);
+    border-top: 1px solid var(--grey-3);
+  }
+
+  .sources-title {
+    font-size: 13px;
+    text-transform: uppercase;
+    letter-spacing: 0.08em;
+    color: var(--spectrum-global-color-gray-600);
+    margin-bottom: var(--spacing-xs);
+  }
+
+  .sources ul {
+    list-style: none;
+    padding: 0;
+    margin: 0;
+    display: flex;
+    flex-direction: column;
+    gap: 4px;
+  }
+
+  .source-item {
+    display: flex;
+    gap: 8px;
+    font-size: 14px;
+    color: var(--spectrum-global-color-gray-900);
+  }
+
+  .source-name {
+    font-weight: 500;
+  }
+
+  .source-count {
+    color: var(--spectrum-global-color-gray-600);
+  }
 </style>