Add stats for AI SDK usage

Author: timokoessler

README.md

@@ -104,6 +104,7 @@ Zen instruments the following AI SDKs to track which models are used and how man
 
 * ✅ [`openai`](https://www.npmjs.com/package/openai) 4.x
 * ✅ [`@aws-sdk/client-bedrock-runtime`](https://www.npmjs.com/package/@aws-sdk/client-bedrock-runtime) 3.x
+* ✅ [`ai`](https://www.npmjs.com/package/ai) 4.x
 
 _Note: Prompt injection attacks are currently not covered by Zen._
 

library/agent/protect.ts

@@ -51,6 +51,7 @@ import { Prisma } from "../sinks/Prisma";
 import { AwsSDKVersion2 } from "../sinks/AwsSDKVersion2";
 import { OpenAI } from "../sinks/OpenAI";
 import { AwsSDKVersion3 } from "../sinks/AwsSDKVersion3";
+import { AiSDK } from "../sinks/AiSDK";
 
 function getLogger(): Logger {
   if (isDebugging()) {
@@ -147,6 +148,7 @@ export function getWrappers() {
     new AwsSDKVersion3(),
     // new Function(), Disabled because functionName.constructor === Function is false after patching global
     new AwsSDKVersion2(),
+    new AiSDK(),
   ];
 }