Skip to content

Commit 94ff620

Browse files
fjtiradofjtirado
committed
[Fix #1087] A2A implementation
Signed-off-by: fjtirado <ftirados@ibm.com>
1 parent 3b5f7d3 commit 94ff620

22 files changed

Lines changed: 1081 additions & 6 deletions

‎impl/a2a/pom.xml‎

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
2+
<modelVersion>4.0.0</modelVersion>
3+
<parent>
4+
<groupId>io.serverlessworkflow</groupId>
5+
<artifactId>serverlessworkflow-impl</artifactId>
6+
<version>8.0.0-SNAPSHOT</version>
7+
</parent>
8+
<artifactId>serverlessworkflow-impl-a2a</artifactId>
9+
<name>Serverless Workflow :: Impl :: A2A</name>
10+
<dependencies>
11+
<dependency>
12+
<groupId>io.serverlessworkflow</groupId>
13+
<artifactId>serverlessworkflow-impl-core</artifactId>
14+
</dependency>
15+
<dependency>
16+
<groupId>org.a2aproject.sdk</groupId>
17+
<artifactId>a2a-java-sdk-client</artifactId>
18+
</dependency>
19+
<dependency>
20+
<groupId>org.junit.jupiter</groupId>
21+
<artifactId>junit-jupiter-engine</artifactId>
22+
<scope>test</scope>
23+
</dependency>
24+
<dependency>
25+
<groupId>org.junit.jupiter</groupId>
26+
<artifactId>junit-jupiter-params</artifactId>
27+
<scope>test</scope>
28+
</dependency>
29+
</dependencies>
30+
</project>

‎impl/a2a/src/main/java/io/serverlessworkflow/impl/executors/a2a/A2AExceptionHandler.java‎

Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,37 @@
1+
/*
2+
* Copyright 2020-Present The Serverless Workflow Specification Authors
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* http://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
package io.serverlessworkflow.impl.executors.a2a;
17+
18+
import io.serverlessworkflow.impl.WorkflowModel;
19+
import io.serverlessworkflow.impl.WorkflowPosition;
20+
import java.util.concurrent.CompletableFuture;
21+
import java.util.function.Consumer;
22+
23+
class A2AExceptionHandler implements Consumer<Throwable> {
24+
25+
private final CompletableFuture<WorkflowModel> future;
26+
private final WorkflowPosition position;
27+
28+
A2AExceptionHandler(CompletableFuture<WorkflowModel> future, WorkflowPosition position) {
29+
this.future = future;
30+
this.position = position;
31+
}
32+
33+
@Override
34+
public void accept(Throwable ex) {
35+
future.completeExceptionally(A2AUtils.workflowException(position, ex));
36+
}
37+
}

‎impl/a2a/src/main/java/io/serverlessworkflow/impl/executors/a2a/A2AExecutor.java‎

Lines changed: 77 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,77 @@
1+
/*
2+
* Copyright 2020-Present The Serverless Workflow Specification Authors
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* http://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
package io.serverlessworkflow.impl.executors.a2a;
17+
18+
import io.serverlessworkflow.impl.TaskContext;
19+
import io.serverlessworkflow.impl.WorkflowContext;
20+
import io.serverlessworkflow.impl.WorkflowModel;
21+
import io.serverlessworkflow.impl.WorkflowValueResolver;
22+
import io.serverlessworkflow.impl.executors.CallableTask;
23+
import java.net.URI;
24+
import java.util.Map;
25+
import java.util.concurrent.CompletableFuture;
26+
import org.a2aproject.sdk.client.Client;
27+
import org.a2aproject.sdk.client.config.ClientConfig;
28+
import org.a2aproject.sdk.client.http.A2ACardResolver;
29+
import org.a2aproject.sdk.client.transport.jsonrpc.JSONRPCTransport;
30+
import org.a2aproject.sdk.client.transport.jsonrpc.JSONRPCTransportConfig;
31+
import org.a2aproject.sdk.spec.A2AClientException;
32+
import org.a2aproject.sdk.spec.AgentCard;
33+
import org.slf4j.Logger;
34+
import org.slf4j.LoggerFactory;
35+
36+
class A2AExecutor implements CallableTask {
37+
38+
private final WorkflowValueResolver<URI> uriSupplier;
39+
private final A2ARequestDispatcher dispatcher;
40+
private final WorkflowValueResolver<Map<String, Object>> mapResolver;
41+
42+
private static final Logger logger = LoggerFactory.getLogger(A2AExecutor.class);
43+
44+
public A2AExecutor(
45+
WorkflowValueResolver<URI> uriSupplier,
46+
A2ARequestDispatcher dispatcher,
47+
WorkflowValueResolver<Map<String, Object>> mapResolver) {
48+
this.uriSupplier = uriSupplier;
49+
this.dispatcher = dispatcher;
50+
this.mapResolver = mapResolver;
51+
}
52+
53+
@Override
54+
public CompletableFuture<WorkflowModel> apply(
55+
WorkflowContext workflowContext, TaskContext taskContext, WorkflowModel input) {
56+
URI uri = uriSupplier.apply(workflowContext, taskContext, input);
57+
try {
58+
AgentCard agentCard =
59+
A2ACardResolver.builder()
60+
.baseUrl(uri.resolve("/").toString())
61+
.agentCardPath(uri.getPath())
62+
.build()
63+
.getAgentCard();
64+
logger.debug("Agent card is {}", agentCard);
65+
return dispatcher.apply(
66+
Client.builder(agentCard)
67+
.clientConfig(new ClientConfig.Builder().build())
68+
.withTransport(JSONRPCTransport.class, new JSONRPCTransportConfig())
69+
.build(),
70+
mapResolver.apply(workflowContext, taskContext, input),
71+
workflowContext,
72+
taskContext);
73+
} catch (A2AClientException ex) {
74+
throw A2AUtils.workflowException(taskContext.position(), ex);
75+
}
76+
}
77+
}

‎impl/a2a/src/main/java/io/serverlessworkflow/impl/executors/a2a/A2AExecutorBuilder.java‎

Lines changed: 83 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,83 @@
1+
/*
2+
* Copyright 2020-Present The Serverless Workflow Specification Authors
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* http://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
package io.serverlessworkflow.impl.executors.a2a;
17+
18+
import io.serverlessworkflow.api.types.A2AArguments;
19+
import io.serverlessworkflow.api.types.CallA2A;
20+
import io.serverlessworkflow.api.types.TaskBase;
21+
import io.serverlessworkflow.impl.WorkflowDefinition;
22+
import io.serverlessworkflow.impl.WorkflowMutablePosition;
23+
import io.serverlessworkflow.impl.WorkflowUtils;
24+
import io.serverlessworkflow.impl.WorkflowValueResolver;
25+
import io.serverlessworkflow.impl.executors.CallableTaskBuilder;
26+
import io.serverlessworkflow.impl.executors.CallableTaskFactory;
27+
import java.net.URI;
28+
29+
public class A2AExecutorBuilder implements CallableTaskBuilder<CallA2A> {
30+
31+
@Override
32+
public boolean accept(Class<? extends TaskBase> clazz) {
33+
return CallA2A.class.equals(clazz);
34+
}
35+
36+
@Override
37+
public CallableTaskFactory init(
38+
CallA2A task, WorkflowDefinition definition, WorkflowMutablePosition position) {
39+
A2AArguments args = task.getWith();
40+
41+
WorkflowValueResolver<URI> uriSupplier;
42+
if (args.getServer() != null) {
43+
uriSupplier = definition.resourceLoader().uriSupplier(args.getServer());
44+
} else if (args.getAgentCard() != null) {
45+
uriSupplier = definition.resourceLoader().uriSupplier(args.getAgentCard().getEndpoint());
46+
} else {
47+
throw new IllegalArgumentException("Neither server nor agent card is set for task:" + task);
48+
}
49+
50+
A2ARequestDispatcher dispatcher =
51+
switch (args.getMethod()) {
52+
case MESSAGE_SEND ->
53+
new MessageDispatcher(
54+
(workflowContext, taskContext, completableFuture) ->
55+
new MessageSendConsumer(workflowContext.definition(), completableFuture));
56+
case MESSAGE_STREAM ->
57+
new MessageDispatcher(
58+
(workflowContext, taskContext, completableFuture) ->
59+
new MessageStreamConsumer(
60+
workflowContext.definition(), completableFuture, taskContext.position()));
61+
case TASKS_LIST -> new ListTaskParamsDispatcher();
62+
case TASKS_GET -> new GetTaskParamsDispatcher();
63+
case TASKS_CANCEL -> new CancelTaskParamsDispatcher();
64+
// TODO handle missing cases
65+
case AGENT_GET_AUTHENTICATED_EXTENDED_CARD,
66+
TASKS_PUSH_NOTIFICATION_CONFIG_DELETE,
67+
TASKS_PUSH_NOTIFICATION_CONFIG_GET,
68+
TASKS_PUSH_NOTIFICATION_CONFIG_LIST,
69+
TASKS_PUSH_NOTIFICATION_CONFIG_SET,
70+
TASKS_RESUBSCRIBE ->
71+
throw new UnsupportedOperationException("Unimplemented case: " + args.getMethod());
72+
};
73+
74+
return () ->
75+
new A2AExecutor(
76+
uriSupplier,
77+
dispatcher,
78+
WorkflowUtils.buildMapResolver(
79+
definition.application(),
80+
args.getParameters().getString(),
81+
args.getParameters().getWithA2AParameters().getAdditionalProperties()));
82+
}
83+
}

‎impl/a2a/src/main/java/io/serverlessworkflow/impl/executors/a2a/A2ARequestDispatcher.java‎

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
/*
2+
* Copyright 2020-Present The Serverless Workflow Specification Authors
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* http://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
package io.serverlessworkflow.impl.executors.a2a;
17+
18+
import io.serverlessworkflow.impl.TaskContext;
19+
import io.serverlessworkflow.impl.WorkflowContext;
20+
import io.serverlessworkflow.impl.WorkflowModel;
21+
import java.util.Map;
22+
import java.util.concurrent.CompletableFuture;
23+
import org.a2aproject.sdk.client.Client;
24+
25+
@FunctionalInterface
26+
interface A2ARequestDispatcher {
27+
CompletableFuture<WorkflowModel> apply(
28+
Client client,
29+
Map<String, Object> parameters,
30+
WorkflowContext workflowContext,
31+
TaskContext taskContext);
32+
}

0 commit comments

Comments
 (0)