This is an automated email from the ASF dual-hosted git repository. xtsong pushed a commit to branch main in repository https://gitbox.apache.org/repos/asf/flink-agents.git
commit 535657494c2eb00439b732f393c6a5374ebfc277 Author: WenjinXie <[email protected]> AuthorDate: Fri Sep 26 11:16:32 2025 +0800 [hotfix] Move test purpose example to e2e tests package in python. --- .../common_tools.py => e2e_tests/__init__.py} | 36 ---------------------- .../{examples => e2e_tests}/agent_example.py | 0 .../{examples => e2e_tests}/chat_model_example.py | 0 .../{examples => e2e_tests}/common_tools.py | 0 .../integrate_datastream_with_agent_example.py | 2 +- .../integrate_table_with_agent_example.py | 2 +- .../integrate_table_with_react_agent_example.py | 2 +- .../{examples => e2e_tests}/my_agent.py | 0 .../{examples => e2e_tests}/react_agent_example.py | 2 +- .../resources/input_data.txt | 0 10 files changed, 4 insertions(+), 40 deletions(-) diff --git a/python/flink_agents/examples/common_tools.py b/python/flink_agents/e2e_tests/__init__.py similarity index 63% copy from python/flink_agents/examples/common_tools.py copy to python/flink_agents/e2e_tests/__init__.py index a165d9e..e154fad 100644 --- a/python/flink_agents/examples/common_tools.py +++ b/python/flink_agents/e2e_tests/__init__.py @@ -15,39 +15,3 @@ # See the License for the specific language governing permissions and # limitations under the License. ################################################################################# - - -def add(a: int, b: int) -> int: - """Calculate the sum of a and b. - - Parameters - ---------- - a : int - The first operand - b : int - The second operand - - Returns: - ------- - int: - The sum of a and b - """ - return a + b - - -def multiply(a: int, b: int) -> int: - """Useful function to multiply two numbers. - - Parameters - ---------- - a : int - The first operand - b : int - The second operand - - Returns: - ------- - int: - The product of a and b - """ - return a * b diff --git a/python/flink_agents/examples/agent_example.py b/python/flink_agents/e2e_tests/agent_example.py similarity index 100% rename from python/flink_agents/examples/agent_example.py rename to python/flink_agents/e2e_tests/agent_example.py diff --git a/python/flink_agents/examples/chat_model_example.py b/python/flink_agents/e2e_tests/chat_model_example.py similarity index 100% rename from python/flink_agents/examples/chat_model_example.py rename to python/flink_agents/e2e_tests/chat_model_example.py diff --git a/python/flink_agents/examples/common_tools.py b/python/flink_agents/e2e_tests/common_tools.py similarity index 100% rename from python/flink_agents/examples/common_tools.py rename to python/flink_agents/e2e_tests/common_tools.py diff --git a/python/flink_agents/examples/integrate_datastream_with_agent_example.py b/python/flink_agents/e2e_tests/integrate_datastream_with_agent_example.py similarity index 97% rename from python/flink_agents/examples/integrate_datastream_with_agent_example.py rename to python/flink_agents/e2e_tests/integrate_datastream_with_agent_example.py index 8e03d69..5f2e391 100644 --- a/python/flink_agents/examples/integrate_datastream_with_agent_example.py +++ b/python/flink_agents/e2e_tests/integrate_datastream_with_agent_example.py @@ -26,7 +26,7 @@ from pyflink.datastream import ( from pyflink.datastream.connectors.file_system import FileSource, StreamFormat from flink_agents.api.execution_environment import AgentsExecutionEnvironment -from flink_agents.examples.my_agent import DataStreamAgent, ItemData +from flink_agents.e2e_tests.my_agent import DataStreamAgent, ItemData class MyKeySelector(KeySelector): diff --git a/python/flink_agents/examples/integrate_table_with_agent_example.py b/python/flink_agents/e2e_tests/integrate_table_with_agent_example.py similarity index 98% rename from python/flink_agents/examples/integrate_table_with_agent_example.py rename to python/flink_agents/e2e_tests/integrate_table_with_agent_example.py index 2069d51..27df5c1 100644 --- a/python/flink_agents/examples/integrate_table_with_agent_example.py +++ b/python/flink_agents/e2e_tests/integrate_table_with_agent_example.py @@ -32,7 +32,7 @@ from pyflink.table import ( ) from flink_agents.api.execution_environment import AgentsExecutionEnvironment -from flink_agents.examples.my_agent import TableAgent +from flink_agents.e2e_tests.my_agent import TableAgent current_dir = Path(__file__).parent diff --git a/python/flink_agents/examples/integrate_table_with_react_agent_example.py b/python/flink_agents/e2e_tests/integrate_table_with_react_agent_example.py similarity index 98% rename from python/flink_agents/examples/integrate_table_with_react_agent_example.py rename to python/flink_agents/e2e_tests/integrate_table_with_react_agent_example.py index c767c2d..e335782 100644 --- a/python/flink_agents/examples/integrate_table_with_react_agent_example.py +++ b/python/flink_agents/e2e_tests/integrate_table_with_react_agent_example.py @@ -32,7 +32,7 @@ from flink_agents.api.execution_environment import AgentsExecutionEnvironment from flink_agents.api.prompts.prompt import Prompt from flink_agents.api.resource import ResourceDescriptor from flink_agents.api.tools.tool import Tool -from flink_agents.examples.common_tools import add, multiply +from flink_agents.e2e_tests.common_tools import add, multiply from flink_agents.integrations.chat_models.ollama_chat_model import ( OllamaChatModelConnection, OllamaChatModelSetup, diff --git a/python/flink_agents/examples/my_agent.py b/python/flink_agents/e2e_tests/my_agent.py similarity index 100% rename from python/flink_agents/examples/my_agent.py rename to python/flink_agents/e2e_tests/my_agent.py diff --git a/python/flink_agents/examples/react_agent_example.py b/python/flink_agents/e2e_tests/react_agent_example.py similarity index 97% rename from python/flink_agents/examples/react_agent_example.py rename to python/flink_agents/e2e_tests/react_agent_example.py index 830bc6a..6cfbfa4 100644 --- a/python/flink_agents/examples/react_agent_example.py +++ b/python/flink_agents/e2e_tests/react_agent_example.py @@ -25,7 +25,7 @@ from flink_agents.api.execution_environment import AgentsExecutionEnvironment from flink_agents.api.prompts.prompt import Prompt from flink_agents.api.resource import ResourceDescriptor from flink_agents.api.tools.tool import Tool -from flink_agents.examples.common_tools import add, multiply +from flink_agents.e2e_tests.common_tools import add, multiply from flink_agents.integrations.chat_models.ollama_chat_model import ( OllamaChatModelConnection, OllamaChatModelSetup, diff --git a/python/flink_agents/examples/resources/input_data.txt b/python/flink_agents/e2e_tests/resources/input_data.txt similarity index 100% rename from python/flink_agents/examples/resources/input_data.txt rename to python/flink_agents/e2e_tests/resources/input_data.txt
