ブログ

// Main.controller.js の抜粋例
onInit: function () {
  var oModel = this.getView().getModel();
  if (!oModel.getProperty("/sessionId")) {
    // 会話のつながり用（初回のみ生成）
    oModel.setProperty("/sessionId", "sess-" + Date.now());
  }
  if (oModel.getProperty("/agentResultText") === undefined) {
    oModel.setProperty("/agentResultText", "");
  }
  // this._url は manifest の dataSources 等から組み立てた CAP のベース URL
},

onSendMessage: function () {
  var oInput = this.byId("messageInput");
  var prompt = (oInput && oInput.getValue()) ? oInput.getValue().trim() : "";
  if (!prompt) {
    sap.m.MessageToast.show("プロンプトを入力してください");
    return;
  }

  var oModel = this.getView().getModel();
  var sessionId = oModel.getProperty("/sessionId");
  var base = this._url; // 例: https://<your-cap-host>
  var url = base + "/odata/v2/MCPService/askAi"
    + "?prompt='" + encodeURIComponent(prompt) + "'"
    + "&sessionId='" + encodeURIComponent(sessionId) + "'";

  sap.ui.core.BusyIndicator.show(0);
  var that = this;
  this._baseController.execAPI("POST", url, null)
    .then(function (data) {
      var text = (data && data.d && data.d.askAi != null)
        ? String(data.d.askAi)
        : "";
      oModel.setProperty("/agentResultText", text);
    })
    .catch(function (err) {
      var msg = (err && err.message) ? err.message : String(err);
      sap.m.MessageToast.show("エラー: " + msg);
    })
    .finally(function () {
      sap.ui.core.BusyIndicator.hide();
    });
}

// ハンドラのサンプル
@Component
@ServiceName("MCPService")
public class McpServiceHandler implements EventHandler {

  @Autowired
  private VertexAgentClient vertexAgentClient;

  @On(event = "askAi")
  public void handleAskAi(AskAiContext context) {
    try {
      String answer = vertexAgentClient.query(
          context.getPrompt(),
          context.getSessionId()
      );
      context.setResult(answer);
    } catch (Exception e) {
      context.setResult("Error: " + e.getMessage());
    }
  }
}

// Vertex AI Agent Engine への POST処理
public String query(String prompt, String sessionId) throws Exception {
  String region = System.getenv("GCP_REGION");
  String project = System.getenv("GCP_PROJECT_ID");
  String agentId = System.getenv("VERTEX_AGENT_ENGINE_ID");

  String token = obtainBearerToken(); // サービスアカウント等（鍵はリポジトリに含めない）

  String url = String.format(
      "https://%s-aiplatform.googleapis.com/v1beta1/projects/%s/locations/%s/reasoningEngines/%s:query",
      region, project, region, agentId);

  Map<String, Object> inner = new HashMap<>();
  inner.put("input", prompt);
  inner.put("session_id", sessionId != null ? sessionId : "");

  Map<String, Object> body = new HashMap<>();
  body.put("input", inner);

  HttpHeaders headers = new HttpHeaders();
  headers.setContentType(MediaType.APPLICATION_JSON);
  headers.setBearerAuth(token);

  RestTemplate rt = new RestTemplate();
  ResponseEntity<Map> res = rt.postForEntity(
      url, new HttpEntity<>(body, headers), Map.class);

  if (res.getBody() == null) return "";
  Object out = res.getBody().get("output");
  return out != null ? out.toString() : "";
}

// obtainBearerToken() は google-auth-library 等で
// サービスアカウント JSON を読み、cloud-platform スコープでトークン取得、など

from google.adk.agents.llm_agent import LlmAgent as Agent
from google.adk.runners import Runner
from google.adk.sessions import InMemorySessionService
from google.genai import types

class DemoAgentApp:
    def __init__(self, toolset, app_name: str = "demo_app"):
        self.app_name = app_name
        self.session_service = InMemorySessionService()
        self.agent = Agent(
            model="gemini-2.0-flash",
            name="demo_bigquery_agent",
            instruction=(
                "与えられたツールだけを使い、事実に基づいて簡潔に日本語で答えてください。"
            ),
            tools=[toolset],
        )

    def query(self, user_text: str, session_id: str, user_id: str = "demo_user") -> str:
        runner = Runner(
            agent=self.agent,
            app_name=self.app_name,
            session_service=self.session_service,
        )
        msg = types.Content(
            role="user",
            parts=[types.Part(text=user_text)],
        )
        final_text = ""
        # 以下は ADK のバージョンに合わせて調整
        for event in runner.run(
            user_id=user_id,
            session_id=session_id,
            new_message=msg,
        ):
            if getattr(event, "content", None) and event.content.parts:
                for p in event.content.parts:
                    if getattr(p, "text", None):
                        final_text = p.text
        return final_text or "(応答なし)"

from google.adk.tools.api_registry import ApiRegistry

def build_bigquery_toolset(project_id: str):
    api_registry = ApiRegistry(project=project_id, location="global")
    mcp_server_name = (
        f"projects/{project_id}/locations/global/mcpServers/"
        f"google-bigquery.googleapis.com-mcp"
    )
    toolset = api_registry.get_toolset(mcp_server_name=mcp_server_name)
    if toolset is None:
        raise RuntimeError("ツールセット取得に失敗しました。Registry・IAM を確認してください。")
    return toolset

# 利用例（エージェント生成時）
# toolset = build_bigquery_toolset("<YOUR_PROJECT_ID>")
# app = DemoAgentApp(toolset=toolset)