fix: SessionOutput type defs

Author: kallebysantos

ext/ai/js/ai.ts

@@ -46,7 +46,12 @@ export type SessionOutput<T extends SessionType, O> = T extends "gte-small"
   ? number[]
   : T extends LLMProviderName
     ? O extends { stream: true }
-      ? AsyncGenerator<LLMProviderInstance<T>["output"]>
+      ? AsyncGenerator<
+        Result<
+          LLMProviderInstance<T>["output"],
+          LLMProviderInstance<T>["error"]
+        >
+      >
     : LLMProviderInstance<T>["output"]
   : never;
 

ext/ai/js/llm/providers/ollama.ts

@@ -33,7 +33,7 @@ export type OllamaMessage = {
 
 export class OllamaLLMSession implements ILLMProvider, ILLMProviderMeta {
   input!: OllamaProviderInput;
-  output!: OllamaProviderOutput;
+  output!: ILLMProviderOutput<OllamaMessage>;
   error!: OllamaProviderError;
   options: OllamaProviderOptions;
 

ext/ai/js/llm/providers/openai.ts

@@ -120,7 +120,7 @@ export type OpenAIProviderError = ILLMProviderError<object>;
 
 export class OpenAILLMSession implements ILLMProvider, ILLMProviderMeta {
   input!: OpenAIProviderInput;
-  output!: OpenAIProviderOutput;
+  output!: ILLMProviderOutput<OpenAIResponse>;
   error!: OpenAIProviderError;
   options: OpenAIProviderOptions;