lazabogdan · August 21, 2024 18:50 · Aug 21, 2024
diff --git a/github_gist_tools.py b/github_gist_tools.py
@@ -0,0 +1,67 @@
+
+import os
+from typing import Dict
+from langchain.tools import tool
+import requests
+
+# Replace with your GitHub token or use environment variables
+GITHUB_TOKEN = "your_github_token_here"
+
+@tool
+def create_gist(description: str, files: Dict[str, str], public: bool = False) -> Dict[str, str]:
+    """
+    Create a GitHub Gist with the given description, files, and visibility.
+
+    Args:
+        description (str): A description of the Gist.
+        files (Dict[str, str]): A dictionary where keys are filenames and values are file contents.
+        public (bool, optional): Whether the Gist should be public. Defaults to False.
+
+    Returns:
+        Dict[str, str]: A dictionary containing the URL of the created Gist or an error message.
+    """
+    url = "https://api.github.com/gists"
+    headers = {
+        "Authorization": f"token {GITHUB_TOKEN}",
+        "Accept": "application/vnd.github.v3+json"
+    }
+    data = {
+        "description": description,
+        "public": public,
+        "files": {filename: {"content": content} for filename, content in files.items()}
+    }
+
+    try:
+        response = requests.post(url, headers=headers, json=data)
+        response.raise_for_status()
+        return {"url": response.json()["html_url"]}
+    except requests.RequestException as e:
+        return {"error": str(e)}
+
+@tool
+def update_gist(gist_id: str, files: Dict[str, str]) -> Dict[str, str]:
+    """
+    Update an existing GitHub Gist with the given ID and files.
+
+    Args:
+        gist_id (str): The ID of the Gist to update.
+        files (Dict[str, str]): A dictionary where keys are filenames and values are file contents.
+
+    Returns:
+        Dict[str, str]: A dictionary containing a success message and the URL of the updated Gist, or an error message.
+    """
+    url = f"https://api.github.com/gists/{gist_id}"
+    headers = {
+        "Authorization": f"token {GITHUB_TOKEN}",
+        "Accept": "application/vnd.github.v3+json"
+    }
+    data = {
+        "files": {filename: {"content": content} for filename, content in files.items()}
+    }
+
+    try:
+        response = requests.patch(url, headers=headers, json=data)
+        response.raise_for_status()
+        return {"detail": "Gist updated successfully", "url": response.json()["html_url"]}
+    except requests.RequestException as e:
+        return {"error": str(e)}