Added new action query-codebase (#12904)

Author: jcortes

components/greptile/actions/query-codebase/query-codebase.mjs

@@ -0,0 +1,68 @@
+import app from "../../greptile.app.mjs";
+import utils from "../../common/utils.mjs";
+
+export default {
+  key: "greptile-query-codebase",
+  name: "Query Codebase",
+  description: "Search the user's codebase using a natural language query. [See the documentation](https://docs.greptile.com/apps/overview)",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    app,
+    query: {
+      type: "string",
+      label: "Query",
+      description: "A string containing the question in natural language.",
+    },
+    repositories: {
+      type: "string[]",
+      label: "Repositories",
+      description: "List of repositories indexed in Greptile to reference while answering your question. Array of JSON objects with keys `remote`, `branch` and `repository` eg. `{\"remote\":\"github\",\"branch\":\"main\",\"repository\":\"PipedreamHQ/pipedream\"}`",
+    },
+    sessionId: {
+      type: "string",
+      label: "Session ID",
+      description: "Only use this if you intend to need to retrieve chat history later.",
+      optional: true,
+    },
+    genius: {
+      type: "boolean",
+      label: "Genius",
+      description: "Genius requests are smarter but 8-10 seconds slower, great for complex usecases like reviewing PR and updating technical docs.",
+      optional: true,
+    },
+  },
+  methods: {
+    queryCodebase(args = {}) {
+      return this.app.post({
+        path: "/query",
+        ...args,
+      });
+    },
+  },
+  async run({ $ }) {
+    const {
+      queryCodebase,
+      query,
+      repositories,
+      sessionId,
+      genius,
+    } = this;
+
+    const response = await queryCodebase({
+      $,
+      data: {
+        messages: [
+          {
+            content: query,
+          },
+        ],
+        repositories: utils.parseArray(repositories),
+        sessionId,
+        genius,
+      },
+    });
+    $.export("$summary", "Successfully queried the codebase.");
+    return response;
+  },
+};

components/greptile/common/constants.mjs

@@ -0,0 +1,7 @@
+const BASE_URL = "https://api.greptile.com";
+const VERSION_PATH = "/v2";
+
+export default {
+  BASE_URL,
+  VERSION_PATH,
+};

components/greptile/common/utils.mjs

@@ -0,0 +1,50 @@
+import { ConfigurationError } from "@pipedream/platform";
+
+function isJson(value) {
+  value =
+    typeof(value) !== "string"
+      ? JSON.stringify(value)
+      : value;
+
+  try {
+    value = JSON.parse(value);
+  } catch (e) {
+    return false;
+  }
+
+  return typeof(value) === "object" && value !== null;
+}
+
+function valueToObject(value) {
+  if (!isJson(value)) {
+    return value;
+  }
+  return JSON.parse(value);
+}
+
+function parseArray(value) {
+  try {
+    if (!value) {
+      return [];
+    }
+
+    if (Array.isArray(value)) {
+      return value;
+    }
+
+    const parsedValue = JSON.parse(value);
+
+    if (!Array.isArray(parsedValue)) {
+      throw new Error("Not an array");
+    }
+
+    return parsedValue;
+
+  } catch (e) {
+    throw new ConfigurationError("Make sure the custom expression contains a valid JSON array object");
+  }
+}
+
+export default {
+  parseArray: (value) => parseArray(value).map(valueToObject),
+};

components/greptile/greptile.app.mjs

@@ -1,11 +1,33 @@
+import { axios } from "@pipedream/platform";
+import constants from "./common/constants.mjs";
+
 export default {
   type: "app",
   app: "greptile",
-  propDefinitions: {},
   methods: {
-    // this.$auth contains connected account data
-    authKeys() {
-      console.log(Object.keys(this.$auth));
+    getUrl(path) {
+      return `${constants.BASE_URL}${constants.VERSION_PATH}${path}`;
+    },
+    getHeaders(headers) {
+      return {
+        ...headers,
+        Authorization: `Bearer ${this.$auth.api_key}`,
+      };
+    },
+    _makeRequest({
+      $ = this, path, headers, ...args
+    } = {}) {
+      return axios($, {
+        ...args,
+        url: this.getUrl(path),
+        headers: this.getHeaders(headers),
+      });
+    },
+    post(args = {}) {
+      return this._makeRequest({
+        method: "POST",
+        ...args,
+      });
     },
   },
-};
+};

components/greptile/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pipedream/greptile",
-  "version": "0.0.1",
+  "version": "0.1.0",
   "description": "Pipedream Greptile Components",
   "main": "greptile.app.mjs",
   "keywords": [
@@ -11,5 +11,8 @@
   "author": "Pipedream <[email protected]> (https://pipedream.com/)",
   "publishConfig": {
     "access": "public"
+  },
+  "dependencies": {
+    "@pipedream/platform": "^3.0.0"
   }
-}
+}