Update app.py

app.py

@@ -66,7 +66,7 @@ OAUTH_CONFIGS = {
         "client_secret": "ec8548ef297ed59d8372e5bc549d8ed9adc996f7",
         "auth_url": "https://github.com/login/oauth/authorize",
         "token_url": "https://github.com/login/oauth/access_token",
-        "scope": "read:user repo"
+        "scope": "read:user repo workflow"
     },
     "figma": {
         "client_id": "Cg79RWBHaeyW2afGJVSQi8",
@@ -137,11 +137,12 @@ STRICT IDENTITY RULES:
    - To Delete/Remove a page: Search for the page_id, then call `notion_delete_page`. NEVER tell the user to delete manually.
    - To Create a new page: Use `notion_create_page`.
 
-4. GITHUB CONNECTOR WORKFLOW RULES:
-   - To view user repositories, use `github_list_repos`.
-   - To read repository content or file text, use `github_get_repo_content`.
-   - To check user details or profile, use `github_get_user_profile`.
-   - To view issues, use `github_list_issues`. To create issue, use `github_create_issue`.
+4. GITHUB CONNECTOR WORKFLOW RULES (FULL CAPABILITY):
+   - READ & SEARCH: Use `github_get_repo_content` to read files/folders, `github_search_code` to search code, `github_list_commits` for history.
+   - EDIT & WRITE: Use `github_create_or_update_file` to create or edit files, `github_delete_file` to remove files.
+   - BRANCHES & PRS: Use `github_create_branch`, `github_delete_branch`, `github_list_pull_requests`, `github_get_pull_request`, `github_create_pull_request`, `github_merge_pull_request`.
+   - ISSUES & COMMENTS: Use `github_list_issues`, `github_create_issue`, `github_update_issue` (to edit/close/reopen), `github_create_comment`.
+   - AUTOMATION: Use `github_list_workflow_runs` to check CI/CD build status.
 
 5. NATURAL CONVERSATION & TOOL RESPONSE STYLE:
    - Do NOT introduce your company, founder, or model name in general responses unless requested.
@@ -495,17 +496,173 @@ TOOLS = [
                 "properties": {
                     "owner": {"type": "string", "description": "Repository owner / username"},
                     "repo": {"type": "string", "description": "Repository name"},
-                    "path": {"type": "string", "description": "Path to file or folder, default empty for root"}
+                    "path": {"type": "string", "description": "Path to file or folder, default empty for root"},
+                    "branch": {"type": "string", "description": "Branch name, default main/master"}
                 },
                 "required": ["owner", "repo"]
             }
         }
     },
+    {
+        "type": "function",
+        "function": {
+            "name": "github_create_or_update_file",
+            "description": "Create a new file or edit/update an existing file in GitHub repository",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "owner": {"type": "string", "description": "Repository owner"},
+                    "repo": {"type": "string", "description": "Repository name"},
+                    "path": {"type": "string", "description": "File path in repository e.g. 'src/App.kt'"},
+                    "content": {"type": "string", "description": "Text content for the file"},
+                    "message": {"type": "string", "description": "Commit message"},
+                    "branch": {"type": "string", "description": "Target branch name"}
+                },
+                "required": ["owner", "repo", "path", "content", "message"]
+            }
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "github_delete_file",
+            "description": "Delete a file from GitHub repository",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "owner": {"type": "string", "description": "Repository owner"},
+                    "repo": {"type": "string", "description": "Repository name"},
+                    "path": {"type": "string", "description": "File path to delete"},
+                    "message": {"type": "string", "description": "Commit message"},
+                    "branch": {"type": "string", "description": "Branch name"}
+                },
+                "required": ["owner", "repo", "path", "message"]
+            }
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "github_create_branch",
+            "description": "Create a new Git branch in GitHub repository",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "owner": {"type": "string", "description": "Repository owner"},
+                    "repo": {"type": "string", "description": "Repository name"},
+                    "new_branch": {"type": "string", "description": "Name of new branch"},
+                    "base_branch": {"type": "string", "description": "Source branch to copy from, default 'main'"}
+                },
+                "required": ["owner", "repo", "new_branch"]
+            }
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "github_delete_branch",
+            "description": "Delete a branch from GitHub repository",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "owner": {"type": "string", "description": "Repository owner"},
+                    "repo": {"type": "string", "description": "Repository name"},
+                    "branch": {"type": "string", "description": "Branch name to delete"}
+                },
+                "required": ["owner", "repo", "branch"]
+            }
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "github_list_commits",
+            "description": "List commit history for a GitHub repository or specific file path",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "owner": {"type": "string", "description": "Repository owner"},
+                    "repo": {"type": "string", "description": "Repository name"},
+                    "path": {"type": "string", "description": "Optional file path filter"},
+                    "max_results": {"type": "integer", "description": "Number of commits, default 10"}
+                },
+                "required": ["owner", "repo"]
+            }
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "github_list_pull_requests",
+            "description": "List Pull Requests (PRs) in a GitHub repository",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "owner": {"type": "string", "description": "Repository owner"},
+                    "repo": {"type": "string", "description": "Repository name"},
+                    "state": {"type": "string", "description": "State e.g. 'open', 'closed', 'all'"}
+                },
+                "required": ["owner", "repo"]
+            }
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "github_get_pull_request",
+            "description": "Get detailed information and diff summary of a Pull Request",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "owner": {"type": "string", "description": "Repository owner"},
+                    "repo": {"type": "string", "description": "Repository name"},
+                    "pull_number": {"type": "integer", "description": "PR number"}
+                },
+                "required": ["owner", "repo", "pull_number"]
+            }
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "github_create_pull_request",
+            "description": "Create a new Pull Request in GitHub repository",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "owner": {"type": "string", "description": "Repository owner"},
+                    "repo": {"type": "string", "description": "Repository name"},
+                    "title": {"type": "string", "description": "PR title"},
+                    "head": {"type": "string", "description": "Branch containing new code"},
+                    "base": {"type": "string", "description": "Target branch to merge into e.g. 'main'"},
+                    "body": {"type": "string", "description": "PR description"}
+                },
+                "required": ["owner", "repo", "title", "head", "base"]
+            }
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "github_merge_pull_request",
+            "description": "Merge a Pull Request in GitHub repository",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "owner": {"type": "string", "description": "Repository owner"},
+                    "repo": {"type": "string", "description": "Repository name"},
+                    "pull_number": {"type": "integer", "description": "PR number"},
+                    "commit_title": {"type": "string", "description": "Optional merge commit message"}
+                },
+                "required": ["owner", "repo", "pull_number"]
+            }
+        }
+    },
     {
         "type": "function",
         "function": {
             "name": "github_list_issues",
-            "description": "List issues or pull requests in a GitHub repository",
+            "description": "List issues in a GitHub repository",
             "parameters": {
                 "type": "object",
                 "properties": {
@@ -533,6 +690,71 @@ TOOLS = [
                 "required": ["owner", "repo", "title"]
             }
         }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "github_update_issue",
+            "description": "Update or close/reopen an existing issue or Pull Request in GitHub",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "owner": {"type": "string", "description": "Repository owner"},
+                    "repo": {"type": "string", "description": "Repository name"},
+                    "issue_number": {"type": "integer", "description": "Issue number"},
+                    "title": {"type": "string", "description": "New title"},
+                    "body": {"type": "string", "description": "New body content"},
+                    "state": {"type": "string", "description": "State 'open' or 'closed'"}
+                },
+                "required": ["owner", "repo", "issue_number"]
+            }
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "github_create_comment",
+            "description": "Add a comment to an issue or Pull Request in GitHub",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "owner": {"type": "string", "description": "Repository owner"},
+                    "repo": {"type": "string", "description": "Repository name"},
+                    "issue_number": {"type": "integer", "description": "Issue or PR number"},
+                    "body": {"type": "string", "description": "Comment text"}
+                },
+                "required": ["owner", "repo", "issue_number", "body"]
+            }
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "github_search_code",
+            "description": "Search code across user repositories on GitHub",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "query": {"type": "string", "description": "Search text or function name e.g. 'function_name user:owner'"}
+                },
+                "required": ["query"]
+            }
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "github_list_workflow_runs",
+            "description": "List GitHub Actions CI/CD workflow runs and build status",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "owner": {"type": "string", "description": "Repository owner"},
+                    "repo": {"type": "string", "description": "Repository name"}
+                },
+                "required": ["owner", "repo"]
+            }
+        }
     }
 ]
 
@@ -665,12 +887,14 @@ async def github_list_repos(user_id: str, max_results: int = 10):
     except Exception as e:
         return f"Error listing GitHub repos: {str(e)}"
 
-async def github_get_repo_content(user_id: str, owner: str, repo: str, path: str = ""):
+async def github_get_repo_content(user_id: str, owner: str, repo: str, path: str = "", branch: str = None):
     headers = await github_get_headers(user_id)
     if not headers:
         return "GitHub is not connected."
     try:
         url = f"https://api.github.com/repos/{owner}/{repo}/contents/{path}"
+        if branch:
+            url += f"?ref={branch}"
         async with httpx.AsyncClient() as c:
             res = await c.get(url, headers=headers)
             if res.status_code != 200:
@@ -685,11 +909,185 @@ async def github_get_repo_content(user_id: str, owner: str, repo: str, path: str
                 if data.get("encoding") == "base64" and data.get("content"):
                     content = base64.b64decode(data.get("content")).decode("utf-8", errors="ignore")
                     return content
-                return f"File Name: {data.get('name')} | Size: {data.get('size')} bytes"
+                return f"File Name: {data.get('name')} | Size: {data.get('size')} bytes | SHA: {data.get('sha')}"
             return "Unable to parse content."
     except Exception as e:
         return f"Error reading GitHub content: {str(e)}"
 
+async def github_create_or_update_file(user_id: str, owner: str, repo: str, path: str, content: str, message: str, branch: str = None):
+    headers = await github_get_headers(user_id)
+    if not headers:
+        return "GitHub is not connected."
+    try:
+        url = f"https://api.github.com/repos/{owner}/{repo}/contents/{path}"
+        encoded_content = base64.b64encode(content.encode("utf-8")).decode("utf-8")
+        payload = {
+            "message": message,
+            "content": encoded_content
+        }
+        if branch:
+            payload["branch"] = branch
+        async with httpx.AsyncClient() as c:
+            get_res = await c.get(url + (f"?ref={branch}" if branch else ""), headers=headers)
+            if get_res.status_code == 200:
+                file_info = get_res.json()
+                if isinstance(file_info, dict) and "sha" in file_info:
+                    payload["sha"] = file_info["sha"]
+            res = await c.put(url, headers=headers, json=payload)
+            if res.status_code not in [200, 201]:
+                return f"Error updating file: {res.text}"
+            data = res.json()
+            return f"Successfully saved file '{path}' in commit: {data.get('commit', {}).get('sha')}"
+    except Exception as e:
+        return f"Error modifying file: {str(e)}"
+
+async def github_delete_file(user_id: str, owner: str, repo: str, path: str, message: str, branch: str = None):
+    headers = await github_get_headers(user_id)
+    if not headers:
+        return "GitHub is not connected."
+    try:
+        url = f"https://api.github.com/repos/{owner}/{repo}/contents/{path}"
+        async with httpx.AsyncClient() as c:
+            get_res = await c.get(url + (f"?ref={branch}" if branch else ""), headers=headers)
+            if get_res.status_code != 200:
+                return f"File '{path}' not found."
+            file_info = get_res.json()
+            sha = file_info.get("sha")
+            payload = {"message": message, "sha": sha}
+            if branch: payload["branch"] = branch
+            del_res = await c.request("DELETE", url, headers=headers, json=payload)
+            if del_res.status_code != 200:
+                return f"Error deleting file: {del_res.text}"
+            return f"Successfully deleted file '{path}'."
+    except Exception as e:
+        return f"Error deleting file: {str(e)}"
+
+async def github_create_branch(user_id: str, owner: str, repo: str, new_branch: str, base_branch: str = "main"):
+    headers = await github_get_headers(user_id)
+    if not headers:
+        return "GitHub is not connected."
+    try:
+        async with httpx.AsyncClient() as c:
+            ref_res = await c.get(f"https://api.github.com/repos/{owner}/{repo}/git/ref/heads/{base_branch}", headers=headers)
+            if ref_res.status_code != 200:
+                ref_res = await c.get(f"https://api.github.com/repos/{owner}/{repo}/git/ref/heads/master", headers=headers)
+                if ref_res.status_code != 200:
+                    return f"Error getting base branch SHA: {ref_res.text}"
+            sha = ref_res.json().get("object", {}).get("sha")
+            payload = {
+                "ref": f"refs/heads/{new_branch}",
+                "sha": sha
+            }
+            res = await c.post(f"https://api.github.com/repos/{owner}/{repo}/git/refs", headers=headers, json=payload)
+            if res.status_code != 201:
+                return f"Error creating branch: {res.text}"
+            return f"Successfully created branch '{new_branch}'."
+    except Exception as e:
+        return f"Error creating branch: {str(e)}"
+
+async def github_delete_branch(user_id: str, owner: str, repo: str, branch: str):
+    headers = await github_get_headers(user_id)
+    if not headers:
+        return "GitHub is not connected."
+    try:
+        url = f"https://api.github.com/repos/{owner}/{repo}/git/refs/heads/{branch}"
+        async with httpx.AsyncClient() as c:
+            res = await c.delete(url, headers=headers)
+            if res.status_code != 204:
+                return f"Error deleting branch: {res.text}"
+            return f"Successfully deleted branch '{branch}'."
+    except Exception as e:
+        return f"Error deleting branch: {str(e)}"
+
+async def github_list_commits(user_id: str, owner: str, repo: str, path: str = None, max_results: int = 10):
+    headers = await github_get_headers(user_id)
+    if not headers:
+        return "GitHub is not connected."
+    try:
+        url = f"https://api.github.com/repos/{owner}/{repo}/commits?per_page={max_results}"
+        if path: url += f"&path={path}"
+        async with httpx.AsyncClient() as c:
+            res = await c.get(url, headers=headers)
+            if res.status_code != 200:
+                return f"Error fetching commits: {res.text}"
+            commits = res.json()
+            items = []
+            for cm in commits:
+                sha = cm.get("sha", "")[:7]
+                msg = cm.get("commit", {}).get("message", "").split("\n")[0]
+                author = cm.get("commit", {}).get("author", {}).get("name", "")
+                date = cm.get("commit", {}).get("author", {}).get("date", "")
+                items.append(f"Commit [{sha}]: {msg} | Author: {author} | Date: {date}")
+            return "\n".join(items)
+    except Exception as e:
+        return f"Error listing commits: {str(e)}"
+
+async def github_list_pull_requests(user_id: str, owner: str, repo: str, state: str = "open"):
+    headers = await github_get_headers(user_id)
+    if not headers:
+        return "GitHub is not connected."
+    try:
+        url = f"https://api.github.com/repos/{owner}/{repo}/pulls?state={state}"
+        async with httpx.AsyncClient() as c:
+            res = await c.get(url, headers=headers)
+            if res.status_code != 200:
+                return f"Error fetching PRs: {res.text}"
+            prs = res.json()
+            if not prs: return "No Pull Requests found."
+            items = []
+            for p in prs:
+                items.append(f"PR #{p.get('number')}: {p.get('title')} | State: {p.get('state')} | User: {p.get('user', {}).get('login')} | Head: {p.get('head', {}).get('ref')} -> Base: {p.get('base', {}).get('ref')}")
+            return "\n".join(items)
+    except Exception as e:
+        return f"Error listing PRs: {str(e)}"
+
+async def github_get_pull_request(user_id: str, owner: str, repo: str, pull_number: int):
+    headers = await github_get_headers(user_id)
+    if not headers:
+        return "GitHub is not connected."
+    try:
+        url = f"https://api.github.com/repos/{owner}/{repo}/pulls/{pull_number}"
+        async with httpx.AsyncClient() as c:
+            res = await c.get(url, headers=headers)
+            if res.status_code != 200:
+                return f"Error fetching PR detail: {res.text}"
+            p = res.json()
+            return f"PR #{p.get('number')}: {p.get('title')}\nState: {p.get('state')}\nMergeable: {p.get('mergeable')}\nBody:\n{p.get('body')}"
+    except Exception as e:
+        return f"Error fetching PR detail: {str(e)}"
+
+async def github_create_pull_request(user_id: str, owner: str, repo: str, title: str, head: str, base: str = "main", body: str = ""):
+    headers = await github_get_headers(user_id)
+    if not headers:
+        return "GitHub is not connected."
+    try:
+        url = f"https://api.github.com/repos/{owner}/{repo}/pulls"
+        payload = {"title": title, "head": head, "base": base, "body": body}
+        async with httpx.AsyncClient() as c:
+            res = await c.post(url, headers=headers, json=payload)
+            if res.status_code not in [200, 201]:
+                return f"Error creating PR: {res.text}"
+            p = res.json()
+            return f"Successfully created PR #{p.get('number')}: {p.get('html_url')}"
+    except Exception as e:
+        return f"Error creating PR: {str(e)}"
+
+async def github_merge_pull_request(user_id: str, owner: str, repo: str, pull_number: int, commit_title: str = None):
+    headers = await github_get_headers(user_id)
+    if not headers:
+        return "GitHub is not connected."
+    try:
+        url = f"https://api.github.com/repos/{owner}/{repo}/pulls/{pull_number}/merge"
+        payload = {}
+        if commit_title: payload["commit_title"] = commit_title
+        async with httpx.AsyncClient() as c:
+            res = await c.put(url, headers=headers, json=payload)
+            if res.status_code != 200:
+                return f"Error merging PR: {res.text}"
+            return f"Successfully merged PR #{pull_number}."
+    except Exception as e:
+        return f"Error merging PR: {str(e)}"
+
 async def github_list_issues(user_id: str, owner: str, repo: str, state: str = "open"):
     headers = await github_get_headers(user_id)
     if not headers:
@@ -726,6 +1124,79 @@ async def github_create_issue(user_id: str, owner: str, repo: str, title: str, b
     except Exception as e:
         return f"Error creating GitHub issue: {str(e)}"
 
+async def github_update_issue(user_id: str, owner: str, repo: str, issue_number: int, title: str = None, body: str = None, state: str = None):
+    headers = await github_get_headers(user_id)
+    if not headers:
+        return "GitHub is not connected."
+    try:
+        url = f"https://api.github.com/repos/{owner}/{repo}/issues/{issue_number}"
+        payload = {}
+        if title: payload["title"] = title
+        if body: payload["body"] = body
+        if state: payload["state"] = state
+        async with httpx.AsyncClient() as c:
+            res = await c.patch(url, headers=headers, json=payload)
+            if res.status_code != 200:
+                return f"Error updating issue: {res.text}"
+            return f"Successfully updated Issue #{issue_number}."
+    except Exception as e:
+        return f"Error updating issue: {str(e)}"
+
+async def github_create_comment(user_id: str, owner: str, repo: str, issue_number: int, body: str):
+    headers = await github_get_headers(user_id)
+    if not headers:
+        return "GitHub is not connected."
+    try:
+        url = f"https://api.github.com/repos/{owner}/{repo}/issues/{issue_number}/comments"
+        payload = {"body": body}
+        async with httpx.AsyncClient() as c:
+            res = await c.post(url, headers=headers, json=payload)
+            if res.status_code not in [200, 201]:
+                return f"Error adding comment: {res.text}"
+            return f"Successfully added comment to Issue/PR #{issue_number}."
+    except Exception as e:
+        return f"Error adding comment: {str(e)}"
+
+async def github_search_code(user_id: str, query: str):
+    headers = await github_get_headers(user_id)
+    if not headers:
+        return "GitHub is not connected."
+    try:
+        encoded_q = urllib.parse.quote(query)
+        url = f"https://api.github.com/search/code?q={encoded_q}"
+        async with httpx.AsyncClient() as c:
+            res = await c.get(url, headers=headers)
+            if res.status_code != 200:
+                return f"Error searching code: {res.text}"
+            data = res.json()
+            items = data.get("items", [])
+            if not items: return "No code matches found."
+            results = []
+            for item in items[:10]:
+                results.append(f"Repo: {item.get('repository', {}).get('full_name')} | File: {item.get('path')} | URL: {item.get('html_url')}")
+            return "\n".join(results)
+    except Exception as e:
+        return f"Error searching code: {str(e)}"
+
+async def github_list_workflow_runs(user_id: str, owner: str, repo: str):
+    headers = await github_get_headers(user_id)
+    if not headers:
+        return "GitHub is not connected."
+    try:
+        url = f"https://api.github.com/repos/{owner}/{repo}/actions/runs?per_page=5"
+        async with httpx.AsyncClient() as c:
+            res = await c.get(url, headers=headers)
+            if res.status_code != 200:
+                return f"Error fetching workflow runs: {res.text}"
+            runs = res.json().get("workflow_runs", [])
+            if not runs: return "No CI/CD workflow runs found."
+            items = []
+            for r in runs:
+                items.append(f"Run #{r.get('run_number')}: {r.get('name')} | Status: {r.get('status')} | Conclusion: {r.get('conclusion')} | Branch: {r.get('head_branch')}")
+            return "\n".join(items)
+    except Exception as e:
+        return f"Error fetching workflow runs: {str(e)}"
+
 async def notion_get_headers(user_id: str):
     token_data = await get_token_from_supabase(user_id, "notion")
     if not token_data:
@@ -1231,11 +1702,37 @@ async def execute_tool_call(user_id: str, func_name: str, args: dict) -> str:
     elif func_name == "github_list_repos":
         return await github_list_repos(user_id, args.get("max_results", 10))
     elif func_name == "github_get_repo_content":
-        return await github_get_repo_content(user_id, args.get("owner", ""), args.get("repo", ""), args.get("path", ""))
+        return await github_get_repo_content(user_id, args.get("owner", ""), args.get("repo", ""), args.get("path", ""), args.get("branch"))
+    elif func_name == "github_create_or_update_file":
+        return await github_create_or_update_file(user_id, args.get("owner", ""), args.get("repo", ""), args.get("path", ""), args.get("content", ""), args.get("message", ""), args.get("branch"))
+    elif func_name == "github_delete_file":
+        return await github_delete_file(user_id, args.get("owner", ""), args.get("repo", ""), args.get("path", ""), args.get("message", ""), args.get("branch"))
+    elif func_name == "github_create_branch":
+        return await github_create_branch(user_id, args.get("owner", ""), args.get("repo", ""), args.get("new_branch", ""), args.get("base_branch", "main"))
+    elif func_name == "github_delete_branch":
+        return await github_delete_branch(user_id, args.get("owner", ""), args.get("repo", ""), args.get("branch", ""))
+    elif func_name == "github_list_commits":
+        return await github_list_commits(user_id, args.get("owner", ""), args.get("repo", ""), args.get("path"), args.get("max_results", 10))
+    elif func_name == "github_list_pull_requests":
+        return await github_list_pull_requests(user_id, args.get("owner", ""), args.get("repo", ""), args.get("state", "open"))
+    elif func_name == "github_get_pull_request":
+        return await github_get_pull_request(user_id, args.get("owner", ""), args.get("repo", ""), args.get("pull_number", 1))
+    elif func_name == "github_create_pull_request":
+        return await github_create_pull_request(user_id, args.get("owner", ""), args.get("repo", ""), args.get("title", ""), args.get("head", ""), args.get("base", "main"), args.get("body", ""))
+    elif func_name == "github_merge_pull_request":
+        return await github_merge_pull_request(user_id, args.get("owner", ""), args.get("repo", ""), args.get("pull_number", 1), args.get("commit_title"))
     elif func_name == "github_list_issues":
         return await github_list_issues(user_id, args.get("owner", ""), args.get("repo", ""), args.get("state", "open"))
     elif func_name == "github_create_issue":
         return await github_create_issue(user_id, args.get("owner", ""), args.get("repo", ""), args.get("title", ""), args.get("body", ""))
+    elif func_name == "github_update_issue":
+        return await github_update_issue(user_id, args.get("owner", ""), args.get("repo", ""), args.get("issue_number", 1), args.get("title"), args.get("body"), args.get("state"))
+    elif func_name == "github_create_comment":
+        return await github_create_comment(user_id, args.get("owner", ""), args.get("repo", ""), args.get("issue_number", 1), args.get("body", ""))
+    elif func_name == "github_search_code":
+        return await github_search_code(user_id, args.get("query", ""))
+    elif func_name == "github_list_workflow_runs":
+        return await github_list_workflow_runs(user_id, args.get("owner", ""), args.get("repo", ""))
     return "Unknown tool function."
 
 def count_tokens(messages):