From 2fd8e02edff7b29bb72b6716c4248396aa774418 Mon Sep 17 00:00:00 2001
From: Max <shadow.iqka@qq.com>
Date: Fri, 6 Mar 2026 08:29:21 +0800
Subject: [PATCH] Enhance CI workflows for Tai service with Docker socket
 integration

- Add Docker socket volume mount to the Docker run commands in both `pr-test.yml` and `unit-test.yml`, allowing the Tai K8s container to interact with the host Docker daemon.
- Update kubeconfig generation steps to dynamically retrieve the K3D IP address, ensuring accurate configuration for Kubernetes interactions.

These changes improve the integration of the Tai service with Kubernetes by enabling better communication with the Docker environment during CI workflows.
---
 .github/workflows/pr-test.yml   | 7 +++++++
 .github/workflows/unit-test.yml | 7 +++++++
 2 files changed, 14 insertions(+)

diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml
index 9b671a0e..8db4b250 100644
--- a/.github/workflows/pr-test.yml
+++ b/.github/workflows/pr-test.yml
@@ -1139,6 +1139,7 @@ jobs:
           docker run -d --name tai-k8s \
             --network k3d-tai-test \
             -p 8081:8080 -p 9101:9100 -p 6443:6443 -p 6081:6080 \
+            -v /var/run/docker.sock:/var/run/docker.sock:ro \
             -v /tmp/kubeconfig-tai-k8s.yml:/etc/tai/kubeconfig.yml:ro \
             -e TAI_K8S_UPSTREAM="tcp://${K3D_IP}:6443" \
             -e TAI_KUBECONFIG=/etc/tai/kubeconfig.yml \
@@ -1981,7 +1982,10 @@ jobs:
           docker run -d --name tai-k8s \
             --network k3d-tai-test \
             -p 8081:8080 -p 9101:9100 -p 6443:6443 -p 6081:6080 \
+            -v /var/run/docker.sock:/var/run/docker.sock:ro \
+            -v /tmp/kubeconfig-tai-k8s.yml:/etc/tai/kubeconfig.yml:ro \
             -e TAI_K8S_UPSTREAM="tcp://${K3D_IP}:6443" \
+            -e TAI_KUBECONFIG=/etc/tai/kubeconfig.yml \
             yaoapp/tai:latest
 
           for i in $(seq 1 30); do
@@ -2006,7 +2010,10 @@ jobs:
 
       - name: Generate kubeconfig for benchmarks
         run: |
+          K3D_IP=$(docker inspect k3d-tai-test-server-0 | jq -r '.[0].NetworkSettings.Networks["k3d-tai-test"].IPAddress')
           k3d kubeconfig get tai-test > /tmp/kubeconfig-k3d.yml
+          sed "s|server: .*|server: https://${K3D_IP}:6443|" /tmp/kubeconfig-k3d.yml \
+            > /tmp/kubeconfig-tai-k8s.yml
           sed 's|server: .*|server: https://127.0.0.1:6443|' /tmp/kubeconfig-k3d.yml \
             > ${{ runner.temp }}/kubeconfig-tai.yml
 
diff --git a/.github/workflows/unit-test.yml b/.github/workflows/unit-test.yml
index 2173f606..31920fc1 100644
--- a/.github/workflows/unit-test.yml
+++ b/.github/workflows/unit-test.yml
@@ -847,6 +847,7 @@ jobs:
           docker run -d --name tai-k8s \
             --network k3d-tai-test \
             -p 8081:8080 -p 9101:9100 -p 6443:6443 -p 6081:6080 \
+            -v /var/run/docker.sock:/var/run/docker.sock:ro \
             -v /tmp/kubeconfig-tai-k8s.yml:/etc/tai/kubeconfig.yml:ro \
             -e TAI_K8S_UPSTREAM="tcp://${K3D_IP}:6443" \
             -e TAI_KUBECONFIG=/etc/tai/kubeconfig.yml \
@@ -1487,7 +1488,10 @@ jobs:
           docker run -d --name tai-k8s \
             --network k3d-tai-test \
             -p 8081:8080 -p 9101:9100 -p 6443:6443 -p 6081:6080 \
+            -v /var/run/docker.sock:/var/run/docker.sock:ro \
+            -v /tmp/kubeconfig-tai-k8s.yml:/etc/tai/kubeconfig.yml:ro \
             -e TAI_K8S_UPSTREAM="tcp://${K3D_IP}:6443" \
+            -e TAI_KUBECONFIG=/etc/tai/kubeconfig.yml \
             yaoapp/tai:latest
 
           for i in $(seq 1 30); do
@@ -1512,7 +1516,10 @@ jobs:
 
       - name: Generate kubeconfig for benchmarks
         run: |
+          K3D_IP=$(docker inspect k3d-tai-test-server-0 | jq -r '.[0].NetworkSettings.Networks["k3d-tai-test"].IPAddress')
           k3d kubeconfig get tai-test > /tmp/kubeconfig-k3d.yml
+          sed "s|server: .*|server: https://${K3D_IP}:6443|" /tmp/kubeconfig-k3d.yml \
+            > /tmp/kubeconfig-tai-k8s.yml
           sed 's|server: .*|server: https://127.0.0.1:6443|' /tmp/kubeconfig-k3d.yml \
             > ${{ runner.temp }}/kubeconfig-tai.yml