feat: custom provider support with streamSimple

- Add resetApiProviders() to clear and re-register built-in providers
- Add createAssistantMessageEventStream() factory for extensions
- Add streamSimple support in ProviderConfig for custom API implementations
- Call resetApiProviders() on /reload to clean up extension providers
- Add custom-provider.md documentation
- Add custom-provider.ts example with full Anthropic implementation
- Update extensions.md with streamSimple config option

packages/ai/CHANGELOG.md

@@ -5,6 +5,8 @@
 ### Added
 
 - Added `azure-openai-responses` provider support for Azure OpenAI Responses API. ([#890](https://github.com/badlogic/pi-mono/pull/890) by [@markusylisiurunen](https://github.com/markusylisiurunen))
+- Added `createAssistantMessageEventStream()` factory function for use in extensions.
+- Added `resetApiProviders()` to clear and re-register built-in API providers.
 
 ### Changed
 

packages/ai/src/index.ts

@@ -8,6 +8,7 @@ export * from "./providers/google-gemini-cli.js";
 export * from "./providers/google-vertex.js";
 export * from "./providers/openai-completions.js";
 export * from "./providers/openai-responses.js";
+export * from "./providers/register-builtins.js";
 export * from "./stream.js";
 export * from "./types.js";
 export * from "./utils/event-stream.js";

packages/ai/src/providers/register-builtins.ts

@@ -1,4 +1,4 @@
-import { registerApiProvider } from "../api-registry.js";
+import { clearApiProviders, registerApiProvider } from "../api-registry.js";
 import { streamBedrock, streamSimpleBedrock } from "./amazon-bedrock.js";
 import { streamAnthropic, streamSimpleAnthropic } from "./anthropic.js";
 import { streamAzureOpenAIResponses, streamSimpleAzureOpenAIResponses } from "./azure-openai-responses.js";
@@ -9,56 +9,65 @@ import { streamOpenAICodexResponses, streamSimpleOpenAICodexResponses } from "./
 import { streamOpenAICompletions, streamSimpleOpenAICompletions } from "./openai-completions.js";
 import { streamOpenAIResponses, streamSimpleOpenAIResponses } from "./openai-responses.js";
 
-registerApiProvider({
-	api: "anthropic-messages",
-	stream: streamAnthropic,
-	streamSimple: streamSimpleAnthropic,
-});
+export function registerBuiltInApiProviders(): void {
+	registerApiProvider({
+		api: "anthropic-messages",
+		stream: streamAnthropic,
+		streamSimple: streamSimpleAnthropic,
+	});
 
-registerApiProvider({
-	api: "openai-completions",
-	stream: streamOpenAICompletions,
-	streamSimple: streamSimpleOpenAICompletions,
-});
+	registerApiProvider({
+		api: "openai-completions",
+		stream: streamOpenAICompletions,
+		streamSimple: streamSimpleOpenAICompletions,
+	});
 
-registerApiProvider({
-	api: "openai-responses",
-	stream: streamOpenAIResponses,
-	streamSimple: streamSimpleOpenAIResponses,
-});
+	registerApiProvider({
+		api: "openai-responses",
+		stream: streamOpenAIResponses,
+		streamSimple: streamSimpleOpenAIResponses,
+	});
 
-registerApiProvider({
-	api: "azure-openai-responses",
-	stream: streamAzureOpenAIResponses,
-	streamSimple: streamSimpleAzureOpenAIResponses,
-});
+	registerApiProvider({
+		api: "azure-openai-responses",
+		stream: streamAzureOpenAIResponses,
+		streamSimple: streamSimpleAzureOpenAIResponses,
+	});
 
-registerApiProvider({
-	api: "openai-codex-responses",
-	stream: streamOpenAICodexResponses,
-	streamSimple: streamSimpleOpenAICodexResponses,
-});
+	registerApiProvider({
+		api: "openai-codex-responses",
+		stream: streamOpenAICodexResponses,
+		streamSimple: streamSimpleOpenAICodexResponses,
+	});
 
-registerApiProvider({
-	api: "google-generative-ai",
-	stream: streamGoogle,
-	streamSimple: streamSimpleGoogle,
-});
+	registerApiProvider({
+		api: "google-generative-ai",
+		stream: streamGoogle,
+		streamSimple: streamSimpleGoogle,
+	});
 
-registerApiProvider({
-	api: "google-gemini-cli",
-	stream: streamGoogleGeminiCli,
-	streamSimple: streamSimpleGoogleGeminiCli,
-});
+	registerApiProvider({
+		api: "google-gemini-cli",
+		stream: streamGoogleGeminiCli,
+		streamSimple: streamSimpleGoogleGeminiCli,
+	});
 
-registerApiProvider({
-	api: "google-vertex",
-	stream: streamGoogleVertex,
-	streamSimple: streamSimpleGoogleVertex,
-});
+	registerApiProvider({
+		api: "google-vertex",
+		stream: streamGoogleVertex,
+		streamSimple: streamSimpleGoogleVertex,
+	});
 
-registerApiProvider({
-	api: "bedrock-converse-stream",
-	stream: streamBedrock,
-	streamSimple: streamSimpleBedrock,
-});
+	registerApiProvider({
+		api: "bedrock-converse-stream",
+		stream: streamBedrock,
+		streamSimple: streamSimpleBedrock,
+	});
+}
+
+export function resetApiProviders(): void {
+	clearApiProviders();
+	registerBuiltInApiProviders();
+}
+
+registerBuiltInApiProviders();

packages/ai/src/utils/event-stream.ts

@@ -80,3 +80,8 @@ export class AssistantMessageEventStream extends EventStream<AssistantMessageEve
 		);
 	}
 }
+
+/** Factory function for AssistantMessageEventStream (for use in extensions) */
+export function createAssistantMessageEventStream(): AssistantMessageEventStream {
+	return new AssistantMessageEventStream();
+}