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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
#!/bin/bash
[ ! -f .env ] || export $(grep -v '^#' .env | xargs)
[ ! -f .teagent ] || export $(grep -v '^#' .teagent | xargs)
if [ -z "$REPO" ]; then
exit 1
fi
# Default 5 minute maximum
TIMEOUT_MAX="${TIMEOUT_MAX:-300}"
SERVER="${OLLAMA_HOST:-http://localhost:11434}"
MODEL="${OLLAMA_MODEL:-llama3.2}"
echo "${SERVER}"
for FILE in agent/ast/*.py; do
PROMPT=$(cat << EOF
You are an engineer in charge of freshening up old code. For this task, you will be given a source file. Please update documentation as necessary and propose any changes to better suit the code. The code is in Python.
To do this task, you will be drafting a PR. Therefore, you will output three pieces: a 'title' for the PR, the 'discussion' of the PR, and the 'code' of the PR.
(You may treat any called functions from the histclass package as correctly working functions)
File contents:
`cat "${FILE}"`
EOF
)
echo "${PROMPT}"
SCHEMA="json"
OPTS='{"num_ctx": 8192}'
SCHEMA='{"type": "object", "properties": {"code": {"type": "string"}, "title": {"type": "string"}, "discussion": {"type": "string"}}, "required": ["code", "title", "dicussion"]}'
CDATA="$(jq -n --arg model "$MODEL" --arg prompt "$PROMPT" --argjson options "$OPTS" --argjson schema "$SCHEMA" \
'{model:$model, prompt:$prompt, options: $options, stream:false, format: $schema}')"
echo "${CDATA}"
RESPONSE=$(timeout "${TIMEOUT_MAX}" curl -s "$SERVER/api/generate" \
-H "Content-Type: application/json" \
-d "${CDATA}")
# '{model:$model, prompt:$prompt, stream:false}')" \
# | jq -r '.response')
echo "${RESPONSE}"
#echo "${RESPONSE}" | jq -r '.response' | jq
echo "${RESPONSE}" | jq -r '.response' | jq -r '.title'
echo "${RESPONSE}" | jq -r '.response' | jq -r '.discussion'
echo "${RESPONSE}" | jq -r '.response' | jq -r '.code'
TITLE="`echo "${RESPONSE}" | jq -r '.response' | jq -r '.title'`"
DISCUSSION="`echo "${RESPONSE}" | jq -r '.response' | jq -r '.discussion'`"
CODE="`echo "${RESPONSE}" | jq -r '.response' | jq -r '.code'`"
SOURCE_FILE="`head -n1 \"${FILE}\"`"
SOURCE_FILE="${SOURCE_FILE### Function derived from }"
BODY=$(cat << EOF
File Source: ${SOURCE_FILE}
# Details
${DISCUSSION}
# Code
\`\`\`python3
${CODE}
\`\`\`
EOF
)
echo "${BODY}"
echo "${TITLE}"
tea issues create --title "${TITLE}" --body "${BODY}" --login teagent --repo "cc/HistClass"
done
|
