Prompt: (raw) (yaml) words:3130 bytes:31962
Review this bash script: #!/usr/bin/env bash # # Ollama Multirun # # Bash shell script to run a prompt against all models in Ollama, and save the output as web pages # # Usage: # - Enter prompt manually: ./multirun.sh # - Enter prompt as argument: ./multirun.sh "your prompt" # - Enter prompt from file: ./multirun.sh < prompt.txt # - Enter prompt from pipe: echo "your prompt" | ./multirun.sh # echo "summarize this file: $(cat filename)" | ./multirun.sh # # - By default, will use all available models # To set a list of models to use, set as a comma-seperated list with -m # example: ./multirun.sh -m deepseek-r1:1.5b,deepseek-r1:8b # # - By default, will use "./results" as the results output directory # To set a results output directory: # ./multirun.sh -r ./path/to/directory # # - By default, will wait 5 minutes for models to respond. # To set new timeout (in seconds): # ./multirun.sh -t 30 NAME="ollama-multirun" VERSION="5.17" URL="https://github.com/attogram/ollama-multirun" TIMEOUT="300" # number of seconds to allow model to respond usage() { me=$(basename "$0") echo "$NAME"; echo echo "Usage:" echo " ./$me [flags]" echo " ./$me [flags] [prompt]" echo; echo "Flags:"; echo " -h -- Help for $NAME" echo " -m model1,model2 -- Use specific models (comma separated list)" echo " -r <dir> -- Set results directory" echo " -t # -- Set timeout, in seconds" echo " -v -- Show version information" echo " [prompt] -- Set the prompt (\"Example prompt\")" } parseCommandLine() { modelsList="" resultsDirectory="results" prompt="" while (( "$#" )); do case "$1" in -h) usage exit 0 ;; -m) # specify models to run if [ -n "$2" ] && [ ${2:0:1} != "-" ]; then modelsList=$2 shift 2 else echo "Error: Argument for $1 is missing" >&2 exit 1 fi ;; -r) # specify results outputDirectory if [ -n "$2" ] && [ ${2:0:1} != "-" ]; then resultsDirectory=$2 shift 2 else echo "Error: Argument for $1 is missing" >&2 exit 1 fi ;; -t) # specify timeout in seconds if [ -n "$2" ] && [ ${2:0:1} != "-" ]; then TIMEOUT=$2 shift 2 else echo "Error: Argument for $1 is missing" >&2 exit 1 fi ;; -v) echo "$NAME v$VERSION" exit 0 ;; -*|--*=) # unsupported flags echo "Error: unsupported argument: $1" >&2 exit 1 #shift 1 ;; *) # preserve positional arguments prompt+="$1" shift ;; esac done # set positional arguments in their proper place eval set -- "$prompt" } getDateTime() { echo $(date '+%Y-%m-%d %H:%M:%S') } setModels() { models=($(ollama list | awk '{if (NR > 1) print $1}' | sort)) # Get list of models, sorted alphabetically if [ -z "$models" ]; then echo "No models found. Please install models with 'ollama pull <model-name>'" >&2 exit 1 fi parsedModels=() if [ -n "$modelsList" ]; then IFS=',' read -ra modelsListArray <<< "$modelsList" # parse csv into modelsListArray for m in "${modelsListArray[@]}"; do if [[ "${models[*]}" =~ "$m" ]]; then # if model exists parsedModels+=("$m") else echo "Error: model not found: $m" >&2 exit 1 fi done fi if [ -n "$parsedModels" ]; then models=("${parsedModels[@]}") fi echo "models:"; echo "${models[@]}" echo } safeString() { local input="$1" # Get the input input=${input:0:42} # Truncate to first 42 characters input=$(echo "$input" | tr '[:upper:]' '[:lower:]') # Convert to lowercase input=$(echo "$input" | sed "s/ /_/g") # Replace spaces with underscores input=$(echo "$input" | sed 's/[^a-zA-Z0-9_]/_/g' | tr -cd 'a-zA-Z0-9_') # Replace non-allowed characters with underscores echo "$input" # Output the sanitized string } createOutputDirectory() { tag=$(safeString "$prompt") tagDatetime=$(date '+%Y%m%d-%H%M%S') outputDirectory="$resultsDirectory/${tag}_${tagDatetime}" echo $(getDateTime) "Output Directory: $outputDirectory/" if [ ! -d "$outputDirectory" ]; then if ! mkdir -p "$outputDirectory"; then echo "Error: Failed to create Output Directory $outputDirectory" >&2 exit 1 fi fi } setPrompt() { if [ -n "$prompt" ]; then # if prompt is already set from command line return fi if [ -t 0 ]; then # Check if input is from a terminal (interactive) echo "Enter prompt:"; read -r prompt # Read prompt from user input return fi prompt=$(cat) # Read from standard input (pipe or file) } savePrompt() { echo $(getDateTime) "Prompt: $prompt" promptFile="$outputDirectory/prompt.txt" echo $(getDateTime) "Creating Prompt Text: $promptFile" echo "$prompt" > "$promptFile" promptWords=$(wc -w < "$promptFile" | awk '{print $1}') promptBytes=$(wc -c < "$promptFile" | awk '{print $1}') promptYamlFile="$outputDirectory/$tag.prompt.yaml" echo $(getDateTime) "Creating Prompt Yaml: $promptYamlFile" generatePromptYaml > "$promptYamlFile" } generatePromptYaml() { # Github Prompt YAML: https://docs.github.com/en/github-models/use-github-models/storing-prompts-in-github-repositories cat << EOF messages: - role: system content: '' - role: user content: | $(while IFS= read -r line; do echo " $line"; done <<< "$prompt") model: '' EOF } textarea() { local content="$1" # Get the input if [ -z "$content" ]; then content="" fi local padding="$2" if [ -z "$padding" ]; then padding=0 fi local max="$3" if [ -z "$max" ]; then max=25 fi local lines=$(echo "$content\n" | wc -l) # Get number of lines in content lines=$((lines + padding)) if [ "$lines" -gt "$max" ]; then lines=$max fi content=$(echo "$content" | sed 's/&/\&amp;/g; s/</\&lt;/g; s/>/\&gt;/g; s/"/\&quot;/g; s/'"'"'/\&#39;/g') # Escape HTML special characters echo "<textarea readonly rows='$lines'>${content}</textarea>" } showPrompt() { echo "<p>Prompt: (<a href='./prompt.txt'>raw</a>) (<a href='./${tag}.prompt.yaml'>yaml</a>)" echo " words:$promptWords bytes:$promptBytes<br />" textarea "$prompt" 2 10 # 0 padding, max 10 lines echo "</p>" } showImages() { if [ -n "$addedImages" ]; then for image in ${addedImages}; do echo -n "<div class='box'>" echo -n "<a target='image' href='$(basename $image)'><img src='$(basename $image)' alt='$image' width='250' /></a>" echo -n "</div>" done echo -n "<br />" fi } clearModel() { echo $(getDateTime) "Clearing model session: $1" ( expect \ -c "spawn ollama run $1" \ -c "expect \">>> \"" \ -c 'send -- "/clear\n"' \ -c "expect \"Cleared session context\"" \ -c 'send -- "/bye\n"' \ -c "expect eof" \ ; ) > /dev/null 2>&1 # Suppress output if [ $? -ne 0 ]; then echo "ERROR: Failed to clear model session: $1" >&2 # exit 1 fi } stopModel() { echo $(getDateTime) "Stopping model: $1" ollama stop "$1" if [ $? -ne 0 ]; then echo "ERROR: Failed to stop model: $1" >&2 # exit 1 fi } showSortableTablesJavascript() { # From: https://github.com/tofsjonas/sortable/ # License: The Unlicense - https://github.com/tofsjonas/sortable/blob/main/LICENSE echo '<style> .sortable thead th:not(.no-sort){cursor:pointer}.sortable thead th:not(.no-sort)::after,.sortable thead th:not(.no-sort)::before{transition:color .1s ease-in-out;font-size:1.2em;color:rgba(0,0,0,0)}.sortable thead th:not(.no-sort)::after{margin-left:3px;content:"▸"}.sortable thead th:not(.no-sort):hover::after{color:inherit}.sortable thead th:not(.no-sort)[aria-sort=descending]::after{color:inherit;content:"▾"}.sortable thead th:not(.no-sort)[aria-sort=ascending]::after{color:inherit;content:"▴"}.sortable thead th:not(.no-sort).indicator-left::after{content:""}.sortable thead th:not(.no-sort).indicator-left::before{margin-right:3px;content:"▸"}.sortable thead th:not(.no-sort).indicator-left:hover::before{color:inherit}.sortable thead th:not(.no-sort).indicator-left[aria-sort=descending]::before{color:inherit;content:"▾"}.sortable thead th:not(.no-sort).indicator-left[aria-sort=ascending]::before{color:inherit;content:"▴"}.sortable{--stripe-color: #e4e4e4;--th-color: #fff;--th-bg: #808080;--td-color: #000;--td-on-stripe-color: #000;border-spacing:0}.sortable.sticky thead th{position:sticky;top:0;z-index:1}.sortable tbody tr:nth-child(odd){background-color:var(--stripe-color);color:var(--td-on-stripe-color)}.sortable thead th{background:var(--th-bg);color:var(--th-color);font-weight:normal;text-align:left;text-transform:capitalize;vertical-align:baseline;white-space:nowrap}.sortable td{color:var(--td-color)}.sortable td,.sortable th{padding:10px}.sortable td:first-child,.sortable th:first-child{border-top-left-radius:4px}.sortable td:last-child,.sortable th:last-child{border-top-right-radius:4px} </style>' echo '<script> function sortSortable(a,m){function n(b){if(b){if(m&&b.dataset.sortAlt)return b.dataset.sortAlt;if(b.dataset.sort)return b.dataset.sort;if(b.textContent)return b.textContent}return""}a.dispatchEvent(new Event("sort-start",{bubbles:!0}));for(var c=a.tHead.querySelector("th[aria-sort]"),k=a.tHead.children[0],p="ascending"===c.getAttribute("aria-sort"),f=a.classList.contains("n-last"),e=function(b,q,d){var g=n(q.cells[d]),r=n(b.cells[d]);if(f){if(""===g&&""!==r)return-1;if(""===r&&""!==g)return 1}var u= +g-+r;g=isNaN(u)?g.localeCompare(r):u;return 0===g&&k.cells[d]&&k.cells[d].hasAttribute("data-sort-tbr")?e(b,q,+k.cells[d].dataset.sortTbr):p?-g:g},h=0;h<a.tBodies.length;h++){var l=a.tBodies[h],v=[].slice.call(l.rows,0);v.sort(function(b,q){var d;return e(b,q,+(null!==(d=c.dataset.sortCol)&&void 0!==d?d:c.cellIndex))});var t=l.cloneNode();t.append.apply(t,v);a.replaceChild(t,l)}a.dispatchEvent(new Event("sort-end",{bubbles:!0}))} function sortableEventListener(a){function m(h,l){return h.nodeName===l?h:m(h.parentNode,l)}try{var n=a.shiftKey||a.altKey,c=m(a.target,"TH"),k=c.parentNode,p=k.parentNode,f=p.parentNode;if("THEAD"===p.nodeName&&f.classList.contains("sortable")&&!c.classList.contains("no-sort")){var e=k.cells;for(a=0;a<e.length;a++)e[a]!==c&&e[a].removeAttribute("aria-sort");e="descending";if("descending"===c.getAttribute("aria-sort")||f.classList.contains("asc")&&"ascending"!==c.getAttribute("aria-sort"))e="ascending"; c.setAttribute("aria-sort",e);f.dataset.timer&&clearTimeout(+f.dataset.timer);f.dataset.timer=setTimeout(function(){sortSortable(f,n)},1).toString()}}catch(h){}}document.addEventListener("click",sortableEventListener); </script>' } showHeader() { title="$1" cat << "EOF" <!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <meta name="viewport" content="width=device-width, initial-scale=1.0"> <style> a:hover { background-color: yellow; color: black; } body { font-family: monospace; } header, footer { background-color: #f0f0f0; padding: 10px; } li { margin: 5px; } table, td, th { border-collapse: collapse; } td, th { border: 1px solid #cccccc; padding: 5px; text-align: right; } tr:hover { background-color: lightyellow; color: black; } textarea { border: 1px solid #cccccc; white-space: pre-wrap; width: 90%; } .box { display: inline-block; margin: 3px; padding: 2px; vertical-align: top; } .left { text-align: left; } .menu { font-size: small; } </style> EOF echo "<title>$title</title></head><body>" } showFooter() { title="$1" echo "<br /><br />" echo "<footer>" echo "<p>$title</p>" echo "<p>Page created: $(getDateTime)</p>" echo "<p>Generated with: <a target='$NAME' href='$URL'>$NAME</a> v$VERSION</p>" echo "</footer></body></html>" } createMenu() { local currentModel="$1" echo "<span class='menu'>" echo "<a href='models.html'>models</a>: " for modelName in ${models[@]}; do if [ "$modelName" == "$currentModel" ]; then echo "<b>$modelName</b> " else echo "<a href='./$(safeString "$modelName").html'>$modelName</a> " fi done echo '</span>'; } setStats() { statsTotalDuration=$(grep -oE "total duration:[[:space:]]+(.*)" "$modelStatsTxt" | awk '{ print $NF }') statsLoadDuration=$(grep -oE "load duration:[[:space:]]+(.*)" "$modelStatsTxt" | awk '{ print $NF }') statsPromptEvalCount=$(grep -oE "prompt eval count:[[:space:]]+(.*)" "$modelStatsTxt" | awk '{ print $4, $5 }') statsPromptEvalDuration=$(grep -oE "prompt eval duration:[[:space:]]+(.*)" "$modelStatsTxt" | awk '{ print $NF }') statsPromptEvalRate=$(grep -oE "prompt eval rate:[[:space:]]+(.*)" "$modelStatsTxt" | awk '{ print $4, $5 }') statsEvalCount=$(grep -oE "^eval count:[[:space:]]+(.*)" "$modelStatsTxt" | awk '{ print $3, $4 }') statsEvalDuration=$(grep -oE "^eval duration:[[:space:]]+(.*)" "$modelStatsTxt" | awk '{ print $NF }') statsEvalRate=$(grep -oE "^eval rate:[[:space:]]+(.*)" "$modelStatsTxt" | awk '{ print $3, $4 }') addedImages=$(grep -oE "Added image '(.*)'" "$modelStatsTxt" | awk '{ print $NF }' | sed "s/'//g") if [ -n "$addedImages" ]; then for image in ${addedImages}; do if ! [ -f "$outputDirectory"/"$(basename $image)" ]; then echo "Copying image: $image" cp $image $outputDirectory fi done fi responseWords=$(wc -w < "$modelOutputTxt" | awk '{print $1}') responseBytes=$(wc -c < "$modelOutputTxt" | awk '{print $1}') } setOllamaStats() { ollamaVersion=$(ollama -v | awk '{print $4}') # ps columns: 1:NAME, 2:ID, 3:SIZE_NUM 4:SIZE_GB, 5:PROCESSOR_% 6:PROCESS_TYPE, 7:CONTEXT, 8:UNTIL ollamaPs=$(ollama ps | awk '{print $1, $2, $3, $4, $5, $6, $7}' | sed '1d') # Get columns from ollama ps output, skipping the header ollamaModel=$(echo "$ollamaPs" | awk '{print $1}') # Get the model name ollamaSize=$(echo "$ollamaPs" | awk '{print $3, $4}') # Get the model size ollamaProcessor=$(echo "$ollamaPs" | awk '{print $5, $6}') # Get the processor ollamaContext=$(echo "$ollamaPs" | awk '{print $7}') # Get the context size } setSystemStats() { systemArch=$(uname -m) # Get hardware platform systemProcessor=$(uname -p) # Get system processor systemOSName=$(uname -s) # Get system OS name systemOSVersion=$(uname -r) # Get system OS version setSystemMemoryStats } setSystemMemoryStats() { systemMemoryUsed="?" systemMemoryAvail="?" #echo "OS Type: $OSTYPE" case "$OSTYPE" in cygwin|msys) #echo "OS Type match: cygwin|msys" if command -v wmic >/dev/null 2>&1; then local totalMemKB=$(wmic OS get TotalVisibleMemorySize /value 2>/dev/null | grep -E "^TotalVisibleMemorySize=" | cut -d'=' -f2 | tr -d '\r') local availMemKB=$(wmic OS get FreePhysicalMemory /value 2>/dev/null | grep -E "^FreePhysicalMemory=" | cut -d'=' -f2 | tr -d '\r') if [ -n "$totalMemKB" ] && [ -n "$availMemKB" ]; then local usedMemKB=$((totalMemKB - availMemKB)) # Convert KB to human readable format (approximate) if [ $usedMemKB -gt 1048576 ]; then systemMemoryUsed="$((usedMemKB / 1048576))G" elif [ $usedMemKB -gt 1024 ]; then systemMemoryUsed="$((usedMemKB / 1024))M" else systemMemoryUsed="${usedMemKB}K" fi if [ $availMemKB -gt 1048576 ]; then systemMemoryAvail="$((availMemKB / 1048576))G" elif [ $availMemKB -gt 1024 ]; then systemMemoryAvail="$((availMemKB / 1024))M" else systemMemoryAvail="${availMemKB}K" fi fi fi ;; darwin*) #echo "OS Type match: darwin" top=$(top -l 1 2>/dev/null || echo "") if [ -n "$top" ]; then systemMemoryUsed=$(echo "$top" | awk '/PhysMem/ {print $2}' || echo "N/A") systemMemoryAvail=$(echo "$top" | awk '/PhysMem/ {print $6}' || echo "N/A") fi ;; *) #echo "OS Type match: *" top=$(top -l 1 2>/dev/null || top -bn1 2>/dev/null || echo "") if [ -n "$top" ]; then systemMemoryUsed=$(echo "$top" | awk '/PhysMem/ {print $2}' || echo "N/A") systemMemoryAvail=$(echo "$top" | awk '/PhysMem/ {print $6}' || echo "N/A") fi ;; esac } showSystemStats() { echo "<div class='box'><table>" echo "<tr><td class='left' colspan='2'>System</td></tr>" echo "<tr><td class='left'>Ollama proc</td><td class='left'>$ollamaProcessor</td></tr>" echo "<tr><td class='left'>Ollama context</td><td class='left'>$ollamaContext</td></tr>" echo "<tr><td class='left'>Ollama version</td><td class='left'>$ollamaVersion</td></tr>" echo "<tr><td class='left'>Multirun timeout</td><td class='left'>$TIMEOUT seconds</td></tr>" echo "<tr><td class='left'>Sys arch</td><td class='left'>$systemArch</td></tr>" echo "<tr><td class='left'>Sys processor</td><td class='left'>$systemProcessor</td></tr>" echo "<tr><td class='left'>sys memory</td><td class='left'>$systemMemoryUsed + $systemMemoryAvail</td></tr>" echo "<tr><td class='left'>Sys OS</td><td class='left'>$systemOSName $systemOSVersion</td></tr>" echo "</table></div>" } createModelInfoTxt() { # Create model info files - for each model, do 'ollama show' and save the results to text file for model in "${models[@]}"; do modelInfoTxt="$outputDirectory/$(safeString "$model").info.txt" echo $(getDateTime) "Creating Model Info Text: $modelInfoTxt" ollama show "$model" > "$modelInfoTxt" done } setModelInfo() { modelInfoTxt="$outputDirectory/$(safeString "$model").info.txt" modelCapabilities=() modelSystemPrompt="" modelTemperature="" section="" while IFS= read -r line; do # Read the content of the file line by line line="$(echo -e "${line}" | sed -e 's/^[[:space:]]*//' -e 's/[[:space:]]*$//')" # Trim leading/trailing whitespace if [[ -z "$line" ]]; then section="" continue; # Skip empty lines fi if [[ $line == "Model"* ]]; then section="Model" continue elif [[ $line == "Capabilities"* ]]; then section="Capabilities" continue elif [[ $line == "System"* ]]; then section="System" continue elif [[ $line == "Parameters"* ]]; then section="Parameters" continue elif [[ $line == "License"* ]]; then section="License" continue elif [[ $line == "Projector"* ]]; then section="Projector" continue fi case $section in "Model") if [[ "$line" == "architecture"* ]]; then modelArchitecture=$(echo "$line" | awk '/architecture/ {print $2}') # Get model architecture fi if [[ "$line" == "parameters"* ]]; then modelParameters=$(echo "$line" | awk '/parameters/ {print $2}') # Get model parameters fi if [[ "$line" == "context length"* ]]; then modelContextLength=$(echo "$line" | awk '/context length/ {print $3}') # Get model context length fi if [[ "$line" == "embedding length"* ]]; then modelEmbeddingLength=$(echo "$line" | awk '/embedding length/ {print $3}') # Get model embedding length fi if [[ "$line" == "quantization"* ]]; then modelQuantization=$(echo "$line" | awk '/quantization/ {print $2}') # Get model quantization fi ;; "Capabilities") modelCapabilities+=("$line") ;; "System") modelSystemPrompt+="$line"$'\n' ;; "Parameters") if [[ "$line" == "temperature"* ]]; then modelTemperature=$(echo "$line" | awk '/temperature/ {print $2}') # Get model temperature fi ;; esac done < "$modelInfoTxt" } createModelsOverviewHtml() { # list of models used in current run modelsIndexHtml="$outputDirectory/models.html" echo $(getDateTime) "Creating Models Index Page: $modelsIndexHtml" { showHeader "$NAME: models" titleLink="<a href='../index.html'>$NAME</a>: <a href='./index.html'>$tag</a>: <b>models</b>: $tagDatetime" echo "<header>$titleLink</header>" cat <<- EOF <br /> <table class="sortable"> <thead> <tr> <th class='left'>model</th> <th>architecture</th> <th>parameters</th> <th>context<br />length</th> <th>embedding<br />length</th> <th>quantization</th> <th>temperature</th> <th>capabilities</th> <th class='left'>system prompt</th> <th>(raw)</th> <th>(index)</th> </tr> </thead> EOF } > "$modelsIndexHtml" for model in "${models[@]}"; do setModelInfo { echo "<tr>" echo "<td class='left'><a href='./$(safeString "$model").html'>$model</a></td>" echo "<td>$modelArchitecture</td>" echo "<td>$modelParameters</td>" echo "<td>$modelContextLength</td>" echo "<td>$modelEmbeddingLength</td>" echo "<td>$modelQuantization</td>" echo "<td>$modelTemperature</td>" echo "<td class='left'>$(printf "%s<br />" "${modelCapabilities[@]}")</td>" echo "<td class='left'>$modelSystemPrompt</td>" echo "<td><a href='./$(safeString "$model").info.txt'>raw</a></td>" echo "<td><a href='../models.html#$(safeString "$model")'>index</a></td>" echo "</tr>" } >> "$modelsIndexHtml" done { echo "</table>" showSortableTablesJavascript showFooter "$titleLink" } >> "$modelsIndexHtml" } createModelOutputHtml() { modelHtmlFile="$outputDirectory/$(safeString "$model").html" echo $(getDateTime) "Creating Model Output Page: $modelHtmlFile" { showHeader "$NAME: $model" titleLink="<a href='../index.html'>$NAME</a>: <a href='./index.html'>$tag</a>: <b>$model</b>: $tagDatetime" echo "<header>$titleLink<br /><br />" createMenu "$model" echo "</header>" showPrompt showImages modelThinkingTxt="$outputDirectory/$(safeString "$model").thinking.txt" if [ -f "$modelThinkingTxt" ]; then echo "<p>Thinking: $model (<a href='./$(safeString "$model").thinking.txt'>raw</a>)<br />" textarea "$(cat "$modelThinkingTxt")" 3 15 # 3 padding, max 15 lines echo "</p>" fi echo "<p>Output: $model (<a href='./$(safeString "$model").output.txt'>raw</a>)<br />" textarea "$(cat "$modelOutputTxt")" 3 25 # 3 padding, max 25 lines echo "</p>" echo "<div class='box'><table>" echo "<tr><td class='left' colspan='2'>Stats (<a href='./$(safeString "$model").stats.txt'>raw</a>)</td></tr>" echo "<tr><td class='left'>words</td><td>$responseWords</td></tr>" echo "<tr><td class='left'>bytes</td><td>$responseBytes</td></tr>" echo "<tr><td class='left'>total duration</td><td>$statsTotalDuration</td></tr>" echo "<tr><td class='left'>load duration</td><td>$statsLoadDuration</td></tr>" echo "<tr><td class='left'>prompt eval count</td><td>$statsPromptEvalCount</td></tr>" echo "<tr><td class='left'>prompt eval duration</td><td>$statsPromptEvalDuration</td></tr>" echo "<tr><td class='left'>prompt eval rate</td><td>$statsPromptEvalRate</td></tr>" echo "<tr><td class='left'>eval count</td><td>$statsEvalCount</td></tr>" echo "<tr><td class='left'>eval duration</td><td>$statsEvalDuration</td></tr>" echo "<tr><td class='left'>eval rate</td><td>$statsEvalRate</td></tr>" echo "</table></div>" echo "<div class='box'><table>" echo "<tr><td class='left' colspan='2'>Model (<a href='./$(safeString "$model").info.txt'>raw</a>)</td></tr>" echo "<tr><td class='left'>name</td><td class='left'><a href='../models.html#$(safeString "$model")'>$model</a></td></tr>" echo "<tr><td class='left'>architecture</td><td class='left'>$modelArchitecture</td></tr>" echo "<tr><td class='left'>size</td><td class='left'>$ollamaSize</td></tr>" echo "<tr><td class='left'>parameters</td><td class='left'>$modelParameters</td></tr>" echo "<tr><td class='left'>context length</td><td class='left'>$modelContextLength</td></tr>" echo "<tr><td class='left'>embedding length</td><td class='left'>$modelEmbeddingLength</td></tr>" echo "<tr><td class='left'>quantization</td><td class='left'>$modelQuantization</td></tr>" echo "<tr><td class='left'>capabilities</td><td class='left'>$(printf "%s<br />" "${modelCapabilities[@]}")</td>" echo "</table></div>" showSystemStats showFooter "$titleLink" } > "$modelHtmlFile" } createOutputIndexHtml() { outputIndexHtml="$outputDirectory/index.html" echo $(getDateTime) "Creating Output Index Page: $outputIndexHtml" { showHeader "$NAME: $tag" titleLink="<a href='../index.html'>$NAME</a>: <b>$tag</b>: $tagDatetime" echo "<header>$titleLink<br /><br />" createMenu "index" echo "</header>" showPrompt echo "<!-- IMAGES -->" cat <<- "EOF" <table class="sortable"> <thead> <tr> <th class='left'>model</th> <th>words</th> <th>bytes</th> <th>total<br />duration</th> <th>load<br />duration</th> <th>prompt eval<br />count</th> <th>prompt eval<br />duration</th> <th>prompt eval<br />rate</th> <th>eval<br />count</th> <th>eval<br />duration</th> <th>eval<br />rate</th> </tr> </thead> EOF } > "$outputIndexHtml" } addModelToOutputIndexHtml() { ( echo "<tr>" echo "<td class='left'><a href='./$(safeString "$model").html'>$model</a></td>" echo "<td>$responseWords</td>" echo "<td>$responseBytes</td>" echo "<td>$statsTotalDuration</td>" echo "<td>$statsLoadDuration</td>" echo "<td>$statsPromptEvalCount</td>" echo "<td>$statsPromptEvalDuration</td>" echo "<td>$statsPromptEvalRate</td>" echo "<td>$statsEvalCount</td>" echo "<td>$statsEvalDuration</td>" echo "<td>$statsEvalRate</td>" echo "</tr>" ) >> "$outputIndexHtml" } finishOutputIndexHtml() { { echo "</table>" echo "<br /><br />" showSystemStats showSortableTablesJavascript titleLink="<a href='../index.html'>$NAME</a>: <b>$tag</b>: $tagDatetime" showFooter "$titleLink" } >> "$outputIndexHtml" imagesHtml=$(showImages) sed -i '' -e "s#<!-- IMAGES -->#${imagesHtml}#" "$outputIndexHtml" } getSortedResultsDirectories() { # Sort directories by datetime at end of directory name echo $(ls -d "$resultsDirectory"/* | awk 'match($0, /[0-9]{8}-[0-9]{6}$/) { print $0, substr($0, RSTART, RLENGTH) }' | sort -k2 -r | cut -d' ' -f1) } createMainIndexHtml() { resultsIndexFile="${resultsDirectory}/index.html" echo $(getDateTime) "Creating Main Index Page: $resultsIndexFile" { showHeader "$NAME: results" titleLink="<b>$NAME</b>" echo "<header>$titleLink</header>" echo "<p><a href='models.html'>Models Index</a></p>" echo "<p>Runs:<ul>" for dir in $(getSortedResultsDirectories); do if [ -d "$dir" ]; then echo "<li><a href='${dir##*/}/index.html'>${dir##*/}</a></li>" fi done echo "</ul></p>" showFooter "$titleLink" } > $resultsIndexFile } createMainModelIndexHtml() { # create table of contents: list all models used in all run results, and links to every individual model run modelsFound=() modelsIndex=() for dir in $(getSortedResultsDirectories); do # for each item in main results directory if [ -d "$dir" ]; then # if is a directory for file in "$dir"/*.html; do # for each *.html file in the directory if [[ $file != *"/index.html" && $file != *"/models.html" ]]; then # skip index.html and models.html fileName="${file##*/}" modelName="${fileName%.html}" # remove .html to get model name if [[ ! "${modelsFound[@]}" =~ "$modelName" ]]; then modelsFound+=("$modelName") fi modelsIndex+=("$modelName:$dir/$fileName") fi done fi done mainModelIndexHtml="$resultsDirectory/models.html" echo; echo $(getDateTime) "Creating Main Model Index Page: $mainModelIndexHtml" { showHeader "$NAME: Model Run Index" titleLink="<b><a href='index.html'>$NAME</a></b>: Model Run Index" echo "<header>$titleLink</header>" echo '<p>Models: ' for foundModel in "${modelsFound[@]}"; do echo "<a href='#$foundModel'>$foundModel</a> " done echo '</p>' echo "<ul>" for foundModel in "${modelsFound[@]}"; do echo " <li id='$foundModel'>$foundModel</li>" echo " <ul>" for modelIndex in "${modelsIndex[@]}"; do modelName=${modelIndex%%:*} # get everything before the : if [ "$modelName" == "$foundModel" ]; then run=${modelIndex#*:} # get everything after the : runLink="${run#$resultsDirectory/}" # remove the results directory from beginning runName="${runLink%/*}" # remove everything after last slash including the slash echo " <li><a href='$runLink'>$runName</a></li>" fi done echo ' </ul>' done echo "</ul>" echo "<p><a href='./models.html'>top</a></p>" showFooter "$titleLink" } > "$mainModelIndexHtml" } runModelWithTimeout() { ollama run --verbose "${model}" -- "${prompt}" > "${modelOutputTxt}" 2> "${modelStatsTxt}" & pid=$! ( sleep $TIMEOUT if kill -0 $pid 2>/dev/null; then echo "[ERROR: Multirun Timeout after ${TIMEOUT} seconds]" > "${modelOutputTxt}" kill $pid 2>/dev/null fi ) & timeout_pid=$! # Wait for the main process to complete if wait $pid 2>/dev/null; then # Main process completed successfully, kill the timeout process if kill -0 $timeout_pid 2>/dev/null; then kill $timeout_pid 2>/dev/null wait $timeout_pid 2>/dev/null # Clean up the timeout process fi else # Main process was killed (likely by timeout), wait for timeout process wait $timeout_pid 2>/dev/null fi } parseThinkingOutput() { local modelThinkingTxt="$outputDirectory/$(safeString "$model").thinking.txt" # Check for either <think> tags or Thinking... patterns if grep -q -E "(<think>|Thinking\.\.\.)" "$modelOutputTxt"; then #echo "Found thinking content in $modelOutputTxt, extracting..." # Read the entire file content local content=$(cat "$modelOutputTxt") # Extract thinking content local thinkingContent="" thinkingContent+=$(echo "$content" | sed -n '/<think>/,/<\/think>/p' | sed '1d;$d') thinkingContent+=$(echo "$content" | sed -n '/Thinking\.\.\./,/\.\.\.done thinking\./p' | sed '1d;$d') # Remove thinking content from original content=$(echo "$content" | sed '/<think>/,/<\/think>/d') content=$(echo "$content" | sed '/Thinking\.\.\./,/\.\.\.done thinking\./d') echo $(getDateTime) "Creating Thinking Text: $modelThinkingTxt" echo "$thinkingContent" > "$modelThinkingTxt" echo $(getDateTime) "Updating Model Output Text: $modelOutputTxt" echo "$content" > "$modelOutputTxt" fi } export OLLAMA_MAX_LOADED_MODELS=1 parseCommandLine "$@" echo; echo "$NAME v$VERSION"; echo setModels setPrompt echo; echo $(getDateTime) "Response Timeout: $TIMEOUT" createOutputDirectory createMainIndexHtml savePrompt createModelInfoTxt createModelsOverviewHtml setSystemStats createOutputIndexHtml for model in "${models[@]}"; do # Loop through each model and run it with the given prompt echo; echo $(getDateTime) "Running model: $model" clearModel "$model" modelOutputTxt="$outputDirectory/$(safeString "$model").output.txt" modelStatsTxt="$outputDirectory/$(safeString "$model").stats.txt" echo $(getDateTime) "Creating Model Output Text: $modelOutputTxt" echo $(getDateTime) "Creating Model Stats Text: $modelStatsTxt" runModelWithTimeout setSystemMemoryStats setOllamaStats parseThinkingOutput setModelInfo setStats createModelOutputHtml addModelToOutputIndexHtml stopModel "$model" done finishOutputIndexHtml createMainModelIndexHtml echo; echo $(getDateTime) "Done: $outputDirectory/"

Output: llama3.2:1b (raw)
The provided bash script is a comprehensive tool for running Ollama models with multirun, a framework that allows you to create complex models and run them in parallel using multiple processes. Here's a review of the script: **Overall structure**: The script follows a well-structured approach, separating different tasks into separate functions. This makes it easy to read and maintain. **Functionality**: * It provides various options for specifying models, prompts, timeouts, and other settings. * The `parseCommandLine` function is used to parse command-line arguments and set default values for some flags (e.g., `-m`, `-r`, `-t`). * The `getDateTime` function returns a formatted date and time string. * The `setPrompt` function sets the prompt based on user input or prompts from files. * The `savePrompt` function saves the generated prompt to a file. * The `generatePromptYaml` function creates a YAML file for storing Ollama prompts. * The `showSortableTablesJavascript` function generates JavaScript code for sorting tables in output pages. * The `showFooter` function adds a footer to the HTML page containing links to the model index and results index. * The `createModelOutputHtml` function generates an HTML page with information about each model run. **Improvements and suggestions**: 1. Error handling: While there are some error handling mechanisms (e.g., checking if input is present), more robust error handling can be implemented for situations like invalid inputs or file operations. 2. Code organization: Consider breaking down the script into multiple files, each responsible for a specific task (e.g., `models.py`, `prompt_generator.py`, etc.). This will make it easier to maintain and extend the codebase. 3. Input validation: Some functions (e.g., `setPrompt`) assume that user input is valid; add checks for invalid inputs to prevent potential issues. 4. Performance optimization: The script uses a lot of system resources, which might impact performance on systems with limited resources. Consider using more efficient data structures or caching mechanisms when possible. 5. Testing: It's essential to write tests for the script to ensure it works as expected in various scenarios. This will help catch bugs and improve overall quality. 6. Comments and documentation: While some functions have brief comments, there is room for more comprehensive documentation (e.g., `README.md` file). 7. Code style: Some code formatting and indentation might benefit from a consistent style guide. **Specific issues**: 1. In the `showHeader` function, it tries to access model names without checking if they exist first. Add checks before accessing. 2. The `parseCommandLine` function assumes that all arguments are provided; add error handling for missing arguments. 3. Some functions (e.g., `createModelInfoTxt`, `setStats`) update files in the same directory, which might not be desirable. Consider moving these functions to separate scripts or using a different file structure. Overall, this script provides a solid foundation for running Ollama models with multirun. With some minor improvements and adherence to best practices, it can become an even more powerful tool for complex modeling tasks.