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