fix(ai): infer provider options for media activities

Author: AdemBenAbdallah

.changeset/soft-falcons-summarize.md

@@ -0,0 +1,5 @@
+---
+'@tanstack/ai': patch
+---
+
+Fix provider-specific model options inference for TTS, transcription, and summarize activities.

packages/ai/src/activities/generateSpeech/index.ts

@@ -35,10 +35,11 @@ export const kind = 'tts' as const
 /**
  * Extract provider options from a TTSAdapter via ~types.
  */
-export type TTSProviderOptions<TAdapter> =
-  TAdapter extends TTSAdapter<any, any>
-    ? TAdapter['~types']['providerOptions']
-    : object
+export type TTSProviderOptions<TAdapter> = TAdapter extends {
+  '~types': { providerOptions: infer P extends object }
+}
+  ? P
+  : object
 
 // ===========================
 // Activity Options Type

packages/ai/src/activities/generateTranscription/index.ts

@@ -35,10 +35,11 @@ export const kind = 'transcription' as const
 /**
  * Extract provider options from a TranscriptionAdapter via ~types.
  */
-export type TranscriptionProviderOptions<TAdapter> =
-  TAdapter extends TranscriptionAdapter<any, any>
-    ? TAdapter['~types']['providerOptions']
-    : object
+export type TranscriptionProviderOptions<TAdapter> = TAdapter extends {
+  '~types': { providerOptions: infer P extends object }
+}
+  ? P
+  : object
 
 // ===========================
 // Activity Options Type

packages/ai/src/activities/summarize/index.ts

@@ -25,10 +25,11 @@ export const kind = 'summarize' as const
 // ===========================
 
 /** Extract provider options from a SummarizeAdapter via ~types */
-export type SummarizeProviderOptions<TAdapter> =
-  TAdapter extends SummarizeAdapter<any, any>
-    ? TAdapter['~types']['providerOptions']
-    : object
+export type SummarizeProviderOptions<TAdapter> = TAdapter extends {
+  '~types': { providerOptions: infer P extends object }
+}
+  ? P
+  : object
 
 // ===========================
 // Activity Options Type
@@ -42,7 +43,7 @@ export type SummarizeProviderOptions<TAdapter> =
  * @template TStream - Whether to stream the output
  */
 export interface SummarizeActivityOptions<
-  TAdapter extends SummarizeAdapter<string, object>,
+  TAdapter extends SummarizeAdapter<string, SummarizeProviderOptions<TAdapter>>,
   TStream extends boolean = false,
 > {
   /** The summarize adapter to use (must be created with a model) */
@@ -150,7 +151,7 @@ function createId(prefix: string): string {
  * ```
  */
 export function summarize<
-  TAdapter extends SummarizeAdapter<string, object>,
+  TAdapter extends SummarizeAdapter<string, SummarizeProviderOptions<TAdapter>>,
   TStream extends boolean = false,
 >(
   options: SummarizeActivityOptions<TAdapter, TStream>,
@@ -159,26 +160,22 @@ export function summarize<
 
   if (stream) {
     return runStreamingSummarize(
-      options as SummarizeActivityOptions<
-        SummarizeAdapter<string, object>,
-        true
-      >,
+      options as SummarizeActivityOptions<TAdapter, true>,
     ) as SummarizeActivityResult<TStream>
   }
 
   return runSummarize(
-    options as SummarizeActivityOptions<
-      SummarizeAdapter<string, object>,
-      false
-    >,
+    options as SummarizeActivityOptions<TAdapter, false>,
   ) as SummarizeActivityResult<TStream>
 }
 
 /**
  * Run non-streaming summarization
  */
-async function runSummarize(
-  options: SummarizeActivityOptions<SummarizeAdapter<string, object>, false>,
+async function runSummarize<
+  TAdapter extends SummarizeAdapter<string, SummarizeProviderOptions<TAdapter>>,
+>(
+  options: SummarizeActivityOptions<TAdapter, false>,
 ): Promise<SummarizationResult> {
   const { adapter, text, maxLength, style, focus, modelOptions } = options
   const model = adapter.model
@@ -247,8 +244,10 @@ async function runSummarize(
  * Uses the adapter's native streaming if available, otherwise falls back
  * to non-streaming and yields the result as a single chunk.
  */
-async function* runStreamingSummarize(
-  options: SummarizeActivityOptions<SummarizeAdapter<string, object>, true>,
+async function* runStreamingSummarize<
+  TAdapter extends SummarizeAdapter<string, SummarizeProviderOptions<TAdapter>>,
+>(
+  options: SummarizeActivityOptions<TAdapter, true>,
 ): AsyncIterable<StreamChunk> {
   const { adapter, text, maxLength, style, focus, modelOptions } = options
   const model = adapter.model
@@ -296,7 +295,7 @@ async function* runStreamingSummarize(
  * Create typed options for the summarize() function without executing.
  */
 export function createSummarizeOptions<
-  TAdapter extends SummarizeAdapter<string, object>,
+  TAdapter extends SummarizeAdapter<string, SummarizeProviderOptions<TAdapter>>,
   TStream extends boolean = false,
 >(
   options: SummarizeActivityOptions<TAdapter, TStream>,