@xpert-ai/xpert-sdk 0.0.1

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 (122) hide show
  1. package/LICENSE +21 -0
  2. package/dist/auth/error.d.ts +9 -0
  3. package/dist/auth/error.js +76 -0
  4. package/dist/auth/error.js.map +1 -0
  5. package/dist/auth/error.mjs +76 -0
  6. package/dist/auth/error.mjs.map +1 -0
  7. package/dist/auth/index.d.ts +18 -0
  8. package/dist/auth/index.js +35 -0
  9. package/dist/auth/index.js.map +1 -0
  10. package/dist/auth/index.mjs +35 -0
  11. package/dist/auth/index.mjs.map +1 -0
  12. package/dist/auth/types.d.ts +297 -0
  13. package/dist/auth/types.js +2 -0
  14. package/dist/auth/types.js.map +1 -0
  15. package/dist/auth/types.mjs +2 -0
  16. package/dist/auth/types.mjs.map +1 -0
  17. package/dist/client.d.ts +662 -0
  18. package/dist/client.js +1233 -0
  19. package/dist/client.js.map +1 -0
  20. package/dist/client.mjs +1233 -0
  21. package/dist/client.mjs.map +1 -0
  22. package/dist/index.d.ts +7 -0
  23. package/dist/index.js +3 -0
  24. package/dist/index.js.map +1 -0
  25. package/dist/index.mjs +3 -0
  26. package/dist/index.mjs.map +1 -0
  27. package/dist/logging/index.d.ts +43 -0
  28. package/dist/logging/index.js +33 -0
  29. package/dist/logging/index.js.map +1 -0
  30. package/dist/logging/index.mjs +33 -0
  31. package/dist/logging/index.mjs.map +1 -0
  32. package/dist/schema.d.ts +270 -0
  33. package/dist/schema.js +2 -0
  34. package/dist/schema.js.map +1 -0
  35. package/dist/schema.mjs +2 -0
  36. package/dist/schema.mjs.map +1 -0
  37. package/dist/singletons/fetch.d.ts +7 -0
  38. package/dist/singletons/fetch.js +24 -0
  39. package/dist/singletons/fetch.js.map +1 -0
  40. package/dist/singletons/fetch.mjs +24 -0
  41. package/dist/singletons/fetch.mjs.map +1 -0
  42. package/dist/tests/fetch.test.d.ts +1 -0
  43. package/dist/tests/fetch.test.js +154 -0
  44. package/dist/tests/fetch.test.js.map +1 -0
  45. package/dist/tests/fetch.test.mjs +154 -0
  46. package/dist/tests/fetch.test.mjs.map +1 -0
  47. package/dist/tests/sse.test.d.ts +1 -0
  48. package/dist/tests/sse.test.js +157 -0
  49. package/dist/tests/sse.test.js.map +1 -0
  50. package/dist/tests/sse.test.mjs +157 -0
  51. package/dist/tests/sse.test.mjs.map +1 -0
  52. package/dist/types.d.ts +176 -0
  53. package/dist/types.js +2 -0
  54. package/dist/types.js.map +1 -0
  55. package/dist/types.messages.d.ts +86 -0
  56. package/dist/types.messages.js +2 -0
  57. package/dist/types.messages.js.map +1 -0
  58. package/dist/types.messages.mjs +2 -0
  59. package/dist/types.messages.mjs.map +1 -0
  60. package/dist/types.mjs +2 -0
  61. package/dist/types.mjs.map +1 -0
  62. package/dist/types.stream.d.ts +212 -0
  63. package/dist/types.stream.js +2 -0
  64. package/dist/types.stream.js.map +1 -0
  65. package/dist/types.stream.mjs +2 -0
  66. package/dist/types.stream.mjs.map +1 -0
  67. package/dist/ui/branching.d.ts +35 -0
  68. package/dist/ui/branching.js +145 -0
  69. package/dist/ui/branching.js.map +1 -0
  70. package/dist/ui/branching.mjs +145 -0
  71. package/dist/ui/branching.mjs.map +1 -0
  72. package/dist/ui/branching.test.d.ts +1 -0
  73. package/dist/ui/branching.test.js +369 -0
  74. package/dist/ui/branching.test.js.map +1 -0
  75. package/dist/ui/branching.test.mjs +369 -0
  76. package/dist/ui/branching.test.mjs.map +1 -0
  77. package/dist/ui/errors.d.ts +12 -0
  78. package/dist/ui/errors.js +10 -0
  79. package/dist/ui/errors.js.map +1 -0
  80. package/dist/ui/errors.mjs +10 -0
  81. package/dist/ui/errors.mjs.map +1 -0
  82. package/dist/ui/manager.d.ts +87 -0
  83. package/dist/ui/manager.js +160 -0
  84. package/dist/ui/manager.js.map +1 -0
  85. package/dist/ui/manager.mjs +160 -0
  86. package/dist/ui/manager.mjs.map +1 -0
  87. package/dist/ui/messages.d.ts +18 -0
  88. package/dist/ui/messages.js +67 -0
  89. package/dist/ui/messages.js.map +1 -0
  90. package/dist/ui/messages.mjs +67 -0
  91. package/dist/ui/messages.mjs.map +1 -0
  92. package/dist/ui/utils.d.ts +2 -0
  93. package/dist/ui/utils.js +11 -0
  94. package/dist/ui/utils.js.map +1 -0
  95. package/dist/ui/utils.mjs +11 -0
  96. package/dist/ui/utils.mjs.map +1 -0
  97. package/dist/utils/async_caller.d.ts +48 -0
  98. package/dist/utils/async_caller.js +147 -0
  99. package/dist/utils/async_caller.js.map +1 -0
  100. package/dist/utils/async_caller.mjs +147 -0
  101. package/dist/utils/async_caller.mjs.map +1 -0
  102. package/dist/utils/env.d.ts +1 -0
  103. package/dist/utils/env.js +13 -0
  104. package/dist/utils/env.js.map +1 -0
  105. package/dist/utils/env.mjs +13 -0
  106. package/dist/utils/env.mjs.map +1 -0
  107. package/dist/utils/signals.d.ts +1 -0
  108. package/dist/utils/signals.js +19 -0
  109. package/dist/utils/signals.js.map +1 -0
  110. package/dist/utils/signals.mjs +19 -0
  111. package/dist/utils/signals.mjs.map +1 -0
  112. package/dist/utils/sse.d.ts +8 -0
  113. package/dist/utils/sse.js +151 -0
  114. package/dist/utils/sse.js.map +1 -0
  115. package/dist/utils/sse.mjs +151 -0
  116. package/dist/utils/sse.mjs.map +1 -0
  117. package/dist/utils/stream.d.ts +13 -0
  118. package/dist/utils/stream.js +109 -0
  119. package/dist/utils/stream.js.map +1 -0
  120. package/dist/utils/stream.mjs +109 -0
  121. package/dist/utils/stream.mjs.map +1 -0
  122. package/package.json +56 -0
@@ -0,0 +1,154 @@
1
+ import { describe, it, expect, beforeEach, afterEach, vi } from "vitest";
2
+ import { Client } from "../client.js";
3
+ import { overrideFetchImplementation } from "../singletons/fetch.js";
4
+ describe.each([["global"], ["mocked"]])("Client uses %s fetch", (description) => {
5
+ let globalFetchMock;
6
+ let overriddenFetch;
7
+ let expectedFetchMock;
8
+ let unexpectedFetchMock;
9
+ beforeEach(() => {
10
+ globalFetchMock = vi.fn(() => Promise.resolve({
11
+ ok: true,
12
+ json: () => Promise.resolve({
13
+ batch_ingest_config: {
14
+ use_multipart_endpoint: true,
15
+ },
16
+ }),
17
+ text: () => Promise.resolve(""),
18
+ headers: new Headers({}),
19
+ }));
20
+ overriddenFetch = vi.fn(() => Promise.resolve({
21
+ ok: true,
22
+ json: () => Promise.resolve({
23
+ batch_ingest_config: {
24
+ use_multipart_endpoint: true,
25
+ },
26
+ }),
27
+ text: () => Promise.resolve(""),
28
+ headers: new Headers({}),
29
+ }));
30
+ expectedFetchMock =
31
+ description === "mocked" ? overriddenFetch : globalFetchMock;
32
+ unexpectedFetchMock =
33
+ description === "mocked" ? globalFetchMock : overriddenFetch;
34
+ if (description === "mocked") {
35
+ overrideFetchImplementation(overriddenFetch);
36
+ }
37
+ else {
38
+ overrideFetchImplementation(globalFetchMock);
39
+ }
40
+ // Mock global fetch
41
+ globalThis.fetch = globalFetchMock;
42
+ });
43
+ afterEach(() => {
44
+ vi.restoreAllMocks();
45
+ });
46
+ describe("createRuns", () => {
47
+ it("should create an example with the given input and generation", async () => {
48
+ const client = new Client({ apiKey: "test-api-key" });
49
+ const thread = await client.threads.create();
50
+ expect(expectedFetchMock).toHaveBeenCalledTimes(1);
51
+ expect(unexpectedFetchMock).not.toHaveBeenCalled();
52
+ vi.clearAllMocks(); // Clear all mocks before the next operation
53
+ // Then clear & run the function
54
+ await client.runs.create(thread.thread_id, "somegraph", {
55
+ input: { foo: "bar" },
56
+ });
57
+ expect(expectedFetchMock).toHaveBeenCalledTimes(1);
58
+ expect(unexpectedFetchMock).not.toHaveBeenCalled();
59
+ });
60
+ });
61
+ describe("header coalescing", () => {
62
+ it("should properly merge headers with conflicting name casing", async () => {
63
+ const client = new Client({ apiKey: "test-api-key" });
64
+ await client.threads.fetch("/test", {
65
+ headers: { "X-Api-Key": "custom-value" },
66
+ });
67
+ expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
68
+ headers: expect.objectContaining({
69
+ "x-api-key": "custom-value",
70
+ }),
71
+ }));
72
+ });
73
+ it("should properly merge headers from multiple sources", async () => {
74
+ const client = new Client({
75
+ apiKey: "test-api-key",
76
+ defaultHeaders: {
77
+ "x-default": "default-value",
78
+ "x-override": "default-value",
79
+ },
80
+ });
81
+ await client.threads.fetch("/test", {
82
+ headers: {
83
+ "x-custom": "custom-value",
84
+ "x-override": "custom-value",
85
+ },
86
+ });
87
+ expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
88
+ headers: expect.objectContaining({
89
+ "x-api-key": "test-api-key",
90
+ "x-default": "default-value",
91
+ "x-custom": "custom-value",
92
+ "x-override": "custom-value",
93
+ }),
94
+ }));
95
+ vi.clearAllMocks();
96
+ // Test with null/undefined values
97
+ await client.threads.fetch("/test", {
98
+ headers: {
99
+ "x-null": null,
100
+ "x-undefined": undefined,
101
+ "x-empty": "",
102
+ },
103
+ });
104
+ expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
105
+ headers: expect.objectContaining({
106
+ "x-api-key": "test-api-key",
107
+ "x-default": "default-value",
108
+ }),
109
+ }));
110
+ expect(expectedFetchMock).not.toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
111
+ headers: expect.objectContaining({
112
+ "x-null": null,
113
+ "x-undefined": undefined,
114
+ }),
115
+ }));
116
+ });
117
+ it("should handle Headers object input", async () => {
118
+ const client = new Client({ apiKey: "test-api-key" });
119
+ const headers = new Headers();
120
+ headers.append("x-custom", "custom-value");
121
+ headers.append("x-multi", "value1");
122
+ headers.append("x-multi", "value2");
123
+ await client.threads.fetch("/test", { headers });
124
+ expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
125
+ headers: expect.objectContaining({
126
+ "x-api-key": "test-api-key",
127
+ "x-custom": "custom-value",
128
+ "x-multi": "value1, value2",
129
+ }),
130
+ }));
131
+ });
132
+ it("should handle array of header tuples", async () => {
133
+ const client = new Client({
134
+ apiKey: "test-api-key",
135
+ defaultHeaders: {
136
+ "x-custom": "custom-value",
137
+ },
138
+ });
139
+ const headers = [
140
+ ["x-multi", "value1"],
141
+ ["x-multi", "value2"],
142
+ ];
143
+ await client.threads.fetch("/test", { headers });
144
+ expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
145
+ headers: expect.objectContaining({
146
+ "x-api-key": "test-api-key",
147
+ "x-custom": "custom-value",
148
+ "x-multi": "value1, value2",
149
+ }),
150
+ }));
151
+ });
152
+ });
153
+ });
154
+ //# sourceMappingURL=fetch.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"fetch.test.js","sourceRoot":"","sources":["../../src/tests/fetch.test.ts"],"names":[],"mappings":"OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,QAAQ;OACjE,EAAE,MAAM,EAAE;OACV,EAAE,2BAA2B,EAAE;AAEtC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CACrC,sBAAsB,EACtB,CAAC,WAAmB,EAAE,EAAE;IACtB,IAAI,eAAyC,CAAC;IAC9C,IAAI,eAAyC,CAAC;IAE9C,IAAI,iBAA2C,CAAC;IAChD,IAAI,mBAA6C,CAAC;IAElD,UAAU,CAAC,GAAG,EAAE;QACd,eAAe,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAC3B,OAAO,CAAC,OAAO,CAAC;YACd,EAAE,EAAE,IAAI;YACR,IAAI,EAAE,GAAG,EAAE,CACT,OAAO,CAAC,OAAO,CAAC;gBACd,mBAAmB,EAAE;oBACnB,sBAAsB,EAAE,IAAI;iBAC7B;aACF,CAAC;YACJ,IAAI,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC/B,OAAO,EAAE,IAAI,OAAO,CAAC,EAAE,CAAC;SACzB,CAAC,CACH,CAAC;QACF,eAAe,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAC3B,OAAO,CAAC,OAAO,CAAC;YACd,EAAE,EAAE,IAAI;YACR,IAAI,EAAE,GAAG,EAAE,CACT,OAAO,CAAC,OAAO,CAAC;gBACd,mBAAmB,EAAE;oBACnB,sBAAsB,EAAE,IAAI;iBAC7B;aACF,CAAC;YACJ,IAAI,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC/B,OAAO,EAAE,IAAI,OAAO,CAAC,EAAE,CAAC;SACzB,CAAC,CACH,CAAC;QACF,iBAAiB;YACf,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;QAC/D,mBAAmB;YACjB,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;QAE/D,IAAI,WAAW,KAAK,QAAQ,EAAE,CAAC;YAC7B,2BAA2B,CAAC,eAAe,CAAC,CAAC;QAC/C,CAAC;aAAM,CAAC;YACN,2BAA2B,CAAC,eAAe,CAAC,CAAC;QAC/C,CAAC;QACD,oBAAoB;QACnB,UAAkB,CAAC,KAAK,GAAG,eAAe,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;YAEtD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YAC7C,MAAM,CAAC,iBAAiB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,mBAAmB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAEnD,EAAE,CAAC,aAAa,EAAE,CAAC,CAAC,4CAA4C;YAEhE,gCAAgC;YAChC,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,WAAW,EAAE;gBACtD,KAAK,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE;aACtB,CAAC,CAAC;YACH,MAAM,CAAC,iBAAiB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,mBAAmB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;YACtD,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE;gBAC3C,OAAO,EAAE,EAAE,WAAW,EAAE,cAAc,EAAE;aACzC,CAAC,CAAC;YACH,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;iBAC5B,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC;gBACxB,MAAM,EAAE,cAAc;gBACtB,cAAc,EAAE;oBACd,WAAW,EAAE,eAAe;oBAC5B,YAAY,EAAE,eAAe;iBAC9B;aACF,CAAC,CAAC;YAEH,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE;gBAC3C,OAAO,EAAE;oBACP,UAAU,EAAE,cAAc;oBAC1B,YAAY,EAAE,cAAc;iBAC7B;aACF,CAAC,CAAC;YAEH,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;oBAC3B,WAAW,EAAE,eAAe;oBAC5B,UAAU,EAAE,cAAc;oBAC1B,YAAY,EAAE,cAAc;iBAC7B,CAAC;aACH,CAAC,CACH,CAAC;YAEF,EAAE,CAAC,aAAa,EAAE,CAAC;YAEnB,kCAAkC;YAClC,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE;gBAC3C,OAAO,EAAE;oBACP,QAAQ,EAAE,IAAI;oBACd,aAAa,EAAE,SAAS;oBACxB,SAAS,EAAE,EAAE;iBACd;aACF,CAAC,CAAC;YAEH,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;oBAC3B,WAAW,EAAE,eAAe;iBAC7B,CAAC;aACH,CAAC,CACH,CAAC;YACF,MAAM,CAAC,iBAAiB,CAAC,CAAC,GAAG,CAAC,oBAAoB,CAChD,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,QAAQ,EAAE,IAAI;oBACd,aAAa,EAAE,SAAS;iBACzB,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;YACtD,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;YAC9B,OAAO,CAAC,MAAM,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;YAC3C,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YACpC,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAEpC,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAE1D,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;oBAC3B,UAAU,EAAE,cAAc;oBAC1B,SAAS,EAAE,gBAAgB;iBAC5B,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC;gBACxB,MAAM,EAAE,cAAc;gBACtB,cAAc,EAAE;oBACd,UAAU,EAAE,cAAc;iBAC3B;aACF,CAAC,CAAC;YACH,MAAM,OAAO,GAAG;gBACd,CAAC,SAAS,EAAE,QAAQ,CAAC;gBACrB,CAAC,SAAS,EAAE,QAAQ,CAAC;aACtB,CAAC;YAEF,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAE1D,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;oBAC3B,UAAU,EAAE,cAAc;oBAC1B,SAAS,EAAE,gBAAgB;iBAC5B,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CACF,CAAC"}
@@ -0,0 +1,154 @@
1
+ import { describe, it, expect, beforeEach, afterEach, vi } from "vitest";
2
+ import { Client } from "../client.mjs";
3
+ import { overrideFetchImplementation } from "../singletons/fetch.mjs";
4
+ describe.each([["global"], ["mocked"]])("Client uses %s fetch", (description) => {
5
+ let globalFetchMock;
6
+ let overriddenFetch;
7
+ let expectedFetchMock;
8
+ let unexpectedFetchMock;
9
+ beforeEach(() => {
10
+ globalFetchMock = vi.fn(() => Promise.resolve({
11
+ ok: true,
12
+ json: () => Promise.resolve({
13
+ batch_ingest_config: {
14
+ use_multipart_endpoint: true,
15
+ },
16
+ }),
17
+ text: () => Promise.resolve(""),
18
+ headers: new Headers({}),
19
+ }));
20
+ overriddenFetch = vi.fn(() => Promise.resolve({
21
+ ok: true,
22
+ json: () => Promise.resolve({
23
+ batch_ingest_config: {
24
+ use_multipart_endpoint: true,
25
+ },
26
+ }),
27
+ text: () => Promise.resolve(""),
28
+ headers: new Headers({}),
29
+ }));
30
+ expectedFetchMock =
31
+ description === "mocked" ? overriddenFetch : globalFetchMock;
32
+ unexpectedFetchMock =
33
+ description === "mocked" ? globalFetchMock : overriddenFetch;
34
+ if (description === "mocked") {
35
+ overrideFetchImplementation(overriddenFetch);
36
+ }
37
+ else {
38
+ overrideFetchImplementation(globalFetchMock);
39
+ }
40
+ // Mock global fetch
41
+ globalThis.fetch = globalFetchMock;
42
+ });
43
+ afterEach(() => {
44
+ vi.restoreAllMocks();
45
+ });
46
+ describe("createRuns", () => {
47
+ it("should create an example with the given input and generation", async () => {
48
+ const client = new Client({ apiKey: "test-api-key" });
49
+ const thread = await client.threads.create();
50
+ expect(expectedFetchMock).toHaveBeenCalledTimes(1);
51
+ expect(unexpectedFetchMock).not.toHaveBeenCalled();
52
+ vi.clearAllMocks(); // Clear all mocks before the next operation
53
+ // Then clear & run the function
54
+ await client.runs.create(thread.thread_id, "somegraph", {
55
+ input: { foo: "bar" },
56
+ });
57
+ expect(expectedFetchMock).toHaveBeenCalledTimes(1);
58
+ expect(unexpectedFetchMock).not.toHaveBeenCalled();
59
+ });
60
+ });
61
+ describe("header coalescing", () => {
62
+ it("should properly merge headers with conflicting name casing", async () => {
63
+ const client = new Client({ apiKey: "test-api-key" });
64
+ await client.threads.fetch("/test", {
65
+ headers: { "X-Api-Key": "custom-value" },
66
+ });
67
+ expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
68
+ headers: expect.objectContaining({
69
+ "x-api-key": "custom-value",
70
+ }),
71
+ }));
72
+ });
73
+ it("should properly merge headers from multiple sources", async () => {
74
+ const client = new Client({
75
+ apiKey: "test-api-key",
76
+ defaultHeaders: {
77
+ "x-default": "default-value",
78
+ "x-override": "default-value",
79
+ },
80
+ });
81
+ await client.threads.fetch("/test", {
82
+ headers: {
83
+ "x-custom": "custom-value",
84
+ "x-override": "custom-value",
85
+ },
86
+ });
87
+ expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
88
+ headers: expect.objectContaining({
89
+ "x-api-key": "test-api-key",
90
+ "x-default": "default-value",
91
+ "x-custom": "custom-value",
92
+ "x-override": "custom-value",
93
+ }),
94
+ }));
95
+ vi.clearAllMocks();
96
+ // Test with null/undefined values
97
+ await client.threads.fetch("/test", {
98
+ headers: {
99
+ "x-null": null,
100
+ "x-undefined": undefined,
101
+ "x-empty": "",
102
+ },
103
+ });
104
+ expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
105
+ headers: expect.objectContaining({
106
+ "x-api-key": "test-api-key",
107
+ "x-default": "default-value",
108
+ }),
109
+ }));
110
+ expect(expectedFetchMock).not.toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
111
+ headers: expect.objectContaining({
112
+ "x-null": null,
113
+ "x-undefined": undefined,
114
+ }),
115
+ }));
116
+ });
117
+ it("should handle Headers object input", async () => {
118
+ const client = new Client({ apiKey: "test-api-key" });
119
+ const headers = new Headers();
120
+ headers.append("x-custom", "custom-value");
121
+ headers.append("x-multi", "value1");
122
+ headers.append("x-multi", "value2");
123
+ await client.threads.fetch("/test", { headers });
124
+ expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
125
+ headers: expect.objectContaining({
126
+ "x-api-key": "test-api-key",
127
+ "x-custom": "custom-value",
128
+ "x-multi": "value1, value2",
129
+ }),
130
+ }));
131
+ });
132
+ it("should handle array of header tuples", async () => {
133
+ const client = new Client({
134
+ apiKey: "test-api-key",
135
+ defaultHeaders: {
136
+ "x-custom": "custom-value",
137
+ },
138
+ });
139
+ const headers = [
140
+ ["x-multi", "value1"],
141
+ ["x-multi", "value2"],
142
+ ];
143
+ await client.threads.fetch("/test", { headers });
144
+ expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
145
+ headers: expect.objectContaining({
146
+ "x-api-key": "test-api-key",
147
+ "x-custom": "custom-value",
148
+ "x-multi": "value1, value2",
149
+ }),
150
+ }));
151
+ });
152
+ });
153
+ });
154
+ //# sourceMappingURL=fetch.test.mjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"fetch.test.mjs","sourceRoot":"","sources":["../../src/tests/fetch.test.ts"],"names":[],"mappings":"OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,QAAQ;OACjE,EAAE,MAAM,EAAE;OACV,EAAE,2BAA2B,EAAE;AAEtC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CACrC,sBAAsB,EACtB,CAAC,WAAmB,EAAE,EAAE;IACtB,IAAI,eAAyC,CAAC;IAC9C,IAAI,eAAyC,CAAC;IAE9C,IAAI,iBAA2C,CAAC;IAChD,IAAI,mBAA6C,CAAC;IAElD,UAAU,CAAC,GAAG,EAAE;QACd,eAAe,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAC3B,OAAO,CAAC,OAAO,CAAC;YACd,EAAE,EAAE,IAAI;YACR,IAAI,EAAE,GAAG,EAAE,CACT,OAAO,CAAC,OAAO,CAAC;gBACd,mBAAmB,EAAE;oBACnB,sBAAsB,EAAE,IAAI;iBAC7B;aACF,CAAC;YACJ,IAAI,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC/B,OAAO,EAAE,IAAI,OAAO,CAAC,EAAE,CAAC;SACzB,CAAC,CACH,CAAC;QACF,eAAe,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAC3B,OAAO,CAAC,OAAO,CAAC;YACd,EAAE,EAAE,IAAI;YACR,IAAI,EAAE,GAAG,EAAE,CACT,OAAO,CAAC,OAAO,CAAC;gBACd,mBAAmB,EAAE;oBACnB,sBAAsB,EAAE,IAAI;iBAC7B;aACF,CAAC;YACJ,IAAI,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC/B,OAAO,EAAE,IAAI,OAAO,CAAC,EAAE,CAAC;SACzB,CAAC,CACH,CAAC;QACF,iBAAiB;YACf,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;QAC/D,mBAAmB;YACjB,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;QAE/D,IAAI,WAAW,KAAK,QAAQ,EAAE,CAAC;YAC7B,2BAA2B,CAAC,eAAe,CAAC,CAAC;QAC/C,CAAC;aAAM,CAAC;YACN,2BAA2B,CAAC,eAAe,CAAC,CAAC;QAC/C,CAAC;QACD,oBAAoB;QACnB,UAAkB,CAAC,KAAK,GAAG,eAAe,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;YAEtD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YAC7C,MAAM,CAAC,iBAAiB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,mBAAmB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAEnD,EAAE,CAAC,aAAa,EAAE,CAAC,CAAC,4CAA4C;YAEhE,gCAAgC;YAChC,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,WAAW,EAAE;gBACtD,KAAK,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE;aACtB,CAAC,CAAC;YACH,MAAM,CAAC,iBAAiB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,mBAAmB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;YACtD,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE;gBAC3C,OAAO,EAAE,EAAE,WAAW,EAAE,cAAc,EAAE;aACzC,CAAC,CAAC;YACH,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;iBAC5B,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC;gBACxB,MAAM,EAAE,cAAc;gBACtB,cAAc,EAAE;oBACd,WAAW,EAAE,eAAe;oBAC5B,YAAY,EAAE,eAAe;iBAC9B;aACF,CAAC,CAAC;YAEH,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE;gBAC3C,OAAO,EAAE;oBACP,UAAU,EAAE,cAAc;oBAC1B,YAAY,EAAE,cAAc;iBAC7B;aACF,CAAC,CAAC;YAEH,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;oBAC3B,WAAW,EAAE,eAAe;oBAC5B,UAAU,EAAE,cAAc;oBAC1B,YAAY,EAAE,cAAc;iBAC7B,CAAC;aACH,CAAC,CACH,CAAC;YAEF,EAAE,CAAC,aAAa,EAAE,CAAC;YAEnB,kCAAkC;YAClC,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE;gBAC3C,OAAO,EAAE;oBACP,QAAQ,EAAE,IAAI;oBACd,aAAa,EAAE,SAAS;oBACxB,SAAS,EAAE,EAAE;iBACd;aACF,CAAC,CAAC;YAEH,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;oBAC3B,WAAW,EAAE,eAAe;iBAC7B,CAAC;aACH,CAAC,CACH,CAAC;YACF,MAAM,CAAC,iBAAiB,CAAC,CAAC,GAAG,CAAC,oBAAoB,CAChD,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,QAAQ,EAAE,IAAI;oBACd,aAAa,EAAE,SAAS;iBACzB,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;YACtD,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;YAC9B,OAAO,CAAC,MAAM,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;YAC3C,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YACpC,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAEpC,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAE1D,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;oBAC3B,UAAU,EAAE,cAAc;oBAC1B,SAAS,EAAE,gBAAgB;iBAC5B,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC;gBACxB,MAAM,EAAE,cAAc;gBACtB,cAAc,EAAE;oBACd,UAAU,EAAE,cAAc;iBAC3B;aACF,CAAC,CAAC;YACH,MAAM,OAAO,GAAG;gBACd,CAAC,SAAS,EAAE,QAAQ,CAAC;gBACrB,CAAC,SAAS,EAAE,QAAQ,CAAC;aACtB,CAAC;YAEF,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAE1D,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;oBAC3B,UAAU,EAAE,cAAc;oBAC1B,SAAS,EAAE,gBAAgB;iBAC5B,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CACF,CAAC"}
@@ -0,0 +1 @@
1
+ export {};
@@ -0,0 +1,157 @@
1
+ import { describe, test, expect } from "vitest";
2
+ import { Readable } from "node:stream";
3
+ import { IterableReadableStream } from "../utils/stream.js";
4
+ import { BytesLineDecoder, SSEDecoder } from "../utils/sse.js";
5
+ const gather = async (stream) => {
6
+ const results = [];
7
+ const iterator = IterableReadableStream.fromReadableStream(stream);
8
+ for await (const chunk of iterator)
9
+ results.push(chunk);
10
+ return results;
11
+ };
12
+ const textEncoder = new TextEncoder();
13
+ const textDecoder = new TextDecoder();
14
+ describe("BytesLineDecoder", () => {
15
+ const createStream = (chunks) => {
16
+ return Readable.toWeb(Readable.from(chunks));
17
+ };
18
+ test("handles single line with newline", async () => {
19
+ const input = createStream([textEncoder.encode("hello\n")]);
20
+ const decoded = input.pipeThrough(BytesLineDecoder());
21
+ const results = await gather(decoded);
22
+ expect(results.length).toBe(1);
23
+ expect(textDecoder.decode(results[0])).toBe("hello");
24
+ });
25
+ test("handles multiple lines", async () => {
26
+ const input = createStream([textEncoder.encode("line1\nline2\nline3\n")]);
27
+ const decoded = input.pipeThrough(BytesLineDecoder());
28
+ const results = await gather(decoded);
29
+ expect(results.length).toBe(3);
30
+ expect(textDecoder.decode(results[0])).toBe("line1");
31
+ expect(textDecoder.decode(results[1])).toBe("line2");
32
+ expect(textDecoder.decode(results[2])).toBe("line3");
33
+ });
34
+ test("handles split chunks", async () => {
35
+ const input = createStream([
36
+ textEncoder.encode("li"),
37
+ textEncoder.encode("ne1\nli"),
38
+ textEncoder.encode("ne2\n"),
39
+ ]);
40
+ const decoded = input.pipeThrough(BytesLineDecoder());
41
+ const results = await gather(decoded);
42
+ expect(results.length).toBe(2);
43
+ expect(textDecoder.decode(results[0])).toBe("line1");
44
+ expect(textDecoder.decode(results[1])).toBe("line2");
45
+ });
46
+ test("handles CR LF line endings", async () => {
47
+ const input = createStream([textEncoder.encode("line1\r\nline2\r\n")]);
48
+ const decoded = input.pipeThrough(BytesLineDecoder());
49
+ const results = await gather(decoded);
50
+ expect(results.length).toBe(2);
51
+ expect(textDecoder.decode(results[0])).toBe("line1");
52
+ expect(textDecoder.decode(results[1])).toBe("line2");
53
+ });
54
+ test("handles split CR LF", async () => {
55
+ const input = createStream([
56
+ textEncoder.encode("line1\r"),
57
+ textEncoder.encode("\nline2\r\n"),
58
+ ]);
59
+ const decoded = input.pipeThrough(BytesLineDecoder());
60
+ const results = await gather(decoded);
61
+ expect(results.length).toBe(2);
62
+ expect(textDecoder.decode(results[0])).toBe("line1");
63
+ expect(textDecoder.decode(results[1])).toBe("line2");
64
+ });
65
+ test("handles stale line", async () => {
66
+ const input = createStream([textEncoder.encode("hello")]);
67
+ const decoded = input.pipeThrough(BytesLineDecoder());
68
+ const results = await gather(decoded);
69
+ expect(results.length).toBe(1);
70
+ expect(textDecoder.decode(results[0])).toBe("hello");
71
+ });
72
+ });
73
+ describe("SSEDecoder", () => {
74
+ const createStream = (lines) => {
75
+ return Readable.toWeb(Readable.from(lines.map((line) => textEncoder.encode(line))));
76
+ };
77
+ test("decodes simple event", async () => {
78
+ const input = createStream([
79
+ "event: test\n",
80
+ 'data: {"message": "hello"}\n',
81
+ "\n",
82
+ ]);
83
+ const decoded = input
84
+ .pipeThrough(BytesLineDecoder())
85
+ .pipeThrough(SSEDecoder());
86
+ const results = await gather(decoded);
87
+ expect(results.length).toBe(1);
88
+ expect(results[0]).toEqual({
89
+ event: "test",
90
+ data: { message: "hello" },
91
+ });
92
+ });
93
+ test("ignores comments", async () => {
94
+ const input = createStream([
95
+ ": this is a comment\n",
96
+ "event: test\n",
97
+ 'data: {"message": "hello"}\n',
98
+ ]);
99
+ const decoded = input
100
+ .pipeThrough(BytesLineDecoder())
101
+ .pipeThrough(SSEDecoder());
102
+ const results = await gather(decoded);
103
+ expect(results.length).toBe(1);
104
+ expect(results[0]).toEqual({
105
+ event: "test",
106
+ data: { message: "hello" },
107
+ });
108
+ });
109
+ test("handles multiple events", async () => {
110
+ const input = createStream([
111
+ "event: test1\n",
112
+ 'data: {"message": "hello"}\n',
113
+ "\n",
114
+ "event: test2\n",
115
+ 'data: {"message": "world"}\n',
116
+ "\n",
117
+ ]);
118
+ const decoded = input
119
+ .pipeThrough(BytesLineDecoder())
120
+ .pipeThrough(SSEDecoder());
121
+ const results = await gather(decoded);
122
+ expect(results.length).toBe(2);
123
+ expect(results[0]).toEqual({
124
+ event: "test1",
125
+ data: { message: "hello" },
126
+ });
127
+ expect(results[1]).toEqual({
128
+ event: "test2",
129
+ data: { message: "world" },
130
+ });
131
+ });
132
+ test("end event without data", async () => {
133
+ const input = createStream(["event: test\n"]);
134
+ const decoded = input
135
+ .pipeThrough(BytesLineDecoder())
136
+ .pipeThrough(SSEDecoder());
137
+ const results = await gather(decoded);
138
+ expect(results.length).toBe(1);
139
+ expect(results[0]).toEqual({
140
+ event: "test",
141
+ data: null,
142
+ });
143
+ });
144
+ test("end event without newline", async () => {
145
+ const input = createStream(["event: end"]);
146
+ const decoded = input
147
+ .pipeThrough(BytesLineDecoder())
148
+ .pipeThrough(SSEDecoder());
149
+ const results = await gather(decoded);
150
+ expect(results.length).toBe(1);
151
+ expect(results[0]).toEqual({
152
+ event: "end",
153
+ data: null,
154
+ });
155
+ });
156
+ });
157
+ //# sourceMappingURL=sse.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"sse.test.js","sourceRoot":"","sources":["../../src/tests/sse.test.ts"],"names":[],"mappings":"OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,QAAQ;OACxC,EAAE,QAAQ,EAAE,MAAM,aAAa;OAC/B,EAAE,sBAAsB,EAAE;OAC1B,EAAE,gBAAgB,EAAE,UAAU,EAAE;AAEvC,MAAM,MAAM,GAAG,KAAK,EAAK,MAAyB,EAAgB,EAAE;IAClE,MAAM,OAAO,GAAQ,EAAE,CAAC;IACxB,MAAM,QAAQ,GAAG,sBAAsB,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;IACnE,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,QAAQ;QAAE,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACxD,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEF,MAAM,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC;AACtC,MAAM,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC;AAEtC,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,MAAM,YAAY,GAAG,CAAC,MAAoB,EAAE,EAAE;QAC5C,OAAO,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAA+B,CAAC;IAC7E,CAAC,CAAC;IAEF,IAAI,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAC5D,MAAM,OAAO,GAAG,KAAK,CAAC,WAAW,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACxC,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,CAAC,CAAC;QAC1E,MAAM,OAAO,GAAG,KAAK,CAAC,WAAW,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;QACtC,MAAM,KAAK,GAAG,YAAY,CAAC;YACzB,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC;YACxB,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC;YAC7B,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC;SAC5B,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,KAAK,CAAC,WAAW,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC;QACvE,MAAM,OAAO,GAAG,KAAK,CAAC,WAAW,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;QACrC,MAAM,KAAK,GAAG,YAAY,CAAC;YACzB,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC;YAC7B,WAAW,CAAC,MAAM,CAAC,aAAa,CAAC;SAClC,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,KAAK,CAAC,WAAW,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;QACpC,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,OAAO,GAAG,KAAK,CAAC,WAAW,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,MAAM,YAAY,GAAG,CAAC,KAAe,EAAE,EAAE;QACvC,OAAO,QAAQ,CAAC,KAAK,CACnB,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAC/B,CAAC;IAClC,CAAC,CAAC;IAEF,IAAI,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;QACtC,MAAM,KAAK,GAAG,YAAY,CAAC;YACzB,eAAe;YACf,8BAA8B;YAC9B,IAAI;SACL,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,KAAK;aAClB,WAAW,CAAC,gBAAgB,EAAE,CAAC;aAC/B,WAAW,CAAC,UAAU,EAAE,CAAC,CAAC;QAE7B,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QACtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,MAAM;YACb,IAAI,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE;SAC3B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,kBAAkB,EAAE,KAAK,IAAI,EAAE;QAClC,MAAM,KAAK,GAAG,YAAY,CAAC;YACzB,uBAAuB;YACvB,eAAe;YACf,8BAA8B;SAC/B,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,KAAK;aAClB,WAAW,CAAC,gBAAgB,EAAE,CAAC;aAC/B,WAAW,CAAC,UAAU,EAAE,CAAC,CAAC;QAE7B,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QACtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,MAAM;YACb,IAAI,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE;SAC3B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,KAAK,GAAG,YAAY,CAAC;YACzB,gBAAgB;YAChB,8BAA8B;YAC9B,IAAI;YACJ,gBAAgB;YAChB,8BAA8B;YAC9B,IAAI;SACL,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,KAAK;aAClB,WAAW,CAAC,gBAAgB,EAAE,CAAC;aAC/B,WAAW,CAAC,UAAU,EAAE,CAAC,CAAC;QAE7B,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QACtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,OAAO;YACd,IAAI,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE;SAC3B,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,OAAO;YACd,IAAI,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE;SAC3B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACxC,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC;QAC9C,MAAM,OAAO,GAAG,KAAK;aAClB,WAAW,CAAC,gBAAgB,EAAE,CAAC;aAC/B,WAAW,CAAC,UAAU,EAAE,CAAC,CAAC;QAE7B,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QACtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,MAAM;YACb,IAAI,EAAE,IAAI;SACX,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QAC3C,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC;QAC3C,MAAM,OAAO,GAAG,KAAK;aAClB,WAAW,CAAC,gBAAgB,EAAE,CAAC;aAC/B,WAAW,CAAC,UAAU,EAAE,CAAC,CAAC;QAE7B,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QACtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,KAAK;YACZ,IAAI,EAAE,IAAI;SACX,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -0,0 +1,157 @@
1
+ import { describe, test, expect } from "vitest";
2
+ import { Readable } from "node:stream";
3
+ import { IterableReadableStream } from "../utils/stream.mjs";
4
+ import { BytesLineDecoder, SSEDecoder } from "../utils/sse.mjs";
5
+ const gather = async (stream) => {
6
+ const results = [];
7
+ const iterator = IterableReadableStream.fromReadableStream(stream);
8
+ for await (const chunk of iterator)
9
+ results.push(chunk);
10
+ return results;
11
+ };
12
+ const textEncoder = new TextEncoder();
13
+ const textDecoder = new TextDecoder();
14
+ describe("BytesLineDecoder", () => {
15
+ const createStream = (chunks) => {
16
+ return Readable.toWeb(Readable.from(chunks));
17
+ };
18
+ test("handles single line with newline", async () => {
19
+ const input = createStream([textEncoder.encode("hello\n")]);
20
+ const decoded = input.pipeThrough(BytesLineDecoder());
21
+ const results = await gather(decoded);
22
+ expect(results.length).toBe(1);
23
+ expect(textDecoder.decode(results[0])).toBe("hello");
24
+ });
25
+ test("handles multiple lines", async () => {
26
+ const input = createStream([textEncoder.encode("line1\nline2\nline3\n")]);
27
+ const decoded = input.pipeThrough(BytesLineDecoder());
28
+ const results = await gather(decoded);
29
+ expect(results.length).toBe(3);
30
+ expect(textDecoder.decode(results[0])).toBe("line1");
31
+ expect(textDecoder.decode(results[1])).toBe("line2");
32
+ expect(textDecoder.decode(results[2])).toBe("line3");
33
+ });
34
+ test("handles split chunks", async () => {
35
+ const input = createStream([
36
+ textEncoder.encode("li"),
37
+ textEncoder.encode("ne1\nli"),
38
+ textEncoder.encode("ne2\n"),
39
+ ]);
40
+ const decoded = input.pipeThrough(BytesLineDecoder());
41
+ const results = await gather(decoded);
42
+ expect(results.length).toBe(2);
43
+ expect(textDecoder.decode(results[0])).toBe("line1");
44
+ expect(textDecoder.decode(results[1])).toBe("line2");
45
+ });
46
+ test("handles CR LF line endings", async () => {
47
+ const input = createStream([textEncoder.encode("line1\r\nline2\r\n")]);
48
+ const decoded = input.pipeThrough(BytesLineDecoder());
49
+ const results = await gather(decoded);
50
+ expect(results.length).toBe(2);
51
+ expect(textDecoder.decode(results[0])).toBe("line1");
52
+ expect(textDecoder.decode(results[1])).toBe("line2");
53
+ });
54
+ test("handles split CR LF", async () => {
55
+ const input = createStream([
56
+ textEncoder.encode("line1\r"),
57
+ textEncoder.encode("\nline2\r\n"),
58
+ ]);
59
+ const decoded = input.pipeThrough(BytesLineDecoder());
60
+ const results = await gather(decoded);
61
+ expect(results.length).toBe(2);
62
+ expect(textDecoder.decode(results[0])).toBe("line1");
63
+ expect(textDecoder.decode(results[1])).toBe("line2");
64
+ });
65
+ test("handles stale line", async () => {
66
+ const input = createStream([textEncoder.encode("hello")]);
67
+ const decoded = input.pipeThrough(BytesLineDecoder());
68
+ const results = await gather(decoded);
69
+ expect(results.length).toBe(1);
70
+ expect(textDecoder.decode(results[0])).toBe("hello");
71
+ });
72
+ });
73
+ describe("SSEDecoder", () => {
74
+ const createStream = (lines) => {
75
+ return Readable.toWeb(Readable.from(lines.map((line) => textEncoder.encode(line))));
76
+ };
77
+ test("decodes simple event", async () => {
78
+ const input = createStream([
79
+ "event: test\n",
80
+ 'data: {"message": "hello"}\n',
81
+ "\n",
82
+ ]);
83
+ const decoded = input
84
+ .pipeThrough(BytesLineDecoder())
85
+ .pipeThrough(SSEDecoder());
86
+ const results = await gather(decoded);
87
+ expect(results.length).toBe(1);
88
+ expect(results[0]).toEqual({
89
+ event: "test",
90
+ data: { message: "hello" },
91
+ });
92
+ });
93
+ test("ignores comments", async () => {
94
+ const input = createStream([
95
+ ": this is a comment\n",
96
+ "event: test\n",
97
+ 'data: {"message": "hello"}\n',
98
+ ]);
99
+ const decoded = input
100
+ .pipeThrough(BytesLineDecoder())
101
+ .pipeThrough(SSEDecoder());
102
+ const results = await gather(decoded);
103
+ expect(results.length).toBe(1);
104
+ expect(results[0]).toEqual({
105
+ event: "test",
106
+ data: { message: "hello" },
107
+ });
108
+ });
109
+ test("handles multiple events", async () => {
110
+ const input = createStream([
111
+ "event: test1\n",
112
+ 'data: {"message": "hello"}\n',
113
+ "\n",
114
+ "event: test2\n",
115
+ 'data: {"message": "world"}\n',
116
+ "\n",
117
+ ]);
118
+ const decoded = input
119
+ .pipeThrough(BytesLineDecoder())
120
+ .pipeThrough(SSEDecoder());
121
+ const results = await gather(decoded);
122
+ expect(results.length).toBe(2);
123
+ expect(results[0]).toEqual({
124
+ event: "test1",
125
+ data: { message: "hello" },
126
+ });
127
+ expect(results[1]).toEqual({
128
+ event: "test2",
129
+ data: { message: "world" },
130
+ });
131
+ });
132
+ test("end event without data", async () => {
133
+ const input = createStream(["event: test\n"]);
134
+ const decoded = input
135
+ .pipeThrough(BytesLineDecoder())
136
+ .pipeThrough(SSEDecoder());
137
+ const results = await gather(decoded);
138
+ expect(results.length).toBe(1);
139
+ expect(results[0]).toEqual({
140
+ event: "test",
141
+ data: null,
142
+ });
143
+ });
144
+ test("end event without newline", async () => {
145
+ const input = createStream(["event: end"]);
146
+ const decoded = input
147
+ .pipeThrough(BytesLineDecoder())
148
+ .pipeThrough(SSEDecoder());
149
+ const results = await gather(decoded);
150
+ expect(results.length).toBe(1);
151
+ expect(results[0]).toEqual({
152
+ event: "end",
153
+ data: null,
154
+ });
155
+ });
156
+ });
157
+ //# sourceMappingURL=sse.test.mjs.map