From 3a4d3b249f4b529ead2f80dd0df19a9cdc0c10a6 Mon Sep 17 00:00:00 2001
From: Dax Raad <d@ironbay.co>
Date: Wed, 28 May 2025 13:22:48 -0400
Subject: generate message type

---
 js/src/server/server.ts | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

(limited to 'js/src/server/server.ts')

diff --git a/js/src/server/server.ts b/js/src/server/server.ts
index 57f52a37c..ea70cd5ef 100644
--- a/js/src/server/server.ts
+++ b/js/src/server/server.ts
@@ -9,6 +9,7 @@ import { z } from "zod";
 import "zod-openapi/extend";
 import { Config } from "../app/config";
 import { LLM } from "../llm/llm";
+import { SessionMessage } from "./message";
 
 const SessionInfo = Session.Info.openapi({
   ref: "Session.Info",
@@ -118,7 +119,7 @@ export namespace Server {
               description: "Successfully created session",
               content: {
                 "application/json": {
-                  schema: resolver(z.custom<Session.Message[]>()),
+                  schema: resolver(SessionMessage.array()),
                 },
               },
             },
@@ -165,12 +166,12 @@ export namespace Server {
             sessionID: z.string(),
             providerID: z.string(),
             modelID: z.string(),
-            parts: z.custom<Session.Message["parts"]>(),
+            parts: SessionMessage.shape.parts,
           }),
         ),
         async (c) => {
           const body = c.req.valid("json");
-          const msg = await Session.chat(body);
+          const msg = await Session.chat(body as any);
           return c.json(msg);
         },
       )
-- 
cgit v1.2.3