@ank1015/providers 0.0.1 → 0.0.2

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 (169) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +93 -383
  3. package/dist/agent/conversation.d.ts +97 -0
  4. package/dist/agent/conversation.d.ts.map +1 -0
  5. package/dist/agent/conversation.js +328 -0
  6. package/dist/agent/conversation.js.map +1 -0
  7. package/dist/agent/runner.d.ts +37 -0
  8. package/dist/agent/runner.d.ts.map +1 -0
  9. package/dist/agent/runner.js +169 -0
  10. package/dist/agent/runner.js.map +1 -0
  11. package/dist/agent/tools/calculate.d.ts +15 -0
  12. package/dist/agent/tools/calculate.d.ts.map +1 -0
  13. package/dist/agent/tools/calculate.js +23 -0
  14. package/dist/agent/tools/calculate.js.map +1 -0
  15. package/dist/agent/tools/get-current-time.d.ts +15 -0
  16. package/dist/agent/tools/get-current-time.d.ts.map +1 -0
  17. package/dist/agent/tools/get-current-time.js +38 -0
  18. package/dist/agent/tools/get-current-time.js.map +1 -0
  19. package/dist/agent/tools/index.d.ts +3 -0
  20. package/dist/agent/tools/index.d.ts.map +1 -0
  21. package/dist/agent/tools/index.js +3 -0
  22. package/dist/agent/tools/index.js.map +1 -0
  23. package/dist/agent/types.d.ts +53 -31
  24. package/dist/agent/types.d.ts.map +1 -1
  25. package/dist/agent/types.js +1 -2
  26. package/dist/agent/utils.d.ts +14 -0
  27. package/dist/agent/utils.d.ts.map +1 -0
  28. package/dist/agent/utils.js +59 -0
  29. package/dist/agent/utils.js.map +1 -0
  30. package/dist/index.d.ts +16 -9
  31. package/dist/index.d.ts.map +1 -1
  32. package/dist/index.js +16 -28
  33. package/dist/index.js.map +1 -1
  34. package/dist/llm.d.ts +15 -0
  35. package/dist/llm.d.ts.map +1 -0
  36. package/dist/llm.js +92 -0
  37. package/dist/llm.js.map +1 -0
  38. package/dist/models.d.ts +8 -1
  39. package/dist/models.d.ts.map +1 -1
  40. package/dist/models.generated.d.ts +25 -112
  41. package/dist/models.generated.d.ts.map +1 -1
  42. package/dist/models.generated.js +72 -227
  43. package/dist/models.generated.js.map +1 -1
  44. package/dist/models.js +30 -32
  45. package/dist/models.js.map +1 -1
  46. package/dist/providers/google/complete.d.ts +3 -0
  47. package/dist/providers/google/complete.d.ts.map +1 -0
  48. package/dist/providers/google/complete.js +53 -0
  49. package/dist/providers/google/complete.js.map +1 -0
  50. package/dist/providers/google/index.d.ts +6 -0
  51. package/dist/providers/google/index.d.ts.map +1 -0
  52. package/dist/providers/google/index.js +6 -0
  53. package/dist/providers/google/index.js.map +1 -0
  54. package/dist/providers/google/stream.d.ts +3 -0
  55. package/dist/providers/google/stream.d.ts.map +1 -0
  56. package/dist/providers/{google.js → google/stream.js} +67 -231
  57. package/dist/providers/google/stream.js.map +1 -0
  58. package/dist/providers/google/types.d.ts +8 -0
  59. package/dist/providers/google/types.d.ts.map +1 -0
  60. package/dist/providers/google/types.js +2 -0
  61. package/dist/providers/google/types.js.map +1 -0
  62. package/dist/providers/google/utils.d.ts +30 -0
  63. package/dist/providers/google/utils.d.ts.map +1 -0
  64. package/dist/providers/google/utils.js +354 -0
  65. package/dist/providers/google/utils.js.map +1 -0
  66. package/dist/providers/openai/complete.d.ts +3 -0
  67. package/dist/providers/openai/complete.d.ts.map +1 -0
  68. package/dist/providers/openai/complete.js +57 -0
  69. package/dist/providers/openai/complete.js.map +1 -0
  70. package/dist/providers/openai/index.d.ts +4 -0
  71. package/dist/providers/openai/index.d.ts.map +1 -0
  72. package/dist/providers/openai/index.js +4 -0
  73. package/dist/providers/openai/index.js.map +1 -0
  74. package/dist/providers/openai/stream.d.ts +3 -0
  75. package/dist/providers/openai/stream.d.ts.map +1 -0
  76. package/dist/providers/{openai.js → openai/stream.js} +74 -152
  77. package/dist/providers/openai/stream.js.map +1 -0
  78. package/dist/providers/openai/types.d.ts +8 -0
  79. package/dist/providers/openai/types.d.ts.map +1 -0
  80. package/dist/providers/openai/types.js +2 -0
  81. package/dist/providers/openai/types.js.map +1 -0
  82. package/dist/providers/openai/utils.d.ts +13 -0
  83. package/dist/providers/openai/utils.d.ts.map +1 -0
  84. package/dist/providers/openai/utils.js +285 -0
  85. package/dist/providers/openai/utils.js.map +1 -0
  86. package/dist/types.d.ts +95 -87
  87. package/dist/types.d.ts.map +1 -1
  88. package/dist/types.js +1 -9
  89. package/dist/types.js.map +1 -1
  90. package/dist/utils/event-stream.d.ts +2 -2
  91. package/dist/utils/event-stream.d.ts.map +1 -1
  92. package/dist/utils/event-stream.js +2 -7
  93. package/dist/utils/event-stream.js.map +1 -1
  94. package/dist/utils/json-parse.js +3 -6
  95. package/dist/utils/json-parse.js.map +1 -1
  96. package/dist/utils/overflow.d.ts +51 -0
  97. package/dist/utils/overflow.d.ts.map +1 -0
  98. package/dist/utils/overflow.js +106 -0
  99. package/dist/utils/overflow.js.map +1 -0
  100. package/dist/utils/sanitize-unicode.js +1 -4
  101. package/dist/utils/sanitize-unicode.js.map +1 -1
  102. package/dist/utils/uuid.d.ts +6 -0
  103. package/dist/utils/uuid.d.ts.map +1 -0
  104. package/dist/utils/uuid.js +9 -0
  105. package/dist/utils/uuid.js.map +1 -0
  106. package/dist/utils/validation.d.ts +10 -3
  107. package/dist/utils/validation.d.ts.map +1 -1
  108. package/dist/utils/validation.js +20 -12
  109. package/dist/utils/validation.js.map +1 -1
  110. package/package.json +45 -8
  111. package/biome.json +0 -43
  112. package/dist/agent/agent-loop.d.ts +0 -5
  113. package/dist/agent/agent-loop.d.ts.map +0 -1
  114. package/dist/agent/agent-loop.js +0 -219
  115. package/dist/agent/agent-loop.js.map +0 -1
  116. package/dist/providers/convert.d.ts +0 -6
  117. package/dist/providers/convert.d.ts.map +0 -1
  118. package/dist/providers/convert.js +0 -207
  119. package/dist/providers/convert.js.map +0 -1
  120. package/dist/providers/google.d.ts +0 -26
  121. package/dist/providers/google.d.ts.map +0 -1
  122. package/dist/providers/google.js.map +0 -1
  123. package/dist/providers/openai.d.ts +0 -17
  124. package/dist/providers/openai.d.ts.map +0 -1
  125. package/dist/providers/openai.js.map +0 -1
  126. package/dist/stream.d.ts +0 -4
  127. package/dist/stream.d.ts.map +0 -1
  128. package/dist/stream.js +0 -40
  129. package/dist/stream.js.map +0 -1
  130. package/dist/test-google-agent-loop.d.ts +0 -2
  131. package/dist/test-google-agent-loop.d.ts.map +0 -1
  132. package/dist/test-google-agent-loop.js +0 -186
  133. package/dist/test-google-agent-loop.js.map +0 -1
  134. package/dist/test-google.d.ts +0 -2
  135. package/dist/test-google.d.ts.map +0 -1
  136. package/dist/test-google.js +0 -41
  137. package/dist/test-google.js.map +0 -1
  138. package/src/agent/agent-loop.ts +0 -275
  139. package/src/agent/types.ts +0 -80
  140. package/src/index.ts +0 -72
  141. package/src/models.generated.ts +0 -314
  142. package/src/models.ts +0 -45
  143. package/src/providers/convert.ts +0 -222
  144. package/src/providers/google.ts +0 -496
  145. package/src/providers/openai.ts +0 -437
  146. package/src/stream.ts +0 -60
  147. package/src/types.ts +0 -198
  148. package/src/utils/event-stream.ts +0 -60
  149. package/src/utils/json-parse.ts +0 -28
  150. package/src/utils/sanitize-unicode.ts +0 -25
  151. package/src/utils/validation.ts +0 -69
  152. package/test/core/agent-loop.test.ts +0 -958
  153. package/test/core/stream.test.ts +0 -409
  154. package/test/data/red-circle.png +0 -0
  155. package/test/data/superintelligentwill.pdf +0 -0
  156. package/test/edge-cases/general.test.ts +0 -565
  157. package/test/integration/e2e.test.ts +0 -530
  158. package/test/models/cost.test.ts +0 -499
  159. package/test/models/registry.test.ts +0 -298
  160. package/test/providers/convert.test.ts +0 -846
  161. package/test/providers/google-schema.test.ts +0 -666
  162. package/test/providers/google-stream.test.ts +0 -369
  163. package/test/providers/openai-stream.test.ts +0 -251
  164. package/test/utils/event-stream.test.ts +0 -289
  165. package/test/utils/json-parse.test.ts +0 -344
  166. package/test/utils/sanitize-unicode.test.ts +0 -329
  167. package/test/utils/validation.test.ts +0 -614
  168. package/tsconfig.json +0 -21
  169. package/vitest.config.ts +0 -9
@@ -0,0 +1,38 @@
1
+ import { Type } from "@sinclair/typebox";
2
+ export async function getCurrentTime(timezone) {
3
+ const date = new Date();
4
+ if (timezone) {
5
+ try {
6
+ const timeStr = date.toLocaleString("en-US", {
7
+ timeZone: timezone,
8
+ dateStyle: "full",
9
+ timeStyle: "long",
10
+ });
11
+ return {
12
+ content: [{ type: "text", content: timeStr }],
13
+ details: { utcTimestamp: date.getTime() },
14
+ };
15
+ }
16
+ catch (e) {
17
+ throw new Error(`Invalid timezone: ${timezone}. Current UTC time: ${date.toISOString()}`);
18
+ }
19
+ }
20
+ const timeStr = date.toLocaleString("en-US", { dateStyle: "full", timeStyle: "long" });
21
+ return {
22
+ content: [{ type: "text", content: timeStr }],
23
+ details: { utcTimestamp: date.getTime() },
24
+ };
25
+ }
26
+ const getCurrentTimeSchema = Type.Object({
27
+ timezone: Type.Optional(Type.String({ description: "Optional timezone (e.g., 'America/New_York', 'Europe/London')" })),
28
+ });
29
+ export const getCurrentTimeTool = {
30
+ label: "Current Time",
31
+ name: "get_current_time",
32
+ description: "Get the current date and time",
33
+ parameters: getCurrentTimeSchema,
34
+ execute: async (_toolCallId, args) => {
35
+ return getCurrentTime(args.timezone);
36
+ },
37
+ };
38
+ //# sourceMappingURL=get-current-time.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"get-current-time.js","sourceRoot":"","sources":["../../../src/agent/tools/get-current-time.ts"],"names":[],"mappings":"AAAA,OAAO,EAAe,IAAI,EAAE,MAAM,mBAAmB,CAAC;AAMtD,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,QAAiB;IACrD,MAAM,IAAI,GAAG,IAAI,IAAI,EAAE,CAAC;IACxB,IAAI,QAAQ,EAAE,CAAC;QACd,IAAI,CAAC;YACJ,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE;gBAC5C,QAAQ,EAAE,QAAQ;gBAClB,SAAS,EAAE,MAAM;gBACjB,SAAS,EAAE,MAAM;aACjB,CAAC,CAAC;YACH,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;gBAC7C,OAAO,EAAE,EAAE,YAAY,EAAE,IAAI,CAAC,OAAO,EAAE,EAAE;aACzC,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,uBAAuB,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QAC3F,CAAC;IACF,CAAC;IACD,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC;IACvF,OAAO;QACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;QAC7C,OAAO,EAAE,EAAE,YAAY,EAAE,IAAI,CAAC,OAAO,EAAE,EAAE;KACzC,CAAC;AACH,CAAC;AAED,MAAM,oBAAoB,GAAG,IAAI,CAAC,MAAM,CAAC;IACxC,QAAQ,EAAE,IAAI,CAAC,QAAQ,CACtB,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,+DAA+D,EAAE,CAAC,CAC7F;CACD,CAAC,CAAC;AAIH,MAAM,CAAC,MAAM,kBAAkB,GAAqE;IACnG,KAAK,EAAE,cAAc;IACrB,IAAI,EAAE,kBAAkB;IACxB,WAAW,EAAE,+BAA+B;IAC5C,UAAU,EAAE,oBAAoB;IAChC,OAAO,EAAE,KAAK,EAAE,WAAmB,EAAE,IAA0B,EAAE,EAAE;QAClE,OAAO,cAAc,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACtC,CAAC;CACD,CAAC"}
@@ -0,0 +1,3 @@
1
+ export { calculate, calculateTool } from "./calculate.js";
2
+ export { getCurrentTime, getCurrentTimeTool } from "./get-current-time.js";
3
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agent/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAC1D,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC"}
@@ -0,0 +1,3 @@
1
+ export { calculate, calculateTool } from "./calculate.js";
2
+ export { getCurrentTime, getCurrentTimeTool } from "./get-current-time.js";
3
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/agent/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAC1D,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC"}
@@ -1,17 +1,48 @@
1
- import { Message, Tool, UserTextContent, UserImageContent, UserFileContent, ToolResultMessage, AssistantMessageEvent, AssistantMessage, Model, OptionsForApi, Api, UserMessage } from "../types";
1
+ import { Api, BaseAssistantEvent, Content, Message, Model, OptionsForApi, Tool } from "../types.js";
2
2
  import type { Static, TSchema } from "@sinclair/typebox";
3
3
  export interface AgentToolResult<T> {
4
- content: (UserTextContent | UserImageContent | UserFileContent)[];
4
+ content: Content;
5
5
  details: T;
6
6
  }
7
+ /**
8
+ * Attachment type definition.
9
+ * TO make it easy for applications to pass inputs
10
+ */
11
+ export interface Attachment {
12
+ id: string;
13
+ type: "image" | "file";
14
+ fileName: string;
15
+ mimeType: string;
16
+ size?: number;
17
+ content: string;
18
+ }
19
+ export type AgentToolUpdateCallback<T = any> = (partialResult: AgentToolResult<T>) => void;
7
20
  export interface AgentTool<TParameters extends TSchema = TSchema, TDetails = any> extends Tool<TParameters> {
8
21
  label: string;
9
- execute: (toolCallId: string, params: Static<TParameters>, signal?: AbortSignal) => Promise<AgentToolResult<TDetails>>;
22
+ execute: (toolCallId: string, params: Static<TParameters>, signal?: AbortSignal, onUpdate?: AgentToolUpdateCallback<TDetails>) => Promise<AgentToolResult<TDetails>>;
23
+ }
24
+ export interface Provider<TApi extends Api> {
25
+ model: Model<TApi>;
26
+ providerOptions: OptionsForApi<TApi>;
10
27
  }
11
- export interface AgentContext {
28
+ export interface QueuedMessage<TApp = Message> {
29
+ original: TApp;
30
+ llm?: Message;
31
+ }
32
+ export interface AgentState {
12
33
  systemPrompt?: string;
34
+ provider: Provider<Api>;
13
35
  messages: Message[];
14
- tools?: readonly AgentTool<any>[];
36
+ tools: AgentTool[];
37
+ isStreaming: boolean;
38
+ pendingToolCalls: Set<string>;
39
+ error?: string;
40
+ }
41
+ export interface AgentLoopConfig {
42
+ systemPrompt?: string;
43
+ tools: AgentTool[];
44
+ provider: Provider<Api>;
45
+ getQueuedMessages: <T>() => Promise<QueuedMessage<T>[]>;
15
46
  }
16
47
  export type AgentEvent = {
17
48
  type: "agent_start";
@@ -19,49 +50,40 @@ export type AgentEvent = {
19
50
  type: "turn_start";
20
51
  } | {
21
52
  type: "message_start";
22
- message: AssistantMessage | UserMessage | ToolResultMessage;
53
+ messageType: 'user' | 'assistant' | 'toolResult' | 'custom';
54
+ messageId: string;
55
+ message: Message;
23
56
  } | {
24
57
  type: "message_update";
25
- assistantMessageEvent: AssistantMessageEvent;
26
- message: AssistantMessage;
58
+ messageType: 'assistant' | 'custom';
59
+ messageId: string;
60
+ message: Message | BaseAssistantEvent<Api>;
27
61
  } | {
28
62
  type: "message_end";
29
- message: AssistantMessage | UserMessage | ToolResultMessage;
63
+ messageType: 'user' | 'assistant' | 'toolResult' | 'custom';
64
+ messageId: string;
65
+ message: Message;
30
66
  } | {
31
67
  type: "tool_execution_start";
32
68
  toolCallId: string;
33
69
  toolName: string;
34
70
  args: any;
71
+ } | {
72
+ type: "tool_execution_update";
73
+ toolCallId: string;
74
+ toolName: string;
75
+ args: any;
76
+ partialResult: AgentToolResult<any>;
35
77
  } | {
36
78
  type: "tool_execution_end";
37
79
  toolCallId: string;
38
80
  toolName: string;
39
- result: AgentToolResult<any> | string;
81
+ result: AgentToolResult<any>;
40
82
  isError: boolean;
41
83
  } | {
42
84
  type: "turn_end";
43
- message: AssistantMessage;
44
- toolResults: ToolResultMessage[];
45
85
  } | {
46
86
  type: "agent_end";
47
- messages: AgentContext["messages"];
48
- status: "completed" | "aborted" | "error";
87
+ agentMessages: Message[];
49
88
  };
50
- export interface QueuedMessage<TApp = Message> {
51
- original: TApp;
52
- llm?: Message;
53
- }
54
- export type ReasoningEffort = "minimal" | "low" | "medium" | "high";
55
- export interface SimpleProviderOptions {
56
- reasoning?: ReasoningEffort;
57
- signal?: AbortSignal;
58
- apiKey?: string;
59
- maxTokens?: number;
60
- }
61
- export interface AgentLoopConfig<TApi extends Api> {
62
- model: Model<TApi>;
63
- preprocessor?: (messages: AgentContext["messages"], signal?: AbortSignal) => Promise<AgentContext["messages"]>;
64
- providerOptions: OptionsForApi<TApi>;
65
- getQueuedMessages?: <T>() => Promise<QueuedMessage<T>[]>;
66
- }
67
89
  //# sourceMappingURL=types.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/agent/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,eAAe,EAAE,gBAAgB,EAAE,eAAe,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,gBAAgB,EAAE,KAAK,EAAE,aAAa,EAAE,GAAG,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AACjM,OAAO,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAGzD,MAAM,WAAW,eAAe,CAAC,CAAC;IAEjC,OAAO,EAAE,CAAC,eAAe,GAAG,gBAAgB,GAAG,eAAe,CAAC,EAAE,CAAC;IAElE,OAAO,EAAE,CAAC,CAAC;CACX;AAGD,MAAM,WAAW,SAAS,CAAC,WAAW,SAAS,OAAO,GAAG,OAAO,EAAE,QAAQ,GAAG,GAAG,CAAE,SAAQ,IAAI,CAAC,WAAW,CAAC;IAE1G,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,CACR,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,MAAM,CAAC,WAAW,CAAC,EAC3B,MAAM,CAAC,EAAE,WAAW,KAChB,OAAO,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,CAAC;CACxC;AAGD,MAAM,WAAW,YAAY;IAC5B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,KAAK,CAAC,EAAE,SAAS,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC;CAClC;AAID,MAAM,MAAM,UAAU,GAEnB;IAAE,IAAI,EAAE,aAAa,CAAA;CAAE,GAEvB;IAAE,IAAI,EAAE,YAAY,CAAA;CAAE,GAEtB;IAAE,IAAI,EAAE,eAAe,CAAC;IAAC,OAAO,EAAE,gBAAgB,GAAG,WAAW,GAAG,iBAAiB,CAAA;CAAE,GAEtF;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,qBAAqB,EAAE,qBAAqB,CAAC;IAAC,OAAO,EAAE,gBAAgB,CAAA;CAAE,GAEnG;IAAE,IAAI,EAAE,aAAa,CAAC;IAAC,OAAO,EAAE,gBAAgB,GAAG,WAAW,GAAG,iBAAiB,CAAA;CAAE,GAEpF;IAAE,IAAI,EAAE,sBAAsB,CAAC;IAAC,UAAU,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,GAAG,CAAA;CAAE,GAEjF;IACA,IAAI,EAAE,oBAAoB,CAAC;IAC3B,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,eAAe,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC;IACtC,OAAO,EAAE,OAAO,CAAC;CAChB,GAED;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,OAAO,EAAE,gBAAgB,CAAC;IAAC,WAAW,EAAE,iBAAiB,EAAE,CAAA;CAAE,GAGjF;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,QAAQ,EAAE,YAAY,CAAC,UAAU,CAAC,CAAC;IAAC,MAAM,EAAE,WAAW,GAAG,SAAS,GAAG,OAAO,CAAA;CAAE,CAAC;AAGxG,MAAM,WAAW,aAAa,CAAC,IAAI,GAAG,OAAO;IAC5C,QAAQ,EAAE,IAAI,CAAC;IACf,GAAG,CAAC,EAAE,OAAO,CAAC;CACd;AAED,MAAM,MAAM,eAAe,GAAG,SAAS,GAAG,KAAK,GAAG,QAAQ,GAAG,MAAM,CAAC;AAEpE,MAAM,WAAW,qBAAqB;IACrC,SAAS,CAAC,EAAE,eAAe,CAAC;IAC5B,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;CACnB;AAGD,MAAM,WAAW,eAAe,CAAC,IAAI,SAAS,GAAG;IAChD,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IACnB,YAAY,CAAC,EAAE,CAAC,QAAQ,EAAE,YAAY,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,CAAC;IAC5G,eAAe,EAAE,aAAa,CAAC,IAAI,CAAC,CAAC;IACxC,iBAAiB,CAAC,EAAE,CAAC,CAAC,OAAO,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;CACzD"}
1
+ {"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/agent/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,kBAAkB,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,IAAI,EAAsB,MAAM,aAAa,CAAC;AACxH,OAAO,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAGzD,MAAM,WAAW,eAAe,CAAC,CAAC;IAEjC,OAAO,EAAE,OAAO,CAAC;IAEjB,OAAO,EAAE,CAAC,CAAC;CACX;AAED;;;GAGG;AACH,MAAM,WAAW,UAAU;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,OAAO,GAAG,MAAM,CAAC;IACvB,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;CAChB;AAGD,MAAM,MAAM,uBAAuB,CAAC,CAAC,GAAG,GAAG,IAAI,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC;AAG3F,MAAM,WAAW,SAAS,CAAC,WAAW,SAAS,OAAO,GAAG,OAAO,EAAE,QAAQ,GAAG,GAAG,CAAE,SAAQ,IAAI,CAAC,WAAW,CAAC;IAE1G,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,CACR,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,MAAM,CAAC,WAAW,CAAC,EAC3B,MAAM,CAAC,EAAE,WAAW,EACpB,QAAQ,CAAC,EAAE,uBAAuB,CAAC,QAAQ,CAAC,KACxC,OAAO,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,CAAC;CACxC;AAED,MAAM,WAAW,QAAQ,CAAC,IAAI,SAAS,GAAG;IACzC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IACnB,eAAe,EAAE,aAAa,CAAC,IAAI,CAAC,CAAA;CACpC;AAGD,MAAM,WAAW,aAAa,CAAC,IAAI,GAAG,OAAO;IAC5C,QAAQ,EAAE,IAAI,CAAC;IACf,GAAG,CAAC,EAAE,OAAO,CAAC;CACd;AAED,MAAM,WAAW,UAAU;IAC1B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC;IACxB,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,KAAK,EAAE,SAAS,EAAE,CAAA;IAClB,WAAW,EAAE,OAAO,CAAC;IACrB,gBAAgB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAC9B,KAAK,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,eAAe;IAC/B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE,SAAS,EAAE,CAAC;IACnB,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC;IACxB,iBAAiB,EAAE,CAAC,CAAC,OAAO,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC,CAAA;CACvD;AAMD,MAAM,MAAM,UAAU,GAEnB;IAAE,IAAI,EAAE,aAAa,CAAA;CAAE,GAEvB;IAAE,IAAI,EAAE,YAAY,CAAA;CAAE,GAEtB;IAAE,IAAI,EAAE,eAAe,CAAC;IAAC,WAAW,EAAE,MAAM,GAAG,WAAW,GAAG,YAAY,GAAG,QAAQ,CAAC;IAAC,SAAS,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,OAAO,CAAA;CAAE,GAE3H;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,WAAW,EAAE,WAAW,GAAG,QAAQ,CAAC;IAAC,SAAS,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,OAAO,GAAG,kBAAkB,CAAC,GAAG,CAAC,CAAA;CAAE,GAE9H;IAAE,IAAI,EAAE,aAAa,CAAC;IAAC,WAAW,EAAE,MAAM,GAAG,WAAW,GAAG,YAAY,GAAG,QAAQ,CAAC;IAAC,SAAS,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,OAAO,CAAA;CAAE,GAEzH;IAAE,IAAI,EAAE,sBAAsB,CAAC;IAAC,UAAU,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,GAAG,CAAA;CAAE,GAEjF;IACD,IAAI,EAAE,uBAAuB,CAAC;IAC9B,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,GAAG,CAAC;IACV,aAAa,EAAE,eAAe,CAAC,GAAG,CAAC,CAAA;CACnC,GAEC;IACD,IAAI,EAAE,oBAAoB,CAAC;IAC3B,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC;IAC7B,OAAO,EAAE,OAAO,CAAC;CACjB,GAEC;IAAE,IAAI,EAAE,UAAU,CAAA;CAAE,GAGpB;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,aAAa,EAAE,OAAO,EAAE,CAAA;CAAE,CAAC"}
@@ -1,3 +1,2 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
1
+ export {};
3
2
  //# sourceMappingURL=types.js.map
@@ -0,0 +1,14 @@
1
+ import { ToolResultMessage, UserMessage } from "../types.js";
2
+ import { AgentToolResult, Attachment } from "./types.js";
3
+ import { AssistantToolCall } from "../types.js";
4
+ /**
5
+ * Builds a UserMessage from text input and optional attachments.
6
+ * Pure function with no side effects.
7
+ */
8
+ export declare function buildUserMessage(input: string, attachments?: Attachment[]): UserMessage;
9
+ /**
10
+ * Builds a ToolResultMessage from tool execution results.
11
+ * Pure function with no side effects.
12
+ */
13
+ export declare function buildToolResultMessage(toolCall: AssistantToolCall, result: AgentToolResult<unknown>, isError: boolean, errorDetails?: ToolResultMessage['error']): ToolResultMessage;
14
+ //# sourceMappingURL=utils.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/agent/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAW,iBAAiB,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AACtE,OAAO,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AACzD,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAGhD;;;GAGG;AACH,wBAAgB,gBAAgB,CAC/B,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,UAAU,EAAE,GACxB,WAAW,CAqCb;AAED;;;GAGG;AACH,wBAAgB,sBAAsB,CACrC,QAAQ,EAAE,iBAAiB,EAC3B,MAAM,EAAE,eAAe,CAAC,OAAO,CAAC,EAChC,OAAO,EAAE,OAAO,EAChB,YAAY,CAAC,EAAE,iBAAiB,CAAC,OAAO,CAAC,GACvC,iBAAiB,CAYnB"}
@@ -0,0 +1,59 @@
1
+ import { generateUUID } from "../utils/uuid.js";
2
+ /**
3
+ * Builds a UserMessage from text input and optional attachments.
4
+ * Pure function with no side effects.
5
+ */
6
+ export function buildUserMessage(input, attachments) {
7
+ const content = [{ type: 'text', content: input }];
8
+ if (attachments?.length) {
9
+ for (const attachment of attachments) {
10
+ if (attachment.type === 'image') {
11
+ content.push({
12
+ type: 'image',
13
+ data: attachment.content,
14
+ mimeType: attachment.mimeType,
15
+ metadata: {
16
+ fileName: attachment.fileName,
17
+ size: attachment.size || 0
18
+ }
19
+ });
20
+ }
21
+ if (attachment.type === 'file') {
22
+ content.push({
23
+ type: 'file',
24
+ data: attachment.content,
25
+ mimeType: attachment.mimeType,
26
+ filename: attachment.fileName,
27
+ metadata: {
28
+ fileName: attachment.fileName,
29
+ size: attachment.size || 0
30
+ }
31
+ });
32
+ }
33
+ }
34
+ }
35
+ return {
36
+ role: 'user',
37
+ id: generateUUID(),
38
+ timestamp: Date.now(),
39
+ content
40
+ };
41
+ }
42
+ /**
43
+ * Builds a ToolResultMessage from tool execution results.
44
+ * Pure function with no side effects.
45
+ */
46
+ export function buildToolResultMessage(toolCall, result, isError, errorDetails) {
47
+ return {
48
+ role: 'toolResult',
49
+ id: generateUUID(),
50
+ toolCallId: toolCall.toolCallId,
51
+ toolName: toolCall.name,
52
+ content: result.content,
53
+ details: result.details,
54
+ isError,
55
+ error: errorDetails,
56
+ timestamp: Date.now(),
57
+ };
58
+ }
59
+ //# sourceMappingURL=utils.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/agent/utils.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAEhD;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAC/B,KAAa,EACb,WAA0B;IAE1B,MAAM,OAAO,GAAY,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;IAE5D,IAAI,WAAW,EAAE,MAAM,EAAE,CAAC;QACzB,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACtC,IAAI,UAAU,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBACjC,OAAO,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,UAAU,CAAC,OAAO;oBACxB,QAAQ,EAAE,UAAU,CAAC,QAAQ;oBAC7B,QAAQ,EAAE;wBACT,QAAQ,EAAE,UAAU,CAAC,QAAQ;wBAC7B,IAAI,EAAE,UAAU,CAAC,IAAI,IAAI,CAAC;qBAC1B;iBACD,CAAC,CAAC;YACJ,CAAC;YACD,IAAI,UAAU,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAChC,OAAO,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,UAAU,CAAC,OAAO;oBACxB,QAAQ,EAAE,UAAU,CAAC,QAAQ;oBAC7B,QAAQ,EAAE,UAAU,CAAC,QAAQ;oBAC7B,QAAQ,EAAE;wBACT,QAAQ,EAAE,UAAU,CAAC,QAAQ;wBAC7B,IAAI,EAAE,UAAU,CAAC,IAAI,IAAI,CAAC;qBAC1B;iBACD,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;IACF,CAAC;IAED,OAAO;QACN,IAAI,EAAE,MAAM;QACZ,EAAE,EAAE,YAAY,EAAE;QAClB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,OAAO;KACP,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,sBAAsB,CACrC,QAA2B,EAC3B,MAAgC,EAChC,OAAgB,EAChB,YAAyC;IAEzC,OAAO;QACN,IAAI,EAAE,YAAY;QAClB,EAAE,EAAE,YAAY,EAAE;QAClB,UAAU,EAAE,QAAQ,CAAC,UAAU;QAC/B,QAAQ,EAAE,QAAQ,CAAC,IAAI;QACvB,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,OAAO;QACP,KAAK,EAAE,YAAY;QACnB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;KACrB,CAAC;AACH,CAAC"}
package/dist/index.d.ts CHANGED
@@ -1,10 +1,17 @@
1
- export { stream } from "./stream";
2
- export { agentLoop } from "./agent/agent-loop";
3
- export type { Message, UserMessage, ToolResultMessage, NativeAssistantMessage, NativeOpenAIMessage, UserTextContent, UserImageContent, UserFileContent, AssistantTextContent, AssistantThinkingContent, AssistantToolCall, AbstractedImageContent, Context, Tool, Api, Model, AssistantMessage, AssistantMessageEvent, Usage, StopReason, OptionsForApi, StreamFunction, } from "./types";
4
- export type { AgentContext, AgentTool, AgentToolResult, AgentEvent, AgentLoopConfig, QueuedMessage, } from "./agent/types";
5
- export { MODELS } from "./models.generated";
6
- export { calculateCost } from "./models";
7
- export { defineTool } from "./types";
8
- export type { ToolName, ToolNames } from "./types";
9
- export { EventStream, AssistantMessageEventStream } from "./utils/event-stream";
1
+ export * from './types.js';
2
+ export * from './agent/types.js';
3
+ export * from './models.js';
4
+ export * from './models.generated.js';
5
+ export * from './llm.js';
6
+ export * from './providers/openai/index.js';
7
+ export * from './providers/google/index.js';
8
+ export * from './agent/conversation.js';
9
+ export * from './agent/utils.js';
10
+ export * from './utils/sanitize-unicode.js';
11
+ export * from './utils/uuid.js';
12
+ export * from './utils/validation.js';
13
+ export * from './utils/overflow.js';
14
+ export * from './utils/event-stream.js';
15
+ export * from './utils/json-parse.js';
16
+ export * from './agent/tools/index.js';
10
17
  //# sourceMappingURL=index.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAClC,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAK/C,YAAY,EAEX,OAAO,EACP,WAAW,EACX,iBAAiB,EACjB,sBAAsB,EACtB,mBAAmB,EAGnB,eAAe,EACf,gBAAgB,EAChB,eAAe,EACf,oBAAoB,EACpB,wBAAwB,EACxB,iBAAiB,EACjB,sBAAsB,EAGtB,OAAO,EACP,IAAI,EACJ,GAAG,EACH,KAAK,EAGL,gBAAgB,EAChB,qBAAqB,EACrB,KAAK,EACL,UAAU,EAGV,aAAa,EACb,cAAc,GACd,MAAM,SAAS,CAAC;AAKjB,YAAY,EAEX,YAAY,EACZ,SAAS,EACT,eAAe,EACf,UAAU,EACV,eAAe,EACf,aAAa,GACb,MAAM,eAAe,CAAC;AAKvB,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAKzC,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,YAAY,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAKnD,OAAO,EAAE,WAAW,EAAE,2BAA2B,EAAE,MAAM,sBAAsB,CAAC"}
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAC3B,cAAc,kBAAkB,CAAC;AAEjC,cAAc,aAAa,CAAC;AAC5B,cAAc,uBAAuB,CAAC;AAEtC,cAAc,UAAU,CAAC;AACzB,cAAc,6BAA6B,CAAC;AAC5C,cAAc,6BAA6B,CAAC;AAE5C,cAAc,yBAAyB,CAAC;AACxC,cAAc,kBAAkB,CAAC;AAEjC,cAAc,6BAA6B,CAAC;AAC5C,cAAc,iBAAiB,CAAC;AAChC,cAAc,uBAAuB,CAAC;AACtC,cAAc,qBAAqB,CAAA;AACnC,cAAc,yBAAyB,CAAA;AACvC,cAAc,uBAAuB,CAAA;AAErC,cAAc,wBAAwB,CAAA"}
package/dist/index.js CHANGED
@@ -1,29 +1,17 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.AssistantMessageEventStream = exports.EventStream = exports.defineTool = exports.calculateCost = exports.MODELS = exports.agentLoop = exports.stream = void 0;
4
- // ============================================================================
5
- // Main Functions
6
- // ============================================================================
7
- var stream_1 = require("./stream");
8
- Object.defineProperty(exports, "stream", { enumerable: true, get: function () { return stream_1.stream; } });
9
- var agent_loop_1 = require("./agent/agent-loop");
10
- Object.defineProperty(exports, "agentLoop", { enumerable: true, get: function () { return agent_loop_1.agentLoop; } });
11
- // ============================================================================
12
- // Models and Registry
13
- // ============================================================================
14
- var models_generated_1 = require("./models.generated");
15
- Object.defineProperty(exports, "MODELS", { enumerable: true, get: function () { return models_generated_1.MODELS; } });
16
- var models_1 = require("./models");
17
- Object.defineProperty(exports, "calculateCost", { enumerable: true, get: function () { return models_1.calculateCost; } });
18
- // ============================================================================
19
- // Utilities
20
- // ============================================================================
21
- var types_1 = require("./types");
22
- Object.defineProperty(exports, "defineTool", { enumerable: true, get: function () { return types_1.defineTool; } });
23
- // ============================================================================
24
- // Event Streams
25
- // ============================================================================
26
- var event_stream_1 = require("./utils/event-stream");
27
- Object.defineProperty(exports, "EventStream", { enumerable: true, get: function () { return event_stream_1.EventStream; } });
28
- Object.defineProperty(exports, "AssistantMessageEventStream", { enumerable: true, get: function () { return event_stream_1.AssistantMessageEventStream; } });
1
+ export * from './types.js';
2
+ export * from './agent/types.js';
3
+ export * from './models.js';
4
+ export * from './models.generated.js';
5
+ export * from './llm.js';
6
+ export * from './providers/openai/index.js';
7
+ export * from './providers/google/index.js';
8
+ export * from './agent/conversation.js';
9
+ export * from './agent/utils.js';
10
+ export * from './utils/sanitize-unicode.js';
11
+ export * from './utils/uuid.js';
12
+ export * from './utils/validation.js';
13
+ export * from './utils/overflow.js';
14
+ export * from './utils/event-stream.js';
15
+ export * from './utils/json-parse.js';
16
+ export * from './agent/tools/index.js';
29
17
  //# sourceMappingURL=index.js.map
package/dist/index.js.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,+EAA+E;AAC/E,iBAAiB;AACjB,+EAA+E;AAC/E,mCAAkC;AAAzB,gGAAA,MAAM,OAAA;AACf,iDAA+C;AAAtC,uGAAA,SAAS,OAAA;AAoDlB,+EAA+E;AAC/E,sBAAsB;AACtB,+EAA+E;AAC/E,uDAA4C;AAAnC,0GAAA,MAAM,OAAA;AACf,mCAAyC;AAAhC,uGAAA,aAAa,OAAA;AAEtB,+EAA+E;AAC/E,YAAY;AACZ,+EAA+E;AAC/E,iCAAqC;AAA5B,mGAAA,UAAU,OAAA;AAGnB,+EAA+E;AAC/E,gBAAgB;AAChB,+EAA+E;AAC/E,qDAAgF;AAAvE,2GAAA,WAAW,OAAA;AAAE,2HAAA,2BAA2B,OAAA"}
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAC3B,cAAc,kBAAkB,CAAC;AAEjC,cAAc,aAAa,CAAC;AAC5B,cAAc,uBAAuB,CAAC;AAEtC,cAAc,UAAU,CAAC;AACzB,cAAc,6BAA6B,CAAC;AAC5C,cAAc,6BAA6B,CAAC;AAE5C,cAAc,yBAAyB,CAAC;AACxC,cAAc,kBAAkB,CAAC;AAEjC,cAAc,6BAA6B,CAAC;AAC5C,cAAc,iBAAiB,CAAC;AAChC,cAAc,uBAAuB,CAAC;AACtC,cAAc,qBAAqB,CAAA;AACnC,cAAc,yBAAyB,CAAA;AACvC,cAAc,uBAAuB,CAAA;AAErC,cAAc,wBAAwB,CAAA"}
package/dist/llm.d.ts ADDED
@@ -0,0 +1,15 @@
1
+ import { Model, Api, Context, OptionsForApi, BaseAssistantMessage } from "./types.js";
2
+ import { AssistantMessageEventStream } from "./utils/event-stream.js";
3
+ export declare function getApiKeyFromEnv(api: Api): string | undefined;
4
+ export declare function complete<TApi extends Api>(model: Model<TApi>, context: Context, options?: OptionsForApi<TApi>, id?: string): Promise<BaseAssistantMessage<TApi>>;
5
+ export declare function stream<TApi extends Api>(model: Model<TApi>, context: Context, options?: OptionsForApi<TApi>, id?: string): AssistantMessageEventStream<TApi>;
6
+ export declare function getMockMessage(model: Model<Api>): BaseAssistantMessage<Api>;
7
+ export interface LLMClient {
8
+ complete<TApi extends Api>(model: Model<TApi>, context: Context, options?: OptionsForApi<TApi>, id?: string): Promise<BaseAssistantMessage<TApi>>;
9
+ stream<TApi extends Api>(model: Model<TApi>, context: Context, options?: OptionsForApi<TApi>, id?: string): AssistantMessageEventStream<TApi>;
10
+ }
11
+ export declare class DefaultLLMClient implements LLMClient {
12
+ complete<TApi extends Api>(model: Model<TApi>, context: Context, options?: OptionsForApi<TApi>, id?: string): Promise<BaseAssistantMessage<TApi>>;
13
+ stream<TApi extends Api>(model: Model<TApi>, context: Context, options?: OptionsForApi<TApi>, id?: string): AssistantMessageEventStream<TApi>;
14
+ }
15
+ //# sourceMappingURL=llm.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"llm.d.ts","sourceRoot":"","sources":["../src/llm.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,aAAa,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC;AACtF,OAAO,EAAE,2BAA2B,EAAE,MAAM,yBAAyB,CAAC;AAStE,wBAAgB,gBAAgB,CAAC,GAAG,EAAE,GAAG,sBAGxC;AAGD,wBAAsB,QAAQ,CAAC,IAAI,SAAS,GAAG,EAC3C,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,EAClB,OAAO,EAAE,OAAO,EAChB,OAAO,CAAC,EAAE,aAAa,CAAC,IAAI,CAAC,EAC7B,EAAE,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC,CAgCrC;AAED,wBAAgB,MAAM,CAAC,IAAI,SAAS,GAAG,EACnC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,EAClB,OAAO,EAAE,OAAO,EAChB,OAAO,CAAC,EAAE,aAAa,CAAC,IAAI,CAAC,EAC7B,EAAE,CAAC,EAAE,MAAM,GACZ,2BAA2B,CAAC,IAAI,CAAC,CAgCnC;AAED,wBAAgB,cAAc,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,oBAAoB,CAAC,GAAG,CAAC,CA4B3E;AAED,MAAM,WAAW,SAAS;IACtB,QAAQ,CAAC,IAAI,SAAS,GAAG,EACrB,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,EAClB,OAAO,EAAE,OAAO,EAChB,OAAO,CAAC,EAAE,aAAa,CAAC,IAAI,CAAC,EAC7B,EAAE,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC,CAAC;IAEvC,MAAM,CAAC,IAAI,SAAS,GAAG,EACnB,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,EAClB,OAAO,EAAE,OAAO,EAChB,OAAO,CAAC,EAAE,aAAa,CAAC,IAAI,CAAC,EAC7B,EAAE,CAAC,EAAE,MAAM,GACZ,2BAA2B,CAAC,IAAI,CAAC,CAAC;CACxC;AAED,qBAAa,gBAAiB,YAAW,SAAS;IACxC,QAAQ,CAAC,IAAI,SAAS,GAAG,EAC3B,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,EAClB,OAAO,EAAE,OAAO,EAChB,OAAO,CAAC,EAAE,aAAa,CAAC,IAAI,CAAC,EAC7B,EAAE,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;IAItC,MAAM,CAAC,IAAI,SAAS,GAAG,EACnB,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,EAClB,OAAO,EAAE,OAAO,EAChB,OAAO,CAAC,EAAE,aAAa,CAAC,IAAI,CAAC,EAC7B,EAAE,CAAC,EAAE,MAAM,GACZ,2BAA2B,CAAC,IAAI,CAAC;CAGvC"}
package/dist/llm.js ADDED
@@ -0,0 +1,92 @@
1
+ import { completeGoogle, streamGoogle } from "./providers/google/index.js";
2
+ import { getMockGoogleMessage } from "./providers/google/utils.js";
3
+ import { completeOpenAI, streamOpenAI } from "./providers/openai/index.js";
4
+ import { getMockOpenaiMessage } from "./providers/openai/utils.js";
5
+ import { generateUUID } from "./utils/uuid.js";
6
+ const envMap = {
7
+ openai: "OPENAI_API_KEY",
8
+ google: "GEMINI_API_KEY",
9
+ };
10
+ export function getApiKeyFromEnv(api) {
11
+ const envVar = envMap[api];
12
+ return process.env[envVar];
13
+ }
14
+ export async function complete(model, context, options, id) {
15
+ // Type-safe apiKey extraction - works because all provider options have apiKey
16
+ const apiKey = options?.apiKey ?? getApiKeyFromEnv(model.api);
17
+ if (!apiKey) {
18
+ throw new Error(`No API key for provider: ${model.api}`);
19
+ }
20
+ // Ensure providerOptions has required apiKey
21
+ const providerOptions = { ...options, apiKey };
22
+ const messageId = id ?? generateUUID();
23
+ switch (model.api) {
24
+ case 'openai':
25
+ return completeOpenAI(model, context, providerOptions, messageId);
26
+ case 'google':
27
+ return completeGoogle(model, context, providerOptions, messageId);
28
+ default: {
29
+ const _exhaustive = model.api;
30
+ throw new Error(`Unhandled API: ${_exhaustive}`);
31
+ }
32
+ }
33
+ }
34
+ export function stream(model, context, options, id) {
35
+ // Type-safe apiKey extraction - works because all provider options have apiKey
36
+ const apiKey = options?.apiKey ?? getApiKeyFromEnv(model.api);
37
+ if (!apiKey) {
38
+ throw new Error(`No API key for provider: ${model.api}`);
39
+ }
40
+ // Ensure providerOptions has required apiKey
41
+ const providerOptions = { ...options, apiKey };
42
+ const messageId = id ?? generateUUID();
43
+ switch (model.api) {
44
+ case 'openai':
45
+ return streamOpenAI(model, context, providerOptions, messageId);
46
+ case 'google':
47
+ return streamGoogle(model, context, providerOptions, messageId);
48
+ default: {
49
+ const _exhaustive = model.api;
50
+ throw new Error(`Unhandled API: ${_exhaustive}`);
51
+ }
52
+ }
53
+ }
54
+ export function getMockMessage(model) {
55
+ const messageId = generateUUID();
56
+ let message;
57
+ if (model.api === 'openai') {
58
+ message = getMockOpenaiMessage();
59
+ }
60
+ else if (model.api === 'google') {
61
+ message = getMockGoogleMessage();
62
+ }
63
+ const baseMessage = {
64
+ role: 'assistant',
65
+ message: message,
66
+ api: model.api,
67
+ id: messageId,
68
+ model: model,
69
+ timestamp: Date.now(),
70
+ duration: 0,
71
+ stopReason: 'stop',
72
+ content: [],
73
+ usage: {
74
+ input: 0,
75
+ output: 0,
76
+ cacheRead: 0,
77
+ cacheWrite: 0,
78
+ totalTokens: 0,
79
+ cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 }
80
+ }
81
+ };
82
+ return baseMessage;
83
+ }
84
+ export class DefaultLLMClient {
85
+ async complete(model, context, options, id) {
86
+ return complete(model, context, options, id);
87
+ }
88
+ stream(model, context, options, id) {
89
+ return stream(model, context, options, id);
90
+ }
91
+ }
92
+ //# sourceMappingURL=llm.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"llm.js","sourceRoot":"","sources":["../src/llm.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAyB,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAClG,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AACnE,OAAO,EAAE,cAAc,EAAyB,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAClG,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AAGnE,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAE/C,MAAM,MAAM,GAAwB;IAChC,MAAM,EAAE,gBAAgB;IACxB,MAAM,EAAE,gBAAgB;CAC3B,CAAC;AAGF,MAAM,UAAU,gBAAgB,CAAC,GAAQ;IACrC,MAAM,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,CAAA;IAC1B,OAAO,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;AAC9B,CAAC;AAGD,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC1B,KAAkB,EAClB,OAAgB,EAChB,OAA6B,EAC7B,EAAW;IAGX,+EAA+E;IAC/E,MAAM,MAAM,GAAI,OAAe,EAAE,MAAM,IAAI,gBAAgB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACvE,IAAI,CAAC,MAAM,EAAE,CAAC;QACV,MAAM,IAAI,KAAK,CAAC,4BAA4B,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;IAC7D,CAAC;IAED,6CAA6C;IAC7C,MAAM,eAAe,GAAG,EAAE,GAAG,OAAO,EAAE,MAAM,EAAyB,CAAC;IACtE,MAAM,SAAS,GAAG,EAAE,IAAI,YAAY,EAAE,CAAC;IAEvC,QAAQ,KAAK,CAAC,GAAG,EAAE,CAAC;QAChB,KAAK,QAAQ;YACT,OAAO,cAAc,CACjB,KAAwB,EACxB,OAAO,EACP,eAAwC,EACxC,SAAS,CAC2B,CAAC;QAC7C,KAAK,QAAQ;YACT,OAAO,cAAc,CACjB,KAAwB,EACxB,OAAO,EACP,eAAwC,EACxC,SAAS,CAC2B,CAAC;QAC7C,OAAO,CAAC,CAAC,CAAC;YACN,MAAM,WAAW,GAAU,KAAK,CAAC,GAAG,CAAC;YACrC,MAAM,IAAI,KAAK,CAAC,kBAAkB,WAAW,EAAE,CAAC,CAAC;QACrD,CAAC;IACL,CAAC;AACL,CAAC;AAED,MAAM,UAAU,MAAM,CAClB,KAAkB,EAClB,OAAgB,EAChB,OAA6B,EAC7B,EAAW;IAGX,+EAA+E;IAC/E,MAAM,MAAM,GAAI,OAAe,EAAE,MAAM,IAAI,gBAAgB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACvE,IAAI,CAAC,MAAM,EAAE,CAAC;QACV,MAAM,IAAI,KAAK,CAAC,4BAA4B,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;IAC7D,CAAC;IAED,6CAA6C;IAC7C,MAAM,eAAe,GAAG,EAAE,GAAG,OAAO,EAAE,MAAM,EAAyB,CAAC;IACtE,MAAM,SAAS,GAAG,EAAE,IAAI,YAAY,EAAE,CAAC;IAEvC,QAAQ,KAAK,CAAC,GAAG,EAAE,CAAC;QAChB,KAAK,QAAQ;YACT,OAAO,YAAY,CACf,KAAwB,EACxB,OAAO,EACP,eAAwC,EACxC,SAAS,CACoC,CAAC;QACtD,KAAK,QAAQ;YACT,OAAO,YAAY,CACf,KAAwB,EACxB,OAAO,EACP,eAAwC,EACxC,SAAS,CACoC,CAAC;QACtD,OAAO,CAAC,CAAC,CAAC;YACN,MAAM,WAAW,GAAU,KAAK,CAAC,GAAG,CAAC;YACrC,MAAM,IAAI,KAAK,CAAC,kBAAkB,WAAW,EAAE,CAAC,CAAC;QACrD,CAAC;IACL,CAAC;AACL,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,KAAiB;IAC5C,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;IACjC,IAAI,OAAO,CAAC;IACZ,IAAI,KAAK,CAAC,GAAG,KAAK,QAAQ,EAAE,CAAC;QACzB,OAAO,GAAG,oBAAoB,EAAE,CAAA;IACpC,CAAC;SAAM,IAAI,KAAK,CAAC,GAAG,KAAK,QAAQ,EAAE,CAAC;QAChC,OAAO,GAAG,oBAAoB,EAAE,CAAA;IACpC,CAAC;IACD,MAAM,WAAW,GAA8B;QAC3C,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE,OAAQ;QACjB,GAAG,EAAE,KAAK,CAAC,GAAG;QACd,EAAE,EAAE,SAAS;QACb,KAAK,EAAE,KAAK;QACZ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,QAAQ,EAAE,CAAC;QACX,UAAU,EAAE,MAAM;QAClB,OAAO,EAAE,EAAE;QACX,KAAK,EAAE;YACH,KAAK,EAAE,CAAC;YACR,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,CAAC;YACZ,UAAU,EAAE,CAAC;YACb,WAAW,EAAE,CAAC;YACd,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;SACvE;KACJ,CAAA;IACD,OAAO,WAAW,CAAA;AACtB,CAAC;AAkBD,MAAM,OAAO,gBAAgB;IACzB,KAAK,CAAC,QAAQ,CACV,KAAkB,EAClB,OAAgB,EAChB,OAA6B,EAC7B,EAAW;QAEX,OAAO,QAAQ,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;IACjD,CAAC;IAED,MAAM,CACF,KAAkB,EAClB,OAAgB,EAChB,OAA6B,EAC7B,EAAW;QAEX,OAAO,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;IAC/C,CAAC;CACJ"}
package/dist/models.d.ts CHANGED
@@ -1,3 +1,10 @@
1
- import type { Api, Model, Usage } from "./types.js";
1
+ import { Api, Model, Usage } from "./types.js";
2
+ import { MODELS } from "./models.generated.js";
3
+ type ModelIdsForApi<TApi extends Api> = TApi extends keyof typeof MODELS ? keyof (typeof MODELS)[TApi] : never;
4
+ export declare function getProviders(): Api[];
5
+ export declare function getModel<TApi extends Api>(api: TApi, modelId: ModelIdsForApi<TApi>): Model<TApi> | undefined;
6
+ export declare function getModels<TApi extends Api>(api: TApi): Model<TApi>[];
2
7
  export declare function calculateCost<TApi extends Api>(model: Model<TApi>, usage: Usage): Usage["cost"];
8
+ export declare function getAvailableModels(): Model<Api>[];
9
+ export {};
3
10
  //# sourceMappingURL=models.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../src/models.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAoCpD,wBAAgB,aAAa,CAAC,IAAI,SAAS,GAAG,EAAE,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAO/F"}
1
+ {"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../src/models.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAa,KAAK,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAC1D,OAAO,EAAE,MAAM,EAAE,MAAM,uBAAuB,CAAC;AAI/C,KAAK,cAAc,CAAC,IAAI,SAAS,GAAG,IAAI,IAAI,SAAS,MAAM,OAAO,MAAM,GACrE,MAAM,CAAC,OAAO,MAAM,CAAC,CAAC,IAAI,CAAC,GAC3B,KAAK,CAAC;AAET,wBAAgB,YAAY,IAAI,GAAG,EAAE,CAEpC;AAED,wBAAgB,QAAQ,CAAC,IAAI,SAAS,GAAG,EACxC,GAAG,EAAE,IAAI,EACT,OAAO,EAAE,cAAc,CAAC,IAAI,CAAC,GAC3B,KAAK,CAAC,IAAI,CAAC,GAAG,SAAS,CAMzB;AAED,wBAAgB,SAAS,CAAC,IAAI,SAAS,GAAG,EAAE,GAAG,EAAE,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,CAIpE;AAED,wBAAgB,aAAa,CAAC,IAAI,SAAS,GAAG,EAAE,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAO/F;AAED,wBAAgB,kBAAkB,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE,CAYjD"}