example/rocketmq-multiagent-base-adk/WeatherAgent/src/main/java/agent/AgentCardProducer.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package agent;

import java.util.Collections;
import java.util.List;
import io.a2a.server.PublicAgentCard;
import io.a2a.spec.AgentCapabilities;
import io.a2a.spec.AgentCard;
import io.a2a.spec.AgentSkill;
import jakarta.enterprise.context.ApplicationScoped;
import jakarta.enterprise.inject.Produces;
import org.apache.commons.lang3.StringUtils;

import static org.apache.rocketmq.a2a.common.RocketMQA2AConstant.ROCKETMQ_PROTOCOL;

@ApplicationScoped
public class AgentCardProducer {
    private static final String ROCKETMQ_ENDPOINT = System.getProperty("rocketMQEndpoint", "");
    private static final String ROCKETMQ_INSTANCE_ID = System.getProperty("rocketMQInstanceID", "");
    private static final String BIZ_TOPIC = System.getProperty("bizTopic", "");

    @Produces
    @PublicAgentCard
    public AgentCard agentCard() {

        return new AgentCard.Builder()
                .name("天气查询助手Agent")
                .description("对未来一段时间内的天气情况进行查询")
                .url(buildRocketMQUrl())
                .version("1.0.0")
                .documentationUrl("http://example.com/docs")
                .capabilities(new AgentCapabilities.Builder()
                        .streaming(true)
                        .pushNotifications(true)
                        .stateTransitionHistory(true)
                        .build())
                .defaultInputModes(Collections.singletonList("text"))
                .defaultOutputModes(Collections.singletonList("text"))
                .skills(Collections.singletonList(new AgentSkill.Builder()
                                .id("天气查询助手Agent")
                                .name("天气查询助手Agent")
                                .description("对未来一段时间内的天气情况进行查询")
                                .tags(Collections.singletonList("天气查询"))
                                .examples(List.of("下周的天气怎么样"))
                                .build()))
                .preferredTransport(ROCKETMQ_PROTOCOL)
                .protocolVersion("0.3.0")
                .build();
    }

    private static String buildRocketMQUrl() {
        if (StringUtils.isEmpty(ROCKETMQ_ENDPOINT) || StringUtils.isEmpty(ROCKETMQ_INSTANCE_ID) || StringUtils.isEmpty(BIZ_TOPIC)) {
            throw new RuntimeException("buildRocketMQUrl param error, please check rocketmq config");
        }
        return "http://" + ROCKETMQ_ENDPOINT + "/" + ROCKETMQ_INSTANCE_ID + "/" + BIZ_TOPIC;
    }

}