Spaces:
Configuration error
Configuration error
| """ | |
| Connector Manager — Manus-style connector ecosystem | |
| Manages OAuth tokens, connection state, API access | |
| """ | |
| import json | |
| import os | |
| import time | |
| from typing import Dict, List, Optional | |
| import structlog | |
| log = structlog.get_logger() | |
| CONNECTORS_CONFIG = [ | |
| { | |
| "id": "github", | |
| "name": "GitHub", | |
| "icon": "github", | |
| "color": "#24292e", | |
| "env_key": "GITHUB_TOKEN", | |
| "description": "Repos, Issues, PRs, Commits", | |
| "scopes": ["repo", "issues", "pull_requests"], | |
| "category": "code", | |
| }, | |
| { | |
| "id": "huggingface", | |
| "name": "HuggingFace", | |
| "icon": "huggingface", | |
| "color": "#ff9d00", | |
| "env_key": "HF_TOKEN", | |
| "description": "Spaces, Models, Datasets", | |
| "scopes": ["spaces", "models"], | |
| "category": "ai", | |
| }, | |
| { | |
| "id": "vercel", | |
| "name": "Vercel", | |
| "icon": "vercel", | |
| "color": "#000000", | |
| "env_key": "VERCEL_TOKEN", | |
| "description": "Deployments, Domains, Functions", | |
| "scopes": ["deployments", "projects"], | |
| "category": "deploy", | |
| }, | |
| { | |
| "id": "openai", | |
| "name": "OpenAI", | |
| "icon": "openai", | |
| "color": "#10a37f", | |
| "env_key": "OPENAI_API_KEY", | |
| "description": "GPT-4o, Embeddings, DALL-E", | |
| "scopes": ["chat", "embeddings"], | |
| "category": "ai", | |
| }, | |
| { | |
| "id": "groq", | |
| "name": "Groq", | |
| "icon": "groq", | |
| "color": "#f55036", | |
| "env_key": "GROQ_API_KEY", | |
| "description": "Llama 3.3 70B — Ultra Fast", | |
| "scopes": ["chat"], | |
| "category": "ai", | |
| }, | |
| { | |
| "id": "cerebras", | |
| "name": "Cerebras", | |
| "icon": "cerebras", | |
| "color": "#7c3aed", | |
| "env_key": "CEREBRAS_API_KEY", | |
| "description": "Llama 3.1 70B — Long Context", | |
| "scopes": ["chat"], | |
| "category": "ai", | |
| }, | |
| { | |
| "id": "openrouter", | |
| "name": "OpenRouter", | |
| "icon": "openrouter", | |
| "color": "#6366f1", | |
| "env_key": "OPENROUTER_API_KEY", | |
| "description": "Multi-model router, free tier", | |
| "scopes": ["chat"], | |
| "category": "ai", | |
| }, | |
| { | |
| "id": "anthropic", | |
| "name": "Anthropic", | |
| "icon": "anthropic", | |
| "color": "#d4a27f", | |
| "env_key": "ANTHROPIC_API_KEY", | |
| "description": "Claude 3.5 Sonnet", | |
| "scopes": ["chat"], | |
| "category": "ai", | |
| }, | |
| { | |
| "id": "n8n", | |
| "name": "n8n", | |
| "icon": "n8n", | |
| "color": "#ea4b71", | |
| "env_key": "N8N_URL", | |
| "description": "Workflow automation engine", | |
| "scopes": ["workflows", "executions"], | |
| "category": "workflow", | |
| }, | |
| { | |
| "id": "telegram", | |
| "name": "Telegram", | |
| "icon": "telegram", | |
| "color": "#0088cc", | |
| "env_key": "TELEGRAM_BOT_TOKEN", | |
| "description": "Bot API, messages, webhooks", | |
| "scopes": ["messages", "bots"], | |
| "category": "messaging", | |
| }, | |
| { | |
| "id": "discord", | |
| "name": "Discord", | |
| "icon": "discord", | |
| "color": "#5865f2", | |
| "env_key": "DISCORD_BOT_TOKEN", | |
| "description": "Bot, channels, webhooks", | |
| "scopes": ["messages", "bots"], | |
| "category": "messaging", | |
| }, | |
| { | |
| "id": "slack", | |
| "name": "Slack", | |
| "icon": "slack", | |
| "color": "#4a154b", | |
| "env_key": "SLACK_BOT_TOKEN", | |
| "description": "Messages, channels, workflows", | |
| "scopes": ["messages", "channels"], | |
| "category": "messaging", | |
| }, | |
| { | |
| "id": "cloudflare", | |
| "name": "Cloudflare", | |
| "icon": "cloudflare", | |
| "color": "#f38020", | |
| "env_key": "CLOUDFLARE_API_TOKEN", | |
| "description": "Workers, KV, Pages", | |
| "scopes": ["workers", "kv", "pages"], | |
| "category": "infra", | |
| }, | |
| ] | |
| class ConnectorManager: | |
| """Manages all platform connectors — connection state, tokens, status.""" | |
| def __init__(self): | |
| self._configs = {c["id"]: c for c in CONNECTORS_CONFIG} | |
| def get_all(self) -> List[Dict]: | |
| """Get all connectors with connection status.""" | |
| result = [] | |
| for cfg in CONNECTORS_CONFIG: | |
| token = os.environ.get(cfg["env_key"], "") | |
| result.append({ | |
| **cfg, | |
| "connected": bool(token), | |
| "token_preview": f"{token[:8]}..." if token else None, | |
| }) | |
| return result | |
| def get_connected(self) -> List[Dict]: | |
| """Get only connected connectors.""" | |
| return [c for c in self.get_all() if c["connected"]] | |
| def get_by_category(self, category: str) -> List[Dict]: | |
| """Get connectors by category.""" | |
| return [c for c in self.get_all() if c["category"] == category] | |
| def is_connected(self, connector_id: str) -> bool: | |
| cfg = self._configs.get(connector_id) | |
| if not cfg: | |
| return False | |
| return bool(os.environ.get(cfg["env_key"], "")) | |
| def get_token(self, connector_id: str) -> Optional[str]: | |
| cfg = self._configs.get(connector_id) | |
| if not cfg: | |
| return None | |
| return os.environ.get(cfg["env_key"]) or None | |
| def set_token(self, connector_id: str, token: str): | |
| """Set connector token at runtime (does not persist across restarts).""" | |
| cfg = self._configs.get(connector_id) | |
| if cfg: | |
| os.environ[cfg["env_key"]] = token | |
| log.info("Connector token set", connector=connector_id) | |
| def get_summary(self) -> Dict: | |
| all_c = self.get_all() | |
| connected = [c for c in all_c if c["connected"]] | |
| by_cat = {} | |
| for c in all_c: | |
| cat = c["category"] | |
| if cat not in by_cat: | |
| by_cat[cat] = {"total": 0, "connected": 0} | |
| by_cat[cat]["total"] += 1 | |
| if c["connected"]: | |
| by_cat[cat]["connected"] += 1 | |
| return { | |
| "total": len(all_c), | |
| "connected": len(connected), | |
| "by_category": by_cat, | |
| "ai_ready": self.is_connected("openai") or self.is_connected("groq") | |
| or self.is_connected("openrouter") or self.is_connected("anthropic") | |
| or self.is_connected("cerebras"), | |
| } | |