tweak context preset

agnaistic · Sep 8, 2024 · bf8b594 · bf8b594
1 parent 01d3c0c
commit bf8b594
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/web/pages/Chat/ChatHeader.tsx b/web/pages/Chat/ChatHeader.tsx
@@ -78,6 +78,8 @@ const ChatNav: Component<NavProps> = (props) => {
     () => props.ctx.chat?.userId === props.ctx.user?._id && props.ctx.chat?.mode !== 'companion'
   )
 
+  const canModel = createMemo(() => props.ctx.info?.adapter === 'agnaistic')
+
   const size = 20
 
   return (
@@ -126,7 +128,7 @@ const ChatNav: Component<NavProps> = (props) => {
         </Nav.Item>
       </Show>
 
-      <Show when={props.ctx.info?.adapter === 'agnaistic'}>
+      <Show when={canModel()}>
         <div class="flex w-full justify-center">
           <AgnaisticModel inherit={props.ctx.preset} />
         </div>

diff --git a/web/store/context.tsx b/web/store/context.tsx
@@ -14,6 +14,7 @@ import { getRgbaFromVar } from '../shared/colors'
 import { MsgState, msgStore } from './message'
 import { ChatTree } from '/common/chat'
 import { presetStore } from './presets'
+import { getChatPreset } from '/common/prompt'
 
 export type ContextState = {
   tooltip?: string | JSX.Element
@@ -139,8 +140,8 @@ export function ContextProvider(props: { children: any }) {
   })
 
   const preset = createMemo(() => {
-    const id = chats.active?.chat.genPreset || users.user?.defaultPreset
-    const match = presets.presets.find((p) => p._id === id)
+    if (!chats.active?.chat || !users.user) return
+    const match = getChatPreset(chats.active.chat, users.user, presets.presets)
     return match
   })