feat(deploy): 支持提交部署审批后显示issue地址 & 获取审批情况时检查 issue 状态并同步数据表 (#527)

* fix(login): fix the login component login url * fix(deploy_website): 判断部署网站链接url合法性 * docs(doc): update i18n doc * feat(deploy): 支持显示部署issue地址 & 获取审批情况时检查 issue 状态并同步数据表 * renmae bot_approval_dao function name * add space for text * fix(botmodel): set default value for top_p
petercat-ai · Nov 27, 2024 · cc0bdb3 · cc0bdb3
1 parent 1746b5a
commit cc0bdb3
Show file tree

Hide file tree

Showing 9 changed files with 94 additions and 14 deletions.
diff --git a/client/app/factory/edit/components/DeployBotModal/index.tsx b/client/app/factory/edit/components/DeployBotModal/index.tsx
@@ -41,6 +41,7 @@ const MyBotDeployModal: React.FC<IModalProps> = ({ isOpen, onClose }) => {
   } = useBindBotToRepo();
   const {
     publicBot,
+    data: publicBotResult,
     isLoading: isPublicBotLoading,
     isSuccess: isPublicBotSuccess,
   } = usePublicBot();
@@ -51,6 +52,7 @@ const MyBotDeployModal: React.FC<IModalProps> = ({ isOpen, onClose }) => {
   } = useUnPublicBot();
   const {
     deployWebsite,
+    data: deployWebsiteResult,
     isLoading: isDeployWebsiteLoading,
     isSuccess: isDeployWebsiteSuccess,
   } = useDeployWebsite();
@@ -75,10 +77,11 @@ const MyBotDeployModal: React.FC<IModalProps> = ({ isOpen, onClose }) => {
         checked: !!botProfile.domain_whitelist[0],
         targetUrl: botProfile.domain_whitelist[0],
       },
-      appInstalledRepo: peterCatBotRepos?.map((item: any) => ({
-        repo_id: item.repo_id,
-        checked: item.robot_id === botProfile.id,
-      }))??[],
+      appInstalledRepo:
+        peterCatBotRepos?.map((item: any) => ({
+          repo_id: item.repo_id,
+          checked: item.robot_id === botProfile.id,
+        })) ?? [],
     };
   }, [botProfile, peterCatBotRepos]);
 
@@ -148,6 +151,25 @@ const MyBotDeployModal: React.FC<IModalProps> = ({ isOpen, onClose }) => {
     isDeployWebsiteSuccess ||
     isBindBotSuccess;
 
+  const renderResultPath = (result: { approval_path: any }, text: string) => {
+    return (
+      <>
+        {result?.approval_path ? (
+          <div className="w-full flex flex-col gap-1 justify-center items-center">
+            <span>
+              {text}
+              {" "}
+              {I18N.DeployBotModal.DeployContent.shenHeZhong}
+            </span>
+            <a href={result.approval_path} target="_blank" className="block max-w-full break-words">
+              {result.approval_path}
+            </a>
+          </div>
+        ) : null}
+      </>
+    );
+  };
+
   return (
     <>
       <Modal
@@ -169,6 +191,16 @@ const MyBotDeployModal: React.FC<IModalProps> = ({ isOpen, onClose }) => {
                       <span>{I18N.DeployBotModal.index.buShuChengGong}</span>
                     </div>
                   </ModalHeader>
+                  <ModalBody className="py-[0px] flex flex-col gap-2 justify-center items-center">
+                    {renderResultPath(
+                      publicBotResult,
+                      I18N.DeployBotModal.DeployContent.gongKaiDaoPE,
+                    )}
+                    {renderResultPath(
+                      deployWebsiteResult,
+                      I18N.DeployBotModal.DeployContent.buShuDaoWoDe,
+                    )}
+                  </ModalBody>
                   <ModalFooter className="flex justify-center items-center">
                     <Button
                       className="border-[1.5px] border-[#3F3F46] rounded-[46px] bg-[#3F3F46] text-white"
@@ -196,7 +228,7 @@ const MyBotDeployModal: React.FC<IModalProps> = ({ isOpen, onClose }) => {
                         deployInfo={deployInfo}
                         websiteApproval={websiteApproval}
                         marketApproval={marketApproval}
-                        peterCatBotRepos={peterCatBotRepos??[]}
+                        peterCatBotRepos={peterCatBotRepos ?? []}
                         onChange={handleDeployChange}
                       />
                     </Spinner>

diff --git a/client/app/hooks/useBot.ts b/client/app/hooks/useBot.ts
@@ -153,7 +153,7 @@ export function usePublicBot() {
     mutationFn: publicBot,
   });
   return {
-    data: mutation.data,
+    data: mutation.data?.data,
     publicBot: mutation.mutate,
     isLoading: mutation.isPending,
     error: mutation.error,
@@ -178,7 +178,7 @@ export function useDeployWebsite() {
     mutationFn: deployWebsite,
   });
   return {
-    data: mutation.data,
+    data: mutation.data?.data,
     deployWebsite: mutation.mutate,
     isLoading: mutation.isPending,
     error: mutation.error,

diff --git a/client/app/market/page.tsx b/client/app/market/page.tsx
@@ -66,7 +66,6 @@ export default function Market() {
   return (
     <div>
       <div className="grid grid-flow-row-dense gap-8 justify-items-center px-[40px] grid-cols-2 sm:grid-cols-1 md:grid-cols-2 lg:grid-cols-3 xl:grid-cols-4 2xl:grid-cols-5">
-        {/* <PublishBotEntity area="list" /> */}
         {!isEmpty(bots) &&
           map(bots, (bot: Bot) => (
             <BotCard key={bot.id} bot={bot} handleCardClick={handleCardClick} />

diff --git a/server/bot/router.py b/server/bot/router.py
@@ -433,15 +433,32 @@ def get_bot_approval_config(
     status: Optional[ApprovalStatus] = Query(
         None, description="approval status ,open or closed"
     ),
+    user: Annotated[User | None, Depends(get_user)] = None,
 ):
     try:
         bot_approval_dao = BotApprovalDAO()
-        (success, data) = bot_approval_dao.query_by_bot_id(bot_id)
+        (success, data) = bot_approval_dao.query_by_id_status(bot_id, status.value)
         if not success:
             raise ValueError(data)
-        # filter data by status
         if status:
             data = list(filter(lambda x: x["approval_status"] == status.value, data))
+            auth = Auth.Token(token=user.access_token)
+            g = Github(auth=auth)
+            repo = g.get_repo(OFFICIAL_REPO)
+            for item in data:
+                if not item["approval_path"]:
+                    continue
+                issue_num = int(item["approval_path"].split("/")[-1])
+                issue = repo.get_issue(issue_num)
+                if issue.state == "closed":
+                    item["approval_status"] = ApprovalStatus.CLOSED
+                    # update approval status to closed
+                    bot_approval_dao.update_approval_status(
+                        item["id"], ApprovalStatus.CLOSED.value
+                    )
+        # filter again
+        data = list(filter(lambda x: x["approval_status"] == status.value, data))
+        # check data item issue is closed
         return {"data": data, "status": 200}
     except Exception as e:
         return JSONResponse(content={"success": False, "errorMessage": e})
diff --git a/server/core/dao/botApprovalDAO.py b/server/core/dao/botApprovalDAO.py
@@ -27,12 +27,13 @@ def create(self, data: BotApproval):
             print("Error: ", e)
             return False, {"message": "approval creation failed"}
 
-    def query_by_bot_id(self, bot_id: str):
+    def query_by_id_status(self, bot_id: str, status: str):
         try:
             bot_approval = (
                 self.client.table("bot_approval")
                 .select("*")
                 .eq("bot_id", bot_id)
+                .eq("approval_status", status)
                 .execute()
             )
             if bot_approval:
@@ -42,3 +43,19 @@ def query_by_bot_id(self, bot_id: str):
         except Exception as e:
             print("Error: ", e)
             return False, {"message": "approval query failed"}
+
+    def update_approval_status(self, id: str, status: str):
+        try:
+            bot_approval = (
+                self.client.table("bot_approval")
+                .update({"approval_status": status})
+                .eq("id", id)
+                .execute()
+            )
+            if bot_approval:
+                return True, {"message": "approval updated successfully"}
+            else:
+                return False, {"message": "approval update failed"}
+        except Exception as e:
+            print("Error: ", e)
+            return False, {"message": "approval update failed"}
diff --git a/server/core/models/bot.py b/server/core/models/bot.py
@@ -18,7 +18,8 @@ class BotModel(BaseModel):
     domain_whitelist: Optional[list[str]] = []
     temperature: Optional[float] = 0.2
     n: Optional[int] = 1
-    top_p: Optional[float]
+    top_p: Optional[float] = None
+
 
 
 class RepoBindBotConfigVO(BaseModel):

diff --git a/server/github_app/router.py b/server/github_app/router.py
@@ -28,6 +28,7 @@
     get_installation_repositories,
     get_jwt,
     get_private_key,
+    get_user_orgs,
 )
 
 from petercat_utils import get_env_variable
@@ -145,7 +146,7 @@ def get_user_repos_installed_app_(
         auth = Auth.Token(token=user.access_token)
         g = Github(auth=auth)
         github_user = g.get_user()
-        orgs = github_user.get_orgs()
+        orgs = get_user_orgs(github_user.login, auth.token)
         repository_config_dao = RepositoryConfigDAO()
         installations = repository_config_dao.query_by_owners(
             [org.id for org in orgs] + [github_user.id]

diff --git a/server/github_app/utils.py b/server/github_app/utils.py
@@ -58,3 +58,16 @@ def get_installation_repositories(access_token: str):
         },
     )
     return resp.json()
+
+
+def get_user_orgs(username, access_token: str):
+    url = f"https://api.github.com/users/{username}/orgs"
+    resp = requests.get(
+        url,
+        headers={
+            "X-GitHub-Api-Version": "2022-11-28",
+            "Accept": "application/vnd.github+json",
+            "Authorization": f"Bearer {access_token}",
+        },
+    )
+    return resp.json()
diff --git a/server/requirements.txt b/server/requirements.txt
@@ -7,7 +7,7 @@ langserve
 langchain_community>=0.2.11
 langchain>=0.2.12
 langchain_google_genai
-PyGithub==2.3.0
+PyGithub
 GitPython
 python-multipart
 httpx[socks]