lint

Signed-off-by: Samhita Alla <[email protected]>
flyteorg · Aug 13, 2024 · 387d9b9 · 387d9b9
1 parent 1297f7a
commit 387d9b9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/plugins/flytekit-inference/flytekitplugins/inference/ollama/serve.py b/plugins/flytekit-inference/flytekitplugins/inference/ollama/serve.py
@@ -39,9 +39,10 @@ def setup_ollama_pod_template(self):
         )
 
         container_name = "create-model" if self._model_modelfile else "pull-model"
+        modelfile_escaped = self._model_modelfile.replace("\n", "\\n") if self._model_modelfile else None
         command = (
-            f'sleep 15; curl -X POST {self.base_url}/api/create -d \'{{"name": "{self._model_name}", "modelfile": "{self._model_modelfile.replace("\n", "\\n")}"}}\''
-            if self._model_modelfile
+            f'sleep 15; curl -X POST {self.base_url}/api/create -d \'{{"name": "{self._model_name}", "modelfile": "{modelfile_escaped}"}}\''
+            if modelfile_escaped
             else f'sleep 15; curl -X POST {self.base_url}/api/pull -d \'{{"name": "{self._model_name}"}}\''
         )