@ainetwork/adk 0.1.5 → 0.1.7
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/dist/cjs/chunk-3JKRJYJ6.cjs +111 -0
- package/dist/cjs/chunk-3JKRJYJ6.cjs.map +1 -0
- package/dist/cjs/chunk-466ED5WN.cjs +41 -0
- package/dist/cjs/chunk-466ED5WN.cjs.map +1 -0
- package/dist/cjs/chunk-4ESGHH2M.cjs +20 -0
- package/dist/cjs/chunk-4ESGHH2M.cjs.map +1 -0
- package/dist/cjs/chunk-4HY26I2C.cjs +168 -0
- package/dist/cjs/chunk-4HY26I2C.cjs.map +1 -0
- package/dist/cjs/chunk-5OWM5IH4.cjs +20 -0
- package/dist/cjs/chunk-5OWM5IH4.cjs.map +1 -0
- package/dist/cjs/chunk-6S6GEY3D.cjs +37 -0
- package/dist/cjs/chunk-6S6GEY3D.cjs.map +1 -0
- package/dist/cjs/chunk-7QNBYJ22.cjs +37 -0
- package/dist/cjs/chunk-7QNBYJ22.cjs.map +1 -0
- package/dist/cjs/chunk-DWJF754D.cjs +48 -0
- package/dist/cjs/chunk-DWJF754D.cjs.map +1 -0
- package/dist/cjs/chunk-FGV256AQ.cjs +175 -0
- package/dist/cjs/chunk-FGV256AQ.cjs.map +1 -0
- package/dist/cjs/chunk-GZKCC6VR.cjs +37 -0
- package/dist/cjs/chunk-GZKCC6VR.cjs.map +1 -0
- package/dist/cjs/chunk-ICFTUZP5.cjs +111 -0
- package/dist/cjs/chunk-ICFTUZP5.cjs.map +1 -0
- package/dist/cjs/chunk-K3EK37A7.cjs +28 -0
- package/dist/cjs/chunk-K3EK37A7.cjs.map +1 -0
- package/dist/cjs/chunk-LRBRG2DS.cjs +43 -0
- package/dist/cjs/chunk-LRBRG2DS.cjs.map +1 -0
- package/dist/cjs/chunk-MMWEU2MO.cjs +100 -0
- package/dist/cjs/chunk-MMWEU2MO.cjs.map +1 -0
- package/dist/cjs/chunk-N26HPJTO.cjs +27 -0
- package/dist/cjs/chunk-N26HPJTO.cjs.map +1 -0
- package/dist/cjs/chunk-NYYEZ4ER.cjs +22 -0
- package/dist/cjs/chunk-NYYEZ4ER.cjs.map +1 -0
- package/dist/cjs/chunk-OBUCVZNU.cjs +37 -0
- package/dist/cjs/chunk-OBUCVZNU.cjs.map +1 -0
- package/dist/cjs/chunk-QUWZWKSZ.cjs +166 -0
- package/dist/cjs/chunk-QUWZWKSZ.cjs.map +1 -0
- package/dist/cjs/chunk-QVIS5ELA.cjs +37 -0
- package/dist/cjs/chunk-QVIS5ELA.cjs.map +1 -0
- package/dist/cjs/chunk-RMRFCEP4.cjs +8 -0
- package/dist/cjs/chunk-RMRFCEP4.cjs.map +1 -0
- package/dist/cjs/chunk-RPMERDHU.cjs +44 -0
- package/dist/cjs/chunk-RPMERDHU.cjs.map +1 -0
- package/dist/cjs/chunk-RWQDV2XY.cjs +166 -0
- package/dist/cjs/chunk-RWQDV2XY.cjs.map +1 -0
- package/dist/cjs/chunk-S3RQLMWY.cjs +27 -0
- package/dist/cjs/chunk-S3RQLMWY.cjs.map +1 -0
- package/dist/cjs/chunk-UQ72NNHV.cjs +225 -0
- package/dist/cjs/chunk-UQ72NNHV.cjs.map +1 -0
- package/dist/cjs/chunk-UR5BG625.cjs +27 -0
- package/dist/cjs/chunk-UR5BG625.cjs.map +1 -0
- package/dist/cjs/chunk-VBXOEQ3N.cjs +22 -0
- package/dist/cjs/chunk-VBXOEQ3N.cjs.map +1 -0
- package/dist/cjs/chunk-XZAKXJYO.cjs +20 -0
- package/dist/cjs/chunk-XZAKXJYO.cjs.map +1 -0
- package/dist/cjs/chunk-ZAZ3KYP5.cjs +43 -0
- package/dist/cjs/chunk-ZAZ3KYP5.cjs.map +1 -0
- package/dist/cjs/controllers/a2a.controller.cjs +3 -3
- package/dist/cjs/controllers/api/session.api.controller.cjs +2 -2
- package/dist/cjs/controllers/index.cjs +6 -6
- package/dist/cjs/controllers/index.cjs.map +1 -1
- package/dist/cjs/controllers/query.controller.cjs +2 -2
- package/dist/cjs/index.cjs +26 -25
- package/dist/cjs/index.cjs.map +1 -1
- package/dist/cjs/middlewares/auth.middleware.cjs +2 -2
- package/dist/cjs/middlewares/error.middleware.cjs +3 -3
- package/dist/cjs/modules/a2a/a2a.module.cjs +3 -3
- package/dist/cjs/modules/index.cjs +10 -10
- package/dist/cjs/modules/index.cjs.map +1 -1
- package/dist/cjs/modules/mcp/mcp.module.cjs +3 -3
- package/dist/cjs/modules/models/base.model.cjs +2 -2
- package/dist/cjs/routes/a2a.routes.cjs +7 -7
- package/dist/cjs/routes/api/sessions.routes.cjs +3 -3
- package/dist/cjs/routes/api.routes.cjs +4 -4
- package/dist/cjs/routes/index.cjs +15 -14
- package/dist/cjs/routes/index.cjs.map +1 -1
- package/dist/cjs/routes/query.routes.cjs +7 -6
- package/dist/cjs/routes/query.routes.cjs.map +1 -1
- package/dist/cjs/services/a2a.service.cjs +3 -3
- package/dist/cjs/services/index.cjs +9 -5
- package/dist/cjs/services/index.cjs.map +1 -1
- package/dist/cjs/services/query-stream.service.cjs +9 -0
- package/dist/cjs/services/query-stream.service.cjs.map +1 -0
- package/dist/cjs/services/query.service.cjs +4 -4
- package/dist/cjs/types/memory.cjs +2 -2
- package/dist/cjs/types/stream.cjs +1 -0
- package/dist/cjs/types/stream.cjs.map +1 -0
- package/dist/cjs/utils/logger.cjs +2 -2
- package/dist/esm/{chunk-GZFRRFOE.js → chunk-2BITOA2R.js} +1 -1
- package/dist/esm/chunk-2BITOA2R.js.map +1 -0
- package/dist/esm/{chunk-RB4IFZJA.js → chunk-5EPK37PE.js} +2 -2
- package/dist/esm/{chunk-RB4IFZJA.js.map → chunk-5EPK37PE.js.map} +1 -1
- package/dist/esm/{chunk-BTZVK3EA.js → chunk-624FPEEV.js} +2 -2
- package/dist/esm/{chunk-B3R4A2NL.js → chunk-A3W52LKL.js} +4 -4
- package/dist/esm/{chunk-SQWZXP4P.js → chunk-DJRRMFKH.js} +2 -2
- package/dist/esm/chunk-EFMGNCWG.js +225 -0
- package/dist/esm/chunk-EFMGNCWG.js.map +1 -0
- package/dist/esm/{chunk-EODMRV4H.js → chunk-HAE333YO.js} +2 -2
- package/dist/esm/{chunk-ILFWNK3W.js → chunk-MAU2MAK2.js} +2 -2
- package/dist/esm/{chunk-356AV44C.js → chunk-MC7CPQ6W.js} +13 -21
- package/dist/esm/{chunk-356AV44C.js.map → chunk-MC7CPQ6W.js.map} +1 -1
- package/dist/esm/{chunk-64GEXWSE.js → chunk-N7VYM2R5.js} +2 -1
- package/dist/esm/chunk-N7VYM2R5.js.map +1 -0
- package/dist/esm/{chunk-6BK7LDIA.js → chunk-PN6NMQBM.js} +1 -1
- package/dist/esm/chunk-PN6NMQBM.js.map +1 -0
- package/dist/esm/{chunk-RSSLUA2N.js → chunk-QROUPWWO.js} +2 -2
- package/dist/esm/{chunk-F727XG4N.js → chunk-VAZWF5IK.js} +1 -1
- package/dist/esm/{chunk-F727XG4N.js.map → chunk-VAZWF5IK.js.map} +1 -1
- package/dist/esm/chunk-WGHAZ4HG.js +44 -0
- package/dist/esm/chunk-WGHAZ4HG.js.map +1 -0
- package/dist/esm/{chunk-AQ5YZERE.js → chunk-Y6E73I7I.js} +2 -2
- package/dist/esm/{chunk-NSKXMJWE.js → chunk-YT6PD6UF.js} +4 -4
- package/dist/esm/chunk-ZFSIP5KN.js +43 -0
- package/dist/esm/chunk-ZFSIP5KN.js.map +1 -0
- package/dist/esm/controllers/a2a.controller.d.ts +1 -0
- package/dist/esm/controllers/a2a.controller.js +2 -2
- package/dist/esm/controllers/api/model.api.controller.d.ts +1 -0
- package/dist/esm/controllers/api/session.api.controller.js +1 -1
- package/dist/esm/controllers/index.d.ts +2 -0
- package/dist/esm/controllers/index.js +4 -4
- package/dist/esm/controllers/query.controller.d.ts +5 -1
- package/dist/esm/controllers/query.controller.js +1 -1
- package/dist/esm/index.d.ts +2 -1
- package/dist/esm/index.js +19 -18
- package/dist/esm/index.js.map +1 -1
- package/dist/esm/middlewares/auth.middleware.js +1 -1
- package/dist/esm/middlewares/error.middleware.js +2 -2
- package/dist/esm/modules/a2a/a2a.module.js +2 -2
- package/dist/esm/modules/index.d.ts +1 -0
- package/dist/esm/modules/index.js +10 -10
- package/dist/esm/modules/mcp/mcp.module.js +2 -2
- package/dist/esm/modules/memory/base.memory.d.ts +6 -6
- package/dist/esm/modules/models/base.model.d.ts +13 -0
- package/dist/esm/modules/models/base.model.js +1 -1
- package/dist/esm/modules/models/model.module.d.ts +1 -0
- package/dist/esm/routes/a2a.routes.d.ts +1 -0
- package/dist/esm/routes/a2a.routes.js +6 -6
- package/dist/esm/routes/api/models.routes.d.ts +1 -0
- package/dist/esm/routes/api/sessions.routes.js +2 -2
- package/dist/esm/routes/api.routes.d.ts +1 -0
- package/dist/esm/routes/api.routes.js +3 -3
- package/dist/esm/routes/index.d.ts +1 -0
- package/dist/esm/routes/index.js +14 -13
- package/dist/esm/routes/query.routes.d.ts +2 -1
- package/dist/esm/routes/query.routes.js +6 -5
- package/dist/esm/services/a2a.service.d.ts +1 -0
- package/dist/esm/services/a2a.service.js +2 -2
- package/dist/esm/services/index.d.ts +3 -0
- package/dist/esm/services/index.js +9 -5
- package/dist/esm/services/query-stream.service.d.ts +74 -0
- package/dist/esm/services/query-stream.service.js +9 -0
- package/dist/esm/services/query-stream.service.js.map +1 -0
- package/dist/esm/services/query.service.d.ts +1 -0
- package/dist/esm/services/query.service.js +3 -3
- package/dist/esm/types/memory.d.ts +9 -1
- package/dist/esm/types/memory.js +1 -1
- package/dist/esm/types/stream.d.ts +47 -0
- package/dist/esm/types/stream.js +1 -0
- package/dist/esm/types/stream.js.map +1 -0
- package/dist/esm/utils/logger.d.ts +1 -0
- package/dist/esm/utils/logger.js +1 -1
- package/package.json +1 -1
- package/dist/esm/chunk-64GEXWSE.js.map +0 -1
- package/dist/esm/chunk-6BK7LDIA.js.map +0 -1
- package/dist/esm/chunk-GZFRRFOE.js.map +0 -1
- package/dist/esm/chunk-O5OBYREI.js +0 -27
- package/dist/esm/chunk-O5OBYREI.js.map +0 -1
- package/dist/esm/chunk-WIJDRP2A.js +0 -26
- package/dist/esm/chunk-WIJDRP2A.js.map +0 -1
- /package/dist/esm/{chunk-BTZVK3EA.js.map → chunk-624FPEEV.js.map} +0 -0
- /package/dist/esm/{chunk-B3R4A2NL.js.map → chunk-A3W52LKL.js.map} +0 -0
- /package/dist/esm/{chunk-SQWZXP4P.js.map → chunk-DJRRMFKH.js.map} +0 -0
- /package/dist/esm/{chunk-EODMRV4H.js.map → chunk-HAE333YO.js.map} +0 -0
- /package/dist/esm/{chunk-ILFWNK3W.js.map → chunk-MAU2MAK2.js.map} +0 -0
- /package/dist/esm/{chunk-RSSLUA2N.js.map → chunk-QROUPWWO.js.map} +0 -0
- /package/dist/esm/{chunk-AQ5YZERE.js.map → chunk-Y6E73I7I.js.map} +0 -0
- /package/dist/esm/{chunk-NSKXMJWE.js.map → chunk-YT6PD6UF.js.map} +0 -0
|
@@ -0,0 +1,175 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; }
|
|
2
|
+
|
|
3
|
+
var _chunkJTAXZJQ2cjs = require('./chunk-JTAXZJQ2.cjs');
|
|
4
|
+
|
|
5
|
+
// src/services/query.service.ts
|
|
6
|
+
var QueryService = class {
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
|
|
10
|
+
|
|
11
|
+
|
|
12
|
+
constructor(modelModule, a2aModule, mcpModule, memoryModule, prompts) {
|
|
13
|
+
this.modelModule = modelModule;
|
|
14
|
+
this.a2aModule = a2aModule;
|
|
15
|
+
this.mcpModule = mcpModule;
|
|
16
|
+
this.memoryModule = memoryModule;
|
|
17
|
+
this.prompts = prompts;
|
|
18
|
+
}
|
|
19
|
+
/**
|
|
20
|
+
* Detects the intent from a user query.
|
|
21
|
+
*
|
|
22
|
+
* @param query - The user's input query
|
|
23
|
+
* @returns The detected intent (currently returns the query as-is)
|
|
24
|
+
* @todo Implement actual intent detection logic
|
|
25
|
+
*/
|
|
26
|
+
async intentTriggering(query) {
|
|
27
|
+
return query;
|
|
28
|
+
}
|
|
29
|
+
/**
|
|
30
|
+
* Fulfills the detected intent by generating a response.
|
|
31
|
+
*
|
|
32
|
+
* Manages the complete inference loop including:
|
|
33
|
+
* - Loading prompts and conversation history
|
|
34
|
+
* - Collecting available tools from modules
|
|
35
|
+
* - Executing model inference with tool support
|
|
36
|
+
* - Processing tool calls iteratively until completion
|
|
37
|
+
*
|
|
38
|
+
* @param query - The user's input query
|
|
39
|
+
* @param sessionId - Session identifier for context
|
|
40
|
+
* @param sessionHistory - Previous conversation history
|
|
41
|
+
* @returns Object containing process steps and final response
|
|
42
|
+
*/
|
|
43
|
+
async intentFulfilling(query, sessionId, sessionHistory) {
|
|
44
|
+
const systemPrompt = `
|
|
45
|
+
Today is ${(/* @__PURE__ */ new Date()).toLocaleDateString()}.
|
|
46
|
+
|
|
47
|
+
${_optionalChain([this, 'access', _ => _.prompts, 'optionalAccess', _2 => _2.agent]) || ""}
|
|
48
|
+
|
|
49
|
+
${_optionalChain([this, 'access', _3 => _3.prompts, 'optionalAccess', _4 => _4.system]) || ""}
|
|
50
|
+
`;
|
|
51
|
+
const modelInstance = this.modelModule.getModel();
|
|
52
|
+
const messages = modelInstance.generateMessages({
|
|
53
|
+
query,
|
|
54
|
+
sessionHistory,
|
|
55
|
+
systemPrompt: systemPrompt.trim()
|
|
56
|
+
});
|
|
57
|
+
const tools = [];
|
|
58
|
+
if (this.mcpModule) {
|
|
59
|
+
tools.push(...this.mcpModule.getTools());
|
|
60
|
+
}
|
|
61
|
+
if (this.a2aModule) {
|
|
62
|
+
tools.push(...await this.a2aModule.getTools());
|
|
63
|
+
}
|
|
64
|
+
const functions = modelInstance.convertToolsToFunctions(tools);
|
|
65
|
+
const processList = [];
|
|
66
|
+
let finalMessage = "";
|
|
67
|
+
let didCallTool = false;
|
|
68
|
+
while (true) {
|
|
69
|
+
const response = await modelInstance.fetchWithContextMessage(
|
|
70
|
+
messages,
|
|
71
|
+
functions
|
|
72
|
+
);
|
|
73
|
+
didCallTool = false;
|
|
74
|
+
_chunkJTAXZJQ2cjs.loggers.intent.debug("messages", { messages });
|
|
75
|
+
const { content, toolCalls } = response;
|
|
76
|
+
_chunkJTAXZJQ2cjs.loggers.intent.debug("content", { content });
|
|
77
|
+
_chunkJTAXZJQ2cjs.loggers.intent.debug("tool_calls", { ...toolCalls });
|
|
78
|
+
if (toolCalls) {
|
|
79
|
+
const messagePayload = _optionalChain([this, 'access', _5 => _5.a2aModule, 'optionalAccess', _6 => _6.getMessagePayload, 'call', _7 => _7(
|
|
80
|
+
query,
|
|
81
|
+
sessionId
|
|
82
|
+
)]);
|
|
83
|
+
for (const toolCall of toolCalls) {
|
|
84
|
+
const toolName = toolCall.name;
|
|
85
|
+
didCallTool = true;
|
|
86
|
+
const selectedTool = tools.filter((tool) => tool.id === toolName)[0];
|
|
87
|
+
let toolResult = "";
|
|
88
|
+
if (this.mcpModule && selectedTool.protocol === "MCP" /* MCP */) {
|
|
89
|
+
const toolArgs = toolCall.arguments;
|
|
90
|
+
_chunkJTAXZJQ2cjs.loggers.intent.debug("MCP tool call", { toolName, toolArgs });
|
|
91
|
+
const result = await this.mcpModule.useTool(
|
|
92
|
+
selectedTool,
|
|
93
|
+
toolArgs
|
|
94
|
+
);
|
|
95
|
+
toolResult = `[Bot Called MCP Tool ${toolName} with args ${JSON.stringify(toolArgs)}]
|
|
96
|
+
` + JSON.stringify(result.content, null, 2);
|
|
97
|
+
} else if (this.a2aModule && selectedTool.protocol === "A2A" /* A2A */) {
|
|
98
|
+
const result = await this.a2aModule.useTool(
|
|
99
|
+
selectedTool,
|
|
100
|
+
messagePayload,
|
|
101
|
+
sessionId
|
|
102
|
+
);
|
|
103
|
+
toolResult = `[Bot Called A2A Tool ${toolName}]
|
|
104
|
+
${result.join("\n")}`;
|
|
105
|
+
} else {
|
|
106
|
+
_chunkJTAXZJQ2cjs.loggers.intent.warn(
|
|
107
|
+
`Unrecognized tool type: ${selectedTool.protocol}`
|
|
108
|
+
);
|
|
109
|
+
continue;
|
|
110
|
+
}
|
|
111
|
+
_chunkJTAXZJQ2cjs.loggers.intent.debug("toolResult", { toolResult });
|
|
112
|
+
processList.push(toolResult);
|
|
113
|
+
modelInstance.appendMessages(messages, toolResult);
|
|
114
|
+
}
|
|
115
|
+
} else if (content) {
|
|
116
|
+
processList.push(content);
|
|
117
|
+
finalMessage = content;
|
|
118
|
+
}
|
|
119
|
+
if (!didCallTool) break;
|
|
120
|
+
}
|
|
121
|
+
const botResponse = {
|
|
122
|
+
process: processList.join("\n"),
|
|
123
|
+
response: finalMessage
|
|
124
|
+
};
|
|
125
|
+
return botResponse;
|
|
126
|
+
}
|
|
127
|
+
/**
|
|
128
|
+
* Main entry point for processing user queries.
|
|
129
|
+
*
|
|
130
|
+
* Handles the complete query lifecycle:
|
|
131
|
+
* 1. Loads session history from memory
|
|
132
|
+
* 2. Detects intent from the query
|
|
133
|
+
* 3. Fulfills the intent with AI response
|
|
134
|
+
* 4. Updates conversation history
|
|
135
|
+
*
|
|
136
|
+
* @param query - The user's input query
|
|
137
|
+
* @param sessionId - Unique session identifier
|
|
138
|
+
* @param userId - Unique user identifier
|
|
139
|
+
* @returns Object containing the AI-generated response
|
|
140
|
+
*/
|
|
141
|
+
async handleQuery(query, sessionId, userId) {
|
|
142
|
+
console.log("cccccccccccccccccccccccc");
|
|
143
|
+
const queryStartAt = Date.now();
|
|
144
|
+
const sessionMemory = _optionalChain([this, 'access', _8 => _8.memoryModule, 'optionalAccess', _9 => _9.getSessionMemory, 'call', _10 => _10()]);
|
|
145
|
+
const session = !userId ? void 0 : await _optionalChain([sessionMemory, 'optionalAccess', _11 => _11.getSession, 'call', _12 => _12(sessionId, userId)]);
|
|
146
|
+
const intent = this.intentTriggering(query);
|
|
147
|
+
const result = await this.intentFulfilling(query, sessionId, session);
|
|
148
|
+
if (userId) {
|
|
149
|
+
await _optionalChain([sessionMemory, 'optionalAccess', _13 => _13.addChatToSession, 'call', _14 => _14(
|
|
150
|
+
sessionId,
|
|
151
|
+
{
|
|
152
|
+
role: "USER" /* USER */,
|
|
153
|
+
timestamp: queryStartAt,
|
|
154
|
+
content: { type: "text", parts: [query] }
|
|
155
|
+
},
|
|
156
|
+
userId
|
|
157
|
+
)]);
|
|
158
|
+
await _optionalChain([sessionMemory, 'optionalAccess', _15 => _15.addChatToSession, 'call', _16 => _16(
|
|
159
|
+
sessionId,
|
|
160
|
+
{
|
|
161
|
+
role: "MODEL" /* MODEL */,
|
|
162
|
+
timestamp: Date.now(),
|
|
163
|
+
content: { type: "text", parts: [result.response] }
|
|
164
|
+
},
|
|
165
|
+
userId
|
|
166
|
+
)]);
|
|
167
|
+
}
|
|
168
|
+
return { content: result.response };
|
|
169
|
+
}
|
|
170
|
+
};
|
|
171
|
+
|
|
172
|
+
|
|
173
|
+
|
|
174
|
+
exports.QueryService = QueryService;
|
|
175
|
+
//# sourceMappingURL=chunk-FGV256AQ.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-FGV256AQ.cjs","../../src/services/query.service.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACmBO,IAAM,aAAA,EAAN,MAAmB;AAAA,EACjB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,WAAA,CACC,WAAA,EACA,SAAA,EACA,SAAA,EACA,YAAA,EACA,OAAA,EACC;AACD,IAAA,IAAA,CAAK,YAAA,EAAc,WAAA;AACnB,IAAA,IAAA,CAAK,UAAA,EAAY,SAAA;AACjB,IAAA,IAAA,CAAK,UAAA,EAAY,SAAA;AACjB,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AACpB,IAAA,IAAA,CAAK,QAAA,EAAU,OAAA;AAAA,EAChB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,gBAAA,CAAiB,KAAA,EAAe;AAE7C,IAAA,OAAO,KAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAgBA,MAAc,gBAAA,CACb,KAAA,EACA,SAAA,EACA,cAAA,EACC;AAED,IAAA,MAAM,aAAA,EAAe,CAAA;AAAA,SAAA,EAAA,iBACZ,IAAI,IAAA,CAAK,CAAA,CAAA,CAAE,kBAAA,CAAmB,CAAC,CAAA;AAAA;AAAA,kBAExC,IAAA,mBAAK,OAAA,6BAAS,QAAA,GAAS,EAAE,CAAA;AAAA;AAAA,kBAEzB,IAAA,qBAAK,OAAA,6BAAS,SAAA,GAAU,EAAE,CAAA;AAAA,IAAA,CAAA;AAG1B,IAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,WAAA,CAAY,QAAA,CAAS,CAAA;AAChD,IAAA,MAAM,SAAA,EAAW,aAAA,CAAc,gBAAA,CAAiB;AAAA,MAC/C,KAAA;AAAA,MACA,cAAA;AAAA,MACA,YAAA,EAAc,YAAA,CAAa,IAAA,CAAK;AAAA,IACjC,CAAC,CAAA;AAED,IAAA,MAAM,MAAA,EAAsB,CAAC,CAAA;AAC7B,IAAA,GAAA,CAAI,IAAA,CAAK,SAAA,EAAW;AACnB,MAAA,KAAA,CAAM,IAAA,CAAK,GAAG,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAC,CAAA;AAAA,IACxC;AACA,IAAA,GAAA,CAAI,IAAA,CAAK,SAAA,EAAW;AACnB,MAAA,KAAA,CAAM,IAAA,CAAK,GAAI,MAAM,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAE,CAAA;AAAA,IAChD;AACA,IAAA,MAAM,UAAA,EAAY,aAAA,CAAc,uBAAA,CAAwB,KAAK,CAAA;AAE7D,IAAA,MAAM,YAAA,EAAwB,CAAC,CAAA;AAC/B,IAAA,IAAI,aAAA,EAAe,EAAA;AACnB,IAAA,IAAI,YAAA,EAAc,KAAA;AAElB,IAAA,MAAA,CAAO,IAAA,EAAM;AACZ,MAAA,MAAM,SAAA,EAAW,MAAM,aAAA,CAAc,uBAAA;AAAA,QACpC,QAAA;AAAA,QACA;AAAA,MACD,CAAA;AACA,MAAA,YAAA,EAAc,KAAA;AAEd,MAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,UAAA,EAAY,EAAE,SAAS,CAAC,CAAA;AAE7C,MAAA,MAAM,EAAE,OAAA,EAAS,UAAU,EAAA,EAAI,QAAA;AAE/B,MAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,SAAA,EAAW,EAAE,QAAQ,CAAC,CAAA;AAC3C,MAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,YAAA,EAAc,EAAE,GAAG,UAAU,CAAC,CAAA;AAEnD,MAAA,GAAA,CAAI,SAAA,EAAW;AACd,QAAA,MAAM,eAAA,kBAAiB,IAAA,qBAAK,SAAA,6BAAW,iBAAA;AAAA,UACtC,KAAA;AAAA,UACA;AAAA,QACD,GAAA;AAEA,QAAA,IAAA,CAAA,MAAW,SAAA,GAAY,SAAA,EAAW;AACjC,UAAA,MAAM,SAAA,EAAW,QAAA,CAAS,IAAA;AAC1B,UAAA,YAAA,EAAc,IAAA;AACd,UAAA,MAAM,aAAA,EAAe,KAAA,CAAM,MAAA,CAAO,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,GAAA,IAAO,QAAQ,CAAA,CAAE,CAAC,CAAA;AAEnE,UAAA,IAAI,WAAA,EAAa,EAAA;AACjB,UAAA,GAAA,CACC,IAAA,CAAK,UAAA,GACL,YAAA,CAAa,SAAA,IAAA,eAAA,EACZ;AACD,YAAA,MAAM,SAAA,EAAW,QAAA,CAAS,SAAA;AAG1B,YAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,eAAA,EAAiB,EAAE,QAAA,EAAU,SAAS,CAAC,CAAA;AAC5D,YAAA,MAAM,OAAA,EAAS,MAAM,IAAA,CAAK,SAAA,CAAU,OAAA;AAAA,cACnC,YAAA;AAAA,cACA;AAAA,YACD,CAAA;AACA,YAAA,WAAA,EACC,CAAA,qBAAA,EAAwB,QAAQ,CAAA,WAAA,EAAc,IAAA,CAAK,SAAA,CAAU,QAAQ,CAAC,CAAA;AAAA,EAAA,EACtE,IAAA,CAAK,SAAA,CAAU,MAAA,CAAO,OAAA,EAAS,IAAA,EAAM,CAAC,CAAA;AAAA,UACxC,EAAA,KAAA,GAAA,CACC,IAAA,CAAK,UAAA,GACL,YAAA,CAAa,SAAA,IAAA,eAAA,EACZ;AACD,YAAA,MAAM,OAAA,EAAS,MAAM,IAAA,CAAK,SAAA,CAAU,OAAA;AAAA,cACnC,YAAA;AAAA,cACA,cAAA;AAAA,cACA;AAAA,YACD,CAAA;AACA,YAAA,WAAA,EAAa,CAAA,qBAAA,EAAwB,QAAQ,CAAA;AAAA,EAAM,MAAA,CAAO,IAAA,CAAK,IAAI,CAAC,CAAA,CAAA;AAC9D,UAAA;AAEE,YAAA;AACP,cAAA;AACD,YAAA;AACA,YAAA;AACD,UAAA;AAEQ,UAAA;AAER,UAAA;AACA,UAAA;AACD,QAAA;AACU,MAAA;AACE,QAAA;AACZ,QAAA;AACD,MAAA;AAEK,MAAA;AACN,IAAA;AAEM,IAAA;AACI,MAAA;AACC,MAAA;AACX,IAAA;AAEO,IAAA;AACR,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAgByB,EAAA;AAEZ,IAAA;AACN,IAAA;AACA,IAAA;AACU,IAAA;AAKD,IAAA;AAGA,IAAA;AACH,IAAA;AACL,MAAA;AACL,QAAA;AACA,QAAA;AACC,UAAA;AACA,UAAA;AACS,UAAA;AACV,QAAA;AACA,QAAA;AACD,MAAA;AACM,MAAA;AACL,QAAA;AACA,QAAA;AACC,UAAA;AACA,UAAA;AACS,UAAA;AACV,QAAA;AACA,QAAA;AACD,MAAA;AACD,IAAA;AAES,IAAA;AACV,EAAA;AACD;ADjEoB;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-FGV256AQ.cjs","sourcesContent":[null,"import type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport type { AinAgentPrompts } from \"@/types/agent.js\";\nimport { ChatRole, type SessionObject } from \"@/types/memory.js\";\nimport {\n\ttype IA2ATool,\n\ttype IAgentTool,\n\ttype IMCPTool,\n\tTOOL_PROTOCOL_TYPE,\n} from \"@/types/tool.js\";\nimport { loggers } from \"@/utils/logger.js\";\n\n/**\n * Service for processing user queries through the agent's AI pipeline.\n *\n * Orchestrates the query processing workflow including intent detection,\n * model inference, tool execution, and response generation. Manages\n * conversation context and coordinates between different modules.\n */\nexport class QueryService {\n\tprivate modelModule: ModelModule;\n\tprivate a2aModule?: A2AModule;\n\tprivate mcpModule?: MCPModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate prompts?: AinAgentPrompts;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t\tprompts?: AinAgentPrompts,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.a2aModule = a2aModule;\n\t\tthis.mcpModule = mcpModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.prompts = prompts;\n\t}\n\n\t/**\n\t * Detects the intent from a user query.\n\t *\n\t * @param query - The user's input query\n\t * @returns The detected intent (currently returns the query as-is)\n\t * @todo Implement actual intent detection logic\n\t */\n\tprivate async intentTriggering(query: string) {\n\t\t/* TODO */\n\t\treturn query;\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a response.\n\t *\n\t * Manages the complete inference loop including:\n\t * - Loading prompts and conversation history\n\t * - Collecting available tools from modules\n\t * - Executing model inference with tool support\n\t * - Processing tool calls iteratively until completion\n\t *\n\t * @param query - The user's input query\n\t * @param sessionId - Session identifier for context\n\t * @param sessionHistory - Previous conversation history\n\t * @returns Object containing process steps and final response\n\t */\n\tprivate async intentFulfilling(\n\t\tquery: string,\n\t\tsessionId: string,\n\t\tsessionHistory: SessionObject | undefined,\n\t) {\n\t\t// 1. Load agent / system prompt from memory\n\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n `;\n\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery,\n\t\t\tsessionHistory,\n\t\t\tsystemPrompt: systemPrompt.trim(),\n\t\t});\n\n\t\tconst tools: IAgentTool[] = [];\n\t\tif (this.mcpModule) {\n\t\t\ttools.push(...this.mcpModule.getTools());\n\t\t}\n\t\tif (this.a2aModule) {\n\t\t\ttools.push(...(await this.a2aModule.getTools()));\n\t\t}\n\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\n\t\tconst processList: string[] = [];\n\t\tlet finalMessage = \"\";\n\t\tlet didCallTool = false;\n\n\t\twhile (true) {\n\t\t\tconst response = await modelInstance.fetchWithContextMessage(\n\t\t\t\tmessages,\n\t\t\t\tfunctions,\n\t\t\t);\n\t\t\tdidCallTool = false;\n\n\t\t\tloggers.intent.debug(\"messages\", { messages });\n\n\t\t\tconst { content, toolCalls } = response;\n\n\t\t\tloggers.intent.debug(\"content\", { content });\n\t\t\tloggers.intent.debug(\"tool_calls\", { ...toolCalls });\n\n\t\t\tif (toolCalls) {\n\t\t\t\tconst messagePayload = this.a2aModule?.getMessagePayload(\n\t\t\t\t\tquery,\n\t\t\t\t\tsessionId,\n\t\t\t\t);\n\n\t\t\t\tfor (const toolCall of toolCalls) {\n\t\t\t\t\tconst toolName = toolCall.name;\n\t\t\t\t\tdidCallTool = true;\n\t\t\t\t\tconst selectedTool = tools.filter((tool) => tool.id === toolName)[0];\n\n\t\t\t\t\tlet toolResult = \"\";\n\t\t\t\t\tif (\n\t\t\t\t\t\tthis.mcpModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.MCP\n\t\t\t\t\t) {\n\t\t\t\t\t\tconst toolArgs = toolCall.arguments as\n\t\t\t\t\t\t\t| { [x: string]: unknown }\n\t\t\t\t\t\t\t| undefined;\n\t\t\t\t\t\tloggers.intent.debug(\"MCP tool call\", { toolName, toolArgs });\n\t\t\t\t\t\tconst result = await this.mcpModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IMCPTool,\n\t\t\t\t\t\t\ttoolArgs,\n\t\t\t\t\t\t);\n\t\t\t\t\t\ttoolResult =\n\t\t\t\t\t\t\t`[Bot Called MCP Tool ${toolName} with args ${JSON.stringify(toolArgs)}]\\n` +\n\t\t\t\t\t\t\tJSON.stringify(result.content, null, 2);\n\t\t\t\t\t} else if (\n\t\t\t\t\t\tthis.a2aModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.A2A\n\t\t\t\t\t) {\n\t\t\t\t\t\tconst result = await this.a2aModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IA2ATool,\n\t\t\t\t\t\t\tmessagePayload!,\n\t\t\t\t\t\t\tsessionId,\n\t\t\t\t\t\t);\n\t\t\t\t\t\ttoolResult = `[Bot Called A2A Tool ${toolName}]\\n${result.join(\"\\n\")}`;\n\t\t\t\t\t} else {\n\t\t\t\t\t\t// Unrecognized tool type. It cannot be happened...\n\t\t\t\t\t\tloggers.intent.warn(\n\t\t\t\t\t\t\t`Unrecognized tool type: ${selectedTool.protocol}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\n\t\t\t\t\tloggers.intent.debug(\"toolResult\", { toolResult });\n\n\t\t\t\t\tprocessList.push(toolResult);\n\t\t\t\t\tmodelInstance.appendMessages(messages, toolResult);\n\t\t\t\t}\n\t\t\t} else if (content) {\n\t\t\t\tprocessList.push(content);\n\t\t\t\tfinalMessage = content;\n\t\t\t}\n\n\t\t\tif (!didCallTool) break;\n\t\t}\n\n\t\tconst botResponse = {\n\t\t\tprocess: processList.join(\"\\n\"),\n\t\t\tresponse: finalMessage,\n\t\t};\n\n\t\treturn botResponse;\n\t}\n\n\t/**\n\t * Main entry point for processing user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads session history from memory\n\t * 2. Detects intent from the query\n\t * 3. Fulfills the intent with AI response\n\t * 4. Updates conversation history\n\t *\n\t * @param query - The user's input query\n\t * @param sessionId - Unique session identifier\n\t * @param userId - Unique user identifier\n\t * @returns Object containing the AI-generated response\n\t */\n\tpublic async handleQuery(query: string, sessionId: string, userId?: string) {\n\t\t// 1. Load session history with sessionId\n\t\tconsole.log('cccccccccccccccccccccccc');\n\t\tconst queryStartAt = Date.now();\n\t\tconst sessionMemory = this.memoryModule?.getSessionMemory();\n\t\tconst session = !userId\n\t\t\t? undefined\n\t\t\t: await sessionMemory?.getSession(sessionId, userId);\n\n\t\t// 2. intent triggering\n\t\tconst intent = this.intentTriggering(query);\n\n\t\t// 3. intent fulfillment\n\t\tconst result = await this.intentFulfilling(query, sessionId, session);\n\t\tif (userId) {\n\t\t\tawait sessionMemory?.addChatToSession(\n\t\t\t\tsessionId,\n\t\t\t\t{\n\t\t\t\t\trole: ChatRole.USER,\n\t\t\t\t\ttimestamp: queryStartAt,\n\t\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t\t},\n\t\t\t\tuserId,\n\t\t\t);\n\t\t\tawait sessionMemory?.addChatToSession(\n\t\t\t\tsessionId,\n\t\t\t\t{\n\t\t\t\t\trole: ChatRole.MODEL,\n\t\t\t\t\ttimestamp: Date.now(),\n\t\t\t\t\tcontent: { type: \"text\", parts: [result.response] },\n\t\t\t\t},\n\t\t\t\tuserId,\n\t\t\t);\n\t\t}\n\n\t\treturn { content: result.response };\n\t}\n}\n"]}
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});
|
|
2
|
+
|
|
3
|
+
var _chunk7BZO57TGcjs = require('./chunk-7BZO57TG.cjs');
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
var _chunkQUWZWKSZcjs = require('./chunk-QUWZWKSZ.cjs');
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
var _chunkILGIEJCXcjs = require('./chunk-ILGIEJCX.cjs');
|
|
10
|
+
|
|
11
|
+
// src/routes/a2a.routes.ts
|
|
12
|
+
var _server = require('@a2a-js/sdk/server');
|
|
13
|
+
var _express = require('express');
|
|
14
|
+
var createA2ARouter = (agent) => {
|
|
15
|
+
const router = _express.Router.call(void 0, );
|
|
16
|
+
const taskStore = new (0, _server.InMemoryTaskStore)();
|
|
17
|
+
const queryService = new (0, _chunkQUWZWKSZcjs.QueryService)(
|
|
18
|
+
agent.modelModule,
|
|
19
|
+
agent.a2aModule,
|
|
20
|
+
agent.mcpModule,
|
|
21
|
+
agent.memoryModule,
|
|
22
|
+
agent.manifest.prompts
|
|
23
|
+
);
|
|
24
|
+
const a2aService = new (0, _chunk7BZO57TGcjs.A2AService)(queryService);
|
|
25
|
+
const a2aController = new (0, _chunkILGIEJCXcjs.A2AController)(
|
|
26
|
+
a2aService,
|
|
27
|
+
taskStore,
|
|
28
|
+
agent.generateAgentCard
|
|
29
|
+
);
|
|
30
|
+
router.post("/a2a", a2aController.handleA2ARequest);
|
|
31
|
+
return router;
|
|
32
|
+
};
|
|
33
|
+
|
|
34
|
+
|
|
35
|
+
|
|
36
|
+
exports.createA2ARouter = createA2ARouter;
|
|
37
|
+
//# sourceMappingURL=chunk-GZKCC6VR.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-GZKCC6VR.cjs","../../src/routes/a2a.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACVA,4CAAkC;AAClC,kCAAuB;AAchB,IAAM,gBAAA,EAAkB,CAAC,KAAA,EAAA,GAA4B;AAC3D,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAGtB,EAAA,MAAM,UAAA,EAAY,IAAI,8BAAA,CAAkB,CAAA;AACxC,EAAA,MAAM,aAAA,EAAe,IAAI,mCAAA;AAAA,IACxB,KAAA,CAAM,WAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,YAAA;AAAA,IACN,KAAA,CAAM,QAAA,CAAS;AAAA,EAChB,CAAA;AACA,EAAA,MAAM,WAAA,EAAa,IAAI,iCAAA,CAAW,YAAY,CAAA;AAC9C,EAAA,MAAM,cAAA,EAAgB,IAAI,oCAAA;AAAA,IACzB,UAAA;AAAA,IACA,SAAA;AAAA,IACA,KAAA,CAAM;AAAA,EACP,CAAA;AAGA,EAAA,MAAA,CAAO,IAAA,CAAK,MAAA,EAAQ,aAAA,CAAc,gBAAgB,CAAA;AAElD,EAAA,OAAO,MAAA;AACR,CAAA;ADNA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-GZKCC6VR.cjs","sourcesContent":[null,"import { InMemoryTaskStore } from \"@a2a-js/sdk/server\";\nimport { Router } from \"express\";\nimport type { AINAgent } from \"@/index.js\";\nimport { QueryService } from \"@/services/query.service.js\";\nimport { A2AController } from \"../controllers/a2a.controller.js\";\nimport { A2AService } from \"../services/a2a.service.js\";\n\n/**\n * Creates and configures the A2A router.\n * This function is a \"composition root\" for the A2A feature,\n * creating and injecting all necessary dependencies.\n * @param intentAnalyzer The core intent analyzer.\n * @param agentCard The agent's card.\n * @returns An Express Router instance.\n */\nexport const createA2ARouter = (agent: AINAgent): Router => {\n\tconst router = Router();\n\n\t// 1. Create dependencies for the A2A feature\n\tconst taskStore = new InMemoryTaskStore();\n\tconst queryService = new QueryService(\n\t\tagent.modelModule,\n\t\tagent.a2aModule,\n\t\tagent.mcpModule,\n\t\tagent.memoryModule,\n\t\tagent.manifest.prompts,\n\t);\n\tconst a2aService = new A2AService(queryService);\n\tconst a2aController = new A2AController(\n\t\ta2aService,\n\t\ttaskStore,\n\t\tagent.generateAgentCard,\n\t);\n\n\t// 2. Define the route\n\trouter.post(\"/a2a\", a2aController.handleA2ARequest);\n\n\treturn router;\n};\n"]}
|
|
@@ -0,0 +1,111 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } var _class;
|
|
2
|
+
|
|
3
|
+
var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
|
|
4
|
+
|
|
5
|
+
// src/controllers/a2a.controller.ts
|
|
6
|
+
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
|
|
10
|
+
|
|
11
|
+
var _server = require('@a2a-js/sdk/server');
|
|
12
|
+
var AINRequestHandler = class extends _server.DefaultRequestHandler {
|
|
13
|
+
|
|
14
|
+
constructor(cardGenerator, taskStore, executor, eventBusManager = new (0, _server.DefaultExecutionEventBusManager)()) {
|
|
15
|
+
const card = cardGenerator();
|
|
16
|
+
super(card, taskStore, executor, eventBusManager);
|
|
17
|
+
this.cardGenerator = cardGenerator;
|
|
18
|
+
}
|
|
19
|
+
async getAgentCard() {
|
|
20
|
+
return this.cardGenerator();
|
|
21
|
+
}
|
|
22
|
+
};
|
|
23
|
+
var A2AController = (_class = class {
|
|
24
|
+
|
|
25
|
+
constructor(a2aService, taskStore, cardGenerator) {;_class.prototype.__init.call(this);
|
|
26
|
+
const requestHandler = new AINRequestHandler(
|
|
27
|
+
cardGenerator,
|
|
28
|
+
taskStore,
|
|
29
|
+
a2aService
|
|
30
|
+
// Injecting the service here as the executor
|
|
31
|
+
);
|
|
32
|
+
this.jsonRpcTransportHandler = new (0, _server.JsonRpcTransportHandler)(requestHandler);
|
|
33
|
+
}
|
|
34
|
+
/**
|
|
35
|
+
* Handles the POST /a2a request.
|
|
36
|
+
*/
|
|
37
|
+
__init() {this.handleA2ARequest = async (req, res) => {
|
|
38
|
+
try {
|
|
39
|
+
const rpcResponseOrStream = await this.jsonRpcTransportHandler.handle(
|
|
40
|
+
req.body
|
|
41
|
+
);
|
|
42
|
+
if (typeof _optionalChain([rpcResponseOrStream, 'optionalAccess', _2 => _2[Symbol.asyncIterator]]) === "function") {
|
|
43
|
+
const stream = rpcResponseOrStream;
|
|
44
|
+
this.streamSse(req, res, stream);
|
|
45
|
+
} else {
|
|
46
|
+
const rpcResponse = rpcResponseOrStream;
|
|
47
|
+
res.status(200).json(rpcResponse);
|
|
48
|
+
}
|
|
49
|
+
} catch (error) {
|
|
50
|
+
this.handleError(req, res, error);
|
|
51
|
+
}
|
|
52
|
+
}}
|
|
53
|
+
async streamSse(req, res, stream) {
|
|
54
|
+
res.setHeader("Content-Type", "text/event-stream");
|
|
55
|
+
res.setHeader("Cache-Control", "no-cache");
|
|
56
|
+
res.setHeader("Connection", "keep-alive");
|
|
57
|
+
res.flushHeaders();
|
|
58
|
+
try {
|
|
59
|
+
for await (const event of stream) {
|
|
60
|
+
res.write(`id: ${Date.now()}
|
|
61
|
+
`);
|
|
62
|
+
res.write(`data: ${JSON.stringify(event)}
|
|
63
|
+
|
|
64
|
+
`);
|
|
65
|
+
}
|
|
66
|
+
} catch (streamError) {
|
|
67
|
+
_chunk466ED5WNcjs.loggers.server.error(
|
|
68
|
+
`Error during SSE streaming (request ${_optionalChain([req, 'access', _3 => _3.body, 'optionalAccess', _4 => _4.id])}):`,
|
|
69
|
+
streamError
|
|
70
|
+
);
|
|
71
|
+
const a2aError = streamError instanceof _server.A2AError ? streamError : _server.A2AError.internalError("Streaming error");
|
|
72
|
+
const errorResponse = {
|
|
73
|
+
jsonrpc: "2.0",
|
|
74
|
+
id: _optionalChain([req, 'access', _5 => _5.body, 'optionalAccess', _6 => _6.id]) || null,
|
|
75
|
+
error: a2aError.toJSONRPCError()
|
|
76
|
+
};
|
|
77
|
+
res.write(`id: ${Date.now()}
|
|
78
|
+
`);
|
|
79
|
+
res.write("event: error\n");
|
|
80
|
+
res.write(`data: ${JSON.stringify(errorResponse)}
|
|
81
|
+
|
|
82
|
+
`);
|
|
83
|
+
} finally {
|
|
84
|
+
if (!res.writableEnded) {
|
|
85
|
+
res.end();
|
|
86
|
+
}
|
|
87
|
+
}
|
|
88
|
+
}
|
|
89
|
+
handleError(req, res, error) {
|
|
90
|
+
_chunk466ED5WNcjs.loggers.server.error(
|
|
91
|
+
"Unhandled error in AINAgent A2A POST handler:",
|
|
92
|
+
error
|
|
93
|
+
);
|
|
94
|
+
const a2aError = error instanceof _server.A2AError ? error : _server.A2AError.internalError("General processing error");
|
|
95
|
+
const errorResponse = {
|
|
96
|
+
jsonrpc: "2.0",
|
|
97
|
+
id: _optionalChain([req, 'access', _7 => _7.body, 'optionalAccess', _8 => _8.id]) || null,
|
|
98
|
+
error: a2aError.toJSONRPCError()
|
|
99
|
+
};
|
|
100
|
+
if (!res.headersSent) {
|
|
101
|
+
res.status(500).json(errorResponse);
|
|
102
|
+
} else if (!res.writableEnded) {
|
|
103
|
+
res.end();
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
}, _class);
|
|
107
|
+
|
|
108
|
+
|
|
109
|
+
|
|
110
|
+
exports.A2AController = A2AController;
|
|
111
|
+
//# sourceMappingURL=chunk-ICFTUZP5.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-ICFTUZP5.cjs","../../src/controllers/a2a.controller.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACOA;AACC;AACA;AACA;AACA;AAAA,4CACM;AAUP,IAAM,kBAAA,EAAN,MAAA,QAAgC,8BAAsB;AAAA,EAC7C;AAAA,EAER,WAAA,CACC,aAAA,EACA,SAAA,EACA,QAAA,EACA,gBAAA,EAA4C,IAAI,4CAAA,CAAgC,CAAA,EAC/E;AACD,IAAA,MAAM,KAAA,EAAO,aAAA,CAAc,CAAA;AAC3B,IAAA,KAAA,CAAM,IAAA,EAAM,SAAA,EAAW,QAAA,EAAU,eAAe,CAAA;AAChD,IAAA,IAAA,CAAK,cAAA,EAAgB,aAAA;AAAA,EACtB;AAAA,EAEA,MAAM,YAAA,CAAA,EAAmC;AACxC,IAAA,OAAO,IAAA,CAAK,aAAA,CAAc,CAAA;AAAA,EAC3B;AACD,CAAA;AAEO,IAAM,cAAA,YAAN,MAAoB;AAAA,EAClB;AAAA,EAER,WAAA,CACC,UAAA,EACA,SAAA,EACA,aAAA,EACC;AAED,IAAA,MAAM,eAAA,EAAiB,IAAI,iBAAA;AAAA,MAC1B,aAAA;AAAA,MACA,SAAA;AAAA,MACA;AAAA;AAAA,IACD,CAAA;AACA,IAAA,IAAA,CAAK,wBAAA,EAA0B,IAAI,oCAAA,CAAwB,cAAc,CAAA;AAAA,EAC1E;AAAA;AAAA;AAAA;AAAA,iBAKO,iBAAA,EAAmB,MAAA,CAAO,GAAA,EAAc,GAAA,EAAA,GAAkB;AAChE,IAAA,IAAI;AACH,MAAA,MAAM,oBAAA,EAAsB,MAAM,IAAA,CAAK,uBAAA,CAAwB,MAAA;AAAA,QAC9D,GAAA,CAAI;AAAA,MACL,CAAA;AAGA,MAAA,GAAA,CACC,uBAAQ,mBAAA,4BAAA,CAA8B,MAAA,CAAO,aAAa,IAAA,IAC1D,UAAA,EACC;AACD,QAAA,MAAM,OAAA,EAAS,mBAAA;AAKf,QAAA,IAAA,CAAK,SAAA,CAAU,GAAA,EAAK,GAAA,EAAK,MAAM,CAAA;AAAA,MAChC,EAAA,KAAO;AAEN,QAAA,MAAM,YAAA,EAAc,mBAAA;AACpB,QAAA,GAAA,CAAI,MAAA,CAAO,GAAG,CAAA,CAAE,IAAA,CAAK,WAAW,CAAA;AAAA,MACjC;AAAA,IACD,EAAA,MAAA,CAAS,KAAA,EAAgB;AACxB,MAAA,IAAA,CAAK,WAAA,CAAY,GAAA,EAAK,GAAA,EAAK,KAAK,CAAA;AAAA,IACjC;AAAA,EACD,EAAA;AAAA,EAEA,MAAc,SAAA,CACb,GAAA,EACA,GAAA,EACA,MAAA,EACC;AACD,IAAA,GAAA,CAAI,SAAA,CAAU,cAAA,EAAgB,mBAAmB,CAAA;AACjD,IAAA,GAAA,CAAI,SAAA,CAAU,eAAA,EAAiB,UAAU,CAAA;AACzC,IAAA,GAAA,CAAI,SAAA,CAAU,YAAA,EAAc,YAAY,CAAA;AACxC,IAAA,GAAA,CAAI,YAAA,CAAa,CAAA;AAEjB,IAAA,IAAI;AACH,MAAA,IAAA,MAAA,CAAA,MAAiB,MAAA,GAAS,MAAA,EAAQ;AACjC,QAAA,GAAA,CAAI,KAAA,CAAM,CAAA,IAAA,EAAO,IAAA,CAAK,GAAA,CAAI,CAAC,CAAA;AAAA,CAAI,CAAA;AAC/B,QAAA,GAAA,CAAI,KAAA,CAAM,CAAA,MAAA,EAAS,IAAA,CAAK,SAAA,CAAU,KAAK,CAAC,CAAA;AAAA;AAAA,CAAM,CAAA;AAAA,MAC/C;AAAA,IACD,EAAA,MAAA,CAAS,WAAA,EAAsB;AAC9B,MAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA;AAAA,QACd,CAAA,oCAAA,kBAAuC,GAAA,qBAAI,IAAA,6BAAM,IAAE,CAAA,EAAA,CAAA;AAAA,QACnD;AAAA,MACD,CAAA;AACA,MAAA,MAAM,SAAA,EACL,YAAA,WAAuB,iBAAA,EACpB,YAAA,EACA,gBAAA,CAAS,aAAA,CAAc,iBAAiB,CAAA;AAC5C,MAAA,MAAM,cAAA,EAAsC;AAAA,QAC3C,OAAA,EAAS,KAAA;AAAA,QACT,EAAA,kBAAI,GAAA,qBAAI,IAAA,6BAAM,KAAA,GAAM,IAAA;AAAA,QACpB,KAAA,EAAO,QAAA,CAAS,cAAA,CAAe;AAAA,MAChC,CAAA;AACA,MAAA,GAAA,CAAI,KAAA,CAAM,CAAA,IAAA,EAAO,IAAA,CAAK,GAAA,CAAI,CAAC,CAAA;AAAA,CAAI,CAAA;AAC/B,MAAA,GAAA,CAAI,KAAA,CAAM,gBAAgB,CAAA;AAC1B,MAAA,GAAA,CAAI,KAAA,CAAM,CAAA,MAAA,EAAS,IAAA,CAAK,SAAA,CAAU,aAAa,CAAC,CAAA;AAAA;AAAA,CAAM,CAAA;AAAA,IACvD,EAAA,QAAE;AACD,MAAA,GAAA,CAAI,CAAC,GAAA,CAAI,aAAA,EAAe;AACvB,QAAA,GAAA,CAAI,GAAA,CAAI,CAAA;AAAA,MACT;AAAA,IACD;AAAA,EACD;AAAA,EAEQ,WAAA,CAAY,GAAA,EAAc,GAAA,EAAe,KAAA,EAAgB;AAChE,IAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA;AAAA,MACd,+CAAA;AAAA,MACA;AAAA,IACD,CAAA;AACA,IAAA,MAAM,SAAA,EACL,MAAA,WAAiB,iBAAA,EACd,MAAA,EACA,gBAAA,CAAS,aAAA,CAAc,0BAA0B,CAAA;AACrD,IAAA,MAAM,cAAA,EAAsC;AAAA,MAC3C,OAAA,EAAS,KAAA;AAAA,MACT,EAAA,kBAAI,GAAA,qBAAI,IAAA,6BAAM,KAAA,GAAM,IAAA;AAAA,MACpB,KAAA,EAAO,QAAA,CAAS,cAAA,CAAe;AAAA,IAChC,CAAA;AACA,IAAA,GAAA,CAAI,CAAC,GAAA,CAAI,WAAA,EAAa;AACrB,MAAA,GAAA,CAAI,MAAA,CAAO,GAAG,CAAA,CAAE,IAAA,CAAK,aAAa,CAAA;AAAA,IACnC,EAAA,KAAA,GAAA,CAAW,CAAC,GAAA,CAAI,aAAA,EAAe;AAC9B,MAAA,GAAA,CAAI,GAAA,CAAI,CAAA;AAAA,IACT;AAAA,EACD;AACD,UAAA;AD7CA;AACA;AACE;AACF,sCAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-ICFTUZP5.cjs","sourcesContent":[null,"import type {\n\tA2AResponse,\n\tAgentCard,\n\tJSONRPCErrorResponse,\n\tJSONRPCSuccessResponse,\n} from \"@a2a-js/sdk\";\nimport type {\n\tAgentExecutor,\n\tExecutionEventBusManager,\n\tTaskStore,\n} from \"@a2a-js/sdk/server\";\nimport {\n\tA2AError,\n\tDefaultExecutionEventBusManager,\n\tDefaultRequestHandler,\n\tJsonRpcTransportHandler,\n} from \"@a2a-js/sdk/server\";\nimport type { Request, Response } from \"express\";\nimport type { A2AService } from \"@/services/a2a.service.js\";\nimport { loggers } from \"@/utils/logger.js\";\n\n/**\n * Handles the transport layer for A2A communication.\n * It orchestrates the A2A-JS SDK components and manages the HTTP request/response lifecycle.\n */\n\nclass AINRequestHandler extends DefaultRequestHandler {\n\tprivate cardGenerator: () => AgentCard;\n\n\tconstructor(\n\t\tcardGenerator: () => AgentCard,\n\t\ttaskStore: TaskStore,\n\t\texecutor: AgentExecutor,\n\t\teventBusManager: ExecutionEventBusManager = new DefaultExecutionEventBusManager(),\n\t) {\n\t\tconst card = cardGenerator();\n\t\tsuper(card, taskStore, executor, eventBusManager);\n\t\tthis.cardGenerator = cardGenerator;\n\t}\n\n\tasync getAgentCard(): Promise<AgentCard> {\n\t\treturn this.cardGenerator();\n\t}\n}\n\nexport class A2AController {\n\tprivate jsonRpcTransportHandler: JsonRpcTransportHandler;\n\n\tconstructor(\n\t\ta2aService: A2AService,\n\t\ttaskStore: TaskStore,\n\t\tcardGenerator: () => AgentCard,\n\t) {\n\t\t// The controller is responsible for setting up the SDK components.\n\t\tconst requestHandler = new AINRequestHandler(\n\t\t\tcardGenerator,\n\t\t\ttaskStore,\n\t\t\ta2aService, // Injecting the service here as the executor\n\t\t);\n\t\tthis.jsonRpcTransportHandler = new JsonRpcTransportHandler(requestHandler);\n\t}\n\n\t/**\n\t * Handles the POST /a2a request.\n\t */\n\tpublic handleA2ARequest = async (req: Request, res: Response) => {\n\t\ttry {\n\t\t\tconst rpcResponseOrStream = await this.jsonRpcTransportHandler.handle(\n\t\t\t\treq.body,\n\t\t\t);\n\n\t\t\t// Handle streaming responses (AsyncGenerator)\n\t\t\tif (\n\t\t\t\ttypeof (rpcResponseOrStream as any)?.[Symbol.asyncIterator] ===\n\t\t\t\t\"function\"\n\t\t\t) {\n\t\t\t\tconst stream = rpcResponseOrStream as AsyncGenerator<\n\t\t\t\t\tJSONRPCSuccessResponse,\n\t\t\t\t\tvoid,\n\t\t\t\t\tundefined\n\t\t\t\t>;\n\t\t\t\tthis.streamSse(req, res, stream);\n\t\t\t} else {\n\t\t\t\t// Handle single JSON-RPC response\n\t\t\t\tconst rpcResponse = rpcResponseOrStream as A2AResponse;\n\t\t\t\tres.status(200).json(rpcResponse);\n\t\t\t}\n\t\t} catch (error: unknown) {\n\t\t\tthis.handleError(req, res, error);\n\t\t}\n\t};\n\n\tprivate async streamSse(\n\t\treq: Request,\n\t\tres: Response,\n\t\tstream: AsyncGenerator<JSONRPCSuccessResponse, void, undefined>,\n\t) {\n\t\tres.setHeader(\"Content-Type\", \"text/event-stream\");\n\t\tres.setHeader(\"Cache-Control\", \"no-cache\");\n\t\tres.setHeader(\"Connection\", \"keep-alive\");\n\t\tres.flushHeaders();\n\n\t\ttry {\n\t\t\tfor await (const event of stream) {\n\t\t\t\tres.write(`id: ${Date.now()}\\n`);\n\t\t\t\tres.write(`data: ${JSON.stringify(event)}\\n\\n`);\n\t\t\t}\n\t\t} catch (streamError: unknown) {\n\t\t\tloggers.server.error(\n\t\t\t\t`Error during SSE streaming (request ${req.body?.id}):`,\n\t\t\t\tstreamError,\n\t\t\t);\n\t\t\tconst a2aError =\n\t\t\t\tstreamError instanceof A2AError\n\t\t\t\t\t? streamError\n\t\t\t\t\t: A2AError.internalError(\"Streaming error\");\n\t\t\tconst errorResponse: JSONRPCErrorResponse = {\n\t\t\t\tjsonrpc: \"2.0\",\n\t\t\t\tid: req.body?.id || null,\n\t\t\t\terror: a2aError.toJSONRPCError(),\n\t\t\t};\n\t\t\tres.write(`id: ${Date.now()}\\n`);\n\t\t\tres.write(\"event: error\\n\");\n\t\t\tres.write(`data: ${JSON.stringify(errorResponse)}\\n\\n`);\n\t\t} finally {\n\t\t\tif (!res.writableEnded) {\n\t\t\t\tres.end();\n\t\t\t}\n\t\t}\n\t}\n\n\tprivate handleError(req: Request, res: Response, error: unknown) {\n\t\tloggers.server.error(\n\t\t\t\"Unhandled error in AINAgent A2A POST handler:\",\n\t\t\terror,\n\t\t);\n\t\tconst a2aError =\n\t\t\terror instanceof A2AError\n\t\t\t\t? error\n\t\t\t\t: A2AError.internalError(\"General processing error\");\n\t\tconst errorResponse: JSONRPCErrorResponse = {\n\t\t\tjsonrpc: \"2.0\",\n\t\t\tid: req.body?.id || null,\n\t\t\terror: a2aError.toJSONRPCError(),\n\t\t};\n\t\tif (!res.headersSent) {\n\t\t\tres.status(500).json(errorResponse);\n\t\t} else if (!res.writableEnded) {\n\t\t\tres.end();\n\t\t}\n\t}\n}\n"]}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});
|
|
2
|
+
|
|
3
|
+
var _chunkN26HPJTOcjs = require('./chunk-N26HPJTO.cjs');
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
var _chunkFGV256AQcjs = require('./chunk-FGV256AQ.cjs');
|
|
7
|
+
|
|
8
|
+
// src/routes/query.routes.ts
|
|
9
|
+
var _express = require('express');
|
|
10
|
+
var createQueryRouter = (agent) => {
|
|
11
|
+
const router = _express.Router.call(void 0, );
|
|
12
|
+
const queryService = new (0, _chunkFGV256AQcjs.QueryService)(
|
|
13
|
+
agent.modelModule,
|
|
14
|
+
agent.a2aModule,
|
|
15
|
+
agent.mcpModule,
|
|
16
|
+
agent.memoryModule,
|
|
17
|
+
agent.manifest.prompts
|
|
18
|
+
);
|
|
19
|
+
const queryController = new (0, _chunkN26HPJTOcjs.QueryController)(queryService);
|
|
20
|
+
console.log("aaaaaaaaaaaaaaaaaaaaaa");
|
|
21
|
+
router.post("/query", queryController.handleQueryRequest);
|
|
22
|
+
return router;
|
|
23
|
+
};
|
|
24
|
+
|
|
25
|
+
|
|
26
|
+
|
|
27
|
+
exports.createQueryRouter = createQueryRouter;
|
|
28
|
+
//# sourceMappingURL=chunk-K3EK37A7.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-K3EK37A7.cjs","../../src/routes/query.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,kCAAuB;AAKhB,IAAM,kBAAA,EAAoB,CAAC,KAAA,EAAA,GAA4B;AAC7D,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAEtB,EAAA,MAAM,aAAA,EAAe,IAAI,mCAAA;AAAA,IACxB,KAAA,CAAM,WAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,YAAA;AAAA,IACN,KAAA,CAAM,QAAA,CAAS;AAAA,EAChB,CAAA;AACA,EAAA,MAAM,gBAAA,EAAkB,IAAI,sCAAA,CAAgB,YAAY,CAAA;AACxD,EAAA,OAAA,CAAQ,GAAA,CAAI,wBAAwB,CAAA;AACpC,EAAA,MAAA,CAAO,IAAA,CAAK,QAAA,EAAU,eAAA,CAAgB,kBAAkB,CAAA;AAExD,EAAA,OAAO,MAAA;AACR,CAAA;ADGA;AACA;AACE;AACF,8CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-K3EK37A7.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport { QueryController } from \"@/controllers/query.controller.js\";\nimport type { AINAgent } from \"@/index.js\";\nimport { QueryService } from \"@/services/query.service.js\";\n\nexport const createQueryRouter = (agent: AINAgent): Router => {\n\tconst router = Router();\n\n\tconst queryService = new QueryService(\n\t\tagent.modelModule,\n\t\tagent.a2aModule,\n\t\tagent.mcpModule,\n\t\tagent.memoryModule,\n\t\tagent.manifest.prompts,\n\t);\n\tconst queryController = new QueryController(queryService);\n\tconsole.log('aaaaaaaaaaaaaaaaaaaaaa');\n\trouter.post(\"/query\", queryController.handleQueryRequest);\n\n\treturn router;\n};\n"]}
|
|
@@ -0,0 +1,43 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});
|
|
2
|
+
|
|
3
|
+
var _chunkDX2JH3FJcjs = require('./chunk-DX2JH3FJ.cjs');
|
|
4
|
+
|
|
5
|
+
// src/middlewares/auth.middleware.ts
|
|
6
|
+
var _httpstatuscodes = require('http-status-codes');
|
|
7
|
+
var AuthMiddleware = class {
|
|
8
|
+
|
|
9
|
+
constructor(auth) {
|
|
10
|
+
this.auth = auth;
|
|
11
|
+
}
|
|
12
|
+
middleware() {
|
|
13
|
+
return async (req, res, next) => {
|
|
14
|
+
try {
|
|
15
|
+
const authRes = await this.auth.authenticate(req, res);
|
|
16
|
+
if (authRes.isAuthenticated) {
|
|
17
|
+
res.locals.userId = authRes.userId;
|
|
18
|
+
next();
|
|
19
|
+
} else {
|
|
20
|
+
const error = new (0, _chunkDX2JH3FJcjs.AinHttpError)(
|
|
21
|
+
_httpstatuscodes.StatusCodes.UNAUTHORIZED,
|
|
22
|
+
"Unauthorized"
|
|
23
|
+
);
|
|
24
|
+
throw error;
|
|
25
|
+
}
|
|
26
|
+
} catch (e) {
|
|
27
|
+
if (!e.status) {
|
|
28
|
+
const error = new (0, _chunkDX2JH3FJcjs.AinHttpError)(
|
|
29
|
+
_httpstatuscodes.StatusCodes.INTERNAL_SERVER_ERROR,
|
|
30
|
+
`Authentication error: ${JSON.stringify(e)}`
|
|
31
|
+
);
|
|
32
|
+
throw error;
|
|
33
|
+
}
|
|
34
|
+
throw e;
|
|
35
|
+
}
|
|
36
|
+
};
|
|
37
|
+
}
|
|
38
|
+
};
|
|
39
|
+
|
|
40
|
+
|
|
41
|
+
|
|
42
|
+
exports.AuthMiddleware = AuthMiddleware;
|
|
43
|
+
//# sourceMappingURL=chunk-LRBRG2DS.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-LRBRG2DS.cjs","../../src/middlewares/auth.middleware.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACHA,oDAA4B;AAMrB,IAAM,eAAA,EAAN,MAAqB;AAAA,EACnB;AAAA,EACR,WAAA,CAAY,IAAA,EAAgB;AAC3B,IAAA,IAAA,CAAK,KAAA,EAAO,IAAA;AAAA,EACb;AAAA,EAEO,UAAA,CAAA,EAA6B;AACnC,IAAA,OAAO,MAAA,CAAO,GAAA,EAAc,GAAA,EAAe,IAAA,EAAA,GAAuB;AACjE,MAAA,IAAI;AACH,QAAA,MAAM,QAAA,EAAwB,MAAM,IAAA,CAAK,IAAA,CAAK,YAAA,CAAa,GAAA,EAAK,GAAG,CAAA;AACnE,QAAA,GAAA,CAAI,OAAA,CAAQ,eAAA,EAAiB;AAC5B,UAAA,GAAA,CAAI,MAAA,CAAO,OAAA,EAAS,OAAA,CAAQ,MAAA;AAC5B,UAAA,IAAA,CAAK,CAAA;AAAA,QACN,EAAA,KAAO;AACN,UAAA,MAAM,MAAA,EAAsB,IAAI,mCAAA;AAAA,YAC/B,4BAAA,CAAY,YAAA;AAAA,YACZ;AAAA,UACD,CAAA;AACA,UAAA,MAAM,KAAA;AAAA,QACP;AAAA,MACD,EAAA,MAAA,CAAS,CAAA,EAAQ;AAChB,QAAA,GAAA,CAAI,CAAC,CAAA,CAAE,MAAA,EAAQ;AACd,UAAA,MAAM,MAAA,EAAsB,IAAI,mCAAA;AAAA,YAC/B,4BAAA,CAAY,qBAAA;AAAA,YACZ,CAAA,sBAAA,EAAyB,IAAA,CAAK,SAAA,CAAU,CAAC,CAAC,CAAA;AAAA,UAAA;AAE3C,UAAA;AAAM,QAAA;AAEP,QAAA;AAAM,MAAA;AACP,IAAA;AACD,EAAA;AAEF;ADDA;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-LRBRG2DS.cjs","sourcesContent":[null,"import type { NextFunction, Request, RequestHandler, Response } from \"express\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type { BaseAuth } from \"@/modules/auth/base.auth\";\nimport { AinHttpError } from \"@/types/agent\";\nimport type { AuthResponse } from \"@/types/auth\";\nimport { loggers } from \"@/utils/logger\";\n\nexport class AuthMiddleware {\n\tprivate auth: BaseAuth;\n\tconstructor(auth: BaseAuth) {\n\t\tthis.auth = auth;\n\t}\n\n\tpublic middleware(): RequestHandler {\n\t\treturn async (req: Request, res: Response, next: NextFunction) => {\n\t\t\ttry {\n\t\t\t\tconst authRes: AuthResponse = await this.auth.authenticate(req, res);\n\t\t\t\tif (authRes.isAuthenticated) {\n\t\t\t\t\tres.locals.userId = authRes.userId;\n\t\t\t\t\tnext();\n\t\t\t\t} else {\n\t\t\t\t\tconst error: AinHttpError = new AinHttpError(\n\t\t\t\t\t\tStatusCodes.UNAUTHORIZED,\n\t\t\t\t\t\t\"Unauthorized\",\n\t\t\t\t\t);\n\t\t\t\t\tthrow error;\n\t\t\t\t}\n\t\t\t} catch (e: any) {\n\t\t\t\tif (!e.status) {\n\t\t\t\t\tconst error: AinHttpError = new AinHttpError(\n\t\t\t\t\t\tStatusCodes.INTERNAL_SERVER_ERROR,\n\t\t\t\t\t\t`Authentication error: ${JSON.stringify(e)}`,\n\t\t\t\t\t);\n\t\t\t\t\tthrow error;\n\t\t\t\t}\n\t\t\t\tthrow e;\n\t\t\t}\n\t\t};\n\t}\n}\n"]}
|
|
@@ -0,0 +1,100 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true}); var _class;
|
|
2
|
+
|
|
3
|
+
var _chunkE433ONENcjs = require('./chunk-E433ONEN.cjs');
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
|
|
7
|
+
|
|
8
|
+
// src/modules/mcp/mcp.module.ts
|
|
9
|
+
var _indexjs = require('@modelcontextprotocol/sdk/client/index.js');
|
|
10
|
+
var _stdiojs = require('@modelcontextprotocol/sdk/client/stdio.js');
|
|
11
|
+
var MCPModule = (_class = class {constructor() { _class.prototype.__init.call(this);_class.prototype.__init2.call(this);_class.prototype.__init3.call(this); }
|
|
12
|
+
/** Map of MCP server names to their client instances */
|
|
13
|
+
__init() {this.mcpMap = /* @__PURE__ */ new Map()}
|
|
14
|
+
/** Map of MCP server names to their transport instances */
|
|
15
|
+
__init2() {this.transportMap = /* @__PURE__ */ new Map()}
|
|
16
|
+
/** Array of all discovered tools from connected MCP servers */
|
|
17
|
+
__init3() {this.tools = []}
|
|
18
|
+
/**
|
|
19
|
+
* Connects to MCP servers based on the provided configuration.
|
|
20
|
+
*
|
|
21
|
+
* For each server in the config, establishes a connection, discovers
|
|
22
|
+
* available tools, and adds them to the module's tool collection.
|
|
23
|
+
* Skips servers that are already connected.
|
|
24
|
+
*
|
|
25
|
+
* @param mcpConfig - Configuration object mapping server names to connection details
|
|
26
|
+
* @throws Error if connection to any MCP server fails
|
|
27
|
+
*/
|
|
28
|
+
async addMCPConfig(mcpConfig) {
|
|
29
|
+
try {
|
|
30
|
+
for (const [name, conf] of Object.entries(mcpConfig)) {
|
|
31
|
+
if (this.mcpMap.get(name) && this.transportMap.get(name)) continue;
|
|
32
|
+
const transport = new (0, _stdiojs.StdioClientTransport)(conf);
|
|
33
|
+
this.transportMap.set(name, transport);
|
|
34
|
+
const mcp = new (0, _indexjs.Client)({ name: "mcp-client-cli", version: "1.0.0" });
|
|
35
|
+
await mcp.connect(transport);
|
|
36
|
+
this.mcpMap.set(name, mcp);
|
|
37
|
+
const toolsResult = await mcp.listTools();
|
|
38
|
+
this.tools.push(
|
|
39
|
+
...toolsResult.tools.map((tool) => {
|
|
40
|
+
return new (0, _chunkE433ONENcjs.MCPTool)(name, tool);
|
|
41
|
+
})
|
|
42
|
+
);
|
|
43
|
+
}
|
|
44
|
+
_chunk466ED5WNcjs.loggers.mcp.info("Connected to MCP server with tools:", {
|
|
45
|
+
tools: this.tools.map((tool) => tool.id)
|
|
46
|
+
});
|
|
47
|
+
} catch (error) {
|
|
48
|
+
_chunk466ED5WNcjs.loggers.mcp.error("Failed to connect to MCP server:", { error });
|
|
49
|
+
throw error;
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
/**
|
|
53
|
+
* Returns all available tools from connected MCP servers.
|
|
54
|
+
*
|
|
55
|
+
* @returns Array of MCPTool instances representing available tools
|
|
56
|
+
*/
|
|
57
|
+
getTools() {
|
|
58
|
+
return this.tools;
|
|
59
|
+
}
|
|
60
|
+
/**
|
|
61
|
+
* Executes a tool on its corresponding MCP server.
|
|
62
|
+
*
|
|
63
|
+
* @param tool - The MCPTool instance to execute
|
|
64
|
+
* @param _args - Arguments to pass to the tool
|
|
65
|
+
* @returns Promise resolving to the tool's execution result
|
|
66
|
+
* @throws Error if the MCP server for the tool is not found
|
|
67
|
+
*/
|
|
68
|
+
async useTool(tool, _args) {
|
|
69
|
+
const { serverName, mcpTool } = tool;
|
|
70
|
+
const toolName = mcpTool.name;
|
|
71
|
+
const mcp = this.mcpMap.get(serverName);
|
|
72
|
+
if (!mcp) {
|
|
73
|
+
throw new Error(`Invalid MCP Tool ${serverName}-${mcpTool.name}`);
|
|
74
|
+
}
|
|
75
|
+
const result = await mcp.callTool({
|
|
76
|
+
name: toolName,
|
|
77
|
+
arguments: _args
|
|
78
|
+
});
|
|
79
|
+
const toolResult = `[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]
|
|
80
|
+
` + JSON.stringify(result.content, null, 2);
|
|
81
|
+
_chunk466ED5WNcjs.loggers.mcp.debug("MCP useTool result:", toolResult);
|
|
82
|
+
return result;
|
|
83
|
+
}
|
|
84
|
+
/**
|
|
85
|
+
* Closes all MCP client connections.
|
|
86
|
+
*
|
|
87
|
+
* Should be called when shutting down the application to ensure
|
|
88
|
+
* all MCP connections are properly closed.
|
|
89
|
+
*/
|
|
90
|
+
async cleanup() {
|
|
91
|
+
this.mcpMap.forEach((mcp) => {
|
|
92
|
+
mcp.close();
|
|
93
|
+
});
|
|
94
|
+
}
|
|
95
|
+
}, _class);
|
|
96
|
+
|
|
97
|
+
|
|
98
|
+
|
|
99
|
+
exports.MCPModule = MCPModule;
|
|
100
|
+
//# sourceMappingURL=chunk-MMWEU2MO.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-MMWEU2MO.cjs","../../src/modules/mcp/mcp.module.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,oEAAuB;AACvB,oEAAqC;AA0B9B,IAAM,UAAA,YAAN,MAAgB;AAAA;AAAA,iBAEd,OAAA,kBAA8B,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA,kBAEtC,aAAA,kBAAkD,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA,kBAE1D,MAAA,EAAmB,CAAC,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAY5B,MAAM,YAAA,CAAa,SAAA,EAAsB;AACxC,IAAA,IAAI;AACH,MAAA,IAAA,CAAA,MAAW,CAAC,IAAA,EAAM,IAAI,EAAA,GAAK,MAAA,CAAO,OAAA,CAAQ,SAAS,CAAA,EAAG;AAErD,QAAA,GAAA,CAAI,IAAA,CAAK,MAAA,CAAO,GAAA,CAAI,IAAI,EAAA,GAAK,IAAA,CAAK,YAAA,CAAa,GAAA,CAAI,IAAI,CAAA,EAAG,QAAA;AAE1D,QAAA,MAAM,UAAA,EAAY,IAAI,kCAAA,CAAqB,IAAI,CAAA;AAC/C,QAAA,IAAA,CAAK,YAAA,CAAa,GAAA,CAAI,IAAA,EAAM,SAAS,CAAA;AACrC,QAAA,MAAM,IAAA,EAAM,IAAI,oBAAA,CAAO,EAAE,IAAA,EAAM,gBAAA,EAAkB,OAAA,EAAS,QAAQ,CAAC,CAAA;AACnE,QAAA,MAAM,GAAA,CAAI,OAAA,CAAQ,SAAS,CAAA;AAC3B,QAAA,IAAA,CAAK,MAAA,CAAO,GAAA,CAAI,IAAA,EAAM,GAAG,CAAA;AAEzB,QAAA,MAAM,YAAA,EAAc,MAAM,GAAA,CAAI,SAAA,CAAU,CAAA;AACxC,QAAA,IAAA,CAAK,KAAA,CAAM,IAAA;AAAA,UACV,GAAG,WAAA,CAAY,KAAA,CAAM,GAAA,CAAI,CAAC,IAAA,EAAA,GAAS;AAClC,YAAA,OAAO,IAAI,8BAAA,CAAQ,IAAA,EAAM,IAAI,CAAA;AAAA,UAC9B,CAAC;AAAA,QACF,CAAA;AAAA,MACD;AACA,MAAA,yBAAA,CAAQ,GAAA,CAAI,IAAA,CAAK,qCAAA,EAAuC;AAAA,QACvD,KAAA,EAAO,IAAA,CAAK,KAAA,CAAM,GAAA,CAAI,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,EAAE;AAAA,MACxC,CAAC,CAAA;AAAA,IACF,EAAA,MAAA,CAAS,KAAA,EAAgB;AACxB,MAAA,yBAAA,CAAQ,GAAA,CAAI,KAAA,CAAM,kCAAA,EAAoC,EAAE,MAAM,CAAC,CAAA;AAC/D,MAAA,MAAM,KAAA;AAAA,IACP;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,QAAA,CAAA,EAAW;AACV,IAAA,OAAO,IAAA,CAAK,KAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAM,OAAA,CAAQ,IAAA,EAAe,KAAA,EAA2B;AACvD,IAAA,MAAM,EAAE,UAAA,EAAY,QAAQ,EAAA,EAAI,IAAA;AAChC,IAAA,MAAM,SAAA,EAAW,OAAA,CAAQ,IAAA;AACzB,IAAA,MAAM,IAAA,EAAM,IAAA,CAAK,MAAA,CAAO,GAAA,CAAI,UAAU,CAAA;AAEtC,IAAA,GAAA,CAAI,CAAC,GAAA,EAAK;AACT,MAAA,MAAM,IAAI,KAAA,CAAM,CAAA,iBAAA,EAAoB,UAAU,CAAA,CAAA,EAAI,OAAA,CAAQ,IAAI,CAAA,CAAA;AAC/D,IAAA;AAEkC,IAAA;AAC3B,MAAA;AACK,MAAA;AACX,IAAA;AAE0C,IAAA;AACJ;AAEY,IAAA;AAC5C,IAAA;AACR,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAQgB,EAAA;AACsB,IAAA;AAC1B,MAAA;AACV,IAAA;AACF,EAAA;AACD;AD3BqE;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-MMWEU2MO.cjs","sourcesContent":[null,"import { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport type { MCPConfig } from \"@/types/mcp.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { MCPTool } from \"./mcp.tool.js\";\n\n/**\n * Module for managing Model Context Protocol (MCP) server connections.\n *\n * This module handles the lifecycle of MCP client connections, discovers\n * available tools from connected servers, and provides an interface for\n * executing those tools. Multiple MCP servers can be connected simultaneously.\n *\n * @example\n * ```typescript\n * const mcpModule = new MCPModule();\n * await mcpModule.addMCPConfig({\n * \"filesystem\": {\n * command: \"npx\",\n * args: [\"@modelcontextprotocol/server-filesystem\", \"/path/to/files\"]\n * }\n * });\n *\n * const tools = mcpModule.getTools();\n * const result = await mcpModule.useTool(tools[0], { path: \"/example.txt\" });\n * ```\n */\nexport class MCPModule {\n\t/** Map of MCP server names to their client instances */\n\tprivate mcpMap: Map<string, Client> = new Map();\n\t/** Map of MCP server names to their transport instances */\n\tprivate transportMap: Map<string, StdioClientTransport> = new Map();\n\t/** Array of all discovered tools from connected MCP servers */\n\tprivate tools: MCPTool[] = [];\n\n\t/**\n\t * Connects to MCP servers based on the provided configuration.\n\t *\n\t * For each server in the config, establishes a connection, discovers\n\t * available tools, and adds them to the module's tool collection.\n\t * Skips servers that are already connected.\n\t *\n\t * @param mcpConfig - Configuration object mapping server names to connection details\n\t * @throws Error if connection to any MCP server fails\n\t */\n\tasync addMCPConfig(mcpConfig: MCPConfig) {\n\t\ttry {\n\t\t\tfor (const [name, conf] of Object.entries(mcpConfig)) {\n\t\t\t\t// FIXME(yoojin): Need strict duplication check.\n\t\t\t\tif (this.mcpMap.get(name) && this.transportMap.get(name)) continue; // Duplicated mcp: skip\n\n\t\t\t\tconst transport = new StdioClientTransport(conf);\n\t\t\t\tthis.transportMap.set(name, transport);\n\t\t\t\tconst mcp = new Client({ name: \"mcp-client-cli\", version: \"1.0.0\" });\n\t\t\t\tawait mcp.connect(transport);\n\t\t\t\tthis.mcpMap.set(name, mcp);\n\n\t\t\t\tconst toolsResult = await mcp.listTools();\n\t\t\t\tthis.tools.push(\n\t\t\t\t\t...toolsResult.tools.map((tool) => {\n\t\t\t\t\t\treturn new MCPTool(name, tool);\n\t\t\t\t\t}),\n\t\t\t\t);\n\t\t\t}\n\t\t\tloggers.mcp.info(\"Connected to MCP server with tools:\", {\n\t\t\t\ttools: this.tools.map((tool) => tool.id),\n\t\t\t});\n\t\t} catch (error: unknown) {\n\t\t\tloggers.mcp.error(\"Failed to connect to MCP server:\", { error });\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\t/**\n\t * Returns all available tools from connected MCP servers.\n\t *\n\t * @returns Array of MCPTool instances representing available tools\n\t */\n\tgetTools() {\n\t\treturn this.tools;\n\t}\n\n\t/**\n\t * Executes a tool on its corresponding MCP server.\n\t *\n\t * @param tool - The MCPTool instance to execute\n\t * @param _args - Arguments to pass to the tool\n\t * @returns Promise resolving to the tool's execution result\n\t * @throws Error if the MCP server for the tool is not found\n\t */\n\tasync useTool(tool: MCPTool, _args?: any): Promise<any> {\n\t\tconst { serverName, mcpTool } = tool;\n\t\tconst toolName = mcpTool.name;\n\t\tconst mcp = this.mcpMap.get(serverName);\n\n\t\tif (!mcp) {\n\t\t\tthrow new Error(`Invalid MCP Tool ${serverName}-${mcpTool.name}`);\n\t\t}\n\n\t\tconst result = await mcp.callTool({\n\t\t\tname: toolName,\n\t\t\targuments: _args,\n\t\t});\n\t\tconst toolResult =\n\t\t\t`[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]\\n` +\n\t\t\tJSON.stringify(result.content, null, 2);\n\n\t\tloggers.mcp.debug(\"MCP useTool result:\", toolResult);\n\t\treturn result;\n\t}\n\n\t/**\n\t * Closes all MCP client connections.\n\t *\n\t * Should be called when shutting down the application to ensure\n\t * all MCP connections are properly closed.\n\t */\n\tasync cleanup() {\n\t\tthis.mcpMap.forEach((mcp: Client) => {\n\t\t\tmcp.close();\n\t\t});\n\t}\n}\n"]}
|
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true}); var _class;// src/controllers/query.controller.ts
|
|
2
|
+
var QueryController = (_class = class {
|
|
3
|
+
|
|
4
|
+
constructor(queryService) {;_class.prototype.__init.call(this);
|
|
5
|
+
this.queryService = queryService;
|
|
6
|
+
}
|
|
7
|
+
__init() {this.handleQueryRequest = async (req, res, next) => {
|
|
8
|
+
console.log("bbbbbbbbbbbbbbbbbbbb");
|
|
9
|
+
try {
|
|
10
|
+
const { message, sessionId } = req.body;
|
|
11
|
+
const userId = res.locals.userId;
|
|
12
|
+
const result = await this.queryService.handleQuery(
|
|
13
|
+
message,
|
|
14
|
+
sessionId,
|
|
15
|
+
userId
|
|
16
|
+
);
|
|
17
|
+
res.status(200).json(result);
|
|
18
|
+
} catch (error) {
|
|
19
|
+
next(error);
|
|
20
|
+
}
|
|
21
|
+
}}
|
|
22
|
+
}, _class);
|
|
23
|
+
|
|
24
|
+
|
|
25
|
+
|
|
26
|
+
exports.QueryController = QueryController;
|
|
27
|
+
//# sourceMappingURL=chunk-N26HPJTO.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-N26HPJTO.cjs","../../src/controllers/query.controller.ts"],"names":[],"mappings":"AAAA;ACGO,IAAM,gBAAA,YAAN,MAAsB;AAAA,EACpB;AAAA,EAER,WAAA,CAAY,YAAA,EAA4B;AACvC,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AAAA,EACrB;AAAA,iBAEO,mBAAA,EAAqB,MAAA,CAC3B,GAAA,EACA,GAAA,EACA,IAAA,EAAA,GACI;AACJ,IAAA,OAAA,CAAQ,GAAA,CAAI,sBAAsB,CAAA;AAClC,IAAA,IAAI;AACH,MAAA,MAAM,EAAE,OAAA,EAAS,UAAU,EAAA,EAAI,GAAA,CAAI,IAAA;AACnC,MAAA,MAAM,OAAA,EAAS,GAAA,CAAI,MAAA,CAAO,MAAA;AAE1B,MAAA,MAAM,OAAA,EAAS,MAAM,IAAA,CAAK,YAAA,CAAa,WAAA;AAAA,QACtC,OAAA;AAAA,QACA,SAAA;AAAA,QACA;AAAA,MACD,CAAA;AAEA,MAAA,GAAA,CAAI,MAAA,CAAO,GAAG,CAAA,CAAE,IAAA,CAAK,MAAM,CAAA;AAAA,IAC5B,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,IAAA,CAAK,KAAK,CAAA;AAAA,IACX;AAAA,EACD,EAAA;AACD,UAAA;ADTA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-N26HPJTO.cjs","sourcesContent":[null,"import type { NextFunction, Request, Response } from \"express\";\nimport type { QueryService } from \"@/services\";\n\nexport class QueryController {\n\tprivate queryService;\n\n\tconstructor(queryService: QueryService) {\n\t\tthis.queryService = queryService;\n\t}\n\n\tpublic handleQueryRequest = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\tconsole.log('bbbbbbbbbbbbbbbbbbbb');\n\t\ttry {\n\t\t\tconst { message, sessionId } = req.body;\n\t\t\tconst userId = res.locals.userId;\n\n\t\t\tconst result = await this.queryService.handleQuery(\n\t\t\t\tmessage,\n\t\t\t\tsessionId,\n\t\t\t\tuserId,\n\t\t\t);\n\n\t\t\tres.status(200).json(result);\n\t\t} catch (error) {\n\t\t\tnext(error);\n\t\t}\n\t};\n}\n"]}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});
|
|
2
|
+
|
|
3
|
+
var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
|
|
4
|
+
|
|
5
|
+
// src/middlewares/error.middleware.ts
|
|
6
|
+
var errorMiddleware = (error, req, res, next) => {
|
|
7
|
+
try {
|
|
8
|
+
const status = error.status || 500;
|
|
9
|
+
const message = error.message || "Something went wrong";
|
|
10
|
+
_chunk466ED5WNcjs.logger.error(
|
|
11
|
+
`[${req.method}] ${req.path} >> StatusCode:: ${status}, Message:: ${message}`
|
|
12
|
+
);
|
|
13
|
+
res.status(status).json({ message });
|
|
14
|
+
} catch (error2) {
|
|
15
|
+
next(error2);
|
|
16
|
+
}
|
|
17
|
+
};
|
|
18
|
+
|
|
19
|
+
|
|
20
|
+
|
|
21
|
+
exports.errorMiddleware = errorMiddleware;
|
|
22
|
+
//# sourceMappingURL=chunk-NYYEZ4ER.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-NYYEZ4ER.cjs","../../src/middlewares/error.middleware.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACAO,IAAM,gBAAA,EAAkB,CAC9B,KAAA,EACA,GAAA,EACA,GAAA,EACA,IAAA,EAAA,GACU;AACV,EAAA,IAAI;AACH,IAAA,MAAM,OAAA,EAAiB,KAAA,CAAM,OAAA,GAAU,GAAA;AACvC,IAAA,MAAM,QAAA,EAAkB,KAAA,CAAM,QAAA,GAAW,sBAAA;AAEzC,IAAA,wBAAA,CAAO,KAAA;AAAA,MACN,CAAA,CAAA,EAAI,GAAA,CAAI,MAAM,CAAA,EAAA,EAAK,GAAA,CAAI,IAAI,CAAA,iBAAA,EAAoB,MAAM,CAAA,YAAA,EAAe,OAAO,CAAA;AAAA,IAAA;AAE5E,IAAA;AAAmC,EAAA;AAEnC,IAAA;AAAU,EAAA;AAEZ;ADJA;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-NYYEZ4ER.cjs","sourcesContent":[null,"import type { NextFunction, Request, Response } from \"express\";\nimport type { AinHttpError } from \"@/types/agent\";\nimport { logger } from \"@/utils/logger\";\n\nexport const errorMiddleware = (\n\terror: AinHttpError,\n\treq: Request,\n\tres: Response,\n\tnext: NextFunction,\n): void => {\n\ttry {\n\t\tconst status: number = error.status || 500;\n\t\tconst message: string = error.message || \"Something went wrong\";\n\n\t\tlogger.error(\n\t\t\t`[${req.method}] ${req.path} >> StatusCode:: ${status}, Message:: ${message}`,\n\t\t);\n\t\tres.status(status).json({ message });\n\t} catch (error) {\n\t\tnext(error);\n\t}\n};\n"]}
|