Skip to content

Commit 5091211

Browse files
committed
wip
1 parent 10041d8 commit 5091211

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

gpt/xs/call.nu

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,13 @@
1818

1919
let p = anthropic provider
2020
let prepared = do $p.prepare-request $thread []
21-
let response = $prepared | do $p.call $key "claude-3-5-haiku-20241022" | do $p.response_stream_aggregate
21+
let response = $prepared | do $p.call $key "claude-3-5-haiku-20241022" | tee {
22+
each {|chunk|
23+
let event = do $p.response_stream_streamer $chunk
24+
if $event == null { return }
25+
26+
$event | to json | .append gpt.recv} } |
27+
do $p.response_stream_aggregate
2228

2329
$response
2430
}

tests/end-to-end/bootstrap.nu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ let req = .append gpt.call --meta {continues: (.head gpt.turn).id}
1919

2020
.cat -f | update hash { .cas } | take until {|frame|
2121
print ($frame | table -e)
22-
($frame.topic == "gpt.response") and ($frame.meta?.frame_id == $req.id)
22+
($frame.topic in ["gpt.error" "gpt.response"]) and ($frame.meta?.frame_id == $req.id)
2323
}
2424

2525
sleep 50ms

0 commit comments

Comments
 (0)