@smythos/sre 1.5.53 → 1.5.54

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