fix: status code

StartUpNationLabs · Oct 20, 2024 · 935d56d · 935d56d
1 parent 5d4fdbc
commit 935d56d
Show file tree

Hide file tree

Showing 5 changed files with 24 additions and 29 deletions.
diff --git a/config/ollama/deployment.yaml b/config/ollama/deployment.yaml
@@ -22,6 +22,8 @@ spec:
           env:
             - name: OLLAMA_LOG_LEVEL
               value: "DEBUG"
+            - name: OLLAMA_DEBUG
+              value: "1"
 ---
 apiVersion: v1
 kind: Service

diff --git a/config/samples/ollama_v1_custommodel.yaml b/config/samples/ollama_v1_custommodel.yaml
@@ -6,7 +6,7 @@ metadata:
     app.kubernetes.io/managed-by: kustomize
   name: mario-custommodel-sample
 spec:
-  modelName: mario-custommodel-sample
+  modelName: mario-custommodel-sample:v1
   modelFile: >-
     FROM llama3.2
     # sets the temperature to 1 [higher is more creative, lower is more coherent]

diff --git a/internal/controller/custommodel_controller.go b/internal/controller/custommodel_controller.go
@@ -155,7 +155,7 @@ func (r *CustomModelReconciler) Reconcile(ctx context.Context, req ctrl.Request)
 		Modelfile: &CustomModel.Spec.ModelFile,
 		Stream:    &stream,
 	})
-	if err != nil {
+	if err != nil || res.StatusCode() != 200 {
 		logger.Error(err, "unable to create CustomModel")
 		return ctrl.Result{}, err
 	}

diff --git a/internal/ollama_client/api.gen.go b/internal/ollama_client/api.gen.go
diff --git a/openapi.yaml b/openapi.yaml
@@ -83,9 +83,6 @@ paths:
                 name:
                   type: string
                   description: Name of the model to create (required)
-                path:
-                  type: string
-                  description: Path to the Modelfile
                 modelfile:
                   type: string
                   description: The Modelfile contents