scoped_api_client.py 1.3 KB

1234567891011121314151617181920212223242526272829303132333435
  1. from services.api_client import JsonTransport
  2. class ScopedApiClient:
  3. def __init__(self, base_url, transport=None, timeout=10):
  4. self.base_url = (base_url or '').rstrip('/')
  5. self.transport = transport or JsonTransport()
  6. self.timeout = int(timeout)
  7. def call_tool(self, token, tool_code, route_path, payload, request_id, client_ip=''):
  8. token = str(token or '').strip()
  9. if not token:
  10. raise RuntimeError('mcp token missing')
  11. url = self.base_url + '/' + route_path.lstrip('/')
  12. headers = {
  13. 'Authorization': 'Bearer {0}'.format(token),
  14. 'X-MCP-Tool-Code': tool_code,
  15. 'X-Request-Id': request_id,
  16. }
  17. client_ip = str(client_ip or '').strip()
  18. if client_ip:
  19. headers['X-MCP-Client-IP'] = client_ip
  20. return self.transport.post_json(url, payload, headers, self.timeout)
  21. def list_enabled_tools(self, token, request_id=''):
  22. token = str(token or '').strip()
  23. if not token:
  24. raise RuntimeError('mcp token missing')
  25. url = self.base_url + '/mcp/tools/listEnabledTools'
  26. headers = {
  27. 'Authorization': 'Bearer {0}'.format(token),
  28. 'X-Request-Id': str(request_id or '').strip(),
  29. }
  30. return self.transport.post_json(url, {}, headers, self.timeout)