chore: code format

Author: tmg0

.github/workflows/publish.yml

@@ -1,4 +1,4 @@
-name: 'publish'
+name: publish
 
 on:
   workflow_dispatch:
@@ -14,11 +14,11 @@ jobs:
       fail-fast: false
       matrix:
         include:
-          - platform: 'macos-latest' # for Arm based macs (M1 and above).
-            args: '--target aarch64-apple-darwin'
-          - platform: 'macos-latest' # for Intel based macs.
-            args: '--target x86_64-apple-darwin'
-          - platform: 'windows-latest'
+          - platform: macos-latest # for Arm based macs (M1 and above).
+            args: --target aarch64-apple-darwin
+          - platform: macos-latest # for Intel based macs.
+            args: --target x86_64-apple-darwin
+          - platform: windows-latest
             args: ''
 
     runs-on: ${{ matrix.platform }}
@@ -33,7 +33,7 @@ jobs:
         uses: actions/setup-node@v4
         with:
           node-version: lts/*
-          cache: 'pnpm' # Set this to npm, yarn or pnpm.
+          cache: pnpm # Set this to npm, yarn or pnpm.
 
       - name: install Rust stable
         uses: dtolnay/rust-toolchain@stable # Set this to dtolnay/rust-toolchain@nightly
@@ -44,7 +44,7 @@ jobs:
       - name: Rust cache
         uses: swatinem/rust-cache@v2
         with:
-          workspaces: './src-tauri -> target'
+          workspaces: ./src-tauri -> target
 
       - name: install frontend dependencies
         # If you don't have `beforeBuildCommand` configured you may want to build your frontend here too.
@@ -55,8 +55,8 @@ jobs:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         with:
           tagName: v__VERSION__ # the action automatically replaces \_\_VERSION\_\_ with the app version.
-          releaseName: 'v__VERSION__'
-          releaseBody: 'See the assets to download this version and install.'
+          releaseName: v__VERSION__
+          releaseBody: See the assets to download this version and install.
           releaseDraft: true
           prerelease: false
-          args: ${{ matrix.args }}
+          args: ${{ matrix.args }}

app/components/AsssistantPopover.vue

@@ -3,27 +3,22 @@ import { platform } from '@tauri-apps/plugin-os'
 import ChevronUpDown from '~icons/heroicons/chevron-up-down'
 import Sparkles from '~icons/heroicons/sparkles-solid'
 
-const props = defineProps<{
-  table?: string
-  cursor?: Database
-}>()
-
 const IS_MACOS = platform() === 'macos'
 
-const { cursor, table } = toRefs(props)
 const router = useRouter()
 const domRef = ref()
 const msgRefs = useTemplateRef('msgRef')
 const { focused } = useFocus(domRef)
 const { enter } = useMagicKeys()
 const isLoading = ref(false)
 const { md } = useMdit()
-const isReady = computed(() => !!cursor.value)
 
 const { model: modelKey } = storeToRefs(useSettingsStore())
+const { system } = storeToRefs(useAssistantStore())
 const model = computed(() => MODULES.find(m => m.model === modelKey.value) ?? {})
 
-const { data, prompt, messages, execute, system } = useStreamText({
+const { data, prompt, messages, execute } = useStreamText({
+  system,
   onFinish() {
     messages.value.push({
       role: 'assistant',
@@ -36,15 +31,6 @@ const { data, prompt, messages, execute, system } = useStreamText({
   },
 })
 
-watchImmediate(() => [isReady.value, table.value], async ([v, t]) => {
-  if (v) {
-    const prompt = await generateTableSchemaPromptWithIndexRows([t], cursor.value!)
-    system.value = usePromptTemplate(TABLE_ASSISTANT_SYSTEM_PROMPT, {
-      tableInfo: prompt,
-    }).value
-  }
-})
-
 const conversation = computed(() => {
   return {
     messages: data.value ? [...messages.value, { role: 'assistant', content: data.value }] : messages.value,

app/composables/useAi.ts

@@ -1,6 +1,7 @@
 import { type CoreMessage, streamText } from 'ai'
 
 export interface UseStreamTextOptions {
+  system?: MaybeRef<string>
   onFinish?: () => void
 }
 
@@ -43,7 +44,6 @@ export function useAiProvider() {
 
 export function useStreamText(options: UseStreamTextOptions = {}) {
   const data = ref('')
-  const system = ref('')
   const prompt = ref('')
   const messages = ref<CoreMessage[]>([])
   const { model } = useAiProvider()
@@ -62,7 +62,7 @@ export function useStreamText(options: UseStreamTextOptions = {}) {
     const result = streamText({
       model: model.value,
       messages: [
-        system.value ? { role: 'system', content: system.value } : undefined,
+        unref(options.system) ? { role: 'system', content: unref(options.system) } : undefined,
         ...messages.value,
       ].filter(Boolean) as CoreMessage[],
       onFinish,
@@ -77,7 +77,6 @@ export function useStreamText(options: UseStreamTextOptions = {}) {
 
   return {
     data,
-    system,
     prompt,
     messages,
     execute,

app/pages/[id].vue

@@ -20,7 +20,6 @@ const cursor = computed(() => cursors.value[id.value])
 const instance = ref<Database | undefined>(undefined)
 const db = computed(() => parseConnectionURL(instance.value?.path).database)
 const { isFullscreen } = useTauriWindow()
-const table = useRouteParams<string>('name', '')
 
 const tabs = computed(() => [
   cursor.value ? { key: 'id-tables-name', icon: CircleStack } : undefined,
@@ -45,6 +44,10 @@ const abort = watchImmediate(connections, async (cnxs) => {
   }
 })
 
+function onNavi(key: string) {
+  router.replace({ name: key })
+}
+
 provide('__TABLITE:CURSOR', instance)
 
 preloadRouteComponents({ name: 'id-queries' })
@@ -62,14 +65,14 @@ preloadRouteComponents({ name: 'id-queries' })
         </Button>
       </div>
 
-      <AsssistantPopover :table="table" :cursor="instance" />
+      <AsssistantPopover />
     </div>
 
     <Separator />
 
     <div class="flex flex-1 h-0">
       <div class="flex flex-col items-center flex-shrink-0 border-r border-r-zinc-200 bg-zinc-100">
-        <div v-for="tab in tabs" :key="tab.key" class="flex items-center cursor-pointer justify-center relative" :class="[route.name === tab.key ? 'bg-zinc-200 text-zinc-600' : 'text-zinc-600/50 hover:text-zinc-600']" @click="router.replace({ name: tab.key })">
+        <div v-for="tab in tabs" :key="tab.key" class="flex items-center cursor-pointer justify-center relative" :class="[route.name === tab.key ? 'bg-zinc-200 text-zinc-600' : 'text-zinc-600/50 hover:text-zinc-600']" @click="onNavi(tab.key)">
           <component :is="tab.icon" class="flex-shrink-0 size-[18px] m-4" />
           <div v-if="route.name === tab.key" class="absolute top-0 bottom-0 left-0 w-0.5 bg-zinc-800" />
         </div>

app/pages/[id]/histories/index.vue

@@ -19,6 +19,10 @@ definePageMeta({
   keepalive: true,
 })
 
+defineAssistantContext({
+  system: '',
+})
+
 let editor: monaco.editor.IStandaloneCodeEditor
 
 const PLATFORM = platform()

app/pages/[id]/queries/index.vue

@@ -1,6 +1,10 @@
 <script setup lang="ts">
 import { hash } from 'ohash'
 
+defineAssistantContext({
+  system: '',
+})
+
 const store = useSettingsStore()
 const { language, alias, tags, googleAPIKey, deepseekApiKey, openrouterApiKey, model } = storeToRefs(store)
 const { connections } = storeToRefs(useConnectionStore())

app/pages/[id]/settings/index.vue

@@ -18,15 +18,29 @@ definePageMeta({
 })
 
 const cursor = inject<Ref<Database> | undefined>('__TABLITE:CURSOR', undefined)
-const selectedTable = useRouteParams('name', '')
-const { data, limit, offset, count, structure, primaryKeys, isLoading, backend, where, setup, execute } = useTable(selectedTable, cursor)
+const table = useRouteParams('name', '')
+const { data, limit, offset, count, structure, primaryKeys, isLoading, backend, where, execute } = useTable(table, cursor)
 const mode = ref('data')
 const columns = computed(() => structure.value.map(({ columnName }) => columnName))
 const changes = ref<Record<string, any>>({})
 const updates = ref<Update[]>([])
 const inserts = ref<Record<string, any>[]>([])
 const deletes = ref<string[]>([])
 const selectedRowKeys = ref([])
+const isReady = computed(() => !!cursor.value)
+
+watchImmediate(() => [isReady.value, table.value], async ([v, t]) => {
+  if (v) {
+    defineAssistantContext({
+      async system() {
+        const prompt = await generateTableSchemaPromptWithIndexRows([t], cursor.value!)
+        return usePromptTemplate(TABLE_ASSISTANT_SYSTEM_PROMPT, {
+          tableInfo: prompt,
+        })
+      },
+    })
+  }
+})
 
 const page = computed({
   get() {
@@ -69,7 +83,7 @@ const hasChanged = computed(() => {
 async function onSelectTable() {
   toast.dismiss()
   where.value = ''
-  changes.value = { [selectedTable.value]: {} }
+  changes.value = { [table.value]: {} }
   page.value = 1
   inserts.value = []
   deletes.value = []
@@ -88,7 +102,7 @@ async function onApplyFliters(value: string) {
 }
 
 async function onDisvardChanges() {
-  changes.value = selectedTable.value ? { [selectedTable.value]: {} } : {}
+  changes.value = table.value ? { [table.value]: {} } : {}
   inserts.value = []
   deletes.value = []
   toast.dismiss()
@@ -120,14 +134,14 @@ function onOpenUpdatesPreview(visible: boolean) {
 
   for (const _n of inserts.value) {
     const keys = Object.keys(_n)
-    sqls.push(`INSERT INTO \`${selectedTable.value}\` (${keys.join(', ')}) VALUES (${keys.map(k => normalizeQueryValue(_n[k], dataTypeMap.value[k])).join(', ')})`)
+    sqls.push(`INSERT INTO \`${table.value}\` (${keys.join(', ')}) VALUES (${keys.map(k => normalizeQueryValue(_n[k], dataTypeMap.value[k])).join(', ')})`)
   }
 
   for (const _d of deletes.value) {
     try {
       const json = JSON.parse(_d)
       const where = Object.entries(json).map(([k, v]) => `${k} = ${normalizeQueryValue(v, dataTypeMap.value[k])}`).join(' AND ')
-      sqls.push(`DELETE FROM \`${selectedTable.value}\` WHERE ${where}`)
+      sqls.push(`DELETE FROM \`${table.value}\` WHERE ${where}`)
     }
     catch {
       continue
@@ -154,7 +168,7 @@ async function onSave() {
   toast.dismiss()
   await applyUpdates()
   await execute()
-  changes.value = selectedTable.value ? { [selectedTable.value]: {} } : {}
+  changes.value = table.value ? { [table.value]: {} } : {}
   updates.value = []
   inserts.value = []
   deletes.value = []
@@ -177,21 +191,21 @@ function onDeleteRecords() {
   <div class="flex-1 h-full w-full flex flex-col">
     <ResizablePanelGroup direction="horizontal" class="h-0 flex-1 w-full">
       <ResizablePanel :default-size="22" :min-size="10" :max-size="50">
-        <TableSelector v-model:value="selectedTable" :cursor="cursor" :loading="isLoading" @after-select="onSelectTable" />
+        <TableSelector v-model:value="table" :cursor="cursor" :loading="isLoading" @after-select="onSelectTable" />
       </ResizablePanel>
 
       <ResizableHandle />
 
       <ResizablePanel class="h-full bg-white">
-        <div v-if="!selectedTable" class="w-full h-full flex items-center justify-center flex-col cursor-default">
+        <div v-if="!table" class="w-full h-full flex items-center justify-center flex-col cursor-default">
           <TextHoverEffect class="w-1/2 px-1" :stroke-width="1" text="TABLITE" />
         </div>
 
-        <div v-show="selectedTable" class="flex flex-col h-full">
+        <div v-show="table" class="flex flex-col h-full">
           <div class="p-4 flex justify-between items-center">
             <div class="ml-2 flex-1">
               <div class="font-semibold uppercase cursor-default">
-                {{ selectedTable }}
+                {{ table }}
               </div>
             </div>
 
@@ -229,7 +243,7 @@ function onDeleteRecords() {
 
           <div class="w-full h-0 flex-1 flex flex-col bg-zinc-50 -m-px">
             <VisTable
-              v-model:changes="changes[selectedTable]"
+              v-model:changes="changes[table]"
               v-model:inserts="inserts"
               v-model:selected-row-keys="selectedRowKeys"
               editable

app/pages/[id]/tables/[[name]].vue

@@ -17,4 +17,4 @@
     "lib": "@/lib"
   },
   "iconLibrary": "lucide"
-}
+}

components.json

@@ -29,6 +29,7 @@
     "ai": "^4.1.61",
     "destr": "^2.0.3",
     "embla-carousel-vue": "^8.5.2",
+    "klona": "^2.0.6",
     "lucide-vue-next": "^0.469.0",
     "markdown-it": "^14.1.0",
     "monaco-editor": "^0.52.2",