dont consume reader when calculating digest

Add wizard vicuna uncensored model link

README.md

@@ -54,12 +54,13 @@ Hello! It's your friend Mario.
 
 Ollama includes a library of open-source, pre-trained models. More models are coming soon.
 
-| Model       | Parameters | Size  | Download                  |
-| ----------- | ---------- | ----- | ------------------------- |
-| Llama2      | 7B         | 3.8GB | `ollama pull llama2`      |
-| Orca Mini   | 3B         | 1.9GB | `ollama pull orca`        |
-| Vicuna      | 7B         | 3.8GB | `ollama pull vicuna`      |
-| Nous-Hermes | 13B         | 7.3GB | `ollama pull nous-hermes` |
+| Model                     | Parameters | Size  | Download                    |
+| ----------------------    | ---------- | ----- | --------------------------- |
+| Llama2                    | 7B         | 3.8GB | `ollama pull llama2`        |
+| Orca Mini                 | 3B         | 1.9GB | `ollama pull orca`          |
+| Vicuna                    | 7B         | 3.8GB | `ollama pull vicuna`        |
+| Nous-Hermes               | 13B        | 7.3GB | `ollama pull nous-hermes`   |
+| Wizard Vicuna Uncensored  | 13B        | 6.8GB | `ollama pull wizard-vicuna` |
 
 ## Building
 

app/src/install.ts

@@ -13,7 +13,9 @@ export function installed() {
 }
 
 export async function install() {
-  const command = `do shell script "ln -F -s ${ollama} ${symlinkPath}" with administrator privileges`
+  const command = `do shell script "mkdir -p ${path.dirname(
+    symlinkPath
+  )} && ln -F -s ${ollama} ${symlinkPath}" with administrator privileges`
 
   try {
     await exec(`osascript -e '${command}'`)

parser/parser.go

@@ -46,8 +46,8 @@ func Parse(reader io.Reader) ([]Command, error) {
 				return nil, fmt.Errorf("no model specified in FROM line")
 			}
 			foundModel = true
-		case "PROMPT":
-			command.Name = "prompt"
+		case "PROMPT", "LICENSE":
+			command.Name = strings.ToLower(fields[0])
 			if fields[1] == `"""` {
 				multiline = true
 				multilineCommand = &command

server/images.go

@@ -215,6 +215,16 @@ func CreateModel(name string, mf io.Reader, fn func(status string)) error {
 			}
 			l.MediaType = "application/vnd.ollama.image.prompt"
 			layers = append(layers, l)
+		case "license":
+			fn("creating license layer")
+			license := strings.NewReader(c.Arg)
+			l, err := CreateLayer(license)
+			if err != nil {
+				fn(fmt.Sprintf("couldn't create license layer: %v", err))
+				return fmt.Errorf("failed to create layer: %v", err)
+			}
+			l.MediaType = "application/vnd.ollama.image.license"
+			layers = append(layers, l)
 		default:
 			params[c.Name] = c.Arg
 		}
@@ -641,8 +651,7 @@ func createConfigLayer(layers []string) (*LayerReader, error) {
 		return nil, err
 	}
 
-	buf := bytes.NewBuffer(configJSON)
-	digest, size := GetSHA256Digest(buf)
+	digest, size := GetSHA256Digest(bytes.NewBuffer(configJSON))
 
 	layer := &LayerReader{
 		Layer: Layer{
@@ -650,7 +659,7 @@ func createConfigLayer(layers []string) (*LayerReader, error) {
 			Digest:    digest,
 			Size:      size,
 		},
-		Reader: buf,
+		Reader: bytes.NewBuffer(configJSON),
 	}
 	return layer, nil
 }