Add model and system prompt variables
This commit is contained in:
parent
4f57a52163
commit
69ad5c3ff3
1 changed files with 6 additions and 3 deletions
9
main.sh
9
main.sh
|
|
@ -5,13 +5,17 @@ source .env
|
||||||
PROMPT="$*"
|
PROMPT="$*"
|
||||||
TEMP_FILE=$(mktemp)
|
TEMP_FILE=$(mktemp)
|
||||||
prompt=$(echo $PROMPT)
|
prompt=$(echo $PROMPT)
|
||||||
|
ANTHROPIC_MODEL=${MODEL:-"claude-sonnet-4-5-20250929"}
|
||||||
|
SYSTEM_PROMPT=${SYSTEM_PROMPT:-""}
|
||||||
|
|
||||||
JSON_PAYLOAD=$(jq -n \
|
JSON_PAYLOAD=$(jq -n \
|
||||||
--arg model "claude-sonnet-4-5-20250929" \
|
--arg model "$ANTHROPIC_MODEL" \
|
||||||
--arg prompt "$PROMPT" \
|
--arg prompt "$PROMPT" \
|
||||||
|
--arg system_prompt "$SYSTEM_PROMPT" \
|
||||||
'{
|
'{
|
||||||
model: $model,
|
model: $model,
|
||||||
max_tokens: 1024,
|
system: $system_prompt,
|
||||||
|
max_tokens: 2048,
|
||||||
messages: [{role: "user", content: $prompt}]
|
messages: [{role: "user", content: $prompt}]
|
||||||
}')
|
}')
|
||||||
|
|
||||||
|
|
@ -29,4 +33,3 @@ echo -ne "\r\033[K"
|
||||||
|
|
||||||
# Display response
|
# Display response
|
||||||
jq -r '.content[0].text' "$TEMP_FILE"
|
jq -r '.content[0].text' "$TEMP_FILE"
|
||||||
rm "$TEMP_FILE"
|
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue