ai-pieces-custom 0.0.6 → 0.0.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.
|
Binary file
|
package/package.json
CHANGED
|
@@ -47,6 +47,38 @@ function flattenMcpServers(servers) {
|
|
|
47
47
|
}
|
|
48
48
|
return { mcpClients, tools };
|
|
49
49
|
}
|
|
50
|
+
function buildAuthHeaders(authConfig, token) {
|
|
51
|
+
let headers = {};
|
|
52
|
+
console.log('Token from build auth', authConfig);
|
|
53
|
+
console.log({ authConfig });
|
|
54
|
+
console.log('Token :');
|
|
55
|
+
console.log({ token });
|
|
56
|
+
switch (authConfig.type) {
|
|
57
|
+
case shared_1.McpAuthType.NONE:
|
|
58
|
+
break;
|
|
59
|
+
case shared_1.McpAuthType.HEADERS: {
|
|
60
|
+
headers = authConfig.headers;
|
|
61
|
+
break;
|
|
62
|
+
}
|
|
63
|
+
case shared_1.McpAuthType.ACCESS_TOKEN: {
|
|
64
|
+
if (token != null) {
|
|
65
|
+
console.log("Setting oauth token");
|
|
66
|
+
headers['Authorization'] = `Bearer ${token}`;
|
|
67
|
+
}
|
|
68
|
+
else {
|
|
69
|
+
console.log("Normal user defined access token inuse");
|
|
70
|
+
headers['Authorization'] = `Bearer ${authConfig.accessToken}`;
|
|
71
|
+
}
|
|
72
|
+
break;
|
|
73
|
+
}
|
|
74
|
+
case shared_1.McpAuthType.API_KEY: {
|
|
75
|
+
const headerName = authConfig.apiKeyHeader;
|
|
76
|
+
headers[headerName] = authConfig.apiKey;
|
|
77
|
+
break;
|
|
78
|
+
}
|
|
79
|
+
}
|
|
80
|
+
return headers;
|
|
81
|
+
}
|
|
50
82
|
function constructMcpServersTools(tools, token) {
|
|
51
83
|
return tslib_1.__awaiter(this, void 0, void 0, function* () {
|
|
52
84
|
const collected = [];
|
|
@@ -56,7 +88,7 @@ function constructMcpServersTools(tools, token) {
|
|
|
56
88
|
for (const tool of tools) {
|
|
57
89
|
try {
|
|
58
90
|
const mcpClient = yield (0, mcp_1.experimental_createMCPClient)({
|
|
59
|
-
transport: createTransportConfig(tool.protocol, tool.serverUrl,
|
|
91
|
+
transport: createTransportConfig(tool.protocol, tool.serverUrl, buildAuthHeaders(tool.auth, token)),
|
|
60
92
|
});
|
|
61
93
|
const mcpTools = yield mcpClient.tools();
|
|
62
94
|
collected.push({
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../../../../../../../../packages/pieces/custom/ai-custom/src/lib/actions/agents/tools.ts"],"names":[],"mappings":";;
|
|
1
|
+
{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../../../../../../../../packages/pieces/custom/ai-custom/src/lib/actions/agents/tools.ts"],"names":[],"mappings":";;AAuIA,kDAkFC;;AAzND,2BAAsD;AACtD,sDAAoB;AACpB,mCAAqC;AAErC,iDAA4L;AAC5L,0FAAkG;AAElG,qCAAuG;AAEvG,SAAS,qBAAqB,CAC1B,QAAqB,EACrB,SAAiB,EACjB,UAAkC,EAAE;IAEpC,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,CAAA;IAE9B,QAAQ,QAAQ,EAAE,CAAC;QACf,KAAK,oBAAW,CAAC,WAAW,CAAC,CAAC,CAAC;YAC3B,OAAO;gBACH,IAAI,EAAE,MAAM;gBACZ,GAAG,EAAE,SAAS;gBACd,OAAO;aACV,CAAA;QACL,CAAC;QACD,KAAK,oBAAW,CAAC,eAAe,CAAC,CAAC,CAAC;YAC/B,OAAO,IAAI,iDAA6B,CAAC,GAAG,EAAE;gBAC1C,WAAW,EAAE;oBACT,OAAO;iBACV;aACJ,CAAC,CAAA;QACN,CAAC;QACD,KAAK,oBAAW,CAAC,GAAG,CAAC,CAAC,CAAC;YACnB,OAAO;gBACH,IAAI,EAAE,KAAK;gBACX,GAAG,EAAE,SAAS;gBACd,OAAO;aACV,CAAA;QACL,CAAC;QACD;YACI,MAAM,IAAI,KAAK,CAAC,kCAAkC,QAAQ,EAAE,CAAC,CAAA;IACrE,CAAC;AACL,CAAC;AAOD,SAAS,iBAAiB,CACtB,OAAyB;IAEzB,MAAM,UAAU,GAAgB,EAAE,CAAC;IACnC,MAAM,KAAK,GAAyB,EAAE,CAAC;IAEvC,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAElC,KAAK,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1D,KAAK,CAAC,GAAG,QAAQ,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;AACjC,CAAC;AAED,SAAS,gBAAgB,CAAC,UAAyB,EAAE,KAAoB;IACrE,IAAI,OAAO,GAA2B,EAAE,CAAA;IACxC,OAAO,CAAC,GAAG,CAAC,uBAAuB,EAAE,UAAU,CAAC,CAAA;IAChD,OAAO,CAAC,GAAG,CAAC,EAAE,UAAU,EAAE,CAAC,CAAA;IAC3B,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAA;IACtB,OAAO,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;IACtB,QAAQ,UAAU,CAAC,IAAI,EAAE,CAAC;QACtB,KAAK,oBAAW,CAAC,IAAI;YACjB,MAAK;QACT,KAAK,oBAAW,CAAC,OAAO,CAAC,CAAC,CAAC;YACvB,OAAO,GAAG,UAAU,CAAC,OAAO,CAAA;YAC5B,MAAK;QACT,CAAC;QACD,KAAK,oBAAW,CAAC,YAAY,CAAC,CAAC,CAAC;YAC5B,IAAI,KAAK,IAAI,IAAI,EAAI,CAAC;gBAClB,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAA;gBAClC,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,KAAK,EAAE,CAAA;YAChD,CAAC;iBAAM,CAAC;gBACJ,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAA;gBACrD,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,UAAU,CAAC,WAAW,EAAE,CAAA;YACjE,CAAC;YACD,MAAK;QACT,CAAC;QACD,KAAK,oBAAW,CAAC,OAAO,CAAC,CAAC,CAAC;YACvB,MAAM,UAAU,GAAG,UAAU,CAAC,YAAY,CAAA;YAC1C,OAAO,CAAC,UAAU,CAAC,GAAG,UAAU,CAAC,MAAM,CAAA;YACvC,MAAK;QACT,CAAC;IACL,CAAC;IAED,OAAO,OAAO,CAAA;AAClB,CAAC;AAED,SAAe,wBAAwB,CACnC,KAAqB,EACrB,KAAoB;;QAEpB,MAAM,SAAS,GAAqB,EAAE,CAAC;QACvC,OAAO,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAA;QAC5C,OAAO,CAAC,GAAG,CAAC,EAAC,KAAK,EAAC,CAAC,CAAA;QACpB,OAAO,CAAC,GAAG,CAAC,EAAC,KAAK,EAAC,CAAC,CAAA;QAEpB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,MAAM,SAAS,GAAG,MAAM,IAAA,kCAAe,EAAC;oBACtC,SAAS,EAAE,qBAAqB,CAC9B,IAAI,CAAC,QAAQ,EACb,IAAI,CAAC,SAAS,EACd,gBAAgB,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,CACnB;iBAClB,CAAC,CAAC;gBAEH,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC;gBAEzC,SAAS,CAAC,IAAI,CAAC;oBACb,OAAO,EAAE,IAAI,CAAC,QAAQ;oBACtB,SAAS;oBACT,KAAK,EAAE,QAAgC;iBACxC,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CACX,mCAAmC,IAAI,CAAC,SAAS,GAAG,EACpD,KAAK,CACN,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;CAAA;AAEH,SAAsB,mBAAmB,CACvC,MAAgC;;;QAG9B,MAAM,EAAE,aAAa,EAAE,gBAAgB,EAAE,UAAU,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;QAC/E,MAAM,eAAe,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC;YAChD,KAAK,EAAE,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,sBAAa,CAAC,KAAK,CAAC;YACnE,KAAK,EAAE,KAAK;SACb,CAAC,CAAC;QACH,MAAM,UAAU,GAAG,MAAM,kBAAU,CAAC,mBAAmB,CAAC;YACtD,KAAK,EAAE,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,sBAAa,CAAC,IAAI,CAAC;YAClE,UAAU,EAAE,OAAO,CAAC,KAAK,CAAC,IAAI;YAC9B,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,SAAS;YACnC,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK;SAC5B,CAAC,CAAA;QACF,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,cAAc,CAAC;QACjD,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAA;QACzB,OAAO,CAAC,GAAG,CAAC,EAAC,UAAU,EAAC,CAAC,CAAA;QACzB,MAAM,cAAc,GAAG,MAAM,wBAAwB,CACnD,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,sBAAa,CAAC,GAAG,CAAC,EAC1D,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAC/B,CAAA;QACD,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,iBAAiB,CAAC,cAAc,CAAC,CAAA;QAEzE,MAAM,aAAa,iDACd,eAAe,GACf,UAAU,GACV,QAAQ,CACZ,CAAC;QAEF,OAAO;YACH,UAAU;YACV,KAAK,kCACE,aAAa,KAChB,CAAC,kCAAyB,CAAC,EAAE,IAAA,gBAAW,EAAC;oBAC3C,WAAW,EACT,iQAAiQ;oBACnQ,WAAW,EAAE,aAAC,CAAC,MAAM,iBACnB,OAAO,EAAE,aAAC;6BACP,OAAO,EAAE;6BACT,QAAQ,CACP,8FAA8F,CAC/F,IACA,CAAC,CAAC,IAAA,cAAK,EAAC,gBAAgB,CAAC;wBAC1B,CAAC,CAAC;4BACE,MAAM,EAAE,aAAC;iCACN,MAAM,CACL,MAAA,MAAA,kBAAU,CAAC,sBAAsB,CAAC,gBAAgB,CAAC,0CAAE,KAAK,mCAAI,EAAE,CACjE;iCACA,QAAQ,EAAE;iCACV,QAAQ,CACP,4GAA4G,CAC7G;yBACJ;wBACH,CAAC,CAAC;4BACE,MAAM,EAAE,aAAC;iCACN,MAAM,EAAE;iCACR,QAAQ,EAAE;iCACV,QAAQ,CACP,8HAA8H,CAC/H;yBACJ,CAAC,EACN;oBACF,OAAO,EAAE,CAAO,MAAM,EAAE,EAAE;wBACxB,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,MAG3B,CAAC;wBACF,aAAa,CAAC,SAAS,CACrB,OAAO,CAAC,CAAC,CAAC,wBAAe,CAAC,SAAS,CAAC,CAAC,CAAC,wBAAe,CAAC,MAAM,CAC7D,CAAC;wBACF,IAAI,CAAC,IAAA,cAAK,EAAC,gBAAgB,CAAC,IAAI,CAAC,IAAA,cAAK,EAAC,MAAM,CAAC,EAAE,CAAC;4BAC/C,aAAa,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;wBAC5C,CAAC;wBACD,IAAI,CAAC,IAAA,cAAK,EAAC,gBAAgB,CAAC,IAAI,CAAC,IAAA,cAAK,EAAC,MAAM,CAAC,EAAE,CAAC;4BAC/C,aAAa,CAAC,WAAW,CAAC,MAA2B,CAAC,CAAC;wBACzD,CAAC;wBACD,OAAO,EAAE,CAAC;oBACZ,CAAC,CAAA;iBACF,CAAC,GACL;SACA,CAAA;IACL,CAAC;CAAA"}
|