Prechádzať zdrojové kódy

feat: allow insecure pull

Timothy J. Baek 11 mesiacov pred
rodič
commit
555e4e32d5
1 zmenil súbory, kde vykonal 4 pridanie a 1 odobranie
  1. 4 1
      backend/apps/ollama/main.py

+ 4 - 1
backend/apps/ollama/main.py

@@ -306,6 +306,9 @@ async def pull_model(
 
     r = None
 
+    # Admin should be able to pull models from any source
+    payload = {**form_data.model_dump(exclude_none=True), "insecure": True}
+
     def get_request():
         nonlocal url
         nonlocal r
@@ -333,7 +336,7 @@ async def pull_model(
             r = requests.request(
                 method="POST",
                 url=f"{url}/api/pull",
-                data=form_data.model_dump_json(exclude_none=True).encode(),
+                data=json.dumps(payload),
                 stream=True,
             )