70 lines
2.0 KiB
Ruby
70 lines
2.0 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
RSpec.describe DiscourseAi::Completions::StructuredOutput do
|
|
subject(:structured_output) do
|
|
described_class.new(
|
|
{
|
|
message: {
|
|
type: "string",
|
|
},
|
|
bool: {
|
|
type: "boolean",
|
|
},
|
|
number: {
|
|
type: "integer",
|
|
},
|
|
status: {
|
|
type: "string",
|
|
},
|
|
},
|
|
)
|
|
end
|
|
|
|
describe "Parsing structured output on the fly" do
|
|
it "acts as a buffer for an streamed JSON" do
|
|
chunks = [
|
|
+"{\"message\": \"Line 1\\n",
|
|
+"Line 2\\n",
|
|
+"Line 3\", ",
|
|
+"\"bool\": true,",
|
|
+"\"number\": 4",
|
|
+"2,",
|
|
+"\"status\": \"o",
|
|
+"\\\"k\\\"\"}",
|
|
]
|
|
|
|
structured_output << chunks[0]
|
|
expect(structured_output.read_latest_buffered_chunk).to eq({ message: "Line 1\n" })
|
|
|
|
structured_output << chunks[1]
|
|
expect(structured_output.read_latest_buffered_chunk).to eq({ message: "Line 2\n" })
|
|
|
|
structured_output << chunks[2]
|
|
expect(structured_output.read_latest_buffered_chunk).to eq({ message: "Line 3" })
|
|
|
|
structured_output << chunks[3]
|
|
expect(structured_output.read_latest_buffered_chunk).to eq({ bool: true })
|
|
|
|
# Waiting for number to be fully buffered.
|
|
structured_output << chunks[4]
|
|
expect(structured_output.read_latest_buffered_chunk).to eq({ bool: true })
|
|
|
|
structured_output << chunks[5]
|
|
expect(structured_output.read_latest_buffered_chunk).to eq({ bool: true, number: 42 })
|
|
|
|
structured_output << chunks[6]
|
|
expect(structured_output.read_latest_buffered_chunk).to eq(
|
|
{ bool: true, number: 42, status: "o" },
|
|
)
|
|
|
|
structured_output << chunks[7]
|
|
expect(structured_output.read_latest_buffered_chunk).to eq(
|
|
{ bool: true, number: 42, status: "\"k\"" },
|
|
)
|
|
|
|
# No partial string left to read.
|
|
expect(structured_output.read_latest_buffered_chunk).to eq({ bool: true, number: 42 })
|
|
end
|
|
end
|
|
end
|