f15847a40f
* More typing for automations * undo test updates * remove casting of loop step and add more types * remove unused type * pro * Update types and fix ts issues * add branching types * remove any cast --------- Co-authored-by: Adria Navarro <adria@budibase.com>
109 lines
2.3 KiB
TypeScript
109 lines
2.3 KiB
TypeScript
import { OpenAI } from "openai"
|
|
|
|
import {
|
|
AutomationActionStepId,
|
|
AutomationStepDefinition,
|
|
AutomationStepType,
|
|
AutomationIOType,
|
|
OpenAIStepInputs,
|
|
OpenAIStepOutputs,
|
|
} from "@budibase/types"
|
|
import { env } from "@budibase/backend-core"
|
|
import * as automationUtils from "../automationUtils"
|
|
|
|
enum Model {
|
|
GPT_35_TURBO = "gpt-3.5-turbo",
|
|
// will only work with api keys that have access to the GPT4 API
|
|
GPT_4 = "gpt-4",
|
|
}
|
|
|
|
export const definition: AutomationStepDefinition = {
|
|
name: "OpenAI",
|
|
tagline: "Send prompts to ChatGPT",
|
|
icon: "Algorithm",
|
|
description: "Interact with the OpenAI ChatGPT API.",
|
|
type: AutomationStepType.ACTION,
|
|
internal: true,
|
|
features: {},
|
|
stepId: AutomationActionStepId.OPENAI,
|
|
inputs: {
|
|
prompt: "",
|
|
},
|
|
schema: {
|
|
inputs: {
|
|
properties: {
|
|
prompt: {
|
|
type: AutomationIOType.STRING,
|
|
title: "Prompt",
|
|
},
|
|
model: {
|
|
type: AutomationIOType.STRING,
|
|
title: "Model",
|
|
enum: Object.values(Model),
|
|
},
|
|
},
|
|
required: ["prompt", "model"],
|
|
},
|
|
outputs: {
|
|
properties: {
|
|
success: {
|
|
type: AutomationIOType.BOOLEAN,
|
|
description: "Whether the action was successful",
|
|
},
|
|
response: {
|
|
type: AutomationIOType.STRING,
|
|
description: "What was output",
|
|
},
|
|
},
|
|
required: ["success", "response"],
|
|
},
|
|
},
|
|
}
|
|
|
|
export async function run({
|
|
inputs,
|
|
}: {
|
|
inputs: OpenAIStepInputs
|
|
}): Promise<OpenAIStepOutputs> {
|
|
if (!env.OPENAI_API_KEY) {
|
|
return {
|
|
success: false,
|
|
response:
|
|
"OpenAI API Key not configured - please add the OPENAI_API_KEY environment variable.",
|
|
}
|
|
}
|
|
|
|
if (inputs.prompt == null) {
|
|
return {
|
|
success: false,
|
|
response: "Budibase OpenAI Automation Failed: No prompt supplied",
|
|
}
|
|
}
|
|
|
|
try {
|
|
const openai = new OpenAI({
|
|
apiKey: env.OPENAI_API_KEY,
|
|
})
|
|
|
|
const completion = await openai.chat.completions.create({
|
|
model: inputs.model,
|
|
messages: [
|
|
{
|
|
role: "user",
|
|
content: inputs.prompt,
|
|
},
|
|
],
|
|
})
|
|
const response = completion?.choices[0]?.message?.content
|
|
|
|
return {
|
|
response,
|
|
success: true,
|
|
}
|
|
} catch (err) {
|
|
return {
|
|
success: false,
|
|
response: automationUtils.getError(err),
|
|
}
|
|
}
|
|
}
|