From f67adede69a57881023d80128f27df1834bfffa0 Mon Sep 17 00:00:00 2001
From: Zhang Minghan <zmh13054618081@dingtalk.com>
Date: Wed, 13 Mar 2024 21:54:39 +0800
Subject: [PATCH] feat: auto switch model when using `use message` actions in
 shared conversation (#116)

---
 app/src/api/sharing.ts                    |  1 +
 app/src/dialogs/ShareManagementDialog.tsx | 26 +++++++++++------------
 app/src/routes/Sharing.tsx                |  7 +++++-
 app/src/store/chat.ts                     |  8 +++++++
 manager/conversation/shared.go            |  5 +++--
 5 files changed, 31 insertions(+), 16 deletions(-)

diff --git a/app/src/api/sharing.ts b/app/src/api/sharing.ts
index 0707b23..40d147d 100644
--- a/app/src/api/sharing.ts
+++ b/app/src/api/sharing.ts
@@ -18,6 +18,7 @@ export type ViewData = {
   name: string;
   username: string;
   time: string;
+  model?: string;
   messages: Message[];
 };
 
diff --git a/app/src/dialogs/ShareManagementDialog.tsx b/app/src/dialogs/ShareManagementDialog.tsx
index 1724c3e..888edfb 100644
--- a/app/src/dialogs/ShareManagementDialog.tsx
+++ b/app/src/dialogs/ShareManagementDialog.tsx
@@ -131,20 +131,20 @@ function ShareManagementDialog() {
     <Dialog open={open} onOpenChange={(open) => dispatch(setDialog(open))}>
       <DialogContent>
         <DialogHeader>
-            <DialogTitle className={`mb-4`}>{t("share.manage")}</DialogTitle>
-            {data.length > 0 ? (
-              <ScrollArea className={`max-h-[60vh]`}>
-                <DialogDescription className={`share-table`}>
-                  <ShareTable data={data} />
-                </DialogDescription>
-              </ScrollArea>
-            ) : (
-              <DialogDescription>
-                <p className={`text-center select-none mt-6 mb-2`}>
-                  {t("conversation.empty")}
-                </p>
+          <DialogTitle className={`mb-4`}>{t("share.manage")}</DialogTitle>
+          {data.length > 0 ? (
+            <ScrollArea className={`max-h-[60vh]`}>
+              <DialogDescription className={`share-table`}>
+                <ShareTable data={data} />
               </DialogDescription>
-            )}
+            </ScrollArea>
+          ) : (
+            <DialogDescription>
+              <p className={`text-center select-none mt-6 mb-2`}>
+                {t("conversation.empty")}
+              </p>
+            </DialogDescription>
+          )}
         </DialogHeader>
         <DialogFooter>
           <Button variant={`outline`} onClick={() => dispatch(closeDialog())}>
diff --git a/app/src/routes/Sharing.tsx b/app/src/routes/Sharing.tsx
index ce1545d..2a3a90d 100644
--- a/app/src/routes/Sharing.tsx
+++ b/app/src/routes/Sharing.tsx
@@ -40,7 +40,7 @@ function SharingForm({ data }: SharingFormProps) {
   const { t } = useTranslation();
   const { toast } = useToast();
   const mobile = useMobile();
-  const { mask: setMask } = useConversationActions();
+  const { mask: setMask, selected: setModel } = useConversationActions();
   const [maximized, setMaximized] = useState(false);
   const container = useRef<HTMLDivElement>(null);
   const date = new Date(data.time);
@@ -163,6 +163,11 @@ function SharingForm({ data }: SharingFormProps) {
               name: data.name,
               context: message,
             });
+            setModel(data?.model);
+
+            console.debug(
+              `[sharing] switch to conversation (name: ${data.name}, model: ${data.model})`,
+            );
             await router.navigate("/");
           }}
         >
diff --git a/app/src/store/chat.ts b/app/src/store/chat.ts
index 430de08..3dedc91 100644
--- a/app/src/store/chat.ts
+++ b/app/src/store/chat.ts
@@ -424,6 +424,8 @@ export function useConversationActions() {
   const current = useSelector(selectCurrent);
   const mask = useSelector(selectMaskItem);
 
+  const supportModels = useSelector(selectSupportModels);
+
   return {
     toggle: async (id: number) => {
       const conversation = conversations[id];
@@ -478,6 +480,12 @@ export function useConversationActions() {
         dispatch(setCurrent(-1));
       }
     },
+    selected: (model?: string) => {
+      if (!model || model === "") return;
+      if (!supportModels.map((item) => item.id).includes(model)) return;
+
+      dispatch(setModel(model));
+    },
   };
 }
 
diff --git a/manager/conversation/shared.go b/manager/conversation/shared.go
index cf249dd..0f80f30 100644
--- a/manager/conversation/shared.go
+++ b/manager/conversation/shared.go
@@ -21,6 +21,7 @@ type SharedForm struct {
 	Username string            `json:"username"`
 	Name     string            `json:"name"`
 	Messages []globals.Message `json:"messages"`
+	Model    string            `json:"model"`
 	Time     time.Time         `json:"time"`
 }
 
@@ -138,12 +139,12 @@ func GetSharedConversation(db *sql.DB, hash string) (*SharedForm, error) {
 	)
 	if err := globals.QueryRowDb(db, `
 		SELECT auth.username, sharing.refs, sharing.updated_at, conversation.conversation_name,
-		       sharing.user_id, sharing.conversation_id
+		       sharing.user_id, sharing.conversation_id, conversation.model
 		FROM sharing
 		INNER JOIN auth ON auth.id = sharing.user_id
 		INNER JOIN conversation ON conversation.conversation_id = sharing.conversation_id AND conversation.user_id = sharing.user_id
 		WHERE sharing.hash = ?
-	`, hash).Scan(&shared.Username, &ref, &updated, &shared.Name, &uid, &cid); err != nil {
+	`, hash).Scan(&shared.Username, &ref, &updated, &shared.Name, &uid, &cid, &shared.Model); err != nil {
 		return nil, err
 	}