#!/bin/bash

[ ! -f .env ] || export $(grep -v '^#' .env | xargs)
SERVER="${OLLAMA_HOST:-http://localhost:11434}"
MODEL="${OLLAMA_MODEL:-llama3.2}"
PROMPT=$(cat << EOF
Please clean up the following code, leaving ample documentation:
def test(a: int, b: float) -> int:
  return a + b ** 0.5
EOF
)
RESPONSE=$(curl -s "$SERVER/api/generate" \
  -H "Content-Type: application/json" \
  -d "$(jq -n --arg model "$MODEL" --arg prompt "$PROMPT" \
        '{model:$model, prompt:$prompt, stream:false}')" \
        | jq -r '.response')
echo "${RESPONSE}"
#curl -s "$SERVER/api/generate" \
#  -H "Content-Type: application/json" \
#  -d "{
#    \"model\": \"$MODEL\",
#    \"prompt\": \"$PROMPT\",
#    \"stream\": false
#  }" | jq -r '.response'