wehub-resource-sync
88 lines
2.3 KiB
TypeScript
88 lines
2.3 KiB
TypeScript
import {
|
|
CopilotRuntime,
|
|
OpenAIAdapter,
|
|
copilotRuntimeNextJSAppRouterEndpoint,
|
|
} from "@copilotkit/runtime";
|
|
import { ChatPromptTemplate } from "@langchain/core/prompts";
|
|
import { ChatOpenAI } from "@langchain/openai";
|
|
|
|
const runtime = new CopilotRuntime({
|
|
actions: [
|
|
{
|
|
name: "sayHello",
|
|
description: "Says hello to someone.",
|
|
parameters: [
|
|
{
|
|
name: "name",
|
|
type: "string",
|
|
description: "The name of the person to say hello to.",
|
|
required: true,
|
|
},
|
|
],
|
|
handler: async ({ name }) => {
|
|
const prompt = ChatPromptTemplate.fromMessages([
|
|
[
|
|
"system",
|
|
"The user tells you their name. Say hello to the person in the most " +
|
|
" ridiculous way, roasting their name.",
|
|
],
|
|
["user", "My name is {name}"],
|
|
]);
|
|
const chain = prompt.pipe(new ChatOpenAI());
|
|
return chain.invoke({
|
|
name: name,
|
|
});
|
|
},
|
|
},
|
|
{
|
|
name: "sayGoodbye",
|
|
description: "Says goodbye to someone.",
|
|
parameters: [
|
|
{
|
|
name: "name",
|
|
type: "string",
|
|
description: "The name of the person to say goodbye to.",
|
|
required: true,
|
|
},
|
|
],
|
|
handler: async ({ name }) => {
|
|
const prompt = ChatPromptTemplate.fromMessages([
|
|
[
|
|
"system",
|
|
"The user tells you their name. Say goodbye to the person in the most " +
|
|
" ridiculous way, roasting their name.",
|
|
],
|
|
["user", "My name is {name}"],
|
|
]);
|
|
const chain = prompt.pipe(new ChatOpenAI());
|
|
return chain.stream({
|
|
name: name,
|
|
});
|
|
},
|
|
},
|
|
],
|
|
langserve: [
|
|
{
|
|
chainUrl: "http://localhost:8000/retriever",
|
|
name: "askAboutAnimals",
|
|
description:
|
|
"Always call this function if the users asks about a certain animal.",
|
|
},
|
|
{
|
|
chainUrl: "http://localhost:8000/agent",
|
|
name: "askAboutEugeneThoughts",
|
|
description:
|
|
"Always call this function if the users asks about Eugene's thoughts on a certain topic.",
|
|
},
|
|
],
|
|
});
|
|
|
|
const serviceAdapter = new OpenAIAdapter();
|
|
|
|
export const { GET, POST, OPTIONS } = copilotRuntimeNextJSAppRouterEndpoint({
|
|
runtime,
|
|
serviceAdapter,
|
|
endpoint: "/api/copilotkit/openai",
|
|
debug: true,
|
|
}) as any;
|