Browse Source

wip(desktop): progress

Adam 2 months ago
parent
commit
f20d6e8555

+ 3 - 0
packages/desktop/src/components/prompt-input.tsx

@@ -489,6 +489,9 @@ export const PromptInput: Component<PromptInputProps> = (props) => {
                   <Show when={!i.cost || i.cost?.input === 0}>
                   <Show when={!i.cost || i.cost?.input === 0}>
                     <Tag>Free</Tag>
                     <Tag>Free</Tag>
                   </Show>
                   </Show>
+                  <Show when={i.latest}>
+                    <Tag>Latest</Tag>
+                  </Show>
                 </div>
                 </div>
               )}
               )}
             </SelectDialog>
             </SelectDialog>

+ 12 - 1
packages/desktop/src/context/local.tsx

@@ -25,6 +25,7 @@ export type View = LocalFile["view"]
 
 
 export type LocalModel = Omit<Model, "provider"> & {
 export type LocalModel = Omit<Model, "provider"> & {
   provider: Provider
   provider: Provider
+  latest?: boolean
 }
 }
 export type ModelKey = { providerID: string; modelID: string }
 export type ModelKey = { providerID: string; modelID: string }
 
 
@@ -114,7 +115,17 @@ export const { use: useLocal, provider: LocalProvider } = createSimpleContext({
       })
       })
 
 
       const list = createMemo(() =>
       const list = createMemo(() =>
-        sync.data.provider.flatMap((p) => Object.values(p.models).map((m) => ({ ...m, provider: p }) as LocalModel)),
+        sync.data.provider.flatMap((p) =>
+          Object.values(p.models).map(
+            (m) =>
+              ({
+                ...m,
+                name: m.name.replace("(latest)", "").trim(),
+                provider: p,
+                latest: m.name.includes("(latest)"),
+              }) as LocalModel,
+          ),
+        ),
       )
       )
       const find = (key: ModelKey) => list().find((m) => m.id === key?.modelID && m.provider.id === key.providerID)
       const find = (key: ModelKey) => list().find((m) => m.id === key?.modelID && m.provider.id === key.providerID)