modelfusion 0.0.44
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/LICENSE +21 -0
- package/README.md +429 -0
- package/composed-function/index.cjs +22 -0
- package/composed-function/index.d.ts +6 -0
- package/composed-function/index.js +6 -0
- package/composed-function/summarize/SummarizationFunction.cjs +2 -0
- package/composed-function/summarize/SummarizationFunction.d.ts +4 -0
- package/composed-function/summarize/SummarizationFunction.js +1 -0
- package/composed-function/summarize/summarizeRecursively.cjs +19 -0
- package/composed-function/summarize/summarizeRecursively.d.ts +11 -0
- package/composed-function/summarize/summarizeRecursively.js +15 -0
- package/composed-function/summarize/summarizeRecursivelyWithTextGenerationAndTokenSplitting.cjs +29 -0
- package/composed-function/summarize/summarizeRecursivelyWithTextGenerationAndTokenSplitting.d.ts +24 -0
- package/composed-function/summarize/summarizeRecursivelyWithTextGenerationAndTokenSplitting.js +25 -0
- package/composed-function/use-tool/NoSuchToolError.cjs +17 -0
- package/composed-function/use-tool/NoSuchToolError.d.ts +4 -0
- package/composed-function/use-tool/NoSuchToolError.js +13 -0
- package/composed-function/use-tool/Tool.cjs +43 -0
- package/composed-function/use-tool/Tool.d.ts +15 -0
- package/composed-function/use-tool/Tool.js +39 -0
- package/composed-function/use-tool/useTool.cjs +59 -0
- package/composed-function/use-tool/useTool.d.ts +36 -0
- package/composed-function/use-tool/useTool.js +54 -0
- package/cost/Cost.cjs +38 -0
- package/cost/Cost.d.ts +16 -0
- package/cost/Cost.js +34 -0
- package/cost/CostCalculator.cjs +2 -0
- package/cost/CostCalculator.d.ts +8 -0
- package/cost/CostCalculator.js +1 -0
- package/cost/calculateCost.cjs +28 -0
- package/cost/calculateCost.d.ts +7 -0
- package/cost/calculateCost.js +24 -0
- package/cost/index.cjs +19 -0
- package/cost/index.d.ts +3 -0
- package/cost/index.js +3 -0
- package/index.cjs +25 -0
- package/index.d.ts +9 -0
- package/index.js +9 -0
- package/model-function/AbstractModel.cjs +22 -0
- package/model-function/AbstractModel.d.ts +12 -0
- package/model-function/AbstractModel.js +18 -0
- package/model-function/FunctionOptions.cjs +2 -0
- package/model-function/FunctionOptions.d.ts +6 -0
- package/model-function/FunctionOptions.js +1 -0
- package/model-function/Model.cjs +2 -0
- package/model-function/Model.d.ts +23 -0
- package/model-function/Model.js +1 -0
- package/model-function/ModelCallEvent.cjs +2 -0
- package/model-function/ModelCallEvent.d.ts +18 -0
- package/model-function/ModelCallEvent.js +1 -0
- package/model-function/ModelCallEventSource.cjs +42 -0
- package/model-function/ModelCallEventSource.d.ts +13 -0
- package/model-function/ModelCallEventSource.js +38 -0
- package/model-function/ModelCallObserver.cjs +2 -0
- package/model-function/ModelCallObserver.d.ts +5 -0
- package/model-function/ModelCallObserver.js +1 -0
- package/model-function/ModelInformation.cjs +2 -0
- package/model-function/ModelInformation.d.ts +4 -0
- package/model-function/ModelInformation.js +1 -0
- package/model-function/SuccessfulModelCall.cjs +22 -0
- package/model-function/SuccessfulModelCall.d.ts +9 -0
- package/model-function/SuccessfulModelCall.js +18 -0
- package/model-function/embed-text/TextEmbeddingEvent.cjs +2 -0
- package/model-function/embed-text/TextEmbeddingEvent.d.ts +23 -0
- package/model-function/embed-text/TextEmbeddingEvent.js +1 -0
- package/model-function/embed-text/TextEmbeddingModel.cjs +2 -0
- package/model-function/embed-text/TextEmbeddingModel.d.ts +18 -0
- package/model-function/embed-text/TextEmbeddingModel.js +1 -0
- package/model-function/embed-text/embedText.cjs +90 -0
- package/model-function/embed-text/embedText.d.ts +33 -0
- package/model-function/embed-text/embedText.js +85 -0
- package/model-function/executeCall.cjs +60 -0
- package/model-function/executeCall.d.ts +27 -0
- package/model-function/executeCall.js +56 -0
- package/model-function/generate-image/ImageGenerationEvent.cjs +2 -0
- package/model-function/generate-image/ImageGenerationEvent.d.ts +22 -0
- package/model-function/generate-image/ImageGenerationEvent.js +1 -0
- package/model-function/generate-image/ImageGenerationModel.cjs +2 -0
- package/model-function/generate-image/ImageGenerationModel.d.ts +8 -0
- package/model-function/generate-image/ImageGenerationModel.js +1 -0
- package/model-function/generate-image/generateImage.cjs +63 -0
- package/model-function/generate-image/generateImage.d.ts +23 -0
- package/model-function/generate-image/generateImage.js +59 -0
- package/model-function/generate-json/GenerateJsonModel.cjs +2 -0
- package/model-function/generate-json/GenerateJsonModel.d.ts +10 -0
- package/model-function/generate-json/GenerateJsonModel.js +1 -0
- package/model-function/generate-json/GenerateJsonOrTextModel.cjs +2 -0
- package/model-function/generate-json/GenerateJsonOrTextModel.d.ts +18 -0
- package/model-function/generate-json/GenerateJsonOrTextModel.js +1 -0
- package/model-function/generate-json/JsonGenerationEvent.cjs +2 -0
- package/model-function/generate-json/JsonGenerationEvent.d.ts +22 -0
- package/model-function/generate-json/JsonGenerationEvent.js +1 -0
- package/model-function/generate-json/NoSuchSchemaError.cjs +17 -0
- package/model-function/generate-json/NoSuchSchemaError.d.ts +4 -0
- package/model-function/generate-json/NoSuchSchemaError.js +13 -0
- package/model-function/generate-json/SchemaDefinition.cjs +2 -0
- package/model-function/generate-json/SchemaDefinition.d.ts +6 -0
- package/model-function/generate-json/SchemaDefinition.js +1 -0
- package/model-function/generate-json/SchemaValidationError.cjs +36 -0
- package/model-function/generate-json/SchemaValidationError.d.ts +11 -0
- package/model-function/generate-json/SchemaValidationError.js +32 -0
- package/model-function/generate-json/generateJson.cjs +61 -0
- package/model-function/generate-json/generateJson.d.ts +9 -0
- package/model-function/generate-json/generateJson.js +57 -0
- package/model-function/generate-json/generateJsonOrText.cjs +74 -0
- package/model-function/generate-json/generateJsonOrText.d.ts +25 -0
- package/model-function/generate-json/generateJsonOrText.js +70 -0
- package/model-function/generate-text/AsyncQueue.cjs +66 -0
- package/model-function/generate-text/AsyncQueue.d.ts +17 -0
- package/model-function/generate-text/AsyncQueue.js +62 -0
- package/model-function/generate-text/DeltaEvent.cjs +2 -0
- package/model-function/generate-text/DeltaEvent.d.ts +7 -0
- package/model-function/generate-text/DeltaEvent.js +1 -0
- package/model-function/generate-text/TextDeltaEventSource.cjs +54 -0
- package/model-function/generate-text/TextDeltaEventSource.d.ts +5 -0
- package/model-function/generate-text/TextDeltaEventSource.js +46 -0
- package/model-function/generate-text/TextGenerationEvent.cjs +2 -0
- package/model-function/generate-text/TextGenerationEvent.d.ts +22 -0
- package/model-function/generate-text/TextGenerationEvent.js +1 -0
- package/model-function/generate-text/TextGenerationModel.cjs +2 -0
- package/model-function/generate-text/TextGenerationModel.d.ts +42 -0
- package/model-function/generate-text/TextGenerationModel.js +1 -0
- package/model-function/generate-text/TextStreamingEvent.cjs +2 -0
- package/model-function/generate-text/TextStreamingEvent.d.ts +22 -0
- package/model-function/generate-text/TextStreamingEvent.js +1 -0
- package/model-function/generate-text/extractTextDeltas.cjs +23 -0
- package/model-function/generate-text/extractTextDeltas.d.ts +7 -0
- package/model-function/generate-text/extractTextDeltas.js +19 -0
- package/model-function/generate-text/generateText.cjs +67 -0
- package/model-function/generate-text/generateText.d.ts +20 -0
- package/model-function/generate-text/generateText.js +63 -0
- package/model-function/generate-text/parseEventSourceReadableStream.cjs +30 -0
- package/model-function/generate-text/parseEventSourceReadableStream.d.ts +8 -0
- package/model-function/generate-text/parseEventSourceReadableStream.js +26 -0
- package/model-function/generate-text/streamText.cjs +115 -0
- package/model-function/generate-text/streamText.d.ts +11 -0
- package/model-function/generate-text/streamText.js +111 -0
- package/model-function/index.cjs +47 -0
- package/model-function/index.d.ts +31 -0
- package/model-function/index.js +31 -0
- package/model-function/tokenize-text/Tokenizer.cjs +2 -0
- package/model-function/tokenize-text/Tokenizer.d.ts +19 -0
- package/model-function/tokenize-text/Tokenizer.js +1 -0
- package/model-function/tokenize-text/countTokens.cjs +10 -0
- package/model-function/tokenize-text/countTokens.d.ts +5 -0
- package/model-function/tokenize-text/countTokens.js +6 -0
- package/model-function/transcribe-audio/TranscriptionEvent.cjs +2 -0
- package/model-function/transcribe-audio/TranscriptionEvent.d.ts +22 -0
- package/model-function/transcribe-audio/TranscriptionEvent.js +1 -0
- package/model-function/transcribe-audio/TranscriptionModel.cjs +2 -0
- package/model-function/transcribe-audio/TranscriptionModel.d.ts +8 -0
- package/model-function/transcribe-audio/TranscriptionModel.js +1 -0
- package/model-function/transcribe-audio/transcribe.cjs +62 -0
- package/model-function/transcribe-audio/transcribe.d.ts +22 -0
- package/model-function/transcribe-audio/transcribe.js +58 -0
- package/model-provider/automatic1111/Automatic1111Error.cjs +39 -0
- package/model-provider/automatic1111/Automatic1111Error.d.ts +31 -0
- package/model-provider/automatic1111/Automatic1111Error.js +31 -0
- package/model-provider/automatic1111/Automatic1111ImageGenerationModel.cjs +76 -0
- package/model-provider/automatic1111/Automatic1111ImageGenerationModel.d.ts +54 -0
- package/model-provider/automatic1111/Automatic1111ImageGenerationModel.js +72 -0
- package/model-provider/automatic1111/index.cjs +20 -0
- package/model-provider/automatic1111/index.d.ts +2 -0
- package/model-provider/automatic1111/index.js +2 -0
- package/model-provider/cohere/CohereError.cjs +36 -0
- package/model-provider/cohere/CohereError.d.ts +22 -0
- package/model-provider/cohere/CohereError.js +28 -0
- package/model-provider/cohere/CohereTextEmbeddingModel.cjs +172 -0
- package/model-provider/cohere/CohereTextEmbeddingModel.d.ts +119 -0
- package/model-provider/cohere/CohereTextEmbeddingModel.js +165 -0
- package/model-provider/cohere/CohereTextGenerationModel.cjs +283 -0
- package/model-provider/cohere/CohereTextGenerationModel.d.ts +203 -0
- package/model-provider/cohere/CohereTextGenerationModel.js +276 -0
- package/model-provider/cohere/CohereTokenizer.cjs +136 -0
- package/model-provider/cohere/CohereTokenizer.d.ts +118 -0
- package/model-provider/cohere/CohereTokenizer.js +129 -0
- package/model-provider/cohere/index.cjs +22 -0
- package/model-provider/cohere/index.d.ts +4 -0
- package/model-provider/cohere/index.js +4 -0
- package/model-provider/huggingface/HuggingFaceError.cjs +52 -0
- package/model-provider/huggingface/HuggingFaceError.d.ts +22 -0
- package/model-provider/huggingface/HuggingFaceError.js +44 -0
- package/model-provider/huggingface/HuggingFaceTextGenerationModel.cjs +174 -0
- package/model-provider/huggingface/HuggingFaceTextGenerationModel.d.ts +75 -0
- package/model-provider/huggingface/HuggingFaceTextGenerationModel.js +167 -0
- package/model-provider/huggingface/index.cjs +20 -0
- package/model-provider/huggingface/index.d.ts +2 -0
- package/model-provider/huggingface/index.js +2 -0
- package/model-provider/index.cjs +22 -0
- package/model-provider/index.d.ts +6 -0
- package/model-provider/index.js +6 -0
- package/model-provider/llamacpp/LlamaCppError.cjs +52 -0
- package/model-provider/llamacpp/LlamaCppError.d.ts +22 -0
- package/model-provider/llamacpp/LlamaCppError.js +44 -0
- package/model-provider/llamacpp/LlamaCppTextEmbeddingModel.cjs +96 -0
- package/model-provider/llamacpp/LlamaCppTextEmbeddingModel.d.ts +40 -0
- package/model-provider/llamacpp/LlamaCppTextEmbeddingModel.js +89 -0
- package/model-provider/llamacpp/LlamaCppTextGenerationModel.cjs +245 -0
- package/model-provider/llamacpp/LlamaCppTextGenerationModel.d.ts +399 -0
- package/model-provider/llamacpp/LlamaCppTextGenerationModel.js +238 -0
- package/model-provider/llamacpp/LlamaCppTokenizer.cjs +64 -0
- package/model-provider/llamacpp/LlamaCppTokenizer.d.ts +38 -0
- package/model-provider/llamacpp/LlamaCppTokenizer.js +57 -0
- package/model-provider/llamacpp/index.cjs +22 -0
- package/model-provider/llamacpp/index.d.ts +4 -0
- package/model-provider/llamacpp/index.js +4 -0
- package/model-provider/openai/OpenAICostCalculator.cjs +71 -0
- package/model-provider/openai/OpenAICostCalculator.d.ts +6 -0
- package/model-provider/openai/OpenAICostCalculator.js +67 -0
- package/model-provider/openai/OpenAIError.cjs +50 -0
- package/model-provider/openai/OpenAIError.d.ts +47 -0
- package/model-provider/openai/OpenAIError.js +42 -0
- package/model-provider/openai/OpenAIImageGenerationModel.cjs +124 -0
- package/model-provider/openai/OpenAIImageGenerationModel.d.ts +113 -0
- package/model-provider/openai/OpenAIImageGenerationModel.js +119 -0
- package/model-provider/openai/OpenAIModelSettings.cjs +2 -0
- package/model-provider/openai/OpenAIModelSettings.d.ts +8 -0
- package/model-provider/openai/OpenAIModelSettings.js +1 -0
- package/model-provider/openai/OpenAITextEmbeddingModel.cjs +171 -0
- package/model-provider/openai/OpenAITextEmbeddingModel.d.ts +122 -0
- package/model-provider/openai/OpenAITextEmbeddingModel.js +162 -0
- package/model-provider/openai/OpenAITextGenerationModel.cjs +326 -0
- package/model-provider/openai/OpenAITextGenerationModel.d.ts +254 -0
- package/model-provider/openai/OpenAITextGenerationModel.js +317 -0
- package/model-provider/openai/OpenAITranscriptionModel.cjs +195 -0
- package/model-provider/openai/OpenAITranscriptionModel.d.ts +196 -0
- package/model-provider/openai/OpenAITranscriptionModel.js +187 -0
- package/model-provider/openai/TikTokenTokenizer.cjs +86 -0
- package/model-provider/openai/TikTokenTokenizer.d.ts +35 -0
- package/model-provider/openai/TikTokenTokenizer.js +82 -0
- package/model-provider/openai/chat/OpenAIChatMessage.cjs +24 -0
- package/model-provider/openai/chat/OpenAIChatMessage.d.ts +26 -0
- package/model-provider/openai/chat/OpenAIChatMessage.js +21 -0
- package/model-provider/openai/chat/OpenAIChatModel.cjs +288 -0
- package/model-provider/openai/chat/OpenAIChatModel.d.ts +344 -0
- package/model-provider/openai/chat/OpenAIChatModel.js +279 -0
- package/model-provider/openai/chat/OpenAIChatPrompt.cjs +143 -0
- package/model-provider/openai/chat/OpenAIChatPrompt.d.ts +108 -0
- package/model-provider/openai/chat/OpenAIChatPrompt.js +135 -0
- package/model-provider/openai/chat/OpenAIChatStreamIterable.cjs +112 -0
- package/model-provider/openai/chat/OpenAIChatStreamIterable.d.ts +19 -0
- package/model-provider/openai/chat/OpenAIChatStreamIterable.js +105 -0
- package/model-provider/openai/chat/countOpenAIChatMessageTokens.cjs +28 -0
- package/model-provider/openai/chat/countOpenAIChatMessageTokens.d.ts +20 -0
- package/model-provider/openai/chat/countOpenAIChatMessageTokens.js +23 -0
- package/model-provider/openai/index.cjs +31 -0
- package/model-provider/openai/index.d.ts +13 -0
- package/model-provider/openai/index.js +12 -0
- package/model-provider/stability/StabilityError.cjs +36 -0
- package/model-provider/stability/StabilityError.d.ts +22 -0
- package/model-provider/stability/StabilityError.js +28 -0
- package/model-provider/stability/StabilityImageGenerationModel.cjs +133 -0
- package/model-provider/stability/StabilityImageGenerationModel.d.ts +95 -0
- package/model-provider/stability/StabilityImageGenerationModel.js +129 -0
- package/model-provider/stability/index.cjs +20 -0
- package/model-provider/stability/index.d.ts +2 -0
- package/model-provider/stability/index.js +2 -0
- package/package.json +87 -0
- package/prompt/InstructionPrompt.cjs +2 -0
- package/prompt/InstructionPrompt.d.ts +7 -0
- package/prompt/InstructionPrompt.js +1 -0
- package/prompt/Llama2PromptMapping.cjs +56 -0
- package/prompt/Llama2PromptMapping.d.ts +10 -0
- package/prompt/Llama2PromptMapping.js +51 -0
- package/prompt/OpenAIChatPromptMapping.cjs +62 -0
- package/prompt/OpenAIChatPromptMapping.d.ts +6 -0
- package/prompt/OpenAIChatPromptMapping.js +57 -0
- package/prompt/PromptMapping.cjs +2 -0
- package/prompt/PromptMapping.d.ts +7 -0
- package/prompt/PromptMapping.js +1 -0
- package/prompt/PromptMappingTextGenerationModel.cjs +88 -0
- package/prompt/PromptMappingTextGenerationModel.d.ts +26 -0
- package/prompt/PromptMappingTextGenerationModel.js +84 -0
- package/prompt/TextPromptMapping.cjs +50 -0
- package/prompt/TextPromptMapping.d.ts +14 -0
- package/prompt/TextPromptMapping.js +45 -0
- package/prompt/chat/ChatPrompt.cjs +2 -0
- package/prompt/chat/ChatPrompt.d.ts +33 -0
- package/prompt/chat/ChatPrompt.js +1 -0
- package/prompt/chat/trimChatPrompt.cjs +50 -0
- package/prompt/chat/trimChatPrompt.d.ts +19 -0
- package/prompt/chat/trimChatPrompt.js +46 -0
- package/prompt/chat/validateChatPrompt.cjs +36 -0
- package/prompt/chat/validateChatPrompt.d.ts +8 -0
- package/prompt/chat/validateChatPrompt.js +31 -0
- package/prompt/index.cjs +25 -0
- package/prompt/index.d.ts +9 -0
- package/prompt/index.js +9 -0
- package/run/ConsoleLogger.cjs +12 -0
- package/run/ConsoleLogger.d.ts +6 -0
- package/run/ConsoleLogger.js +8 -0
- package/run/DefaultRun.cjs +78 -0
- package/run/DefaultRun.d.ts +24 -0
- package/run/DefaultRun.js +74 -0
- package/run/IdMetadata.cjs +2 -0
- package/run/IdMetadata.d.ts +7 -0
- package/run/IdMetadata.js +1 -0
- package/run/Run.cjs +2 -0
- package/run/Run.d.ts +27 -0
- package/run/Run.js +1 -0
- package/run/RunFunction.cjs +2 -0
- package/run/RunFunction.d.ts +13 -0
- package/run/RunFunction.js +1 -0
- package/run/Vector.cjs +2 -0
- package/run/Vector.d.ts +5 -0
- package/run/Vector.js +1 -0
- package/run/index.cjs +22 -0
- package/run/index.d.ts +6 -0
- package/run/index.js +6 -0
- package/text-chunk/TextChunk.cjs +2 -0
- package/text-chunk/TextChunk.d.ts +3 -0
- package/text-chunk/TextChunk.js +1 -0
- package/text-chunk/index.cjs +22 -0
- package/text-chunk/index.d.ts +6 -0
- package/text-chunk/index.js +6 -0
- package/text-chunk/retrieve-text-chunks/TextChunkRetriever.cjs +2 -0
- package/text-chunk/retrieve-text-chunks/TextChunkRetriever.d.ts +8 -0
- package/text-chunk/retrieve-text-chunks/TextChunkRetriever.js +1 -0
- package/text-chunk/retrieve-text-chunks/retrieveTextChunks.cjs +10 -0
- package/text-chunk/retrieve-text-chunks/retrieveTextChunks.d.ts +6 -0
- package/text-chunk/retrieve-text-chunks/retrieveTextChunks.js +6 -0
- package/text-chunk/split/SplitFunction.cjs +2 -0
- package/text-chunk/split/SplitFunction.d.ts +4 -0
- package/text-chunk/split/SplitFunction.js +1 -0
- package/text-chunk/split/splitOnSeparator.cjs +12 -0
- package/text-chunk/split/splitOnSeparator.d.ts +8 -0
- package/text-chunk/split/splitOnSeparator.js +7 -0
- package/text-chunk/split/splitRecursively.cjs +41 -0
- package/text-chunk/split/splitRecursively.d.ts +22 -0
- package/text-chunk/split/splitRecursively.js +33 -0
- package/util/DurationMeasurement.cjs +42 -0
- package/util/DurationMeasurement.d.ts +5 -0
- package/util/DurationMeasurement.js +38 -0
- package/util/ErrorHandler.cjs +2 -0
- package/util/ErrorHandler.d.ts +1 -0
- package/util/ErrorHandler.js +1 -0
- package/util/SafeResult.cjs +2 -0
- package/util/SafeResult.d.ts +8 -0
- package/util/SafeResult.js +1 -0
- package/util/api/AbortError.cjs +9 -0
- package/util/api/AbortError.d.ts +3 -0
- package/util/api/AbortError.js +5 -0
- package/util/api/ApiCallError.cjs +45 -0
- package/util/api/ApiCallError.d.ts +15 -0
- package/util/api/ApiCallError.js +41 -0
- package/util/api/RetryError.cjs +24 -0
- package/util/api/RetryError.d.ts +10 -0
- package/util/api/RetryError.js +20 -0
- package/util/api/RetryFunction.cjs +2 -0
- package/util/api/RetryFunction.d.ts +1 -0
- package/util/api/RetryFunction.js +1 -0
- package/util/api/ThrottleFunction.cjs +2 -0
- package/util/api/ThrottleFunction.d.ts +1 -0
- package/util/api/ThrottleFunction.js +1 -0
- package/util/api/callWithRetryAndThrottle.cjs +7 -0
- package/util/api/callWithRetryAndThrottle.d.ts +7 -0
- package/util/api/callWithRetryAndThrottle.js +3 -0
- package/util/api/postToApi.cjs +103 -0
- package/util/api/postToApi.d.ts +29 -0
- package/util/api/postToApi.js +96 -0
- package/util/api/retryNever.cjs +8 -0
- package/util/api/retryNever.d.ts +4 -0
- package/util/api/retryNever.js +4 -0
- package/util/api/retryWithExponentialBackoff.cjs +48 -0
- package/util/api/retryWithExponentialBackoff.d.ts +10 -0
- package/util/api/retryWithExponentialBackoff.js +44 -0
- package/util/api/throttleMaxConcurrency.cjs +65 -0
- package/util/api/throttleMaxConcurrency.d.ts +7 -0
- package/util/api/throttleMaxConcurrency.js +61 -0
- package/util/api/throttleUnlimitedConcurrency.cjs +8 -0
- package/util/api/throttleUnlimitedConcurrency.d.ts +5 -0
- package/util/api/throttleUnlimitedConcurrency.js +4 -0
- package/util/cosineSimilarity.cjs +26 -0
- package/util/cosineSimilarity.d.ts +11 -0
- package/util/cosineSimilarity.js +22 -0
- package/util/index.cjs +26 -0
- package/util/index.d.ts +10 -0
- package/util/index.js +10 -0
- package/util/never.cjs +6 -0
- package/util/never.d.ts +1 -0
- package/util/never.js +2 -0
- package/util/runSafe.cjs +15 -0
- package/util/runSafe.d.ts +2 -0
- package/util/runSafe.js +11 -0
- package/vector-index/VectorIndex.cjs +2 -0
- package/vector-index/VectorIndex.d.ts +18 -0
- package/vector-index/VectorIndex.js +1 -0
- package/vector-index/VectorIndexSimilarTextChunkRetriever.cjs +57 -0
- package/vector-index/VectorIndexSimilarTextChunkRetriever.d.ts +20 -0
- package/vector-index/VectorIndexSimilarTextChunkRetriever.js +53 -0
- package/vector-index/VectorIndexTextChunkStore.cjs +77 -0
- package/vector-index/VectorIndexTextChunkStore.d.ts +35 -0
- package/vector-index/VectorIndexTextChunkStore.js +73 -0
- package/vector-index/index.cjs +22 -0
- package/vector-index/index.d.ts +6 -0
- package/vector-index/index.js +6 -0
- package/vector-index/memory/MemoryVectorIndex.cjs +63 -0
- package/vector-index/memory/MemoryVectorIndex.d.ts +31 -0
- package/vector-index/memory/MemoryVectorIndex.js +56 -0
- package/vector-index/pinecone/PineconeVectorIndex.cjs +66 -0
- package/vector-index/pinecone/PineconeVectorIndex.d.ts +29 -0
- package/vector-index/pinecone/PineconeVectorIndex.js +62 -0
- package/vector-index/upsertTextChunks.cjs +15 -0
- package/vector-index/upsertTextChunks.d.ts +11 -0
- package/vector-index/upsertTextChunks.js +11 -0
@@ -0,0 +1,115 @@
|
|
1
|
+
"use strict";
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
3
|
+
exports.streamText = void 0;
|
4
|
+
const nanoid_1 = require("nanoid");
|
5
|
+
const DurationMeasurement_js_1 = require("../../util/DurationMeasurement.cjs");
|
6
|
+
const AbortError_js_1 = require("../../util/api/AbortError.cjs");
|
7
|
+
const runSafe_js_1 = require("../../util/runSafe.cjs");
|
8
|
+
const ModelCallEventSource_js_1 = require("../ModelCallEventSource.cjs");
|
9
|
+
const extractTextDeltas_js_1 = require("./extractTextDeltas.cjs");
|
10
|
+
async function streamText(model, prompt, options) {
|
11
|
+
if (options?.settings != null) {
|
12
|
+
model = model.withSettings(options.settings);
|
13
|
+
options = {
|
14
|
+
functionId: options.functionId,
|
15
|
+
run: options.run,
|
16
|
+
};
|
17
|
+
}
|
18
|
+
const run = options?.run;
|
19
|
+
const settings = model.settings;
|
20
|
+
const eventSource = new ModelCallEventSource_js_1.ModelCallEventSource({
|
21
|
+
observers: [...(settings.observers ?? []), ...(run?.observers ?? [])],
|
22
|
+
errorHandler: run?.errorHandler,
|
23
|
+
});
|
24
|
+
const durationMeasurement = (0, DurationMeasurement_js_1.startDurationMeasurement)();
|
25
|
+
const startMetadata = {
|
26
|
+
callId: `call-${(0, nanoid_1.nanoid)()}`,
|
27
|
+
runId: run?.runId,
|
28
|
+
sessionId: run?.sessionId,
|
29
|
+
userId: run?.userId,
|
30
|
+
functionId: options?.functionId,
|
31
|
+
model: model.modelInformation,
|
32
|
+
startEpochSeconds: durationMeasurement.startEpochSeconds,
|
33
|
+
};
|
34
|
+
eventSource.notifyModelCallStarted({
|
35
|
+
type: "text-streaming-started",
|
36
|
+
metadata: startMetadata,
|
37
|
+
settings,
|
38
|
+
prompt,
|
39
|
+
});
|
40
|
+
const result = await (0, runSafe_js_1.runSafe)(async () => (0, extractTextDeltas_js_1.extractTextDeltas)({
|
41
|
+
deltaIterable: await model.generateDeltaStreamResponse(prompt, {
|
42
|
+
functionId: options?.functionId,
|
43
|
+
settings,
|
44
|
+
run,
|
45
|
+
}),
|
46
|
+
extractDelta: (fullDelta) => model.extractTextDelta(fullDelta),
|
47
|
+
onDone: (fullText, lastFullDelta) => {
|
48
|
+
const finishMetadata = {
|
49
|
+
...startMetadata,
|
50
|
+
durationInMs: durationMeasurement.durationInMs,
|
51
|
+
};
|
52
|
+
eventSource.notifyModelCallFinished({
|
53
|
+
type: "text-streaming-finished",
|
54
|
+
status: "success",
|
55
|
+
metadata: finishMetadata,
|
56
|
+
settings,
|
57
|
+
prompt,
|
58
|
+
response: lastFullDelta,
|
59
|
+
generatedText: fullText,
|
60
|
+
});
|
61
|
+
},
|
62
|
+
onError: (error) => {
|
63
|
+
const finishMetadata = {
|
64
|
+
...startMetadata,
|
65
|
+
durationInMs: durationMeasurement.durationInMs,
|
66
|
+
};
|
67
|
+
eventSource.notifyModelCallFinished(error instanceof AbortError_js_1.AbortError
|
68
|
+
? {
|
69
|
+
type: "text-streaming-finished",
|
70
|
+
status: "abort",
|
71
|
+
metadata: finishMetadata,
|
72
|
+
settings,
|
73
|
+
prompt,
|
74
|
+
}
|
75
|
+
: {
|
76
|
+
type: "text-streaming-finished",
|
77
|
+
status: "failure",
|
78
|
+
metadata: finishMetadata,
|
79
|
+
settings,
|
80
|
+
prompt,
|
81
|
+
error,
|
82
|
+
});
|
83
|
+
},
|
84
|
+
}));
|
85
|
+
if (!result.ok) {
|
86
|
+
const finishMetadata = {
|
87
|
+
...startMetadata,
|
88
|
+
durationInMs: durationMeasurement.durationInMs,
|
89
|
+
};
|
90
|
+
if (result.isAborted) {
|
91
|
+
eventSource.notifyModelCallFinished({
|
92
|
+
type: "text-streaming-finished",
|
93
|
+
status: "abort",
|
94
|
+
metadata: finishMetadata,
|
95
|
+
settings,
|
96
|
+
prompt,
|
97
|
+
});
|
98
|
+
throw new AbortError_js_1.AbortError();
|
99
|
+
}
|
100
|
+
eventSource.notifyModelCallFinished({
|
101
|
+
type: "text-streaming-finished",
|
102
|
+
status: "failure",
|
103
|
+
metadata: finishMetadata,
|
104
|
+
settings,
|
105
|
+
prompt,
|
106
|
+
error: result.error,
|
107
|
+
});
|
108
|
+
throw result.error;
|
109
|
+
}
|
110
|
+
return {
|
111
|
+
textStream: result.output,
|
112
|
+
metadata: startMetadata,
|
113
|
+
};
|
114
|
+
}
|
115
|
+
exports.streamText = streamText;
|
@@ -0,0 +1,11 @@
|
|
1
|
+
import { FunctionOptions } from "../FunctionOptions.js";
|
2
|
+
import { DeltaEvent } from "./DeltaEvent.js";
|
3
|
+
import { TextGenerationModel, TextGenerationModelSettings } from "./TextGenerationModel.js";
|
4
|
+
import { CallMetadata } from "model-function/executeCall.js";
|
5
|
+
export declare function streamText<PROMPT, FULL_DELTA, SETTINGS extends TextGenerationModelSettings>(model: TextGenerationModel<PROMPT, unknown, FULL_DELTA, SETTINGS> & {
|
6
|
+
generateDeltaStreamResponse: (prompt: PROMPT, options: FunctionOptions<SETTINGS>) => PromiseLike<AsyncIterable<DeltaEvent<FULL_DELTA>>>;
|
7
|
+
extractTextDelta: (fullDelta: FULL_DELTA) => string | undefined;
|
8
|
+
}, prompt: PROMPT, options?: FunctionOptions<SETTINGS>): Promise<{
|
9
|
+
textStream: AsyncIterable<string>;
|
10
|
+
metadata: Omit<CallMetadata<TextGenerationModel<PROMPT, unknown, FULL_DELTA, SETTINGS>>, "durationInMs">;
|
11
|
+
}>;
|
@@ -0,0 +1,111 @@
|
|
1
|
+
import { nanoid as createId } from "nanoid";
|
2
|
+
import { startDurationMeasurement } from "../../util/DurationMeasurement.js";
|
3
|
+
import { AbortError } from "../../util/api/AbortError.js";
|
4
|
+
import { runSafe } from "../../util/runSafe.js";
|
5
|
+
import { ModelCallEventSource } from "../ModelCallEventSource.js";
|
6
|
+
import { extractTextDeltas } from "./extractTextDeltas.js";
|
7
|
+
export async function streamText(model, prompt, options) {
|
8
|
+
if (options?.settings != null) {
|
9
|
+
model = model.withSettings(options.settings);
|
10
|
+
options = {
|
11
|
+
functionId: options.functionId,
|
12
|
+
run: options.run,
|
13
|
+
};
|
14
|
+
}
|
15
|
+
const run = options?.run;
|
16
|
+
const settings = model.settings;
|
17
|
+
const eventSource = new ModelCallEventSource({
|
18
|
+
observers: [...(settings.observers ?? []), ...(run?.observers ?? [])],
|
19
|
+
errorHandler: run?.errorHandler,
|
20
|
+
});
|
21
|
+
const durationMeasurement = startDurationMeasurement();
|
22
|
+
const startMetadata = {
|
23
|
+
callId: `call-${createId()}`,
|
24
|
+
runId: run?.runId,
|
25
|
+
sessionId: run?.sessionId,
|
26
|
+
userId: run?.userId,
|
27
|
+
functionId: options?.functionId,
|
28
|
+
model: model.modelInformation,
|
29
|
+
startEpochSeconds: durationMeasurement.startEpochSeconds,
|
30
|
+
};
|
31
|
+
eventSource.notifyModelCallStarted({
|
32
|
+
type: "text-streaming-started",
|
33
|
+
metadata: startMetadata,
|
34
|
+
settings,
|
35
|
+
prompt,
|
36
|
+
});
|
37
|
+
const result = await runSafe(async () => extractTextDeltas({
|
38
|
+
deltaIterable: await model.generateDeltaStreamResponse(prompt, {
|
39
|
+
functionId: options?.functionId,
|
40
|
+
settings,
|
41
|
+
run,
|
42
|
+
}),
|
43
|
+
extractDelta: (fullDelta) => model.extractTextDelta(fullDelta),
|
44
|
+
onDone: (fullText, lastFullDelta) => {
|
45
|
+
const finishMetadata = {
|
46
|
+
...startMetadata,
|
47
|
+
durationInMs: durationMeasurement.durationInMs,
|
48
|
+
};
|
49
|
+
eventSource.notifyModelCallFinished({
|
50
|
+
type: "text-streaming-finished",
|
51
|
+
status: "success",
|
52
|
+
metadata: finishMetadata,
|
53
|
+
settings,
|
54
|
+
prompt,
|
55
|
+
response: lastFullDelta,
|
56
|
+
generatedText: fullText,
|
57
|
+
});
|
58
|
+
},
|
59
|
+
onError: (error) => {
|
60
|
+
const finishMetadata = {
|
61
|
+
...startMetadata,
|
62
|
+
durationInMs: durationMeasurement.durationInMs,
|
63
|
+
};
|
64
|
+
eventSource.notifyModelCallFinished(error instanceof AbortError
|
65
|
+
? {
|
66
|
+
type: "text-streaming-finished",
|
67
|
+
status: "abort",
|
68
|
+
metadata: finishMetadata,
|
69
|
+
settings,
|
70
|
+
prompt,
|
71
|
+
}
|
72
|
+
: {
|
73
|
+
type: "text-streaming-finished",
|
74
|
+
status: "failure",
|
75
|
+
metadata: finishMetadata,
|
76
|
+
settings,
|
77
|
+
prompt,
|
78
|
+
error,
|
79
|
+
});
|
80
|
+
},
|
81
|
+
}));
|
82
|
+
if (!result.ok) {
|
83
|
+
const finishMetadata = {
|
84
|
+
...startMetadata,
|
85
|
+
durationInMs: durationMeasurement.durationInMs,
|
86
|
+
};
|
87
|
+
if (result.isAborted) {
|
88
|
+
eventSource.notifyModelCallFinished({
|
89
|
+
type: "text-streaming-finished",
|
90
|
+
status: "abort",
|
91
|
+
metadata: finishMetadata,
|
92
|
+
settings,
|
93
|
+
prompt,
|
94
|
+
});
|
95
|
+
throw new AbortError();
|
96
|
+
}
|
97
|
+
eventSource.notifyModelCallFinished({
|
98
|
+
type: "text-streaming-finished",
|
99
|
+
status: "failure",
|
100
|
+
metadata: finishMetadata,
|
101
|
+
settings,
|
102
|
+
prompt,
|
103
|
+
error: result.error,
|
104
|
+
});
|
105
|
+
throw result.error;
|
106
|
+
}
|
107
|
+
return {
|
108
|
+
textStream: result.output,
|
109
|
+
metadata: startMetadata,
|
110
|
+
};
|
111
|
+
}
|
@@ -0,0 +1,47 @@
|
|
1
|
+
"use strict";
|
2
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
3
|
+
if (k2 === undefined) k2 = k;
|
4
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
5
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
6
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
7
|
+
}
|
8
|
+
Object.defineProperty(o, k2, desc);
|
9
|
+
}) : (function(o, m, k, k2) {
|
10
|
+
if (k2 === undefined) k2 = k;
|
11
|
+
o[k2] = m[k];
|
12
|
+
}));
|
13
|
+
var __exportStar = (this && this.__exportStar) || function(m, exports) {
|
14
|
+
for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
|
15
|
+
};
|
16
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
17
|
+
__exportStar(require("./FunctionOptions.cjs"), exports);
|
18
|
+
__exportStar(require("./Model.cjs"), exports);
|
19
|
+
__exportStar(require("./ModelCallEvent.cjs"), exports);
|
20
|
+
__exportStar(require("./ModelCallObserver.cjs"), exports);
|
21
|
+
__exportStar(require("./ModelInformation.cjs"), exports);
|
22
|
+
__exportStar(require("./SuccessfulModelCall.cjs"), exports);
|
23
|
+
__exportStar(require("./embed-text/TextEmbeddingEvent.cjs"), exports);
|
24
|
+
__exportStar(require("./embed-text/TextEmbeddingModel.cjs"), exports);
|
25
|
+
__exportStar(require("./embed-text/embedText.cjs"), exports);
|
26
|
+
__exportStar(require("./generate-image/ImageGenerationEvent.cjs"), exports);
|
27
|
+
__exportStar(require("./generate-image/ImageGenerationModel.cjs"), exports);
|
28
|
+
__exportStar(require("./generate-image/generateImage.cjs"), exports);
|
29
|
+
__exportStar(require("./generate-json/GenerateJsonModel.cjs"), exports);
|
30
|
+
__exportStar(require("./generate-json/GenerateJsonOrTextModel.cjs"), exports);
|
31
|
+
__exportStar(require("./generate-json/JsonGenerationEvent.cjs"), exports);
|
32
|
+
__exportStar(require("./generate-json/NoSuchSchemaError.cjs"), exports);
|
33
|
+
__exportStar(require("./generate-json/SchemaDefinition.cjs"), exports);
|
34
|
+
__exportStar(require("./generate-json/SchemaValidationError.cjs"), exports);
|
35
|
+
__exportStar(require("./generate-json/generateJson.cjs"), exports);
|
36
|
+
__exportStar(require("./generate-json/generateJsonOrText.cjs"), exports);
|
37
|
+
__exportStar(require("./generate-text/TextGenerationEvent.cjs"), exports);
|
38
|
+
__exportStar(require("./generate-text/TextGenerationModel.cjs"), exports);
|
39
|
+
__exportStar(require("./generate-text/generateText.cjs"), exports);
|
40
|
+
__exportStar(require("./generate-text/DeltaEvent.cjs"), exports);
|
41
|
+
__exportStar(require("./generate-text/TextDeltaEventSource.cjs"), exports);
|
42
|
+
__exportStar(require("./generate-text/streamText.cjs"), exports);
|
43
|
+
__exportStar(require("./tokenize-text/Tokenizer.cjs"), exports);
|
44
|
+
__exportStar(require("./tokenize-text/countTokens.cjs"), exports);
|
45
|
+
__exportStar(require("./transcribe-audio/TranscriptionEvent.cjs"), exports);
|
46
|
+
__exportStar(require("./transcribe-audio/TranscriptionModel.cjs"), exports);
|
47
|
+
__exportStar(require("./transcribe-audio/transcribe.cjs"), exports);
|
@@ -0,0 +1,31 @@
|
|
1
|
+
export * from "./FunctionOptions.js";
|
2
|
+
export * from "./Model.js";
|
3
|
+
export * from "./ModelCallEvent.js";
|
4
|
+
export * from "./ModelCallObserver.js";
|
5
|
+
export * from "./ModelInformation.js";
|
6
|
+
export * from "./SuccessfulModelCall.js";
|
7
|
+
export * from "./embed-text/TextEmbeddingEvent.js";
|
8
|
+
export * from "./embed-text/TextEmbeddingModel.js";
|
9
|
+
export * from "./embed-text/embedText.js";
|
10
|
+
export * from "./generate-image/ImageGenerationEvent.js";
|
11
|
+
export * from "./generate-image/ImageGenerationModel.js";
|
12
|
+
export * from "./generate-image/generateImage.js";
|
13
|
+
export * from "./generate-json/GenerateJsonModel.js";
|
14
|
+
export * from "./generate-json/GenerateJsonOrTextModel.js";
|
15
|
+
export * from "./generate-json/JsonGenerationEvent.js";
|
16
|
+
export * from "./generate-json/NoSuchSchemaError.js";
|
17
|
+
export * from "./generate-json/SchemaDefinition.js";
|
18
|
+
export * from "./generate-json/SchemaValidationError.js";
|
19
|
+
export * from "./generate-json/generateJson.js";
|
20
|
+
export * from "./generate-json/generateJsonOrText.js";
|
21
|
+
export * from "./generate-text/TextGenerationEvent.js";
|
22
|
+
export * from "./generate-text/TextGenerationModel.js";
|
23
|
+
export * from "./generate-text/generateText.js";
|
24
|
+
export * from "./generate-text/DeltaEvent.js";
|
25
|
+
export * from "./generate-text/TextDeltaEventSource.js";
|
26
|
+
export * from "./generate-text/streamText.js";
|
27
|
+
export * from "./tokenize-text/Tokenizer.js";
|
28
|
+
export * from "./tokenize-text/countTokens.js";
|
29
|
+
export * from "./transcribe-audio/TranscriptionEvent.js";
|
30
|
+
export * from "./transcribe-audio/TranscriptionModel.js";
|
31
|
+
export * from "./transcribe-audio/transcribe.js";
|
@@ -0,0 +1,31 @@
|
|
1
|
+
export * from "./FunctionOptions.js";
|
2
|
+
export * from "./Model.js";
|
3
|
+
export * from "./ModelCallEvent.js";
|
4
|
+
export * from "./ModelCallObserver.js";
|
5
|
+
export * from "./ModelInformation.js";
|
6
|
+
export * from "./SuccessfulModelCall.js";
|
7
|
+
export * from "./embed-text/TextEmbeddingEvent.js";
|
8
|
+
export * from "./embed-text/TextEmbeddingModel.js";
|
9
|
+
export * from "./embed-text/embedText.js";
|
10
|
+
export * from "./generate-image/ImageGenerationEvent.js";
|
11
|
+
export * from "./generate-image/ImageGenerationModel.js";
|
12
|
+
export * from "./generate-image/generateImage.js";
|
13
|
+
export * from "./generate-json/GenerateJsonModel.js";
|
14
|
+
export * from "./generate-json/GenerateJsonOrTextModel.js";
|
15
|
+
export * from "./generate-json/JsonGenerationEvent.js";
|
16
|
+
export * from "./generate-json/NoSuchSchemaError.js";
|
17
|
+
export * from "./generate-json/SchemaDefinition.js";
|
18
|
+
export * from "./generate-json/SchemaValidationError.js";
|
19
|
+
export * from "./generate-json/generateJson.js";
|
20
|
+
export * from "./generate-json/generateJsonOrText.js";
|
21
|
+
export * from "./generate-text/TextGenerationEvent.js";
|
22
|
+
export * from "./generate-text/TextGenerationModel.js";
|
23
|
+
export * from "./generate-text/generateText.js";
|
24
|
+
export * from "./generate-text/DeltaEvent.js";
|
25
|
+
export * from "./generate-text/TextDeltaEventSource.js";
|
26
|
+
export * from "./generate-text/streamText.js";
|
27
|
+
export * from "./tokenize-text/Tokenizer.js";
|
28
|
+
export * from "./tokenize-text/countTokens.js";
|
29
|
+
export * from "./transcribe-audio/TranscriptionEvent.js";
|
30
|
+
export * from "./transcribe-audio/TranscriptionModel.js";
|
31
|
+
export * from "./transcribe-audio/transcribe.js";
|
@@ -0,0 +1,19 @@
|
|
1
|
+
export interface BasicTokenizer {
|
2
|
+
/**
|
3
|
+
* Get the tokens that represent the given text.
|
4
|
+
*/
|
5
|
+
tokenize: (text: string) => PromiseLike<Array<number>>;
|
6
|
+
}
|
7
|
+
export interface FullTokenizer extends BasicTokenizer {
|
8
|
+
/**
|
9
|
+
* Get the tokens that represent the given text and the text for each token.
|
10
|
+
*/
|
11
|
+
tokenizeWithTexts: (text: string) => PromiseLike<{
|
12
|
+
tokens: Array<number>;
|
13
|
+
tokenTexts: Array<string>;
|
14
|
+
}>;
|
15
|
+
/**
|
16
|
+
* Get the text that represents the given tokens.
|
17
|
+
*/
|
18
|
+
detokenize: (tokens: Array<number>) => PromiseLike<string>;
|
19
|
+
}
|
@@ -0,0 +1 @@
|
|
1
|
+
export {};
|
@@ -0,0 +1,10 @@
|
|
1
|
+
"use strict";
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
3
|
+
exports.countTokens = void 0;
|
4
|
+
/**
|
5
|
+
* Count the number of tokens in the given text.
|
6
|
+
*/
|
7
|
+
async function countTokens(tokenizer, text) {
|
8
|
+
return (await tokenizer.tokenize(text)).length;
|
9
|
+
}
|
10
|
+
exports.countTokens = countTokens;
|
@@ -0,0 +1,22 @@
|
|
1
|
+
import { ModelCallFinishedEventMetadata, ModelCallStartedEventMetadata } from "../ModelCallEvent.js";
|
2
|
+
export type TranscriptionStartedEvent = {
|
3
|
+
type: "transcription-started";
|
4
|
+
metadata: ModelCallStartedEventMetadata;
|
5
|
+
settings: unknown;
|
6
|
+
data: unknown;
|
7
|
+
};
|
8
|
+
export type TranscriptionFinishedEvent = {
|
9
|
+
type: "transcription-finished";
|
10
|
+
metadata: ModelCallFinishedEventMetadata;
|
11
|
+
settings: unknown;
|
12
|
+
data: unknown;
|
13
|
+
} & ({
|
14
|
+
status: "success";
|
15
|
+
response: unknown;
|
16
|
+
transcription: string;
|
17
|
+
} | {
|
18
|
+
status: "failure";
|
19
|
+
error: unknown;
|
20
|
+
} | {
|
21
|
+
status: "abort";
|
22
|
+
});
|
@@ -0,0 +1 @@
|
|
1
|
+
export {};
|
@@ -0,0 +1,8 @@
|
|
1
|
+
import { FunctionOptions } from "../FunctionOptions.js";
|
2
|
+
import { Model, ModelSettings } from "../Model.js";
|
3
|
+
export interface TranscriptionModelSettings extends ModelSettings {
|
4
|
+
}
|
5
|
+
export interface TranscriptionModel<DATA, RESPONSE, SETTINGS> extends Model<SETTINGS> {
|
6
|
+
generateTranscriptionResponse: (data: DATA, options?: FunctionOptions<SETTINGS>) => PromiseLike<RESPONSE>;
|
7
|
+
extractTranscriptionText: (response: RESPONSE) => string;
|
8
|
+
}
|
@@ -0,0 +1 @@
|
|
1
|
+
export {};
|
@@ -0,0 +1,62 @@
|
|
1
|
+
"use strict";
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
3
|
+
exports.transcribe = void 0;
|
4
|
+
const executeCall_js_1 = require("../executeCall.cjs");
|
5
|
+
/**
|
6
|
+
* Transcribe audio data into text.
|
7
|
+
*
|
8
|
+
* @example
|
9
|
+
* const data = await fs.promises.readFile("data/test.mp3");
|
10
|
+
*
|
11
|
+
* const { transcription } = await transcribe(
|
12
|
+
* new OpenAITranscriptionModel({ model: "whisper-1" }),
|
13
|
+
* {
|
14
|
+
* type: "mp3",
|
15
|
+
* data,
|
16
|
+
* }
|
17
|
+
* );
|
18
|
+
*/
|
19
|
+
async function transcribe(model, data, options) {
|
20
|
+
const result = await (0, executeCall_js_1.executeCall)({
|
21
|
+
model,
|
22
|
+
options,
|
23
|
+
generateResponse: (options) => model.generateTranscriptionResponse(data, options),
|
24
|
+
extractOutputValue: model.extractTranscriptionText,
|
25
|
+
getStartEvent: (metadata, settings) => ({
|
26
|
+
type: "transcription-started",
|
27
|
+
metadata,
|
28
|
+
settings,
|
29
|
+
data,
|
30
|
+
}),
|
31
|
+
getAbortEvent: (metadata, settings) => ({
|
32
|
+
type: "transcription-finished",
|
33
|
+
status: "abort",
|
34
|
+
settings,
|
35
|
+
metadata,
|
36
|
+
data,
|
37
|
+
}),
|
38
|
+
getFailureEvent: (metadata, settings, error) => ({
|
39
|
+
type: "transcription-finished",
|
40
|
+
status: "failure",
|
41
|
+
metadata,
|
42
|
+
settings,
|
43
|
+
data,
|
44
|
+
error,
|
45
|
+
}),
|
46
|
+
getSuccessEvent: (metadata, settings, response, output) => ({
|
47
|
+
type: "transcription-finished",
|
48
|
+
status: "success",
|
49
|
+
metadata,
|
50
|
+
settings,
|
51
|
+
data,
|
52
|
+
response,
|
53
|
+
transcription: output,
|
54
|
+
}),
|
55
|
+
});
|
56
|
+
return {
|
57
|
+
transcription: result.output,
|
58
|
+
response: result.response,
|
59
|
+
metadata: result.metadata,
|
60
|
+
};
|
61
|
+
}
|
62
|
+
exports.transcribe = transcribe;
|
@@ -0,0 +1,22 @@
|
|
1
|
+
import { FunctionOptions } from "../FunctionOptions.js";
|
2
|
+
import { CallMetadata } from "../executeCall.js";
|
3
|
+
import { TranscriptionModel, TranscriptionModelSettings } from "./TranscriptionModel.js";
|
4
|
+
/**
|
5
|
+
* Transcribe audio data into text.
|
6
|
+
*
|
7
|
+
* @example
|
8
|
+
* const data = await fs.promises.readFile("data/test.mp3");
|
9
|
+
*
|
10
|
+
* const { transcription } = await transcribe(
|
11
|
+
* new OpenAITranscriptionModel({ model: "whisper-1" }),
|
12
|
+
* {
|
13
|
+
* type: "mp3",
|
14
|
+
* data,
|
15
|
+
* }
|
16
|
+
* );
|
17
|
+
*/
|
18
|
+
export declare function transcribe<DATA, RESPONSE, SETTINGS extends TranscriptionModelSettings>(model: TranscriptionModel<DATA, RESPONSE, SETTINGS>, data: DATA, options?: FunctionOptions<SETTINGS>): Promise<{
|
19
|
+
transcription: string;
|
20
|
+
response: RESPONSE;
|
21
|
+
metadata: CallMetadata<TranscriptionModel<DATA, RESPONSE, SETTINGS>>;
|
22
|
+
}>;
|
@@ -0,0 +1,58 @@
|
|
1
|
+
import { executeCall } from "../executeCall.js";
|
2
|
+
/**
|
3
|
+
* Transcribe audio data into text.
|
4
|
+
*
|
5
|
+
* @example
|
6
|
+
* const data = await fs.promises.readFile("data/test.mp3");
|
7
|
+
*
|
8
|
+
* const { transcription } = await transcribe(
|
9
|
+
* new OpenAITranscriptionModel({ model: "whisper-1" }),
|
10
|
+
* {
|
11
|
+
* type: "mp3",
|
12
|
+
* data,
|
13
|
+
* }
|
14
|
+
* );
|
15
|
+
*/
|
16
|
+
export async function transcribe(model, data, options) {
|
17
|
+
const result = await executeCall({
|
18
|
+
model,
|
19
|
+
options,
|
20
|
+
generateResponse: (options) => model.generateTranscriptionResponse(data, options),
|
21
|
+
extractOutputValue: model.extractTranscriptionText,
|
22
|
+
getStartEvent: (metadata, settings) => ({
|
23
|
+
type: "transcription-started",
|
24
|
+
metadata,
|
25
|
+
settings,
|
26
|
+
data,
|
27
|
+
}),
|
28
|
+
getAbortEvent: (metadata, settings) => ({
|
29
|
+
type: "transcription-finished",
|
30
|
+
status: "abort",
|
31
|
+
settings,
|
32
|
+
metadata,
|
33
|
+
data,
|
34
|
+
}),
|
35
|
+
getFailureEvent: (metadata, settings, error) => ({
|
36
|
+
type: "transcription-finished",
|
37
|
+
status: "failure",
|
38
|
+
metadata,
|
39
|
+
settings,
|
40
|
+
data,
|
41
|
+
error,
|
42
|
+
}),
|
43
|
+
getSuccessEvent: (metadata, settings, response, output) => ({
|
44
|
+
type: "transcription-finished",
|
45
|
+
status: "success",
|
46
|
+
metadata,
|
47
|
+
settings,
|
48
|
+
data,
|
49
|
+
response,
|
50
|
+
transcription: output,
|
51
|
+
}),
|
52
|
+
});
|
53
|
+
return {
|
54
|
+
transcription: result.output,
|
55
|
+
response: result.response,
|
56
|
+
metadata: result.metadata,
|
57
|
+
};
|
58
|
+
}
|
@@ -0,0 +1,39 @@
|
|
1
|
+
"use strict";
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
4
|
+
};
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
6
|
+
exports.failedAutomatic1111CallResponseHandler = exports.Automatic1111Error = exports.automatic1111ErrorDataSchema = void 0;
|
7
|
+
const zod_1 = require("zod");
|
8
|
+
const secure_json_parse_1 = __importDefault(require("secure-json-parse"));
|
9
|
+
const ApiCallError_js_1 = require("../../util/api/ApiCallError.cjs");
|
10
|
+
exports.automatic1111ErrorDataSchema = zod_1.z.object({
|
11
|
+
error: zod_1.z.string(),
|
12
|
+
detail: zod_1.z.string(),
|
13
|
+
body: zod_1.z.string(),
|
14
|
+
errors: zod_1.z.string(),
|
15
|
+
});
|
16
|
+
class Automatic1111Error extends ApiCallError_js_1.ApiCallError {
|
17
|
+
constructor({ data, statusCode, url, requestBodyValues, message = data.detail, }) {
|
18
|
+
super({ message, statusCode, requestBodyValues, url });
|
19
|
+
Object.defineProperty(this, "data", {
|
20
|
+
enumerable: true,
|
21
|
+
configurable: true,
|
22
|
+
writable: true,
|
23
|
+
value: void 0
|
24
|
+
});
|
25
|
+
this.data = data;
|
26
|
+
}
|
27
|
+
}
|
28
|
+
exports.Automatic1111Error = Automatic1111Error;
|
29
|
+
const failedAutomatic1111CallResponseHandler = async ({ response, url, requestBodyValues }) => {
|
30
|
+
const responseBody = await response.text();
|
31
|
+
const parsedError = exports.automatic1111ErrorDataSchema.parse(secure_json_parse_1.default.parse(responseBody));
|
32
|
+
return new Automatic1111Error({
|
33
|
+
url,
|
34
|
+
requestBodyValues,
|
35
|
+
statusCode: response.status,
|
36
|
+
data: parsedError,
|
37
|
+
});
|
38
|
+
};
|
39
|
+
exports.failedAutomatic1111CallResponseHandler = failedAutomatic1111CallResponseHandler;
|