Merge pull request #256 from sarike/ds-openai

feat: openai datasource

Author: hanshuai

server/node-service/package.json

@@ -36,6 +36,7 @@
     "express": "^4.18.2",
     "express-async-errors": "^3.1.1",
     "firebase-admin": "^11.5.0",
+    "formdata-node": "4",
     "jsonpath": "^1.1.1",
     "lodash": "^4.17.21",
     "loglevel": "^1.8.1",

server/node-service/src/common/util.ts

@@ -52,3 +52,19 @@ export function toJsonValue(value: any): any {
     return {};
   }
 }
+
+export function toStringOrJson(value: any): any {
+  if (typeof value !== "string") {
+    return value;
+  }
+  try {
+    const json = relaxedJSONToJSON(value, true);
+    return JSON.parse(json);
+  } catch (e) {
+    if (typeof value === "string") {
+      return value;
+    }
+    console.info("invalid json input:", value);
+    return {};
+  }
+}

server/node-service/src/plugins/index.ts

@@ -7,6 +7,7 @@ import dynamoDBPlugin from "./dynamodb";
 import firebasePlugin from "./firebase";
 import couchdbPlugin from "./couchdb";
 import woocommercePlugin from "./woocommerce";
+import openAiPlugin from "./openAi";
 
 const plugins: (DataSourcePlugin | DataSourcePluginFactory)[] = [
   // helloWorldPlugin,
@@ -17,6 +18,7 @@ const plugins: (DataSourcePlugin | DataSourcePluginFactory)[] = [
   firebasePlugin,
   couchdbPlugin,
   woocommercePlugin,
+  openAiPlugin,
 ];
 
 export default plugins;

server/node-service/src/plugins/openAi/index.ts

@@ -0,0 +1,55 @@
+import _ from "lodash";
+import { OpenAPIV3, OpenAPI } from "openapi-types";
+import { ConfigToType, DataSourcePlugin } from "openblocks-sdk/dataSource";
+import { runOpenApi } from "../openApi";
+import { parseOpenApi, ParseOpenApiOptions } from "../openApi/parse";
+import spec from "./openai-spec.json";
+
+const dataSourceConfig = {
+  type: "dataSource",
+  params: [
+    {
+      key: "ApiKey.value",
+      type: "password",
+      label: "API Key",
+      rules: [{ required: true }],
+      placeholder: "<Your API KEY>",
+      tooltip:
+        "[Document](https://platform.openai.com/account/api-keys) on which you can create your api key",
+    },
+  ],
+} as const;
+
+const parseOptions: ParseOpenApiOptions = {
+  actionLabel: (method: string, path: string, operation: OpenAPI.Operation) => {
+    return _.upperFirst(operation.operationId || "");
+  },
+};
+
+type DataSourceConfigType = ConfigToType<typeof dataSourceConfig>;
+
+const openAiPlugin: DataSourcePlugin<any, DataSourceConfigType> = {
+  id: "OpenAI",
+  name: "OpenAI",
+  icon: "openAI.svg",
+  category: "api",
+  dataSourceConfig,
+  queryConfig: async () => {
+    const { actions } = await parseOpenApi(spec, parseOptions);
+    return {
+      type: "query",
+      label: "Action",
+      actions,
+    };
+  },
+  run: function (actionData, dataSourceConfig): Promise<any> {
+    const runApiDsConfig = {
+      url: "",
+      serverURL: "",
+      dynamicParamsConfig: dataSourceConfig,
+    };
+    return runOpenApi(actionData, runApiDsConfig, spec as OpenAPIV3.Document);
+  },
+};
+
+export default openAiPlugin;