@smythos/sre 1.5.46 → 1.5.50

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 (224) hide show
  1. package/CHANGELOG +98 -90
  2. package/LICENSE +18 -18
  3. package/README.md +135 -135
  4. package/dist/bundle-analysis-lazy.html +4949 -0
  5. package/dist/bundle-analysis.html +4949 -0
  6. package/dist/index.js +3 -3
  7. package/dist/index.js.map +1 -1
  8. package/dist/types/Components/MCPClient.class.d.ts +1 -0
  9. package/dist/types/subsystems/LLMManager/LLM.service/connectors/openai/OpenAIConnector.class.d.ts +1 -6
  10. package/dist/types/utils/package-manager.utils.d.ts +26 -0
  11. package/package.json +1 -1
  12. package/src/Components/APICall/APICall.class.ts +156 -156
  13. package/src/Components/APICall/AccessTokenManager.ts +130 -130
  14. package/src/Components/APICall/ArrayBufferResponse.helper.ts +58 -58
  15. package/src/Components/APICall/OAuth.helper.ts +294 -294
  16. package/src/Components/APICall/mimeTypeCategories.ts +46 -46
  17. package/src/Components/APICall/parseData.ts +167 -167
  18. package/src/Components/APICall/parseHeaders.ts +41 -41
  19. package/src/Components/APICall/parseProxy.ts +68 -68
  20. package/src/Components/APICall/parseUrl.ts +91 -91
  21. package/src/Components/APIEndpoint.class.ts +234 -234
  22. package/src/Components/APIOutput.class.ts +58 -58
  23. package/src/Components/AgentPlugin.class.ts +102 -102
  24. package/src/Components/Async.class.ts +155 -155
  25. package/src/Components/Await.class.ts +90 -90
  26. package/src/Components/Classifier.class.ts +158 -158
  27. package/src/Components/Component.class.ts +132 -132
  28. package/src/Components/ComponentHost.class.ts +38 -38
  29. package/src/Components/DataSourceCleaner.class.ts +92 -92
  30. package/src/Components/DataSourceIndexer.class.ts +181 -181
  31. package/src/Components/DataSourceLookup.class.ts +161 -161
  32. package/src/Components/ECMASandbox.class.ts +71 -71
  33. package/src/Components/FEncDec.class.ts +29 -29
  34. package/src/Components/FHash.class.ts +33 -33
  35. package/src/Components/FSign.class.ts +80 -80
  36. package/src/Components/FSleep.class.ts +25 -25
  37. package/src/Components/FTimestamp.class.ts +25 -25
  38. package/src/Components/FileStore.class.ts +78 -78
  39. package/src/Components/ForEach.class.ts +97 -97
  40. package/src/Components/GPTPlugin.class.ts +70 -70
  41. package/src/Components/GenAILLM.class.ts +586 -586
  42. package/src/Components/HuggingFace.class.ts +314 -314
  43. package/src/Components/Image/imageSettings.config.ts +70 -70
  44. package/src/Components/ImageGenerator.class.ts +502 -502
  45. package/src/Components/JSONFilter.class.ts +54 -54
  46. package/src/Components/LLMAssistant.class.ts +213 -213
  47. package/src/Components/LogicAND.class.ts +28 -28
  48. package/src/Components/LogicAtLeast.class.ts +85 -85
  49. package/src/Components/LogicAtMost.class.ts +86 -86
  50. package/src/Components/LogicOR.class.ts +29 -29
  51. package/src/Components/LogicXOR.class.ts +34 -34
  52. package/src/Components/MCPClient.class.ts +138 -112
  53. package/src/Components/MemoryDeleteKeyVal.class.ts +70 -70
  54. package/src/Components/MemoryReadKeyVal.class.ts +66 -66
  55. package/src/Components/MemoryWriteKeyVal.class.ts +62 -62
  56. package/src/Components/MemoryWriteObject.class.ts +97 -97
  57. package/src/Components/MultimodalLLM.class.ts +128 -128
  58. package/src/Components/OpenAPI.class.ts +72 -72
  59. package/src/Components/PromptGenerator.class.ts +122 -122
  60. package/src/Components/ScrapflyWebScrape.class.ts +159 -159
  61. package/src/Components/ServerlessCode.class.ts +123 -123
  62. package/src/Components/TavilyWebSearch.class.ts +98 -98
  63. package/src/Components/VisionLLM.class.ts +104 -104
  64. package/src/Components/ZapierAction.class.ts +127 -127
  65. package/src/Components/index.ts +97 -97
  66. package/src/Core/AgentProcess.helper.ts +240 -240
  67. package/src/Core/Connector.class.ts +123 -123
  68. package/src/Core/ConnectorsService.ts +197 -197
  69. package/src/Core/DummyConnector.ts +49 -49
  70. package/src/Core/HookService.ts +105 -105
  71. package/src/Core/SmythRuntime.class.ts +235 -235
  72. package/src/Core/SystemEvents.ts +16 -16
  73. package/src/Core/boot.ts +56 -56
  74. package/src/config.ts +15 -15
  75. package/src/constants.ts +126 -126
  76. package/src/data/hugging-face.params.json +579 -579
  77. package/src/helpers/AWSLambdaCode.helper.ts +587 -587
  78. package/src/helpers/BinaryInput.helper.ts +331 -331
  79. package/src/helpers/Conversation.helper.ts +1119 -1119
  80. package/src/helpers/ECMASandbox.helper.ts +54 -54
  81. package/src/helpers/JsonContent.helper.ts +97 -97
  82. package/src/helpers/LocalCache.helper.ts +97 -97
  83. package/src/helpers/Log.helper.ts +274 -274
  84. package/src/helpers/OpenApiParser.helper.ts +150 -150
  85. package/src/helpers/S3Cache.helper.ts +147 -147
  86. package/src/helpers/SmythURI.helper.ts +5 -5
  87. package/src/helpers/Sysconfig.helper.ts +77 -77
  88. package/src/helpers/TemplateString.helper.ts +243 -243
  89. package/src/helpers/TypeChecker.helper.ts +329 -329
  90. package/src/index.ts +3 -3
  91. package/src/index.ts.bak +3 -3
  92. package/src/subsystems/AgentManager/Agent.class.ts +1114 -1114
  93. package/src/subsystems/AgentManager/Agent.helper.ts +3 -3
  94. package/src/subsystems/AgentManager/AgentData.service/AgentDataConnector.ts +230 -230
  95. package/src/subsystems/AgentManager/AgentData.service/connectors/CLIAgentDataConnector.class.ts +66 -66
  96. package/src/subsystems/AgentManager/AgentData.service/connectors/LocalAgentDataConnector.class.ts +142 -142
  97. package/src/subsystems/AgentManager/AgentData.service/connectors/NullAgentData.class.ts +39 -39
  98. package/src/subsystems/AgentManager/AgentData.service/index.ts +18 -18
  99. package/src/subsystems/AgentManager/AgentLogger.class.ts +297 -297
  100. package/src/subsystems/AgentManager/AgentRequest.class.ts +51 -51
  101. package/src/subsystems/AgentManager/AgentRuntime.class.ts +559 -559
  102. package/src/subsystems/AgentManager/AgentSSE.class.ts +101 -101
  103. package/src/subsystems/AgentManager/AgentSettings.class.ts +52 -52
  104. package/src/subsystems/AgentManager/Component.service/ComponentConnector.ts +32 -32
  105. package/src/subsystems/AgentManager/Component.service/connectors/LocalComponentConnector.class.ts +60 -60
  106. package/src/subsystems/AgentManager/Component.service/index.ts +11 -11
  107. package/src/subsystems/AgentManager/EmbodimentSettings.class.ts +47 -47
  108. package/src/subsystems/AgentManager/ForkedAgent.class.ts +154 -154
  109. package/src/subsystems/AgentManager/OSResourceMonitor.ts +77 -77
  110. package/src/subsystems/ComputeManager/Code.service/CodeConnector.ts +98 -98
  111. package/src/subsystems/ComputeManager/Code.service/connectors/AWSLambdaCode.class.ts +172 -172
  112. package/src/subsystems/ComputeManager/Code.service/connectors/ECMASandbox.class.ts +131 -131
  113. package/src/subsystems/ComputeManager/Code.service/index.ts +13 -13
  114. package/src/subsystems/IO/CLI.service/CLIConnector.ts +47 -47
  115. package/src/subsystems/IO/CLI.service/index.ts +9 -9
  116. package/src/subsystems/IO/Log.service/LogConnector.ts +32 -32
  117. package/src/subsystems/IO/Log.service/connectors/ConsoleLog.class.ts +28 -28
  118. package/src/subsystems/IO/Log.service/index.ts +13 -13
  119. package/src/subsystems/IO/NKV.service/NKVConnector.ts +43 -43
  120. package/src/subsystems/IO/NKV.service/connectors/NKVLocalStorage.class.ts +234 -234
  121. package/src/subsystems/IO/NKV.service/connectors/NKVRAM.class.ts +204 -204
  122. package/src/subsystems/IO/NKV.service/connectors/NKVRedis.class.ts +182 -182
  123. package/src/subsystems/IO/NKV.service/index.ts +14 -14
  124. package/src/subsystems/IO/Router.service/RouterConnector.ts +21 -21
  125. package/src/subsystems/IO/Router.service/connectors/ExpressRouter.class.ts +48 -48
  126. package/src/subsystems/IO/Router.service/connectors/NullRouter.class.ts +40 -40
  127. package/src/subsystems/IO/Router.service/index.ts +11 -11
  128. package/src/subsystems/IO/Storage.service/SmythFS.class.ts +489 -489
  129. package/src/subsystems/IO/Storage.service/StorageConnector.ts +66 -66
  130. package/src/subsystems/IO/Storage.service/connectors/LocalStorage.class.ts +327 -327
  131. package/src/subsystems/IO/Storage.service/connectors/S3Storage.class.ts +482 -482
  132. package/src/subsystems/IO/Storage.service/index.ts +13 -13
  133. package/src/subsystems/IO/VectorDB.service/VectorDBConnector.ts +108 -108
  134. package/src/subsystems/IO/VectorDB.service/connectors/MilvusVectorDB.class.ts +454 -454
  135. package/src/subsystems/IO/VectorDB.service/connectors/PineconeVectorDB.class.ts +384 -384
  136. package/src/subsystems/IO/VectorDB.service/connectors/RAMVecrtorDB.class.ts +421 -421
  137. package/src/subsystems/IO/VectorDB.service/embed/BaseEmbedding.ts +107 -107
  138. package/src/subsystems/IO/VectorDB.service/embed/OpenAIEmbedding.ts +109 -109
  139. package/src/subsystems/IO/VectorDB.service/embed/index.ts +21 -21
  140. package/src/subsystems/IO/VectorDB.service/index.ts +14 -14
  141. package/src/subsystems/LLMManager/LLM.helper.ts +251 -251
  142. package/src/subsystems/LLMManager/LLM.inference.ts +339 -339
  143. package/src/subsystems/LLMManager/LLM.service/LLMConnector.ts +489 -489
  144. package/src/subsystems/LLMManager/LLM.service/LLMCredentials.helper.ts +171 -171
  145. package/src/subsystems/LLMManager/LLM.service/connectors/Anthropic.class.ts +659 -659
  146. package/src/subsystems/LLMManager/LLM.service/connectors/Bedrock.class.ts +400 -400
  147. package/src/subsystems/LLMManager/LLM.service/connectors/Echo.class.ts +77 -77
  148. package/src/subsystems/LLMManager/LLM.service/connectors/GoogleAI.class.ts +757 -757
  149. package/src/subsystems/LLMManager/LLM.service/connectors/Groq.class.ts +304 -304
  150. package/src/subsystems/LLMManager/LLM.service/connectors/Perplexity.class.ts +250 -250
  151. package/src/subsystems/LLMManager/LLM.service/connectors/VertexAI.class.ts +423 -423
  152. package/src/subsystems/LLMManager/LLM.service/connectors/openai/OpenAIConnector.class.ts +488 -488
  153. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/ChatCompletionsApiInterface.ts +524 -524
  154. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/OpenAIApiInterface.ts +100 -100
  155. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/OpenAIApiInterfaceFactory.ts +81 -81
  156. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/ResponsesApiInterface.ts +1145 -1145
  157. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/constants.ts +13 -13
  158. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/index.ts +4 -4
  159. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/utils.ts +11 -11
  160. package/src/subsystems/LLMManager/LLM.service/connectors/openai/types.ts +32 -32
  161. package/src/subsystems/LLMManager/LLM.service/connectors/xAI.class.ts +471 -471
  162. package/src/subsystems/LLMManager/LLM.service/index.ts +44 -44
  163. package/src/subsystems/LLMManager/ModelsProvider.service/ModelsProviderConnector.ts +300 -300
  164. package/src/subsystems/LLMManager/ModelsProvider.service/connectors/JSONModelsProvider.class.ts +252 -252
  165. package/src/subsystems/LLMManager/ModelsProvider.service/index.ts +11 -11
  166. package/src/subsystems/LLMManager/custom-models.ts +854 -854
  167. package/src/subsystems/LLMManager/models.ts +2540 -2540
  168. package/src/subsystems/LLMManager/paramMappings.ts +69 -69
  169. package/src/subsystems/MemoryManager/Cache.service/CacheConnector.ts +86 -86
  170. package/src/subsystems/MemoryManager/Cache.service/connectors/LocalStorageCache.class.ts +297 -297
  171. package/src/subsystems/MemoryManager/Cache.service/connectors/RAMCache.class.ts +201 -201
  172. package/src/subsystems/MemoryManager/Cache.service/connectors/RedisCache.class.ts +252 -252
  173. package/src/subsystems/MemoryManager/Cache.service/connectors/S3Cache.class.ts +373 -373
  174. package/src/subsystems/MemoryManager/Cache.service/index.ts +15 -15
  175. package/src/subsystems/MemoryManager/LLMCache.ts +72 -72
  176. package/src/subsystems/MemoryManager/LLMContext.ts +124 -124
  177. package/src/subsystems/MemoryManager/LLMMemory.service/LLMMemoryConnector.ts +26 -26
  178. package/src/subsystems/MemoryManager/RuntimeContext.ts +266 -266
  179. package/src/subsystems/Security/AccessControl/ACL.class.ts +208 -208
  180. package/src/subsystems/Security/AccessControl/AccessCandidate.class.ts +82 -82
  181. package/src/subsystems/Security/AccessControl/AccessRequest.class.ts +52 -52
  182. package/src/subsystems/Security/Account.service/AccountConnector.ts +44 -44
  183. package/src/subsystems/Security/Account.service/connectors/AWSAccount.class.ts +76 -76
  184. package/src/subsystems/Security/Account.service/connectors/DummyAccount.class.ts +130 -130
  185. package/src/subsystems/Security/Account.service/connectors/JSONFileAccount.class.ts +159 -159
  186. package/src/subsystems/Security/Account.service/index.ts +14 -14
  187. package/src/subsystems/Security/Credentials.helper.ts +62 -62
  188. package/src/subsystems/Security/ManagedVault.service/ManagedVaultConnector.ts +38 -38
  189. package/src/subsystems/Security/ManagedVault.service/connectors/NullManagedVault.class.ts +53 -53
  190. package/src/subsystems/Security/ManagedVault.service/connectors/SecretManagerManagedVault.ts +154 -154
  191. package/src/subsystems/Security/ManagedVault.service/index.ts +12 -12
  192. package/src/subsystems/Security/SecureConnector.class.ts +110 -110
  193. package/src/subsystems/Security/Vault.service/Vault.helper.ts +30 -30
  194. package/src/subsystems/Security/Vault.service/VaultConnector.ts +29 -29
  195. package/src/subsystems/Security/Vault.service/connectors/HashicorpVault.class.ts +46 -46
  196. package/src/subsystems/Security/Vault.service/connectors/JSONFileVault.class.ts +221 -221
  197. package/src/subsystems/Security/Vault.service/connectors/NullVault.class.ts +54 -54
  198. package/src/subsystems/Security/Vault.service/connectors/SecretsManager.class.ts +140 -140
  199. package/src/subsystems/Security/Vault.service/index.ts +12 -12
  200. package/src/types/ACL.types.ts +104 -104
  201. package/src/types/AWS.types.ts +10 -10
  202. package/src/types/Agent.types.ts +61 -61
  203. package/src/types/AgentLogger.types.ts +17 -17
  204. package/src/types/Cache.types.ts +1 -1
  205. package/src/types/Common.types.ts +2 -2
  206. package/src/types/LLM.types.ts +496 -496
  207. package/src/types/Redis.types.ts +8 -8
  208. package/src/types/SRE.types.ts +64 -64
  209. package/src/types/Security.types.ts +14 -14
  210. package/src/types/Storage.types.ts +5 -5
  211. package/src/types/VectorDB.types.ts +86 -86
  212. package/src/utils/base64.utils.ts +275 -275
  213. package/src/utils/cli.utils.ts +68 -68
  214. package/src/utils/data.utils.ts +322 -322
  215. package/src/utils/date-time.utils.ts +22 -22
  216. package/src/utils/general.utils.ts +238 -238
  217. package/src/utils/index.ts +12 -12
  218. package/src/utils/lazy-client.ts +261 -261
  219. package/src/utils/numbers.utils.ts +13 -13
  220. package/src/utils/oauth.utils.ts +35 -35
  221. package/src/utils/string.utils.ts +414 -414
  222. package/src/utils/url.utils.ts +19 -19
  223. package/src/utils/validation.utils.ts +74 -74
  224. package/dist/types/subsystems/LLMManager/ModelsProvider.service/connectors/SmythModelsProvider.class.d.ts +0 -39
@@ -1,123 +1,123 @@
1
- import { IAgent as Agent } from '@sre/types/Agent.types';
2
- import { Component } from './Component.class';
3
- import Joi from 'joi';
4
- import { ConnectorService } from '@sre/Core/ConnectorsService';
5
- import { AWSCredentials, AWSRegionConfig } from '@sre/types/AWS.types';
6
- import { calculateExecutionCost, generateCodeFromLegacyComponent, getLambdaCredentials, reportUsage } from '@sre/helpers/AWSLambdaCode.helper';
7
- import { AccessCandidate } from '@sre/Security/AccessControl/AccessCandidate.class';
8
-
9
- export class ServerlessCode extends Component {
10
-
11
- protected configSchema = Joi.object({
12
- code_imports: Joi.string().max(1000).allow('').label('Imports'),
13
- code_body: Joi.string().max(500000).allow('').label('Code'),
14
- code: Joi.string().max(500000).allow('').label('Code').optional(),
15
- deploy_btn: Joi.string().max(500000).allow('').label('Deploy').optional(),
16
- accessKeyId: Joi.string().max(100).allow('').label('AWS Access Key ID').optional(),
17
- secretAccessKey: Joi.string().max(200).allow('').label('AWS Secret Access Key').optional(),
18
- region: Joi.string().label('AWS Region').optional(),
19
- _templateSettings: Joi.object().allow(null).label('Template Settings'),
20
- _templateVars: Joi.object().allow(null).label('Template Variables'),
21
- function_label: Joi.string().max(100).allow('').label('Function Label').optional(),
22
- function_label_end: Joi.string().allow(null).label('Function Label End').optional(),
23
- use_own_keys: Joi.boolean().label('Use Own Keys').optional(),
24
- pricing_note: Joi.string().allow(null).label('Pricing Note').optional(),
25
- });
26
- constructor() {
27
- super();
28
- }
29
- init() { }
30
-
31
- async process(input, config, agent: Agent) {
32
- await new Promise(resolve => setTimeout(resolve, 10000));
33
- await super.process(input, config, agent);
34
- const logger = this.createComponentLogger(agent, config);
35
- try {
36
- logger.debug(`=== Serverless Code Log ===`);
37
- let Output: any = {};
38
- let _error = undefined;
39
- const componentInputs = agent.components[config.id]?.inputs || {};
40
-
41
- let codeInputs = {};
42
- for (let field of componentInputs) {
43
- const _type = typeof input[field.name];
44
- switch (_type) {
45
- case 'string':
46
- try {
47
- codeInputs[field.name] = JSON.parse(input[field.name].replace(/\\"/g, '"'));
48
- } catch (error) {
49
- codeInputs[field.name] = `${input[field.name]}`;
50
- }
51
- break;
52
- case 'number':
53
- case 'boolean':
54
- codeInputs[field.name] = input[field.name];
55
- break;
56
- default:
57
- codeInputs[field.name] = input[field.name];
58
- break;
59
- }
60
- }
61
-
62
- logger.debug(`\nInput Variables: \n${JSON.stringify(codeInputs, null, 2)}\n`);
63
-
64
- let codeConnector = ConnectorService.getCodeConnector();
65
- let codeCredentials: AWSCredentials & AWSRegionConfig & { isUserProvidedKeys: boolean } =
66
- await getLambdaCredentials(agent, config);
67
-
68
- if (codeCredentials.isUserProvidedKeys) {
69
- codeConnector = codeConnector.instance({
70
- region: codeCredentials.region,
71
- accessKeyId: codeCredentials.accessKeyId,
72
- secretAccessKey: codeCredentials.secretAccessKey,
73
- })
74
- }
75
- let code = config?.data?.code;
76
- if (!code) {
77
- code = generateCodeFromLegacyComponent(config.data.code_body, config.data.code_imports, Object.keys(codeInputs))
78
- }
79
- // Deploy lambda function if it doesn't exist or the code hash is different
80
- await codeConnector.agent(agent.id)
81
- .deploy(config.id, {
82
- code,
83
- inputs: codeInputs,
84
- }, {
85
- runtime: 'nodejs',
86
- });
87
-
88
- try {
89
- const executionResponse = await codeConnector.agent(agent.id).execute(config.id, codeInputs);
90
- const executionTime = executionResponse.executionTime;
91
- logger.debug(
92
- `Code result:\n ${typeof executionResponse.output === 'object' ? JSON.stringify(executionResponse.output, null, 2) : executionResponse.output
93
- }\n`,
94
- );
95
- logger.debug(`Execution time: ${executionTime}ms\n`);
96
-
97
- const cost = calculateExecutionCost(executionTime);
98
- if (!codeCredentials.isUserProvidedKeys) {
99
- const accountConnector = ConnectorService.getAccountConnector();
100
- const agentTeam = await accountConnector.getCandidateTeam(AccessCandidate.agent(agent.id));
101
- reportUsage({ cost, agentId: agent.id, teamId: agentTeam });
102
- }
103
-
104
- if (executionResponse.success) {
105
- Output = executionResponse.output;
106
- } else {
107
- Output = undefined;
108
- _error = executionResponse.errors;
109
- }
110
- } catch (error: any) {
111
- logger.error(`Error running code \n${error}\n`);
112
- _error = error?.response?.data || error?.message || error.toString();
113
- Output = undefined; //prevents running next component if the code execution failed
114
- }
115
-
116
- return { Output, _error, _debug: logger.output };
117
- } catch (err: any) {
118
- const _error = err?.response?.data || err?.message || err.toString();
119
- logger.error(` Error running code \n${_error}\n`);
120
- return { Output: undefined, _error, _debug: logger.output };
121
- }
122
- }
123
- }
1
+ import { IAgent as Agent } from '@sre/types/Agent.types';
2
+ import { Component } from './Component.class';
3
+ import Joi from 'joi';
4
+ import { ConnectorService } from '@sre/Core/ConnectorsService';
5
+ import { AWSCredentials, AWSRegionConfig } from '@sre/types/AWS.types';
6
+ import { calculateExecutionCost, generateCodeFromLegacyComponent, getLambdaCredentials, reportUsage } from '@sre/helpers/AWSLambdaCode.helper';
7
+ import { AccessCandidate } from '@sre/Security/AccessControl/AccessCandidate.class';
8
+
9
+ export class ServerlessCode extends Component {
10
+
11
+ protected configSchema = Joi.object({
12
+ code_imports: Joi.string().max(1000).allow('').label('Imports'),
13
+ code_body: Joi.string().max(500000).allow('').label('Code'),
14
+ code: Joi.string().max(500000).allow('').label('Code').optional(),
15
+ deploy_btn: Joi.string().max(500000).allow('').label('Deploy').optional(),
16
+ accessKeyId: Joi.string().max(100).allow('').label('AWS Access Key ID').optional(),
17
+ secretAccessKey: Joi.string().max(200).allow('').label('AWS Secret Access Key').optional(),
18
+ region: Joi.string().label('AWS Region').optional(),
19
+ _templateSettings: Joi.object().allow(null).label('Template Settings'),
20
+ _templateVars: Joi.object().allow(null).label('Template Variables'),
21
+ function_label: Joi.string().max(100).allow('').label('Function Label').optional(),
22
+ function_label_end: Joi.string().allow(null).label('Function Label End').optional(),
23
+ use_own_keys: Joi.boolean().label('Use Own Keys').optional(),
24
+ pricing_note: Joi.string().allow(null).label('Pricing Note').optional(),
25
+ });
26
+ constructor() {
27
+ super();
28
+ }
29
+ init() { }
30
+
31
+ async process(input, config, agent: Agent) {
32
+ await new Promise(resolve => setTimeout(resolve, 10000));
33
+ await super.process(input, config, agent);
34
+ const logger = this.createComponentLogger(agent, config);
35
+ try {
36
+ logger.debug(`=== Serverless Code Log ===`);
37
+ let Output: any = {};
38
+ let _error = undefined;
39
+ const componentInputs = agent.components[config.id]?.inputs || {};
40
+
41
+ let codeInputs = {};
42
+ for (let field of componentInputs) {
43
+ const _type = typeof input[field.name];
44
+ switch (_type) {
45
+ case 'string':
46
+ try {
47
+ codeInputs[field.name] = JSON.parse(input[field.name].replace(/\\"/g, '"'));
48
+ } catch (error) {
49
+ codeInputs[field.name] = `${input[field.name]}`;
50
+ }
51
+ break;
52
+ case 'number':
53
+ case 'boolean':
54
+ codeInputs[field.name] = input[field.name];
55
+ break;
56
+ default:
57
+ codeInputs[field.name] = input[field.name];
58
+ break;
59
+ }
60
+ }
61
+
62
+ logger.debug(`\nInput Variables: \n${JSON.stringify(codeInputs, null, 2)}\n`);
63
+
64
+ let codeConnector = ConnectorService.getCodeConnector();
65
+ let codeCredentials: AWSCredentials & AWSRegionConfig & { isUserProvidedKeys: boolean } =
66
+ await getLambdaCredentials(agent, config);
67
+
68
+ if (codeCredentials.isUserProvidedKeys) {
69
+ codeConnector = codeConnector.instance({
70
+ region: codeCredentials.region,
71
+ accessKeyId: codeCredentials.accessKeyId,
72
+ secretAccessKey: codeCredentials.secretAccessKey,
73
+ })
74
+ }
75
+ let code = config?.data?.code;
76
+ if (!code) {
77
+ code = generateCodeFromLegacyComponent(config.data.code_body, config.data.code_imports, Object.keys(codeInputs))
78
+ }
79
+ // Deploy lambda function if it doesn't exist or the code hash is different
80
+ await codeConnector.agent(agent.id)
81
+ .deploy(config.id, {
82
+ code,
83
+ inputs: codeInputs,
84
+ }, {
85
+ runtime: 'nodejs',
86
+ });
87
+
88
+ try {
89
+ const executionResponse = await codeConnector.agent(agent.id).execute(config.id, codeInputs);
90
+ const executionTime = executionResponse.executionTime;
91
+ logger.debug(
92
+ `Code result:\n ${typeof executionResponse.output === 'object' ? JSON.stringify(executionResponse.output, null, 2) : executionResponse.output
93
+ }\n`,
94
+ );
95
+ logger.debug(`Execution time: ${executionTime}ms\n`);
96
+
97
+ const cost = calculateExecutionCost(executionTime);
98
+ if (!codeCredentials.isUserProvidedKeys) {
99
+ const accountConnector = ConnectorService.getAccountConnector();
100
+ const agentTeam = await accountConnector.getCandidateTeam(AccessCandidate.agent(agent.id));
101
+ reportUsage({ cost, agentId: agent.id, teamId: agentTeam });
102
+ }
103
+
104
+ if (executionResponse.success) {
105
+ Output = executionResponse.output;
106
+ } else {
107
+ Output = undefined;
108
+ _error = executionResponse.errors;
109
+ }
110
+ } catch (error: any) {
111
+ logger.error(`Error running code \n${error}\n`);
112
+ _error = error?.response?.data || error?.message || error.toString();
113
+ Output = undefined; //prevents running next component if the code execution failed
114
+ }
115
+
116
+ return { Output, _error, _debug: logger.output };
117
+ } catch (err: any) {
118
+ const _error = err?.response?.data || err?.message || err.toString();
119
+ logger.error(` Error running code \n${_error}\n`);
120
+ return { Output: undefined, _error, _debug: logger.output };
121
+ }
122
+ }
123
+ }
@@ -1,98 +1,98 @@
1
- import { IAgent as Agent } from '@sre/types/Agent.types';
2
- import { Component } from './Component.class';
3
- import Joi from 'joi';
4
- import SREConfig from '@sre/config';
5
- import axios from 'axios';
6
- import { SystemEvents } from '@sre/Core/SystemEvents';
7
- import { getCredentials } from '../subsystems/Security/Credentials.helper';
8
- import { AccessCandidate } from '../subsystems/Security/AccessControl/AccessCandidate.class';
9
-
10
- export class TavilyWebSearch extends Component {
11
- protected schema = {
12
- name: 'TavilyWebSearch',
13
- description: 'Use this component to generate a responses from an LLM',
14
-
15
- inputs: {
16
- SearchQuery: {
17
- type: 'Text',
18
- description: 'The search query to get the web search results of',
19
- default: true,
20
- },
21
- },
22
- outputs: {
23
- Results: {
24
- type: 'Array',
25
- description: 'The web search results',
26
- default: true,
27
- },
28
- },
29
- };
30
-
31
- protected configSchema = Joi.object({
32
- includeImages: Joi.boolean().default(false).label('Include Image Results'),
33
- sourcesLimit: Joi.number().integer().default(10).label('Sources Limit'),
34
- searchTopic: Joi.string().valid('general', 'news').label('Search Topic'),
35
- includeQAs: Joi.boolean().default(false).label('Include QAs'),
36
- timeRange: Joi.string().valid('None', 'day', 'week', 'month', 'year').label('Time Range'),
37
- includeRawContent: Joi.boolean().default(false).label('Include Raw Content'),
38
- excludeDomains: Joi.string().allow('').label('Exclude Domains'),
39
- });
40
- constructor() {
41
- super();
42
- }
43
- init() {}
44
- async process(input, config, agent: Agent) {
45
- await super.process(input, config, agent);
46
-
47
- const logger = this.createComponentLogger(agent, config);
48
- try {
49
- logger.debug(`=== Web Scrape Log ===`);
50
- let Output: any = {};
51
- let _error = undefined;
52
- let searchQuery = input['SearchQuery'];
53
-
54
- const teamId = agent.teamId;
55
- const api_key = await getCredentials(AccessCandidate.team(teamId), 'tavily');
56
-
57
- logger.debug('Payload:', JSON.stringify(config.data));
58
- const response = await axios({
59
- method: 'post',
60
- url: 'https://api.tavily.com/search',
61
- data: {
62
- api_key,
63
- query: searchQuery,
64
- topic: config.data.searchTopic,
65
- exclude_domains: config.data.excludeDomains?.length ? config.data.excludeDomains.split(',') : [],
66
- max_results: config.data.sourcesLimit,
67
- ...(config.data.timeRange !== 'None' ? { time_range: config.data.timeRange } : {}),
68
- ...(config.data.includeImages ? { include_images: true } : {}),
69
- ...(config.data.includeQAs ? { include_answer: true } : {}),
70
- ...(config.data.includeRawContent ? { include_raw_content: true } : {}),
71
- },
72
- });
73
- Output = {
74
- Results: response.data.results,
75
- ...(config.data.includeImages ? { Images: response.data.images } : {}),
76
- ...(config.data.includeQAs ? { Answer: response.data.answer } : {}),
77
- };
78
- this.reportUsage({
79
- agentId: agent.id,
80
- teamId: agent.teamId,
81
- });
82
- return { ...Output, _error, _debug: logger.output };
83
- } catch (err: any) {
84
- const _error = err?.message || err?.response?.data || err.toString();
85
- logger.error(` Error scraping web \n${JSON.stringify(_error)}\n`);
86
- return { Output: undefined, _error, _debug: logger.output };
87
- }
88
- }
89
-
90
- protected reportUsage({ agentId, teamId }: { agentId: string; teamId: string }) {
91
- SystemEvents.emit('USAGE:API', {
92
- sourceId: 'api:websearch.smyth',
93
- credits: 1,
94
- agentId,
95
- teamId,
96
- });
97
- }
98
- }
1
+ import { IAgent as Agent } from '@sre/types/Agent.types';
2
+ import { Component } from './Component.class';
3
+ import Joi from 'joi';
4
+ import SREConfig from '@sre/config';
5
+ import axios from 'axios';
6
+ import { SystemEvents } from '@sre/Core/SystemEvents';
7
+ import { getCredentials } from '../subsystems/Security/Credentials.helper';
8
+ import { AccessCandidate } from '../subsystems/Security/AccessControl/AccessCandidate.class';
9
+
10
+ export class TavilyWebSearch extends Component {
11
+ protected schema = {
12
+ name: 'TavilyWebSearch',
13
+ description: 'Use this component to generate a responses from an LLM',
14
+
15
+ inputs: {
16
+ SearchQuery: {
17
+ type: 'Text',
18
+ description: 'The search query to get the web search results of',
19
+ default: true,
20
+ },
21
+ },
22
+ outputs: {
23
+ Results: {
24
+ type: 'Array',
25
+ description: 'The web search results',
26
+ default: true,
27
+ },
28
+ },
29
+ };
30
+
31
+ protected configSchema = Joi.object({
32
+ includeImages: Joi.boolean().default(false).label('Include Image Results'),
33
+ sourcesLimit: Joi.number().integer().default(10).label('Sources Limit'),
34
+ searchTopic: Joi.string().valid('general', 'news').label('Search Topic'),
35
+ includeQAs: Joi.boolean().default(false).label('Include QAs'),
36
+ timeRange: Joi.string().valid('None', 'day', 'week', 'month', 'year').label('Time Range'),
37
+ includeRawContent: Joi.boolean().default(false).label('Include Raw Content'),
38
+ excludeDomains: Joi.string().allow('').label('Exclude Domains'),
39
+ });
40
+ constructor() {
41
+ super();
42
+ }
43
+ init() {}
44
+ async process(input, config, agent: Agent) {
45
+ await super.process(input, config, agent);
46
+
47
+ const logger = this.createComponentLogger(agent, config);
48
+ try {
49
+ logger.debug(`=== Web Scrape Log ===`);
50
+ let Output: any = {};
51
+ let _error = undefined;
52
+ let searchQuery = input['SearchQuery'];
53
+
54
+ const teamId = agent.teamId;
55
+ const api_key = await getCredentials(AccessCandidate.team(teamId), 'tavily');
56
+
57
+ logger.debug('Payload:', JSON.stringify(config.data));
58
+ const response = await axios({
59
+ method: 'post',
60
+ url: 'https://api.tavily.com/search',
61
+ data: {
62
+ api_key,
63
+ query: searchQuery,
64
+ topic: config.data.searchTopic,
65
+ exclude_domains: config.data.excludeDomains?.length ? config.data.excludeDomains.split(',') : [],
66
+ max_results: config.data.sourcesLimit,
67
+ ...(config.data.timeRange !== 'None' ? { time_range: config.data.timeRange } : {}),
68
+ ...(config.data.includeImages ? { include_images: true } : {}),
69
+ ...(config.data.includeQAs ? { include_answer: true } : {}),
70
+ ...(config.data.includeRawContent ? { include_raw_content: true } : {}),
71
+ },
72
+ });
73
+ Output = {
74
+ Results: response.data.results,
75
+ ...(config.data.includeImages ? { Images: response.data.images } : {}),
76
+ ...(config.data.includeQAs ? { Answer: response.data.answer } : {}),
77
+ };
78
+ this.reportUsage({
79
+ agentId: agent.id,
80
+ teamId: agent.teamId,
81
+ });
82
+ return { ...Output, _error, _debug: logger.output };
83
+ } catch (err: any) {
84
+ const _error = err?.message || err?.response?.data || err.toString();
85
+ logger.error(` Error scraping web \n${JSON.stringify(_error)}\n`);
86
+ return { Output: undefined, _error, _debug: logger.output };
87
+ }
88
+ }
89
+
90
+ protected reportUsage({ agentId, teamId }: { agentId: string; teamId: string }) {
91
+ SystemEvents.emit('USAGE:API', {
92
+ sourceId: 'api:websearch.smyth',
93
+ credits: 1,
94
+ agentId,
95
+ teamId,
96
+ });
97
+ }
98
+ }
@@ -1,104 +1,104 @@
1
- import Joi from 'joi';
2
-
3
- import { TemplateString } from '@sre/helpers/TemplateString.helper';
4
- import { Component } from './Component.class';
5
- import { LLMInference } from '@sre/LLMManager/LLM.inference';
6
- import { AccessCandidate } from '@sre/Security/AccessControl/AccessCandidate.class';
7
-
8
- export class VisionLLM extends Component {
9
- protected configSchema = Joi.object({
10
- prompt: Joi.string().required().max(8_000_000).label('Prompt'), // 2M tokens is around 8M characters
11
- maxTokens: Joi.number().min(1).label('Maximum Tokens'),
12
- model: Joi.string().max(200).required(),
13
- passthrough: Joi.boolean().optional().label('Passthrough'),
14
- });
15
-
16
- constructor() {
17
- super();
18
- }
19
-
20
- init() {}
21
-
22
- async process(input, config, agent) {
23
- await super.process(input, config, agent);
24
-
25
- const logger = this.createComponentLogger(agent, config);
26
- try {
27
- logger.debug(`=== Vision LLM Log ===`);
28
-
29
- const passThrough: boolean = config.data.passthrough || false;
30
- const model: string = config.data?.model;
31
-
32
- const llmInference: LLMInference = await LLMInference.getInstance(model, AccessCandidate.agent(agent.id));
33
- // if the llm is undefined, then it means we removed the model from our system
34
- if (!llmInference.connector) {
35
- return {
36
- _error: `The model '${model}' is not available. Please try a different one.`,
37
- _debug: logger.output,
38
- };
39
- }
40
-
41
- const modelId = await agent.modelsProvider.getModelId(model);
42
- logger.debug(` Model : ${modelId || model}`);
43
-
44
- let prompt: any = TemplateString(config.data.prompt).parse(input).result;
45
-
46
- logger.debug(` Prompt\n`, prompt, '\n');
47
-
48
- const files = Array.isArray(input.Images) ? input.Images : [input.Images];
49
-
50
- let response: any;
51
- if (passThrough) {
52
- const contentPromise = new Promise(async (resolve, reject) => {
53
- let _content = '';
54
- const eventEmitter: any = await llmInference.multimodalStreamRequestLegacy(prompt, files, config, agent).catch((error) => {
55
- console.error('Error on multimodalStreamRequest: ', error);
56
- reject(error);
57
- });
58
- eventEmitter.on('content', (content) => {
59
- if (typeof agent.callback === 'function') {
60
- agent.callback({ content });
61
- }
62
- agent.sse.send('llm/passthrough/content', content);
63
- _content += content;
64
- });
65
- eventEmitter.on('thinking', (thinking) => {
66
- if (typeof agent.callback === 'function') {
67
- agent.callback({ thinking });
68
- }
69
- agent.sse.send('llm/passthrough/thinking', thinking);
70
- });
71
- eventEmitter.on('end', () => {
72
- console.log('end');
73
- resolve(_content);
74
- });
75
- });
76
- response = await contentPromise;
77
- } else {
78
- response = await llmInference.prompt({ query: prompt, files, params: { ...config, agentId: agent.id } });
79
- }
80
-
81
- // in case we have the response but it's empty string, undefined or null
82
- if (!response) {
83
- return { _error: ' LLM Error = Empty Response!', _debug: logger.output };
84
- }
85
-
86
- if (response?.error) {
87
- const error = response?.error + ' ' + (response?.details || '');
88
- logger.error(` LLM Error=`, error);
89
-
90
- return { Reply: response?.data, _error: error, _debug: logger.output };
91
- }
92
-
93
- logger.debug(' Response \n', response);
94
-
95
- const result = { Reply: response };
96
-
97
- result['_debug'] = logger.output;
98
-
99
- return result;
100
- } catch (error: any) {
101
- return { _error: error.message, _debug: logger.output };
102
- }
103
- }
104
- }
1
+ import Joi from 'joi';
2
+
3
+ import { TemplateString } from '@sre/helpers/TemplateString.helper';
4
+ import { Component } from './Component.class';
5
+ import { LLMInference } from '@sre/LLMManager/LLM.inference';
6
+ import { AccessCandidate } from '@sre/Security/AccessControl/AccessCandidate.class';
7
+
8
+ export class VisionLLM extends Component {
9
+ protected configSchema = Joi.object({
10
+ prompt: Joi.string().required().max(8_000_000).label('Prompt'), // 2M tokens is around 8M characters
11
+ maxTokens: Joi.number().min(1).label('Maximum Tokens'),
12
+ model: Joi.string().max(200).required(),
13
+ passthrough: Joi.boolean().optional().label('Passthrough'),
14
+ });
15
+
16
+ constructor() {
17
+ super();
18
+ }
19
+
20
+ init() {}
21
+
22
+ async process(input, config, agent) {
23
+ await super.process(input, config, agent);
24
+
25
+ const logger = this.createComponentLogger(agent, config);
26
+ try {
27
+ logger.debug(`=== Vision LLM Log ===`);
28
+
29
+ const passThrough: boolean = config.data.passthrough || false;
30
+ const model: string = config.data?.model;
31
+
32
+ const llmInference: LLMInference = await LLMInference.getInstance(model, AccessCandidate.agent(agent.id));
33
+ // if the llm is undefined, then it means we removed the model from our system
34
+ if (!llmInference.connector) {
35
+ return {
36
+ _error: `The model '${model}' is not available. Please try a different one.`,
37
+ _debug: logger.output,
38
+ };
39
+ }
40
+
41
+ const modelId = await agent.modelsProvider.getModelId(model);
42
+ logger.debug(` Model : ${modelId || model}`);
43
+
44
+ let prompt: any = TemplateString(config.data.prompt).parse(input).result;
45
+
46
+ logger.debug(` Prompt\n`, prompt, '\n');
47
+
48
+ const files = Array.isArray(input.Images) ? input.Images : [input.Images];
49
+
50
+ let response: any;
51
+ if (passThrough) {
52
+ const contentPromise = new Promise(async (resolve, reject) => {
53
+ let _content = '';
54
+ const eventEmitter: any = await llmInference.multimodalStreamRequestLegacy(prompt, files, config, agent).catch((error) => {
55
+ console.error('Error on multimodalStreamRequest: ', error);
56
+ reject(error);
57
+ });
58
+ eventEmitter.on('content', (content) => {
59
+ if (typeof agent.callback === 'function') {
60
+ agent.callback({ content });
61
+ }
62
+ agent.sse.send('llm/passthrough/content', content);
63
+ _content += content;
64
+ });
65
+ eventEmitter.on('thinking', (thinking) => {
66
+ if (typeof agent.callback === 'function') {
67
+ agent.callback({ thinking });
68
+ }
69
+ agent.sse.send('llm/passthrough/thinking', thinking);
70
+ });
71
+ eventEmitter.on('end', () => {
72
+ console.log('end');
73
+ resolve(_content);
74
+ });
75
+ });
76
+ response = await contentPromise;
77
+ } else {
78
+ response = await llmInference.prompt({ query: prompt, files, params: { ...config, agentId: agent.id } });
79
+ }
80
+
81
+ // in case we have the response but it's empty string, undefined or null
82
+ if (!response) {
83
+ return { _error: ' LLM Error = Empty Response!', _debug: logger.output };
84
+ }
85
+
86
+ if (response?.error) {
87
+ const error = response?.error + ' ' + (response?.details || '');
88
+ logger.error(` LLM Error=`, error);
89
+
90
+ return { Reply: response?.data, _error: error, _debug: logger.output };
91
+ }
92
+
93
+ logger.debug(' Response \n', response);
94
+
95
+ const result = { Reply: response };
96
+
97
+ result['_debug'] = logger.output;
98
+
99
+ return result;
100
+ } catch (error: any) {
101
+ return { _error: error.message, _debug: logger.output };
102
+ }
103
+ }
104
+ }