Create tools_def.py

tools_def.py

@@ -0,0 +1,200 @@
+TAVILY_TOOL = {
+    "type": "function",
+    "function": {
+        "name": "tavily_search",
+        "description": "Search the live web for real-time information, documentation, news, or current facts.",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "query": {
+                    "type": "string",
+                    "description": "The search query."
+                },
+                "max_results": {
+                    "type": "integer",
+                    "description": "Number of search results to retrieve (default: 8, max: 15).",
+                    "default": 8
+                }
+            },
+            "required": ["query"]
+        }
+    }
+}
+
+GMAIL_TOOL = {
+    "type": "function",
+    "function": {
+        "name": "gmail_api_request",
+        "description": "Execute direct REST API calls to Google Gmail API (Base: https://gmail.googleapis.com).",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "method": {
+                    "type": "string",
+                    "enum": ["GET", "POST", "PUT", "PATCH", "DELETE"],
+                    "description": "HTTP method."
+                },
+                "endpoint": {
+                    "type": "string",
+                    "description": "REST API endpoint path starting with leading slash (e.g., '/gmail/v1/users/me/messages', '/gmail/v1/users/me/drafts', '/gmail/v1/users/me/labels')."
+                },
+                "params": {
+                    "type": "object",
+                    "description": "Optional URL query parameters (e.g., {'q': 'is:unread', 'maxResults': 10})."
+                },
+                "body": {
+                    "type": "object",
+                    "description": "Optional JSON request body for POST/PUT/PATCH requests."
+                }
+            },
+            "required": ["method", "endpoint"]
+        }
+    }
+}
+
+CALENDAR_TOOL = {
+    "type": "function",
+    "function": {
+        "name": "calendar_api_request",
+        "description": "Execute direct REST API calls to Google Calendar API (Base: https://www.googleapis.com).",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "method": {
+                    "type": "string",
+                    "enum": ["GET", "POST", "PUT", "PATCH", "DELETE"],
+                    "description": "HTTP method."
+                },
+                "endpoint": {
+                    "type": "string",
+                    "description": "REST API endpoint path starting with leading slash (e.g., '/calendar/v3/calendars/primary/events', '/calendar/v3/users/me/calendarList')."
+                },
+                "params": {
+                    "type": "object",
+                    "description": "Optional URL query parameters (e.g., {'timeMin': '2026-01-01T00:00:00Z', 'singleEvents': True})."
+                },
+                "body": {
+                    "type": "object",
+                    "description": "Optional JSON request body for POST/PUT/PATCH requests."
+                }
+            },
+            "required": ["method", "endpoint"]
+        }
+    }
+}
+
+DRIVE_TOOL = {
+    "type": "function",
+    "function": {
+        "name": "drive_api_request",
+        "description": "Execute direct REST API calls to Google Drive API (Base: https://www.googleapis.com).",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "method": {
+                    "type": "string",
+                    "enum": ["GET", "POST", "PUT", "PATCH", "DELETE"],
+                    "description": "HTTP method."
+                },
+                "endpoint": {
+                    "type": "string",
+                    "description": "REST API endpoint path starting with leading slash (e.g., '/drive/v3/files', '/drive/v3/about')."
+                },
+                "params": {
+                    "type": "object",
+                    "description": "Optional URL query parameters (e.g., {'q': \"name contains 'document'\", 'fields': 'files(id, name, mimeType)'})."
+                },
+                "body": {
+                    "type": "object",
+                    "description": "Optional JSON request body for POST/PUT/PATCH requests."
+                }
+            },
+            "required": ["method", "endpoint"]
+        }
+    }
+}
+
+NOTION_TOOL = {
+    "type": "function",
+    "function": {
+        "name": "notion_api_request",
+        "description": "Execute direct REST API calls to Notion API v1 (Base: https://api.notion.com).",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "method": {
+                    "type": "string",
+                    "enum": ["GET", "POST", "PATCH", "DELETE"],
+                    "description": "HTTP method."
+                },
+                "endpoint": {
+                    "type": "string",
+                    "description": "REST API endpoint path starting with leading slash (e.g., '/v1/search', '/v1/pages', '/v1/databases/{database_id}/query', '/v1/blocks/{block_id}/children')."
+                },
+                "params": {
+                    "type": "object",
+                    "description": "Optional URL query parameters."
+                },
+                "body": {
+                    "type": "object",
+                    "description": "Optional JSON request body for POST/PATCH requests."
+                }
+            },
+            "required": ["method", "endpoint"]
+        }
+    }
+}
+
+GITHUB_TOOL = {
+    "type": "function",
+    "function": {
+        "name": "github_api_request",
+        "description": "Execute direct REST API calls to GitHub REST API (Base: https://api.github.com).",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "method": {
+                    "type": "string",
+                    "enum": ["GET", "POST", "PUT", "PATCH", "DELETE"],
+                    "description": "HTTP method."
+                },
+                "endpoint": {
+                    "type": "string",
+                    "description": "REST API endpoint path starting with leading slash (e.g., '/user', '/user/repos', '/repos/{owner}/{repo}/issues', '/search/repositories')."
+                },
+                "params": {
+                    "type": "object",
+                    "description": "Optional URL query parameters (e.g., {'per_page': 10, 'state': 'open'})."
+                },
+                "body": {
+                    "type": "object",
+                    "description": "Optional JSON request body for POST/PUT/PATCH requests."
+                }
+            },
+            "required": ["method", "endpoint"]
+        }
+    }
+}
+
+def get_filtered_tools(connectors: list = None, web_search: bool = False) -> list:
+    active_tools = []
+    if web_search:
+        active_tools.append(TAVILY_TOOL)
+    
+    if not connectors or not isinstance(connectors, list):
+        return active_tools
+
+    connector_keys = {str(c).lower().strip() for c in connectors}
+
+    if "gmail" in connector_keys or "google" in connector_keys:
+        active_tools.append(GMAIL_TOOL)
+    if "calendar" in connector_keys or "google" in connector_keys:
+        active_tools.append(CALENDAR_TOOL)
+    if "drive" in connector_keys or "google" in connector_keys:
+        active_tools.append(DRIVE_TOOL)
+    if "notion" in connector_keys:
+        active_tools.append(NOTION_TOOL)
+    if "github" in connector_keys:
+        active_tools.append(GITHUB_TOOL)
+
+    return active_tools