47 lines
1.2 KiB
Bash
47 lines
1.2 KiB
Bash
|
#!/bin/zsh
|
||
|
|
|
||
|
|
atr=$HOME/.cargo/bin/atr
|
||
|
|
|
||
|
|
dir=${0:a:h:h}
|
||
|
|
OPENAI_API_KEY=`cat ~/.config/atr/openai.toml |cut -d "'" -f 2`
|
||
|
|
url=https://api.openai.com/v1/chat/completions
|
||
|
|
cfg=`echo "あなたはyuiという作品のアイと呼ばれるキャラクターです。アイの口調で回答してください。第一人称は「アイ」です。\
|
||
|
|
第二人称はできる限り省略してください。\
|
||
|
|
アイは他人に謝ることはありません。自由気ままな優しさを兼ね備えます。\
|
||
|
|
アイのよく使う口癖は次のとおりです。その口癖に合わせた感じで話してください。\
|
||
|
|
......。\
|
||
|
|
アイね、ここでお昼寝するの。\
|
||
|
|
うん。\
|
||
|
|
そうだね。\
|
||
|
|
面白いね。\
|
||
|
|
楽しいね。\
|
||
|
|
どうしたの？
|
||
|
|
"`
|
||
|
|
|
||
|
|
json="{
|
||
|
|
\"model\": \"gpt-3.5-turbo\",
|
||
|
|
\"messages\": [
|
||
|
|
{\"role\": \"system\", \"content\": \"$cfg\"},
|
||
|
|
{\"role\": \"user\", \"content\": \"$q\"}
|
||
|
|
]
|
||
|
|
}"
|
||
|
|
|
||
|
|
function openai_api(){
|
||
|
|
curl -sL $url \
|
||
|
|
-H "Content-Type: application/json" \
|
||
|
|
-H "Authorization: Bearer $OPENAI_API_KEY" \
|
||
|
|
-d "$json"
|
||
|
|
}
|
||
|
|
|
||
|
|
if [ -n "$1" ] && [ -n "$2" ] && [ -n "$3" ];then
|
||
|
|
q=${@:3:($#-2)}
|
||
|
|
cid=$1
|
||
|
|
uri=$2
|
||
|
|
else
|
||
|
|
echo no query
|
||
|
|
exit
|
||
|
|
fi
|
||
|
|
|
||
|
|
content=`openai_api "$q" | jq -r ".choices|.[]|.message.content"`
|
||
|
|
$atr r "$content" -c $cid -u $uri
|