mirror of
https://github.com/FranP-code/create-better-t-stack.git
synced 2025-10-12 23:52:15 +00:00
feat(cli): upgrade to ai sdk v5 (#487)
This commit is contained in:
@@ -21,32 +21,33 @@ import { Hono } from "hono";
|
||||
import { cors } from "hono/cors";
|
||||
import { logger } from "hono/logger";
|
||||
{{#if (and (includes examples "ai") (or (eq runtime "bun") (eq runtime "node")))}}
|
||||
import { streamText } from "ai";
|
||||
import { streamText, convertToModelMessages } from "ai";
|
||||
import { google } from "@ai-sdk/google";
|
||||
import { stream } from "hono/streaming";
|
||||
{{/if}}
|
||||
{{#if (and (includes examples "ai") (eq runtime "workers"))}}
|
||||
import { streamText } from "ai";
|
||||
import { stream } from "hono/streaming";
|
||||
import { streamText, convertToModelMessages } from "ai";
|
||||
import { createGoogleGenerativeAI } from "@ai-sdk/google";
|
||||
{{/if}}
|
||||
|
||||
const app = new Hono();
|
||||
|
||||
app.use(logger());
|
||||
app.use("/*", cors({
|
||||
{{#if (or (eq runtime "bun") (eq runtime "node"))}}
|
||||
origin: process.env.CORS_ORIGIN || "",
|
||||
{{/if}}
|
||||
{{#if (eq runtime "workers")}}
|
||||
origin: env.CORS_ORIGIN || "",
|
||||
{{/if}}
|
||||
allowMethods: ["GET", "POST", "OPTIONS"],
|
||||
{{#if auth}}
|
||||
allowHeaders: ["Content-Type", "Authorization"],
|
||||
credentials: true,
|
||||
{{/if}}
|
||||
}));
|
||||
app.use(
|
||||
"/*",
|
||||
cors({
|
||||
{{#if (or (eq runtime "bun") (eq runtime "node"))}}
|
||||
origin: process.env.CORS_ORIGIN || "",
|
||||
{{/if}}
|
||||
{{#if (eq runtime "workers")}}
|
||||
origin: env.CORS_ORIGIN || "",
|
||||
{{/if}}
|
||||
allowMethods: ["GET", "POST", "OPTIONS"],
|
||||
{{#if auth}}
|
||||
allowHeaders: ["Content-Type", "Authorization"],
|
||||
credentials: true,
|
||||
{{/if}}
|
||||
})
|
||||
);
|
||||
|
||||
{{#if auth}}
|
||||
app.on(["POST", "GET"], "/api/auth/**", (c) => auth.handler(c.req.raw));
|
||||
@@ -69,44 +70,43 @@ app.use("/rpc/*", async (c, next) => {
|
||||
{{/if}}
|
||||
|
||||
{{#if (eq api "trpc")}}
|
||||
app.use("/trpc/*", trpcServer({
|
||||
router: appRouter,
|
||||
createContext: (_opts, context) => {
|
||||
return createContext({ context });
|
||||
},
|
||||
}));
|
||||
app.use(
|
||||
"/trpc/*",
|
||||
trpcServer({
|
||||
router: appRouter,
|
||||
createContext: (_opts, context) => {
|
||||
return createContext({ context });
|
||||
},
|
||||
})
|
||||
);
|
||||
{{/if}}
|
||||
|
||||
{{#if (and (includes examples "ai") (or (eq runtime "bun") (eq runtime "node")))}}
|
||||
app.post("/ai", async (c) => {
|
||||
const body = await c.req.json();
|
||||
const messages = body.messages || [];
|
||||
const uiMessages = body.messages || [];
|
||||
const result = streamText({
|
||||
model: google("gemini-1.5-flash"),
|
||||
messages,
|
||||
messages: convertToModelMessages(uiMessages),
|
||||
});
|
||||
|
||||
c.header("X-Vercel-AI-Data-Stream", "v1");
|
||||
c.header("Content-Type", "text/plain; charset=utf-8");
|
||||
return stream(c, (stream) => stream.pipe(result.toDataStream()));
|
||||
return result.toUIMessageStreamResponse();
|
||||
});
|
||||
{{/if}}
|
||||
|
||||
{{#if (and (includes examples "ai") (eq runtime "workers"))}}
|
||||
app.post("/ai", async (c) => {
|
||||
const body = await c.req.json();
|
||||
const messages = body.messages || [];
|
||||
const uiMessages = body.messages || [];
|
||||
const google = createGoogleGenerativeAI({
|
||||
apiKey: env.GOOGLE_GENERATIVE_AI_API_KEY,
|
||||
});
|
||||
const result = streamText({
|
||||
model: google("gemini-1.5-flash"),
|
||||
messages,
|
||||
messages: convertToModelMessages(uiMessages),
|
||||
});
|
||||
|
||||
c.header("X-Vercel-AI-Data-Stream", "v1");
|
||||
c.header("Content-Type", "text/plain; charset=utf-8");
|
||||
return stream(c, (stream) => stream.pipe(result.toDataStream()));
|
||||
return result.toUIMessageStreamResponse();
|
||||
});
|
||||
{{/if}}
|
||||
|
||||
@@ -117,17 +117,20 @@ app.get("/", (c) => {
|
||||
{{#if (eq runtime "node")}}
|
||||
import { serve } from "@hono/node-server";
|
||||
|
||||
serve({
|
||||
fetch: app.fetch,
|
||||
port: 3000,
|
||||
}, (info) => {
|
||||
console.log(`Server is running on http://localhost:${info.port}`);
|
||||
});
|
||||
serve(
|
||||
{
|
||||
fetch: app.fetch,
|
||||
port: 3000,
|
||||
},
|
||||
(info) => {
|
||||
console.log(`Server is running on http://localhost:${info.port}`);
|
||||
}
|
||||
);
|
||||
{{else}}
|
||||
{{#if (eq runtime "bun")}}
|
||||
{{#if (eq runtime "bun")}}
|
||||
export default app;
|
||||
{{/if}}
|
||||
{{#if (eq runtime "workers")}}
|
||||
export default app;
|
||||
{{/if}}
|
||||
{{/if}}
|
||||
{{#if (eq runtime "workers")}}
|
||||
export default app;
|
||||
{{/if}}
|
||||
{{/if}}
|
||||
Reference in New Issue
Block a user