From 7400bed1bef7d3104f4778a272e7db912bade52c Mon Sep 17 00:00:00 2001
From: "hanzhi.421" <hanzhi.421@bytedance.com>
Date: Tue, 9 Dec 2025 23:36:35 +0800
Subject: [PATCH] fix: headers

---
 veadk/tools/builtin_tools/image_generate.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/veadk/tools/builtin_tools/image_generate.py b/veadk/tools/builtin_tools/image_generate.py
index a31b549f..31387c03 100644
--- a/veadk/tools/builtin_tools/image_generate.py
+++ b/veadk/tools/builtin_tools/image_generate.py
@@ -146,6 +146,14 @@ def handle_single_task_sync(
                 response = client.images.generate(
                     model=getenv("MODEL_IMAGE_NAME", DEFAULT_IMAGE_GENERATE_MODEL_NAME),
                     **inputs,
+                    extra_headers={
+                        "veadk-source": "veadk",
+                        "veadk-version": VERSION,
+                        "User-Agent": f"VeADK/{VERSION}",
+                        "X-Client-Request-Id": getenv(
+                            "MODEL_AGENT_CLIENT_REQ_ID", f"veadk/{VERSION}"
+                        ),
+                    },
                 )
 
             if not response.error: