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
The following commit(s) were added to refs/heads/main by this push:
new a27fe601 [hotfix] Use qwen3:1.7b in Python Integration Test to improve
stability (#488)
a27fe601 is described below
commit a27fe60122389919f5b758ef0446274036a95f44
Author: Eugene <[email protected]>
AuthorDate: Tue Jan 27 11:41:05 2026 +0800
[hotfix] Use qwen3:1.7b in Python Integration Test to improve stability
(#488)
---
.../integrations/chat_models/tests/start_ollama_server.sh | 4 ++--
.../integrations/chat_models/tests/test_ollama_chat_model.py | 6 ++++--
2 files changed, 6 insertions(+), 4 deletions(-)
diff --git
a/python/flink_agents/integrations/chat_models/tests/start_ollama_server.sh
b/python/flink_agents/integrations/chat_models/tests/start_ollama_server.sh
index 75b369f8..bf7a50a0 100644
--- a/python/flink_agents/integrations/chat_models/tests/start_ollama_server.sh
+++ b/python/flink_agents/integrations/chat_models/tests/start_ollama_server.sh
@@ -35,6 +35,6 @@ if [[ $os == "Linux" ]]; then
sleep 10 # wait for ollama to start
fi
- ollama pull qwen3:0.6b
- ollama run qwen3:0.6b
+ echo "ollama pull $1"
+ ollama pull $1
fi
\ No newline at end of file
diff --git
a/python/flink_agents/integrations/chat_models/tests/test_ollama_chat_model.py
b/python/flink_agents/integrations/chat_models/tests/test_ollama_chat_model.py
index 653fb76b..321b499c 100644
---
a/python/flink_agents/integrations/chat_models/tests/test_ollama_chat_model.py
+++
b/python/flink_agents/integrations/chat_models/tests/test_ollama_chat_model.py
@@ -32,14 +32,16 @@ from
flink_agents.integrations.chat_models.ollama_chat_model import (
)
from flink_agents.plan.tools.function_tool import FunctionTool, from_callable
-test_model = os.environ.get("OLLAMA_CHAT_MODEL", "qwen3:0.6b")
+test_model = os.environ.get("OLLAMA_CHAT_MODEL", "qwen3:1.7b")
current_dir = Path(__file__).parent
try:
# only auto setup ollama in ci with python 3.10 to reduce ci cost.
if "3.10" in sys.version:
subprocess.run(
- ["bash", f"{current_dir}/start_ollama_server.sh"], timeout=300,
check=True
+ ["bash", f"{current_dir}/start_ollama_server.sh", test_model],
+ timeout=300,
+ check=True,
)
client = Client()
models = client.list()