Przeglądaj źródła

refac: tag restoration

Timothy J. Baek 6 miesięcy temu
rodzic
commit
e378f70f34
1 zmienionych plików z 3 dodań i 2 usunięć
  1. 3 2
      backend/open_webui/apps/webui/routers/chats.py

+ 3 - 2
backend/open_webui/apps/webui/routers/chats.py

@@ -114,11 +114,12 @@ async def import_chat(form_data: ChatImportForm, user=Depends(get_verified_user)
             tags = chat.meta.get("tags", [])
             for tag_id in tags:
                 tag_id = tag_id.replace(" ", "_").lower()
+                tag_name = " ".join([word.capitalize() for word in tag_id.split("_")])
                 if (
                     tag_id != "none"
-                    and Tags.get_tag_by_name_and_user_id(tag_id, user.id) is None
+                    and Tags.get_tag_by_name_and_user_id(tag_name, user.id) is None
                 ):
-                    Tags.insert_new_tag(tag_id, user.id)
+                    Tags.insert_new_tag(tag_name, user.id)
 
         return ChatResponse(**chat.model_dump())
     except Exception as e: