ベンダーアダプター

"""DataInsight Analytics APIのためのベンダーアダプタ。"""
from typing import Any, Dict, List, Optional


class DataInsightAdapter:
    """DataInsight APIの規約に合わせてリクエストを変換します。"""

    def transform_body(
        self,
        body: Dict[str, Any],
        operation_id: str,
        prompt_params: Optional[Dict[str, Any]] = None,
    ) -> Dict[str, Any]:
        """
        DataInsight API用にリクエストボディを変換します。

        Args:
            body: LLMからの元のリクエストボディ
            operation_id: OpenAPI操作ID
            prompt_params: セッションコンテキストパラメータ（オーケストレーターによって注入）

        Returns:
            DataInsight APIの期待に合った変換されたボディ
        """
        if not isinstance(body, dict):
            return body

        # セッションコンテキストを注入（prompt_paramsからのaccount_id、user_id）
        if prompt_params and isinstance(prompt_params, dict):
            if "account_id" in prompt_params and "account_id" not in body:
                body["account_id"] = prompt_params["account_id"]
            if "user_id" in prompt_params and "user_id" not in body:
                body["user_id"] = prompt_params["user_id"]

        # 操作特有の変換
        if operation_id == "queryMetrics":
            body = self._transform_query_metrics(body)
        elif operation_id == "getDimensionValues":
            body = self._transform_dimension_values(body)

        return body

    def _transform_query_metrics(self, body: Dict) -> Dict:
        """メトリクスクエリリクエストを変換します。"""
        # メトリクス名を正規化（省略形をサポート）
        metric_aliases = {
            "users": "di:unique_users",
            "sessions": "di:total_sessions",
            "pageviews": "di:page_views",
            "bounce_rate": "di:bounce_rate",
        }

        if isinstance(body.get("metrics"), list):
            body["metrics"] = [
                metric_aliases.get(m, m) for m in body["metrics"]
            ]

        # 時間範囲フォーマットを正規化
        if "timeRange" in body and isinstance(body["timeRange"], dict):
            tr = body["timeRange"]
            # startTime/endTimeを確保（start/endではなく）
            if "start" in tr:
                tr["startTime"] = tr.pop("start")
            if "end" in tr:
                tr["endTime"] = tr.pop("end")

        # sortを期待されるフォーマットに変換
        if isinstance(body.get("sort"), dict):
            field = body["sort"].get("field")
            order = body["sort"].get("order", "DESC").upper()
            body["sort"] = {"field": field, "direction": order}

        return body

    def _transform_dimension_values(self, body: Dict) -> Dict:
        """次元値リクエストを変換します。"""
        dimension_aliases = {
            "country": "di:geo_country",
            "device": "di:device_type",
            "source": "di:traffic_source",
        }

        if "dimension" in body:
            body["dimension"] = dimension_aliases.get(
                body["dimension"], body["dimension"]
            )

        return body

from typing import Optional


def get_vendor_adapter(name: str):
    """名前によってベンダーアダプタインスタンスを返す。利用できない場合はNoneを返す。"""
    if not name:
        return None
    try:
        if name.lower() == "datainsight":
            from .datainsight import DataInsightAdapter
            return DataInsightAdapter()
        # ここに他のベンダーを追加
        # elif name.lower() == "othervendor":
        #     from .othervendor import OtherVendorAdapter
        #     return OtherVendorAdapter()
    except Exception:
        return None
    return None

# DataInsight Analytics統合
# 使用法: SHANNON_CONFIG_PATH=config/overlays/shannon.datainsight.yaml

openapi_tools:
  datainsight_analytics:
    enabled: true
    spec_path: config/openapi_specs/datainsight_api.yaml
    auth_type: bearer
    auth_config:
      vendor: datainsight  # これがアダプタの読み込みをトリガー
      token: "${DATAINSIGHT_API_TOKEN}"
      extra_headers:
        X-Account-ID: "{{body.account_id}}"  # リクエストボディから動的
        X-User-ID: "${DATAINSIGHT_USER_ID}"   # 環境から静的
    category: analytics
    base_cost_per_use: 0.002
    rate_limit: 60
    timeout_seconds: 30
    operations:
      - queryMetrics
      - getDimensionValues

"""DataInsight Analyticsエージェントロールプリセット。"""

ANALYTICS_AGENT_PRESET = {
    "name": "datainsight_analytics",
    "system_prompt": """あなたはDataInsight Analytics APIにアクセスできる専門的なデータアナリティクスエージェントです。

あなたの使命：ウェブアナリティクスデータから実用的な洞察を提供すること。

## 利用可能なツール
- queryMetrics: ユーザー、セッション、ページビュー、バウンス率などのメトリクスを取得
- getDimensionValues: 次元値を取得（国、デバイス、トラフィックソース）

## 出力フォーマット
常に応答を次のように構成してください：
1. **dataResult**ブロック（JSON） - 可視化用
2. **Summary**セクション - 主要な発見
3. **Insights**セクション - 実用的な推奨事項

## ベストプラクティス
- 常にクエリに時間範囲を含める
- メトリクスエイリアスを使用する："users"、"sessions"、"pageviews"（自動変換）
- コンテキストに関連する次元をリクエストする
- 可能な場合は比較分析を提供する
- 異常やトレンドを強調する

覚えておいてください：あなたの目標は、ユーザーがデータを理解し、より良い意思決定を行うのを助けることです。""",

    "allowed_tools": [
        "queryMetrics",
        "getDimensionValues",
    ],

    "temperature": 0.7,
    "response_format": None,
}

# _load_presets()関数の最後に：
try:
    from .datainsight.analytics_agent import ANALYTICS_AGENT_PRESET
    _PRESETS["datainsight_analytics"] = ANALYTICS_AGENT_PRESET
except ImportError:
    pass  # ベンダーモジュールが利用できない場合の優雅なフォールバック

# DataInsight設定
SHANNON_CONFIG_PATH=config/overlays/shannon.datainsight.yaml
DATAINSIGHT_API_TOKEN=your_bearer_token_here
DATAINSIGHT_USER_ID=your_user_id_here

# ドメイン許可リスト（開発：*を使用、製品：特定のドメイン）
OPENAPI_ALLOWED_DOMAINS=api.datainsight.com

# サービスを再構築
docker compose -f deploy/compose/docker-compose.yml build --no-cache llm-service orchestrator
docker compose -f deploy/compose/docker-compose.yml up -d

# 健康状態を待つ
sleep 10

# gRPC経由でテスト
SESSION_ID="test-$(date +%s)"
grpcurl -plaintext -d '{
  "metadata": {
    "user_id": "test-user",
    "session_id": "'$SESSION_ID'"
  },
  "query": "過去30日間のユーザー成長トレンドを見せて",
  "context": {
    "role": "datainsight_analytics",
    "prompt_params": {
      "account_id": "acct_12345",
      "user_id": "user_67890"
    }
  }
}' localhost:50052 shannon.orchestrator.OrchestratorService/SubmitTask