chore(api/core): apply ruff reformatting (#7624)

api/core/tools/provider/builtin/gitlab/gitlab.py

@@ -9,13 +9,13 @@ from core.tools.provider.builtin_tool_provider import BuiltinToolProviderControl
 class GitlabProvider(BuiltinToolProviderController):
     def _validate_credentials(self, credentials: dict[str, Any]) -> None:
         try:
-            if 'access_tokens' not in credentials or not credentials.get('access_tokens'):
+            if "access_tokens" not in credentials or not credentials.get("access_tokens"):
                 raise ToolProviderCredentialValidationError("Gitlab Access Tokens is required.")
-            
-            if 'site_url' not in credentials or not credentials.get('site_url'):
-                site_url = 'https://gitlab.com'
+
+            if "site_url" not in credentials or not credentials.get("site_url"):
+                site_url = "https://gitlab.com"
             else:
-                site_url = credentials.get('site_url')
+                site_url = credentials.get("site_url")
 
             try:
                 headers = {
@@ -23,12 +23,10 @@ class GitlabProvider(BuiltinToolProviderController):
                     "Authorization": f"Bearer {credentials.get('access_tokens')}",
                 }
 
-                response = requests.get(
-                    url= f"{site_url}/api/v4/user",
-                    headers=headers)
+                response = requests.get(url=f"{site_url}/api/v4/user", headers=headers)
                 if response.status_code != 200:
-                    raise ToolProviderCredentialValidationError((response.json()).get('message'))
+                    raise ToolProviderCredentialValidationError((response.json()).get("message"))
             except Exception as e:
                 raise ToolProviderCredentialValidationError("Gitlab Access Tokens is invalid. {}".format(e))
         except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))

api/core/tools/provider/builtin/gitlab/tools/gitlab_commits.py

@@ -9,39 +9,47 @@ from core.tools.tool.builtin_tool import BuiltinTool
 
 
 class GitlabCommitsTool(BuiltinTool):
-    def _invoke(self, 
-                user_id: str,
-                tool_parameters: dict[str, Any]
-        ) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
-
-        project = tool_parameters.get('project', '')
-        employee = tool_parameters.get('employee', '')
-        start_time = tool_parameters.get('start_time', '')
-        end_time = tool_parameters.get('end_time', '')
-        change_type = tool_parameters.get('change_type', 'all')
+    def _invoke(
+        self, user_id: str, tool_parameters: dict[str, Any]
+    ) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
+        project = tool_parameters.get("project", "")
+        employee = tool_parameters.get("employee", "")
+        start_time = tool_parameters.get("start_time", "")
+        end_time = tool_parameters.get("end_time", "")
+        change_type = tool_parameters.get("change_type", "all")
 
         if not project:
-            return self.create_text_message('Project is required')
+            return self.create_text_message("Project is required")
 
         if not start_time:
             start_time = (datetime.utcnow() - timedelta(days=1)).isoformat()
         if not end_time:
             end_time = datetime.utcnow().isoformat()
 
-        access_token = self.runtime.credentials.get('access_tokens')
-        site_url = self.runtime.credentials.get('site_url')
+        access_token = self.runtime.credentials.get("access_tokens")
+        site_url = self.runtime.credentials.get("site_url")
 
-        if 'access_tokens' not in self.runtime.credentials or not self.runtime.credentials.get('access_tokens'):
+        if "access_tokens" not in self.runtime.credentials or not self.runtime.credentials.get("access_tokens"):
             return self.create_text_message("Gitlab API Access Tokens is required.")
-        if 'site_url' not in self.runtime.credentials or not self.runtime.credentials.get('site_url'):
-            site_url = 'https://gitlab.com'
-        
+        if "site_url" not in self.runtime.credentials or not self.runtime.credentials.get("site_url"):
+            site_url = "https://gitlab.com"
+
         # Get commit content
         result = self.fetch(user_id, site_url, access_token, project, employee, start_time, end_time, change_type)
 
         return [self.create_json_message(item) for item in result]
-    
-    def fetch(self,user_id: str, site_url: str, access_token: str, project: str, employee: str = None, start_time: str = '', end_time: str = '', change_type: str = '') -> list[dict[str, Any]]:
+
+    def fetch(
+        self,
+        user_id: str,
+        site_url: str,
+        access_token: str,
+        project: str,
+        employee: str = None,
+        start_time: str = "",
+        end_time: str = "",
+        change_type: str = "",
+    ) -> list[dict[str, Any]]:
         domain = site_url
         headers = {"PRIVATE-TOKEN": access_token}
         results = []
@@ -53,59 +61,66 @@ class GitlabCommitsTool(BuiltinTool):
             response.raise_for_status()
             projects = response.json()
 
-            filtered_projects = [p for p in projects if project == "*" or p['name'] == project]
+            filtered_projects = [p for p in projects if project == "*" or p["name"] == project]
 
             for project in filtered_projects:
-                project_id = project['id']
-                project_name = project['name']
+                project_id = project["id"]
+                project_name = project["name"]
                 print(f"Project: {project_name}")
 
                 # Get all of project commits
                 commits_url = f"{domain}/api/v4/projects/{project_id}/repository/commits"
-                params = {
-                    'since': start_time,
-                    'until': end_time
-                }
+                params = {"since": start_time, "until": end_time}
                 if employee:
-                    params['author'] = employee
+                    params["author"] = employee
 
                 commits_response = requests.get(commits_url, headers=headers, params=params)
                 commits_response.raise_for_status()
                 commits = commits_response.json()
 
                 for commit in commits:
-                    commit_sha = commit['id']
-                    author_name = commit['author_name']
+                    commit_sha = commit["id"]
+                    author_name = commit["author_name"]
 
                     diff_url = f"{domain}/api/v4/projects/{project_id}/repository/commits/{commit_sha}/diff"
                     diff_response = requests.get(diff_url, headers=headers)
                     diff_response.raise_for_status()
                     diffs = diff_response.json()
-                    
+
                     for diff in diffs:
                         # Calculate code lines of changed
-                        added_lines = diff['diff'].count('\n+')
-                        removed_lines = diff['diff'].count('\n-')
+                        added_lines = diff["diff"].count("\n+")
+                        removed_lines = diff["diff"].count("\n-")
                         total_changes = added_lines + removed_lines
 
                         if change_type == "new":
                             if added_lines > 1:
-                                final_code = ''.join([line[1:] for line in diff['diff'].split('\n') if line.startswith('+') and not line.startswith('+++')])
-                                results.append({
-                                    "commit_sha": commit_sha,
-                                    "author_name": author_name,
-                                    "diff": final_code
-                                })
+                                final_code = "".join(
+                                    [
+                                        line[1:]
+                                        for line in diff["diff"].split("\n")
+                                        if line.startswith("+") and not line.startswith("+++")
+                                    ]
+                                )
+                                results.append(
+                                    {"commit_sha": commit_sha, "author_name": author_name, "diff": final_code}
+                                )
                         else:
                             if total_changes > 1:
-                                final_code = ''.join([line[1:] for line in diff['diff'].split('\n') if (line.startswith('+') or line.startswith('-')) and not line.startswith('+++') and not line.startswith('---')])
+                                final_code = "".join(
+                                    [
+                                        line[1:]
+                                        for line in diff["diff"].split("\n")
+                                        if (line.startswith("+") or line.startswith("-"))
+                                        and not line.startswith("+++")
+                                        and not line.startswith("---")
+                                    ]
+                                )
                                 final_code_escaped = json.dumps(final_code)[1:-1]  # Escape the final code
-                                results.append({
-                                    "commit_sha": commit_sha,
-                                    "author_name": author_name,
-                                    "diff": final_code_escaped
-                                })
+                                results.append(
+                                    {"commit_sha": commit_sha, "author_name": author_name, "diff": final_code_escaped}
+                                )
         except requests.RequestException as e:
             print(f"Error fetching data from GitLab: {e}")
-        
-        return results
+
+        return results

api/core/tools/provider/builtin/gitlab/tools/gitlab_files.py

@@ -7,32 +7,29 @@ from core.tools.tool.builtin_tool import BuiltinTool
 
 
 class GitlabFilesTool(BuiltinTool):
-    def _invoke(self, 
-                user_id: str,
-                tool_parameters: dict[str, Any]
-        ) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
-        
-        project = tool_parameters.get('project', '')
-        branch = tool_parameters.get('branch', '')
-        path = tool_parameters.get('path', '')
-
+    def _invoke(
+        self, user_id: str, tool_parameters: dict[str, Any]
+    ) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
+        project = tool_parameters.get("project", "")
+        branch = tool_parameters.get("branch", "")
+        path = tool_parameters.get("path", "")
 
         if not project:
-            return self.create_text_message('Project is required')
+            return self.create_text_message("Project is required")
         if not branch:
-            return self.create_text_message('Branch is required')
+            return self.create_text_message("Branch is required")
 
         if not path:
-            return self.create_text_message('Path is required')
+            return self.create_text_message("Path is required")
 
-        access_token = self.runtime.credentials.get('access_tokens')
-        site_url = self.runtime.credentials.get('site_url')
+        access_token = self.runtime.credentials.get("access_tokens")
+        site_url = self.runtime.credentials.get("site_url")
 
-        if 'access_tokens' not in self.runtime.credentials or not self.runtime.credentials.get('access_tokens'):
+        if "access_tokens" not in self.runtime.credentials or not self.runtime.credentials.get("access_tokens"):
             return self.create_text_message("Gitlab API Access Tokens is required.")
-        if 'site_url' not in self.runtime.credentials or not self.runtime.credentials.get('site_url'):
-            site_url = 'https://gitlab.com'
-    
+        if "site_url" not in self.runtime.credentials or not self.runtime.credentials.get("site_url"):
+            site_url = "https://gitlab.com"
+
         # Get project ID from project name
         project_id = self.get_project_id(site_url, access_token, project)
         if not project_id:
@@ -42,9 +39,9 @@ class GitlabFilesTool(BuiltinTool):
         result = self.fetch(user_id, project_id, site_url, access_token, branch, path)
 
         return [self.create_json_message(item) for item in result]
-    
+
     def extract_project_name_and_path(self, path: str) -> tuple[str, str]:
-        parts = path.split('/', 1)
+        parts = path.split("/", 1)
         if len(parts) < 2:
             return None, None
         return parts[0], parts[1]
@@ -57,13 +54,15 @@ class GitlabFilesTool(BuiltinTool):
             response.raise_for_status()
             projects = response.json()
             for project in projects:
-                if project['name'] == project_name:
-                    return project['id']
+                if project["name"] == project_name:
+                    return project["id"]
         except requests.RequestException as e:
             print(f"Error fetching project ID from GitLab: {e}")
         return None
-    
-    def fetch(self,user_id: str, project_id: str, site_url: str, access_token: str, branch: str, path: str = None) -> list[dict[str, Any]]:
+
+    def fetch(
+        self, user_id: str, project_id: str, site_url: str, access_token: str, branch: str, path: str = None
+    ) -> list[dict[str, Any]]:
         domain = site_url
         headers = {"PRIVATE-TOKEN": access_token}
         results = []
@@ -76,20 +75,16 @@ class GitlabFilesTool(BuiltinTool):
             items = response.json()
 
             for item in items:
-                item_path = item['path']
-                if item['type'] == 'tree':  # It's a directory
+                item_path = item["path"]
+                if item["type"] == "tree":  # It's a directory
                     results.extend(self.fetch(project_id, site_url, access_token, branch, item_path))
                 else:  # It's a file
                     file_url = f"{domain}/api/v4/projects/{project_id}/repository/files/{item_path}/raw?ref={branch}"
                     file_response = requests.get(file_url, headers=headers)
                     file_response.raise_for_status()
                     file_content = file_response.text
-                    results.append({
-                        "path": item_path,
-                        "branch": branch,
-                        "content": file_content
-                    })
+                    results.append({"path": item_path, "branch": branch, "content": file_content})
         except requests.RequestException as e:
             print(f"Error fetching data from GitLab: {e}")
-        
-        return results
+
+        return results