@friendliai/ai-provider 0.2.6 → 0.2.7-alpha.1

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # @friendliai/ai-provider
 
+## 0.2.7-alpha.1
+
+### Patch Changes
+
+- dda6520: Added supported type model to friendli-provider
+
+## 0.2.7-alpha.0
+
+### Patch Changes
+
+- 5ec8524: Initial build for ai sdk v5 support (broken Friendli compatibility)
+
 ## 0.2.6
 
 ### Patch Changes

package/README.md

@@ -12,7 +12,7 @@ npm i @friendliai/ai-provider
 
 ## Credentials
 
-The tokens required for model usage can be obtained from the [Friendli suite](https://suite.friendli.ai/).
+The tokens required for model usage can be obtained from the [Friendli suite](https://friendli.ai/suite/).
 
 To use the provider, you need to set the `FRIENDLI_TOKEN` environment variable with your personal access token.
 

package/dist/index.d.mts

@@ -1,42 +1,14 @@
-import { LanguageModelV1 } from '@ai-sdk/provider';
+import { ProviderV2, LanguageModelV2, EmbeddingModelV2, ImageModelV2, TranscriptionModelV1, SpeechModelV1 } from '@ai-sdk/provider';
 import { FetchFunction } from '@ai-sdk/provider-utils';
-import { OpenAICompatibleChatSettings } from '@ai-sdk/openai-compatible';
 import { z } from 'zod';
 
 declare const FriendliAIServerlessModelIds: readonly ["meta-llama-3.1-8b-instruct", "meta-llama-3.3-70b-instruct", "deepseek-r1"];
 type FriendliAIServerlessModelId = (typeof FriendliAIServerlessModelIds)[number];
 type FriendliAILanguageModelId = FriendliAIServerlessModelId | (string & {});
-type FriendliAIBetaChatModelId = string & {};
-interface FriendliAISharedSettings {
-    /**
-     * Sets the endpoint to which the request will be sent.
-     * auto: automatically selected based on model_id
-     * dedicated: Fixed to "/dedicated/v1"
-     * serverless: automatically selected as one of "/serverless/beta", "/serverless/v1", or "/serverless/tools/v1"
-     * Ignored if baseURL is specified.
-     */
-    endpoint?: 'auto' | 'dedicated' | 'serverless';
-}
-interface FriendliAIChatSettings extends FriendliAISharedSettings, OpenAICompatibleChatSettings {
-    /**
-     * BETA FEATURE: Include the model's training loss in the response.
-     */
-    tools?: Array<{
-        type: 'web:url' | 'web:search' | 'math:calendar' | 'math:statistics' | 'math:calculator' | 'code:python-interpreter';
-    }>;
-    /**
-     * Whether to enable parallel function calling during tool use. Default to true.
-     */
-    parallelToolCalls?: boolean;
-    /**
-     * BETA FEATURE: You can write a regular expression to force output that satisfies that regular expression.
-     */
-    regex?: RegExp;
-}
 
 interface FriendliAIProviderSettings {
     /**
-     * FriendliAI API key. (FRIENDLI__TOKEN)
+     * FriendliAI API key. (FRIENDLI_TOKEN)
      */
     apiKey?: string;
     /**
@@ -57,35 +29,52 @@ interface FriendliAIProviderSettings {
      */
     fetch?: FetchFunction;
 }
-interface FriendliAIProvider {
+interface FriendliAIProvider extends ProviderV2 {
     /**
      * Creates a model for text generation.
      */
-    (modelId: FriendliAILanguageModelId, settings?: FriendliAIChatSettings): LanguageModelV1;
+    (modelId: FriendliAILanguageModelId): LanguageModelV2;
     /**
-     * A model that has not yet been officially released
+     * Creates a chat model for text generation.
      */
-    beta(modelId: FriendliAIBetaChatModelId, settings?: FriendliAIChatSettings): LanguageModelV1;
+    languageModel(modelId: FriendliAILanguageModelId): LanguageModelV2;
     /**
      * Creates a chat model for text generation.
      */
-    chat(modelId: FriendliAILanguageModelId, settings?: FriendliAIChatSettings): LanguageModelV1;
-    chatModel(modelId: FriendliAILanguageModelId, settings?: FriendliAIChatSettings): LanguageModelV1;
+    chat(modelId: FriendliAILanguageModelId): LanguageModelV2;
     /**
      * Creates a completion model for text generation.
      */
-    completion(modelId: FriendliAILanguageModelId, settings?: FriendliAIChatSettings): LanguageModelV1;
-    completionModel(modelId: FriendliAILanguageModelId, settings?: FriendliAIChatSettings): LanguageModelV1;
+    completion(modelId: FriendliAILanguageModelId): LanguageModelV2;
     /**
      * Creates a text embedding model for text generation.
+     * TODO: Implement for Dedicated users
+     */
+    embedding(modelId: string & {}): EmbeddingModelV2<string>;
+    textEmbeddingModel(modelId: string & {}): EmbeddingModelV2<string>;
+    /**
+     * Creates a model for image generation.
+     * TODO: Implement for Dedicated users
+     */
+    imageModel(modelId: string & {}): ImageModelV2;
+    /**
+     * Creates a model for transcription.
+     * TODO: Implement for Dedicated users
      */
-    embedding(modelId: string & {}, settings?: FriendliAIChatSettings): LanguageModelV1;
-    textEmbeddingModel(modelId: string & {}, settings?: FriendliAIChatSettings): LanguageModelV1;
+    transcription(modelId: string & {}): TranscriptionModelV1;
+    /**
+     * Creates a model for speech generation.
+     * TODO: Implement for Dedicated users
+     */
+    speech(modelId: string & {}): SpeechModelV1;
 }
 /**
 Create an FriendliAI provider instance.
  */
 declare function createFriendli(options?: FriendliAIProviderSettings): FriendliAIProvider;
+/**
+ * Default FriendliAI provider instance.
+ */
 declare const friendli: FriendliAIProvider;
 
 declare const friendliaiErrorSchema: z.ZodObject<{

package/dist/index.d.ts

@@ -1,42 +1,14 @@
-import { LanguageModelV1 } from '@ai-sdk/provider';
+import { ProviderV2, LanguageModelV2, EmbeddingModelV2, ImageModelV2, TranscriptionModelV1, SpeechModelV1 } from '@ai-sdk/provider';
 import { FetchFunction } from '@ai-sdk/provider-utils';
-import { OpenAICompatibleChatSettings } from '@ai-sdk/openai-compatible';
 import { z } from 'zod';
 
 declare const FriendliAIServerlessModelIds: readonly ["meta-llama-3.1-8b-instruct", "meta-llama-3.3-70b-instruct", "deepseek-r1"];
 type FriendliAIServerlessModelId = (typeof FriendliAIServerlessModelIds)[number];
 type FriendliAILanguageModelId = FriendliAIServerlessModelId | (string & {});
-type FriendliAIBetaChatModelId = string & {};
-interface FriendliAISharedSettings {
-    /**
-     * Sets the endpoint to which the request will be sent.
-     * auto: automatically selected based on model_id
-     * dedicated: Fixed to "/dedicated/v1"
-     * serverless: automatically selected as one of "/serverless/beta", "/serverless/v1", or "/serverless/tools/v1"
-     * Ignored if baseURL is specified.
-     */
-    endpoint?: 'auto' | 'dedicated' | 'serverless';
-}
-interface FriendliAIChatSettings extends FriendliAISharedSettings, OpenAICompatibleChatSettings {
-    /**
-     * BETA FEATURE: Include the model's training loss in the response.
-     */
-    tools?: Array<{
-        type: 'web:url' | 'web:search' | 'math:calendar' | 'math:statistics' | 'math:calculator' | 'code:python-interpreter';
-    }>;
-    /**
-     * Whether to enable parallel function calling during tool use. Default to true.
-     */
-    parallelToolCalls?: boolean;
-    /**
-     * BETA FEATURE: You can write a regular expression to force output that satisfies that regular expression.
-     */
-    regex?: RegExp;
-}
 
 interface FriendliAIProviderSettings {
     /**
-     * FriendliAI API key. (FRIENDLI__TOKEN)
+     * FriendliAI API key. (FRIENDLI_TOKEN)
      */
     apiKey?: string;
     /**
@@ -57,35 +29,52 @@ interface FriendliAIProviderSettings {
      */
     fetch?: FetchFunction;
 }
-interface FriendliAIProvider {
+interface FriendliAIProvider extends ProviderV2 {
     /**
      * Creates a model for text generation.
      */
-    (modelId: FriendliAILanguageModelId, settings?: FriendliAIChatSettings): LanguageModelV1;
+    (modelId: FriendliAILanguageModelId): LanguageModelV2;
     /**
-     * A model that has not yet been officially released
+     * Creates a chat model for text generation.
      */
-    beta(modelId: FriendliAIBetaChatModelId, settings?: FriendliAIChatSettings): LanguageModelV1;
+    languageModel(modelId: FriendliAILanguageModelId): LanguageModelV2;
     /**
      * Creates a chat model for text generation.
      */
-    chat(modelId: FriendliAILanguageModelId, settings?: FriendliAIChatSettings): LanguageModelV1;
-    chatModel(modelId: FriendliAILanguageModelId, settings?: FriendliAIChatSettings): LanguageModelV1;
+    chat(modelId: FriendliAILanguageModelId): LanguageModelV2;
     /**
      * Creates a completion model for text generation.
      */
-    completion(modelId: FriendliAILanguageModelId, settings?: FriendliAIChatSettings): LanguageModelV1;
-    completionModel(modelId: FriendliAILanguageModelId, settings?: FriendliAIChatSettings): LanguageModelV1;
+    completion(modelId: FriendliAILanguageModelId): LanguageModelV2;
     /**
      * Creates a text embedding model for text generation.
+     * TODO: Implement for Dedicated users
+     */
+    embedding(modelId: string & {}): EmbeddingModelV2<string>;
+    textEmbeddingModel(modelId: string & {}): EmbeddingModelV2<string>;
+    /**
+     * Creates a model for image generation.
+     * TODO: Implement for Dedicated users
+     */
+    imageModel(modelId: string & {}): ImageModelV2;
+    /**
+     * Creates a model for transcription.
+     * TODO: Implement for Dedicated users
      */
-    embedding(modelId: string & {}, settings?: FriendliAIChatSettings): LanguageModelV1;
-    textEmbeddingModel(modelId: string & {}, settings?: FriendliAIChatSettings): LanguageModelV1;
+    transcription(modelId: string & {}): TranscriptionModelV1;
+    /**
+     * Creates a model for speech generation.
+     * TODO: Implement for Dedicated users
+     */
+    speech(modelId: string & {}): SpeechModelV1;
 }
 /**
 Create an FriendliAI provider instance.
  */
 declare function createFriendli(options?: FriendliAIProviderSettings): FriendliAIProvider;
+/**
+ * Default FriendliAI provider instance.
+ */
 declare const friendli: FriendliAIProvider;
 
 declare const friendliaiErrorSchema: z.ZodObject<{