@smythos/sre 1.6.1 → 1.6.8

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