From 5c145570747161e98e34940fe1bfb2dfdf83f480 Mon Sep 17 00:00:00 2001
From: ishandhanani <ishandhanani@gmail.com>
Date: Mon, 6 May 2024 21:45:34 -0400
Subject: [PATCH] proper verb yaml

---
 pkg/cmd/ollama/cverb.yaml                     | 19 -------------------
 pkg/cmd/ollama/{verb.yaml => ollamaverb.yaml} |  9 +++++----
 2 files changed, 5 insertions(+), 23 deletions(-)
 delete mode 100644 pkg/cmd/ollama/cverb.yaml
 rename pkg/cmd/ollama/{verb.yaml => ollamaverb.yaml} (54%)

diff --git a/pkg/cmd/ollama/cverb.yaml b/pkg/cmd/ollama/cverb.yaml
deleted file mode 100644
index 58d856d6..00000000
--- a/pkg/cmd/ollama/cverb.yaml
+++ /dev/null
@@ -1,19 +0,0 @@
-build:
-  system_packages: []
-  python_version: '3.10'
-  cuda: 12.0.1
-  python_packages:
-    - jupyterlab
-  run:
-    - sh -c "$(curl -fsSL https://raw.githubusercontent.com/ohmyzsh/ohmyzsh/master/tools/install.sh)" "" --unattended
-user:
-  shell: zsh
-  authorized_keys_path: /home/ubuntu/.ssh/authorized_keys
-ports:
-  - '2222:22'
-  - '8000:8000'
-services:
-  - name: jupyter
-    entrypoint: jupyter-lab --ip=0.0.0.0 --no-browser --NotebookApp.token='' --NotebookApp.password=''
-    ports:
-      - 127.0.0.1:8888:8888
\ No newline at end of file
diff --git a/pkg/cmd/ollama/verb.yaml b/pkg/cmd/ollama/ollamaverb.yaml
similarity index 54%
rename from pkg/cmd/ollama/verb.yaml
rename to pkg/cmd/ollama/ollamaverb.yaml
index eeb6249f..71935d5a 100644
--- a/pkg/cmd/ollama/verb.yaml
+++ b/pkg/cmd/ollama/ollamaverb.yaml
@@ -1,18 +1,19 @@
 build:
-  python_version: '3.10'
+  python_version: "3.10"
   cuda: 12.0.1
   python_packages:
     - jupyterlab
   run:
     - curl -fsSL https://ollama.com/install.sh | sh
+    - ollama serve & sleep 10; ollama pull llama3; echo "kill 'ollama serve' process"; ps -ef | grep 'ollama serve' | grep -v grep | awk '{print $2}' | xargs -r kill -9
 user:
   shell: zsh
   authorized_keys_path: /home/ubuntu/.ssh/authorized_keys
 ports:
-  - '2222:22'
-  - '8000:8000'
+  - "2222:22"
+  - "8000:8000"
 services:
-  - name: ollama
+  - name: ollama-server
     entrypoint: OLLAMA_HOST=0.0.0.0 ollama serve
     ports:
       - 127.0.0.1:11434:11434
\ No newline at end of file