Skip to content

Commit 79c39fa

Browse files
fix(a2a): use A2aAgentExecutor instead of VeAgentExecutor to successfully handler a2a session info (#281)
Co-authored-by: jinhao.2019 <[email protected]>
1 parent 29ac679 commit 79c39fa

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

veadk/a2a/ve_a2a_server.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -19,20 +19,23 @@
1919

2020
from veadk import Agent
2121
from veadk.a2a.agent_card import get_agent_card
22-
from veadk.a2a.ve_agent_executor import VeAgentExecutor
22+
from veadk.runner import Runner
2323
from veadk.memory.short_term_memory import ShortTermMemory
2424

25+
from google.adk.a2a.executor.a2a_agent_executor import A2aAgentExecutor
26+
2527

2628
class VeA2AServer:
2729
def __init__(
2830
self, agent: Agent, url: str, app_name: str, short_term_memory: ShortTermMemory
2931
):
3032
self.agent_card = get_agent_card(agent, url)
3133

32-
self.agent_executor = VeAgentExecutor(
33-
app_name=app_name,
34-
agent=agent,
35-
short_term_memory=short_term_memory,
34+
self.agent_executor = A2aAgentExecutor(
35+
runner=Runner(
36+
agent=agent,
37+
app_name=app_name,
38+
),
3639
)
3740

3841
self.task_store = InMemoryTaskStore()

0 commit comments

Comments
 (0)