Ver código fonte

feat: save tags to chat data

Timothy J. Baek 1 ano atrás
pai
commit
1eec176313

+ 0 - 2
backend/apps/web/models/tags.py

@@ -101,7 +101,6 @@ class TagTable:
         if tag == None:
             tag = self.insert_new_tag(form_data.tag_name, user_id)
 
-        print(tag)
         id = str(uuid.uuid4())
         chatIdTag = ChatIdTagModel(
             **{
@@ -131,7 +130,6 @@ class TagTable:
             .order_by(ChatIdTag.timestamp.desc())
         ]
 
-        print(tag_names)
         return [
             TagModel(**model_to_dict(tag))
             for tag in Tag.select().where(Tag.name.in_(tag_names))

+ 0 - 1
backend/apps/web/routers/chats.py

@@ -134,7 +134,6 @@ async def get_chat_tags_by_id(id: str, user=Depends(get_current_user)):
     tags = Tags.get_tags_by_chat_id_and_user_id(id, user.id)
 
     if tags != None:
-        print(tags)
         return tags
     else:
         raise HTTPException(

+ 9 - 0
src/routes/(app)/+page.svelte

@@ -189,6 +189,7 @@
 						},
 						messages: messages,
 						history: history,
+						tags: [],
 						timestamp: Date.now()
 					});
 					await chats.set(await getChatList(localStorage.token));
@@ -690,11 +691,19 @@
 	const addTag = async (tagName) => {
 		const res = await addTagById(localStorage.token, $chatId, tagName);
 		tags = await getTags();
+
+		chat = await updateChatById(localStorage.token, $chatId, {
+			tags: tags
+		});
 	};
 
 	const deleteTag = async (tagName) => {
 		const res = await deleteTagById(localStorage.token, $chatId, tagName);
 		tags = await getTags();
+
+		chat = await updateChatById(localStorage.token, $chatId, {
+			tags: tags
+		});
 	};
 
 	const setChatTitle = async (_chatId, _title) => {

+ 8 - 0
src/routes/(app)/c/[id]/+page.svelte

@@ -707,11 +707,19 @@
 	const addTag = async (tagName) => {
 		const res = await addTagById(localStorage.token, $chatId, tagName);
 		tags = await getTags();
+
+		chat = await updateChatById(localStorage.token, $chatId, {
+			tags: tags.map((tag) => tag.name)
+		});
 	};
 
 	const deleteTag = async (tagName) => {
 		const res = await deleteTagById(localStorage.token, $chatId, tagName);
 		tags = await getTags();
+
+		chat = await updateChatById(localStorage.token, $chatId, {
+			tags: tags.map((tag) => tag.name)
+		});
 	};
 
 	onMount(async () => {