@smythos/sre 1.5.45 → 1.5.50

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