fix: updates to the ai sdk provider

Author: dosco

README.md

@@ -21,6 +21,7 @@ We've renamed from "llmclient" to "ax" to highlight our focus on powering agenti
 - Build Agents that can call other agents
 - Convert docs of any format to text
 - RAG, smart chunking, embedding, querying
+- Works with Vercel AI SDK
 - Output validation while streaming
 - Multi-modal DSPy supported
 - Automatic prompt tuning using optimizers
@@ -289,6 +290,68 @@ if (tag === "technicalSupport") {
 }
 ```
 
+## Vercel AI SDK Integration
+
+Install the ax provider package
+
+```shell
+npm i @ax-llm/ax-ai-sdk-provider
+```
+
+Then use it with the AI SDK, you can either use the AI provider or the Agent Provider
+
+```typescript
+const ai = new AxAI({
+    name: 'openai',
+    apiKey: process.env['OPENAI_APIKEY'] ?? "",
+});
+
+// Create a model using the provider
+const model = new AxAIProvider(ai);
+
+export const foodAgent = new AxAgent(ai, {
+  name: 'food-search',
+  description:
+    'Use this agent to find restaurants based on what the customer wants',
+  signature,
+  functions
+})
+
+// Get vercel ai sdk state
+const aiState = getMutableAIState()
+
+// Create an agent for a specific task
+const foodAgent = new AxAgentProvider({
+    agent: foodAgent,
+    updateState: (state) => {
+         aiState.done({ ...aiState.get(), state })
+    },
+    generate: async ({ restaurant, priceRange }) => {
+        return (
+            <BotCard>
+                <h1>{restaurant as string} {priceRange as string}</h1>
+            </BotCard>
+        )
+    }
+})
+
+// Use with streamUI a critical part of building chat UIs in the AI SDK
+const result = await streamUI({
+    model,
+    initial: <SpinnerMessage />,
+    messages: [
+        // ...
+    ],
+    text: ({ content, done, delta }) => {
+        // ...
+    },
+    tools: {
+        // @ts-ignore
+        'find-food': foodAgent,
+    }
+})
+```
+
 ## OpenTelemetry support
 
 The ability to trace and observe your llm workflow is critical to building production workflows. OpenTelemetry is an industry-standard, and we support the new `gen_ai` attribute namespace.