Skip to content

Commit 04f4f17

Browse files
committed
fixup! [plan][python] Built-in actions support async execution.
cross language doesn't support async execution
1 parent b4d0198 commit 04f4f17

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

python/flink_agents/e2e_tests/e2e_tests_resource_cross_language/chat_model_cross_language_test.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
StreamingFileSink,
3333
)
3434

35+
from flink_agents.api.core_options import AgentExecutionOptions
3536
from flink_agents.api.execution_environment import AgentsExecutionEnvironment
3637
from flink_agents.e2e_tests.e2e_tests_resource_cross_language.chat_model_cross_language_agent import (
3738
ChatModelCrossLanguageAgent,
@@ -68,6 +69,7 @@ def test_java_chat_model_integration(tmp_path: Path) -> None: # noqa: D103
6869
)
6970

7071
agents_env = AgentsExecutionEnvironment.get_execution_environment(env=env)
72+
agents_env.get_config().set(AgentExecutionOptions.CHAT_ASYNC, False)
7173
output_datastream = (
7274
agents_env.from_datastream(
7375
input=deserialize_datastream, key_selector= lambda x: "orderKey"

python/flink_agents/e2e_tests/e2e_tests_resource_cross_language/vector_store_cross_language_test.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
StreamingFileSink,
3333
)
3434

35+
from flink_agents.api.core_options import AgentExecutionOptions
3536
from flink_agents.api.execution_environment import AgentsExecutionEnvironment
3637
from flink_agents.e2e_tests.e2e_tests_resource_cross_language.vector_store_cross_language_agent import (
3738
VectorStoreCrossLanguageAgent,
@@ -43,7 +44,7 @@
4344
OLLAMA_MODEL = os.environ.get("OLLAMA_EMBEDDING_MODEL", "nomic-embed-text:latest")
4445
os.environ["OLLAMA_EMBEDDING_MODEL"] = OLLAMA_MODEL
4546

46-
ES_HOST = os.environ.get("ES_HOST")
47+
ES_HOST = "localhost:9200"
4748

4849
client = pull_model(OLLAMA_MODEL)
4950

@@ -73,6 +74,7 @@ def test_java_embedding_model_integration(tmp_path: Path, embedding_type: str) -
7374
)
7475

7576
agents_env = AgentsExecutionEnvironment.get_execution_environment(env=env)
77+
agents_env.get_config().set(AgentExecutionOptions.RAG_ASYNC, False)
7678
output_datastream = (
7779
agents_env.from_datastream(
7880
input=deserialize_datastream, key_selector= lambda x: "orderKey"

0 commit comments

Comments
 (0)