cmd/sketch: add skaband-supported qwen
Co-Authored-By: sketch <hello@sketch.dev>
Change-ID: scbc3e984d79ceda6k
diff --git a/dockerimg/dockerimg.go b/dockerimg/dockerimg.go
index 535ad65..80830e8 100644
--- a/dockerimg/dockerimg.go
+++ b/dockerimg/dockerimg.go
@@ -46,6 +46,9 @@
// ModelURL is the URL of the LLM service.
ModelURL string
+ // OAIModelName is the openai model name of the LLM model to use.
+ OAIModelName string
+
// ModelAPIKey is the API key for LLM service.
ModelAPIKey string
@@ -550,6 +553,9 @@
if config.ModelURL != "" {
cmdArgs = append(cmdArgs, "-e", "SKETCH_MODEL_URL="+config.ModelURL)
}
+ if config.OAIModelName != "" {
+ cmdArgs = append(cmdArgs, "-e", "SKETCH_OAI_MODEL_NAME="+config.OAIModelName)
+ }
if config.SketchPubKey != "" {
cmdArgs = append(cmdArgs, "-e", "SKETCH_PUB_KEY="+config.SketchPubKey)
}