Skip to content

Commit 42b8158

Browse files
authored
fix: headers (#361)
1 parent 21fb381 commit 42b8158

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

veadk/tools/builtin_tools/image_generate.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -146,6 +146,14 @@ def handle_single_task_sync(
146146
response = client.images.generate(
147147
model=getenv("MODEL_IMAGE_NAME", DEFAULT_IMAGE_GENERATE_MODEL_NAME),
148148
**inputs,
149+
extra_headers={
150+
"veadk-source": "veadk",
151+
"veadk-version": VERSION,
152+
"User-Agent": f"VeADK/{VERSION}",
153+
"X-Client-Request-Id": getenv(
154+
"MODEL_AGENT_CLIENT_REQ_ID", f"veadk/{VERSION}"
155+
),
156+
},
149157
)
150158

151159
if not response.error:

0 commit comments

Comments
 (0)