add functionality to clone shared chats

2025-01-03 00:08:49 -05:00
parent 87ba39df57
commit 2444327a47
4 changed files with 90 additions and 3 deletions
@@ -463,6 +463,31 @@ async def clone_chat_by_id(id: str, user=Depends(get_verified_user)):
        )


+############################
+# CloneChatByShareId
+############################
+
+
+@router.post("/{share_id}/clone_shared", response_model=Optional[ChatResponse])
+async def clone_chat_by_share_id(share_id: str, user=Depends(get_verified_user)):
+    chat = Chats.get_chat_by_share_id(share_id)
+    if chat:
+        updated_chat = {
+            **chat.chat,
+            "originalChatId": chat.id,
+            "branchPointMessageId": chat.chat["history"]["currentId"],
+            "title": f"Clone of {chat.title}",
+        }
+
+        chat = Chats.insert_new_chat(user.id, ChatForm(**{"chat": updated_chat}))
+        return ChatResponse(**chat.model_dump())
+    else:
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED, detail=ERROR_MESSAGES.DEFAULT()
+        )
+
+
+
 ############################
 # ArchiveChat
 ############################