@gravityai-dev/openai
Version:
OpenAI integration for Gravity workflow system
118 lines • 4.34 kB
JavaScript
;
/**
* OpenAI Node Definition
* Provides AI text generation capabilities using OpenAI's GPT models
*/
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.definition = exports.OpenAINode = void 0;
exports.createNodeDefinition = createNodeDefinition;
const plugin_base_1 = require("@gravityai-dev/plugin-base");
const executor_1 = __importDefault(require("./executor"));
// Export a function that creates the definition after platform deps are set
function createNodeDefinition() {
const { NodeInputType } = (0, plugin_base_1.getPlatformDependencies)();
return {
packageVersion: "1.0.25",
type: "OpenAI",
isService: false,
name: "OpenAI",
description: "Generate text using OpenAI's models",
category: "AI",
color: "#2F6F66",
logoUrl: "https://res.cloudinary.com/sonik/image/upload/v1749262616/gravity/icons/ChatGPT-Logo.svg.webp",
inputs: [
{
name: "signal",
type: NodeInputType.OBJECT,
description: "Data from previous nodes that can be referenced in templates",
},
],
outputs: [
{
name: "text",
type: NodeInputType.STRING,
description: "The generated text response from ChatGPT",
},
{
name: "usage",
type: NodeInputType.OBJECT,
description: "Token burn",
},
],
configSchema: {
type: "object",
properties: {
model: {
type: "string",
title: "Model",
description: "Select the OpenAI model to use",
enum: ["gpt-3.5-turbo", "gpt-4", "gpt-4-turbo-preview", "gpt-4o"],
enumNames: ["GPT-3.5 Turbo", "GPT-4", "GPT-4 Turbo", "GPT-4o"],
default: "gpt-3.5-turbo",
},
temperature: {
type: "number",
title: "Temperature",
description: "Controls randomness (0-2)",
default: 0.7,
minimum: 0,
maximum: 2,
},
maxTokens: {
type: "number",
title: "Max Tokens",
description: "Maximum number of tokens to generate",
default: 256,
minimum: 1,
maximum: 4096,
},
systemPrompt: {
type: "string",
title: "System Prompt",
description: "System message prompt. Supports template syntax like {{input.fieldName}} to reference input data.",
default: "",
"ui:field": "template",
},
prompt: {
type: "string",
title: "Prompt",
description: "User message/prompt. Supports template syntax like {{input.fieldName}} to reference input data.",
default: "",
"ui:field": "template",
},
history: {
type: "object",
title: "History",
description: "Message hostory [] for context",
default: "",
"ui:field": "template",
},
},
required: ["model"],
},
// This is where we declare credential requirements
credentials: [
{
name: "openAICredential",
required: true,
displayName: "OpenAI API",
description: "OpenAI API credentials for authentication",
},
],
capabilities: {
isTrigger: false,
},
};
}
// Create and export the node
const definition = createNodeDefinition();
exports.definition = definition;
// Export as enhanced node
exports.OpenAINode = {
definition,
executor: executor_1.default,
};
//# sourceMappingURL=index.js.map