wehub-resource-sync
41 lines
1.3 KiB
TypeScript
Vendored
41 lines
1.3 KiB
TypeScript
Vendored
'use server';
|
|
|
|
import { openai } from '@ai-sdk/openai';
|
|
import { createPortkey } from '@portkey-ai/vercel-provider';
|
|
import { streamText, generateText } from 'ai';
|
|
import { createStreamableValue } from 'ai/rsc';
|
|
import { z } from 'zod';
|
|
|
|
export const generateTextAction = async (location: string) => {
|
|
const llmClient = createPortkey({
|
|
apiKey: 'PORTKEY_API_KEY',
|
|
virtualKey: 'YOUR_OPENAI_VIRTUAL_KEY',
|
|
});
|
|
|
|
('use server');
|
|
const { toolResults, toolCalls } = await generateText({
|
|
model: llmClient.completionModel('gpt-3.5-turbo-instruct'),
|
|
temperature: 0.8,
|
|
prompt: `You are a funny chatbot. users location: ${location}`,
|
|
tools: {
|
|
weather: {
|
|
description: "Get the weather for the user's location",
|
|
parameters: z.object({
|
|
location: z.string().describe("user's location"),
|
|
}),
|
|
execute: async ({ location }) => {
|
|
const temperature = Math.floor(Math.random() * 31); // call external api for {location}
|
|
return { temperature };
|
|
},
|
|
},
|
|
},
|
|
});
|
|
if (toolResults && toolCalls) {
|
|
const joke = await streamText({
|
|
model: openai('gpt-4o'),
|
|
prompt: `Tell me a joke that incorporates ${location} and it's current temperature (${toolResults[0].result.temperature})`,
|
|
});
|
|
return createStreamableValue(joke.textStream).value;
|
|
}
|
|
};
|