from __future__ import annotations


class GraphIdResolver:
    _GRAPH_MAP = {
        "CODE_QA": "CodeQAGraph",
        "DOCUMENTATION_EXPLAIN": "DocsQAGraph",
        "OPENAPI_GENERATION": "DocsQAGraph",
        "GENERAL_QA": "DocsQAGraph",
        "DOCUMENTATION_DISCOVERY": "DocsQAGraph",
        "DOCUMENTATION_NAVIGATION": "DocsQAGraph",
        "OPENAPI_FROM_DOCUMENTATION": "DocsQAGraph",
        "GENERATE_DOCS_FROM_CODE": "GenerateDocsFromCodeGraph",
        "FALLBACK": "DocsQAGraph",
    }

    def resolve(self, intent: str) -> str:
        return self._GRAPH_MAP[intent]
