Skip to content

Commit 77a2ca4

Browse files
committed
[ci] Fix test failures due to event refactor.
1 parent 74b5e08 commit 77a2ca4

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

python/flink_agents/e2e_tests/e2e_tests_integration/agent_skills_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -111,8 +111,8 @@ def system_prompt() -> Prompt:
111111
@staticmethod
112112
def process_input(event: Event, ctx: RunnerContext) -> None:
113113
input_event = InputEvent.from_event(event)
114-
if isinstance(input_event.input, Operation):
115-
input: Operation = input_event.input
114+
if isinstance(input_event.input, dict):
115+
input: Operation = Operation(**input_event.input)
116116
ctx.send_event(
117117
ChatRequestEvent(
118118
model="openai_setup",

python/flink_agents/e2e_tests/e2e_tests_integration/long_term_memory_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ def __init__(self, value: Any) -> None:
119119
def from_event(cls, event: Event) -> "MyEvent":
120120
"""Reconstruct a MyEvent from a generic Event."""
121121
assert "value" in event.attributes
122-
return MyEvent(value=event.attributes["value"])
122+
return MyEvent(value=Record.model_validate(event.attributes["value"]))
123123

124124
@property
125125
def value(self) -> Any:

0 commit comments

Comments
 (0)