@smythos/sre 1.5.43 → 1.5.45

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