How to create new tRPC api route calling external api

import { z } from "zod";

import {
  createTRPCRouter,
  protectedProcedure,
} from "~/server/api/trpc";
import { env } from "~/env.mjs";
import { Configuration, OpenAIApi } from "openai";

const configuration = new Configuration({
    apiKey: env.OPEN_AI_KEY
});

const openai = new OpenAIApi(configuration);

const getStory = async (storySummary: string, title: string) => {
    const response = await openai.createCompletion({
        model: "text-davinci-003",
        prompt: `Write me a story based on the story summary:
        ${storySummary}

        with the title:
        ${title}
        `
    })

    return response;
}
export const promptRouter = createTRPCRouter({
  createStory: protectedProcedure
    .input(z.object({ 
        storySummary: z.string(),
        title: z.string()
    }))
    .query(async ({ input }) => {
        const response = await getStory(input.storySummary, input.title);

        return response.data;
    })
});

import React from 'react'
import { api } from '~/utils/api'
const Prompt = () => {
  const { data: response } = api.prompt.createStory.useQuery({ 
    storySummary: 'A little piggy lost her family in a giant thunderstorm and had to live by herself',
    title: 'The piggy tale'
  })
  return (
    <div className='bg-white md:px-6 mb-6 mt-2 shadow-lg rounded-md'>
        {response?.choices[0]?.text ?? 'Response is undefined'}
    </div>
  )
}

export default Prompt;