@smythos/sre 1.6.8 → 1.6.10

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 (239) hide show
  1. package/CHANGELOG +111 -111
  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 +2 -2
  7. package/dist/index.js.map +1 -1
  8. package/dist/types/Components/Triggers/Gmail.trigger.d.ts +58 -0
  9. package/dist/types/Components/Triggers/GmailTrigger.class.d.ts +44 -0
  10. package/dist/types/Components/Triggers/Trigger.class.d.ts +21 -0
  11. package/dist/types/Components/Triggers/WhatsApp.trigger.d.ts +22 -0
  12. package/dist/types/helpers/AIPerformanceAnalyzer.helper.d.ts +45 -0
  13. package/dist/types/helpers/AIPerformanceCollector.helper.d.ts +111 -0
  14. package/dist/types/subsystems/IO/Storage.service/connectors/AzureBlobStorage.class.d.ts +211 -0
  15. package/dist/types/subsystems/IO/VectorDB.service/connectors/WeaviateVectorDB.class.d.ts +187 -0
  16. package/dist/types/subsystems/PerformanceManager/Performance.service/PerformanceConnector.d.ts +102 -0
  17. package/dist/types/subsystems/PerformanceManager/Performance.service/connectors/LocalPerformanceConnector.class.d.ts +100 -0
  18. package/dist/types/subsystems/PerformanceManager/Performance.service/index.d.ts +22 -0
  19. package/dist/types/subsystems/Security/Credentials/Credentials.class.d.ts +2 -0
  20. package/dist/types/subsystems/Security/Credentials/ManagedOAuth2Credentials.class.d.ts +18 -0
  21. package/dist/types/subsystems/Security/Credentials/OAuth2Credentials.class.d.ts +14 -0
  22. package/dist/types/types/Performance.types.d.ts +468 -0
  23. package/dist/types/utils/package-manager.utils.d.ts +26 -0
  24. package/package.json +1 -1
  25. package/src/Components/APICall/APICall.class.ts +161 -161
  26. package/src/Components/APICall/AccessTokenManager.ts +166 -166
  27. package/src/Components/APICall/ArrayBufferResponse.helper.ts +58 -58
  28. package/src/Components/APICall/OAuth.helper.ts +447 -447
  29. package/src/Components/APICall/mimeTypeCategories.ts +46 -46
  30. package/src/Components/APICall/parseData.ts +167 -167
  31. package/src/Components/APICall/parseHeaders.ts +41 -41
  32. package/src/Components/APICall/parseProxy.ts +68 -68
  33. package/src/Components/APICall/parseUrl.ts +91 -91
  34. package/src/Components/APIEndpoint.class.ts +234 -234
  35. package/src/Components/APIOutput.class.ts +58 -58
  36. package/src/Components/AgentPlugin.class.ts +102 -102
  37. package/src/Components/Async.class.ts +155 -155
  38. package/src/Components/Await.class.ts +90 -90
  39. package/src/Components/Classifier.class.ts +158 -158
  40. package/src/Components/Component.class.ts +147 -147
  41. package/src/Components/ComponentHost.class.ts +38 -38
  42. package/src/Components/DataSourceCleaner.class.ts +92 -92
  43. package/src/Components/DataSourceIndexer.class.ts +181 -181
  44. package/src/Components/DataSourceLookup.class.ts +161 -161
  45. package/src/Components/ECMASandbox.class.ts +72 -72
  46. package/src/Components/FEncDec.class.ts +29 -29
  47. package/src/Components/FHash.class.ts +33 -33
  48. package/src/Components/FSign.class.ts +80 -80
  49. package/src/Components/FSleep.class.ts +25 -25
  50. package/src/Components/FTimestamp.class.ts +66 -66
  51. package/src/Components/FileStore.class.ts +78 -78
  52. package/src/Components/ForEach.class.ts +97 -97
  53. package/src/Components/GPTPlugin.class.ts +70 -70
  54. package/src/Components/GenAILLM.class.ts +586 -586
  55. package/src/Components/HuggingFace.class.ts +313 -313
  56. package/src/Components/Image/imageSettings.config.ts +70 -70
  57. package/src/Components/ImageGenerator.class.ts +483 -483
  58. package/src/Components/JSONFilter.class.ts +54 -54
  59. package/src/Components/LLMAssistant.class.ts +213 -213
  60. package/src/Components/LogicAND.class.ts +28 -28
  61. package/src/Components/LogicAtLeast.class.ts +85 -85
  62. package/src/Components/LogicAtMost.class.ts +86 -86
  63. package/src/Components/LogicOR.class.ts +29 -29
  64. package/src/Components/LogicXOR.class.ts +34 -34
  65. package/src/Components/MCPClient.class.ts +137 -137
  66. package/src/Components/MemoryDeleteKeyVal.class.ts +70 -70
  67. package/src/Components/MemoryReadKeyVal.class.ts +67 -67
  68. package/src/Components/MemoryWriteKeyVal.class.ts +62 -62
  69. package/src/Components/MemoryWriteObject.class.ts +97 -97
  70. package/src/Components/MultimodalLLM.class.ts +128 -128
  71. package/src/Components/OpenAPI.class.ts +72 -72
  72. package/src/Components/PromptGenerator.class.ts +122 -122
  73. package/src/Components/ScrapflyWebScrape.class.ts +183 -183
  74. package/src/Components/ServerlessCode.class.ts +123 -123
  75. package/src/Components/TavilyWebSearch.class.ts +103 -103
  76. package/src/Components/VisionLLM.class.ts +104 -104
  77. package/src/Components/ZapierAction.class.ts +127 -127
  78. package/src/Components/index.ts +97 -97
  79. package/src/Core/AgentProcess.helper.ts +240 -240
  80. package/src/Core/Connector.class.ts +123 -123
  81. package/src/Core/ConnectorsService.ts +197 -197
  82. package/src/Core/DummyConnector.ts +49 -49
  83. package/src/Core/HookService.ts +105 -105
  84. package/src/Core/SmythRuntime.class.ts +241 -241
  85. package/src/Core/SystemEvents.ts +16 -16
  86. package/src/Core/boot.ts +56 -56
  87. package/src/config.ts +15 -15
  88. package/src/constants.ts +126 -126
  89. package/src/data/hugging-face.params.json +579 -579
  90. package/src/helpers/AWSLambdaCode.helper.ts +624 -624
  91. package/src/helpers/BinaryInput.helper.ts +331 -331
  92. package/src/helpers/Conversation.helper.ts +1157 -1157
  93. package/src/helpers/ECMASandbox.helper.ts +64 -64
  94. package/src/helpers/JsonContent.helper.ts +97 -97
  95. package/src/helpers/LocalCache.helper.ts +97 -97
  96. package/src/helpers/Log.helper.ts +274 -274
  97. package/src/helpers/OpenApiParser.helper.ts +150 -150
  98. package/src/helpers/S3Cache.helper.ts +147 -147
  99. package/src/helpers/SmythURI.helper.ts +5 -5
  100. package/src/helpers/Sysconfig.helper.ts +95 -95
  101. package/src/helpers/TemplateString.helper.ts +243 -243
  102. package/src/helpers/TypeChecker.helper.ts +329 -329
  103. package/src/index.ts +3 -3
  104. package/src/index.ts.bak +3 -3
  105. package/src/subsystems/AgentManager/Agent.class.ts +1114 -1114
  106. package/src/subsystems/AgentManager/Agent.helper.ts +3 -3
  107. package/src/subsystems/AgentManager/AgentData.service/AgentDataConnector.ts +230 -230
  108. package/src/subsystems/AgentManager/AgentData.service/connectors/CLIAgentDataConnector.class.ts +66 -66
  109. package/src/subsystems/AgentManager/AgentData.service/connectors/LocalAgentDataConnector.class.ts +145 -145
  110. package/src/subsystems/AgentManager/AgentData.service/connectors/NullAgentData.class.ts +39 -39
  111. package/src/subsystems/AgentManager/AgentData.service/index.ts +18 -18
  112. package/src/subsystems/AgentManager/AgentLogger.class.ts +301 -301
  113. package/src/subsystems/AgentManager/AgentRequest.class.ts +51 -51
  114. package/src/subsystems/AgentManager/AgentRuntime.class.ts +557 -557
  115. package/src/subsystems/AgentManager/AgentSSE.class.ts +101 -101
  116. package/src/subsystems/AgentManager/AgentSettings.class.ts +52 -52
  117. package/src/subsystems/AgentManager/Component.service/ComponentConnector.ts +32 -32
  118. package/src/subsystems/AgentManager/Component.service/connectors/LocalComponentConnector.class.ts +60 -60
  119. package/src/subsystems/AgentManager/Component.service/index.ts +11 -11
  120. package/src/subsystems/AgentManager/EmbodimentSettings.class.ts +47 -47
  121. package/src/subsystems/AgentManager/ForkedAgent.class.ts +154 -154
  122. package/src/subsystems/AgentManager/OSResourceMonitor.ts +77 -77
  123. package/src/subsystems/ComputeManager/Code.service/CodeConnector.ts +98 -98
  124. package/src/subsystems/ComputeManager/Code.service/connectors/AWSLambdaCode.class.ts +171 -171
  125. package/src/subsystems/ComputeManager/Code.service/connectors/ECMASandbox.class.ts +131 -131
  126. package/src/subsystems/ComputeManager/Code.service/index.ts +13 -13
  127. package/src/subsystems/IO/CLI.service/CLIConnector.ts +47 -47
  128. package/src/subsystems/IO/CLI.service/index.ts +9 -9
  129. package/src/subsystems/IO/Log.service/LogConnector.ts +32 -32
  130. package/src/subsystems/IO/Log.service/connectors/ConsoleLog.class.ts +28 -28
  131. package/src/subsystems/IO/Log.service/index.ts +13 -13
  132. package/src/subsystems/IO/NKV.service/NKVConnector.ts +43 -43
  133. package/src/subsystems/IO/NKV.service/connectors/NKVLocalStorage.class.ts +234 -234
  134. package/src/subsystems/IO/NKV.service/connectors/NKVRAM.class.ts +204 -204
  135. package/src/subsystems/IO/NKV.service/connectors/NKVRedis.class.ts +182 -182
  136. package/src/subsystems/IO/NKV.service/index.ts +14 -14
  137. package/src/subsystems/IO/Router.service/RouterConnector.ts +21 -21
  138. package/src/subsystems/IO/Router.service/connectors/ExpressRouter.class.ts +48 -48
  139. package/src/subsystems/IO/Router.service/connectors/NullRouter.class.ts +40 -40
  140. package/src/subsystems/IO/Router.service/index.ts +11 -11
  141. package/src/subsystems/IO/Storage.service/SmythFS.class.ts +488 -488
  142. package/src/subsystems/IO/Storage.service/StorageConnector.ts +66 -66
  143. package/src/subsystems/IO/Storage.service/connectors/LocalStorage.class.ts +327 -327
  144. package/src/subsystems/IO/Storage.service/connectors/S3Storage.class.ts +482 -482
  145. package/src/subsystems/IO/Storage.service/index.ts +13 -13
  146. package/src/subsystems/IO/VectorDB.service/VectorDBConnector.ts +108 -108
  147. package/src/subsystems/IO/VectorDB.service/connectors/MilvusVectorDB.class.ts +465 -465
  148. package/src/subsystems/IO/VectorDB.service/connectors/PineconeVectorDB.class.ts +387 -387
  149. package/src/subsystems/IO/VectorDB.service/connectors/RAMVecrtorDB.class.ts +408 -408
  150. package/src/subsystems/IO/VectorDB.service/embed/BaseEmbedding.ts +107 -107
  151. package/src/subsystems/IO/VectorDB.service/embed/GoogleEmbedding.ts +118 -118
  152. package/src/subsystems/IO/VectorDB.service/embed/OpenAIEmbedding.ts +109 -109
  153. package/src/subsystems/IO/VectorDB.service/embed/index.ts +26 -26
  154. package/src/subsystems/IO/VectorDB.service/index.ts +14 -14
  155. package/src/subsystems/LLMManager/LLM.helper.ts +251 -251
  156. package/src/subsystems/LLMManager/LLM.inference.ts +345 -345
  157. package/src/subsystems/LLMManager/LLM.service/LLMConnector.ts +492 -492
  158. package/src/subsystems/LLMManager/LLM.service/LLMCredentials.helper.ts +171 -171
  159. package/src/subsystems/LLMManager/LLM.service/connectors/Anthropic.class.ts +666 -666
  160. package/src/subsystems/LLMManager/LLM.service/connectors/Bedrock.class.ts +407 -407
  161. package/src/subsystems/LLMManager/LLM.service/connectors/Echo.class.ts +92 -92
  162. package/src/subsystems/LLMManager/LLM.service/connectors/GoogleAI.class.ts +983 -983
  163. package/src/subsystems/LLMManager/LLM.service/connectors/Groq.class.ts +319 -319
  164. package/src/subsystems/LLMManager/LLM.service/connectors/Ollama.class.ts +361 -361
  165. package/src/subsystems/LLMManager/LLM.service/connectors/Perplexity.class.ts +257 -257
  166. package/src/subsystems/LLMManager/LLM.service/connectors/VertexAI.class.ts +430 -430
  167. package/src/subsystems/LLMManager/LLM.service/connectors/openai/OpenAIConnector.class.ts +503 -503
  168. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/ChatCompletionsApiInterface.ts +524 -524
  169. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/OpenAIApiInterface.ts +100 -100
  170. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/OpenAIApiInterfaceFactory.ts +81 -81
  171. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/ResponsesApiInterface.ts +1145 -1145
  172. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/constants.ts +13 -13
  173. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/index.ts +4 -4
  174. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/utils.ts +11 -11
  175. package/src/subsystems/LLMManager/LLM.service/connectors/openai/types.ts +32 -32
  176. package/src/subsystems/LLMManager/LLM.service/connectors/xAI.class.ts +478 -478
  177. package/src/subsystems/LLMManager/LLM.service/index.ts +47 -47
  178. package/src/subsystems/LLMManager/ModelsProvider.service/ModelsProviderConnector.ts +303 -303
  179. package/src/subsystems/LLMManager/ModelsProvider.service/connectors/JSONModelsProvider.class.ts +280 -271
  180. package/src/subsystems/LLMManager/ModelsProvider.service/index.ts +11 -11
  181. package/src/subsystems/LLMManager/custom-models.ts +854 -854
  182. package/src/subsystems/LLMManager/models.ts +2540 -2540
  183. package/src/subsystems/LLMManager/paramMappings.ts +69 -69
  184. package/src/subsystems/MemoryManager/Cache.service/CacheConnector.ts +86 -86
  185. package/src/subsystems/MemoryManager/Cache.service/connectors/LocalStorageCache.class.ts +297 -297
  186. package/src/subsystems/MemoryManager/Cache.service/connectors/RAMCache.class.ts +214 -214
  187. package/src/subsystems/MemoryManager/Cache.service/connectors/RedisCache.class.ts +252 -252
  188. package/src/subsystems/MemoryManager/Cache.service/connectors/S3Cache.class.ts +373 -373
  189. package/src/subsystems/MemoryManager/Cache.service/index.ts +15 -15
  190. package/src/subsystems/MemoryManager/LLMCache.ts +72 -72
  191. package/src/subsystems/MemoryManager/LLMContext.ts +124 -124
  192. package/src/subsystems/MemoryManager/LLMMemory.service/LLMMemoryConnector.ts +26 -26
  193. package/src/subsystems/MemoryManager/RuntimeContext.ts +277 -277
  194. package/src/subsystems/Security/AccessControl/ACL.class.ts +208 -208
  195. package/src/subsystems/Security/AccessControl/AccessCandidate.class.ts +82 -82
  196. package/src/subsystems/Security/AccessControl/AccessRequest.class.ts +52 -52
  197. package/src/subsystems/Security/Account.service/AccountConnector.ts +44 -44
  198. package/src/subsystems/Security/Account.service/connectors/DummyAccount.class.ts +130 -130
  199. package/src/subsystems/Security/Account.service/connectors/JSONFileAccount.class.ts +170 -170
  200. package/src/subsystems/Security/Account.service/connectors/MySQLAccount.class.ts +76 -76
  201. package/src/subsystems/Security/Account.service/index.ts +14 -14
  202. package/src/subsystems/Security/Credentials.helper.ts +62 -62
  203. package/src/subsystems/Security/ManagedVault.service/ManagedVaultConnector.ts +38 -38
  204. package/src/subsystems/Security/ManagedVault.service/connectors/NullManagedVault.class.ts +53 -53
  205. package/src/subsystems/Security/ManagedVault.service/connectors/SecretManagerManagedVault.ts +154 -154
  206. package/src/subsystems/Security/ManagedVault.service/index.ts +12 -12
  207. package/src/subsystems/Security/SecureConnector.class.ts +110 -110
  208. package/src/subsystems/Security/Vault.service/Vault.helper.ts +30 -30
  209. package/src/subsystems/Security/Vault.service/VaultConnector.ts +29 -29
  210. package/src/subsystems/Security/Vault.service/connectors/HashicorpVault.class.ts +46 -46
  211. package/src/subsystems/Security/Vault.service/connectors/JSONFileVault.class.ts +221 -221
  212. package/src/subsystems/Security/Vault.service/connectors/NullVault.class.ts +54 -54
  213. package/src/subsystems/Security/Vault.service/connectors/SecretsManager.class.ts +140 -140
  214. package/src/subsystems/Security/Vault.service/index.ts +12 -12
  215. package/src/types/ACL.types.ts +104 -104
  216. package/src/types/AWS.types.ts +10 -10
  217. package/src/types/Agent.types.ts +61 -61
  218. package/src/types/AgentLogger.types.ts +17 -17
  219. package/src/types/Cache.types.ts +1 -1
  220. package/src/types/Common.types.ts +2 -2
  221. package/src/types/LLM.types.ts +520 -520
  222. package/src/types/Redis.types.ts +8 -8
  223. package/src/types/SRE.types.ts +64 -64
  224. package/src/types/Security.types.ts +14 -14
  225. package/src/types/Storage.types.ts +5 -5
  226. package/src/types/VectorDB.types.ts +86 -86
  227. package/src/utils/base64.utils.ts +275 -275
  228. package/src/utils/cli.utils.ts +68 -68
  229. package/src/utils/data.utils.ts +322 -322
  230. package/src/utils/date-time.utils.ts +22 -22
  231. package/src/utils/general.utils.ts +238 -238
  232. package/src/utils/index.ts +12 -12
  233. package/src/utils/lazy-client.ts +261 -261
  234. package/src/utils/numbers.utils.ts +13 -13
  235. package/src/utils/oauth.utils.ts +35 -35
  236. package/src/utils/string.utils.ts +414 -414
  237. package/src/utils/url.utils.ts +19 -19
  238. package/src/utils/validation.utils.ts +74 -74
  239. package/dist/types/subsystems/LLMManager/ModelsProvider.service/connectors/SmythModelsProvider.class.d.ts +0 -39
@@ -1,854 +1,854 @@
1
- //! ***IMPORTANT***
2
- // * WE HAVE SAME FILE IN THREE PLACE, SO ANY CHANGES SHOULD BE SYNCED
3
- // * - /smyth-builder-ui/src/shared/custom-models.ts
4
- // * - /smyth-builder-server/src/services/LLMHelper/custom-models.ts
5
- // * - /smyth-runtime/src/subsystems/LLMManager/custom-models.ts
6
- //! ***IMPORTANT***
7
- /**
8
- * Custom model configurations for various AI providers.
9
- * This template is useful for maintaining up-to-date model aliases and token information,
10
- * even though we store custom model data in the database.
11
- */
12
-
13
- export const customModels = {
14
- /**************************************************
15
- * AWS Bedrock Models
16
- **************************************************/
17
-
18
- /*
19
- Context token information sourced from the AWS Bedrock Console:
20
- https://us-east-1.console.aws.amazon.com/bedrock/home?region=us-east-1#/overview
21
-
22
- Completion token information sourced from the AWS Bedrock Text Playground:
23
- https://us-east-1.console.aws.amazon.com/bedrock/home?region=us-east-1#/text-playground
24
-
25
- Supported model features information sourced from - https://docs.aws.amazon.com/bedrock/latest/userguide/conversation-inference-supported-models-features.html
26
- */
27
-
28
- //#region AI21 Labs Models
29
- 'ai21.jamba-1-5-mini-v1:0': {
30
- llm: 'Bedrock',
31
- label: 'AI21 Labs - Jamba 1.5 Mini',
32
- tokens: 256000,
33
- completionTokens: 4096,
34
- supportsSystemPrompt: true,
35
- supportsStreamingToolUse: false,
36
- components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'GenAILLM'],
37
- tags: ['new'],
38
- },
39
- 'ai21.jamba-1-5-large-v1:0': {
40
- llm: 'Bedrock',
41
- label: 'AI21 Labs - Jamba 1.5 Large',
42
- tokens: 256000,
43
- completionTokens: 4096,
44
- supportsSystemPrompt: true,
45
- supportsStreamingToolUse: false,
46
- components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'GenAILLM'],
47
- tags: ['new'],
48
- },
49
- 'ai21.jamba-instruct-v1:0': {
50
- llm: 'Bedrock',
51
- label: 'AI21 Labs - Jamba-Instruct',
52
- tokens: 256000,
53
- completionTokens: 4096,
54
- supportsSystemPrompt: true,
55
- supportsStreamingToolUse: false,
56
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
57
- },
58
- 'ai21.j2-ultra-v1': {
59
- llm: 'Bedrock',
60
- label: 'AI21 Labs - Jurassic-2 Ultra',
61
- tokens: 8191,
62
- completionTokens: 8191,
63
- supportsSystemPrompt: false,
64
- supportsStreamingToolUse: false,
65
- components: ['PromptGenerator', 'GenAILLM'],
66
- },
67
- 'ai21.j2-mid-v1': {
68
- llm: 'Bedrock',
69
- label: 'AI21 Labs - Jurassic-2 Mid',
70
- tokens: 8191,
71
- completionTokens: 8191,
72
- supportsSystemPrompt: false,
73
- supportsStreamingToolUse: false,
74
- components: ['PromptGenerator', 'GenAILLM'],
75
- },
76
- //#endregion AI21 Labs Models
77
-
78
- //#region Amazon Models
79
- 'amazon.titan-text-premier-v1:0': {
80
- llm: 'Bedrock',
81
- label: 'Amazon - Titan Text Premier',
82
- tokens: 32000,
83
- completionTokens: 3000,
84
- supportsSystemPrompt: false,
85
- supportsStreamingToolUse: false,
86
- components: ['PromptGenerator', 'GenAILLM'],
87
- },
88
- 'amazon.titan-text-express-v1': {
89
- llm: 'Bedrock',
90
- label: 'Amazon - Titan Text G1 - Express',
91
- tokens: 8192,
92
- completionTokens: 4096,
93
- supportsSystemPrompt: false,
94
- supportsStreamingToolUse: false,
95
- components: ['PromptGenerator', 'GenAILLM'],
96
- },
97
- 'amazon.titan-text-lite-v1': {
98
- llm: 'Bedrock',
99
- label: 'Amazon - Titan Text G1 - Lite',
100
- tokens: 4096,
101
- completionTokens: 4096,
102
- supportsSystemPrompt: false,
103
- supportsStreamingToolUse: false,
104
- components: ['PromptGenerator', 'GenAILLM'],
105
- },
106
- // 'Amazon - Titan Embeddings G1 - Text': { alias: 'amazon.titan-embed-text-v1' }, // Converse API doesn't support
107
- // 'Amazon - Titan Embedding Text v2': { alias: 'amazon.titan-embed-text-v2:0' }, // Converse API doesn't support
108
- // 'Amazon - Titan Multimodal Embeddings G1': { alias: 'amazon.titan-embed-image-v1' }, // Converse API doesn't support
109
- // 'Amazon - Titan Image Generator G1 V1': { alias: 'amazon.titan-image-generator-v1' }, // Converse API doesn't support
110
- // 'Amazon - Titan Image Generator G1 V2': { alias: 'amazon.titan-image-generator-v2:0' }, // Converse API doesn't support
111
-
112
- //#endregion Amazon Models
113
-
114
- //#region Anthropic Models
115
- // * NOTE: It's required to submit business info to get access for Anthropic models
116
- // * @Ref of Anthropic tokens and completionTokens - https://docs.anthropic.com/en/docs/about-claude/models
117
- 'us.anthropic.claude-3-5-haiku-20241022-v1:0': {
118
- llm: 'Bedrock',
119
- label: 'Anthropic - Claude 3.5 Haiku',
120
- tokens: 200000,
121
- completionTokens: 8192,
122
- supportsSystemPrompt: true,
123
- supportsStreamingToolUse: true,
124
- components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'Chatbot'],
125
- tags: ['v1:0', 'new'],
126
- },
127
- 'us.anthropic.claude-3-5-sonnet-20241022-v2:0': {
128
- llm: 'Bedrock',
129
- label: 'Anthropic - Claude 3.5 Sonnet',
130
- tokens: 200000,
131
- completionTokens: 8192,
132
- supportsSystemPrompt: true,
133
- supportsStreamingToolUse: true,
134
- components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'Chatbot'],
135
- tags: ['v2:0', 'new'],
136
- },
137
- 'anthropic.claude-3-5-sonnet-20240620-v1:0': {
138
- llm: 'Bedrock',
139
- label: 'Anthropic - Claude 3.5 Sonnet',
140
- tokens: 200000,
141
- completionTokens: 8192,
142
- supportsSystemPrompt: true,
143
- supportsStreamingToolUse: true,
144
- components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'Chatbot'],
145
- tags: ['v1:0'],
146
- },
147
- 'anthropic.claude-3-sonnet-20240229-v1:0': {
148
- llm: 'Bedrock',
149
- label: 'Anthropic - Claude 3 Sonnet',
150
- tokens: 200000,
151
- completionTokens: 4096,
152
- supportsSystemPrompt: true,
153
- supportsStreamingToolUse: true,
154
- components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'Chatbot'],
155
- tags: ['v1:0'],
156
- },
157
- 'anthropic.claude-3-haiku-20240307-v1:0': {
158
- llm: 'Bedrock',
159
- label: 'Anthropic - Claude 3 Haiku',
160
- tokens: 200000,
161
- completionTokens: 4096,
162
- supportsSystemPrompt: true,
163
- supportsStreamingToolUse: true,
164
- components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'Chatbot'],
165
- tags: ['v1:0'],
166
- },
167
- 'anthropic.claude-3-opus-20240229-v1:0': {
168
- llm: 'Bedrock',
169
- label: 'Anthropic - Claude 3 Opus',
170
- tokens: 200000,
171
- completionTokens: 4096,
172
- supportsSystemPrompt: true,
173
- supportsStreamingToolUse: true,
174
- components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'Chatbot'],
175
- tags: ['v1:0'],
176
- },
177
- 'anthropic.claude-v2:1': {
178
- llm: 'Bedrock',
179
- label: 'Anthropic - Claude 2.1',
180
- tokens: 200000,
181
- completionTokens: 4096,
182
- supportsSystemPrompt: true,
183
- supportsStreamingToolUse: false,
184
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
185
- tags: ['v2:1'],
186
- },
187
- 'anthropic.claude-v2': {
188
- llm: 'Bedrock',
189
- label: 'Anthropic - Claude 2.0',
190
- tokens: 100000,
191
- completionTokens: 4096,
192
- supportsSystemPrompt: true,
193
- supportsStreamingToolUse: false,
194
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
195
- },
196
- 'anthropic.claude-instant-v1': {
197
- llm: 'Bedrock',
198
- label: 'Anthropic - Claude Instant',
199
- tokens: 100000,
200
- completionTokens: 4096,
201
- supportsSystemPrompt: true,
202
- supportsStreamingToolUse: false,
203
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
204
- },
205
- //#endregion Anthropic Models
206
-
207
- //#region Cohere Models
208
- 'cohere.command-r-plus-v1:0': {
209
- llm: 'Bedrock',
210
- label: 'Cohere - Command R+',
211
- tokens: 128000,
212
- completionTokens: 4000, // Found 4000 Max tokens in the Playground
213
- supportsSystemPrompt: true,
214
- supportsStreamingToolUse: true,
215
- components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'Chatbot'],
216
- },
217
- 'cohere.command-r-v1:0': {
218
- llm: 'Bedrock',
219
- label: 'Cohere - Command R',
220
- tokens: 128000,
221
- completionTokens: 4000, // Found 4000 Max tokens in the Playground
222
- supportsSystemPrompt: true,
223
- supportsStreamingToolUse: true,
224
- components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'Chatbot'],
225
- },
226
- 'cohere.command-text-v14': {
227
- llm: 'Bedrock',
228
- label: 'Cohere - Command',
229
- tokens: 4000, // Found 4k
230
- completionTokens: 4000, // Found 4000 Max tokens in the Playground
231
- supportsSystemPrompt: false,
232
- supportsStreamingToolUse: false,
233
- components: ['PromptGenerator', 'GenAILLM'],
234
- },
235
- 'cohere.command-light-text-v14': {
236
- llm: 'Bedrock',
237
- label: 'Cohere - Command Light',
238
- tokens: 4000, // 4k tokens
239
- completionTokens: 4000, // Found 4000 Max tokens in the Playground
240
- supportsSystemPrompt: false,
241
- supportsStreamingToolUse: false,
242
- components: ['PromptGenerator', 'GenAILLM'],
243
- },
244
- // 'Cohere - Embed English': { alias: 'cohere.embed-english-v3' }, // Converse API doesn't support
245
- // 'Cohere - Embed Multilingual': { alias: 'cohere.embed-multilingual-v3' }, // Converse API doesn't support
246
- //#endregion Cohere Models
247
-
248
- //#region Meta Models
249
-
250
- // 'Meta - Llama 2 Chat 13B' : { alias: 'meta.llama2-13b-chat-v1' } // Don't have access to the model
251
- // 'Meta - Llama 2 Chat 70B' : { alias: 'meta.llama2-70b-chat-v1' } // Don't have access to the model
252
- 'us.meta.llama3-2-1b-instruct-v1:0': {
253
- llm: 'Bedrock',
254
- label: 'Meta - Llama 3.2 1B Instruct',
255
- tokens: 200000,
256
- completionTokens: 2048,
257
- supportsSystemPrompt: true,
258
- supportsStreamingToolUse: false,
259
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
260
- tags: ['new'],
261
- },
262
- 'us.meta.llama3-2-3b-instruct-v1:0': {
263
- llm: 'Bedrock',
264
- label: 'Meta - Llama 3.2 3B Instruct',
265
- tokens: 200000,
266
- completionTokens: 2048,
267
- supportsSystemPrompt: true,
268
- supportsStreamingToolUse: false,
269
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
270
- tags: ['new'],
271
- },
272
- 'meta.llama3-8b-instruct-v1:0': {
273
- llm: 'Bedrock',
274
- label: 'Meta - Llama 3 8B Instruct',
275
- tokens: 8192,
276
- completionTokens: 2048,
277
- supportsSystemPrompt: true,
278
- supportsStreamingToolUse: false,
279
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
280
- },
281
- 'meta.llama3-70b-instruct-v1:0': {
282
- llm: 'Bedrock',
283
- label: 'Meta - Llama 3 70B Instruct',
284
- tokens: 8192,
285
- completionTokens: 2048,
286
- supportsSystemPrompt: true,
287
- supportsStreamingToolUse: false,
288
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
289
- },
290
- // 'Meta - Llama 3.1 8B Instruct': { alias: 'meta.llama3-1-8b-instruct-v1:0' }, // The provided model identifier is invalid.
291
- // 'Meta - Llama 3.1 70B Instruct': { alias: 'meta.llama3-1-70b-instruct-v1:0' }, // The provided model identifier is invalid.
292
- // 'Meta - Llama 3.1 405B Instruct': { alias: 'meta.llama3-1-405b-instruct-v1:0' }, // The provided model identifier is invalid.
293
-
294
- //#endregion Meta Models
295
-
296
- //#region Mistral Models
297
- 'mistral.mistral-7b-instruct-v0:2': {
298
- llm: 'Bedrock',
299
- label: 'Mistral AI - Mistral 7B Instruct',
300
- tokens: 32000,
301
- completionTokens: 8192,
302
- supportsSystemPrompt: false,
303
- supportsStreamingToolUse: false,
304
- components: ['PromptGenerator', 'GenAILLM'],
305
- },
306
- 'mistral.mixtral-8x7b-instruct-v0:1': {
307
- llm: 'Bedrock',
308
- label: 'Mistral AI - Mixtral 8X7B Instruct',
309
- tokens: 32000,
310
- completionTokens: 4096,
311
- supportsSystemPrompt: false,
312
- supportsStreamingToolUse: false,
313
- components: ['PromptGenerator', 'GenAILLM'],
314
- },
315
- 'mistral.mistral-large-2402-v1:0': {
316
- llm: 'Bedrock',
317
- label: 'Mistral AI - Mistral Large',
318
- tokens: 32000,
319
- completionTokens: 8192,
320
- supportsSystemPrompt: true,
321
- supportsStreamingToolUse: false,
322
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
323
- },
324
- // 'Mistral AI - Mistral Large 2 (24.07)' : { alias: 'mistral.mistral-large-2407-v1:0' } // The provided model identifier is invalid.
325
- 'mistral.mistral-small-2402-v1:0': {
326
- llm: 'Bedrock',
327
- label: 'Mistral AI - Mistral Small',
328
- tokens: 32000,
329
- completionTokens: 8192,
330
- supportsSystemPrompt: true,
331
- supportsStreamingToolUse: false,
332
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
333
- },
334
- //#endregion Mistral Models
335
-
336
- //#region Stability Models
337
- // 'Stability AI - Stable Diffusion XL 0.x' : { alias: 'stability.stable-diffusion-xl-v0' } // Converse API doesn't support
338
- // 'Stability AI - Stable Diffusion XL 1.x' : { alias: 'stability.stable-diffusion-xl-v1' } // Converse API doesn't support
339
- //#endregion Stability Models
340
-
341
- /**************************************************
342
- * Vertex AI Models
343
- **************************************************/
344
-
345
- /*
346
- Context token information sourced from :
347
- Vertex AI documentation - https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models
348
- Anthropic - https://docs.anthropic.com/en/docs/about-claude/models
349
-
350
- Completion token information sourced from:
351
- Vertex AI Text Playground - https://console.cloud.google.com/vertex-ai/generative/language/create/text?model=text-bison@001&authuser=1&project=opt-smythos-vertexia-432522
352
- */
353
-
354
- 'gemini-1.5-flash': {
355
- llm: 'VertexAI',
356
- label: 'Gemini 1.5 Flash',
357
- supportsSystemPrompt: true,
358
- tokens: 1048576,
359
- completionTokens: 8192,
360
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
361
- },
362
- 'gemini-1.5-pro': {
363
- llm: 'VertexAI',
364
- label: 'Gemini 1.5 Pro',
365
- supportsSystemPrompt: true,
366
- tokens: 2097152,
367
- completionTokens: 8192,
368
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
369
- },
370
- 'gemini-1.0-pro': {
371
- llm: 'VertexAI',
372
- label: 'Gemini 1.0 Pro',
373
- supportsSystemPrompt: true,
374
- tokens: 32760,
375
- completionTokens: 8192,
376
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
377
- },
378
- 'claude-3-5-haiku': {
379
- llm: 'VertexAI',
380
- label: 'Claude 3.5 Haiku',
381
- supportsSystemPrompt: true,
382
- tokens: 200000,
383
- completionTokens: 8192,
384
- tags: ['new'],
385
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
386
- },
387
- 'claude-3-5-sonnet-v2': {
388
- llm: 'VertexAI',
389
- label: 'Claude 3.5 Sonnet',
390
- supportsSystemPrompt: true,
391
- tokens: 200000,
392
- completionTokens: 8192,
393
- tags: ['v2', 'new'],
394
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
395
- },
396
- 'claude-3-5-sonnet': {
397
- llm: 'VertexAI',
398
- label: 'Claude 3.5 Sonnet',
399
- supportsSystemPrompt: true,
400
- tokens: 200000,
401
- completionTokens: 8192,
402
- tags: ['v1'],
403
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
404
- },
405
- 'claude-3-sonnet': {
406
- llm: 'VertexAI',
407
- label: 'Claude 3 Sonnet',
408
- supportsSystemPrompt: true,
409
- tokens: 200000,
410
- completionTokens: 4096,
411
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
412
- },
413
- 'claude-3-opus': {
414
- llm: 'VertexAI',
415
- label: 'Claude 3 Opus',
416
- supportsSystemPrompt: true,
417
- tokens: 200000,
418
- completionTokens: 4096,
419
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
420
- },
421
- 'claude-3-haiku': {
422
- llm: 'VertexAI',
423
- label: 'Claude 3 Haiku',
424
- supportsSystemPrompt: true,
425
- tokens: 200000,
426
- completionTokens: 4096,
427
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
428
- },
429
- gemma2: {
430
- llm: 'VertexAI',
431
- label: 'Gemma 2',
432
- supportsSystemPrompt: true,
433
- tokens: 8192, // @Ref https://huggingface.co/blog/gemma2
434
- completionTokens: 4096, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/google/model-garden/gemma2?authuser=1&project=opt-smythos-vertexia-432522
435
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
436
- },
437
- gemma: {
438
- llm: 'VertexAI',
439
- label: 'Gemma',
440
- supportsSystemPrompt: true,
441
- tokens: 8192, // @Ref https://huggingface.co/blog/gemma2
442
- completionTokens: 4096, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/google/model-garden/gemma?authuser=1&project=opt-smythos-vertexia-432522
443
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
444
- },
445
- codegemma: {
446
- llm: 'VertexAI',
447
- label: 'CodeGemma',
448
- supportsSystemPrompt: true,
449
- tokens: 8192, // @Ref - https://huggingface.co/google/gemma-7b-it/discussions/73
450
- completionTokens: 4096, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/google/model-garden/codegemma?authuser=1&project=opt-smythos-vertexia-432522
451
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
452
- },
453
- 'llama3-405b-instruct-maas': {
454
- llm: 'VertexAI',
455
- label: 'Llama 3.1 API Service',
456
- supportsSystemPrompt: true,
457
- tokens: 4096, // @Ref - https://docs.together.ai/docs/chat-models
458
- completionTokens: 4096, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/meta/model-garden/llama3-405b-instruct-maas?authuser=1&project=opt-smythos-vertexia-432522
459
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
460
- },
461
- llama3_1: {
462
- llm: 'VertexAI',
463
- label: 'Llama 3.1',
464
- supportsSystemPrompt: true,
465
- tokens: 4096, // by querying with Gemini - https://gemini.google.com/
466
- completionTokens: 4096, // by querying with Gemini - https://gemini.google.com/
467
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
468
- },
469
- 'llama-guard': {
470
- llm: 'VertexAI',
471
- label: 'Llama Guard',
472
- supportsSystemPrompt: true,
473
- tokens: 8192, // @Ref - by querying with Gemini - https://gemini.google.com/
474
- completionTokens: 4096, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/meta/model-garden/llama-guard?authuser=1&project=opt-smythos-vertexia-432522
475
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
476
- },
477
- llama3: {
478
- llm: 'VertexAI',
479
- label: 'Llama 3',
480
- supportsSystemPrompt: true,
481
- tokens: 8192, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/meta/model-garden/llama3?_ga=2.79473366.358158393.1724662649-247251619.1697981116&authuser=1&project=opt-smythos-vertexia-432522
482
- completionTokens: 4096, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/meta/model-garden/llama3?_ga=2.79473366.358158393.1724662649-247251619.1697981116&authuser=1&project=opt-smythos-vertexia-432522
483
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
484
- },
485
- llama2: {
486
- llm: 'VertexAI',
487
- label: 'Llama 2',
488
- value: 'llama2',
489
- supportsSystemPrompt: true,
490
- tokens: 4096, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/meta/model-garden/llama2?authuser=1&project=opt-smythos-vertexia-432522
491
- completionTokens: 4096, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/meta/model-garden/llama2?authuser=1&project=opt-smythos-vertexia-432522
492
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
493
- },
494
- 'mistral-large': {
495
- llm: 'VertexAI',
496
- label: 'Mistral Large (2407)',
497
- supportsSystemPrompt: true,
498
- tokens: 128000, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/mistralai/model-garden/mistral-large?authuser=1&project=opt-smythos-vertexia-432522
499
- completionTokens: 8192, // Guessing from the context window (tokens)
500
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
501
- },
502
- 'mistral-nemo': {
503
- llm: 'VertexAI',
504
- label: 'Mistral Nemo',
505
- supportsSystemPrompt: true,
506
- tokens: 128000, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/mistralai/model-garden/mistral-nemo?authuser=1&project=opt-smythos-vertexia-432522
507
- completionTokens: 8192, // Guessing from the context window (tokens)
508
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
509
- },
510
- codestral: {
511
- llm: 'VertexAI',
512
- label: 'Codestral',
513
- supportsSystemPrompt: true,
514
- tokens: 32000, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/mistralai/model-garden/codestral?authuser=1&project=opt-smythos-vertexia-432522
515
- completionTokens: 4096, // Guessing from the context window (tokens)
516
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
517
- },
518
- mixtral: {
519
- llm: 'VertexAI',
520
- label: 'Mixtral',
521
- supportsSystemPrompt: true,
522
- tokens: 32000, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/mistral-ai/model-garden/mixtral?authuser=1&project=opt-smythos-vertexia-432522
523
- completionTokens: 4096, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/mistral-ai/model-garden/mixtral?authuser=1&project=opt-smythos-vertexia-432522
524
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
525
- },
526
- 'chat-bison': {
527
- llm: 'VertexAI',
528
- label: 'PaLM 2 Chat Bison',
529
- supportsSystemPrompt: true,
530
- tokens: 4096, // @Ref - by querying with Gemini - https://gemini.google.com/
531
- completionTokens: 2048, // @Ref - https://console.cloud.google.com/vertex-ai/generative/language/create/chat?model=chat-bison@002&authuser=1&project=opt-smythos-vertexia-432522 [Playground]
532
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
533
- },
534
- 'text-bison': {
535
- llm: 'VertexAI',
536
- label: 'PaLM 2 Text Bison',
537
- supportsSystemPrompt: true,
538
- tokens: 32768, // by querying with Gemini - https://gemini.google.com/
539
- completionTokens: 2048, // @Ref - https://console.cloud.google.com/vertex-ai/generative/language/create/text?model=text-bison@001&authuser=1&project=opt-smythos-vertexia-432522 [Playground]
540
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
541
- },
542
- phi3: {
543
- llm: 'VertexAI',
544
- label: 'Phi-3',
545
- supportsSystemPrompt: true,
546
- tokens: 128000, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/microsoft/model-garden/phi3?authuser=1&project=opt-smythos-vertexia-432522
547
- completionTokens: 8192, // Guessing from the context window (tokens)
548
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
549
- },
550
- qwen2: {
551
- llm: 'VertexAI',
552
- label: 'Qwen2',
553
- supportsSystemPrompt: true,
554
- tokens: 131072, // @Ref - https://huggingface.co/Qwen/Qwen2-72B-Instruct
555
- completionTokens: 8192, // Guessing from the context window (tokens)
556
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
557
- },
558
- mammut: {
559
- llm: 'VertexAI',
560
- label: 'MaMMUT',
561
- supportsSystemPrompt: true,
562
- tokens: 4096, // @Ref - by querying with Gemini - https://gemini.google.com/
563
- completionTokens: 2048, // @Ref - by querying with Gemini - https://gemini.google.com/
564
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
565
- },
566
- 'lmsys-vicuna-7b': {
567
- llm: 'VertexAI',
568
- label: 'Vicuna',
569
- supportsSystemPrompt: true,
570
- tokens: 4096, // @Ref - https://docs.together.ai/docs/chat-models
571
- completionTokens: 2048, // @Ref - Guessing from the context window (tokens)
572
- components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
573
- },
574
- 'bio-gpt': {
575
- llm: 'VertexAI',
576
- label: 'BioGPT',
577
- supportsSystemPrompt: false,
578
- tokens: 4096, // Guessed value, no reference
579
- completionTokens: 4096, // Guessed value, no reference
580
- components: ['PromptGenerator', 'GenAILLM'],
581
- },
582
- 'microsoft-biomedclip': {
583
- llm: 'VertexAI',
584
- label: 'BiomedCLIP',
585
- supportsSystemPrompt: false,
586
- tokens: 4096, // Guessed value, no reference
587
- completionTokens: 4096, // Guessed value, no reference
588
- components: ['PromptGenerator', 'GenAILLM'],
589
- },
590
- mistral: {
591
- llm: 'VertexAI',
592
- label: 'Mistral Self-host (7B & Nemo)',
593
- supportsSystemPrompt: false,
594
- tokens: 4096, // Guessed value, no reference
595
- completionTokens: 4096, // Guessed value, no reference
596
- components: ['PromptGenerator', 'GenAILLM'],
597
- },
598
- nllb: {
599
- llm: 'VertexAI',
600
- label: 'NLLB',
601
- supportsSystemPrompt: false,
602
- tokens: 4096, // Guessed value, no reference
603
- completionTokens: 4096, // Guessed value, no reference
604
- components: ['PromptGenerator', 'GenAILLM'],
605
- },
606
- 'codellama-7b-hf': {
607
- llm: 'VertexAI',
608
- label: 'Code Llama',
609
- supportsSystemPrompt: false,
610
- tokens: 4096, // Guessed value, no reference
611
- completionTokens: 4096, // Guessed value, no reference
612
- components: ['PromptGenerator', 'GenAILLM'],
613
- },
614
- 'palmyra-med': {
615
- llm: 'VertexAI',
616
- label: 'Palmyra Med (Writer)',
617
- supportsSystemPrompt: false,
618
- tokens: 4096, // Guessed value, no reference
619
- completionTokens: 4096, // Guessed value, no reference
620
- components: ['PromptGenerator', 'GenAILLM'],
621
- },
622
- 'llama-2-quantized': {
623
- llm: 'VertexAI',
624
- label: 'Llama 2 (Quantized)',
625
- supportsSystemPrompt: false,
626
- tokens: 4096, // Guessed value, no reference
627
- completionTokens: 4096, // Guessed value, no reference
628
- components: ['PromptGenerator', 'GenAILLM'],
629
- },
630
- 'bert-base-uncased': {
631
- llm: 'VertexAI',
632
- label: 'BERT (PEFT)',
633
- supportsSystemPrompt: false,
634
- tokens: 4096, // Guessed value, no reference
635
- completionTokens: 4096, // Guessed value, no reference
636
- components: ['PromptGenerator', 'GenAILLM'],
637
- },
638
- 'falcon-instruct-7b-peft': {
639
- llm: 'VertexAI',
640
- label: 'Falcon-instruct (PEFT)',
641
- supportsSystemPrompt: false,
642
- tokens: 4096, // Guessed value, no reference
643
- completionTokens: 4096, // Guessed value, no reference
644
- components: ['PromptGenerator', 'GenAILLM'],
645
- },
646
- openllama: {
647
- llm: 'VertexAI',
648
- label: 'OpenLLaMA (PEFT)',
649
- supportsSystemPrompt: false,
650
- tokens: 4096, // Guessed value, no reference
651
- completionTokens: 4096, // Guessed value, no reference
652
- components: ['PromptGenerator', 'GenAILLM'],
653
- },
654
- 'roberta-large': {
655
- llm: 'VertexAI',
656
- label: 'RoBERTa-large (PEFT)',
657
- supportsSystemPrompt: false,
658
- tokens: 4096, // Guessed value, no reference
659
- completionTokens: 4096, // Guessed value, no reference
660
- components: ['PromptGenerator', 'GenAILLM'],
661
- },
662
- 'xlm-roberta-large': {
663
- llm: 'VertexAI',
664
- label: 'XLM-RoBERTa-large (PEFT)',
665
- supportsSystemPrompt: false,
666
- tokens: 4096, // Guessed value, no reference
667
- completionTokens: 4096, // Guessed value, no reference
668
- components: ['PromptGenerator', 'GenAILLM'],
669
- },
670
- 'bart-large-cnn': {
671
- llm: 'VertexAI',
672
- label: 'Bart-large-cnn',
673
- supportsSystemPrompt: false,
674
- tokens: 4096, // Guessed value, no reference
675
- completionTokens: 4096, // Guessed value, no reference
676
- components: ['PromptGenerator', 'GenAILLM'],
677
- },
678
- 'dolly-v2': {
679
- llm: 'VertexAI',
680
- label: 'Dolly-v2',
681
- supportsSystemPrompt: false,
682
- tokens: 4096, // Guessed value, no reference
683
- completionTokens: 4096, // Guessed value, no reference
684
- components: ['PromptGenerator', 'GenAILLM'],
685
- },
686
- imagetext: {
687
- llm: 'VertexAI',
688
- label: 'Imagen for Captioning & VQA',
689
- supportsSystemPrompt: false,
690
- tokens: 4096, // Guessed value, no reference
691
- completionTokens: 4096, // Guessed value, no reference
692
- components: ['PromptGenerator', 'GenAILLM'],
693
- },
694
- 'codechat-bison': {
695
- llm: 'VertexAI',
696
- label: 'Codey for Code Chat',
697
- supportsSystemPrompt: false,
698
- tokens: 4096, // Guessed value, no reference
699
- completionTokens: 4096, // Guessed value, no reference
700
- components: ['PromptGenerator', 'GenAILLM'],
701
- },
702
- 'code-bison': {
703
- llm: 'VertexAI',
704
- label: 'Codey for Code Generation',
705
- supportsSystemPrompt: false,
706
- tokens: 4096, // Guessed value, no reference
707
- completionTokens: 4096, // Guessed value, no reference
708
- components: ['PromptGenerator', 'GenAILLM'],
709
- },
710
- 'code-gecko': {
711
- llm: 'VertexAI',
712
- label: 'Codey for Code Completion',
713
- supportsSystemPrompt: false,
714
- tokens: 4096, // Guessed value, no reference
715
- completionTokens: 4096, // Guessed value, no reference
716
- components: ['PromptGenerator', 'GenAILLM'],
717
- },
718
- 'text-unicorn': {
719
- llm: 'VertexAI',
720
- label: 'PaLM 2 Text Unicorn',
721
- supportsSystemPrompt: false,
722
- tokens: 4096, // Guessed value, no reference
723
- completionTokens: 4096, // Guessed value, no reference
724
- components: ['PromptGenerator', 'GenAILLM'],
725
- },
726
- 'textembedding-gecko': {
727
- llm: 'VertexAI',
728
- label: 'Embeddings for Text',
729
- supportsSystemPrompt: false,
730
- tokens: 4096, // Guessed value, no reference
731
- completionTokens: 4096, // Guessed value, no reference
732
- components: ['PromptGenerator', 'GenAILLM'],
733
- },
734
- 't5-flan': {
735
- llm: 'VertexAI',
736
- label: 'T5-FLAN',
737
- supportsSystemPrompt: false,
738
- tokens: 4096, // Guessed value, no reference
739
- completionTokens: 4096, // Guessed value, no reference
740
- components: ['PromptGenerator', 'GenAILLM'],
741
- },
742
- 't5-1.1': {
743
- llm: 'VertexAI',
744
- label: 'T5-1.1',
745
- supportsSystemPrompt: false,
746
- tokens: 4096, // Guessed value, no reference
747
- completionTokens: 4096, // Guessed value, no reference
748
- components: ['PromptGenerator', 'GenAILLM'],
749
- },
750
- 'blip2-opt-2.7-b': {
751
- llm: 'VertexAI',
752
- label: 'BLIP2',
753
- supportsSystemPrompt: false,
754
- tokens: 4096, // Guessed value, no reference
755
- completionTokens: 4096, // Guessed value, no reference
756
- components: ['PromptGenerator', 'GenAILLM'],
757
- },
758
- 'instruct-pix2pix': {
759
- llm: 'VertexAI',
760
- label: 'InstructPix2Pix',
761
- supportsSystemPrompt: false,
762
- tokens: 4096, // Guessed value, no reference
763
- completionTokens: 4096, // Guessed value, no reference
764
- components: ['PromptGenerator', 'GenAILLM'],
765
- },
766
- 'bert-base': {
767
- llm: 'VertexAI',
768
- label: 'BERT',
769
- supportsSystemPrompt: false,
770
- tokens: 4096, // Guessed value, no reference
771
- completionTokens: 4096, // Guessed value, no reference
772
- components: ['PromptGenerator', 'GenAILLM'],
773
- },
774
- 'mediapipe-mobile-bert-classifier': {
775
- llm: 'VertexAI',
776
- label: 'MobileBERT Classifier (MediaPipe)',
777
- supportsSystemPrompt: false,
778
- tokens: 4096, // Guessed value, no reference
779
- completionTokens: 4096, // Guessed value, no reference
780
- components: ['PromptGenerator', 'GenAILLM'],
781
- },
782
- 'mediapipe-average-word-embedding-classifier': {
783
- llm: 'VertexAI',
784
- label: 'Average Word Embedding Classifier (MediaPipe)',
785
- supportsSystemPrompt: false,
786
- tokens: 4096, // Guessed value, no reference
787
- completionTokens: 4096, // Guessed value, no reference
788
- components: ['PromptGenerator', 'GenAILLM'],
789
- },
790
- pic2word: {
791
- llm: 'VertexAI',
792
- label: 'Pic2Word Composed Image Retrieval',
793
- supportsSystemPrompt: false,
794
- tokens: 4096, // Guessed value, no reference
795
- completionTokens: 4096, // Guessed value, no reference
796
- components: ['PromptGenerator', 'GenAILLM'],
797
- },
798
- 'text-translation': {
799
- llm: 'VertexAI',
800
- label: 'Text Translation',
801
- supportsSystemPrompt: false,
802
- tokens: 4096, // Guessed value, no reference
803
- completionTokens: 4096, // Guessed value, no reference
804
- components: ['PromptGenerator', 'GenAILLM'],
805
- },
806
- 'language-v1-moderate-text': {
807
- llm: 'VertexAI',
808
- label: 'Text Moderation',
809
- supportsSystemPrompt: false,
810
- tokens: 4096, // Guessed value, no reference
811
- completionTokens: 4096, // Guessed value, no reference
812
- components: ['PromptGenerator', 'GenAILLM'],
813
- },
814
- 'language-v1-analyze-syntax': {
815
- llm: 'VertexAI',
816
- label: 'Syntax analysis',
817
- supportsSystemPrompt: false,
818
- tokens: 4096, // Guessed value, no reference
819
- completionTokens: 4096, // Guessed value, no reference
820
- components: ['PromptGenerator', 'GenAILLM'],
821
- },
822
- 'language-v1-analyze-entity-sentiment': {
823
- llm: 'VertexAI',
824
- label: 'Entity sentiment analysis',
825
- supportsSystemPrompt: false,
826
- tokens: 4096, // Guessed value, no reference
827
- completionTokens: 4096, // Guessed value, no reference
828
- components: ['PromptGenerator', 'GenAILLM'],
829
- },
830
- 'language-v1-analyze-sentiment': {
831
- llm: 'VertexAI',
832
- label: 'Sentiment analysis',
833
- supportsSystemPrompt: false,
834
- tokens: 4096, // Guessed value, no reference
835
- completionTokens: 4096, // Guessed value, no reference
836
- components: ['PromptGenerator', 'GenAILLM'],
837
- },
838
- 'language-v1-classify-text-v1': {
839
- llm: 'VertexAI',
840
- label: 'Content classification',
841
- supportsSystemPrompt: false,
842
- tokens: 4096, // Guessed value, no reference
843
- completionTokens: 4096, // Guessed value, no reference
844
- components: ['PromptGenerator', 'GenAILLM'],
845
- },
846
- 'pt-test': {
847
- llm: 'VertexAI',
848
- label: 'Entity analysis',
849
- supportsSystemPrompt: false,
850
- tokens: 4096, // Guessed value, no reference
851
- completionTokens: 4096, // Guessed value, no reference
852
- components: ['PromptGenerator', 'GenAILLM'],
853
- },
854
- };
1
+ //! ***IMPORTANT***
2
+ // * WE HAVE SAME FILE IN THREE PLACE, SO ANY CHANGES SHOULD BE SYNCED
3
+ // * - /smyth-builder-ui/src/shared/custom-models.ts
4
+ // * - /smyth-builder-server/src/services/LLMHelper/custom-models.ts
5
+ // * - /smyth-runtime/src/subsystems/LLMManager/custom-models.ts
6
+ //! ***IMPORTANT***
7
+ /**
8
+ * Custom model configurations for various AI providers.
9
+ * This template is useful for maintaining up-to-date model aliases and token information,
10
+ * even though we store custom model data in the database.
11
+ */
12
+
13
+ export const customModels = {
14
+ /**************************************************
15
+ * AWS Bedrock Models
16
+ **************************************************/
17
+
18
+ /*
19
+ Context token information sourced from the AWS Bedrock Console:
20
+ https://us-east-1.console.aws.amazon.com/bedrock/home?region=us-east-1#/overview
21
+
22
+ Completion token information sourced from the AWS Bedrock Text Playground:
23
+ https://us-east-1.console.aws.amazon.com/bedrock/home?region=us-east-1#/text-playground
24
+
25
+ Supported model features information sourced from - https://docs.aws.amazon.com/bedrock/latest/userguide/conversation-inference-supported-models-features.html
26
+ */
27
+
28
+ //#region AI21 Labs Models
29
+ 'ai21.jamba-1-5-mini-v1:0': {
30
+ llm: 'Bedrock',
31
+ label: 'AI21 Labs - Jamba 1.5 Mini',
32
+ tokens: 256000,
33
+ completionTokens: 4096,
34
+ supportsSystemPrompt: true,
35
+ supportsStreamingToolUse: false,
36
+ components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'GenAILLM'],
37
+ tags: ['new'],
38
+ },
39
+ 'ai21.jamba-1-5-large-v1:0': {
40
+ llm: 'Bedrock',
41
+ label: 'AI21 Labs - Jamba 1.5 Large',
42
+ tokens: 256000,
43
+ completionTokens: 4096,
44
+ supportsSystemPrompt: true,
45
+ supportsStreamingToolUse: false,
46
+ components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'GenAILLM'],
47
+ tags: ['new'],
48
+ },
49
+ 'ai21.jamba-instruct-v1:0': {
50
+ llm: 'Bedrock',
51
+ label: 'AI21 Labs - Jamba-Instruct',
52
+ tokens: 256000,
53
+ completionTokens: 4096,
54
+ supportsSystemPrompt: true,
55
+ supportsStreamingToolUse: false,
56
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
57
+ },
58
+ 'ai21.j2-ultra-v1': {
59
+ llm: 'Bedrock',
60
+ label: 'AI21 Labs - Jurassic-2 Ultra',
61
+ tokens: 8191,
62
+ completionTokens: 8191,
63
+ supportsSystemPrompt: false,
64
+ supportsStreamingToolUse: false,
65
+ components: ['PromptGenerator', 'GenAILLM'],
66
+ },
67
+ 'ai21.j2-mid-v1': {
68
+ llm: 'Bedrock',
69
+ label: 'AI21 Labs - Jurassic-2 Mid',
70
+ tokens: 8191,
71
+ completionTokens: 8191,
72
+ supportsSystemPrompt: false,
73
+ supportsStreamingToolUse: false,
74
+ components: ['PromptGenerator', 'GenAILLM'],
75
+ },
76
+ //#endregion AI21 Labs Models
77
+
78
+ //#region Amazon Models
79
+ 'amazon.titan-text-premier-v1:0': {
80
+ llm: 'Bedrock',
81
+ label: 'Amazon - Titan Text Premier',
82
+ tokens: 32000,
83
+ completionTokens: 3000,
84
+ supportsSystemPrompt: false,
85
+ supportsStreamingToolUse: false,
86
+ components: ['PromptGenerator', 'GenAILLM'],
87
+ },
88
+ 'amazon.titan-text-express-v1': {
89
+ llm: 'Bedrock',
90
+ label: 'Amazon - Titan Text G1 - Express',
91
+ tokens: 8192,
92
+ completionTokens: 4096,
93
+ supportsSystemPrompt: false,
94
+ supportsStreamingToolUse: false,
95
+ components: ['PromptGenerator', 'GenAILLM'],
96
+ },
97
+ 'amazon.titan-text-lite-v1': {
98
+ llm: 'Bedrock',
99
+ label: 'Amazon - Titan Text G1 - Lite',
100
+ tokens: 4096,
101
+ completionTokens: 4096,
102
+ supportsSystemPrompt: false,
103
+ supportsStreamingToolUse: false,
104
+ components: ['PromptGenerator', 'GenAILLM'],
105
+ },
106
+ // 'Amazon - Titan Embeddings G1 - Text': { alias: 'amazon.titan-embed-text-v1' }, // Converse API doesn't support
107
+ // 'Amazon - Titan Embedding Text v2': { alias: 'amazon.titan-embed-text-v2:0' }, // Converse API doesn't support
108
+ // 'Amazon - Titan Multimodal Embeddings G1': { alias: 'amazon.titan-embed-image-v1' }, // Converse API doesn't support
109
+ // 'Amazon - Titan Image Generator G1 V1': { alias: 'amazon.titan-image-generator-v1' }, // Converse API doesn't support
110
+ // 'Amazon - Titan Image Generator G1 V2': { alias: 'amazon.titan-image-generator-v2:0' }, // Converse API doesn't support
111
+
112
+ //#endregion Amazon Models
113
+
114
+ //#region Anthropic Models
115
+ // * NOTE: It's required to submit business info to get access for Anthropic models
116
+ // * @Ref of Anthropic tokens and completionTokens - https://docs.anthropic.com/en/docs/about-claude/models
117
+ 'us.anthropic.claude-3-5-haiku-20241022-v1:0': {
118
+ llm: 'Bedrock',
119
+ label: 'Anthropic - Claude 3.5 Haiku',
120
+ tokens: 200000,
121
+ completionTokens: 8192,
122
+ supportsSystemPrompt: true,
123
+ supportsStreamingToolUse: true,
124
+ components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'Chatbot'],
125
+ tags: ['v1:0', 'new'],
126
+ },
127
+ 'us.anthropic.claude-3-5-sonnet-20241022-v2:0': {
128
+ llm: 'Bedrock',
129
+ label: 'Anthropic - Claude 3.5 Sonnet',
130
+ tokens: 200000,
131
+ completionTokens: 8192,
132
+ supportsSystemPrompt: true,
133
+ supportsStreamingToolUse: true,
134
+ components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'Chatbot'],
135
+ tags: ['v2:0', 'new'],
136
+ },
137
+ 'anthropic.claude-3-5-sonnet-20240620-v1:0': {
138
+ llm: 'Bedrock',
139
+ label: 'Anthropic - Claude 3.5 Sonnet',
140
+ tokens: 200000,
141
+ completionTokens: 8192,
142
+ supportsSystemPrompt: true,
143
+ supportsStreamingToolUse: true,
144
+ components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'Chatbot'],
145
+ tags: ['v1:0'],
146
+ },
147
+ 'anthropic.claude-3-sonnet-20240229-v1:0': {
148
+ llm: 'Bedrock',
149
+ label: 'Anthropic - Claude 3 Sonnet',
150
+ tokens: 200000,
151
+ completionTokens: 4096,
152
+ supportsSystemPrompt: true,
153
+ supportsStreamingToolUse: true,
154
+ components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'Chatbot'],
155
+ tags: ['v1:0'],
156
+ },
157
+ 'anthropic.claude-3-haiku-20240307-v1:0': {
158
+ llm: 'Bedrock',
159
+ label: 'Anthropic - Claude 3 Haiku',
160
+ tokens: 200000,
161
+ completionTokens: 4096,
162
+ supportsSystemPrompt: true,
163
+ supportsStreamingToolUse: true,
164
+ components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'Chatbot'],
165
+ tags: ['v1:0'],
166
+ },
167
+ 'anthropic.claude-3-opus-20240229-v1:0': {
168
+ llm: 'Bedrock',
169
+ label: 'Anthropic - Claude 3 Opus',
170
+ tokens: 200000,
171
+ completionTokens: 4096,
172
+ supportsSystemPrompt: true,
173
+ supportsStreamingToolUse: true,
174
+ components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'Chatbot'],
175
+ tags: ['v1:0'],
176
+ },
177
+ 'anthropic.claude-v2:1': {
178
+ llm: 'Bedrock',
179
+ label: 'Anthropic - Claude 2.1',
180
+ tokens: 200000,
181
+ completionTokens: 4096,
182
+ supportsSystemPrompt: true,
183
+ supportsStreamingToolUse: false,
184
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
185
+ tags: ['v2:1'],
186
+ },
187
+ 'anthropic.claude-v2': {
188
+ llm: 'Bedrock',
189
+ label: 'Anthropic - Claude 2.0',
190
+ tokens: 100000,
191
+ completionTokens: 4096,
192
+ supportsSystemPrompt: true,
193
+ supportsStreamingToolUse: false,
194
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
195
+ },
196
+ 'anthropic.claude-instant-v1': {
197
+ llm: 'Bedrock',
198
+ label: 'Anthropic - Claude Instant',
199
+ tokens: 100000,
200
+ completionTokens: 4096,
201
+ supportsSystemPrompt: true,
202
+ supportsStreamingToolUse: false,
203
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
204
+ },
205
+ //#endregion Anthropic Models
206
+
207
+ //#region Cohere Models
208
+ 'cohere.command-r-plus-v1:0': {
209
+ llm: 'Bedrock',
210
+ label: 'Cohere - Command R+',
211
+ tokens: 128000,
212
+ completionTokens: 4000, // Found 4000 Max tokens in the Playground
213
+ supportsSystemPrompt: true,
214
+ supportsStreamingToolUse: true,
215
+ components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'Chatbot'],
216
+ },
217
+ 'cohere.command-r-v1:0': {
218
+ llm: 'Bedrock',
219
+ label: 'Cohere - Command R',
220
+ tokens: 128000,
221
+ completionTokens: 4000, // Found 4000 Max tokens in the Playground
222
+ supportsSystemPrompt: true,
223
+ supportsStreamingToolUse: true,
224
+ components: ['PromptGenerator', 'LLMAssistant', 'AgentPlugin', 'Chatbot'],
225
+ },
226
+ 'cohere.command-text-v14': {
227
+ llm: 'Bedrock',
228
+ label: 'Cohere - Command',
229
+ tokens: 4000, // Found 4k
230
+ completionTokens: 4000, // Found 4000 Max tokens in the Playground
231
+ supportsSystemPrompt: false,
232
+ supportsStreamingToolUse: false,
233
+ components: ['PromptGenerator', 'GenAILLM'],
234
+ },
235
+ 'cohere.command-light-text-v14': {
236
+ llm: 'Bedrock',
237
+ label: 'Cohere - Command Light',
238
+ tokens: 4000, // 4k tokens
239
+ completionTokens: 4000, // Found 4000 Max tokens in the Playground
240
+ supportsSystemPrompt: false,
241
+ supportsStreamingToolUse: false,
242
+ components: ['PromptGenerator', 'GenAILLM'],
243
+ },
244
+ // 'Cohere - Embed English': { alias: 'cohere.embed-english-v3' }, // Converse API doesn't support
245
+ // 'Cohere - Embed Multilingual': { alias: 'cohere.embed-multilingual-v3' }, // Converse API doesn't support
246
+ //#endregion Cohere Models
247
+
248
+ //#region Meta Models
249
+
250
+ // 'Meta - Llama 2 Chat 13B' : { alias: 'meta.llama2-13b-chat-v1' } // Don't have access to the model
251
+ // 'Meta - Llama 2 Chat 70B' : { alias: 'meta.llama2-70b-chat-v1' } // Don't have access to the model
252
+ 'us.meta.llama3-2-1b-instruct-v1:0': {
253
+ llm: 'Bedrock',
254
+ label: 'Meta - Llama 3.2 1B Instruct',
255
+ tokens: 200000,
256
+ completionTokens: 2048,
257
+ supportsSystemPrompt: true,
258
+ supportsStreamingToolUse: false,
259
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
260
+ tags: ['new'],
261
+ },
262
+ 'us.meta.llama3-2-3b-instruct-v1:0': {
263
+ llm: 'Bedrock',
264
+ label: 'Meta - Llama 3.2 3B Instruct',
265
+ tokens: 200000,
266
+ completionTokens: 2048,
267
+ supportsSystemPrompt: true,
268
+ supportsStreamingToolUse: false,
269
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
270
+ tags: ['new'],
271
+ },
272
+ 'meta.llama3-8b-instruct-v1:0': {
273
+ llm: 'Bedrock',
274
+ label: 'Meta - Llama 3 8B Instruct',
275
+ tokens: 8192,
276
+ completionTokens: 2048,
277
+ supportsSystemPrompt: true,
278
+ supportsStreamingToolUse: false,
279
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
280
+ },
281
+ 'meta.llama3-70b-instruct-v1:0': {
282
+ llm: 'Bedrock',
283
+ label: 'Meta - Llama 3 70B Instruct',
284
+ tokens: 8192,
285
+ completionTokens: 2048,
286
+ supportsSystemPrompt: true,
287
+ supportsStreamingToolUse: false,
288
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
289
+ },
290
+ // 'Meta - Llama 3.1 8B Instruct': { alias: 'meta.llama3-1-8b-instruct-v1:0' }, // The provided model identifier is invalid.
291
+ // 'Meta - Llama 3.1 70B Instruct': { alias: 'meta.llama3-1-70b-instruct-v1:0' }, // The provided model identifier is invalid.
292
+ // 'Meta - Llama 3.1 405B Instruct': { alias: 'meta.llama3-1-405b-instruct-v1:0' }, // The provided model identifier is invalid.
293
+
294
+ //#endregion Meta Models
295
+
296
+ //#region Mistral Models
297
+ 'mistral.mistral-7b-instruct-v0:2': {
298
+ llm: 'Bedrock',
299
+ label: 'Mistral AI - Mistral 7B Instruct',
300
+ tokens: 32000,
301
+ completionTokens: 8192,
302
+ supportsSystemPrompt: false,
303
+ supportsStreamingToolUse: false,
304
+ components: ['PromptGenerator', 'GenAILLM'],
305
+ },
306
+ 'mistral.mixtral-8x7b-instruct-v0:1': {
307
+ llm: 'Bedrock',
308
+ label: 'Mistral AI - Mixtral 8X7B Instruct',
309
+ tokens: 32000,
310
+ completionTokens: 4096,
311
+ supportsSystemPrompt: false,
312
+ supportsStreamingToolUse: false,
313
+ components: ['PromptGenerator', 'GenAILLM'],
314
+ },
315
+ 'mistral.mistral-large-2402-v1:0': {
316
+ llm: 'Bedrock',
317
+ label: 'Mistral AI - Mistral Large',
318
+ tokens: 32000,
319
+ completionTokens: 8192,
320
+ supportsSystemPrompt: true,
321
+ supportsStreamingToolUse: false,
322
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
323
+ },
324
+ // 'Mistral AI - Mistral Large 2 (24.07)' : { alias: 'mistral.mistral-large-2407-v1:0' } // The provided model identifier is invalid.
325
+ 'mistral.mistral-small-2402-v1:0': {
326
+ llm: 'Bedrock',
327
+ label: 'Mistral AI - Mistral Small',
328
+ tokens: 32000,
329
+ completionTokens: 8192,
330
+ supportsSystemPrompt: true,
331
+ supportsStreamingToolUse: false,
332
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
333
+ },
334
+ //#endregion Mistral Models
335
+
336
+ //#region Stability Models
337
+ // 'Stability AI - Stable Diffusion XL 0.x' : { alias: 'stability.stable-diffusion-xl-v0' } // Converse API doesn't support
338
+ // 'Stability AI - Stable Diffusion XL 1.x' : { alias: 'stability.stable-diffusion-xl-v1' } // Converse API doesn't support
339
+ //#endregion Stability Models
340
+
341
+ /**************************************************
342
+ * Vertex AI Models
343
+ **************************************************/
344
+
345
+ /*
346
+ Context token information sourced from :
347
+ Vertex AI documentation - https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models
348
+ Anthropic - https://docs.anthropic.com/en/docs/about-claude/models
349
+
350
+ Completion token information sourced from:
351
+ Vertex AI Text Playground - https://console.cloud.google.com/vertex-ai/generative/language/create/text?model=text-bison@001&authuser=1&project=opt-smythos-vertexia-432522
352
+ */
353
+
354
+ 'gemini-1.5-flash': {
355
+ llm: 'VertexAI',
356
+ label: 'Gemini 1.5 Flash',
357
+ supportsSystemPrompt: true,
358
+ tokens: 1048576,
359
+ completionTokens: 8192,
360
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
361
+ },
362
+ 'gemini-1.5-pro': {
363
+ llm: 'VertexAI',
364
+ label: 'Gemini 1.5 Pro',
365
+ supportsSystemPrompt: true,
366
+ tokens: 2097152,
367
+ completionTokens: 8192,
368
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
369
+ },
370
+ 'gemini-1.0-pro': {
371
+ llm: 'VertexAI',
372
+ label: 'Gemini 1.0 Pro',
373
+ supportsSystemPrompt: true,
374
+ tokens: 32760,
375
+ completionTokens: 8192,
376
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
377
+ },
378
+ 'claude-3-5-haiku': {
379
+ llm: 'VertexAI',
380
+ label: 'Claude 3.5 Haiku',
381
+ supportsSystemPrompt: true,
382
+ tokens: 200000,
383
+ completionTokens: 8192,
384
+ tags: ['new'],
385
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
386
+ },
387
+ 'claude-3-5-sonnet-v2': {
388
+ llm: 'VertexAI',
389
+ label: 'Claude 3.5 Sonnet',
390
+ supportsSystemPrompt: true,
391
+ tokens: 200000,
392
+ completionTokens: 8192,
393
+ tags: ['v2', 'new'],
394
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
395
+ },
396
+ 'claude-3-5-sonnet': {
397
+ llm: 'VertexAI',
398
+ label: 'Claude 3.5 Sonnet',
399
+ supportsSystemPrompt: true,
400
+ tokens: 200000,
401
+ completionTokens: 8192,
402
+ tags: ['v1'],
403
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
404
+ },
405
+ 'claude-3-sonnet': {
406
+ llm: 'VertexAI',
407
+ label: 'Claude 3 Sonnet',
408
+ supportsSystemPrompt: true,
409
+ tokens: 200000,
410
+ completionTokens: 4096,
411
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
412
+ },
413
+ 'claude-3-opus': {
414
+ llm: 'VertexAI',
415
+ label: 'Claude 3 Opus',
416
+ supportsSystemPrompt: true,
417
+ tokens: 200000,
418
+ completionTokens: 4096,
419
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
420
+ },
421
+ 'claude-3-haiku': {
422
+ llm: 'VertexAI',
423
+ label: 'Claude 3 Haiku',
424
+ supportsSystemPrompt: true,
425
+ tokens: 200000,
426
+ completionTokens: 4096,
427
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
428
+ },
429
+ gemma2: {
430
+ llm: 'VertexAI',
431
+ label: 'Gemma 2',
432
+ supportsSystemPrompt: true,
433
+ tokens: 8192, // @Ref https://huggingface.co/blog/gemma2
434
+ completionTokens: 4096, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/google/model-garden/gemma2?authuser=1&project=opt-smythos-vertexia-432522
435
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
436
+ },
437
+ gemma: {
438
+ llm: 'VertexAI',
439
+ label: 'Gemma',
440
+ supportsSystemPrompt: true,
441
+ tokens: 8192, // @Ref https://huggingface.co/blog/gemma2
442
+ completionTokens: 4096, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/google/model-garden/gemma?authuser=1&project=opt-smythos-vertexia-432522
443
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
444
+ },
445
+ codegemma: {
446
+ llm: 'VertexAI',
447
+ label: 'CodeGemma',
448
+ supportsSystemPrompt: true,
449
+ tokens: 8192, // @Ref - https://huggingface.co/google/gemma-7b-it/discussions/73
450
+ completionTokens: 4096, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/google/model-garden/codegemma?authuser=1&project=opt-smythos-vertexia-432522
451
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
452
+ },
453
+ 'llama3-405b-instruct-maas': {
454
+ llm: 'VertexAI',
455
+ label: 'Llama 3.1 API Service',
456
+ supportsSystemPrompt: true,
457
+ tokens: 4096, // @Ref - https://docs.together.ai/docs/chat-models
458
+ completionTokens: 4096, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/meta/model-garden/llama3-405b-instruct-maas?authuser=1&project=opt-smythos-vertexia-432522
459
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
460
+ },
461
+ llama3_1: {
462
+ llm: 'VertexAI',
463
+ label: 'Llama 3.1',
464
+ supportsSystemPrompt: true,
465
+ tokens: 4096, // by querying with Gemini - https://gemini.google.com/
466
+ completionTokens: 4096, // by querying with Gemini - https://gemini.google.com/
467
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
468
+ },
469
+ 'llama-guard': {
470
+ llm: 'VertexAI',
471
+ label: 'Llama Guard',
472
+ supportsSystemPrompt: true,
473
+ tokens: 8192, // @Ref - by querying with Gemini - https://gemini.google.com/
474
+ completionTokens: 4096, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/meta/model-garden/llama-guard?authuser=1&project=opt-smythos-vertexia-432522
475
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
476
+ },
477
+ llama3: {
478
+ llm: 'VertexAI',
479
+ label: 'Llama 3',
480
+ supportsSystemPrompt: true,
481
+ tokens: 8192, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/meta/model-garden/llama3?_ga=2.79473366.358158393.1724662649-247251619.1697981116&authuser=1&project=opt-smythos-vertexia-432522
482
+ completionTokens: 4096, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/meta/model-garden/llama3?_ga=2.79473366.358158393.1724662649-247251619.1697981116&authuser=1&project=opt-smythos-vertexia-432522
483
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
484
+ },
485
+ llama2: {
486
+ llm: 'VertexAI',
487
+ label: 'Llama 2',
488
+ value: 'llama2',
489
+ supportsSystemPrompt: true,
490
+ tokens: 4096, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/meta/model-garden/llama2?authuser=1&project=opt-smythos-vertexia-432522
491
+ completionTokens: 4096, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/meta/model-garden/llama2?authuser=1&project=opt-smythos-vertexia-432522
492
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
493
+ },
494
+ 'mistral-large': {
495
+ llm: 'VertexAI',
496
+ label: 'Mistral Large (2407)',
497
+ supportsSystemPrompt: true,
498
+ tokens: 128000, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/mistralai/model-garden/mistral-large?authuser=1&project=opt-smythos-vertexia-432522
499
+ completionTokens: 8192, // Guessing from the context window (tokens)
500
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
501
+ },
502
+ 'mistral-nemo': {
503
+ llm: 'VertexAI',
504
+ label: 'Mistral Nemo',
505
+ supportsSystemPrompt: true,
506
+ tokens: 128000, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/mistralai/model-garden/mistral-nemo?authuser=1&project=opt-smythos-vertexia-432522
507
+ completionTokens: 8192, // Guessing from the context window (tokens)
508
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
509
+ },
510
+ codestral: {
511
+ llm: 'VertexAI',
512
+ label: 'Codestral',
513
+ supportsSystemPrompt: true,
514
+ tokens: 32000, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/mistralai/model-garden/codestral?authuser=1&project=opt-smythos-vertexia-432522
515
+ completionTokens: 4096, // Guessing from the context window (tokens)
516
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
517
+ },
518
+ mixtral: {
519
+ llm: 'VertexAI',
520
+ label: 'Mixtral',
521
+ supportsSystemPrompt: true,
522
+ tokens: 32000, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/mistral-ai/model-garden/mixtral?authuser=1&project=opt-smythos-vertexia-432522
523
+ completionTokens: 4096, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/mistral-ai/model-garden/mixtral?authuser=1&project=opt-smythos-vertexia-432522
524
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
525
+ },
526
+ 'chat-bison': {
527
+ llm: 'VertexAI',
528
+ label: 'PaLM 2 Chat Bison',
529
+ supportsSystemPrompt: true,
530
+ tokens: 4096, // @Ref - by querying with Gemini - https://gemini.google.com/
531
+ completionTokens: 2048, // @Ref - https://console.cloud.google.com/vertex-ai/generative/language/create/chat?model=chat-bison@002&authuser=1&project=opt-smythos-vertexia-432522 [Playground]
532
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
533
+ },
534
+ 'text-bison': {
535
+ llm: 'VertexAI',
536
+ label: 'PaLM 2 Text Bison',
537
+ supportsSystemPrompt: true,
538
+ tokens: 32768, // by querying with Gemini - https://gemini.google.com/
539
+ completionTokens: 2048, // @Ref - https://console.cloud.google.com/vertex-ai/generative/language/create/text?model=text-bison@001&authuser=1&project=opt-smythos-vertexia-432522 [Playground]
540
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
541
+ },
542
+ phi3: {
543
+ llm: 'VertexAI',
544
+ label: 'Phi-3',
545
+ supportsSystemPrompt: true,
546
+ tokens: 128000, // @Ref - https://console.cloud.google.com/vertex-ai/publishers/microsoft/model-garden/phi3?authuser=1&project=opt-smythos-vertexia-432522
547
+ completionTokens: 8192, // Guessing from the context window (tokens)
548
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
549
+ },
550
+ qwen2: {
551
+ llm: 'VertexAI',
552
+ label: 'Qwen2',
553
+ supportsSystemPrompt: true,
554
+ tokens: 131072, // @Ref - https://huggingface.co/Qwen/Qwen2-72B-Instruct
555
+ completionTokens: 8192, // Guessing from the context window (tokens)
556
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
557
+ },
558
+ mammut: {
559
+ llm: 'VertexAI',
560
+ label: 'MaMMUT',
561
+ supportsSystemPrompt: true,
562
+ tokens: 4096, // @Ref - by querying with Gemini - https://gemini.google.com/
563
+ completionTokens: 2048, // @Ref - by querying with Gemini - https://gemini.google.com/
564
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
565
+ },
566
+ 'lmsys-vicuna-7b': {
567
+ llm: 'VertexAI',
568
+ label: 'Vicuna',
569
+ supportsSystemPrompt: true,
570
+ tokens: 4096, // @Ref - https://docs.together.ai/docs/chat-models
571
+ completionTokens: 2048, // @Ref - Guessing from the context window (tokens)
572
+ components: ['PromptGenerator', 'LLMAssistant', 'GenAILLM'],
573
+ },
574
+ 'bio-gpt': {
575
+ llm: 'VertexAI',
576
+ label: 'BioGPT',
577
+ supportsSystemPrompt: false,
578
+ tokens: 4096, // Guessed value, no reference
579
+ completionTokens: 4096, // Guessed value, no reference
580
+ components: ['PromptGenerator', 'GenAILLM'],
581
+ },
582
+ 'microsoft-biomedclip': {
583
+ llm: 'VertexAI',
584
+ label: 'BiomedCLIP',
585
+ supportsSystemPrompt: false,
586
+ tokens: 4096, // Guessed value, no reference
587
+ completionTokens: 4096, // Guessed value, no reference
588
+ components: ['PromptGenerator', 'GenAILLM'],
589
+ },
590
+ mistral: {
591
+ llm: 'VertexAI',
592
+ label: 'Mistral Self-host (7B & Nemo)',
593
+ supportsSystemPrompt: false,
594
+ tokens: 4096, // Guessed value, no reference
595
+ completionTokens: 4096, // Guessed value, no reference
596
+ components: ['PromptGenerator', 'GenAILLM'],
597
+ },
598
+ nllb: {
599
+ llm: 'VertexAI',
600
+ label: 'NLLB',
601
+ supportsSystemPrompt: false,
602
+ tokens: 4096, // Guessed value, no reference
603
+ completionTokens: 4096, // Guessed value, no reference
604
+ components: ['PromptGenerator', 'GenAILLM'],
605
+ },
606
+ 'codellama-7b-hf': {
607
+ llm: 'VertexAI',
608
+ label: 'Code Llama',
609
+ supportsSystemPrompt: false,
610
+ tokens: 4096, // Guessed value, no reference
611
+ completionTokens: 4096, // Guessed value, no reference
612
+ components: ['PromptGenerator', 'GenAILLM'],
613
+ },
614
+ 'palmyra-med': {
615
+ llm: 'VertexAI',
616
+ label: 'Palmyra Med (Writer)',
617
+ supportsSystemPrompt: false,
618
+ tokens: 4096, // Guessed value, no reference
619
+ completionTokens: 4096, // Guessed value, no reference
620
+ components: ['PromptGenerator', 'GenAILLM'],
621
+ },
622
+ 'llama-2-quantized': {
623
+ llm: 'VertexAI',
624
+ label: 'Llama 2 (Quantized)',
625
+ supportsSystemPrompt: false,
626
+ tokens: 4096, // Guessed value, no reference
627
+ completionTokens: 4096, // Guessed value, no reference
628
+ components: ['PromptGenerator', 'GenAILLM'],
629
+ },
630
+ 'bert-base-uncased': {
631
+ llm: 'VertexAI',
632
+ label: 'BERT (PEFT)',
633
+ supportsSystemPrompt: false,
634
+ tokens: 4096, // Guessed value, no reference
635
+ completionTokens: 4096, // Guessed value, no reference
636
+ components: ['PromptGenerator', 'GenAILLM'],
637
+ },
638
+ 'falcon-instruct-7b-peft': {
639
+ llm: 'VertexAI',
640
+ label: 'Falcon-instruct (PEFT)',
641
+ supportsSystemPrompt: false,
642
+ tokens: 4096, // Guessed value, no reference
643
+ completionTokens: 4096, // Guessed value, no reference
644
+ components: ['PromptGenerator', 'GenAILLM'],
645
+ },
646
+ openllama: {
647
+ llm: 'VertexAI',
648
+ label: 'OpenLLaMA (PEFT)',
649
+ supportsSystemPrompt: false,
650
+ tokens: 4096, // Guessed value, no reference
651
+ completionTokens: 4096, // Guessed value, no reference
652
+ components: ['PromptGenerator', 'GenAILLM'],
653
+ },
654
+ 'roberta-large': {
655
+ llm: 'VertexAI',
656
+ label: 'RoBERTa-large (PEFT)',
657
+ supportsSystemPrompt: false,
658
+ tokens: 4096, // Guessed value, no reference
659
+ completionTokens: 4096, // Guessed value, no reference
660
+ components: ['PromptGenerator', 'GenAILLM'],
661
+ },
662
+ 'xlm-roberta-large': {
663
+ llm: 'VertexAI',
664
+ label: 'XLM-RoBERTa-large (PEFT)',
665
+ supportsSystemPrompt: false,
666
+ tokens: 4096, // Guessed value, no reference
667
+ completionTokens: 4096, // Guessed value, no reference
668
+ components: ['PromptGenerator', 'GenAILLM'],
669
+ },
670
+ 'bart-large-cnn': {
671
+ llm: 'VertexAI',
672
+ label: 'Bart-large-cnn',
673
+ supportsSystemPrompt: false,
674
+ tokens: 4096, // Guessed value, no reference
675
+ completionTokens: 4096, // Guessed value, no reference
676
+ components: ['PromptGenerator', 'GenAILLM'],
677
+ },
678
+ 'dolly-v2': {
679
+ llm: 'VertexAI',
680
+ label: 'Dolly-v2',
681
+ supportsSystemPrompt: false,
682
+ tokens: 4096, // Guessed value, no reference
683
+ completionTokens: 4096, // Guessed value, no reference
684
+ components: ['PromptGenerator', 'GenAILLM'],
685
+ },
686
+ imagetext: {
687
+ llm: 'VertexAI',
688
+ label: 'Imagen for Captioning & VQA',
689
+ supportsSystemPrompt: false,
690
+ tokens: 4096, // Guessed value, no reference
691
+ completionTokens: 4096, // Guessed value, no reference
692
+ components: ['PromptGenerator', 'GenAILLM'],
693
+ },
694
+ 'codechat-bison': {
695
+ llm: 'VertexAI',
696
+ label: 'Codey for Code Chat',
697
+ supportsSystemPrompt: false,
698
+ tokens: 4096, // Guessed value, no reference
699
+ completionTokens: 4096, // Guessed value, no reference
700
+ components: ['PromptGenerator', 'GenAILLM'],
701
+ },
702
+ 'code-bison': {
703
+ llm: 'VertexAI',
704
+ label: 'Codey for Code Generation',
705
+ supportsSystemPrompt: false,
706
+ tokens: 4096, // Guessed value, no reference
707
+ completionTokens: 4096, // Guessed value, no reference
708
+ components: ['PromptGenerator', 'GenAILLM'],
709
+ },
710
+ 'code-gecko': {
711
+ llm: 'VertexAI',
712
+ label: 'Codey for Code Completion',
713
+ supportsSystemPrompt: false,
714
+ tokens: 4096, // Guessed value, no reference
715
+ completionTokens: 4096, // Guessed value, no reference
716
+ components: ['PromptGenerator', 'GenAILLM'],
717
+ },
718
+ 'text-unicorn': {
719
+ llm: 'VertexAI',
720
+ label: 'PaLM 2 Text Unicorn',
721
+ supportsSystemPrompt: false,
722
+ tokens: 4096, // Guessed value, no reference
723
+ completionTokens: 4096, // Guessed value, no reference
724
+ components: ['PromptGenerator', 'GenAILLM'],
725
+ },
726
+ 'textembedding-gecko': {
727
+ llm: 'VertexAI',
728
+ label: 'Embeddings for Text',
729
+ supportsSystemPrompt: false,
730
+ tokens: 4096, // Guessed value, no reference
731
+ completionTokens: 4096, // Guessed value, no reference
732
+ components: ['PromptGenerator', 'GenAILLM'],
733
+ },
734
+ 't5-flan': {
735
+ llm: 'VertexAI',
736
+ label: 'T5-FLAN',
737
+ supportsSystemPrompt: false,
738
+ tokens: 4096, // Guessed value, no reference
739
+ completionTokens: 4096, // Guessed value, no reference
740
+ components: ['PromptGenerator', 'GenAILLM'],
741
+ },
742
+ 't5-1.1': {
743
+ llm: 'VertexAI',
744
+ label: 'T5-1.1',
745
+ supportsSystemPrompt: false,
746
+ tokens: 4096, // Guessed value, no reference
747
+ completionTokens: 4096, // Guessed value, no reference
748
+ components: ['PromptGenerator', 'GenAILLM'],
749
+ },
750
+ 'blip2-opt-2.7-b': {
751
+ llm: 'VertexAI',
752
+ label: 'BLIP2',
753
+ supportsSystemPrompt: false,
754
+ tokens: 4096, // Guessed value, no reference
755
+ completionTokens: 4096, // Guessed value, no reference
756
+ components: ['PromptGenerator', 'GenAILLM'],
757
+ },
758
+ 'instruct-pix2pix': {
759
+ llm: 'VertexAI',
760
+ label: 'InstructPix2Pix',
761
+ supportsSystemPrompt: false,
762
+ tokens: 4096, // Guessed value, no reference
763
+ completionTokens: 4096, // Guessed value, no reference
764
+ components: ['PromptGenerator', 'GenAILLM'],
765
+ },
766
+ 'bert-base': {
767
+ llm: 'VertexAI',
768
+ label: 'BERT',
769
+ supportsSystemPrompt: false,
770
+ tokens: 4096, // Guessed value, no reference
771
+ completionTokens: 4096, // Guessed value, no reference
772
+ components: ['PromptGenerator', 'GenAILLM'],
773
+ },
774
+ 'mediapipe-mobile-bert-classifier': {
775
+ llm: 'VertexAI',
776
+ label: 'MobileBERT Classifier (MediaPipe)',
777
+ supportsSystemPrompt: false,
778
+ tokens: 4096, // Guessed value, no reference
779
+ completionTokens: 4096, // Guessed value, no reference
780
+ components: ['PromptGenerator', 'GenAILLM'],
781
+ },
782
+ 'mediapipe-average-word-embedding-classifier': {
783
+ llm: 'VertexAI',
784
+ label: 'Average Word Embedding Classifier (MediaPipe)',
785
+ supportsSystemPrompt: false,
786
+ tokens: 4096, // Guessed value, no reference
787
+ completionTokens: 4096, // Guessed value, no reference
788
+ components: ['PromptGenerator', 'GenAILLM'],
789
+ },
790
+ pic2word: {
791
+ llm: 'VertexAI',
792
+ label: 'Pic2Word Composed Image Retrieval',
793
+ supportsSystemPrompt: false,
794
+ tokens: 4096, // Guessed value, no reference
795
+ completionTokens: 4096, // Guessed value, no reference
796
+ components: ['PromptGenerator', 'GenAILLM'],
797
+ },
798
+ 'text-translation': {
799
+ llm: 'VertexAI',
800
+ label: 'Text Translation',
801
+ supportsSystemPrompt: false,
802
+ tokens: 4096, // Guessed value, no reference
803
+ completionTokens: 4096, // Guessed value, no reference
804
+ components: ['PromptGenerator', 'GenAILLM'],
805
+ },
806
+ 'language-v1-moderate-text': {
807
+ llm: 'VertexAI',
808
+ label: 'Text Moderation',
809
+ supportsSystemPrompt: false,
810
+ tokens: 4096, // Guessed value, no reference
811
+ completionTokens: 4096, // Guessed value, no reference
812
+ components: ['PromptGenerator', 'GenAILLM'],
813
+ },
814
+ 'language-v1-analyze-syntax': {
815
+ llm: 'VertexAI',
816
+ label: 'Syntax analysis',
817
+ supportsSystemPrompt: false,
818
+ tokens: 4096, // Guessed value, no reference
819
+ completionTokens: 4096, // Guessed value, no reference
820
+ components: ['PromptGenerator', 'GenAILLM'],
821
+ },
822
+ 'language-v1-analyze-entity-sentiment': {
823
+ llm: 'VertexAI',
824
+ label: 'Entity sentiment analysis',
825
+ supportsSystemPrompt: false,
826
+ tokens: 4096, // Guessed value, no reference
827
+ completionTokens: 4096, // Guessed value, no reference
828
+ components: ['PromptGenerator', 'GenAILLM'],
829
+ },
830
+ 'language-v1-analyze-sentiment': {
831
+ llm: 'VertexAI',
832
+ label: 'Sentiment analysis',
833
+ supportsSystemPrompt: false,
834
+ tokens: 4096, // Guessed value, no reference
835
+ completionTokens: 4096, // Guessed value, no reference
836
+ components: ['PromptGenerator', 'GenAILLM'],
837
+ },
838
+ 'language-v1-classify-text-v1': {
839
+ llm: 'VertexAI',
840
+ label: 'Content classification',
841
+ supportsSystemPrompt: false,
842
+ tokens: 4096, // Guessed value, no reference
843
+ completionTokens: 4096, // Guessed value, no reference
844
+ components: ['PromptGenerator', 'GenAILLM'],
845
+ },
846
+ 'pt-test': {
847
+ llm: 'VertexAI',
848
+ label: 'Entity analysis',
849
+ supportsSystemPrompt: false,
850
+ tokens: 4096, // Guessed value, no reference
851
+ completionTokens: 4096, // Guessed value, no reference
852
+ components: ['PromptGenerator', 'GenAILLM'],
853
+ },
854
+ };