summaryrefslogtreecommitdiffstats
path: root/krebs/5pkgs/simple/vicuna-chat/default.nix
diff options
context:
space:
mode:
Diffstat (limited to 'krebs/5pkgs/simple/vicuna-chat/default.nix')
-rw-r--r--krebs/5pkgs/simple/vicuna-chat/default.nix33
1 files changed, 0 insertions, 33 deletions
diff --git a/krebs/5pkgs/simple/vicuna-chat/default.nix b/krebs/5pkgs/simple/vicuna-chat/default.nix
deleted file mode 100644
index db15899d6..000000000
--- a/krebs/5pkgs/simple/vicuna-chat/default.nix
+++ /dev/null
@@ -1,33 +0,0 @@
-{ pkgs, ... }:
-pkgs.writers.writeDashBin "vicuna-chat" ''
- set -efu
-
- export PATH=${with pkgs; lib.makeBinPath [
- coreutils
- curl
- jq
- ]}
-
- CONTEXT=''${CONTEXT:-$(date -Id)}
- PROMPT=$*
-
- if ! test -e "$CONTEXT"; then
- echo -n 'null' > "$CONTEXT"
- fi
-
- add_to_context() {
- jq -rc --argjson message "$1" '. + [$message]' "$CONTEXT" > "$CONTEXT.tmp"
- mv "$CONTEXT.tmp" "$CONTEXT"
- }
-
- add_to_context "{\"role\": \"user\", \"content\": \"$PROMPT\"}"
- response=$(
- jq -nc --slurpfile context "$CONTEXT" '{
- model: "vicuna-13b-v1.5-16k",
- messages: $context[0],
- }' |
- curl -Ss http://vicuna.r/v1/chat/completions -H 'Content-Type: application/json' -d @-
- )
- add_to_context "$(jq -rcn --argjson response "$response" '$response.choices[0].message')"
- jq -rcn --argjson response "$response" '$response.choices[0].message.content'
-''