DocsGPT/application/tools/implementations/api_tool.py

import json

import requests
from application.tools.base import Tool


class APITool(Tool):
    """
    API Tool
    A flexible tool for performing various API actions (e.g., sending messages, retrieving data) via custom user-specified APIs
    """

    def __init__(self, config):
        self.config = config
        self.url = config.get("url", "")
        self.method = config.get("method", "GET")
        self.headers = config.get("headers", {"Content-Type": "application/json"})
        self.query_params = config.get("query_params", {})

    def execute_action(self, action_name, **kwargs):
        return self._make_api_call(
            self.url, self.method, self.headers, self.query_params, kwargs
        )

    def _make_api_call(self, url, method, headers, query_params, body):
        if query_params:
            url = f"{url}?{requests.compat.urlencode(query_params)}"
        if isinstance(body, dict):
            body = json.dumps(body)
        try:
            print(f"Making API call: {method} {url} with body: {body}")
            response = requests.request(method, url, headers=headers, data=body)
            response.raise_for_status()
            try:
                data = response.json()
            except ValueError:
                data = None

            return {
                "status_code": response.status_code,
                "data": data,
                "message": "API call successful.",
            }
        except requests.exceptions.RequestException as e:
            return {
                "status_code": response.status_code if response else None,
                "message": f"API call failed: {str(e)}",
            }

    def get_actions_metadata(self):
        return []

    def get_config_requirements(self):
        return {}