@nestjs-adk/core 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 (47) hide show
  1. package/README.md +28 -0
  2. package/dist/index.cjs +1452 -0
  3. package/dist/index.d.ts +49 -0
  4. package/dist/index.mjs +1394 -0
  5. package/dist/lib/abstracts/adk-agent.d.ts +17 -0
  6. package/dist/lib/abstracts/adk-engine.d.ts +6 -0
  7. package/dist/lib/abstracts/adk-skill.d.ts +3 -0
  8. package/dist/lib/abstracts/adk-tool.d.ts +6 -0
  9. package/dist/lib/abstracts/adk-workflow.d.ts +6 -0
  10. package/dist/lib/abstracts/artifact-store.d.ts +10 -0
  11. package/dist/lib/abstracts/embedder.d.ts +13 -0
  12. package/dist/lib/abstracts/memory-store.d.ts +13 -0
  13. package/dist/lib/abstracts/session-store.d.ts +8 -0
  14. package/dist/lib/constants.d.ts +8 -0
  15. package/dist/lib/decorators/agent.decorator.d.ts +2 -0
  16. package/dist/lib/decorators/skill.decorator.d.ts +6 -0
  17. package/dist/lib/decorators/tool.decorator.d.ts +6 -0
  18. package/dist/lib/decorators/workflow-agent.decorator.d.ts +2 -0
  19. package/dist/lib/embeddings/similarity.d.ts +3 -0
  20. package/dist/lib/errors/adk.error.d.ts +4 -0
  21. package/dist/lib/errors/boot.errors.d.ts +43 -0
  22. package/dist/lib/errors/index.d.ts +3 -0
  23. package/dist/lib/errors/runtime.errors.d.ts +51 -0
  24. package/dist/lib/models/context-policy.d.ts +16 -0
  25. package/dist/lib/models/model-specs.d.ts +48 -0
  26. package/dist/lib/module/adk-options.d.ts +24 -0
  27. package/dist/lib/module/adk.module.d.ts +8 -0
  28. package/dist/lib/prompts/adk-prompt.d.ts +8 -0
  29. package/dist/lib/prompts/prompt-files.d.ts +11 -0
  30. package/dist/lib/registry/agent-definition.d.ts +44 -0
  31. package/dist/lib/registry/agent-ref.d.ts +26 -0
  32. package/dist/lib/registry/agent-registry.d.ts +28 -0
  33. package/dist/lib/runner/agent-runner.d.ts +44 -0
  34. package/dist/lib/runner/instruction-builder.d.ts +7 -0
  35. package/dist/lib/runner/run-logger.d.ts +13 -0
  36. package/dist/lib/runner/state-bag.d.ts +9 -0
  37. package/dist/lib/sessions/agent-sessions.d.ts +15 -0
  38. package/dist/lib/stores/in-memory-artifact-store.d.ts +12 -0
  39. package/dist/lib/stores/in-memory-session-store.d.ts +11 -0
  40. package/dist/lib/testing/scripted-engine.d.ts +27 -0
  41. package/dist/lib/testing/scripted-model.d.ts +8 -0
  42. package/dist/lib/types/events.d.ts +113 -0
  43. package/dist/lib/types/options.d.ts +44 -0
  44. package/dist/lib/types/resolved-agent.d.ts +22 -0
  45. package/dist/lib/types/tool-context.d.ts +15 -0
  46. package/dist/lib/types/toolset.d.ts +10 -0
  47. package/package.json +35 -0
@@ -0,0 +1,113 @@
1
+ export interface TokenUsage {
2
+ promptTokens: number;
3
+ outputTokens: number;
4
+ totalTokens: number;
5
+ cachedTokens?: number;
6
+ }
7
+ export interface RawRef {
8
+ event?: unknown;
9
+ response?: unknown;
10
+ }
11
+ export type AgentEvent = {
12
+ type: "run_start";
13
+ agent: string;
14
+ sessionId?: string;
15
+ raw?: RawRef;
16
+ } | {
17
+ type: "llm_response";
18
+ agent: string;
19
+ text?: string;
20
+ usage?: TokenUsage;
21
+ raw?: RawRef;
22
+ } | {
23
+ type: "tool_call";
24
+ agent: string;
25
+ callId: string;
26
+ tool: string;
27
+ args: unknown;
28
+ raw?: RawRef;
29
+ } | {
30
+ type: "tool_result";
31
+ agent: string;
32
+ callId: string;
33
+ tool: string;
34
+ result: unknown;
35
+ raw?: RawRef;
36
+ } | {
37
+ type: "agent_transfer";
38
+ from: string;
39
+ to: string;
40
+ raw?: RawRef;
41
+ } | {
42
+ type: "model_rerouted";
43
+ from: string;
44
+ to: string;
45
+ reason: string;
46
+ raw?: RawRef;
47
+ } | {
48
+ type: "approval_required";
49
+ agent: string;
50
+ callId: string;
51
+ tool: string;
52
+ args: unknown;
53
+ raw?: RawRef;
54
+ } | {
55
+ type: "final";
56
+ agent: string;
57
+ text: string;
58
+ usage: TokenUsage;
59
+ raw?: RawRef;
60
+ };
61
+ export interface RunInput {
62
+ message: string;
63
+ sessionId?: string;
64
+ userId?: string;
65
+ state?: Record<string, unknown>;
66
+ attributes?: Record<string, unknown>;
67
+ labels?: Record<string, string>;
68
+ signal?: AbortSignal;
69
+ history?: SessionEvent[];
70
+ }
71
+ export interface PendingApproval {
72
+ callId: string;
73
+ tool: string;
74
+ args: unknown;
75
+ agent: string;
76
+ }
77
+ export interface RunResult<TOutput = unknown> {
78
+ text: string;
79
+ output?: TOutput;
80
+ usage: TokenUsage;
81
+ events: AgentEvent[];
82
+ status: "completed" | "pending_approval";
83
+ pending?: PendingApproval[];
84
+ }
85
+ export interface SessionEvent {
86
+ v: 1;
87
+ id: string;
88
+ at: number;
89
+ author: "user" | "agent" | "system" | "tool";
90
+ type: string;
91
+ data: Record<string, unknown>;
92
+ }
93
+ export interface Session {
94
+ id: string;
95
+ userId?: string;
96
+ state: Record<string, unknown>;
97
+ events: SessionEvent[];
98
+ createdAt: Date;
99
+ updatedAt: Date;
100
+ }
101
+ export interface SessionInit {
102
+ id?: string;
103
+ userId?: string;
104
+ state?: Record<string, unknown>;
105
+ }
106
+ export interface ArtifactRef {
107
+ sessionId: string;
108
+ name: string;
109
+ }
110
+ export interface ArtifactPart {
111
+ mimeType: string;
112
+ data: string;
113
+ }
@@ -0,0 +1,44 @@
1
+ import type { Type } from "@nestjs/common";
2
+ import type { ZodObject, ZodRawShape } from "zod";
3
+ import type { AdkAgent } from "../abstracts/adk-agent";
4
+ import type { AdkSkill } from "../abstracts/adk-skill";
5
+ import type { AdkTool } from "../abstracts/adk-tool";
6
+ import type { SessionStore } from "../abstracts/session-store";
7
+ import type { ToolContext } from "./tool-context";
8
+ export type ModelInput = string | object;
9
+ export type AnyZodObject = ZodObject<ZodRawShape>;
10
+ export interface AgentOptions {
11
+ name: string;
12
+ description: string;
13
+ model?: ModelInput;
14
+ prompt?: string | Type<import("../prompts/adk-prompt").AdkPrompt>;
15
+ promptFile?: string;
16
+ tools?: Array<Type<AdkTool<AnyZodObject>> | import("./toolset").ToolsetRef>;
17
+ skills?: Type<AdkSkill>[];
18
+ subAgents?: Type<AdkAgent>[];
19
+ session?: Type<SessionStore>;
20
+ output?: ZodObject<ZodRawShape>;
21
+ outputKey?: string;
22
+ context?: import("../models/context-policy").ContextPolicy;
23
+ }
24
+ export interface ToolOptions<S extends AnyZodObject = AnyZodObject> {
25
+ name?: string;
26
+ description: string;
27
+ schema: S;
28
+ requiresApproval?: boolean | ((input: unknown, ctx: ToolContext) => boolean | Promise<boolean>);
29
+ offload?: boolean;
30
+ }
31
+ export type SkillMode = "on-demand" | "always";
32
+ export interface SkillOptions {
33
+ name: string;
34
+ description: string;
35
+ mode?: SkillMode;
36
+ }
37
+ export type WorkflowMode = "sequential" | "parallel" | "loop";
38
+ export interface WorkflowOptions {
39
+ name: string;
40
+ description?: string;
41
+ mode: WorkflowMode;
42
+ agents: Type<AdkAgent | object>[];
43
+ maxIterations?: number;
44
+ }
@@ -0,0 +1,22 @@
1
+ import type { AnyZodObject, ModelInput, WorkflowMode } from "./options";
2
+ export interface ResolvedTool {
3
+ name: string;
4
+ description: string;
5
+ schema: AnyZodObject;
6
+ execute(input: unknown): Promise<unknown>;
7
+ }
8
+ export interface ResolvedAgent {
9
+ name: string;
10
+ description: string;
11
+ instruction?: string;
12
+ model?: ModelInput;
13
+ tools: ResolvedTool[];
14
+ subAgents: ResolvedAgent[];
15
+ workflow?: {
16
+ mode: WorkflowMode;
17
+ agents: ResolvedAgent[];
18
+ maxIterations?: number;
19
+ };
20
+ outputSchema?: AnyZodObject;
21
+ context?: import("../models/context-policy").ContextPolicy;
22
+ }
@@ -0,0 +1,15 @@
1
+ export interface ToolContext {
2
+ agentName: string;
3
+ sessionId?: string;
4
+ userId?: string;
5
+ state: StateBag;
6
+ attributes: Record<string, unknown>;
7
+ signal: AbortSignal;
8
+ actions: {
9
+ endRun(): void;
10
+ };
11
+ }
12
+ export interface StateBag {
13
+ get<T = unknown>(key: string): T | undefined;
14
+ set(key: string, value: unknown): void;
15
+ }
@@ -0,0 +1,10 @@
1
+ import type { ResolvedTool } from "./resolved-agent";
2
+ export interface ToolsetRef {
3
+ readonly __adkToolset: string;
4
+ filter?: string[];
5
+ }
6
+ export declare function toolset(name: string, filter?: string[]): ToolsetRef;
7
+ export declare function isToolsetRef(value: unknown): value is ToolsetRef;
8
+ export declare abstract class ToolsetResolver {
9
+ abstract resolve(ref: ToolsetRef): Promise<ResolvedTool[]>;
10
+ }
package/package.json ADDED
@@ -0,0 +1,35 @@
1
+ {
2
+ "name": "@nestjs-adk/core",
3
+ "version": "0.0.1",
4
+ "description": "NestJS-native agent framework: decorators, DI and abstract contracts over pluggable AI agent engines (Google ADK first).",
5
+ "license": "MIT",
6
+ "sideEffects": false,
7
+ "main": "dist/index.cjs",
8
+ "module": "dist/index.mjs",
9
+ "types": "dist/index.d.ts",
10
+ "exports": {
11
+ ".": {
12
+ "types": "./dist/index.d.ts",
13
+ "import": "./dist/index.mjs",
14
+ "require": "./dist/index.cjs"
15
+ },
16
+ "./package.json": "./package.json"
17
+ },
18
+ "files": ["dist"],
19
+ "engines": {
20
+ "node": ">=20"
21
+ },
22
+ "scripts": {
23
+ "prebuild": "rimraf -g dist",
24
+ "build": "rollup -c"
25
+ },
26
+ "peerDependencies": {
27
+ "@nestjs/common": "^10.0.0 || ^11.0.0",
28
+ "@nestjs/core": "^10.0.0 || ^11.0.0",
29
+ "reflect-metadata": "^0.1.13 || ^0.2.0",
30
+ "zod": "^3.23.0 || ^4.0.0"
31
+ },
32
+ "publishConfig": {
33
+ "access": "public"
34
+ }
35
+ }