@gravityai-dev/openai
Version:
OpenAI integration for Gravity workflow system
97 lines • 4.69 kB
JavaScript
;
/**
* @gravity/openai
* OpenAI integration for Gravity platform
*/
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || (function () {
var ownKeys = function(o) {
ownKeys = Object.getOwnPropertyNames || function (o) {
var ar = [];
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
return ar;
};
return ownKeys(o);
};
return function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
__setModuleDefault(result, mod);
return result;
};
})();
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const plugin_base_1 = require("@gravityai-dev/plugin-base");
const package_json_1 = __importDefault(require("../package.json"));
const plugin = (0, plugin_base_1.createPlugin)({
name: package_json_1.default.name,
version: package_json_1.default.version,
description: package_json_1.default.description,
async setup(api) {
// First, set up platform dependencies
const { setPlatformDependencies } = await Promise.resolve().then(() => __importStar(require("@gravityai-dev/plugin-base")));
setPlatformDependencies({
PromiseNode: api.classes.PromiseNode,
CallbackNode: api.classes.CallbackNode,
NodeInputType: api.types.NodeInputType,
NodeConcurrency: api.types.NodeConcurrency,
getNodeCredentials: api.getNodeCredentials,
getConfig: api.getConfig,
createLogger: api.createLogger,
saveTokenUsage: api.saveTokenUsage,
NodeInput: null,
NodeOutput: null,
NodeDefinition: null,
NodeExecutor: null,
NodeExecutionContext: null,
NodeLifecycle: null,
WorkflowNode: null,
EnhancedNodeDefinition: null,
NodeCredential: null,
ValidationResult: null,
});
// Import nodes after dependencies are set
const { OpenAINode } = await Promise.resolve().then(() => __importStar(require("./OpenAI/node")));
const { OpenAIServiceNode } = await Promise.resolve().then(() => __importStar(require("./OpenAIService/node")));
const { OpenAIStreamNode } = await Promise.resolve().then(() => __importStar(require("./OpenAIStream/node")));
const { OpenAIEmbeddingServiceNode } = await Promise.resolve().then(() => __importStar(require("./OpenAIEmbeddingService/node")));
// Import credential
const { OpenAICredential } = await Promise.resolve().then(() => __importStar(require("./credentials")));
// Register nodes - pass the complete node objects
api.registerNode(OpenAINode);
api.registerNode(OpenAIServiceNode);
api.registerNode(OpenAIStreamNode);
api.registerNode(OpenAIEmbeddingServiceNode);
// Register credential
api.registerCredential(OpenAICredential);
// Import and register services
const { createEmbedding } = await Promise.resolve().then(() => __importStar(require("./OpenAIEmbeddingService/service/embeddings")));
const { queryChatGPT } = await Promise.resolve().then(() => __importStar(require("./OpenAI/service/queryChatGPT")));
const { streamCompletion } = await Promise.resolve().then(() => __importStar(require("./OpenAIStream/service/streaming")));
// Register services for platform use
api.registerService("openai-embeddings", createEmbedding);
api.registerService("openai-chat", queryChatGPT);
api.registerService("openai-stream", streamCompletion);
},
});
exports.default = plugin;
//# sourceMappingURL=index.js.map