@ank1015/providers 0.0.1 → 0.0.3
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.
- package/LICENSE +21 -0
- package/README.md +93 -383
- package/dist/agent/conversation.d.ts +97 -0
- package/dist/agent/conversation.d.ts.map +1 -0
- package/dist/agent/conversation.js +328 -0
- package/dist/agent/conversation.js.map +1 -0
- package/dist/agent/runner.d.ts +37 -0
- package/dist/agent/runner.d.ts.map +1 -0
- package/dist/agent/runner.js +169 -0
- package/dist/agent/runner.js.map +1 -0
- package/dist/agent/tools/calculate.d.ts +15 -0
- package/dist/agent/tools/calculate.d.ts.map +1 -0
- package/dist/agent/tools/calculate.js +23 -0
- package/dist/agent/tools/calculate.js.map +1 -0
- package/dist/agent/tools/get-current-time.d.ts +15 -0
- package/dist/agent/tools/get-current-time.d.ts.map +1 -0
- package/dist/agent/tools/get-current-time.js +38 -0
- package/dist/agent/tools/get-current-time.js.map +1 -0
- package/dist/agent/tools/index.d.ts +3 -0
- package/dist/agent/tools/index.d.ts.map +1 -0
- package/dist/agent/tools/index.js +3 -0
- package/dist/agent/tools/index.js.map +1 -0
- package/dist/agent/types.d.ts +53 -31
- package/dist/agent/types.d.ts.map +1 -1
- package/dist/agent/types.js +1 -2
- package/dist/agent/utils.d.ts +14 -0
- package/dist/agent/utils.d.ts.map +1 -0
- package/dist/agent/utils.js +59 -0
- package/dist/agent/utils.js.map +1 -0
- package/dist/index.d.ts +16 -9
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +16 -28
- package/dist/index.js.map +1 -1
- package/dist/llm.d.ts +15 -0
- package/dist/llm.d.ts.map +1 -0
- package/dist/llm.js +92 -0
- package/dist/llm.js.map +1 -0
- package/dist/models.d.ts +8 -1
- package/dist/models.d.ts.map +1 -1
- package/dist/models.generated.d.ts +25 -112
- package/dist/models.generated.d.ts.map +1 -1
- package/dist/models.generated.js +72 -227
- package/dist/models.generated.js.map +1 -1
- package/dist/models.js +30 -32
- package/dist/models.js.map +1 -1
- package/dist/providers/google/complete.d.ts +3 -0
- package/dist/providers/google/complete.d.ts.map +1 -0
- package/dist/providers/google/complete.js +53 -0
- package/dist/providers/google/complete.js.map +1 -0
- package/dist/providers/google/index.d.ts +6 -0
- package/dist/providers/google/index.d.ts.map +1 -0
- package/dist/providers/google/index.js +6 -0
- package/dist/providers/google/index.js.map +1 -0
- package/dist/providers/google/stream.d.ts +3 -0
- package/dist/providers/google/stream.d.ts.map +1 -0
- package/dist/providers/{google.js → google/stream.js} +67 -231
- package/dist/providers/google/stream.js.map +1 -0
- package/dist/providers/google/types.d.ts +8 -0
- package/dist/providers/google/types.d.ts.map +1 -0
- package/dist/providers/google/types.js +2 -0
- package/dist/providers/google/types.js.map +1 -0
- package/dist/providers/google/utils.d.ts +30 -0
- package/dist/providers/google/utils.d.ts.map +1 -0
- package/dist/providers/google/utils.js +354 -0
- package/dist/providers/google/utils.js.map +1 -0
- package/dist/providers/openai/complete.d.ts +3 -0
- package/dist/providers/openai/complete.d.ts.map +1 -0
- package/dist/providers/openai/complete.js +57 -0
- package/dist/providers/openai/complete.js.map +1 -0
- package/dist/providers/openai/index.d.ts +4 -0
- package/dist/providers/openai/index.d.ts.map +1 -0
- package/dist/providers/openai/index.js +4 -0
- package/dist/providers/openai/index.js.map +1 -0
- package/dist/providers/openai/stream.d.ts +3 -0
- package/dist/providers/openai/stream.d.ts.map +1 -0
- package/dist/providers/{openai.js → openai/stream.js} +74 -152
- package/dist/providers/openai/stream.js.map +1 -0
- package/dist/providers/openai/types.d.ts +8 -0
- package/dist/providers/openai/types.d.ts.map +1 -0
- package/dist/providers/openai/types.js +2 -0
- package/dist/providers/openai/types.js.map +1 -0
- package/dist/providers/openai/utils.d.ts +13 -0
- package/dist/providers/openai/utils.d.ts.map +1 -0
- package/dist/providers/openai/utils.js +285 -0
- package/dist/providers/openai/utils.js.map +1 -0
- package/dist/types.d.ts +95 -87
- package/dist/types.d.ts.map +1 -1
- package/dist/types.js +1 -9
- package/dist/types.js.map +1 -1
- package/dist/utils/event-stream.d.ts +2 -2
- package/dist/utils/event-stream.d.ts.map +1 -1
- package/dist/utils/event-stream.js +2 -7
- package/dist/utils/event-stream.js.map +1 -1
- package/dist/utils/json-parse.js +3 -6
- package/dist/utils/json-parse.js.map +1 -1
- package/dist/utils/overflow.d.ts +51 -0
- package/dist/utils/overflow.d.ts.map +1 -0
- package/dist/utils/overflow.js +106 -0
- package/dist/utils/overflow.js.map +1 -0
- package/dist/utils/sanitize-unicode.js +1 -4
- package/dist/utils/sanitize-unicode.js.map +1 -1
- package/dist/utils/uuid.d.ts +6 -0
- package/dist/utils/uuid.d.ts.map +1 -0
- package/dist/utils/uuid.js +9 -0
- package/dist/utils/uuid.js.map +1 -0
- package/dist/utils/validation.d.ts +10 -3
- package/dist/utils/validation.d.ts.map +1 -1
- package/dist/utils/validation.js +20 -12
- package/dist/utils/validation.js.map +1 -1
- package/package.json +47 -8
- package/biome.json +0 -43
- package/dist/agent/agent-loop.d.ts +0 -5
- package/dist/agent/agent-loop.d.ts.map +0 -1
- package/dist/agent/agent-loop.js +0 -219
- package/dist/agent/agent-loop.js.map +0 -1
- package/dist/providers/convert.d.ts +0 -6
- package/dist/providers/convert.d.ts.map +0 -1
- package/dist/providers/convert.js +0 -207
- package/dist/providers/convert.js.map +0 -1
- package/dist/providers/google.d.ts +0 -26
- package/dist/providers/google.d.ts.map +0 -1
- package/dist/providers/google.js.map +0 -1
- package/dist/providers/openai.d.ts +0 -17
- package/dist/providers/openai.d.ts.map +0 -1
- package/dist/providers/openai.js.map +0 -1
- package/dist/stream.d.ts +0 -4
- package/dist/stream.d.ts.map +0 -1
- package/dist/stream.js +0 -40
- package/dist/stream.js.map +0 -1
- package/dist/test-google-agent-loop.d.ts +0 -2
- package/dist/test-google-agent-loop.d.ts.map +0 -1
- package/dist/test-google-agent-loop.js +0 -186
- package/dist/test-google-agent-loop.js.map +0 -1
- package/dist/test-google.d.ts +0 -2
- package/dist/test-google.d.ts.map +0 -1
- package/dist/test-google.js +0 -41
- package/dist/test-google.js.map +0 -1
- package/src/agent/agent-loop.ts +0 -275
- package/src/agent/types.ts +0 -80
- package/src/index.ts +0 -72
- package/src/models.generated.ts +0 -314
- package/src/models.ts +0 -45
- package/src/providers/convert.ts +0 -222
- package/src/providers/google.ts +0 -496
- package/src/providers/openai.ts +0 -437
- package/src/stream.ts +0 -60
- package/src/types.ts +0 -198
- package/src/utils/event-stream.ts +0 -60
- package/src/utils/json-parse.ts +0 -28
- package/src/utils/sanitize-unicode.ts +0 -25
- package/src/utils/validation.ts +0 -69
- package/test/core/agent-loop.test.ts +0 -958
- package/test/core/stream.test.ts +0 -409
- package/test/data/red-circle.png +0 -0
- package/test/data/superintelligentwill.pdf +0 -0
- package/test/edge-cases/general.test.ts +0 -565
- package/test/integration/e2e.test.ts +0 -530
- package/test/models/cost.test.ts +0 -499
- package/test/models/registry.test.ts +0 -298
- package/test/providers/convert.test.ts +0 -846
- package/test/providers/google-schema.test.ts +0 -666
- package/test/providers/google-stream.test.ts +0 -369
- package/test/providers/openai-stream.test.ts +0 -251
- package/test/utils/event-stream.test.ts +0 -289
- package/test/utils/json-parse.test.ts +0 -344
- package/test/utils/sanitize-unicode.test.ts +0 -329
- package/test/utils/validation.test.ts +0 -614
- package/tsconfig.json +0 -21
- 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 @@
|
|
|
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 @@
|
|
|
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"}
|
package/dist/agent/types.d.ts
CHANGED
|
@@ -1,17 +1,48 @@
|
|
|
1
|
-
import {
|
|
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:
|
|
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
|
|
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
|
|
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
|
-
|
|
53
|
+
messageType: 'user' | 'assistant' | 'toolResult' | 'custom';
|
|
54
|
+
messageId: string;
|
|
55
|
+
message: Message;
|
|
23
56
|
} | {
|
|
24
57
|
type: "message_update";
|
|
25
|
-
|
|
26
|
-
|
|
58
|
+
messageType: 'assistant' | 'custom';
|
|
59
|
+
messageId: string;
|
|
60
|
+
message: Message | BaseAssistantEvent<Api>;
|
|
27
61
|
} | {
|
|
28
62
|
type: "message_end";
|
|
29
|
-
|
|
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
|
|
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
|
-
|
|
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,
|
|
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"}
|
package/dist/agent/types.js
CHANGED
|
@@ -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
|
|
2
|
-
export
|
|
3
|
-
export
|
|
4
|
-
export
|
|
5
|
-
export
|
|
6
|
-
export
|
|
7
|
-
export
|
|
8
|
-
export
|
|
9
|
-
export
|
|
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
|
package/dist/index.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"
|
|
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
|
-
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
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":"
|
|
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
|
package/dist/llm.js.map
ADDED
|
@@ -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
|
|
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
|
package/dist/models.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../src/models.ts"],"names":[],"mappings":"
|
|
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"}
|