wip(desktop): progress

2 files changed, 15 insertions, 1 deletions

diff --git a/packages/desktop/src/components/prompt-input.tsx b/packages/desktop/src/components/prompt-input.tsx
index fbb643e58..bbd638e44 100644
--- a/packages/desktop/src/components/prompt-input.tsx
+++ b/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}>
                     <Tag>Free</Tag>
                   </Show>
+                  <Show when={i.latest}>
+                    <Tag>Latest</Tag>
+                  </Show>
                 </div>
               )}
             </SelectDialog>
diff --git a/packages/desktop/src/context/local.tsx b/packages/desktop/src/context/local.tsx
index 8223a36b9..58a65b0de 100644
--- a/packages/desktop/src/context/local.tsx
+++ b/packages/desktop/src/context/local.tsx
@@ -25,6 +25,7 @@ export type View = LocalFile["view"]
 
 export type LocalModel = Omit<Model, "provider"> & {
   provider: Provider
+  latest?: boolean
 }
 export type ModelKey = { providerID: string; modelID: string }
 
@@ -114,7 +115,17 @@ export const { use: useLocal, provider: LocalProvider } = createSimpleContext({
       })
 
       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)