@ampt/ai - npm Package Compare versions

Comparing version 2.0.1 to 2.0.2

dist/index.d.ts

@@ -1,10 +0,12 @@
-type ModelId = 'anthropic.claude-v2' | 'anthropic.claude-instant-v1' | 'anthropic.claude-v1' | 'stability.stable-diffusion-xl-v0';
+type ModelId = 'anthropic.claude-v2' | 'anthropic.claude-instant-v1' | 'anthropic.claude-v1' | 'stability.stable-diffusion-xl-v0' | 'amazon.titan-embed-text-v1';
 declare function models(): Promise<ModelId[]>;
 declare enum ModelType {
+    Embeddings = "embeddings",
     Language = "language",
     Image = "image"
 }
+type EmbeddingsInferenceCostFunction = (body: Uint8Array) => number;
 type LanguageInferenceCostFunction = (inputSize: number, outputSize: number) => number;
 type ImageInferenceCostFunction = (width: number, height: number, steps: number) => number;
-type InferenceCostFunction = LanguageInferenceCostFunction | ImageInferenceCostFunction;
+type InferenceCostFunction = EmbeddingsInferenceCostFunction | LanguageInferenceCostFunction | ImageInferenceCostFunction;
 interface ModelInfo {
@@ -27,3 +29,7 @@ name: string;
 }
-declare function getModelInfo(id: ModelId): Promise<LanguageModelInfo | ImageModelInfo>;
+interface EmbeddingsModelInfo extends ModelInfo {
+    type: ModelType.Embeddings;
+    inferenceUsage: EmbeddingsInferenceCostFunction;
+}
+declare function getModelInfo(id: ModelId): Promise<LanguageModelInfo | ImageModelInfo | EmbeddingsModelInfo>;
 
@@ -72,2 +78,14 @@ interface InvokeOptions {
 
+interface Embedding {
+    embedding: number[];
+    inputTokenCount: number;
+}
+interface EmbedOptions {
+    modelId?: ModelId;
+}
+declare function embed(input: string, options?: EmbedOptions): Promise<Embedding>;
+declare namespace embed {
+    var models: () => Promise<ModelId[]>;
+}
+
 interface RenderOptions {
@@ -86,2 +104,2 @@ width?: number;
 
-export { ChatMessage, ChatOptions, ImageModelInfo, InvokeOptions, LanguageModelInfo, MessageRole, ModelId, ModelInfo, ModelType, RenderOptions, buildPrompt, chat, getModelInfo, invoke, models, render };
+export { ChatMessage, ChatOptions, EmbedOptions, Embedding, EmbeddingsModelInfo, ImageModelInfo, InvokeOptions, LanguageModelInfo, MessageRole, ModelId, ModelInfo, ModelType, RenderOptions, buildPrompt, chat, embed, getModelInfo, invoke, models, render };

package.json

 {
   "name": "@ampt/ai",
-  "version": "2.0.1",
+  "version": "2.0.2",
   "type": "module",
@@ -5,0 +5,0 @@ "main": "./dist/index.cjs",

Fixed alerts

New author

Supply chain risk

A new npm collaborator published a version of the package for the first time. New collaborators are usually benign additions to a project, but do indicate a change to the security surface area of a package.

Found 1 instance in 1 package

Improved metrics

Total package byte prevSize

3102144

increased by0.62%

Lines of code

80195

increased by0.68%

Number of medium supply chain risk alerts

4

decreased by-20%

No dependency changes