@llumiverse/drivers 1.2.0 → 1.4.0-dev.20260629.090753Z
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.
- package/lib/{types/adobe → adobe}/firefly.d.ts +2 -1
- package/lib/adobe/firefly.d.ts.map +1 -0
- package/lib/adobe/firefly.js +120 -0
- package/lib/adobe/firefly.js.map +1 -0
- package/lib/anthropic/index.d.ts +23 -0
- package/lib/anthropic/index.d.ts.map +1 -0
- package/lib/anthropic/index.js +68 -0
- package/lib/anthropic/index.js.map +1 -0
- package/lib/{types/azure → azure}/azure_foundry.d.ts +9 -7
- package/lib/azure/azure_foundry.d.ts.map +1 -0
- package/lib/{esm/azure → azure}/azure_foundry.js +138 -124
- package/lib/azure/azure_foundry.js.map +1 -0
- package/lib/{types/bedrock → bedrock}/converse.d.ts +2 -2
- package/lib/bedrock/converse.d.ts.map +1 -0
- package/lib/{esm/bedrock → bedrock}/converse.js +86 -79
- package/lib/bedrock/converse.js.map +1 -0
- package/lib/bedrock/embeddings.d.ts +4 -0
- package/lib/bedrock/embeddings.d.ts.map +1 -0
- package/lib/bedrock/embeddings.js +359 -0
- package/lib/bedrock/embeddings.js.map +1 -0
- package/lib/{types/bedrock → bedrock}/index.d.ts +33 -9
- package/lib/bedrock/index.d.ts.map +1 -0
- package/lib/{esm/bedrock → bedrock}/index.js +599 -324
- package/lib/bedrock/index.js.map +1 -0
- package/lib/{types/bedrock → bedrock}/nova-image-payload.d.ts +5 -5
- package/lib/bedrock/nova-image-payload.d.ts.map +1 -0
- package/lib/{esm/bedrock → bedrock}/nova-image-payload.js +31 -36
- package/lib/bedrock/nova-image-payload.js.map +1 -0
- package/lib/{types/bedrock → bedrock}/payloads.d.ts +1 -1
- package/lib/bedrock/payloads.d.ts.map +1 -0
- package/lib/{cjs/bedrock → bedrock}/payloads.js.map +1 -1
- package/lib/{types/bedrock → bedrock}/s3.d.ts +3 -2
- package/lib/bedrock/s3.d.ts.map +1 -0
- package/lib/{esm/bedrock → bedrock}/s3.js +27 -23
- package/lib/bedrock/s3.js.map +1 -0
- package/lib/{types/bedrock → bedrock}/twelvelabs.d.ts +8 -9
- package/lib/bedrock/twelvelabs.d.ts.map +1 -0
- package/lib/{esm/bedrock → bedrock}/twelvelabs.js +17 -21
- package/lib/bedrock/twelvelabs.js.map +1 -0
- package/lib/{types/groq → groq}/index.d.ts +6 -4
- package/lib/groq/index.d.ts.map +1 -0
- package/lib/{esm/groq → groq}/index.js +72 -39
- package/lib/groq/index.js.map +1 -0
- package/lib/{types/huggingface_ie.d.ts → huggingface_ie.d.ts} +4 -3
- package/lib/huggingface_ie.d.ts.map +1 -0
- package/lib/{esm/huggingface_ie.js → huggingface_ie.js} +32 -31
- package/lib/huggingface_ie.js.map +1 -0
- package/lib/index.d.ts +16 -0
- package/lib/index.d.ts.map +1 -0
- package/lib/index.js +16 -0
- package/lib/index.js.map +1 -0
- package/lib/{types/mistral → mistral}/index.d.ts +6 -5
- package/lib/mistral/index.d.ts.map +1 -0
- package/lib/{esm/mistral → mistral}/index.js +66 -38
- package/lib/mistral/index.js.map +1 -0
- package/lib/{types/mistral → mistral}/types.d.ts +1 -0
- package/lib/mistral/types.d.ts.map +1 -0
- package/lib/mistral/types.js.map +1 -0
- package/lib/{types/openai → openai}/azure_openai.d.ts +5 -4
- package/lib/openai/azure_openai.d.ts.map +1 -0
- package/lib/{esm/openai → openai}/azure_openai.js +17 -14
- package/lib/openai/azure_openai.js.map +1 -0
- package/lib/openai/drivers.d.ts +4 -0
- package/lib/openai/drivers.d.ts.map +1 -0
- package/lib/openai/drivers.js +9 -0
- package/lib/openai/drivers.js.map +1 -0
- package/lib/{types/openai → openai}/index.d.ts +17 -5
- package/lib/openai/index.d.ts.map +1 -0
- package/lib/{esm/openai → openai}/index.js +254 -148
- package/lib/openai/index.js.map +1 -0
- package/lib/{types/openai → openai}/openai.d.ts +3 -3
- package/lib/openai/openai.d.ts.map +1 -0
- package/lib/openai/openai.js +15 -0
- package/lib/openai/openai.js.map +1 -0
- package/lib/{types/openai → openai}/openai_compatible.d.ts +3 -3
- package/lib/openai/openai_compatible.d.ts.map +1 -0
- package/lib/{esm/openai → openai}/openai_compatible.js +14 -11
- package/lib/openai/openai_compatible.js.map +1 -0
- package/lib/{types/openai → openai}/openai_format.d.ts +3 -2
- package/lib/openai/openai_format.d.ts.map +1 -0
- package/lib/{esm/openai → openai}/openai_format.js +66 -24
- package/lib/openai/openai_format.js.map +1 -0
- package/lib/{types/replicate.d.ts → replicate.d.ts} +3 -2
- package/lib/replicate.d.ts.map +1 -0
- package/lib/{esm/replicate.js → replicate.js} +59 -49
- package/lib/replicate.js.map +1 -0
- package/lib/shared/claude-messages.d.ts +99 -0
- package/lib/shared/claude-messages.d.ts.map +1 -0
- package/lib/shared/claude-messages.js +869 -0
- package/lib/shared/claude-messages.js.map +1 -0
- package/lib/{types/shared → shared}/claude-thinking.d.ts +1 -1
- package/lib/shared/claude-thinking.d.ts.map +1 -0
- package/lib/{esm/shared → shared}/claude-thinking.js +5 -7
- package/lib/shared/claude-thinking.js.map +1 -0
- package/lib/shared/debug-prompt.d.ts +5 -0
- package/lib/shared/debug-prompt.d.ts.map +1 -0
- package/lib/shared/debug-prompt.js +25 -0
- package/lib/shared/debug-prompt.js.map +1 -0
- package/lib/{types/test-driver → test-driver}/TestErrorCompletionStream.d.ts +1 -1
- package/lib/test-driver/TestErrorCompletionStream.d.ts.map +1 -0
- package/lib/test-driver/TestErrorCompletionStream.js +16 -0
- package/lib/test-driver/TestErrorCompletionStream.js.map +1 -0
- package/lib/{types/test-driver → test-driver}/TestValidationErrorCompletionStream.d.ts +1 -1
- package/lib/test-driver/TestValidationErrorCompletionStream.d.ts.map +1 -0
- package/lib/{esm/test-driver → test-driver}/TestValidationErrorCompletionStream.js +4 -4
- package/lib/test-driver/TestValidationErrorCompletionStream.js.map +1 -0
- package/lib/{types/test-driver → test-driver}/index.d.ts +3 -3
- package/lib/test-driver/index.d.ts.map +1 -0
- package/lib/{esm/test-driver → test-driver}/index.js +20 -20
- package/lib/test-driver/index.js.map +1 -0
- package/lib/{types/test-driver → test-driver}/utils.d.ts +1 -1
- package/lib/test-driver/utils.d.ts.map +1 -0
- package/lib/{esm/test-driver → test-driver}/utils.js +4 -4
- package/lib/test-driver/utils.js.map +1 -0
- package/lib/togetherai/index.d.ts +15 -0
- package/lib/togetherai/index.d.ts.map +1 -0
- package/lib/togetherai/index.js +54 -0
- package/lib/togetherai/index.js.map +1 -0
- package/lib/{types/togetherai → togetherai}/interfaces.d.ts +1 -1
- package/lib/togetherai/interfaces.d.ts.map +1 -0
- package/lib/{cjs/watsonx → togetherai}/interfaces.js.map +1 -1
- package/lib/vertexai/debug.d.ts +2 -0
- package/lib/vertexai/debug.d.ts.map +1 -0
- package/lib/{esm/vertexai → vertexai}/debug.js +1 -1
- package/lib/vertexai/debug.js.map +1 -0
- package/lib/vertexai/embeddings/embed-legacy-multimodal.d.ts +9 -0
- package/lib/vertexai/embeddings/embed-legacy-multimodal.d.ts.map +1 -0
- package/lib/vertexai/embeddings/embed-legacy-multimodal.js +141 -0
- package/lib/vertexai/embeddings/embed-legacy-multimodal.js.map +1 -0
- package/lib/vertexai/embeddings/embed.d.ts +13 -0
- package/lib/vertexai/embeddings/embed.d.ts.map +1 -0
- package/lib/vertexai/embeddings/embed.js +162 -0
- package/lib/vertexai/embeddings/embed.js.map +1 -0
- package/lib/vertexai/embeddings/source-utils.d.ts +9 -0
- package/lib/vertexai/embeddings/source-utils.d.ts.map +1 -0
- package/lib/vertexai/embeddings/source-utils.js +18 -0
- package/lib/vertexai/embeddings/source-utils.js.map +1 -0
- package/lib/{types/vertexai → vertexai}/index.d.ts +31 -19
- package/lib/vertexai/index.d.ts.map +1 -0
- package/lib/{esm/vertexai → vertexai}/index.js +221 -159
- package/lib/vertexai/index.js.map +1 -0
- package/lib/vertexai/models/claude.d.ts +16 -0
- package/lib/vertexai/models/claude.d.ts.map +1 -0
- package/lib/vertexai/models/claude.js +65 -0
- package/lib/vertexai/models/claude.js.map +1 -0
- package/lib/{types/vertexai → vertexai}/models/gemini.d.ts +29 -5
- package/lib/vertexai/models/gemini.d.ts.map +1 -0
- package/lib/{esm/vertexai → vertexai}/models/gemini.js +255 -127
- package/lib/vertexai/models/gemini.js.map +1 -0
- package/lib/{types/vertexai → vertexai}/models/imagen.d.ts +12 -11
- package/lib/vertexai/models/imagen.d.ts.map +1 -0
- package/lib/{esm/vertexai → vertexai}/models/imagen.js +74 -47
- package/lib/vertexai/models/imagen.js.map +1 -0
- package/lib/{types/vertexai → vertexai}/models/llama.d.ts +4 -4
- package/lib/vertexai/models/llama.d.ts.map +1 -0
- package/lib/{esm/vertexai → vertexai}/models/llama.js +37 -33
- package/lib/vertexai/models/llama.js.map +1 -0
- package/lib/vertexai/models.d.ts +21 -0
- package/lib/vertexai/models.d.ts.map +1 -0
- package/lib/{esm/vertexai → vertexai}/models.js +12 -9
- package/lib/vertexai/models.js.map +1 -0
- package/lib/{types/watsonx → watsonx}/index.d.ts +4 -3
- package/lib/watsonx/index.d.ts.map +1 -0
- package/lib/{esm/watsonx → watsonx}/index.js +69 -41
- package/lib/watsonx/index.js.map +1 -0
- package/lib/watsonx/interfaces.d.ts.map +1 -0
- package/lib/{esm/watsonx → watsonx}/interfaces.js.map +1 -1
- package/lib/{types/xai → xai}/index.d.ts +4 -4
- package/lib/xai/index.d.ts.map +1 -0
- package/lib/{esm/xai → xai}/index.js +22 -18
- package/lib/xai/index.js.map +1 -0
- package/package.json +88 -36
- package/src/adobe/firefly.ts +60 -43
- package/src/anthropic/index.ts +121 -0
- package/src/azure/azure_foundry.ts +210 -165
- package/src/bedrock/converse.ts +124 -101
- package/src/bedrock/embeddings.test.ts +508 -0
- package/src/bedrock/embeddings.ts +517 -0
- package/src/bedrock/error-handling.test.ts +62 -14
- package/src/bedrock/index.ts +832 -466
- package/src/bedrock/nova-image-payload.ts +98 -91
- package/src/bedrock/payloads.ts +9 -9
- package/src/bedrock/s3.test.ts +39 -0
- package/src/bedrock/s3.ts +57 -38
- package/src/bedrock/streaming-tool-use.test.ts +86 -69
- package/src/bedrock/twelvelabs.ts +35 -32
- package/src/groq/index.ts +143 -60
- package/src/http-timeout-wiring.test.ts +263 -0
- package/src/huggingface_ie.ts +45 -61
- package/src/index.ts +15 -15
- package/src/mistral/index.ts +97 -48
- package/src/mistral/types.ts +13 -12
- package/src/openai/azure_openai.ts +35 -27
- package/src/openai/drivers.ts +9 -0
- package/src/openai/error-handling.test.ts +308 -113
- package/src/openai/index.ts +389 -254
- package/src/openai/openai.ts +6 -13
- package/src/openai/openai_compatible.ts +35 -25
- package/src/openai/openai_format.ts +90 -40
- package/src/replicate.ts +108 -86
- package/src/shared/claude-messages.test.ts +37 -0
- package/src/shared/claude-messages.ts +1085 -0
- package/src/shared/claude-thinking.ts +6 -12
- package/src/shared/debug-prompt.ts +28 -0
- package/src/test-driver/TestErrorCompletionStream.ts +9 -10
- package/src/test-driver/TestValidationErrorCompletionStream.ts +9 -10
- package/src/test-driver/index.ts +43 -27
- package/src/test-driver/utils.ts +8 -8
- package/src/togetherai/index.test.ts +32 -0
- package/src/togetherai/index.ts +64 -128
- package/src/togetherai/interfaces.ts +5 -5
- package/src/vertexai/debug.ts +3 -3
- package/src/vertexai/embeddings/embed-legacy-multimodal.ts +211 -0
- package/src/vertexai/embeddings/embed.ts +206 -0
- package/src/vertexai/embeddings/embeddings.test.ts +497 -0
- package/src/vertexai/embeddings/source-utils.test.ts +54 -0
- package/src/vertexai/embeddings/source-utils.ts +29 -0
- package/src/vertexai/index.ts +339 -253
- package/src/vertexai/models/claude-error-handling.test.ts +86 -39
- package/src/vertexai/models/claude-streaming-spacing.test.ts +82 -75
- package/src/vertexai/models/claude.ts +84 -1041
- package/src/vertexai/models/gemini-conversation-mutation.test.ts +65 -23
- package/src/vertexai/models/gemini-error-handling.test.ts +126 -10
- package/src/vertexai/models/gemini-orphaned-tool-result.test.ts +118 -0
- package/src/vertexai/models/gemini.ts +372 -182
- package/src/vertexai/models/imagen.ts +164 -100
- package/src/vertexai/models/llama.ts +70 -48
- package/src/vertexai/models.ts +36 -16
- package/src/watsonx/index.ts +101 -65
- package/src/watsonx/interfaces.ts +10 -15
- package/src/xai/index.ts +30 -30
- package/lib/cjs/adobe/firefly.js +0 -120
- package/lib/cjs/adobe/firefly.js.map +0 -1
- package/lib/cjs/azure/azure_foundry.js +0 -432
- package/lib/cjs/azure/azure_foundry.js.map +0 -1
- package/lib/cjs/bedrock/converse.js +0 -359
- package/lib/cjs/bedrock/converse.js.map +0 -1
- package/lib/cjs/bedrock/index.js +0 -1508
- package/lib/cjs/bedrock/index.js.map +0 -1
- package/lib/cjs/bedrock/nova-image-payload.js +0 -207
- package/lib/cjs/bedrock/nova-image-payload.js.map +0 -1
- package/lib/cjs/bedrock/payloads.js +0 -3
- package/lib/cjs/bedrock/s3.js +0 -107
- package/lib/cjs/bedrock/s3.js.map +0 -1
- package/lib/cjs/bedrock/twelvelabs.js +0 -87
- package/lib/cjs/bedrock/twelvelabs.js.map +0 -1
- package/lib/cjs/groq/index.js +0 -328
- package/lib/cjs/groq/index.js.map +0 -1
- package/lib/cjs/huggingface_ie.js +0 -201
- package/lib/cjs/huggingface_ie.js.map +0 -1
- package/lib/cjs/index.js +0 -31
- package/lib/cjs/index.js.map +0 -1
- package/lib/cjs/mistral/index.js +0 -176
- package/lib/cjs/mistral/index.js.map +0 -1
- package/lib/cjs/mistral/types.js +0 -83
- package/lib/cjs/mistral/types.js.map +0 -1
- package/lib/cjs/openai/azure_openai.js +0 -72
- package/lib/cjs/openai/azure_openai.js.map +0 -1
- package/lib/cjs/openai/index.js +0 -1129
- package/lib/cjs/openai/index.js.map +0 -1
- package/lib/cjs/openai/openai.js +0 -21
- package/lib/cjs/openai/openai.js.map +0 -1
- package/lib/cjs/openai/openai_compatible.js +0 -63
- package/lib/cjs/openai/openai_compatible.js.map +0 -1
- package/lib/cjs/openai/openai_format.js +0 -131
- package/lib/cjs/openai/openai_format.js.map +0 -1
- package/lib/cjs/package.json +0 -3
- package/lib/cjs/replicate.js +0 -275
- package/lib/cjs/replicate.js.map +0 -1
- package/lib/cjs/shared/claude-thinking.js +0 -60
- package/lib/cjs/shared/claude-thinking.js.map +0 -1
- package/lib/cjs/test-driver/TestErrorCompletionStream.js +0 -20
- package/lib/cjs/test-driver/TestErrorCompletionStream.js.map +0 -1
- package/lib/cjs/test-driver/TestValidationErrorCompletionStream.js +0 -24
- package/lib/cjs/test-driver/TestValidationErrorCompletionStream.js.map +0 -1
- package/lib/cjs/test-driver/index.js +0 -109
- package/lib/cjs/test-driver/index.js.map +0 -1
- package/lib/cjs/test-driver/utils.js +0 -30
- package/lib/cjs/test-driver/utils.js.map +0 -1
- package/lib/cjs/togetherai/index.js +0 -126
- package/lib/cjs/togetherai/index.js.map +0 -1
- package/lib/cjs/togetherai/interfaces.js +0 -3
- package/lib/cjs/togetherai/interfaces.js.map +0 -1
- package/lib/cjs/vertexai/debug.js +0 -12
- package/lib/cjs/vertexai/debug.js.map +0 -1
- package/lib/cjs/vertexai/embeddings/embeddings-image.js +0 -27
- package/lib/cjs/vertexai/embeddings/embeddings-image.js.map +0 -1
- package/lib/cjs/vertexai/embeddings/embeddings-text.js +0 -23
- package/lib/cjs/vertexai/embeddings/embeddings-text.js.map +0 -1
- package/lib/cjs/vertexai/index.js +0 -644
- package/lib/cjs/vertexai/index.js.map +0 -1
- package/lib/cjs/vertexai/models/claude.js +0 -917
- package/lib/cjs/vertexai/models/claude.js.map +0 -1
- package/lib/cjs/vertexai/models/gemini.js +0 -858
- package/lib/cjs/vertexai/models/gemini.js.map +0 -1
- package/lib/cjs/vertexai/models/imagen.js +0 -303
- package/lib/cjs/vertexai/models/imagen.js.map +0 -1
- package/lib/cjs/vertexai/models/llama.js +0 -183
- package/lib/cjs/vertexai/models/llama.js.map +0 -1
- package/lib/cjs/vertexai/models.js +0 -35
- package/lib/cjs/vertexai/models.js.map +0 -1
- package/lib/cjs/watsonx/index.js +0 -161
- package/lib/cjs/watsonx/index.js.map +0 -1
- package/lib/cjs/watsonx/interfaces.js +0 -3
- package/lib/cjs/xai/index.js +0 -65
- package/lib/cjs/xai/index.js.map +0 -1
- package/lib/esm/adobe/firefly.js +0 -116
- package/lib/esm/adobe/firefly.js.map +0 -1
- package/lib/esm/azure/azure_foundry.js.map +0 -1
- package/lib/esm/bedrock/converse.js.map +0 -1
- package/lib/esm/bedrock/index.js.map +0 -1
- package/lib/esm/bedrock/nova-image-payload.js.map +0 -1
- package/lib/esm/bedrock/payloads.js.map +0 -1
- package/lib/esm/bedrock/s3.js.map +0 -1
- package/lib/esm/bedrock/twelvelabs.js.map +0 -1
- package/lib/esm/groq/index.js.map +0 -1
- package/lib/esm/huggingface_ie.js.map +0 -1
- package/lib/esm/index.js +0 -15
- package/lib/esm/index.js.map +0 -1
- package/lib/esm/mistral/index.js.map +0 -1
- package/lib/esm/mistral/types.js.map +0 -1
- package/lib/esm/openai/azure_openai.js.map +0 -1
- package/lib/esm/openai/index.js.map +0 -1
- package/lib/esm/openai/openai.js +0 -14
- package/lib/esm/openai/openai.js.map +0 -1
- package/lib/esm/openai/openai_compatible.js.map +0 -1
- package/lib/esm/openai/openai_format.js.map +0 -1
- package/lib/esm/replicate.js.map +0 -1
- package/lib/esm/shared/claude-thinking.js.map +0 -1
- package/lib/esm/test-driver/TestErrorCompletionStream.js +0 -16
- package/lib/esm/test-driver/TestErrorCompletionStream.js.map +0 -1
- package/lib/esm/test-driver/TestValidationErrorCompletionStream.js.map +0 -1
- package/lib/esm/test-driver/index.js.map +0 -1
- package/lib/esm/test-driver/utils.js.map +0 -1
- package/lib/esm/togetherai/index.js +0 -122
- package/lib/esm/togetherai/index.js.map +0 -1
- package/lib/esm/togetherai/interfaces.js.map +0 -1
- package/lib/esm/vertexai/debug.js.map +0 -1
- package/lib/esm/vertexai/embeddings/embeddings-image.js +0 -24
- package/lib/esm/vertexai/embeddings/embeddings-image.js.map +0 -1
- package/lib/esm/vertexai/embeddings/embeddings-text.js +0 -20
- package/lib/esm/vertexai/embeddings/embeddings-text.js.map +0 -1
- package/lib/esm/vertexai/index.js.map +0 -1
- package/lib/esm/vertexai/models/claude.js +0 -906
- package/lib/esm/vertexai/models/claude.js.map +0 -1
- package/lib/esm/vertexai/models/gemini.js.map +0 -1
- package/lib/esm/vertexai/models/imagen.js.map +0 -1
- package/lib/esm/vertexai/models/llama.js.map +0 -1
- package/lib/esm/vertexai/models.js.map +0 -1
- package/lib/esm/watsonx/index.js.map +0 -1
- package/lib/esm/xai/index.js.map +0 -1
- package/lib/types/adobe/firefly.d.ts.map +0 -1
- package/lib/types/azure/azure_foundry.d.ts.map +0 -1
- package/lib/types/bedrock/converse.d.ts.map +0 -1
- package/lib/types/bedrock/index.d.ts.map +0 -1
- package/lib/types/bedrock/nova-image-payload.d.ts.map +0 -1
- package/lib/types/bedrock/payloads.d.ts.map +0 -1
- package/lib/types/bedrock/s3.d.ts.map +0 -1
- package/lib/types/bedrock/twelvelabs.d.ts.map +0 -1
- package/lib/types/groq/index.d.ts.map +0 -1
- package/lib/types/huggingface_ie.d.ts.map +0 -1
- package/lib/types/index.d.ts +0 -15
- package/lib/types/index.d.ts.map +0 -1
- package/lib/types/mistral/index.d.ts.map +0 -1
- package/lib/types/mistral/types.d.ts.map +0 -1
- package/lib/types/openai/azure_openai.d.ts.map +0 -1
- package/lib/types/openai/index.d.ts.map +0 -1
- package/lib/types/openai/openai.d.ts.map +0 -1
- package/lib/types/openai/openai_compatible.d.ts.map +0 -1
- package/lib/types/openai/openai_format.d.ts.map +0 -1
- package/lib/types/replicate.d.ts.map +0 -1
- package/lib/types/shared/claude-thinking.d.ts.map +0 -1
- package/lib/types/test-driver/TestErrorCompletionStream.d.ts.map +0 -1
- package/lib/types/test-driver/TestValidationErrorCompletionStream.d.ts.map +0 -1
- package/lib/types/test-driver/index.d.ts.map +0 -1
- package/lib/types/test-driver/utils.d.ts.map +0 -1
- package/lib/types/togetherai/index.d.ts +0 -23
- package/lib/types/togetherai/index.d.ts.map +0 -1
- package/lib/types/togetherai/interfaces.d.ts.map +0 -1
- package/lib/types/vertexai/debug.d.ts +0 -2
- package/lib/types/vertexai/debug.d.ts.map +0 -1
- package/lib/types/vertexai/embeddings/embeddings-image.d.ts +0 -11
- package/lib/types/vertexai/embeddings/embeddings-image.d.ts.map +0 -1
- package/lib/types/vertexai/embeddings/embeddings-text.d.ts +0 -10
- package/lib/types/vertexai/embeddings/embeddings-text.d.ts.map +0 -1
- package/lib/types/vertexai/index.d.ts.map +0 -1
- package/lib/types/vertexai/models/claude.d.ts +0 -119
- package/lib/types/vertexai/models/claude.d.ts.map +0 -1
- package/lib/types/vertexai/models/gemini.d.ts.map +0 -1
- package/lib/types/vertexai/models/imagen.d.ts.map +0 -1
- package/lib/types/vertexai/models/llama.d.ts.map +0 -1
- package/lib/types/vertexai/models.d.ts +0 -20
- package/lib/types/vertexai/models.d.ts.map +0 -1
- package/lib/types/watsonx/index.d.ts.map +0 -1
- package/lib/types/watsonx/interfaces.d.ts.map +0 -1
- package/lib/types/xai/index.d.ts.map +0 -1
- package/src/vertexai/embeddings/embeddings-image.ts +0 -52
- package/src/vertexai/embeddings/embeddings-text.ts +0 -54
- /package/lib/{esm/bedrock → bedrock}/payloads.js +0 -0
- /package/lib/{esm/mistral → mistral}/types.js +0 -0
- /package/lib/{esm/togetherai → togetherai}/interfaces.js +0 -0
- /package/lib/{types/watsonx → watsonx}/interfaces.d.ts +0 -0
- /package/lib/{esm/watsonx → watsonx}/interfaces.js +0 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../../src/vertexai/models/gemini.ts"],"names":[],"mappings":"AACA,OAAO,EAEH,YAAY,EACZ,yBAAyB,EAKzB,kBAAkB,EAClB,YAAY,EACZ,QAAQ,EAER,eAAe,EAGf,aAAa,GAEhB,MAAM,eAAe,CAAC;AACvB,OAAO,EAOH,mBAAmB,EACnB,qBAAqB,EACrB,yBAAyB,EACzB,uBAAuB,EAEvB,eAAe,EAEf,SAAS,EAET,UAAU,EAEV,kBAAkB,EAElB,iCAAiC,EACjC,+BAA+B,EAG/B,+BAA+B,EAC/B,uBAAuB,GAE1B,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AAClD,OAAO,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AAMtE,SAAS,wBAAwB,CAAC,OAAsB;IACpD,+DAA+D;IAC/D,OAAO,CAAC,CAAC,OAAO,CAAC,aAAa,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;AACvE,CAAC;AAED,MAAM,oBAAoB,GAAoB;IAC1C;QACI,QAAQ,EAAE,YAAY,CAAC,+BAA+B;QACtD,SAAS,EAAE,kBAAkB,CAAC,eAAe;KAChD;IACD;QACI,QAAQ,EAAE,YAAY,CAAC,wBAAwB;QAC/C,SAAS,EAAE,kBAAkB,CAAC,eAAe;KAChD;IACD;QACI,QAAQ,EAAE,YAAY,CAAC,+BAA+B;QACtD,SAAS,EAAE,kBAAkB,CAAC,eAAe;KAChD;IACD;QACI,QAAQ,EAAE,YAAY,CAAC,yBAAyB;QAChD,SAAS,EAAE,kBAAkB,CAAC,eAAe;KAChD;IACD;QACI,QAAQ,EAAE,YAAY,CAAC,yBAAyB;QAChD,SAAS,EAAE,kBAAkB,CAAC,eAAe;KAChD;IACD;QACI,QAAQ,EAAE,YAAY,CAAC,6BAA6B;QACpD,SAAS,EAAE,kBAAkB,CAAC,eAAe;KAChD;CACJ,CAAC;AAEF,SAAS,2BAA2B,CAAC,OAAgB;IACjD,OAAO;QACH,GAAG,OAAO;QACV,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YAC/B,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC;gBACzB,OAAO,IAAI,CAAC;YAChB,CAAC;YACD,OAAO;gBACH,GAAG,IAAI;gBACP,UAAU,EAAE;oBACR,GAAG,IAAI,CAAC,UAAU;oBAClB,IAAI,EAAE,sBAAsB,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;iBACrD;aACW,CAAC;QACrB,CAAC,CAAC;KACL,CAAC;AACN,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,MAA6B;IACjE,OAAO;QACH,GAAG,MAAM;QACT,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,2BAA2B,CAAC;QAC1D,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,2BAA2B,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS;KACjF,CAAC;AACN,CAAC;AAED,kGAAkG;AAClG,SAAS,wBAAwB,CAC7B,eAAsG;IAEtG,QAAQ,eAAe,EAAE,CAAC;QACtB,KAAK,wBAAwB;YACzB,OAAO,eAAe,CAAC,sBAAsB,CAAC;QAClD,KAAK,wBAAwB;YACzB,OAAO,eAAe,CAAC,sBAAsB,CAAC;QAClD,KAAK,8BAA8B;YAC/B,OAAO,eAAe,CAAC,4BAA4B,CAAC;QACxD;YACI,OAAO,SAAS,CAAC;IACzB,CAAC;AACL,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,OAAyB,EAAE,MAA6B;IACrF,MAAM,aAAa,GAAG,OAAO,CAAC,aAAkD,CAAC;IACjF,MAAM,KAAK,GAAG,kBAAkB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAEhD,2FAA2F;IAC3F,6EAA6E;IAC7E,2EAA2E;IAC3E,IAAI,eAAe,GAAG,oBAAoB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC5D,IAAI,CAAC,KAAK,IAAI,eAAe,EAAE,CAAC;QAC5B,MAAM,YAAY,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,IAAI,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC;QAC7G,IAAI,YAAY,EAAE,CAAC;YACf,eAAe,GAAG,gCAAgC,CAAC,eAAe,CAAC,CAAC;QACxE,CAAC;IACL,CAAC;IACD,6EAA6E;IAC7E,6EAA6E;IAC7E,IAAI,eAAe,EAAE,CAAC;QAClB,eAAe,GAAG,sBAAsB,CAAC,eAAe,CAAC,CAAC;IAC9D,CAAC;IAED,MAAM,mBAAmB,GAAG,wBAAwB,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC;IAExE,MAAM,gBAAgB,GAA0B;QAC5C,iBAAiB,EAAE,MAAM,CAAC,MAAM;QAChC,cAAc,EAAE,oBAAoB;QACpC,kBAAkB,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,KAAK,CAAC,EAAE,2EAA2E;QAChI,cAAc,EAAE,CAAC;QACjB,eAAe;QACf,WAAW,EAAE,aAAa,EAAE,WAAW;QACvC,IAAI,EAAE,aAAa,EAAE,KAAK;QAC1B,eAAe,EAAE,aAAa,EAAE,UAAU;QAC1C,aAAa,EAAE,aAAa,EAAE,aAAa;QAC3C,cAAc,EAAE,oBAAoB,CAAC,OAAO,CAAC;QAC7C,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,WAAW,EAAE;YACT,SAAS,EAAE,aAAa,EAAE,UAAU;YACpC,WAAW,EAAE,aAAa,EAAE,kBAAkB;YAC9C,gBAAgB,EAAE,aAAa,EAAE,iBAAiB;YAClD,eAAe,EAAE,wBAAwB,CAAC,aAAa,EAAE,gBAAgB,CAAC;YAC1E,cAAc,EAAE,aAAa,EAAE,gBAAgB;YAC/C,wBAAwB,EAAE,aAAa,EAAE,0BAA0B;SACtE;KACJ,CAAC;IAEF,MAAM,MAAM,GAA0B;QAClC,iBAAiB,EAAE,MAAM,CAAC,MAAM;QAChC,cAAc,EAAE,oBAAoB;QACpC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS;QAClC,UAAU,EAAE,KAAK;YACb,CAAC,CAAC;gBACI,qBAAqB,EAAE;oBACnB,IAAI,EAAE,yBAAyB,CAAC,IAAI;iBACvC;aACJ;YACH,CAAC,CAAC,SAAS;QACf,cAAc,EAAE,CAAC;QACjB,wBAAwB;QACxB,gBAAgB,EAAE,mBAAmB,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,SAAS;QACtE,kBAAkB,EAAE,mBAAmB,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS;QAC3E,eAAe;QACf,WAAW,EAAE,aAAa,EAAE,WAAW;QACvC,IAAI,EAAE,aAAa,EAAE,KAAK;QAC1B,IAAI,EAAE,aAAa,EAAE,KAAK;QAC1B,eAAe,EAAE,aAAa,EAAE,UAAU;QAC1C,aAAa,EAAE,aAAa,EAAE,aAAa;QAC3C,eAAe,EAAE,aAAa,EAAE,gBAAgB;QAChD,gBAAgB,EAAE,aAAa,EAAE,iBAAiB;QAClD,IAAI,EAAE,aAAa,EAAE,IAAI;QACzB,cAAc,EAAE,oBAAoB,CAAC,OAAO,CAAC;QAC7C,MAAM,EAAE,OAAO,CAAC,MAAM;KACzB,CAAC;IAEF,OAAO;QACH,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,QAAQ,EAAE,eAAe;QACzB,MAAM,EAAE,OAAO,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,MAAM;KACpF,CAAC;AACN,CAAC;AAED;;;GAGG;AACH,SAAS,wBAAwB,CAAC,OAAgB;IAC9C,MAAM,OAAO,GAAuB,EAAE,CAAC;IACvC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;IAC5B,IAAI,KAAK,EAAE,CAAC;QACR,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACvB,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACZ,OAAO,CAAC,IAAI,CAAC;oBACT,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,IAAI,CAAC,IAAI;iBACnB,CAAC,CAAC;YACP,CAAC;iBAAM,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBACzB,MAAM,gBAAgB,GAAW,IAAI,CAAC,UAAU,CAAC,IAAI,IAAI,EAAE,CAAC;gBAC5D,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,IAAI,WAAW,CAAC;gBACzD,MAAM,QAAQ,GAAG,QAAQ,QAAQ,WAAW,gBAAgB,EAAE,CAAC;gBAC/D,OAAO,CAAC,IAAI,CAAC;oBACT,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,QAAQ;iBAClB,CAAC,CAAC;YACP,CAAC;QACL,CAAC;IACL,CAAC;IACD,OAAO,OAAO,CAAC;AACnB,CAAC;AAED,SAAS,mBAAmB,CAAC,OAAgB;IACzC,MAAM,GAAG,GAAc,EAAE,CAAC;IAC1B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC;IAClC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACvB,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACpB,MAAM,OAAO,GAAY;gBACrB,EAAE,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,IAAI,EAAE;gBAChC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,IAAI,EAAE;gBACvC,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,IAAkB;aACnD,CAAC;YACF,mEAAmE;YACnE,sDAAsD;YACtD,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACxB,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC,gBAAgB,CAAC;YACtD,CAAC;YACD,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACtB,CAAC;IACL,CAAC;IACD,OAAO,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC;AAC5C,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,QAA+B;IAChE,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAElD,MAAM,YAAY,GAAG,QAAQ,CAAC,IAAI,CAC9B,CAAC,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CACnF,CAAC;IACF,8CAA8C;IAC9C,IAAI,CAAC,YAAY,EAAE,CAAC;QAChB,OAAO,QAAQ,CAAC;IACpB,CAAC;IAED,MAAM,MAAM,GAAc,EAAE,CAAC;IAC7B,IAAI,cAAc,GAAG,EAAE,GAAG,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;IAE/E,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACvC,IAAI,cAAc,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YAC3C,mEAAmE;YACnE,cAAc,CAAC,KAAK,GAAG,CAAC,cAAc,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC;QAC7F,CAAC;aAAM,CAAC;YACJ,8CAA8C;YAC9C,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC5B,cAAc,GAAG,EAAE,GAAG,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;QAC/E,CAAC;IACL,CAAC;IAED,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAC5B,OAAO,MAAM,CAAC;AAClB,CAAC;AAED;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,sBAAsB,CAAC,QAAmB;IACtD,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,QAAQ,CAAC;IAC3C,MAAM,MAAM,GAAc,EAAE,CAAC;IAC7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACvC,MAAM,OAAO,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;QAC5B,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YAC5C,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrB,SAAS;QACb,CAAC;QACD,MAAM,mBAAmB,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAChF,IAAI,CAAC,mBAAmB,EAAE,CAAC;YACvB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrB,SAAS;QACb,CAAC;QACD,MAAM,IAAI,GAAG,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAC7B,MAAM,YAAY,GAAG,IAAI,GAAG,EAAU,CAAC;QACvC,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC9C,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBAC5B,IAAI,IAAI,CAAC,YAAY,EAAE,IAAI;oBAAE,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAC1E,CAAC;QACL,CAAC;QACD,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAC3C,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CACpF,CAAC;QACF,mEAAmE;QACnE,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,SAAS;QACpC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;IACtG,CAAC;IACD,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,MAAM,sBAAsB,GAAmB;IAC3C,YAAY,CAAC,UAAU;IACvB,YAAY,CAAC,IAAI;IACjB,YAAY,CAAC,yBAAyB;CACzC,CAAC;AAEF,mFAAmF;AACnF,6EAA6E;AAC7E,4DAA4D;AAC5D,MAAM,0BAA0B,GAAG;IAC/B,sBAAsB,EAAE,kCAAkC;CAC7D,CAAC;AAEF,SAAS,oBAAoB,CAAC,MAAiC;IAC3D,MAAM,aAAa,GAAG,MAAM,CAAC,aAAkD,CAAC;IAChF,gFAAgF;IAChF,IAAI,aAAa,EAAE,sBAAsB,KAAK,SAAS,EAAE,CAAC;QACtD,OAAO,aAAa,CAAC,sBAAsB,CAAC;IAChD,CAAC;IACD,IAAI,aAAa,EAAE,MAAM,EAAE,CAAC;QACxB,OAAO,qBAAqB,CAAC,MAAM,CAAC,KAAK,EAAE,aAAa,CAAC,MAAM,CAAC,CAAC;IACrE,CAAC;IACD,yCAAyC;IACzC,kEAAkE;IAClE,IAAI,qBAAqB,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,KAAK,EAAE,CAAC;QAChD,IAAI,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,GAAG,CAAC;QACf,CAAC;QACD,OAAO,CAAC,CAAC;IACb,CAAC;IACD,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,SAAS,4BAA4B,CACjC,KAAa,EACb,MAAuC;IAEvC,IAAI,KAAK,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;QACvC,OAAO,aAAa,CAAC,IAAI,CAAC;IAC9B,CAAC;IACD,IAAI,KAAK,CAAC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,CAAC;QAC3C,OAAO,MAAM,KAAK,KAAK,CAAC,CAAC,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,aAAa,CAAC,IAAI,CAAC;IACzE,CAAC;IACD,QAAQ,MAAM,EAAE,CAAC;QACb,KAAK,KAAK;YACN,OAAO,aAAa,CAAC,GAAG,CAAC;QAC7B,KAAK,QAAQ;YACT,OAAO,aAAa,CAAC,MAAM,CAAC;QAChC,KAAK,MAAM;YACP,OAAO,aAAa,CAAC,IAAI,CAAC;QAC9B;YACI,OAAO,SAAS,CAAC;IACzB,CAAC;AACL,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAa,EAAE,MAAoD;IAC9F,MAAM,WAAW,GAAG,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;IACjD,MAAM,OAAO,GAAG,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC;IACxD,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAEpC,IAAI,MAAM,KAAK,KAAK,EAAE,CAAC;QACnB,IAAI,KAAK;YAAE,OAAO,GAAG,CAAC;QACtB,IAAI,WAAW;YAAE,OAAO,GAAG,CAAC;QAC5B,IAAI,OAAO;YAAE,OAAO,CAAC,CAAC;QACtB,OAAO,IAAI,CAAC;IAChB,CAAC;IACD,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;QACtB,OAAO,IAAI,CAAC;IAChB,CAAC;IACD,IAAI,KAAK;QAAE,OAAO,KAAK,CAAC;IACxB,IAAI,OAAO,IAAI,WAAW;QAAE,OAAO,KAAK,CAAC;IACzC,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,SAAS,oBAAoB,CAAC,MAAiC;IAC3D,MAAM,aAAa,GAAG,MAAM,CAAC,aAAkD,CAAC;IAEhF,6EAA6E;IAC7E,MAAM,gBAAgB,GAAG,aAAa,EAAE,gBAAgB,IAAI,KAAK,CAAC;IAClE,IAAI,aAAa,EAAE,sBAAsB,KAAK,SAAS,IAAI,aAAa,EAAE,cAAc,EAAE,CAAC;QACvF,OAAO;YACH,eAAe,EAAE,gBAAgB;YACjC,cAAc,EAAE,aAAa,CAAC,sBAAsB;YACpD,aAAa,EAAE,aAAa,CAAC,cAAc;SAC9C,CAAC;IACN,CAAC;IACD,IAAI,aAAa,EAAE,MAAM,EAAE,CAAC;QACxB,IAAI,uBAAuB,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC;YAC/C,OAAO;gBACH,eAAe,EAAE,gBAAgB;gBACjC,aAAa,EAAE,4BAA4B,CAAC,MAAM,CAAC,KAAK,EAAE,aAAa,CAAC,MAAM,CAAC;aAClF,CAAC;QACN,CAAC;QACD,OAAO;YACH,eAAe,EAAE,gBAAgB;YACjC,cAAc,EAAE,qBAAqB,CAAC,MAAM,CAAC,KAAK,EAAE,aAAa,CAAC,MAAM,CAAC;SAC5E,CAAC;IACN,CAAC;IAED,uCAAuC;IACvC,kEAAkE;IAClE,sEAAsE;IACtE,IAAI,uBAAuB,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC;QAC/C,OAAO;YACH,eAAe,EAAE,gBAAgB;YACjC,aAAa,EAAE,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC,GAAG;SACtG,CAAC;IACN,CAAC;IACD,IAAI,uBAAuB,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC;QAC/C,MAAM,sBAAsB,GAAG,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACjE,OAAO;YACH,eAAe,EAAE,gBAAgB;YACjC,cAAc,EAAE,sBAAsB;SACzC,CAAC;IACN,CAAC;AACL,CAAC;AAED,MAAM,OAAO,qBAAqB;IAC9B,KAAK,CAAU;IAEf,YAAY,OAAe;QACvB,IAAI,CAAC,KAAK,GAAG;YACT,EAAE,EAAE,OAAO;YACX,IAAI,EAAE,OAAO;YACb,QAAQ,EAAE,UAAU;YACpB,IAAI,EAAE,SAAS,CAAC,IAAI;YACpB,UAAU,EAAE,IAAI;SACD,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,YAAY,CACd,OAAuB,EACvB,QAAyB,EACzB,OAAyB;QAEzB,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACxC,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC5C,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QAE3C,MAAM,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;QACrC,IAAI,QAAQ,GAAc,EAAE,CAAC;QAC7B,IAAI,MAAM,GAAwB,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,2CAA2C;QAE1G,MAAM,MAAM,GAAc,EAAE,CAAC;QAE7B,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YACzB,yBAAyB;YACzB,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC,MAAM,EAAE,CAAC;gBACjC,gCAAgC;gBAChC,IAAI,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACpC,MAAM,IAAI,KAAK,CACX,6FAA6F,CAChG,CAAC;gBACN,CAAC;gBAED,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;oBACd,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC;wBACf,IAAI,EAAE,GAAG,CAAC,OAAO;qBACpB,CAAC,CAAC;gBACP,CAAC;YACL,CAAC;iBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC,IAAI,EAAE,CAAC;gBACtC,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;oBACnB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;gBACzD,CAAC;gBACD,yFAAyF;gBACzF,MAAM,oBAAoB,GAAS;oBAC/B,gBAAgB,EAAE;wBACd,IAAI,EAAE,GAAG,CAAC,WAAW;wBACrB,QAAQ,EAAE,sBAAsB,CAAC,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC;qBACtD;oBACD,wFAAwF;oBACxF,gBAAgB,EAAE,GAAG,CAAC,iBAAiB;iBAC1C,CAAC;gBACF,QAAQ,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,CAAC,oBAAoB,CAAC;iBAChC,CAAC,CAAC;YACP,CAAC;iBAAM,CAAC;gBACJ,2DAA2D;gBAC3D,MAAM,KAAK,GAAW,EAAE,CAAC;gBACzB,wBAAwB;gBACxB,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;oBACd,KAAK,CAAC,IAAI,CAAC;wBACP,IAAI,EAAE,GAAG,CAAC,OAAO;qBACpB,CAAC,CAAC;gBACP,CAAC;gBAED,wBAAwB;gBACxB,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;oBACZ,KAAK,MAAM,CAAC,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;wBACxB,MAAM,OAAO,GAAG,MAAM,CAAC,CAAC,MAAM,EAAE,CAAC;wBACjC,MAAM,OAAO,GACT,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,iCAAiC,CAAC,CAAC;wBAEzF,IAAI,OAAO,EAAE,CAAC;4BACV,KAAK,CAAC,IAAI,CAAC;gCACP,QAAQ,EAAE;oCACN,OAAO;oCACP,QAAQ,EAAE,CAAC,CAAC,SAAS;iCACxB;6BACJ,CAAC,CAAC;wBACP,CAAC;6BAAM,CAAC;4BACJ,uBAAuB;4BACvB,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,SAAS,EAAE,CAAC;4BACnC,MAAM,IAAI,GAAG,MAAM,kBAAkB,CAAC,MAAM,CAAC,CAAC;4BAC9C,KAAK,CAAC,IAAI,CAAC;gCACP,UAAU,EAAE;oCACR,IAAI;oCACJ,QAAQ,EAAE,CAAC,CAAC,SAAS;iCACxB;6BACJ,CAAC,CAAC;wBACP,CAAC;oBACL,CAAC;gBACL,CAAC;gBAED,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACnB,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC,MAAM,EAAE,CAAC;wBACjC,MAAM,CAAC,IAAI,CAAC;4BACR,IAAI,EAAE,MAAM;4BACZ,KAAK;yBACR,CAAC,CAAC;oBACP,CAAC;yBAAM,CAAC;wBACJ,QAAQ,CAAC,IAAI,CAAC;4BACV,IAAI,EAAE,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;4BAC1D,KAAK;yBACR,CAAC,CAAC;oBACP,CAAC;gBACL,CAAC;YACL,CAAC;QACL,CAAC;QAED,uCAAuC;QACvC,IAAI,MAAM,EAAE,CAAC;YACT,IAAI,wBAAwB,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;gBACtD,6FAA6F;gBAC7F,qFAAqF;gBACrF,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,IAAI,EAAE,iDAAiD,EAAE,CAAC,CAAC;YACpF,CAAC;iBAAM,CAAC;gBACJ,6FAA6F;gBAC7F,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;oBAChB,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC;wBACf,IAAI,EAAE,8FAA8F,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE;qBAC/H,CAAC,CAAC;gBACP,CAAC;qBAAM,CAAC;oBACJ,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC;wBACf,IAAI,EAAE,sEAAsE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE;qBACvG,CAAC,CAAC;gBACP,CAAC;YACL,CAAC;QACL,CAAC;QAED,kDAAkD;QAClD,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7C,MAAM,GAAG,SAAS,CAAC;QACvB,CAAC;QAED,qGAAqG;QACrG,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpB,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACvC,CAAC;QAED,wIAAwI;QACxI,QAAQ,GAAG,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QAE1C,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;IAChC,CAAC;IAED,yBAAyB,CACrB,MAAsB,EACtB,aAA+D;QAE/D,IAAI,CAAC,aAAa,EAAE,eAAe,EAAE,CAAC;YAClC,OAAO,EAAE,CAAC;QACd,CAAC;QACD,MAAM,UAAU,GAAwB;YACpC,KAAK,EAAE,aAAa,CAAC,eAAe;YACpC,MAAM,EAAE,aAAa,CAAC,gBAAgB;YACtC,aAAa,EAAE,aAAa,CAAC,uBAAuB,IAAI,SAAS;YACjE,UAAU,EAAE,CAAC,aAAa,CAAC,gBAAgB,IAAI,CAAC,CAAC,GAAG,CAAC,aAAa,CAAC,uBAAuB,IAAI,CAAC,CAAC;SACnG,CAAC;QAEF,sBAAsB;QACtB,UAAU,CAAC,MAAM;YACb,CAAC,aAAa,CAAC,oBAAoB,IAAI,CAAC,CAAC;gBACzC,CAAC,aAAa,CAAC,kBAAkB,IAAI,CAAC,CAAC;gBACvC,CAAC,aAAa,CAAC,uBAAuB,IAAI,CAAC,CAAC,CAAC;QAEjD,IAAI,CAAC,UAAU,CAAC,KAAK,IAAI,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC;YAC3E,mEAAmE;YACnE,mEAAmE;YACnE,kEAAkE;YAClE,+DAA+D;YAC/D,MAAM,CAAC,MAAM,CAAC,IAAI,CACd,EAAE,KAAK,EAAE,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,EACjF,8EAA8E,CACjF,CAAC;QACN,CAAC;QAED,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;YACrB,UAAU,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC,kCAAkC;QACrE,CAAC;QAED,OAAO,UAAU,CAAC;IACtB,CAAC;IAED,KAAK,CAAC,qBAAqB,CACvB,MAAsB,EACtB,MAA6B,EAC7B,OAAyB;QAEzB,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACxC,IAAI,MAA0B,CAAC;QAC/B,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,WAAW,IAAI,MAAM,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YAClD,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QACvB,CAAC;QACD,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC5C,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QAE3C,iEAAiE;QACjE,2FAA2F;QAC3F,yFAAyF;QACzF,0FAA0F;QAC1F,MAAM,cAAc,GAAG,6BAA6B,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QAC3E,IAAI,cAAc,EAAE,CAAC;YACjB,MAAM,CAAC,MAAM,GAAG,cAAc,CAAC;QACnC,CAAC;QAED,IAAI,YAAY,GAAG,kBAAkB,CAAC,OAAO,CAAC,YAAY,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC7E,MAAM,CAAC,QAAQ,GAAG,YAAY,CAAC;QAE/B,6DAA6D;QAC7D,IAAI,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,CAAC;YACnD,MAAM,GAAG,QAAQ,CAAC,CAAC,oFAAoF;QAC3G,CAAC;QAED,MAAM,aAAa,GAAG,OAAO,CAAC,aAAkD,CAAC;QACjF,MAAM,MAAM,GAAG,MAAM,CAAC,oBAAoB,CAAC,MAAM,EAAE,aAAa,EAAE,IAAI,IAAI,KAAK,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;QAEtG,MAAM,OAAO,GAAG,gBAAgB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAClD,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QAE9D,MAAM,WAAW,GAAwB,IAAI,CAAC,yBAAyB,CAAC,MAAM,EAAE,QAAQ,CAAC,aAAa,CAAC,CAAC;QAExG,IAAI,QAA+B,CAAC;QACpC,IAAI,aAAiC,EAAE,MAAsC,CAAC;QAC9E,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;QAC3C,IAAI,SAAS,EAAE,CAAC;YACZ,QAAQ,SAAS,CAAC,YAAY,EAAE,CAAC;gBAC7B,KAAK,YAAY,CAAC,UAAU;oBACxB,aAAa,GAAG,QAAQ,CAAC;oBACzB,MAAM;gBACV,KAAK,YAAY,CAAC,IAAI;oBAClB,aAAa,GAAG,MAAM,CAAC;oBACvB,MAAM;gBACV;oBACI,aAAa,GAAG,SAAS,CAAC,YAAY,CAAC;YAC/C,CAAC;YACD,MAAM,OAAO,GAAG,SAAS,CAAC,OAAO,CAAC;YAElC,+EAA+E;YAC/E,MAAM,qBAAqB,GAAG,0BAA0B,CAAC,QAAQ,CAAC,SAAS,CAAC,YAAsB,CAAC,CAAC;YACpG,IACI,SAAS,CAAC,YAAY;gBACtB,CAAC,sBAAsB,CAAC,QAAQ,CAAC,SAAS,CAAC,YAAY,CAAC;gBACxD,CAAC,qBAAqB,EACxB,CAAC;gBACC,MAAM,IAAI,KAAK,CACX,8BAA8B,SAAS,CAAC,YAAY,IAAI;oBACpD,mBAAmB,SAAS,CAAC,aAAa,IAAI;oBAC9C,YAAY,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,aAAa,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAClH,CAAC;YACN,CAAC;YAED,IAAI,OAAO,EAAE,CAAC;gBACV,QAAQ,GAAG,mBAAmB,CAAC,OAAO,CAAC,CAAC;gBAExC,wEAAwE;gBACxE,6DAA6D;gBAC7D,yEAAyE;gBACzE,sEAAsE;gBACtE,qEAAqE;gBACrE,IAAI,qBAAqB,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC3D,MAAM,CAAC,MAAM,CAAC,IAAI,CACd,yCAAyC,SAAS,CAAC,YAAY,KAAK;wBAChE,2CAA2C,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAC/F,CAAC;gBACN,CAAC;gBAED,MAAM,GAAG,wBAAwB,CAAC,OAAO,CAAC,CAAC;gBAC3C,YAAY,GAAG,kBAAkB,CAAC,YAAY,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;YAC/D,CAAC;QACL,CAAC;QAED,IAAI,QAAQ,EAAE,CAAC;YACX,aAAa,GAAG,UAAU,CAAC;QAC/B,CAAC;QAED,gDAAgD;QAChD,YAAY,GAAG,yBAAyB,CAAC,YAAY,CAAc,CAAC;QAEpE,uEAAuE;QACvE,MAAM,WAAW,GAAG,mBAAmB,CAAC,YAAY,CAAC,CAAC,UAAU,CAAC;QACjE,MAAM,YAAY,GAAG;YACjB,YAAY,EAAE,OAAO,CAAC,qBAAqB,IAAI,QAAQ;YACvD,WAAW;YACX,aAAa,EAAE,OAAO,CAAC,kBAAkB;SAC5C,CAAC;QACF,IAAI,qBAAqB,GAAG,iCAAiC,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;QAE1F,4CAA4C;QAC5C,qBAAqB,GAAG,+BAA+B,CAAC,qBAAqB,EAAE,YAAY,CAAC,CAAC;QAE7F,sCAAsC;QACtC,qBAAqB,GAAG,+BAA+B,CAAC,qBAAqB,EAAE;YAC3E,YAAY,EAAE,OAAO,CAAC,yBAAyB,IAAI,CAAC;YACpD,WAAW;SACd,CAAC,CAAC;QAEH,qEAAqE;QACrE,MAAM,iBAAiB,GAAG,yBAAyB,CAAC,qBAAqB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;QAE1F,OAAO;YACH,MAAM,EAAE,MAAM,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;YACrF,WAAW,EAAE,WAAW;YACxB,aAAa,EAAE,aAAa;YAC5B,iBAAiB,EAAE,OAAO,CAAC,yBAAyB,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;YAC3E,YAAY,EAAE,iBAAiB;YAC/B,QAAQ;SACU,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,2BAA2B,CAC7B,MAAsB,EACtB,MAA6B,EAC7B,OAAyB;QAEzB,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACxC,IAAI,MAA0B,CAAC;QAC/B,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,WAAW,IAAI,MAAM,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YAClD,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QACvB,CAAC;QACD,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC5C,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QAE3C,iEAAiE;QACjE,2FAA2F;QAC3F,yFAAyF;QACzF,0FAA0F;QAC1F,MAAM,cAAc,GAAG,6BAA6B,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QAC3E,IAAI,cAAc,EAAE,CAAC;YACjB,MAAM,CAAC,MAAM,GAAG,cAAc,CAAC;QACnC,CAAC;QAED,0EAA0E;QAC1E,MAAM,YAAY,GAAG,kBAAkB,CAAC,OAAO,CAAC,YAAY,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC/E,MAAM,CAAC,QAAQ,GAAG,YAAY,CAAC;QAE/B,IAAI,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,CAAC;YACnD,MAAM,GAAG,QAAQ,CAAC,CAAC,oFAAoF;QAC3G,CAAC;QAED,MAAM,aAAa,GAAG,OAAO,CAAC,aAAkD,CAAC;QACjF,MAAM,MAAM,GAAG,MAAM,CAAC,oBAAoB,CAAC,MAAM,EAAE,aAAa,EAAE,IAAI,IAAI,KAAK,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;QAEtG,MAAM,OAAO,GAAG,gBAAgB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAClD,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC;QAEpE,MAAM,MAAM,GAAG,QAAQ,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAC7C,MAAM,WAAW,GAAwB,IAAI,CAAC,yBAAyB,CAAC,MAAM,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;YACpG,IAAI,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAChD,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;oBACtC,IAAI,QAA+B,CAAC;oBACpC,IAAI,aAAiC,CAAC;oBACtC,QAAQ,SAAS,CAAC,YAAY,EAAE,CAAC;wBAC7B,KAAK,YAAY,CAAC,UAAU;4BACxB,aAAa,GAAG,QAAQ,CAAC;4BACzB,MAAM;wBACV,KAAK,YAAY,CAAC,IAAI;4BAClB,aAAa,GAAG,MAAM,CAAC;4BACvB,MAAM;wBACV;4BACI,aAAa,GAAG,SAAS,CAAC,YAAY,CAAC;oBAC/C,CAAC;oBACD,+EAA+E;oBAC/E,MAAM,qBAAqB,GAAG,0BAA0B,CAAC,QAAQ,CAAC,SAAS,CAAC,YAAsB,CAAC,CAAC;oBACpG,IACI,SAAS,CAAC,YAAY;wBACtB,CAAC,sBAAsB,CAAC,QAAQ,CAAC,SAAS,CAAC,YAAY,CAAC;wBACxD,CAAC,qBAAqB,EACxB,CAAC;wBACC,MAAM,IAAI,KAAK,CACX,8BAA8B,SAAS,CAAC,YAAY,IAAI;4BACpD,mBAAmB,SAAS,CAAC,aAAa,IAAI;4BAC9C,YAAY,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,aAAa,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAC5H,CAAC;oBACN,CAAC;oBACD,IAAI,SAAS,CAAC,OAAO,EAAE,IAAI,KAAK,OAAO,EAAE,CAAC;wBACtC,+CAA+C;wBAC/C,MAAM,eAAe,GAAG,wBAAwB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;wBACpE,QAAQ,GAAG,mBAAmB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;wBAClD,IAAI,QAAQ,EAAE,CAAC;4BACX,aAAa,GAAG,UAAU,CAAC;4BAC3B,yDAAyD;4BACzD,yDAAyD;4BACzD,kDAAkD;4BAClD,kBAAkB;4BAClB,IAAI,qBAAqB,EAAE,CAAC;gCACxB,MAAM,CAAC,MAAM,CAAC,IAAI,CACd,yCAAyC,SAAS,CAAC,YAAY,KAAK;oCAChE,2CAA2C,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAC/F,CAAC;4BACN,CAAC;wBACL,CAAC;wBACD,OAAO;4BACH,MAAM,EAAE,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE;4BACzD,WAAW,EAAE,WAAW;4BACxB,aAAa,EAAE,aAAa;4BAC5B,QAAQ;yBACX,CAAC;oBACN,CAAC;gBACL,CAAC;YACL,CAAC;YACD,wDAAwD;YACxD,OAAO;gBACH,MAAM,EAAE,IAAI,CAAC,cAAc,EAAE,kBAAkB;oBAC3C,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,KAAK,EAAE,IAAI,CAAC,cAAc,CAAC,kBAAkB,EAAE,CAAC;oBAC5E,CAAC,CAAC,EAAE;gBACR,aAAa,EAAE,IAAI,CAAC,cAAc,EAAE,WAAW,IAAI,EAAE;gBACrD,WAAW,EAAE,WAAW;aAC3B,CAAC;QACN,CAAC,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAClB,CAAC;IAED;;;;;;;;;;;;;;;;;;;OAmBG;IACH,qBAAqB,CAAC,OAAuB,EAAE,KAAc,EAAE,OAA+B;QAC1F,oDAAoD;QACpD,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;QAEhD,IAAI,CAAC,UAAU,EAAE,CAAC;YACd,+CAA+C;YAC/C,oEAAoE;YACpE,MAAM,KAAK,CAAC;QAChB,CAAC;QAED,MAAM,QAAQ,GAAG,KAAK,CAAC;QACvB,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC;QAEvC,wBAAwB;QACxB,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC;QAEjC,6CAA6C;QAC7C,IAAI,WAAW,GAAG,OAAO,CAAC;QAE1B,2DAA2D;QAC3D,IAAI,cAAc,EAAE,CAAC;YACjB,WAAW,GAAG,IAAI,cAAc,KAAK,WAAW,EAAE,CAAC;QACvD,CAAC;QAED,qDAAqD;QACrD,MAAM,SAAS,GAAG,IAAI,CAAC,sBAAsB,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QAEvE,kDAAkD;QAClD,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAEjD,OAAO,IAAI,eAAe,CACtB,IAAI,OAAO,CAAC,QAAQ,KAAK,WAAW,EAAE,EACtC,SAAS,EACT,OAAO,EACP,KAAK,EACL,cAAc,EACd,SAAS,CACZ,CAAC;IACN,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,KAAc;QACnC,OAAO,CACH,KAAK,KAAK,IAAI;YACd,OAAO,KAAK,KAAK,QAAQ;YACzB,QAAQ,IAAI,KAAK;YACjB,OAAQ,KAA8B,CAAC,MAAM,KAAK,QAAQ;YAC1D,SAAS,IAAI,KAAK;YAClB,OAAQ,KAA+B,CAAC,OAAO,KAAK,QAAQ,CAC/D,CAAC;IACN,CAAC;IAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA6BG;IACK,sBAAsB,CAAC,cAAsB,EAAE,OAAgB;QACnE,IAAI,OAAO,IAAI,IAAI,CAAC,uBAAuB,CAAC,OAAO,CAAC;YAAE,OAAO,KAAK,CAAC;QAEnE,yBAAyB;QACzB,IAAI,cAAc,KAAK,GAAG;YAAE,OAAO,IAAI,CAAC,CAAC,kBAAkB;QAC3D,IAAI,cAAc,KAAK,GAAG;YAAE,OAAO,IAAI,CAAC,CAAC,mBAAmB;QAC5D,IAAI,cAAc,KAAK,GAAG;YAAE,OAAO,IAAI,CAAC,CAAC,sCAAsC;QAC/E,IAAI,cAAc,KAAK,GAAG;YAAE,OAAO,IAAI,CAAC,CAAC,cAAc;QACvD,IAAI,cAAc,KAAK,GAAG;YAAE,OAAO,IAAI,CAAC,CAAC,sBAAsB;QAC/D,IAAI,cAAc,KAAK,GAAG;YAAE,OAAO,IAAI,CAAC,CAAC,kBAAkB;QAC3D,IAAI,cAAc,IAAI,GAAG,IAAI,cAAc,GAAG,GAAG;YAAE,OAAO,IAAI,CAAC,CAAC,0BAA0B;QAE1F,+EAA+E;QAC/E,qEAAqE;QACrE,IAAI,cAAc,KAAK,GAAG,IAAI,OAAO,EAAE,CAAC;YACpC,IACI,OAAO,CAAC,QAAQ,CAAC,wCAAwC,CAAC;gBAC1D,OAAO,CAAC,QAAQ,CAAC,oCAAoC,CAAC,EACxD,CAAC;gBACC,OAAO,IAAI,CAAC;YAChB,CAAC;QACL,CAAC;QAED,2FAA2F;QAC3F,mFAAmF;QACnF,sFAAsF;QACtF,IAAI,OAAO,EAAE,CAAC;YACV,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;YACpC,IAAI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC;gBAAE,OAAO,IAAI,CAAC;QAC5G,CAAC;QAED,kCAAkC;QAClC,IAAI,cAAc,IAAI,GAAG,IAAI,cAAc,GAAG,GAAG;YAAE,OAAO,KAAK,CAAC;QAEhE,4DAA4D;QAC5D,OAAO,SAAS,CAAC;IACrB,CAAC;IAEO,uBAAuB,CAAC,OAAe;QAC3C,MAAM,YAAY,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QAC3C,OAAO,YAAY,CAAC,QAAQ,CAAC,eAAe,CAAC,IAAI,YAAY,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC;IAClG,CAAC;IAED;;;;OAIG;IACK,gBAAgB,CAAC,OAAe;QACpC,+BAA+B;QAC/B,MAAM,QAAQ,GAAG;YACb,+BAA+B,EAAE,sCAAsC;YACvE,aAAa,EAAE,wBAAwB;YACvC,eAAe,EAAE,yBAAyB;YAC1C,cAAc,EAAE,4BAA4B;SAC/C,CAAC;QAEF,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC7B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACrC,IAAI,KAAK,EAAE,CAAC;gBACR,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC;YACpB,CAAC;QACL,CAAC;QAED,OAAO,SAAS,CAAC;IACrB,CAAC;CACJ;AAED;;;;GAIG;AACH,MAAM,UAAU,gCAAgC,CAAC,QAAmB;IAChE,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;QAC5B,IAAI,CAAC,OAAO,CAAC,KAAK;YAAE,OAAO,OAAO,CAAC;QACnC,MAAM,gBAAgB,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,IAAI,CAAC,CAAC,gBAAgB,CAAC,CAAC;QACzF,IAAI,CAAC,gBAAgB;YAAE,OAAO,OAAO,CAAC;QAEtC,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACxC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACpB,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBACrF,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC;gBACrF,OAAO,EAAE,IAAI,EAAE,eAAe,IAAI,CAAC,YAAY,CAAC,IAAI,IAAI,SAAS,IAAI,EAAE,CAAC;YAC5E,CAAC;YACD,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACxB,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ;oBAC1C,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC;oBAChD,CAAC,CAAC,aAAa,CAAC;gBACpB,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC;gBACrF,OAAO,EAAE,IAAI,EAAE,oBAAoB,IAAI,CAAC,gBAAgB,CAAC,IAAI,KAAK,SAAS,GAAG,EAAE,CAAC;YACrF,CAAC;YACD,OAAO,IAAI,CAAC;QAChB,CAAC,CAAC,CAAC;QACH,OAAO,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,kBAAkB,CAAC,KAA0C;IAClE,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/B,OAAO,SAAS,CAAC;IACrB,CAAC;IACD,oDAAoD;IACpD,8DAA8D;IAC9D,OAAO;QACH,oBAAoB,EAAE,KAAK,CAAC,GAAG,CAAC,eAAe,CAAC;KACnD,CAAC;AACN,CAAC;AAED,SAAS,eAAe,CAAC,IAAoB;IACzC,OAAO;QACH,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,WAAW,EAAE,IAAI,CAAC,WAAW;QAC7B,0DAA0D;QAC1D,8EAA8E;QAC9E,sFAAsF;QACtF,oBAAoB,EAAE,IAAI,CAAC,YAAY;KAC1C,CAAC;AACN,CAAC;AAED;;;;;GAKG;AACH,SAAS,kBAAkB,CAAC,YAAqB,EAAE,MAAiB;IAChE,oDAAoD;IACpD,MAAM,SAAS,GAAG,uBAAuB,CAAU,YAAY,CAAC,CAAC;IACjE,MAAM,SAAS,GAAG,SAAS,IAAI,CAAE,YAA0B,IAAI,EAAE,CAAC,CAAC;IACnE,OAAO,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;AACpC,CAAC;AAED,MAAM,UAAU,GAAG,oBAAoB,CAAC;AAExC;;;GAGG;AACH,SAAS,6BAA6B,CAAC,YAAqB;IACxD,IAAI,OAAO,YAAY,KAAK,QAAQ,IAAI,YAAY,KAAK,IAAI,EAAE,CAAC;QAC5D,MAAM,CAAC,GAAG,YAAuC,CAAC;QAClD,IAAI,CAAC,CAAC,UAAU,CAAC,IAAI,OAAO,CAAC,CAAC,UAAU,CAAC,KAAK,QAAQ,EAAE,CAAC;YACrD,OAAO,CAAC,CAAC,UAAU,CAAY,CAAC;QACpC,CAAC;IACL,CAAC;IACD,OAAO,SAAS,CAAC;AACrB,CAAC;AAED;;;;;GAKG;AACH,SAAS,yBAAyB,CAAC,YAAqB,EAAE,MAA2B;IACjF,IAAI,CAAC,MAAM;QAAE,OAAO,YAAY,CAAC;IACjC,IAAI,OAAO,YAAY,KAAK,QAAQ,IAAI,YAAY,KAAK,IAAI,EAAE,CAAC;QAC5D,OAAO,EAAE,GAAI,YAAuB,EAAE,CAAC,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IACjE,CAAC;IACD,OAAO,YAAY,CAAC;AACxB,CAAC;AAED;;;;;;;;;;GAUG;AACH,SAAS,sBAAsB,CAAC,QAAgB;IAC5C,QAAQ,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC3B,IAAI,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QACrD,IAAI,CAAC;YACD,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAChC,CAAC;QAAC,MAAM,CAAC;YACL,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC;QAChC,CAAC;IACL,CAAC;SAAM,CAAC;QACJ,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC;IAChC,CAAC;AACL,CAAC"}
|
|
@@ -1,7 +1,7 @@
|
|
|
1
|
-
import { AIModel, Completion, ExecutionOptions, PromptSegment } from
|
|
2
|
-
import { VertexAIDriver } from
|
|
1
|
+
import { type AIModel, type Completion, type ExecutionOptions, type PromptSegment } from '@llumiverse/core';
|
|
2
|
+
import type { VertexAIDriver } from '../index.js';
|
|
3
3
|
interface ImagenBaseReference {
|
|
4
|
-
referenceType:
|
|
4
|
+
referenceType: 'REFERENCE_TYPE_RAW' | 'REFERENCE_TYPE_MASK' | 'REFERENCE_TYPE_SUBJECT' | 'REFERENCE_TYPE_CONTROL' | 'REFERENCE_TYPE_STYLE';
|
|
5
5
|
referenceId: number;
|
|
6
6
|
referenceImage: {
|
|
7
7
|
bytesBase64Encoded: string;
|
|
@@ -25,10 +25,10 @@ export declare enum ImagenMaskMode {
|
|
|
25
25
|
MASK_MODE_SEMANTIC = "MASK_MODE_SEMANTIC"
|
|
26
26
|
}
|
|
27
27
|
interface ImagenReferenceRaw extends ImagenBaseReference {
|
|
28
|
-
referenceType:
|
|
28
|
+
referenceType: 'REFERENCE_TYPE_RAW';
|
|
29
29
|
}
|
|
30
|
-
interface ImagenReferenceMask extends Omit<ImagenBaseReference,
|
|
31
|
-
referenceType:
|
|
30
|
+
interface ImagenReferenceMask extends Omit<ImagenBaseReference, 'referenceImage'> {
|
|
31
|
+
referenceType: 'REFERENCE_TYPE_MASK';
|
|
32
32
|
maskImageConfig: {
|
|
33
33
|
maskMode?: ImagenMaskMode;
|
|
34
34
|
maskClasses?: number[];
|
|
@@ -39,21 +39,21 @@ interface ImagenReferenceMask extends Omit<ImagenBaseReference, "referenceImage"
|
|
|
39
39
|
};
|
|
40
40
|
}
|
|
41
41
|
interface ImagenReferenceSubject extends ImagenBaseReference {
|
|
42
|
-
referenceType:
|
|
42
|
+
referenceType: 'REFERENCE_TYPE_SUBJECT';
|
|
43
43
|
subjectImageConfig: {
|
|
44
44
|
subjectDescription: string;
|
|
45
|
-
subjectType:
|
|
45
|
+
subjectType: 'SUBJECT_TYPE_PERSON' | 'SUBJECT_TYPE_ANIMAL' | 'SUBJECT_TYPE_PRODUCT' | 'SUBJECT_TYPE_DEFAULT';
|
|
46
46
|
};
|
|
47
47
|
}
|
|
48
48
|
interface ImagenReferenceControl extends ImagenBaseReference {
|
|
49
|
-
referenceType:
|
|
49
|
+
referenceType: 'REFERENCE_TYPE_CONTROL';
|
|
50
50
|
controlImageConfig: {
|
|
51
|
-
controlType:
|
|
51
|
+
controlType: 'CONTROL_TYPE_FACE_MESH' | 'CONTROL_TYPE_CANNY' | 'CONTROL_TYPE_SCRIBBLE';
|
|
52
52
|
enableControlImageComputation?: boolean;
|
|
53
53
|
};
|
|
54
54
|
}
|
|
55
55
|
interface ImagenReferenceStyle extends ImagenBaseReference {
|
|
56
|
-
referenceType:
|
|
56
|
+
referenceType: 'REFERENCE_TYPE_STYLE';
|
|
57
57
|
styleImageConfig: {
|
|
58
58
|
styleDescription?: string;
|
|
59
59
|
};
|
|
@@ -65,6 +65,7 @@ export interface ImagenPrompt {
|
|
|
65
65
|
subjectDescription?: string;
|
|
66
66
|
negativePrompt?: string;
|
|
67
67
|
}
|
|
68
|
+
export declare function formatImagenDebugPrompt(prompt: ImagenPrompt): ImagenPrompt;
|
|
68
69
|
export declare class ImagenModelDefinition {
|
|
69
70
|
model: AIModel;
|
|
70
71
|
constructor(modelId: string);
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"imagen.d.ts","sourceRoot":"","sources":["../../../src/vertexai/models/imagen.ts"],"names":[],"mappings":"AAEA,OAAO,EACH,KAAK,OAAO,EACZ,KAAK,UAAU,EACf,KAAK,gBAAgB,EAIrB,KAAK,aAAa,EAErB,MAAM,kBAAkB,CAAC;AAE1B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAElD,UAAU,mBAAmB;IACzB,aAAa,EACP,oBAAoB,GACpB,qBAAqB,GACrB,wBAAwB,GACxB,wBAAwB,GACxB,sBAAsB,CAAC;IAC7B,WAAW,EAAE,MAAM,CAAC;IACpB,cAAc,EAAE;QACZ,kBAAkB,EAAE,MAAM,CAAC;KAC9B,CAAC;CACL;AAED,oBAAY,cAAc;IACtB,UAAU,eAAe;IACzB,yBAAyB,8BAA8B;IACvD,2BAA2B,gCAAgC;IAC3D,gBAAgB,qBAAqB;IACrC,kBAAkB,uBAAuB;IACzC,qBAAqB,0BAA0B;IAC/C,mBAAmB,wBAAwB;IAC3C,wBAAwB,6BAA6B;IACrD,sBAAsB,2BAA2B;CACpD;AAED,oBAAY,cAAc;IACtB,uBAAuB,4BAA4B;IACnD,oBAAoB,yBAAyB;IAC7C,oBAAoB,yBAAyB;IAC7C,kBAAkB,uBAAuB;CAC5C;AAED,UAAU,kBAAmB,SAAQ,mBAAmB;IACpD,aAAa,EAAE,oBAAoB,CAAC;CACvC;AAED,UAAU,mBAAoB,SAAQ,IAAI,CAAC,mBAAmB,EAAE,gBAAgB,CAAC;IAC7E,aAAa,EAAE,qBAAqB,CAAC;IACrC,eAAe,EAAE;QACb,QAAQ,CAAC,EAAE,cAAc,CAAC;QAC1B,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;QACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;KACrB,CAAC;IACF,cAAc,CAAC,EAAE;QAEb,kBAAkB,EAAE,MAAM,CAAC;KAC9B,CAAC;CACL;AAED,UAAU,sBAAuB,SAAQ,mBAAmB;IACxD,aAAa,EAAE,wBAAwB,CAAC;IACxC,kBAAkB,EAAE;QAChB,kBAAkB,EAAE,MAAM,CAAC;QAC3B,WAAW,EAAE,qBAAqB,GAAG,qBAAqB,GAAG,sBAAsB,GAAG,sBAAsB,CAAC;KAChH,CAAC;CACL;AAED,UAAU,sBAAuB,SAAQ,mBAAmB;IACxD,aAAa,EAAE,wBAAwB,CAAC;IACxC,kBAAkB,EAAE;QAChB,WAAW,EAAE,wBAAwB,GAAG,oBAAoB,GAAG,uBAAuB,CAAC;QACvF,6BAA6B,CAAC,EAAE,OAAO,CAAC;KAC3C,CAAC;CACL;AAED,UAAU,oBAAqB,SAAQ,mBAAmB;IACtD,aAAa,EAAE,sBAAsB,CAAC;IACtC,gBAAgB,EAAE;QACd,gBAAgB,CAAC,EAAE,MAAM,CAAC;KAC7B,CAAC;CACL;AAED,KAAK,aAAa,GACZ,kBAAkB,GAClB,mBAAmB,GACnB,sBAAsB,GACtB,sBAAsB,GACtB,oBAAoB,CAAC;AAE3B,MAAM,WAAW,YAAY;IACzB,MAAM,EAAE,MAAM,CAAC;IACf,eAAe,CAAC,EAAE,aAAa,EAAE,CAAC;IAClC,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,cAAc,CAAC,EAAE,MAAM,CAAC;CAC3B;AAED,wBAAgB,uBAAuB,CAAC,MAAM,EAAE,YAAY,GAAG,YAAY,CAgB1E;AAkED,qBAAa,qBAAqB;IAC9B,KAAK,EAAE,OAAO,CAAC;gBAEH,OAAO,EAAE,MAAM;IAUrB,YAAY,CACd,OAAO,EAAE,cAAc,EACvB,QAAQ,EAAE,aAAa,EAAE,EACzB,OAAO,EAAE,gBAAgB,GAC1B,OAAO,CAAC,YAAY,CAAC;IAkLlB,sBAAsB,CACxB,MAAM,EAAE,cAAc,EACtB,MAAM,EAAE,YAAY,EACpB,OAAO,EAAE,gBAAgB,GAC1B,OAAO,CAAC,UAAU,CAAC;CAoEzB"}
|
|
@@ -1,6 +1,7 @@
|
|
|
1
|
-
import { ModelType, PromptRole, readStreamAsBase64 } from "@llumiverse/core";
|
|
2
1
|
// Import the helper module for converting arbitrary protobuf.Value objects
|
|
3
2
|
import { helpers } from '@google-cloud/aiplatform';
|
|
3
|
+
import { ModelType, PromptRole, readStreamAsBase64, } from '@llumiverse/core';
|
|
4
|
+
import { truncateBinaryForDebug } from '../../shared/debug-prompt.js';
|
|
4
5
|
export var ImagenTaskType;
|
|
5
6
|
(function (ImagenTaskType) {
|
|
6
7
|
ImagenTaskType["TEXT_IMAGE"] = "TEXT_IMAGE";
|
|
@@ -20,13 +21,30 @@ export var ImagenMaskMode;
|
|
|
20
21
|
ImagenMaskMode["MASK_MODE_FOREGROUND"] = "MASK_MODE_FOREGROUND";
|
|
21
22
|
ImagenMaskMode["MASK_MODE_SEMANTIC"] = "MASK_MODE_SEMANTIC";
|
|
22
23
|
})(ImagenMaskMode || (ImagenMaskMode = {}));
|
|
24
|
+
export function formatImagenDebugPrompt(prompt) {
|
|
25
|
+
return {
|
|
26
|
+
...prompt,
|
|
27
|
+
referenceImages: prompt.referenceImages?.map((reference) => {
|
|
28
|
+
if (!reference.referenceImage?.bytesBase64Encoded) {
|
|
29
|
+
return reference;
|
|
30
|
+
}
|
|
31
|
+
return {
|
|
32
|
+
...reference,
|
|
33
|
+
referenceImage: {
|
|
34
|
+
...reference.referenceImage,
|
|
35
|
+
bytesBase64Encoded: truncateBinaryForDebug(reference.referenceImage.bytesBase64Encoded),
|
|
36
|
+
},
|
|
37
|
+
};
|
|
38
|
+
}),
|
|
39
|
+
};
|
|
40
|
+
}
|
|
23
41
|
function getImagenParameters(taskType, options) {
|
|
24
42
|
const commonParameters = {
|
|
25
43
|
sampleCount: options?.number_of_images,
|
|
26
44
|
seed: options?.seed,
|
|
27
45
|
safetySetting: options?.safety_setting,
|
|
28
46
|
personGeneration: options?.person_generation,
|
|
29
|
-
negativePrompt: taskType ? undefined :
|
|
47
|
+
negativePrompt: taskType ? undefined : '', //Filled in later from the prompt
|
|
30
48
|
//TODO: Add more safety and prompt rejection information
|
|
31
49
|
//includeSafetyAttributes: true,
|
|
32
50
|
//includeRaiReason: true,
|
|
@@ -35,7 +53,7 @@ function getImagenParameters(taskType, options) {
|
|
|
35
53
|
case ImagenTaskType.EDIT_MODE_INPAINT_REMOVAL:
|
|
36
54
|
return {
|
|
37
55
|
...commonParameters,
|
|
38
|
-
editMode:
|
|
56
|
+
editMode: 'EDIT_MODE_INPAINT_REMOVAL',
|
|
39
57
|
editConfig: {
|
|
40
58
|
baseSteps: options?.edit_steps,
|
|
41
59
|
},
|
|
@@ -43,7 +61,7 @@ function getImagenParameters(taskType, options) {
|
|
|
43
61
|
case ImagenTaskType.EDIT_MODE_INPAINT_INSERTION:
|
|
44
62
|
return {
|
|
45
63
|
...commonParameters,
|
|
46
|
-
editMode:
|
|
64
|
+
editMode: 'EDIT_MODE_INPAINT_INSERTION',
|
|
47
65
|
editConfig: {
|
|
48
66
|
baseSteps: options?.edit_steps,
|
|
49
67
|
},
|
|
@@ -51,7 +69,7 @@ function getImagenParameters(taskType, options) {
|
|
|
51
69
|
case ImagenTaskType.EDIT_MODE_BGSWAP:
|
|
52
70
|
return {
|
|
53
71
|
...commonParameters,
|
|
54
|
-
editMode:
|
|
72
|
+
editMode: 'EDIT_MODE_BGSWAP',
|
|
55
73
|
editConfig: {
|
|
56
74
|
baseSteps: options?.edit_steps,
|
|
57
75
|
},
|
|
@@ -59,7 +77,7 @@ function getImagenParameters(taskType, options) {
|
|
|
59
77
|
case ImagenTaskType.EDIT_MODE_OUTPAINT:
|
|
60
78
|
return {
|
|
61
79
|
...commonParameters,
|
|
62
|
-
editMode:
|
|
80
|
+
editMode: 'EDIT_MODE_OUTPAINT',
|
|
63
81
|
editConfig: {
|
|
64
82
|
baseSteps: options?.edit_steps,
|
|
65
83
|
},
|
|
@@ -80,7 +98,7 @@ function getImagenParameters(taskType, options) {
|
|
|
80
98
|
...commonParameters,
|
|
81
99
|
};
|
|
82
100
|
default:
|
|
83
|
-
throw new Error(
|
|
101
|
+
throw new Error('Task type not supported');
|
|
84
102
|
}
|
|
85
103
|
}
|
|
86
104
|
export class ImagenModelDefinition {
|
|
@@ -95,13 +113,13 @@ export class ImagenModelDefinition {
|
|
|
95
113
|
};
|
|
96
114
|
}
|
|
97
115
|
async createPrompt(_driver, segments, options) {
|
|
98
|
-
const splits = options.model.split(
|
|
116
|
+
const splits = options.model.split('/');
|
|
99
117
|
const modelName = splits[splits.length - 1];
|
|
100
118
|
options = { ...options, model: modelName };
|
|
101
119
|
const prompt = {
|
|
102
|
-
prompt:
|
|
120
|
+
prompt: '',
|
|
103
121
|
};
|
|
104
|
-
//Collect text prompts, Imagen does not support roles, so everything gets merged together
|
|
122
|
+
//Collect text prompts, Imagen does not support roles, so everything gets merged together
|
|
105
123
|
// however we still respect our typical pattern. System First, Safety Last.
|
|
106
124
|
const system = [];
|
|
107
125
|
const user = [];
|
|
@@ -128,96 +146,104 @@ export class ImagenModelDefinition {
|
|
|
128
146
|
if (!prompt.referenceImages) {
|
|
129
147
|
prompt.referenceImages = [];
|
|
130
148
|
}
|
|
131
|
-
//Always required, but only used by customisation.
|
|
149
|
+
//Always required, but only used by customisation.
|
|
132
150
|
//Each ref ID refers to a single "reference", i.e. object. To provide multiple images of a single ref,
|
|
133
151
|
//include multiple images in one prompt.
|
|
134
152
|
const refId = prompt.referenceImages.length + 1;
|
|
135
153
|
for (const img of msg.files) {
|
|
136
|
-
if (img.mime_type?.includes(
|
|
154
|
+
if (img.mime_type?.includes('image')) {
|
|
137
155
|
if (msg.role !== PromptRole.mask) {
|
|
138
156
|
//Editing based mode requires a reference image
|
|
139
|
-
if (imagenOptions?.edit_mode?.includes(
|
|
157
|
+
if (imagenOptions?.edit_mode?.includes('EDIT_MODE')) {
|
|
140
158
|
prompt.referenceImages.push({
|
|
141
|
-
referenceType:
|
|
159
|
+
referenceType: 'REFERENCE_TYPE_RAW',
|
|
142
160
|
referenceId: refId,
|
|
143
161
|
referenceImage: {
|
|
144
162
|
bytesBase64Encoded: await readStreamAsBase64(await img.getStream()),
|
|
145
|
-
}
|
|
163
|
+
},
|
|
146
164
|
});
|
|
147
165
|
//If mask is auto-generated, add a mask reference
|
|
148
166
|
if (mask_mode !== ImagenMaskMode.MASK_MODE_USER_PROVIDED) {
|
|
149
167
|
prompt.referenceImages.push({
|
|
150
|
-
referenceType:
|
|
168
|
+
referenceType: 'REFERENCE_TYPE_MASK',
|
|
151
169
|
referenceId: refId,
|
|
152
170
|
maskImageConfig: {
|
|
153
171
|
maskMode: mask_mode,
|
|
154
172
|
dilation: imagenOptions?.mask_dilation,
|
|
155
|
-
}
|
|
173
|
+
},
|
|
156
174
|
});
|
|
157
175
|
}
|
|
158
176
|
}
|
|
159
|
-
else if (options.model_options?.edit_mode ===
|
|
177
|
+
else if (options.model_options?.edit_mode ===
|
|
178
|
+
ImagenTaskType.CUSTOMIZATION_SUBJECT) {
|
|
160
179
|
//First image is always the control image
|
|
161
|
-
if (refId
|
|
180
|
+
if (refId === 1) {
|
|
162
181
|
//Customization subject mode requires a control image
|
|
163
182
|
prompt.referenceImages.push({
|
|
164
|
-
referenceType:
|
|
183
|
+
referenceType: 'REFERENCE_TYPE_CONTROL',
|
|
165
184
|
referenceId: refId,
|
|
166
185
|
referenceImage: {
|
|
167
186
|
bytesBase64Encoded: await readStreamAsBase64(await img.getStream()),
|
|
168
187
|
},
|
|
169
188
|
controlImageConfig: {
|
|
170
|
-
controlType: imagenOptions?.controlType ===
|
|
189
|
+
controlType: imagenOptions?.controlType === 'CONTROL_TYPE_FACE_MESH'
|
|
190
|
+
? 'CONTROL_TYPE_FACE_MESH'
|
|
191
|
+
: 'CONTROL_TYPE_CANNY',
|
|
171
192
|
enableControlImageComputation: imagenOptions?.controlImageComputation,
|
|
172
|
-
}
|
|
193
|
+
},
|
|
173
194
|
});
|
|
174
195
|
}
|
|
175
196
|
else {
|
|
176
197
|
// Subject images
|
|
177
198
|
prompt.referenceImages.push({
|
|
178
|
-
referenceType:
|
|
199
|
+
referenceType: 'REFERENCE_TYPE_SUBJECT',
|
|
179
200
|
referenceId: refId,
|
|
180
201
|
referenceImage: {
|
|
181
202
|
bytesBase64Encoded: await readStreamAsBase64(await img.getStream()),
|
|
182
203
|
},
|
|
183
204
|
subjectImageConfig: {
|
|
184
205
|
subjectDescription: prompt.subjectDescription ?? msg.content,
|
|
185
|
-
subjectType: imagenOptions?.subjectType ??
|
|
186
|
-
}
|
|
206
|
+
subjectType: imagenOptions?.subjectType ?? 'SUBJECT_TYPE_DEFAULT',
|
|
207
|
+
},
|
|
187
208
|
});
|
|
188
209
|
}
|
|
189
210
|
}
|
|
190
|
-
else if (options.model_options?.edit_mode ===
|
|
211
|
+
else if (options.model_options?.edit_mode ===
|
|
212
|
+
ImagenTaskType.CUSTOMIZATION_STYLE) {
|
|
191
213
|
// Style images
|
|
192
214
|
prompt.referenceImages.push({
|
|
193
|
-
referenceType:
|
|
215
|
+
referenceType: 'REFERENCE_TYPE_STYLE',
|
|
194
216
|
referenceId: refId,
|
|
195
217
|
referenceImage: {
|
|
196
218
|
bytesBase64Encoded: await readStreamAsBase64(await img.getStream()),
|
|
197
219
|
},
|
|
198
220
|
styleImageConfig: {
|
|
199
221
|
styleDescription: prompt.subjectDescription ?? msg.content,
|
|
200
|
-
}
|
|
222
|
+
},
|
|
201
223
|
});
|
|
202
224
|
}
|
|
203
|
-
else if (options.model_options?.edit_mode ===
|
|
225
|
+
else if (options.model_options?.edit_mode ===
|
|
226
|
+
ImagenTaskType.CUSTOMIZATION_CONTROLLED) {
|
|
204
227
|
// Control images
|
|
205
228
|
prompt.referenceImages.push({
|
|
206
|
-
referenceType:
|
|
229
|
+
referenceType: 'REFERENCE_TYPE_CONTROL',
|
|
207
230
|
referenceId: refId,
|
|
208
231
|
referenceImage: {
|
|
209
232
|
bytesBase64Encoded: await readStreamAsBase64(await img.getStream()),
|
|
210
233
|
},
|
|
211
234
|
controlImageConfig: {
|
|
212
|
-
controlType: imagenOptions?.controlType ===
|
|
235
|
+
controlType: imagenOptions?.controlType === 'CONTROL_TYPE_FACE_MESH'
|
|
236
|
+
? 'CONTROL_TYPE_FACE_MESH'
|
|
237
|
+
: 'CONTROL_TYPE_CANNY',
|
|
213
238
|
enableControlImageComputation: imagenOptions?.controlImageComputation,
|
|
214
|
-
}
|
|
239
|
+
},
|
|
215
240
|
});
|
|
216
241
|
}
|
|
217
|
-
else if (options.model_options?.edit_mode ===
|
|
242
|
+
else if (options.model_options?.edit_mode ===
|
|
243
|
+
ImagenTaskType.CUSTOMIZATION_INSTRUCT) {
|
|
218
244
|
// Control images
|
|
219
245
|
prompt.referenceImages.push({
|
|
220
|
-
referenceType:
|
|
246
|
+
referenceType: 'REFERENCE_TYPE_RAW',
|
|
221
247
|
referenceId: refId,
|
|
222
248
|
referenceImage: {
|
|
223
249
|
bytesBase64Encoded: await readStreamAsBase64(await img.getStream()),
|
|
@@ -228,7 +254,7 @@ export class ImagenModelDefinition {
|
|
|
228
254
|
//If mask is user-provided, add a mask reference
|
|
229
255
|
if (msg.role === PromptRole.mask && mask_mode === ImagenMaskMode.MASK_MODE_USER_PROVIDED) {
|
|
230
256
|
prompt.referenceImages.push({
|
|
231
|
-
referenceType:
|
|
257
|
+
referenceType: 'REFERENCE_TYPE_MASK',
|
|
232
258
|
referenceId: refId,
|
|
233
259
|
referenceImage: {
|
|
234
260
|
bytesBase64Encoded: await readStreamAsBase64(await img.getStream()),
|
|
@@ -236,7 +262,7 @@ export class ImagenModelDefinition {
|
|
|
236
262
|
maskImageConfig: {
|
|
237
263
|
maskMode: mask_mode,
|
|
238
264
|
dilation: imagenOptions?.mask_dilation,
|
|
239
|
-
}
|
|
265
|
+
},
|
|
240
266
|
});
|
|
241
267
|
}
|
|
242
268
|
}
|
|
@@ -244,22 +270,23 @@ export class ImagenModelDefinition {
|
|
|
244
270
|
}
|
|
245
271
|
}
|
|
246
272
|
//Extract the text from the segments
|
|
247
|
-
prompt.prompt += [system.join(
|
|
273
|
+
prompt.prompt += [system.join('\n\n'), user.join('\n\n'), safety.join('\n\n')].join('\n\n');
|
|
248
274
|
//Negative prompt
|
|
249
275
|
if (negative.length > 0) {
|
|
250
|
-
prompt.negativePrompt = negative.join(
|
|
276
|
+
prompt.negativePrompt = negative.join(', ');
|
|
251
277
|
}
|
|
252
278
|
console.log(prompt);
|
|
253
279
|
return prompt;
|
|
254
280
|
}
|
|
255
281
|
async requestImageGeneration(driver, prompt, options) {
|
|
256
|
-
if (options.model_options?._option_id !== undefined &&
|
|
257
|
-
|
|
282
|
+
if (options.model_options?._option_id !== undefined &&
|
|
283
|
+
options.model_options?._option_id !== 'vertexai-imagen') {
|
|
284
|
+
driver.logger.debug({ options: options.model_options }, 'Unexpected option id');
|
|
258
285
|
}
|
|
259
286
|
options.model_options = options.model_options;
|
|
260
287
|
const taskType = options.model_options?.edit_mode ?? ImagenTaskType.TEXT_IMAGE;
|
|
261
|
-
driver.logger.info(
|
|
262
|
-
const modelName = options.model.split(
|
|
288
|
+
driver.logger.info(`Task type: ${taskType}`);
|
|
289
|
+
const modelName = options.model.split('/').pop() ?? '';
|
|
263
290
|
// Configure the parent resource
|
|
264
291
|
// TODO: make location configurable, fixed to us-central1 for now
|
|
265
292
|
const endpoint = `projects/${driver.options.project}/locations/us-central1/publishers/google/models/${modelName}`;
|
|
@@ -268,7 +295,7 @@ export class ImagenModelDefinition {
|
|
|
268
295
|
throw new Error('No instance value found');
|
|
269
296
|
}
|
|
270
297
|
const instances = [instanceValue];
|
|
271
|
-
let parameter = getImagenParameters(taskType, options.model_options ?? { _option_id:
|
|
298
|
+
let parameter = getImagenParameters(taskType, options.model_options ?? { _option_id: 'vertexai-imagen' });
|
|
272
299
|
parameter.negativePrompt = prompt.negativePrompt ?? undefined;
|
|
273
300
|
const numberOfImages = options.model_options?.number_of_images ?? 1;
|
|
274
301
|
// Remove all undefined values
|
|
@@ -287,11 +314,11 @@ export class ImagenModelDefinition {
|
|
|
287
314
|
throw new Error('No predictions found');
|
|
288
315
|
}
|
|
289
316
|
// Extract base64 encoded images from predictions
|
|
290
|
-
const images = predictions.map(prediction => prediction.structValue?.fields?.bytesBase64Encoded?.stringValue ?? '');
|
|
317
|
+
const images = predictions.map((prediction) => prediction.structValue?.fields?.bytesBase64Encoded?.stringValue ?? '');
|
|
291
318
|
return {
|
|
292
|
-
result: images.map(image => ({
|
|
293
|
-
type:
|
|
294
|
-
value: image
|
|
319
|
+
result: images.map((image) => ({
|
|
320
|
+
type: 'image',
|
|
321
|
+
value: image,
|
|
295
322
|
})),
|
|
296
323
|
};
|
|
297
324
|
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"imagen.js","sourceRoot":"","sources":["../../../src/vertexai/models/imagen.ts"],"names":[],"mappings":"AAAA,2EAA2E;AAC3E,OAAO,EAAE,OAAO,EAAe,MAAM,0BAA0B,CAAC;AAChE,OAAO,EAKH,SAAS,EACT,UAAU,EAEV,kBAAkB,GACrB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AAgBtE,MAAM,CAAN,IAAY,cAUX;AAVD,WAAY,cAAc;IACtB,2CAAyB,CAAA;IACzB,yEAAuD,CAAA;IACvD,6EAA2D,CAAA;IAC3D,uDAAqC,CAAA;IACrC,2DAAyC,CAAA;IACzC,iEAA+C,CAAA;IAC/C,6DAA2C,CAAA;IAC3C,uEAAqD,CAAA;IACrD,mEAAiD,CAAA;AACrD,CAAC,EAVW,cAAc,KAAd,cAAc,QAUzB;AAED,MAAM,CAAN,IAAY,cAKX;AALD,WAAY,cAAc;IACtB,qEAAmD,CAAA;IACnD,+DAA6C,CAAA;IAC7C,+DAA6C,CAAA;IAC7C,2DAAyC,CAAA;AAC7C,CAAC,EALW,cAAc,KAAd,cAAc,QAKzB;AAwDD,MAAM,UAAU,uBAAuB,CAAC,MAAoB;IACxD,OAAO;QACH,GAAG,MAAM;QACT,eAAe,EAAE,MAAM,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE;YACvD,IAAI,CAAC,SAAS,CAAC,cAAc,EAAE,kBAAkB,EAAE,CAAC;gBAChD,OAAO,SAAS,CAAC;YACrB,CAAC;YACD,OAAO;gBACH,GAAG,SAAS;gBACZ,cAAc,EAAE;oBACZ,GAAG,SAAS,CAAC,cAAc;oBAC3B,kBAAkB,EAAE,sBAAsB,CAAC,SAAS,CAAC,cAAc,CAAC,kBAAkB,CAAC;iBAC1F;aACJ,CAAC;QACN,CAAC,CAAC;KACL,CAAC;AACN,CAAC;AAED,SAAS,mBAAmB,CAAC,QAAgB,EAAE,OAAsB;IACjE,MAAM,gBAAgB,GAAG;QACrB,WAAW,EAAE,OAAO,EAAE,gBAAgB;QACtC,IAAI,EAAE,OAAO,EAAE,IAAI;QACnB,aAAa,EAAE,OAAO,EAAE,cAAc;QACtC,gBAAgB,EAAE,OAAO,EAAE,iBAAiB;QAC5C,cAAc,EAAE,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,iCAAiC;QAC5E,wDAAwD;QACxD,gCAAgC;QAChC,yBAAyB;KAC5B,CAAC;IACF,QAAQ,QAAQ,EAAE,CAAC;QACf,KAAK,cAAc,CAAC,yBAAyB;YACzC,OAAO;gBACH,GAAG,gBAAgB;gBACnB,QAAQ,EAAE,2BAA2B;gBACrC,UAAU,EAAE;oBACR,SAAS,EAAE,OAAO,EAAE,UAAU;iBACjC;aACJ,CAAC;QACN,KAAK,cAAc,CAAC,2BAA2B;YAC3C,OAAO;gBACH,GAAG,gBAAgB;gBACnB,QAAQ,EAAE,6BAA6B;gBACvC,UAAU,EAAE;oBACR,SAAS,EAAE,OAAO,EAAE,UAAU;iBACjC;aACJ,CAAC;QACN,KAAK,cAAc,CAAC,gBAAgB;YAChC,OAAO;gBACH,GAAG,gBAAgB;gBACnB,QAAQ,EAAE,kBAAkB;gBAC5B,UAAU,EAAE;oBACR,SAAS,EAAE,OAAO,EAAE,UAAU;iBACjC;aACJ,CAAC;QACN,KAAK,cAAc,CAAC,kBAAkB;YAClC,OAAO;gBACH,GAAG,gBAAgB;gBACnB,QAAQ,EAAE,oBAAoB;gBAC9B,UAAU,EAAE;oBACR,SAAS,EAAE,OAAO,EAAE,UAAU;iBACjC;aACJ,CAAC;QACN,KAAK,cAAc,CAAC,UAAU;YAC1B,OAAO;gBACH,GAAG,gBAAgB;gBACnB,6DAA6D;gBAC7D,YAAY,EAAE,OAAO,EAAE,aAAa;gBACpC,WAAW,EAAE,OAAO,EAAE,YAAY;gBAClC,aAAa,EAAE,OAAO,EAAE,cAAc;aACzC,CAAC;QACN,KAAK,cAAc,CAAC,qBAAqB,CAAC;QAC1C,KAAK,cAAc,CAAC,wBAAwB,CAAC;QAC7C,KAAK,cAAc,CAAC,sBAAsB,CAAC;QAC3C,KAAK,cAAc,CAAC,mBAAmB;YACnC,OAAO;gBACH,GAAG,gBAAgB;aACtB,CAAC;QACN;YACI,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IACnD,CAAC;AACL,CAAC;AAED,MAAM,OAAO,qBAAqB;IAC9B,KAAK,CAAU;IAEf,YAAY,OAAe;QACvB,IAAI,CAAC,KAAK,GAAG;YACT,EAAE,EAAE,OAAO;YACX,IAAI,EAAE,OAAO;YACb,QAAQ,EAAE,UAAU;YACpB,IAAI,EAAE,SAAS,CAAC,KAAK;YACrB,UAAU,EAAE,KAAK;SACpB,CAAC;IACN,CAAC;IAED,KAAK,CAAC,YAAY,CACd,OAAuB,EACvB,QAAyB,EACzB,OAAyB;QAEzB,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACxC,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC5C,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QAE3C,MAAM,MAAM,GAAiB;YACzB,MAAM,EAAE,EAAE;SACb,CAAC;QAEF,yFAAyF;QACzF,2EAA2E;QAC3E,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,MAAM,IAAI,GAAa,EAAE,CAAC;QAC1B,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,MAAM,QAAQ,GAAa,EAAE,CAAC;QAE9B,MAAM,SAAS,GAAI,OAAO,CAAC,aAA+B,EAAE,SAAS,CAAC;QACtE,MAAM,aAAa,GAAG,OAAO,CAAC,aAA8B,CAAC;QAE7D,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YACzB,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC,MAAM,EAAE,CAAC;gBACjC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC7B,CAAC;iBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC,MAAM,EAAE,CAAC;gBACxC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC7B,CAAC;iBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC,QAAQ,EAAE,CAAC;gBAC1C,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC/B,CAAC;iBAAM,CAAC;gBACJ,yDAAyD;gBACzD,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC3B,CAAC;YACD,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;gBACZ,0BAA0B;gBAC1B,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;oBAC1B,MAAM,CAAC,eAAe,GAAG,EAAE,CAAC;gBAChC,CAAC;gBAED,kDAAkD;gBAClD,sGAAsG;gBACtG,wCAAwC;gBACxC,MAAM,KAAK,GAAG,MAAM,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC;gBAChD,KAAK,MAAM,GAAG,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;oBAC1B,IAAI,GAAG,CAAC,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;wBACnC,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC,IAAI,EAAE,CAAC;4BAC/B,+CAA+C;4BAC/C,IAAI,aAAa,EAAE,SAAS,EAAE,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;gCAClD,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC;oCACxB,aAAa,EAAE,oBAAoB;oCACnC,WAAW,EAAE,KAAK;oCAClB,cAAc,EAAE;wCACZ,kBAAkB,EAAE,MAAM,kBAAkB,CAAC,MAAM,GAAG,CAAC,SAAS,EAAE,CAAC;qCACtE;iCACJ,CAAC,CAAC;gCACH,iDAAiD;gCACjD,IAAI,SAAS,KAAK,cAAc,CAAC,uBAAuB,EAAE,CAAC;oCACvD,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC;wCACxB,aAAa,EAAE,qBAAqB;wCACpC,WAAW,EAAE,KAAK;wCAClB,eAAe,EAAE;4CACb,QAAQ,EAAE,SAAS;4CACnB,QAAQ,EAAE,aAAa,EAAE,aAAa;yCACzC;qCACJ,CAAC,CAAC;gCACP,CAAC;4BACL,CAAC;iCAAM,IACF,OAAO,CAAC,aAA+B,EAAE,SAAS;gCACnD,cAAc,CAAC,qBAAqB,EACtC,CAAC;gCACC,yCAAyC;gCACzC,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;oCACd,qDAAqD;oCACrD,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC;wCACxB,aAAa,EAAE,wBAAwB;wCACvC,WAAW,EAAE,KAAK;wCAClB,cAAc,EAAE;4CACZ,kBAAkB,EAAE,MAAM,kBAAkB,CAAC,MAAM,GAAG,CAAC,SAAS,EAAE,CAAC;yCACtE;wCACD,kBAAkB,EAAE;4CAChB,WAAW,EACP,aAAa,EAAE,WAAW,KAAK,wBAAwB;gDACnD,CAAC,CAAC,wBAAwB;gDAC1B,CAAC,CAAC,oBAAoB;4CAC9B,6BAA6B,EAAE,aAAa,EAAE,uBAAuB;yCACxE;qCACJ,CAAC,CAAC;gCACP,CAAC;qCAAM,CAAC;oCACJ,iBAAiB;oCACjB,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC;wCACxB,aAAa,EAAE,wBAAwB;wCACvC,WAAW,EAAE,KAAK;wCAClB,cAAc,EAAE;4CACZ,kBAAkB,EAAE,MAAM,kBAAkB,CAAC,MAAM,GAAG,CAAC,SAAS,EAAE,CAAC;yCACtE;wCACD,kBAAkB,EAAE;4CAChB,kBAAkB,EAAE,MAAM,CAAC,kBAAkB,IAAI,GAAG,CAAC,OAAO;4CAC5D,WAAW,EAAE,aAAa,EAAE,WAAW,IAAI,sBAAsB;yCACpE;qCACJ,CAAC,CAAC;gCACP,CAAC;4BACL,CAAC;iCAAM,IACF,OAAO,CAAC,aAA+B,EAAE,SAAS;gCACnD,cAAc,CAAC,mBAAmB,EACpC,CAAC;gCACC,eAAe;gCACf,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC;oCACxB,aAAa,EAAE,sBAAsB;oCACrC,WAAW,EAAE,KAAK;oCAClB,cAAc,EAAE;wCACZ,kBAAkB,EAAE,MAAM,kBAAkB,CAAC,MAAM,GAAG,CAAC,SAAS,EAAE,CAAC;qCACtE;oCACD,gBAAgB,EAAE;wCACd,gBAAgB,EAAE,MAAM,CAAC,kBAAkB,IAAI,GAAG,CAAC,OAAO;qCAC7D;iCACJ,CAAC,CAAC;4BACP,CAAC;iCAAM,IACF,OAAO,CAAC,aAA+B,EAAE,SAAS;gCACnD,cAAc,CAAC,wBAAwB,EACzC,CAAC;gCACC,iBAAiB;gCACjB,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC;oCACxB,aAAa,EAAE,wBAAwB;oCACvC,WAAW,EAAE,KAAK;oCAClB,cAAc,EAAE;wCACZ,kBAAkB,EAAE,MAAM,kBAAkB,CAAC,MAAM,GAAG,CAAC,SAAS,EAAE,CAAC;qCACtE;oCACD,kBAAkB,EAAE;wCAChB,WAAW,EACP,aAAa,EAAE,WAAW,KAAK,wBAAwB;4CACnD,CAAC,CAAC,wBAAwB;4CAC1B,CAAC,CAAC,oBAAoB;wCAC9B,6BAA6B,EAAE,aAAa,EAAE,uBAAuB;qCACxE;iCACJ,CAAC,CAAC;4BACP,CAAC;iCAAM,IACF,OAAO,CAAC,aAA+B,EAAE,SAAS;gCACnD,cAAc,CAAC,sBAAsB,EACvC,CAAC;gCACC,iBAAiB;gCACjB,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC;oCACxB,aAAa,EAAE,oBAAoB;oCACnC,WAAW,EAAE,KAAK;oCAClB,cAAc,EAAE;wCACZ,kBAAkB,EAAE,MAAM,kBAAkB,CAAC,MAAM,GAAG,CAAC,SAAS,EAAE,CAAC;qCACtE;iCACJ,CAAC,CAAC;4BACP,CAAC;wBACL,CAAC;wBACD,gDAAgD;wBAChD,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC,IAAI,IAAI,SAAS,KAAK,cAAc,CAAC,uBAAuB,EAAE,CAAC;4BACvF,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC;gCACxB,aAAa,EAAE,qBAAqB;gCACpC,WAAW,EAAE,KAAK;gCAClB,cAAc,EAAE;oCACZ,kBAAkB,EAAE,MAAM,kBAAkB,CAAC,MAAM,GAAG,CAAC,SAAS,EAAE,CAAC;iCACtE;gCACD,eAAe,EAAE;oCACb,QAAQ,EAAE,SAAS;oCACnB,QAAQ,EAAE,aAAa,EAAE,aAAa;iCACzC;6BACJ,CAAC,CAAC;wBACP,CAAC;oBACL,CAAC;gBACL,CAAC;YACL,CAAC;QACL,CAAC;QAED,oCAAoC;QACpC,MAAM,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAE5F,iBAAiB;QACjB,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtB,MAAM,CAAC,cAAc,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAEpB,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,sBAAsB,CACxB,MAAsB,EACtB,MAAoB,EACpB,OAAyB;QAEzB,IACI,OAAO,CAAC,aAAa,EAAE,UAAU,KAAK,SAAS;YAC/C,OAAO,CAAC,aAAa,EAAE,UAAU,KAAK,iBAAiB,EACzD,CAAC;YACC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,aAAa,EAAE,EAAE,sBAAsB,CAAC,CAAC;QACpF,CAAC;QACD,OAAO,CAAC,aAAa,GAAG,OAAO,CAAC,aAA0C,CAAC;QAE3E,MAAM,QAAQ,GAAW,OAAO,CAAC,aAAa,EAAE,SAAS,IAAI,cAAc,CAAC,UAAU,CAAC;QAEvF,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,QAAQ,EAAE,CAAC,CAAC;QAE7C,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;QAEvD,gCAAgC;QAChC,iEAAiE;QACjE,MAAM,QAAQ,GAAG,YAAY,MAAM,CAAC,OAAO,CAAC,OAAO,mDAAmD,SAAS,EAAE,CAAC;QAElH,MAAM,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC9C,IAAI,CAAC,aAAa,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC/C,CAAC;QACD,MAAM,SAAS,GAAG,CAAC,aAAa,CAAC,CAAC;QAElC,IAAI,SAAS,GAAG,mBAAmB,CAC/B,QAAQ,EACR,OAAO,CAAC,aAAa,IAAI,EAAE,UAAU,EAAE,iBAAiB,EAAE,CACW,CAAC;QAC1E,SAAS,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,SAAS,CAAC;QAE9D,MAAM,cAAc,GAAG,OAAO,CAAC,aAAa,EAAE,gBAAgB,IAAI,CAAC,CAAC;QAEpE,8BAA8B;QAC9B,SAAS,GAAG,MAAM,CAAC,WAAW,CAC1B,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAC5C,CAAC;QAEtB,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAE9C,MAAM,OAAO,GAAsD;YAC/D,QAAQ;YACR,SAAS;YACT,UAAU;SACb,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,eAAe,EAAE,CAAC;QAE9C,kBAAkB;QAClB,MAAM,CAAC,QAAQ,CAAC,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,MAAM,GAAG,cAAc,EAAE,CAAC,CAAC,CAAC,uCAAuC;QAC/H,MAAM,WAAW,GAAG,QAAQ,CAAC,WAAW,CAAC;QAEzC,IAAI,CAAC,WAAW,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC5C,CAAC;QAED,iDAAiD;QACjD,MAAM,MAAM,GAAa,WAAW,CAAC,GAAG,CACpC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,MAAM,EAAE,kBAAkB,EAAE,WAAW,IAAI,EAAE,CACxF,CAAC;QAEF,OAAO;YACH,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBAC3B,IAAI,EAAE,OAAgB;gBACtB,KAAK,EAAE,KAAK;aACf,CAAC,CAAC;SACN,CAAC;IACN,CAAC;CACJ"}
|
|
@@ -1,11 +1,11 @@
|
|
|
1
|
-
import { AIModel, Completion, CompletionChunkObject, ExecutionOptions, PromptOptions, PromptSegment } from
|
|
2
|
-
import { VertexAIDriver } from
|
|
3
|
-
import { ModelDefinition } from
|
|
1
|
+
import { type AIModel, type Completion, type CompletionChunkObject, type ExecutionOptions, type PromptOptions, type PromptSegment } from '@llumiverse/core';
|
|
2
|
+
import type { VertexAIDriver } from '../index.js';
|
|
3
|
+
import type { ModelDefinition } from '../models.js';
|
|
4
4
|
interface LLamaMessage {
|
|
5
5
|
role: string;
|
|
6
6
|
content: string;
|
|
7
7
|
}
|
|
8
|
-
interface LLamaPrompt {
|
|
8
|
+
export interface LLamaPrompt {
|
|
9
9
|
messages: LLamaMessage[];
|
|
10
10
|
}
|
|
11
11
|
export declare class LLamaModelDefinition implements ModelDefinition<LLamaPrompt> {
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"llama.d.ts","sourceRoot":"","sources":["../../../src/vertexai/models/llama.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,KAAK,OAAO,EACZ,KAAK,UAAU,EACf,KAAK,qBAAqB,EAC1B,KAAK,gBAAgB,EAErB,KAAK,aAAa,EAElB,KAAK,aAAa,EAErB,MAAM,kBAAkB,CAAC;AAG1B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAEpD,UAAU,YAAY;IAClB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,WAAW;IACxB,QAAQ,EAAE,YAAY,EAAE,CAAC;CAC5B;AAqED,qBAAa,oBAAqB,YAAW,eAAe,CAAC,WAAW,CAAC;IACrE,KAAK,EAAE,OAAO,CAAC;gBAEH,OAAO,EAAE,MAAM;IAW3B,mBAAmB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM;IASxC,YAAY,CACd,OAAO,EAAE,cAAc,EACvB,QAAQ,EAAE,aAAa,EAAE,EACzB,OAAO,EAAE,aAAa,GACvB,OAAO,CAAC,WAAW,CAAC;IAwCjB,qBAAqB,CACvB,MAAM,EAAE,cAAc,EACtB,MAAM,EAAE,WAAW,EACnB,OAAO,EAAE,gBAAgB,GAC1B,OAAO,CAAC,UAAU,CAAC;IA6DhB,2BAA2B,CAC7B,MAAM,EAAE,cAAc,EACtB,MAAM,EAAE,WAAW,EACnB,OAAO,EAAE,gBAAgB,GAC1B,OAAO,CAAC,aAAa,CAAC,qBAAqB,CAAC,CAAC;CAuDnD"}
|
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { ModelType, PromptRole } from
|
|
2
|
-
import { transformSSEStream } from
|
|
1
|
+
import { ModelType, PromptRole, } from '@llumiverse/core';
|
|
2
|
+
import { transformSSEStream } from '@llumiverse/core/async';
|
|
3
3
|
/**
|
|
4
4
|
* Convert a stream to a string
|
|
5
5
|
*/
|
|
@@ -53,7 +53,7 @@ export class LLamaModelDefinition {
|
|
|
53
53
|
let messageContent = segment.content || '';
|
|
54
54
|
if (segment.files && segment.files.length > 0) {
|
|
55
55
|
for (const file of segment.files) {
|
|
56
|
-
if (file.mime_type?.startsWith(
|
|
56
|
+
if (file.mime_type?.startsWith('text/')) {
|
|
57
57
|
const fileStream = await file.getStream();
|
|
58
58
|
const fileContent = await streamToString(fileStream);
|
|
59
59
|
messageContent += `\n\nFile content:\n${fileContent}`;
|
|
@@ -62,13 +62,13 @@ export class LLamaModelDefinition {
|
|
|
62
62
|
}
|
|
63
63
|
messages.push({
|
|
64
64
|
role: role,
|
|
65
|
-
content: messageContent
|
|
65
|
+
content: messageContent,
|
|
66
66
|
});
|
|
67
67
|
}
|
|
68
68
|
if (options.result_schema) {
|
|
69
69
|
messages.push({
|
|
70
70
|
role: 'user',
|
|
71
|
-
content:
|
|
71
|
+
content: `The answer must be a JSON object using the following JSON Schema:\n${JSON.stringify(options.result_schema)}`,
|
|
72
72
|
});
|
|
73
73
|
}
|
|
74
74
|
// Return the prompt in the format expected by Llama MaaS API
|
|
@@ -77,7 +77,7 @@ export class LLamaModelDefinition {
|
|
|
77
77
|
};
|
|
78
78
|
}
|
|
79
79
|
async requestTextCompletion(driver, prompt, options) {
|
|
80
|
-
const splits = options.model.split(
|
|
80
|
+
const splits = options.model.split('/');
|
|
81
81
|
const modelName = splits[splits.length - 1];
|
|
82
82
|
let conversation = updateConversation(options.conversation, prompt);
|
|
83
83
|
const modelOptions = options.model_options;
|
|
@@ -94,41 +94,43 @@ export class LLamaModelDefinition {
|
|
|
94
94
|
google: {
|
|
95
95
|
model_safety_settings: {
|
|
96
96
|
enabled: false,
|
|
97
|
-
llama_guard_settings: {}
|
|
98
|
-
}
|
|
99
|
-
}
|
|
100
|
-
}
|
|
97
|
+
llama_guard_settings: {},
|
|
98
|
+
},
|
|
99
|
+
},
|
|
100
|
+
},
|
|
101
101
|
};
|
|
102
102
|
// Make POST request to the Llama MaaS API
|
|
103
103
|
const region = this.getLlamaModelRegion(modelName);
|
|
104
104
|
const client = driver.getLLamaClient(region);
|
|
105
105
|
const openaiEndpoint = `endpoints/openapi/chat/completions`;
|
|
106
|
-
const result = await client.post(openaiEndpoint, {
|
|
107
|
-
payload
|
|
108
|
-
});
|
|
106
|
+
const result = (await client.post(openaiEndpoint, {
|
|
107
|
+
payload,
|
|
108
|
+
}));
|
|
109
109
|
// Extract response data
|
|
110
110
|
const assistantMessage = result?.choices[0]?.message;
|
|
111
111
|
const text = assistantMessage?.content;
|
|
112
112
|
// Update conversation with the response
|
|
113
113
|
conversation = updateConversation(conversation, {
|
|
114
|
-
messages: [
|
|
114
|
+
messages: [
|
|
115
|
+
{
|
|
115
116
|
role: assistantMessage?.role,
|
|
116
|
-
content: text
|
|
117
|
-
}
|
|
117
|
+
content: text,
|
|
118
|
+
},
|
|
119
|
+
],
|
|
118
120
|
});
|
|
119
121
|
return {
|
|
120
|
-
result: [{ type:
|
|
122
|
+
result: [{ type: 'text', value: text }],
|
|
121
123
|
token_usage: {
|
|
122
124
|
prompt: result.usage.prompt_tokens,
|
|
123
125
|
result: result.usage.completion_tokens,
|
|
124
|
-
total: result.usage.total_tokens
|
|
126
|
+
total: result.usage.total_tokens,
|
|
125
127
|
},
|
|
126
128
|
finish_reason: result.choices[0].finish_reason,
|
|
127
|
-
conversation
|
|
129
|
+
conversation,
|
|
128
130
|
};
|
|
129
131
|
}
|
|
130
132
|
async requestTextCompletionStream(driver, prompt, options) {
|
|
131
|
-
const splits = options.model.split(
|
|
133
|
+
const splits = options.model.split('/');
|
|
132
134
|
const modelName = splits[splits.length - 1];
|
|
133
135
|
const conversation = updateConversation(options.conversation, prompt);
|
|
134
136
|
const modelOptions = options.model_options;
|
|
@@ -145,10 +147,10 @@ export class LLamaModelDefinition {
|
|
|
145
147
|
google: {
|
|
146
148
|
model_safety_settings: {
|
|
147
149
|
enabled: false,
|
|
148
|
-
llama_guard_settings: {}
|
|
149
|
-
}
|
|
150
|
-
}
|
|
151
|
-
}
|
|
150
|
+
llama_guard_settings: {},
|
|
151
|
+
},
|
|
152
|
+
},
|
|
153
|
+
},
|
|
152
154
|
};
|
|
153
155
|
// Make POST request to the Llama MaaS API
|
|
154
156
|
//TODO: Fix error handling with the fetch client, errors will return a empty response
|
|
@@ -156,22 +158,24 @@ export class LLamaModelDefinition {
|
|
|
156
158
|
const region = this.getLlamaModelRegion(modelName);
|
|
157
159
|
const client = driver.getLLamaClient(region);
|
|
158
160
|
const openaiEndpoint = `endpoints/openapi/chat/completions`;
|
|
159
|
-
const stream = await client.post(openaiEndpoint, {
|
|
161
|
+
const stream = (await client.post(openaiEndpoint, {
|
|
160
162
|
payload,
|
|
161
|
-
reader: 'sse'
|
|
162
|
-
});
|
|
163
|
+
reader: 'sse',
|
|
164
|
+
}));
|
|
163
165
|
return transformSSEStream(stream, (data) => {
|
|
164
166
|
const json = JSON.parse(data);
|
|
165
167
|
const choice = json.choices?.[0];
|
|
166
168
|
const content = choice?.delta?.content ?? '';
|
|
167
169
|
return {
|
|
168
|
-
result: content ? [{ type:
|
|
170
|
+
result: content ? [{ type: 'text', value: content }] : [],
|
|
169
171
|
finish_reason: choice?.finish_reason,
|
|
170
|
-
token_usage: json.usage
|
|
171
|
-
|
|
172
|
-
|
|
173
|
-
|
|
174
|
-
|
|
172
|
+
token_usage: json.usage
|
|
173
|
+
? {
|
|
174
|
+
prompt: json.usage.prompt_tokens,
|
|
175
|
+
result: json.usage.completion_tokens,
|
|
176
|
+
total: json.usage.total_tokens,
|
|
177
|
+
}
|
|
178
|
+
: undefined,
|
|
175
179
|
};
|
|
176
180
|
});
|
|
177
181
|
}
|