fix: correct inference module pod spec generate (#92)

KusionStack · Sep 4, 2024 · 67ba019 · 67ba019
1 parent ce25df7
commit 67ba019
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 6 deletions.
diff --git a/modules/inference/example/dev/example_workspace.yaml b/modules/inference/example/dev/example_workspace.yaml
@@ -2,6 +2,6 @@
 modules: 
   inference: 
     path: oci://ghcr.io/kusionstack/inference
-    version: 0.1.0-beta.1
+    version: 0.1.0-beta.2
     configs:
       default: {}
diff --git a/modules/inference/example/dev/kcl.mod b/modules/inference/example/dev/kcl.mod
@@ -2,7 +2,7 @@
 name = "example"
 
 [dependencies]
-inference = { oci = "oci://ghcr.io/kusionstack/inference", tag = "0.1.0-beta.1" }
+inference = { oci = "oci://ghcr.io/kusionstack/inference", tag = "0.1.0-beta.2" }
 service = {oci = "oci://ghcr.io/kusionstack/service", tag = "0.1.0" }
 kam = { git = "https://github.com/KusionStack/kam.git", tag = "0.2.0" }
 

diff --git a/modules/inference/kcl.mod b/modules/inference/kcl.mod
@@ -1,3 +1,3 @@
 [package]
 name = "inference"
-version = "0.1.0-beta.1"
+version = "0.1.0-beta.2"
diff --git a/modules/inference/src/Makefile b/modules/inference/src/Makefile
@@ -2,7 +2,7 @@ TEST?=$$(go list ./... | grep -v 'vendor')
 ###### chang variables below according to your own modules ###
 NAMESPACE=kusionstack
 NAME=inference
-VERSION=0.1.0-beta.1
+VERSION=0.1.0-beta.2
 BINARY=../bin/kusion-module-${NAME}_${VERSION}
 
 LOCAL_ARCH := $(shell uname -m)

diff --git a/modules/inference/src/inference_generator.go b/modules/inference/src/inference_generator.go
@@ -208,8 +208,8 @@ func (infer *Inference) generatePodSpec(_ *module.GeneratorRequest) (v1.PodSpec,
 	var mountPath string
 	var modelPullCmd []string
 	var containerPort int32
-	switch infer.Framework {
-	case "ollama":
+	switch strings.ToLower(infer.Framework) {
+	case OllamaType:
 		mountPath = "/root/.ollama"
 
 		var builder strings.Builder