Feat/release 1 17

src/bizy_server/api_client.py

@@ -360,6 +360,25 @@ async def fork_model_version(self, version_id: int) -> tuple[None, ErrorNo | Non
             print(f"\033[31m[BizyAir]\033[0m Fail to fork model version: {str(e)}")
             return None, errnos.FORK_MODEL_VERSION
 
+    async def unfork_model_version(
+        self, version_id: int
+    ) -> tuple[None, ErrorNo | None]:
+        server_url = f"{BIZYAIR_SERVER_ADDRESS}/bizy_models/versions/{version_id}/fork"
+
+        headers, err = self.auth_header()
+        if err is not None:
+            return None, err
+
+        try:
+            ret, err = await self.do_delete(server_url, headers=headers)
+            if err is not None:
+                return None, err
+
+            return None, None
+        except Exception as e:
+            print(f"\033[31m[BizyAir]\033[0m Fail to unfork model version: {str(e)}")
+            return None, errnos.UNFORK_MODEL_VERSION
+
     async def update_model(
         self, model_id: int, name: str, type_: str, versions: list[dict]
     ) -> tuple[dict | None, ErrorNo | None]:

src/bizy_server/errno.py

@@ -86,3 +86,4 @@ class errnos:
     QUERY_OFFICIAL_MODELS = ErrorNo(
         500, 500129, None, "Failed to query official models"
     )
+    UNFORK_MODEL_VERSION = ErrorNo(500, 500130, None, "Failed to unfork model version")

src/bizy_server/server.py

@@ -381,6 +381,29 @@ async def fork_model_version(request):
                 print(f"\033[31m[BizyAir]\033[0m Fail to fork model version: {str(e)}")
                 return ErrResponse(errnos.FORK_MODEL_VERSION)
 
+        @self.prompt_server.routes.delete(
+            f"/{COMMUNITY_API}/models/fork/{{model_version_id}}"
+        )
+        async def unfork_model_version(request):
+            try:
+                # 获取version_id参数
+                version_id = request.match_info["model_version_id"]
+                if not version_id:
+                    return ErrResponse(errnos.INVALID_MODEL_VERSION_ID)
+
+                # 调用API fork模型版本
+                _, err = await self.api_client.unfork_model_version(version_id)
+                if err:
+                    return ErrResponse(err)
+
+                return OKResponse(None)
+
+            except Exception as e:
+                print(
+                    f"\033[31m[BizyAir]\033[0m Fail to unfork model version: {str(e)}"
+                )
+                return ErrResponse(errnos.FORK_MODEL_VERSION)
+
         @self.prompt_server.routes.post(
             f"/{COMMUNITY_API}/models/like/{{model_version_id}}"
         )