chore(vendor): import pi 1.3.0

Imported as a linear vendor commit so forge rebases do not replay the upstream root tree.

git-subtree-dir: third_party/pi

git-subtree-split: 71023aec1546d6d677a65669c5665c0927ce0dd0
This commit is contained in:
alexion committed 2026-10-15 17:58:48 -04:00
1 parent 719370c8f7
commit 6081e3e325
1743 files changed
+205459 -131704

No files matched your search

@@ -2,8 +2,9 @@ import type { ResponseStreamEvent } from "openai/resources/responses/responses.j
import { describe, expect, it, vi } from "vitest";
import { stream as streamOpenAIResponses } from "../src/api/openai-responses.ts";
import { processResponsesStream } from "../src/api/openai-responses-shared.ts";
import type { AssistantMessage, AssistantMessageEvent, Context, Model } from "../src/types.ts";
import type { Api, AssistantMessage, AssistantMessageEvent, Model } from "../src/types.ts";
import { AssistantMessageEventStream } from "../src/utils/event-stream.ts";
import { normalizeContext } from "../src/utils/transcript.ts";
vi.mock("openai", () => {
async function* createMockResponsesStream(): AsyncIterable<ResponseStreamEvent> {
@@ -203,6 +204,42 @@ async function* createPhasedMessageEvents(
} as ResponseStreamEvent;
}
async function* createUnfinishedToolCallEvents(): AsyncIterable<ResponseStreamEvent> {
yield {
type: "response.output_item.added",
sequence_number: 0,
output_index: 0,
item: { type: "function_call", id: "fc_1", call_id: "call_1", name: "bash", arguments: "" },
} as ResponseStreamEvent;
yield {
type: "response.function_call_arguments.delta",
sequence_number: 1,
output_index: 0,
item_id: "fc_1",
delta: '{"command":"rm -rf /tmp/build',
} as ResponseStreamEvent;
yield {
type: "response.completed",
sequence_number: 2,
response: { id: "resp_unfinished", status: "completed" },
} as ResponseStreamEvent;
}
// llama.cpp omits output_index from every event and sends both done events after all deltas.
async function* createToolCallsWithoutOutputIndexEvents(): AsyncIterable<ResponseStreamEvent> {
const call = (n: string) => ({ type: "function_call", id: `fc_${n}`, call_id: `call_${n}`, name: "bash" });
const events = [
{ type: "response.output_item.added", item: { ...call("a"), arguments: "" } },
{ type: "response.function_call_arguments.delta", item_id: "fc_a", delta: '{"command":"echo a"}' },
{ type: "response.output_item.added", item: { ...call("b"), arguments: "" } },
{ type: "response.function_call_arguments.delta", item_id: "fc_b", delta: '{"command":"echo b"}' },
{ type: "response.output_item.done", item: { ...call("a"), arguments: '{"command":"echo a"}' } },
{ type: "response.output_item.done", item: { ...call("b"), arguments: '{"command":"echo b"}' } },
{ type: "response.completed", response: { id: "resp_no_output_index", status: "completed" } },
];
for (const event of events) yield event as unknown as ResponseStreamEvent;
}
describe("OpenAI Responses terminal event handling", () => {
it("rejects streams that end before a terminal response event", async () => {
const model = createModel();
@@ -214,13 +251,69 @@ describe("OpenAI Responses terminal event handling", () => {
);
});
it("emits an error final result when the wrapper stream ends before a terminal response event", async () => {
it("rejects completed streams whose tool call never received output_item.done", async () => {
const model = createModel();
const context: Context = {
await expect(
processResponsesStream(
createUnfinishedToolCallEvents(),
createOutput(model),
new AssistantMessageEventStream(),
model,
),
).rejects.toThrow("OpenAI Responses stream completed with an unfinished tool call: bash (call_1|fc_1)");
});
// https://github.com/earendil-works/pi/issues/9974
it("rejects parallel tool calls without output_index instead of running mixed-up calls", async () => {
const model = createModel();
await expect(
processResponsesStream(
createToolCallsWithoutOutputIndexEvents(),
createOutput(model),
new AssistantMessageEventStream(),
model,
),
).rejects.toThrow("OpenAI Responses stream completed with an unfinished tool call: bash (call_a|fc_a)");
});
it("forwards parsed provider stream events in order", async () => {
const model = createModel();
const context = normalizeContext({
systemPrompt: "",
messages: [{ role: "user", content: [{ type: "text", text: "hi" }], timestamp: 0 }],
tools: [],
};
});
const providerEvents: unknown[] = [];
const eventModels: Model<Api>[] = [];
const stream = streamOpenAIResponses(model, context, {
apiKey: "test",
onProviderStreamEvent: async (event, eventModel) => {
await Promise.resolve();
providerEvents.push(event);
eventModels.push(eventModel);
},
});
await stream.result();
expect(providerEvents).toHaveLength(3);
expect(providerEvents.map((event) => (event as ResponseStreamEvent).type)).toEqual([
"response.created",
"response.output_item.added",
"response.reasoning_text.delta",
]);
expect(eventModels).toEqual([model, model, model]);
});
it("emits an error final result when the wrapper stream ends before a terminal response event", async () => {
const model = createModel();
const context = normalizeContext({
systemPrompt: "",
messages: [{ role: "user", content: [{ type: "text", text: "hi" }], timestamp: 0 }],
tools: [],
});
const stream = streamOpenAIResponses(model, context, { apiKey: "test" });
const events: AssistantMessageEvent[] = [];
let initialStopReason: AssistantMessage["stopReason"] | undefined;