blob: 18bc777bfee8cf8f6fe5452b6a80a2d928e6e5cf (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
#!/bin/bash
[ ! -f .env ] || export $(grep -v '^#' .env | xargs)
[ ! -f .teagent ] || export $(grep -v '^#' .teagent | xargs)
if [ -z "$REPO" ]; then
exit 1
fi
SERVER="${OLLAMA_HOST:-http://localhost:11434}"
MODEL="${OLLAMA_MODEL:-llama3.2}"
SOURCE_FILES=`find . -type f -iname \*.py`
for source_file in $SOURCE_FILES; do
if [ ! -s "${source_file}" ]; then
continue
fi
SOURCE=$(<"${source_file}")
PROMPT=$(cat << EOF
Please write tests for the following source, leaving ample documentation.
If there don't seem to be any natural tests, simply write: DONE
EOF
)
PROMPT="${PROMPT}
${SOURCE}"
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')
if [[ "$RESPONSE" = 'DONE'* ]]; then
echo > /dev/null
else
tea issues create --title "Cleanup ${source_file} (TeAgent)" --body "${RESPONSE}" --login teagent --repo "${REPO}"
fi
done
|
