add openapi-categorize-endpoint

.github/workflows/deploy-azure-functions-production.yml

@@ -36,7 +36,7 @@ jobs:
         run: |
           pushd './${{ env.AZURE_FUNCTIONAPP_PACKAGE_PATH }}'
           npm install
-          npm run graphql:codegen --if-present
+          npm run codegen --if-present
           npm run build --if-present
           popd
 

.github/workflows/test.yml

@@ -22,6 +22,6 @@ jobs:
       - name: Install Dependencies
         run: npm ci
       - name: Generate GraphQL types
-        run: npm run graphql:codegen
+        run: npm run codegen
       - name: Run Tests
         run: npm run test

package.json

@@ -19,7 +19,7 @@
     "generate-docs": "npx @redocly/cli build-docs docs/openapi.yaml -o docs/index.html --title 'Booking Api Documentation'",
     "generate-ts": "npm run bundle && npx orval --config ./orval.config.js",
     "postprocess": "node postprocess.js",
-    "graphql:codegen": "npx graphql-codegen -- --watch && npm run postprocess"
+    "codegen": "npx graphql-codegen -- --watch && npm run postprocess"
   },
   "dependencies": {
     "@azure/functions": "^4.3.0",
@@ -34,6 +34,7 @@
     "jsonwebtoken": "^9.0.2",
     "module-alias": "^2.2.3",
     "mongoose": "^7.6.10",
+    "openai": "^4.52.1",
     "zod": "^3.22.4"
   },
   "devDependencies": {

src/functions/openai.function.ts

@@ -0,0 +1,10 @@
+import { app } from "@azure/functions";
+import "module-alias/register";
+import { OpenAIControllerProductCategorize } from "./openai/controllers/product-categorize";
+
+app.http("openaiProductCategorize", {
+  methods: ["POST"],
+  authLevel: "anonymous",
+  route: "openai/products-categorize",
+  handler: OpenAIControllerProductCategorize,
+});

src/functions/openai/controllers/product-categorize.ts

@@ -0,0 +1,23 @@
+import { z } from "zod";
+import { _ } from "~/library/handler";
+import { OpenAIServiceProductCategorize } from "../services/product-categorize";
+
+export type OpenAIControllerProductCategorizeRequest = {
+  body: z.infer<typeof OpenAIServiceProductCategorizeSchema>;
+};
+
+export const OpenAIServiceProductCategorizeSchema = z.object({
+  title: z.string(),
+  description: z.string(),
+});
+
+export type OpenaiProductCategorizeResponse = Awaited<
+  ReturnType<typeof OpenAIServiceProductCategorize>
+>;
+
+export const OpenAIControllerProductCategorize = _(
+  ({ body }: OpenAIControllerProductCategorizeRequest) => {
+    const validateBody = OpenAIServiceProductCategorizeSchema.parse(body);
+    return OpenAIServiceProductCategorize(validateBody);
+  }
+);

src/functions/openai/services/product-categorize.ts

@@ -0,0 +1,94 @@
+import OpenAI from "openai";
+import { shopifyAdmin } from "~/library/shopify";
+import { CollectionsQuery } from "~/types/admin.generated";
+
+const openai = new OpenAI({
+  apiKey: process.env.OPENAI_API_KEY,
+});
+
+export const OpenAIServiceProductCategorize = async ({
+  title,
+  description,
+}: {
+  title: string;
+  description?: string;
+}) => {
+  try {
+    const { data } = await shopifyAdmin().request(COLLECTIONS);
+
+    const collections = data?.collections.nodes;
+
+    // Prepare collections data as context
+    const collectionsContext = JSON.stringify(collections, null, 2);
+
+    const content = `
+Given the following product title and description, response with the collection titles that this product fits into. The JSON structure should be:
+{
+  "collections": [
+    {
+      id: "gid://shopify/Collection/1111",
+      title: "example",
+      ruleSet: {
+        rules: [{
+          column
+          condition
+        }],
+      },
+    },
+  ],
+}
+Where:
+- "collections" includes the existing collections that the product fits into based on the given list of collections.
+### Existing Collections:
+${collectionsContext}
+### Product Details:
+Product Title: ${title}
+Product Description: ${description}
+If you think the product fits multiply collections, it's fine, include them all in the response.
+`;
+
+    const response = await openai.chat.completions.create({
+      model: "gpt-4o-2024-05-13",
+      messages: [
+        {
+          role: "system",
+          content,
+        },
+      ],
+      max_tokens: 300,
+      response_format: {
+        type: "json_object",
+      },
+    });
+
+    return JSON.parse(response.choices[0].message.content as any)
+      .collections as CollectionsQuery["collections"]["nodes"];
+  } catch (error) {
+    console.error("Error:", error);
+  }
+};
+
+const COLLECTION_FRAGMENT = `#graphql
+  fragment CollectionFragment on Collection {
+    id
+    title
+    description
+    ruleSet {
+      rules {
+        column
+        condition
+      }
+    }
+  }
+` as const;
+
+export const COLLECTIONS = `#graphql
+  ${COLLECTION_FRAGMENT}
+  query collections {
+    collections(first: 250, query: "-Alle AND -Subcategory AND -User") {
+      nodes {
+        ...CollectionFragment
+      }
+    }
+  }
+` as const;

src/types/admin.generated.d.ts

@@ -233,9 +233,23 @@ export type DestroyScheduleMetafieldMutationVariables = AdminTypes.Exact<{
 
 export type DestroyScheduleMetafieldMutation = { metafieldDelete?: AdminTypes.Maybe<Pick<AdminTypes.MetafieldDeletePayload, 'deletedId'>> };
 
+export type CollectionFragmentFragment = (
+  Pick<AdminTypes.Collection, 'id' | 'title' | 'description'>
+  & { ruleSet?: AdminTypes.Maybe<{ rules: Array<Pick<AdminTypes.CollectionRule, 'column' | 'condition'>> }> }
+);
+
+export type CollectionsQueryVariables = AdminTypes.Exact<{ [key: string]: never; }>;
+
+
+export type CollectionsQuery = { collections: { nodes: Array<(
+      Pick<AdminTypes.Collection, 'id' | 'title' | 'description'>
+      & { ruleSet?: AdminTypes.Maybe<{ rules: Array<Pick<AdminTypes.CollectionRule, 'column' | 'condition'>> }> }
+    )> } };
+
 interface GeneratedQueryTypes {
   "#graphql\n  query FileGet($id: ID!) {\n    node(id: $id) {\n      ... on MediaImage {\n        preview {\n          image {\n            url\n            width\n            height\n          }\n        }\n      }\n    }\n  }\n": {return: FileGetQuery, variables: FileGetQueryVariables},
   "#graphql\n  query publications {\n    publications(first: 10, catalogType: APP) {\n      nodes {\n        id\n      }\n    }\n  }\n": {return: PublicationsQuery, variables: PublicationsQueryVariables},
+  "#graphql\n  #graphql\n  fragment CollectionFragment on Collection {\n    id\n    title\n    description\n    ruleSet {\n      rules {\n        column\n        condition\n      }\n    }\n  }\n\n  query collections {\n    collections(first: 250, query: \"-Alle AND -Subcategory AND -User\") {\n      nodes {\n        ...CollectionFragment\n      }\n    }\n  }\n": {return: CollectionsQuery, variables: CollectionsQueryVariables},
 }
 
 interface GeneratedMutationTypes {