Sveltekit调用OpenAI生成内容(ai sdk)

src/routes/api/completion/+server.ts:

import { createOpenAI } from '@ai-sdk/openai';
import { streamText } from 'ai';
import type { RequestHandler } from './$types';

import { OPENAI_API_KEY, OPENAI_BASE_URL } from '$env/static/private';

const openai = createOpenAI({
  baseURL: OPENAI_BASE_URL,
  apiKey: OPENAI_API_KEY,
});

export const POST = (async ({ request }) => {
  const { prompt } = await request.json();

  const result = streamText({
    model: openai('gpt-4o-mini'),
    prompt: prompt,
  });

  return result.toDataStreamResponse();
}) satisfies RequestHandler;

src/routes/+page.svelte:

<script>
    import { useCompletion } from "@ai-sdk/svelte";
    import { marked } from "marked";

    const { input, handleSubmit, completion } = useCompletion();
</script>

<main>
    <form on:submit={handleSubmit}>
        <input bind:value={$input} />
        <button type="submit">Send</button>
    </form>
    <div>{@html marked($completion)}</div>
</main>
posted @ 2025-02-16 11:56  卓能文  阅读(38)  评论(0)    收藏  举报