diff --git a/e2e-test/flink-agents-end-to-end-tests-integration/src/test/java/org/apache/flink/agents/integration/test/ChatModelIntegrationAgent.java b/e2e-test/flink-agents-end-to-end-tests-integration/src/test/java/org/apache/flink/agents/integration/test/ChatModelIntegrationAgent.java index 6ce40a53..a3f44b8f 100644 --- a/e2e-test/flink-agents-end-to-end-tests-integration/src/test/java/org/apache/flink/agents/integration/test/ChatModelIntegrationAgent.java +++ b/e2e-test/flink-agents-end-to-end-tests-integration/src/test/java/org/apache/flink/agents/integration/test/ChatModelIntegrationAgent.java @@ -67,7 +67,7 @@ * the model is allowed to call. */ public class ChatModelIntegrationAgent extends Agent { - public static final String OLLAMA_MODEL = "qwen3:0.6b"; + public static final String OLLAMA_MODEL = "qwen3:8b"; @ChatModelConnection public static ResourceDescriptor chatModelConnection() { diff --git a/e2e-test/flink-agents-end-to-end-tests-integration/src/test/java/org/apache/flink/agents/integration/test/ReActAgentTest.java b/e2e-test/flink-agents-end-to-end-tests-integration/src/test/java/org/apache/flink/agents/integration/test/ReActAgentTest.java index 10905fc5..79c18b58 100644 --- a/e2e-test/flink-agents-end-to-end-tests-integration/src/test/java/org/apache/flink/agents/integration/test/ReActAgentTest.java +++ b/e2e-test/flink-agents-end-to-end-tests-integration/src/test/java/org/apache/flink/agents/integration/test/ReActAgentTest.java @@ -53,7 +53,7 @@ import static org.apache.flink.agents.integration.test.OllamaPreparationUtils.pullModel; public class ReActAgentTest { - public static final String OLLAMA_MODEL = "qwen3:1.7b"; + public static final String OLLAMA_MODEL = "qwen3:8b"; @org.apache.flink.agents.api.annotation.Tool( description = "Useful function to add two numbers.") diff --git a/python/flink_agents/e2e_tests/e2e_tests_integration/chat_model_integration_agent.py b/python/flink_agents/e2e_tests/e2e_tests_integration/chat_model_integration_agent.py index 0eaa9fc7..33553078 100644 --- a/python/flink_agents/e2e_tests/e2e_tests_integration/chat_model_integration_agent.py +++ b/python/flink_agents/e2e_tests/e2e_tests_integration/chat_model_integration_agent.py @@ -84,7 +84,7 @@ def math_chat_model() -> ResourceDescriptor: return ResourceDescriptor( clazz=OllamaChatModelSetup, connection="ollama_connection", - model=os.environ.get("OLLAMA_CHAT_MODEL", "qwen3:1.7b"), + model=os.environ.get("OLLAMA_CHAT_MODEL", "qwen3:8b"), tools=["add"], extract_reasoning=True, ) @@ -114,7 +114,7 @@ def creative_chat_model() -> ResourceDescriptor: return ResourceDescriptor( clazz=TongyiChatModelSetup, connection="ollama_connection", - model=os.environ.get("OLLAMA_CHAT_MODEL", "qwen3:1.7b"), + model=os.environ.get("OLLAMA_CHAT_MODEL", "qwen3:8b"), extract_reasoning=True, ) elif model_provider == "OpenAI": diff --git a/python/flink_agents/e2e_tests/e2e_tests_integration/chat_model_integration_test.py b/python/flink_agents/e2e_tests/e2e_tests_integration/chat_model_integration_test.py index 93dd48fd..21ee92ac 100644 --- a/python/flink_agents/e2e_tests/e2e_tests_integration/chat_model_integration_test.py +++ b/python/flink_agents/e2e_tests/e2e_tests_integration/chat_model_integration_test.py @@ -30,7 +30,7 @@ TONGYI_MODEL = os.environ.get("TONGYI_CHAT_MODEL", "qwen-plus") os.environ["TONGYI_CHAT_MODEL"] = TONGYI_MODEL -OLLAMA_MODEL = os.environ.get("OLLAMA_CHAT_MODEL", "qwen3:1.7b") +OLLAMA_MODEL = os.environ.get("OLLAMA_CHAT_MODEL", "qwen3:8b") os.environ["OLLAMA_CHAT_MODEL"] = OLLAMA_MODEL OPENAI_MODEL = os.environ.get("OPENAI_CHAT_MODEL", "gpt-3.5-turbo") os.environ["OPENAI_CHAT_MODEL"] = OPENAI_MODEL diff --git a/python/flink_agents/e2e_tests/e2e_tests_integration/e2e_tests_mcp/mcp_test.py b/python/flink_agents/e2e_tests/e2e_tests_integration/e2e_tests_mcp/mcp_test.py index ea54c404..2a108395 100644 --- a/python/flink_agents/e2e_tests/e2e_tests_integration/e2e_tests_mcp/mcp_test.py +++ b/python/flink_agents/e2e_tests/e2e_tests_integration/e2e_tests_mcp/mcp_test.py @@ -55,7 +55,7 @@ OllamaChatModelSetup, ) -OLLAMA_MODEL = os.environ.get("MCP_OLLAMA_CHAT_MODEL", "qwen3:1.7b") +OLLAMA_MODEL = os.environ.get("MCP_OLLAMA_CHAT_MODEL", "qwen3:8b") MCP_SERVER_ENDPOINT = "http://127.0.0.1:8000/mcp" diff --git a/python/flink_agents/e2e_tests/e2e_tests_integration/react_agent_test.py b/python/flink_agents/e2e_tests/e2e_tests_integration/react_agent_test.py index 630a7f34..b5932103 100644 --- a/python/flink_agents/e2e_tests/e2e_tests_integration/react_agent_test.py +++ b/python/flink_agents/e2e_tests/e2e_tests_integration/react_agent_test.py @@ -48,7 +48,7 @@ os.environ["PYTHONPATH"] = sysconfig.get_paths()["purelib"] -OLLAMA_MODEL = os.environ.get("REACT_OLLAMA_MODEL", "qwen3:1.7b") +OLLAMA_MODEL = os.environ.get("REACT_OLLAMA_MODEL", "qwen3:8b") os.environ["OLLAMA_CHAT_MODEL"] = OLLAMA_MODEL