@mohanscodex/spectra-ai 0.4.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (62) hide show
  1. package/README.md +189 -0
  2. package/dist/__tests__/e2e.test.d.ts +2 -0
  3. package/dist/__tests__/e2e.test.d.ts.map +1 -0
  4. package/dist/__tests__/e2e.test.js +364 -0
  5. package/dist/__tests__/e2e.test.js.map +1 -0
  6. package/dist/__tests__/event-stream.test.d.ts +2 -0
  7. package/dist/__tests__/event-stream.test.d.ts.map +1 -0
  8. package/dist/__tests__/event-stream.test.js +87 -0
  9. package/dist/__tests__/event-stream.test.js.map +1 -0
  10. package/dist/__tests__/provider.test.d.ts +2 -0
  11. package/dist/__tests__/provider.test.d.ts.map +1 -0
  12. package/dist/__tests__/provider.test.js +40 -0
  13. package/dist/__tests__/provider.test.js.map +1 -0
  14. package/dist/event-stream.d.ts +19 -0
  15. package/dist/event-stream.d.ts.map +1 -0
  16. package/dist/event-stream.js +74 -0
  17. package/dist/event-stream.js.map +1 -0
  18. package/dist/index.d.ts +8 -0
  19. package/dist/index.d.ts.map +1 -0
  20. package/dist/index.js +5 -0
  21. package/dist/index.js.map +1 -0
  22. package/dist/models.d.ts +6 -0
  23. package/dist/models.d.ts.map +1 -0
  24. package/dist/models.js +16480 -0
  25. package/dist/models.js.map +1 -0
  26. package/dist/providers/anthropic.d.ts +8 -0
  27. package/dist/providers/anthropic.d.ts.map +1 -0
  28. package/dist/providers/anthropic.js +279 -0
  29. package/dist/providers/anthropic.js.map +1 -0
  30. package/dist/providers/groq.d.ts +16 -0
  31. package/dist/providers/groq.d.ts.map +1 -0
  32. package/dist/providers/groq.js +196 -0
  33. package/dist/providers/groq.js.map +1 -0
  34. package/dist/providers/openai-completions.d.ts +17 -0
  35. package/dist/providers/openai-completions.d.ts.map +1 -0
  36. package/dist/providers/openai-completions.js +319 -0
  37. package/dist/providers/openai-completions.js.map +1 -0
  38. package/dist/providers/openai-responses.d.ts +12 -0
  39. package/dist/providers/openai-responses.d.ts.map +1 -0
  40. package/dist/providers/openai-responses.js +335 -0
  41. package/dist/providers/openai-responses.js.map +1 -0
  42. package/dist/providers/openrouter.d.ts +12 -0
  43. package/dist/providers/openrouter.d.ts.map +1 -0
  44. package/dist/providers/openrouter.js +42 -0
  45. package/dist/providers/openrouter.js.map +1 -0
  46. package/dist/providers/register-builtins.d.ts +2 -0
  47. package/dist/providers/register-builtins.d.ts.map +1 -0
  48. package/dist/providers/register-builtins.js +41 -0
  49. package/dist/providers/register-builtins.js.map +1 -0
  50. package/dist/providers/shared.d.ts +3 -0
  51. package/dist/providers/shared.d.ts.map +1 -0
  52. package/dist/providers/shared.js +12 -0
  53. package/dist/providers/shared.js.map +1 -0
  54. package/dist/registry.d.ts +19 -0
  55. package/dist/registry.d.ts.map +1 -0
  56. package/dist/registry.js +28 -0
  57. package/dist/registry.js.map +1 -0
  58. package/dist/types.d.ts +164 -0
  59. package/dist/types.d.ts.map +1 -0
  60. package/dist/types.js +2 -0
  61. package/dist/types.js.map +1 -0
  62. package/package.json +33 -0
@@ -0,0 +1,87 @@
1
+ import { describe, it, expect } from "vitest";
2
+ import { EventStream, AssistantMessageEventStream } from "../event-stream.js";
3
+ describe("EventStream", () => {
4
+ it("should queue events and deliver them in order", async () => {
5
+ const stream = new EventStream((e) => e === "end", (e) => e);
6
+ const events = [];
7
+ stream.push("hello");
8
+ stream.push("world");
9
+ stream.push("end");
10
+ for await (const event of stream) {
11
+ events.push(event);
12
+ }
13
+ expect(events).toEqual(["hello", "world", "end"]);
14
+ });
15
+ it("should resolve result on completion", async () => {
16
+ const stream = new EventStream((e) => e === "done", (e) => 42);
17
+ stream.push("some event");
18
+ stream.push("done");
19
+ stream.end();
20
+ const result = await stream.result();
21
+ expect(result).toBe(42);
22
+ });
23
+ it("should handle push after partial consumption", async () => {
24
+ const stream = new EventStream((e) => e === -1, (e) => e);
25
+ stream.push(1);
26
+ stream.push(2);
27
+ const iterator = stream[Symbol.asyncIterator]();
28
+ const first = await iterator.next();
29
+ expect(first.value).toBe(1);
30
+ expect(first.done).toBe(false);
31
+ stream.push(3);
32
+ const second = await iterator.next();
33
+ expect(second.value).toBe(2);
34
+ expect(second.done).toBe(false);
35
+ const third = await iterator.next();
36
+ expect(third.value).toBe(3);
37
+ expect(third.done).toBe(false);
38
+ });
39
+ });
40
+ describe("AssistantMessageEventStream", () => {
41
+ it("should resolve result on done event", async () => {
42
+ const stream = new AssistantMessageEventStream();
43
+ stream.push({ type: "start", partial: { role: "assistant", content: [], provider: "test", model: "test", usage: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, totalTokens: 0 }, stopReason: "stop", timestamp: Date.now() } });
44
+ stream.push({ type: "text_start", contentIndex: 0, partial: {} });
45
+ stream.push({ type: "text_delta", contentIndex: 0, delta: "Hello", partial: {} });
46
+ stream.push({
47
+ type: "done",
48
+ reason: "stop",
49
+ message: {
50
+ role: "assistant",
51
+ content: [{ type: "text", text: "Hello" }],
52
+ provider: "test",
53
+ model: "test",
54
+ usage: { input: 5, output: 10, cacheRead: 0, cacheWrite: 0, totalTokens: 15 },
55
+ stopReason: "stop",
56
+ timestamp: Date.now(),
57
+ },
58
+ });
59
+ const result = await stream.result();
60
+ expect(result.stopReason).toBe("stop");
61
+ expect(result.content[0]).toEqual({ type: "text", text: "Hello" });
62
+ });
63
+ it("should resolve result on error event", async () => {
64
+ const stream = new AssistantMessageEventStream();
65
+ stream.push({ type: "error", reason: "error", error: { role: "assistant", content: [], provider: "test", model: "test", usage: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, totalTokens: 0 }, stopReason: "error", errorMessage: "Something went wrong", timestamp: Date.now() } });
66
+ const result = await stream.result();
67
+ expect(result.stopReason).toBe("error");
68
+ expect(result.errorMessage).toBe("Something went wrong");
69
+ });
70
+ it("should iterate events in order", async () => {
71
+ const stream = new AssistantMessageEventStream();
72
+ const events = [];
73
+ stream.push({ type: "start", partial: { role: "assistant", content: [], provider: "test", model: "test", usage: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, totalTokens: 0 }, stopReason: "stop", timestamp: Date.now() } });
74
+ stream.push({ type: "text_start", contentIndex: 0, partial: {} });
75
+ stream.push({ type: "text_delta", contentIndex: 0, delta: "Hi", partial: {} });
76
+ stream.push({
77
+ type: "done",
78
+ reason: "stop",
79
+ message: { role: "assistant", content: [{ type: "text", text: "Hi" }], provider: "test", model: "test", usage: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, totalTokens: 0 }, stopReason: "stop", timestamp: Date.now() },
80
+ });
81
+ for await (const event of stream) {
82
+ events.push(event.type);
83
+ }
84
+ expect(events).toEqual(["start", "text_start", "text_delta", "done"]);
85
+ });
86
+ });
87
+ //# sourceMappingURL=event-stream.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"event-stream.test.js","sourceRoot":"","sources":["../../src/__tests__/event-stream.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAkB,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,WAAW,EAAE,2BAA2B,EAAE,MAAM,oBAAoB,CAAC;AAE9E,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,MAAM,GAAG,IAAI,WAAW,CAC5B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,KAAK,EAClB,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CACT,CAAC;QAEF,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEnB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrB,CAAC;QAED,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,MAAM,GAAG,IAAI,WAAW,CAC5B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,MAAM,EACnB,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CACV,CAAC;QAEF,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC1B,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpB,MAAM,CAAC,GAAG,EAAE,CAAC;QAEb,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC1B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,MAAM,GAAG,IAAI,WAAW,CAC5B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EACf,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CACT,CAAC;QAEF,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACf,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAEf,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;QAChD,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAE/B,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAEf,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC7B,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEhC,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,MAAM,GAAG,IAAI,2BAA2B,EAAE,CAAC;QAEjD,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;QACrO,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,CAAC,EAAE,OAAO,EAAE,EAAS,EAAE,CAAC,CAAC;QACzE,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,EAAS,EAAE,CAAC,CAAC;QACzF,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,MAAM;YACZ,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;gBAC1C,QAAQ,EAAE,MAAM;gBAChB,KAAK,EAAE,MAAM;gBACb,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE;gBAC7E,UAAU,EAAE,MAAM;gBAClB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,MAAM,GAAG,IAAI,2BAA2B,EAAE,CAAC;QAEjD,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,EAAE,UAAU,EAAE,OAAO,EAAE,YAAY,EAAE,sBAAsB,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;QAE3R,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,MAAM,GAAG,IAAI,2BAA2B,EAAE,CAAC;QACjD,MAAM,MAAM,GAAa,EAAE,CAAC;QAE5B,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;QACrO,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,CAAC,EAAE,OAAO,EAAE,EAAS,EAAE,CAAC,CAAC;QACzE,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,EAAS,EAAE,CAAC,CAAC;QACtF,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,MAAM;YACZ,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE;SACjO,CAAC,CAAC;QAEH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC1B,CAAC;QAED,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=provider.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"provider.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/provider.test.ts"],"names":[],"mappings":""}
@@ -0,0 +1,40 @@
1
+ import { describe, it, expect, beforeEach } from "vitest";
2
+ import { registerProvider, getProvider, listProviders } from "../registry.js";
3
+ import { createAnthropicProvider } from "../providers/anthropic.js";
4
+ describe("Provider Registry", () => {
5
+ beforeEach(() => {
6
+ registerProvider(createAnthropicProvider());
7
+ });
8
+ it("should register and retrieve provider", () => {
9
+ const provider = getProvider("anthropic");
10
+ expect(provider).toBeDefined();
11
+ expect(provider?.name).toBe("anthropic");
12
+ });
13
+ it("should list all registered providers", () => {
14
+ const providers = listProviders();
15
+ expect(providers).toContain("anthropic");
16
+ });
17
+ it("should return undefined for unknown provider", () => {
18
+ const provider = getProvider("unknown");
19
+ expect(provider).toBeUndefined();
20
+ });
21
+ });
22
+ describe("Anthropic Provider", () => {
23
+ it("should create provider with correct name", () => {
24
+ const provider = createAnthropicProvider();
25
+ expect(provider.name).toBe("anthropic");
26
+ });
27
+ it("should create stream that emits error without API key", async () => {
28
+ const provider = createAnthropicProvider();
29
+ const model = { id: "claude-sonnet-4-20250514", name: "Claude Sonnet 4", provider: "anthropic", api: "anthropic-messages" };
30
+ const context = { messages: [{ role: "user", content: "Hello", timestamp: Date.now() }] };
31
+ const stream = provider.stream(model, context);
32
+ const events = [];
33
+ for await (const event of stream) {
34
+ events.push(event.type);
35
+ }
36
+ expect(events).toContain("start");
37
+ expect(events).toContain("error");
38
+ });
39
+ });
40
+ //# sourceMappingURL=provider.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"provider.test.js","sourceRoot":"","sources":["../../src/__tests__/provider.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAM,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,gBAAgB,EAAE,WAAW,EAAE,aAAa,EAAU,MAAM,gBAAgB,CAAC;AACtF,OAAO,EAAE,uBAAuB,EAAE,MAAM,2BAA2B,CAAC;AAGpE,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,UAAU,CAAC,GAAG,EAAE;QACd,gBAAgB,CAAC,uBAAuB,EAAE,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,QAAQ,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;QAC/B,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,SAAS,GAAG,aAAa,EAAE,CAAC;QAClC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,QAAQ,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC;QACxC,MAAM,CAAC,QAAQ,CAAC,CAAC,aAAa,EAAE,CAAC;IACnC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,QAAQ,GAAG,uBAAuB,EAAE,CAAC;QAC3C,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,QAAQ,GAAG,uBAAuB,EAAE,CAAC;QAC3C,MAAM,KAAK,GAAU,EAAE,EAAE,EAAE,0BAA0B,EAAE,IAAI,EAAE,iBAAiB,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,EAAE,oBAAoB,EAAE,CAAC;QACnI,MAAM,OAAO,GAAY,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC;QAEnG,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAE/C,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC1B,CAAC;QAED,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -0,0 +1,19 @@
1
+ import type { AssistantMessage, AssistantMessageEvent } from "./types.js";
2
+ export declare class EventStream<T, R = T> implements AsyncIterable<T> {
3
+ private isComplete;
4
+ private extractResult;
5
+ private queue;
6
+ private waiting;
7
+ private done;
8
+ private finalResultPromise;
9
+ private resolveFinalResult;
10
+ constructor(isComplete: (event: T) => boolean, extractResult: (event: T) => R);
11
+ push(event: T): void;
12
+ end(result?: R): void;
13
+ [Symbol.asyncIterator](): AsyncIterator<T>;
14
+ result(): Promise<R>;
15
+ }
16
+ export declare class AssistantMessageEventStream extends EventStream<AssistantMessageEvent, AssistantMessage> {
17
+ constructor();
18
+ }
19
+ //# sourceMappingURL=event-stream.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"event-stream.d.ts","sourceRoot":"","sources":["../src/event-stream.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,qBAAqB,EAAE,MAAM,YAAY,CAAC;AAE1E,qBAAa,WAAW,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAE,YAAW,aAAa,CAAC,CAAC,CAAC;IAQ1D,OAAO,CAAC,UAAU;IAClB,OAAO,CAAC,aAAa;IARvB,OAAO,CAAC,KAAK,CAAW;IACxB,OAAO,CAAC,OAAO,CAA+C;IAC9D,OAAO,CAAC,IAAI,CAAS;IACrB,OAAO,CAAC,kBAAkB,CAAa;IACvC,OAAO,CAAC,kBAAkB,CAAuB;gBAGvC,UAAU,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,OAAO,EACjC,aAAa,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC;IAOxC,IAAI,CAAC,KAAK,EAAE,CAAC,GAAG,IAAI;IAgBpB,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,IAAI;IAWd,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,aAAa,CAAC,CAAC,CAAC;IAgBjD,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC;CAGrB;AAED,qBAAa,2BAA4B,SAAQ,WAAW,CAC1D,qBAAqB,EACrB,gBAAgB,CACjB;;CAcA"}
@@ -0,0 +1,74 @@
1
+ export class EventStream {
2
+ isComplete;
3
+ extractResult;
4
+ queue = [];
5
+ waiting = [];
6
+ done = false;
7
+ finalResultPromise;
8
+ resolveFinalResult;
9
+ constructor(isComplete, extractResult) {
10
+ this.isComplete = isComplete;
11
+ this.extractResult = extractResult;
12
+ this.finalResultPromise = new Promise((resolve) => {
13
+ this.resolveFinalResult = resolve;
14
+ });
15
+ }
16
+ push(event) {
17
+ if (this.done)
18
+ return;
19
+ if (this.isComplete(event)) {
20
+ this.done = true;
21
+ this.resolveFinalResult(this.extractResult(event));
22
+ }
23
+ const waiter = this.waiting.shift();
24
+ if (waiter) {
25
+ waiter({ value: event, done: false });
26
+ }
27
+ else {
28
+ this.queue.push(event);
29
+ }
30
+ }
31
+ end(result) {
32
+ this.done = true;
33
+ if (result !== undefined) {
34
+ this.resolveFinalResult(result);
35
+ }
36
+ while (this.waiting.length > 0) {
37
+ const waiter = this.waiting.shift();
38
+ waiter({ value: undefined, done: true });
39
+ }
40
+ }
41
+ async *[Symbol.asyncIterator]() {
42
+ while (true) {
43
+ if (this.queue.length > 0) {
44
+ yield this.queue.shift();
45
+ }
46
+ else if (this.done) {
47
+ return;
48
+ }
49
+ else {
50
+ const result = await new Promise((resolve) => this.waiting.push(resolve));
51
+ if (result.done)
52
+ return;
53
+ yield result.value;
54
+ }
55
+ }
56
+ }
57
+ result() {
58
+ return this.finalResultPromise;
59
+ }
60
+ }
61
+ export class AssistantMessageEventStream extends EventStream {
62
+ constructor() {
63
+ super((event) => event.type === "done" || event.type === "error", (event) => {
64
+ if (event.type === "done") {
65
+ return event.message;
66
+ }
67
+ else if (event.type === "error") {
68
+ return event.error;
69
+ }
70
+ throw new Error("Unexpected event type for final result");
71
+ });
72
+ }
73
+ }
74
+ //# sourceMappingURL=event-stream.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"event-stream.js","sourceRoot":"","sources":["../src/event-stream.ts"],"names":[],"mappings":"AAEA,MAAM,OAAO,WAAW;IAQZ;IACA;IARF,KAAK,GAAQ,EAAE,CAAC;IAChB,OAAO,GAA4C,EAAE,CAAC;IACtD,IAAI,GAAG,KAAK,CAAC;IACb,kBAAkB,CAAa;IAC/B,kBAAkB,CAAuB;IAEjD,YACU,UAAiC,EACjC,aAA8B;QAD9B,eAAU,GAAV,UAAU,CAAuB;QACjC,kBAAa,GAAb,aAAa,CAAiB;QAEtC,IAAI,CAAC,kBAAkB,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAChD,IAAI,CAAC,kBAAkB,GAAG,OAAO,CAAC;QACpC,CAAC,CAAC,CAAC;IACL,CAAC;IAED,IAAI,CAAC,KAAQ;QACX,IAAI,IAAI,CAAC,IAAI;YAAE,OAAO;QAEtB,IAAI,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;YACjB,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;QACrD,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACpC,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QACxC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAED,GAAG,CAAC,MAAU;QACZ,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACzB,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/B,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAG,CAAC;YACrC,MAAM,CAAC,EAAE,KAAK,EAAE,SAAyB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC;QAC3B,OAAO,IAAI,EAAE,CAAC;YACZ,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1B,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,EAAG,CAAC;YAC5B,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACrB,OAAO;YACT,CAAC;iBAAM,CAAC;gBACN,MAAM,MAAM,GAAG,MAAM,IAAI,OAAO,CAC9B,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CACxC,CAAC;gBACF,IAAI,MAAM,CAAC,IAAI;oBAAE,OAAO;gBACxB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACjC,CAAC;CACF;AAED,MAAM,OAAO,2BAA4B,SAAQ,WAGhD;IACC;QACE,KAAK,CACH,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAC1D,CAAC,KAAK,EAAE,EAAE;YACR,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC1B,OAAO,KAAK,CAAC,OAAO,CAAC;YACvB,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBAClC,OAAO,KAAK,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC5D,CAAC,CACF,CAAC;IACJ,CAAC;CACF"}
@@ -0,0 +1,8 @@
1
+ export { EventStream, AssistantMessageEventStream } from "./event-stream.js";
2
+ export { stream, complete, registerProvider, getProvider, listProviders, getModels } from "./registry.js";
3
+ export type { StreamFunction, Provider } from "./registry.js";
4
+ export { initProviders } from "./providers/register-builtins.js";
5
+ export { getProviderModels } from "./models.js";
6
+ export type { ModelEntry } from "./models.js";
7
+ export type { TextContent, ThinkingContent, ImageContent, ToolCall, StopReason, Usage, UserMessage, AssistantMessage, ToolResultMessage, Message, Tool, Context, AssistantMessageEvent, StreamOptions, Model, } from "./types.js";
8
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,2BAA2B,EAAE,MAAM,mBAAmB,CAAC;AAC7E,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,gBAAgB,EAAE,WAAW,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAC1G,YAAY,EAAE,cAAc,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAC;AACjE,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAChD,YAAY,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAC9C,YAAY,EACV,WAAW,EACX,eAAe,EACf,YAAY,EACZ,QAAQ,EACR,UAAU,EACV,KAAK,EACL,WAAW,EACX,gBAAgB,EAChB,iBAAiB,EACjB,OAAO,EACP,IAAI,EACJ,OAAO,EACP,qBAAqB,EACrB,aAAa,EACb,KAAK,GACN,MAAM,YAAY,CAAC"}
package/dist/index.js ADDED
@@ -0,0 +1,5 @@
1
+ export { EventStream, AssistantMessageEventStream } from "./event-stream.js";
2
+ export { stream, complete, registerProvider, getProvider, listProviders, getModels } from "./registry.js";
3
+ export { initProviders } from "./providers/register-builtins.js";
4
+ export { getProviderModels } from "./models.js";
5
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,2BAA2B,EAAE,MAAM,mBAAmB,CAAC;AAC7E,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,gBAAgB,EAAE,WAAW,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAE1G,OAAO,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAC;AACjE,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC"}
@@ -0,0 +1,6 @@
1
+ export interface ModelEntry {
2
+ id: string;
3
+ name: string;
4
+ }
5
+ export declare function getProviderModels(providerId: string): ModelEntry[];
6
+ //# sourceMappingURL=models.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../src/models.ts"],"names":[],"mappings":"AAGA,MAAM,WAAW,UAAU;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;CACd;AA6lgBD,wBAAgB,iBAAiB,CAAC,UAAU,EAAE,MAAM,GAAG,UAAU,EAAE,CAElE"}