Merge pull request #10 from get-convex/ian/fix-api

Ian/fix api

Author: ianmacartney

README.md

@@ -10,6 +10,7 @@ AI Agent framework built on Convex.
 - RAG for chat context, via hybrid text & vector search, with configuration options.
   Or use the API to query the history yourself and do it your way.
 - Opt-in search for messages from other threads (for the same specified user).
+- Support for generating and storing objects in messages (as JSON).
 - Tool calls via the AI SDK, along with Convex-specific helpers.
 - Easy workflow integration with the [Workflow component](https://convex.dev/components/workflow).
 - Reactive & realtime updates to asynchronous threads.

example/convex/_generated/api.d.ts

@@ -179,6 +179,7 @@ export declare const components: {
               | string
               | string
               | string;
+            error?: string;
             fileId?: string;
             id?: string;
             message?:
@@ -303,6 +304,7 @@ export declare const components: {
               | string
               | string
               | string;
+            error?: string;
             fileId?: string;
             id?: string;
             message?:
@@ -1096,6 +1098,7 @@ export declare const components: {
               | string
               | string
               | string;
+            error?: string;
             fileId?: string;
             id?: string;
             message?:
@@ -1268,6 +1271,7 @@ export declare const components: {
             | string
             | string
             | string;
+          error?: string;
           fileId?: string;
           id?: string;
           message?:
@@ -1397,6 +1401,7 @@ export declare const components: {
             | string
             | string
             | string;
+          error?: string;
           fileId?: string;
           id?: string;
           message?:

example/convex/example.ts

@@ -4,16 +4,17 @@ import { components, internal } from "./_generated/api";
 import { openai } from "@ai-sdk/openai";
 import { action, query } from "./_generated/server";
 import { v } from "convex/values";
+import { z } from "zod";
 
 // Define an agent similarly to the AI SDK
 const supportAgent = new Agent(components.agent, {
-  thread: openai.chat("gpt-4o-mini"),
+  chat: openai.chat("gpt-4o-mini"),
   textEmbedding: openai.embedding("text-embedding-3-small"),
   instructions: "You are a helpful assistant.",
 });
 
 const wordsmithAgent = new Agent(components.agent, {
-  thread: openai.chat("gpt-4o-mini"),
+  chat: openai.chat("gpt-4o-mini"),
   textEmbedding: openai.embedding("text-embedding-3-small"),
   instructions: "You output a spiffy quirky version of what the user says.",
 });
@@ -117,3 +118,19 @@ export const searchMessages = action({
     });
   },
 });
+
+export const generateObject = action({
+  args: { prompt: v.string() },
+  handler: async (ctx, { prompt }) => {
+    const { threadId, thread } = await supportAgent.createThread(ctx, {});
+    const result = await thread.streamObject({
+      output: "object",
+      schema: z.object({
+        name: z.string(),
+        age: z.number(),
+      }),
+      prompt,
+    });
+    return { threadId, object: result.object };
+  },
+});

example/convex/ideaAgents.ts

@@ -75,7 +75,7 @@ export const attachEntryToIdea = createTool({
 
 const ideaManagerAgent = new Agent(components.agent, {
   name: "Idea Manager Agent",
-  thread: openai.chat("gpt-4o"), // Fancier model for discerning between ideas
+  chat: openai.chat("gpt-4o"), // Fancier model for discerning between ideas
   textEmbedding: openai.embedding("text-embedding-3-small"),
   instructions:
     "You are a helpful assistant that helps manage ideas. " +
@@ -96,7 +96,7 @@ const ideaManagerAgent = new Agent(components.agent, {
 
 const ideaDevelopmentAgent = new Agent(components.agent, {
   name: "Idea Development Agent",
-  thread: openai.chat("gpt-4o-mini"),
+  chat: openai.chat("gpt-4o-mini"),
   textEmbedding: openai.embedding("text-embedding-3-small"),
   instructions:
     "You are a helpful assistant that helps develop ideas. " +
@@ -145,13 +145,22 @@ const ideaDeveloper = ideaDevelopmentAgent.asTool({
   maxSteps: 5,
 });
 
+/**
+ * AGENTS AS STANDALONE ACTIONS
+ */
+
+export const ideaManagerAction = ideaManagerAgent.asAction();
+export const ideaDeveloperAction = ideaDevelopmentAgent.asAction({
+  maxSteps: 5,
+});
+
 /**
  * AGENT DISPATCHERS
  */
 
 const ideaTriageAgent = new Agent(components.agent, {
   name: "Idea Triage Agent",
-  thread: openai.chat("gpt-4o-mini"),
+  chat: openai.chat("gpt-4o-mini"),
   textEmbedding: openai.embedding("text-embedding-3-small"),
   instructions:
     "You are a helpful assistant that helps triage ideas. " +

package-lock.json

@@ -7,7 +7,7 @@
     "email": "support@convex.dev",
     "url": "https://github.com/get-convex/agent/issues"
   },
-  "version": "0.0.1",
+  "version": "0.0.3",
   "license": "Apache-2.0",
   "keywords": [
     "convex",