@smythos/sre 1.5.43 → 1.5.45

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 (233) hide show
  1. package/CHANGELOG +90 -90
  2. package/LICENSE +18 -18
  3. package/README.md +135 -135
  4. package/dist/index.js +13 -13
  5. package/dist/index.js.map +1 -1
  6. package/dist/types/Components/GenAILLM.class.d.ts +6 -0
  7. package/dist/types/helpers/AWSLambdaCode.helper.d.ts +8 -5
  8. package/dist/types/index.d.ts +1 -0
  9. package/dist/types/subsystems/LLMManager/LLM.service/connectors/Groq.class.d.ts +5 -0
  10. package/dist/types/subsystems/LLMManager/LLM.service/connectors/openai/OpenAIConnector.class.d.ts +13 -1
  11. package/dist/types/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/ChatCompletionsApiInterface.d.ts +0 -4
  12. package/dist/types/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/ResponsesApiInterface.d.ts +44 -29
  13. package/dist/types/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/constants.d.ts +4 -2
  14. package/dist/types/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/utils.d.ts +6 -0
  15. package/dist/types/subsystems/LLMManager/LLM.service/connectors/openai/types.d.ts +0 -4
  16. package/dist/types/subsystems/LLMManager/ModelsProvider.service/connectors/SmythModelsProvider.class.d.ts +39 -0
  17. package/dist/types/types/LLM.types.d.ts +4 -1
  18. package/package.json +5 -2
  19. package/src/Components/APICall/APICall.class.ts +156 -156
  20. package/src/Components/APICall/AccessTokenManager.ts +130 -130
  21. package/src/Components/APICall/ArrayBufferResponse.helper.ts +58 -58
  22. package/src/Components/APICall/OAuth.helper.ts +294 -294
  23. package/src/Components/APICall/mimeTypeCategories.ts +46 -46
  24. package/src/Components/APICall/parseData.ts +167 -167
  25. package/src/Components/APICall/parseHeaders.ts +41 -41
  26. package/src/Components/APICall/parseProxy.ts +68 -68
  27. package/src/Components/APICall/parseUrl.ts +91 -91
  28. package/src/Components/APIEndpoint.class.ts +234 -234
  29. package/src/Components/APIOutput.class.ts +58 -58
  30. package/src/Components/AgentPlugin.class.ts +102 -102
  31. package/src/Components/Async.class.ts +155 -155
  32. package/src/Components/Await.class.ts +90 -90
  33. package/src/Components/Classifier.class.ts +158 -158
  34. package/src/Components/Component.class.ts +132 -132
  35. package/src/Components/ComponentHost.class.ts +38 -38
  36. package/src/Components/DataSourceCleaner.class.ts +92 -92
  37. package/src/Components/DataSourceIndexer.class.ts +181 -181
  38. package/src/Components/DataSourceLookup.class.ts +161 -161
  39. package/src/Components/ECMASandbox.class.ts +71 -71
  40. package/src/Components/FEncDec.class.ts +29 -29
  41. package/src/Components/FHash.class.ts +33 -33
  42. package/src/Components/FSign.class.ts +80 -80
  43. package/src/Components/FSleep.class.ts +25 -25
  44. package/src/Components/FTimestamp.class.ts +25 -25
  45. package/src/Components/FileStore.class.ts +78 -78
  46. package/src/Components/ForEach.class.ts +97 -97
  47. package/src/Components/GPTPlugin.class.ts +70 -70
  48. package/src/Components/GenAILLM.class.ts +586 -579
  49. package/src/Components/HuggingFace.class.ts +314 -314
  50. package/src/Components/Image/imageSettings.config.ts +70 -70
  51. package/src/Components/ImageGenerator.class.ts +502 -502
  52. package/src/Components/JSONFilter.class.ts +54 -54
  53. package/src/Components/LLMAssistant.class.ts +213 -213
  54. package/src/Components/LogicAND.class.ts +28 -28
  55. package/src/Components/LogicAtLeast.class.ts +85 -85
  56. package/src/Components/LogicAtMost.class.ts +86 -86
  57. package/src/Components/LogicOR.class.ts +29 -29
  58. package/src/Components/LogicXOR.class.ts +34 -34
  59. package/src/Components/MCPClient.class.ts +112 -112
  60. package/src/Components/MemoryDeleteKeyVal.class.ts +70 -70
  61. package/src/Components/MemoryReadKeyVal.class.ts +66 -66
  62. package/src/Components/MemoryWriteKeyVal.class.ts +62 -62
  63. package/src/Components/MemoryWriteObject.class.ts +97 -97
  64. package/src/Components/MultimodalLLM.class.ts +128 -128
  65. package/src/Components/OpenAPI.class.ts +72 -72
  66. package/src/Components/PromptGenerator.class.ts +122 -122
  67. package/src/Components/ScrapflyWebScrape.class.ts +159 -159
  68. package/src/Components/ServerlessCode.class.ts +123 -123
  69. package/src/Components/TavilyWebSearch.class.ts +98 -98
  70. package/src/Components/VisionLLM.class.ts +104 -104
  71. package/src/Components/ZapierAction.class.ts +127 -127
  72. package/src/Components/index.ts +97 -97
  73. package/src/Core/AgentProcess.helper.ts +240 -240
  74. package/src/Core/Connector.class.ts +123 -123
  75. package/src/Core/ConnectorsService.ts +197 -197
  76. package/src/Core/DummyConnector.ts +49 -49
  77. package/src/Core/HookService.ts +105 -105
  78. package/src/Core/SmythRuntime.class.ts +235 -235
  79. package/src/Core/SystemEvents.ts +16 -16
  80. package/src/Core/boot.ts +56 -56
  81. package/src/config.ts +15 -15
  82. package/src/constants.ts +126 -126
  83. package/src/data/hugging-face.params.json +579 -579
  84. package/src/helpers/AWSLambdaCode.helper.ts +588 -528
  85. package/src/helpers/BinaryInput.helper.ts +331 -331
  86. package/src/helpers/Conversation.helper.ts +1119 -1119
  87. package/src/helpers/ECMASandbox.helper.ts +54 -54
  88. package/src/helpers/JsonContent.helper.ts +97 -97
  89. package/src/helpers/LocalCache.helper.ts +97 -97
  90. package/src/helpers/Log.helper.ts +274 -274
  91. package/src/helpers/OpenApiParser.helper.ts +150 -150
  92. package/src/helpers/S3Cache.helper.ts +147 -147
  93. package/src/helpers/SmythURI.helper.ts +5 -5
  94. package/src/helpers/Sysconfig.helper.ts +77 -77
  95. package/src/helpers/TemplateString.helper.ts +243 -243
  96. package/src/helpers/TypeChecker.helper.ts +329 -329
  97. package/src/index.ts +4 -3
  98. package/src/index.ts.bak +4 -3
  99. package/src/subsystems/AgentManager/Agent.class.ts +1114 -1114
  100. package/src/subsystems/AgentManager/Agent.helper.ts +3 -3
  101. package/src/subsystems/AgentManager/AgentData.service/AgentDataConnector.ts +230 -230
  102. package/src/subsystems/AgentManager/AgentData.service/connectors/CLIAgentDataConnector.class.ts +66 -66
  103. package/src/subsystems/AgentManager/AgentData.service/connectors/LocalAgentDataConnector.class.ts +142 -142
  104. package/src/subsystems/AgentManager/AgentData.service/connectors/NullAgentData.class.ts +39 -39
  105. package/src/subsystems/AgentManager/AgentData.service/index.ts +18 -18
  106. package/src/subsystems/AgentManager/AgentLogger.class.ts +297 -297
  107. package/src/subsystems/AgentManager/AgentRequest.class.ts +51 -51
  108. package/src/subsystems/AgentManager/AgentRuntime.class.ts +559 -559
  109. package/src/subsystems/AgentManager/AgentSSE.class.ts +101 -101
  110. package/src/subsystems/AgentManager/AgentSettings.class.ts +52 -52
  111. package/src/subsystems/AgentManager/Component.service/ComponentConnector.ts +32 -32
  112. package/src/subsystems/AgentManager/Component.service/connectors/LocalComponentConnector.class.ts +60 -60
  113. package/src/subsystems/AgentManager/Component.service/index.ts +11 -11
  114. package/src/subsystems/AgentManager/EmbodimentSettings.class.ts +47 -47
  115. package/src/subsystems/AgentManager/ForkedAgent.class.ts +154 -154
  116. package/src/subsystems/AgentManager/OSResourceMonitor.ts +77 -77
  117. package/src/subsystems/ComputeManager/Code.service/CodeConnector.ts +98 -98
  118. package/src/subsystems/ComputeManager/Code.service/connectors/AWSLambdaCode.class.ts +172 -170
  119. package/src/subsystems/ComputeManager/Code.service/connectors/ECMASandbox.class.ts +131 -131
  120. package/src/subsystems/ComputeManager/Code.service/index.ts +13 -13
  121. package/src/subsystems/IO/CLI.service/CLIConnector.ts +47 -47
  122. package/src/subsystems/IO/CLI.service/index.ts +9 -9
  123. package/src/subsystems/IO/Log.service/LogConnector.ts +32 -32
  124. package/src/subsystems/IO/Log.service/connectors/ConsoleLog.class.ts +28 -28
  125. package/src/subsystems/IO/Log.service/index.ts +13 -13
  126. package/src/subsystems/IO/NKV.service/NKVConnector.ts +43 -43
  127. package/src/subsystems/IO/NKV.service/connectors/NKVLocalStorage.class.ts +234 -234
  128. package/src/subsystems/IO/NKV.service/connectors/NKVRAM.class.ts +204 -204
  129. package/src/subsystems/IO/NKV.service/connectors/NKVRedis.class.ts +182 -182
  130. package/src/subsystems/IO/NKV.service/index.ts +14 -14
  131. package/src/subsystems/IO/Router.service/RouterConnector.ts +21 -21
  132. package/src/subsystems/IO/Router.service/connectors/ExpressRouter.class.ts +48 -48
  133. package/src/subsystems/IO/Router.service/connectors/NullRouter.class.ts +40 -40
  134. package/src/subsystems/IO/Router.service/index.ts +11 -11
  135. package/src/subsystems/IO/Storage.service/SmythFS.class.ts +489 -489
  136. package/src/subsystems/IO/Storage.service/StorageConnector.ts +66 -66
  137. package/src/subsystems/IO/Storage.service/connectors/LocalStorage.class.ts +327 -327
  138. package/src/subsystems/IO/Storage.service/connectors/S3Storage.class.ts +482 -482
  139. package/src/subsystems/IO/Storage.service/index.ts +13 -13
  140. package/src/subsystems/IO/VectorDB.service/VectorDBConnector.ts +108 -108
  141. package/src/subsystems/IO/VectorDB.service/connectors/MilvusVectorDB.class.ts +454 -454
  142. package/src/subsystems/IO/VectorDB.service/connectors/PineconeVectorDB.class.ts +384 -384
  143. package/src/subsystems/IO/VectorDB.service/connectors/RAMVecrtorDB.class.ts +421 -421
  144. package/src/subsystems/IO/VectorDB.service/embed/BaseEmbedding.ts +107 -107
  145. package/src/subsystems/IO/VectorDB.service/embed/OpenAIEmbedding.ts +109 -109
  146. package/src/subsystems/IO/VectorDB.service/embed/index.ts +21 -21
  147. package/src/subsystems/IO/VectorDB.service/index.ts +14 -14
  148. package/src/subsystems/LLMManager/LLM.helper.ts +251 -251
  149. package/src/subsystems/LLMManager/LLM.inference.ts +339 -339
  150. package/src/subsystems/LLMManager/LLM.service/LLMConnector.ts +489 -489
  151. package/src/subsystems/LLMManager/LLM.service/LLMCredentials.helper.ts +171 -171
  152. package/src/subsystems/LLMManager/LLM.service/connectors/Anthropic.class.ts +659 -659
  153. package/src/subsystems/LLMManager/LLM.service/connectors/Bedrock.class.ts +400 -400
  154. package/src/subsystems/LLMManager/LLM.service/connectors/Echo.class.ts +77 -77
  155. package/src/subsystems/LLMManager/LLM.service/connectors/GoogleAI.class.ts +757 -757
  156. package/src/subsystems/LLMManager/LLM.service/connectors/Groq.class.ts +304 -291
  157. package/src/subsystems/LLMManager/LLM.service/connectors/Perplexity.class.ts +250 -250
  158. package/src/subsystems/LLMManager/LLM.service/connectors/VertexAI.class.ts +423 -423
  159. package/src/subsystems/LLMManager/LLM.service/connectors/openai/OpenAIConnector.class.ts +488 -455
  160. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/ChatCompletionsApiInterface.ts +528 -528
  161. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/OpenAIApiInterface.ts +100 -100
  162. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/OpenAIApiInterfaceFactory.ts +81 -81
  163. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/ResponsesApiInterface.ts +1168 -853
  164. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/constants.ts +13 -37
  165. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/index.ts +4 -4
  166. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/utils.ts +11 -0
  167. package/src/subsystems/LLMManager/LLM.service/connectors/openai/types.ts +32 -37
  168. package/src/subsystems/LLMManager/LLM.service/connectors/xAI.class.ts +471 -471
  169. package/src/subsystems/LLMManager/LLM.service/index.ts +44 -44
  170. package/src/subsystems/LLMManager/ModelsProvider.service/ModelsProviderConnector.ts +300 -300
  171. package/src/subsystems/LLMManager/ModelsProvider.service/connectors/JSONModelsProvider.class.ts +252 -252
  172. package/src/subsystems/LLMManager/ModelsProvider.service/index.ts +11 -11
  173. package/src/subsystems/LLMManager/custom-models.ts +854 -854
  174. package/src/subsystems/LLMManager/models.ts +2540 -2540
  175. package/src/subsystems/LLMManager/paramMappings.ts +69 -69
  176. package/src/subsystems/MemoryManager/Cache.service/CacheConnector.ts +86 -86
  177. package/src/subsystems/MemoryManager/Cache.service/connectors/LocalStorageCache.class.ts +297 -297
  178. package/src/subsystems/MemoryManager/Cache.service/connectors/RAMCache.class.ts +201 -201
  179. package/src/subsystems/MemoryManager/Cache.service/connectors/RedisCache.class.ts +252 -252
  180. package/src/subsystems/MemoryManager/Cache.service/connectors/S3Cache.class.ts +373 -373
  181. package/src/subsystems/MemoryManager/Cache.service/index.ts +15 -15
  182. package/src/subsystems/MemoryManager/LLMCache.ts +72 -72
  183. package/src/subsystems/MemoryManager/LLMContext.ts +124 -124
  184. package/src/subsystems/MemoryManager/LLMMemory.service/LLMMemoryConnector.ts +26 -26
  185. package/src/subsystems/MemoryManager/RuntimeContext.ts +266 -266
  186. package/src/subsystems/Security/AccessControl/ACL.class.ts +208 -208
  187. package/src/subsystems/Security/AccessControl/AccessCandidate.class.ts +82 -82
  188. package/src/subsystems/Security/AccessControl/AccessRequest.class.ts +52 -52
  189. package/src/subsystems/Security/Account.service/AccountConnector.ts +44 -44
  190. package/src/subsystems/Security/Account.service/connectors/AWSAccount.class.ts +76 -76
  191. package/src/subsystems/Security/Account.service/connectors/DummyAccount.class.ts +130 -130
  192. package/src/subsystems/Security/Account.service/connectors/JSONFileAccount.class.ts +159 -159
  193. package/src/subsystems/Security/Account.service/index.ts +14 -14
  194. package/src/subsystems/Security/Credentials.helper.ts +62 -62
  195. package/src/subsystems/Security/ManagedVault.service/ManagedVaultConnector.ts +38 -38
  196. package/src/subsystems/Security/ManagedVault.service/connectors/NullManagedVault.class.ts +53 -53
  197. package/src/subsystems/Security/ManagedVault.service/connectors/SecretManagerManagedVault.ts +154 -154
  198. package/src/subsystems/Security/ManagedVault.service/index.ts +12 -12
  199. package/src/subsystems/Security/SecureConnector.class.ts +110 -110
  200. package/src/subsystems/Security/Vault.service/Vault.helper.ts +30 -30
  201. package/src/subsystems/Security/Vault.service/VaultConnector.ts +29 -29
  202. package/src/subsystems/Security/Vault.service/connectors/HashicorpVault.class.ts +46 -46
  203. package/src/subsystems/Security/Vault.service/connectors/JSONFileVault.class.ts +221 -221
  204. package/src/subsystems/Security/Vault.service/connectors/NullVault.class.ts +54 -54
  205. package/src/subsystems/Security/Vault.service/connectors/SecretsManager.class.ts +140 -140
  206. package/src/subsystems/Security/Vault.service/index.ts +12 -12
  207. package/src/types/ACL.types.ts +104 -104
  208. package/src/types/AWS.types.ts +10 -10
  209. package/src/types/Agent.types.ts +61 -61
  210. package/src/types/AgentLogger.types.ts +17 -17
  211. package/src/types/Cache.types.ts +1 -1
  212. package/src/types/Common.types.ts +2 -2
  213. package/src/types/LLM.types.ts +496 -491
  214. package/src/types/Redis.types.ts +8 -8
  215. package/src/types/SRE.types.ts +64 -64
  216. package/src/types/Security.types.ts +14 -14
  217. package/src/types/Storage.types.ts +5 -5
  218. package/src/types/VectorDB.types.ts +86 -86
  219. package/src/utils/base64.utils.ts +275 -275
  220. package/src/utils/cli.utils.ts +68 -68
  221. package/src/utils/data.utils.ts +322 -322
  222. package/src/utils/date-time.utils.ts +22 -22
  223. package/src/utils/general.utils.ts +238 -238
  224. package/src/utils/index.ts +12 -12
  225. package/src/utils/lazy-client.ts +261 -261
  226. package/src/utils/numbers.utils.ts +13 -13
  227. package/src/utils/oauth.utils.ts +35 -35
  228. package/src/utils/string.utils.ts +414 -414
  229. package/src/utils/url.utils.ts +19 -19
  230. package/src/utils/validation.utils.ts +74 -74
  231. package/dist/bundle-analysis-lazy.html +0 -4949
  232. package/dist/bundle-analysis.html +0 -4949
  233. package/dist/types/utils/package-manager.utils.d.ts +0 -26
@@ -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
+ }