payload-ai 0.0.52 → 0.0.54

package/README.md

@@ -121,6 +121,28 @@ plugins: [
   }),
 ],
 ```
+#### Change model for string translation
+
+To update the model you can change the collection settings in the same way as with other collections.
+
+```jsx
+plugins: [
+  aiTranslatorPlugin({
+    enabled: true,
+    stringTranslation: {
+      enabled: true
+    }
+    collections: {
+      translations: {
+        settings: {
+          model: 'gpt-4',
+        },
+      }
+    }
+  }),
+],
+```
+
 
 ### Generate SEO
 
@@ -133,6 +155,7 @@ plugins: [
 - custom access control
 - custom overrides for translation
 - generate images based on input
+- generate Open Graph based on content
 
 Payload is built with a robust infrastructure intended to support Plugins with ease. This provides a simple, modular, and reusable way for developers to extend the core capabilities of Payload.
 

package/dev/src/payload.config.ts

@@ -89,6 +89,11 @@ export default buildConfig({
         'examples-with-versions': {
           fields: ['title', 'longText', 'jsonContent'],
         },
+        translations: {
+          settings: {
+            model: 'gpt-4',
+          },
+        },
       },
     }),
     seo({

package/dev/src/server.ts

@@ -21,8 +21,6 @@ export const start = async (args?: Partial<InitOptions>) => {
     ...(args || {}),
   })
 
-  // Add your own express routes here
-
   app.listen(3000)
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "payload-ai",
-  "version": "0.0.52",
+  "version": "0.0.54",
   "homepage:": "https://polyxo.de",
   "repository": "git@github.com:payloadcms/payload-plugin-template.git",
   "description": "Payload AI tools",

package/src/aiTranslate.ts

@@ -4,10 +4,18 @@ import { deepCompareTranslateAndMerge } from './deepCompareAndMerge'
 
 const aiTranslateHook =
   (
-    { collectionOptions, collection }: { collectionOptions: any; collection: object },
+    {
+      collectionOptions,
+      collection,
+      pluginOptions,
+    }: { collectionOptions: any; collection: object; pluginOptions: any },
     fallback?: string,
   ): CollectionAfterChangeHook =>
   async ({ doc, req, previousDoc, context, collection }) => {
+    const settings = pluginOptions.collections?.[collection.slug]?.settings
+
+    console.log('rreeeeq', settings)
+
     return await translateCollection({
       doc,
       req,
@@ -15,6 +23,7 @@ const aiTranslateHook =
       context,
       collection,
       collectionOptions,
+      settings,
     })
   }
 
@@ -29,6 +38,7 @@ export async function translateCollection({
   collectionOptions,
   onlyMissing,
   codes,
+  settings,
 }: any) {
   if (
     context.triggerAfterChange === false ||
@@ -38,15 +48,6 @@ export async function translateCollection({
 
   const localCodes: string[] = req.payload.config.localization.localeCodes
 
-  console.log(
-    'filter codes',
-    localCodes.filter(
-      targetLanguage =>
-        targetLanguage !== req.payload.config.localization.defaultLocale &&
-        (!codes || codes.includes(targetLanguage)),
-    ),
-  )
-
   const translationPromises = localCodes
     .filter(
       targetLanguage =>
@@ -72,6 +73,7 @@ export async function translateCollection({
         previousDoc.id ? 'update' : 'create',
         onlyMissing,
         req.payload.config.localization.defaultLocale,
+        settings,
       )
 
       const { id, _status, updatedAt, createdAt, publishedDate, ...dataNew } =

package/src/components/Translator/index.tsx

@@ -4,6 +4,7 @@ import { Button, Drawer, DrawerToggler } from 'payload/components/elements'
 import { useDocumentInfo, useLocale } from 'payload/components/utilities'
 import { useModal } from '@faceless-ui/modal'
 import './Translator.scss'
+import { SelectInput, useForm } from 'payload/components/forms'
 
 const baseClass = 'after-dashboard'
 
@@ -46,11 +47,18 @@ export const DrawerTogglerAlt: any = ({
 
 export const Translator: React.FC = () => {
   const baseClass = 'ai-translator'
+
+  //const { fields, getDataByPath } = useForm()
+
   const [isLoading, setIsLoading] = React.useState(false)
+  const [selectedModel, setSelectedModel] = React.useState<string>('default')
 
   const locale = useLocale()
   const documentInfo: any = useDocumentInfo()
   const translate = async ({ codes }: any) => {
+    const settings = {
+      model: selectedModel === 'default' ? undefined : selectedModel,
+    }
     setIsLoading(true)
     try {
       const response = await fetch(`/api/${documentInfo.collection.slug}/translate`, {
@@ -61,6 +69,7 @@ export const Translator: React.FC = () => {
         body: JSON.stringify({
           id: documentInfo.id,
           codes,
+          settings,
         }),
       })
 
@@ -87,23 +96,52 @@ export const Translator: React.FC = () => {
   /*
  <Button onClick={handleClick}>Results are loading...</Button>
  */
+
+  const options = [
+    {
+      label: 'Default',
+      value: 'default',
+    },
+    {
+      label: 'GPT-3.5 Turbo (1106)',
+      value: 'gpt-3.5-turbo-1106',
+    },
+    {
+      label: 'GPT-4',
+      value: 'gpt-4',
+    },
+  ]
   return (
     <div className={baseClass}>
       <DrawerToggler slug="ai-translator" className={`${baseClass}__drawer__toggler`}>
         Translator
       </DrawerToggler>
+
       <Drawer title="Translator" slug="ai-translator">
         {isLoading ? (
           <Button disabled={true}>Results are loading...</Button>
         ) : (
-          <div className={`${baseClass}__translation-buttons`}>
-            <Button disabled={isLoading} onClick={() => translate({})}>
-              <span>Translate content to all languages</span>
-            </Button>
-            <Button disabled={isLoading} onClick={() => translate({ codes: [locale.code] })}>
-              <span>Translate only {locale.label as string}</span>
-            </Button>
-          </div>
+          <>
+            <div>
+              <SelectInput
+                onChange={(e: any) => {
+                  setSelectedModel(e.value)
+                }}
+                name="selectedModel"
+                value={selectedModel}
+                path="model"
+                options={options}
+              />
+            </div>
+            <div className={`${baseClass}__translation-buttons`}>
+              <Button disabled={isLoading} onClick={() => translate({})}>
+                <span>Translate content to all languages</span>
+              </Button>
+              <Button disabled={isLoading} onClick={() => translate({ codes: [locale.code] })}>
+                <span>Translate only {locale.label as string}</span>
+              </Button>
+            </div>
+          </>
         )}
       </Drawer>
     </div>

package/src/deepCompareAndMerge.ts

@@ -17,6 +17,7 @@ export async function deepCompareTranslateAndMerge(
   action?: 'create' | 'update',
   onlyMissing?: boolean,
   sourceLanguage?: string,
+  settings?: any,
 ): Promise<CollectionObjType> {
   if (Array.isArray(newOriginalObj)) {
     return Promise.all(
@@ -30,6 +31,7 @@ export async function deepCompareTranslateAndMerge(
           action,
           onlyMissing,
           sourceLanguage,
+          settings,
         ),
       ),
     )
@@ -45,11 +47,12 @@ export async function deepCompareTranslateAndMerge(
             // Translate the text and merge it into the English object
 
             if (!onlyMissing || targetObj[prop] === undefined || targetObj[prop] === '') {
-              targetObj[prop] = await translateTextOrObject(
-                newOriginalObj[prop],
+              targetObj[prop] = await translateTextOrObject({
+                text: newOriginalObj[prop],
                 language,
                 sourceLanguage,
-              )
+                settings,
+              })
             } else {
               // targetObj[prop] = 'not translated'
             }
@@ -67,6 +70,7 @@ export async function deepCompareTranslateAndMerge(
             action,
             onlyMissing,
             sourceLanguage,
+            settings,
           )
         }
       }

package/src/handleTranslate.ts

@@ -13,6 +13,8 @@ export const createTranslatorHandler = (translatorConfig: PluginTypes): PayloadH
 
     const collectionOptions = translatorConfig.collections[req.collection.config.slug]
 
+    const settings = req.body.settings || {}
+
     //  const translator = aiTranslate({ collectionOptions, collection: req.collection }, 'fallback')
 
     const result = await translateCollection({
@@ -24,6 +26,7 @@ export const createTranslatorHandler = (translatorConfig: PluginTypes): PayloadH
       collection: req.collection.config,
       onlyMissing: req.body.onlyMissing,
       codes: req.body.codes,
+      settings,
     })
     /*if (translatorConfig.access) {
       const hasAccesses = await translatorConfig.access(req)

package/src/plugin.ts

@@ -125,7 +125,7 @@ export const aiTranslatorPlugin =
       return config
     }
 
-    config.collections = [...(config.collections || []), stringTranslations]
+    config.collections = [...(config.collections || []), stringTranslations(pluginOptions)]
 
     config.globals = [
       ...(config.globals || []),

package/src/stringTranslations.ts

@@ -19,26 +19,28 @@ const TextAreaBlock: Block = {
   ],
 }
 
-const stringTranslations: CollectionConfig = {
-  slug: 'translations',
-  admin: {
-    useAsTitle: 'key',
-  },
-  access: {
-    read: anyone,
-    /*update: admins,
+const stringTranslations = (pluginOptions: any): CollectionConfig => {
+  return {
+    slug: 'translations',
+    admin: {
+      useAsTitle: 'key',
+    },
+    access: {
+      read: anyone,
+      /*update: admins,
     create: admins,
     delete: admins,*/
-  },
+    },
 
-  hooks: {
-    afterChange: [
-      aiTranslate({
-        collection: { slug: 'translations' },
-        collectionOptions: { fields: ['content'] },
-      }),
-      async (req: any) => {
-        /*if (req.user && req.user.role === 'admin') {
+    hooks: {
+      afterChange: [
+        aiTranslate({
+          collection: { slug: 'translations' },
+          collectionOptions: { fields: ['content'] },
+          pluginOptions,
+        }),
+        async (req: any) => {
+          /*if (req.user && req.user.role === 'admin') {
         return req
       }
      
@@ -52,94 +54,95 @@ const stringTranslations: CollectionConfig = {
           },
         },
       }*/
-      },
-    ],
-  },
-  endpoints: [
-    {
-      path: '/create-missing',
-      //path: '/:id/tracking',
-      method: 'post',
-      handler: async (req: any, res: any, next: any) => {
-        const posts = await req.payload.find({
-          collection: 'translations',
-          where: {
-            key: {
-              equals: req.body.key,
-            },
-          },
-        })
-
-        if (posts.docs.length > 0) {
-          res.status(200).send(posts.docs)
-        } else {
-          const newPost = await req.payload.create({
+        },
+      ],
+    },
+    endpoints: [
+      {
+        path: '/create-missing',
+        //path: '/:id/tracking',
+        method: 'post',
+        handler: async (req: any, res: any, next: any) => {
+          const posts = await req.payload.find({
             collection: 'translations',
-            data: {
-              key: req.body.key,
-              namespace: req.body.namespace,
-              translation: [
-                {
-                  content: req.body.content,
-                  blockType: 'translation-textarea',
-                },
-              ],
+            where: {
+              key: {
+                equals: req.body.key,
+              },
             },
           })
 
-          res.status(200).send(newPost)
-          //res.status(404).send({ error: 'not found' })
-        }
+          if (posts.docs.length > 0) {
+            res.status(200).send(posts.docs)
+          } else {
+            const newPost = await req.payload.create({
+              collection: 'translations',
+              data: {
+                key: req.body.key,
+                namespace: req.body.namespace,
+                translation: [
+                  {
+                    content: req.body.content,
+                    blockType: 'translation-textarea',
+                  },
+                ],
+              },
+            })
+
+            res.status(200).send(newPost)
+            //res.status(404).send({ error: 'not found' })
+          }
+        },
       },
-    },
-    {
-      path: '/translate',
-      method: 'post',
-      handler: createTranslatorHandler({
-        collections: {
-          translations: {
-            fields: ['content'],
+      {
+        path: '/translate',
+        method: 'post',
+        handler: createTranslatorHandler({
+          collections: {
+            translations: {
+              fields: ['content'],
+            },
           },
-        },
-      }),
-    },
-  ],
-  fields: [
-    {
-      type: 'row',
-      fields: [
-        {
-          name: 'key',
-          type: 'text',
-          required: true,
-          admin: {
-            width: '50%',
+        }),
+      },
+    ],
+    fields: [
+      {
+        type: 'row',
+        fields: [
+          {
+            name: 'key',
+            type: 'text',
+            required: true,
+            admin: {
+              width: '50%',
+            },
           },
-        },
-        {
-          name: 'namespace',
-          type: 'text',
-        },
-      ],
-    },
-    {
-      name: 'translation',
-      type: 'blocks',
-      minRows: 1,
-      maxRows: 1,
-      blocks: [TextAreaBlock],
-    },
-    {
-      name: 'translator',
-      type: 'ui',
-      admin: {
-        position: 'sidebar',
-        components: {
-          Field: Translator,
-        },
+          {
+            name: 'namespace',
+            type: 'text',
+          },
+        ],
       },
-    } as Field,
-  ],
+      {
+        name: 'translation',
+        type: 'blocks',
+        minRows: 1,
+        maxRows: 1,
+        blocks: [TextAreaBlock],
+      },
+      {
+        name: 'translator',
+        type: 'ui',
+        admin: {
+          position: 'sidebar',
+          components: {
+            Field: Translator,
+          },
+        },
+      } as Field,
+    ],
+  }
 }
 
 export default stringTranslations

package/src/translateTextAndObjects.ts

@@ -94,12 +94,21 @@ function reapplyText(
   }
 }
 
-export async function translateTextOrObject(
-  text: any,
-  language: string,
-  sourceLanguage?: string,
-  retryCount: number = 0,
-) {
+interface TranslateTextOrObject {
+  text: any
+  language: string
+  sourceLanguage?: string
+  retryCount?: number
+  setting: any
+}
+
+export async function translateTextOrObject({
+  text,
+  language,
+  sourceLanguage,
+  retryCount = 0,
+  settings,
+}: any) {
   function isTranslateNode(node: any, key: string) {
     return (key === 'text' && typeof node[key] === 'string') || key === 'name'
   }
@@ -115,7 +124,6 @@ export async function translateTextOrObject(
       extractAndCapitalizeText(text.root, ['root'], textMap, isTranslateNode)
     }
 
-    console.log('textMap', JSON.stringify(text, null, 2))
     const chatCompletion = await openai.chat.completions.create({
       model: /* textAsString.length > 2000 ? 'gpt-3.5-turbo-16k' :*/ 'gpt-3.5-turbo-1106', /// 'gpt-3.5-turbo', // gpt-3.5-turbo-1106 // gpt-3.5-turbo-16k-0613
       messages:
@@ -131,6 +139,7 @@ export async function translateTextOrObject(
       top_p: 1,
       frequency_penalty: 0,
       presence_penalty: 0,
+      ...settings,
     })
 
     console.log(
@@ -162,12 +171,14 @@ export async function translateTextOrObject(
       )
 
       await new Promise(resolve => setTimeout(resolve, error.headers['retry-after-ms']))
-      const newResult: any = await translateTextOrObject(
+      const newResult: any = await translateTextOrObject({
         text,
         language,
         sourceLanguage,
-        retryCount + 1,
-      )
+        retryCount: retryCount + 1,
+        settings,
+      })
+
       return newResult
     } else {
       console.log(

package/dist/access/admins.d.ts

@@ -1,4 +0,0 @@
-import type { AccessArgs } from 'payload/config';
-type isAdmin = (args: AccessArgs<unknown, any>) => boolean;
-export declare const admins: isAdmin;
-export {};

package/dist/access/admins.js

@@ -1,12 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.admins = void 0;
-var checkRole_1 = require("./checkRole");
-var admins = function (_a) {
-    var user = _a.req.user;
-    if (!user)
-        return false;
-    return (0, checkRole_1.checkRole)(['admin'], user);
-};
-exports.admins = admins;
-//# sourceMappingURL=admins.js.map

package/dist/access/admins.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"admins.js","sourceRoot":"","sources":["../../src/access/admins.ts"],"names":[],"mappings":";;;AAEA,yCAAuC;AAKhC,IAAM,MAAM,GAAY,UAAC,EAAiB;QAAR,IAAI,cAAA;IAC3C,IAAI,CAAC,IAAI;QAAE,OAAO,KAAK,CAAA;IACvB,OAAO,IAAA,qBAAS,EAAC,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,CAAA;AACnC,CAAC,CAAA;AAHY,QAAA,MAAM,UAGlB"}

package/dist/access/adminsOrPublished.d.ts

@@ -1,2 +0,0 @@
-import type { Access } from 'payload/config';
-export declare const adminsOrPublished: Access;

package/dist/access/adminsOrPublished.js

@@ -1,18 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.adminsOrPublished = void 0;
-var checkRole_1 = require("./checkRole");
-var adminsOrPublished = function (_a) {
-    var user = _a.req.user;
-    if (user && (0, checkRole_1.checkRole)(['admin'], user)) {
-        return true;
-    }
-    return true;
-    return {
-        _status: {
-            equals: 'published',
-        },
-    };
-};
-exports.adminsOrPublished = adminsOrPublished;
-//# sourceMappingURL=adminsOrPublished.js.map

package/dist/access/adminsOrPublished.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"adminsOrPublished.js","sourceRoot":"","sources":["../../src/access/adminsOrPublished.ts"],"names":[],"mappings":";;;AAEA,yCAAuC;AAEhC,IAAM,iBAAiB,GAAW,UAAC,EAAiB;QAAR,IAAI,cAAA;IACrD,IAAI,IAAI,IAAI,IAAA,qBAAS,EAAC,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,EAAE;QACtC,OAAO,IAAI,CAAA;KACZ;IAED,OAAO,IAAI,CAAA;IACX,OAAO;QACL,OAAO,EAAE;YACP,MAAM,EAAE,WAAW;SACpB;KACF,CAAA;AACH,CAAC,CAAA;AAXY,QAAA,iBAAiB,qBAW7B"}

package/dist/access/anyone.d.ts

@@ -1,2 +0,0 @@
-import type { Access } from 'payload/config';
-export declare const anyone: Access;

package/dist/access/anyone.js

@@ -1,6 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.anyone = void 0;
-var anyone = function () { return true; };
-exports.anyone = anyone;
-//# sourceMappingURL=anyone.js.map

package/dist/access/anyone.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"anyone.js","sourceRoot":"","sources":["../../src/access/anyone.ts"],"names":[],"mappings":";;;AAEO,IAAM,MAAM,GAAW,cAAM,OAAA,IAAI,EAAJ,CAAI,CAAA;AAA3B,QAAA,MAAM,UAAqB"}

package/dist/access/checkRole.d.ts

@@ -1 +0,0 @@
-export declare const checkRole: (allRoles: any, user?: any) => boolean;

package/dist/access/checkRole.js

@@ -1,18 +0,0 @@
-"use strict";
-//import type { User } from '../../payload-types'
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.checkRole = void 0;
-var checkRole = function (allRoles /* User['roles'] = [] */, user) {
-    if (user) {
-        if (allRoles.some(function (role) {
-            var _a;
-            return (_a = user === null || user === void 0 ? void 0 : user.roles) === null || _a === void 0 ? void 0 : _a.some(function (individualRole) {
-                return individualRole === role;
-            });
-        }))
-            return true;
-    }
-    return false;
-};
-exports.checkRole = checkRole;
-//# sourceMappingURL=checkRole.js.map

package/dist/access/checkRole.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"checkRole.js","sourceRoot":"","sources":["../../src/access/checkRole.ts"],"names":[],"mappings":";AAAA,iDAAiD;;;AAE1C,IAAM,SAAS,GAAG,UAAC,QAAa,CAAC,wBAAwB,EAAE,IAAU;IAC1E,IAAI,IAAI,EAAE;QACR,IACE,QAAQ,CAAC,IAAI,CAAC,UAAC,IAAS;;YACtB,OAAO,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,KAAK,0CAAE,IAAI,CAAC,UAAC,cAAmB;gBAC3C,OAAO,cAAc,KAAK,IAAI,CAAA;YAChC,CAAC,CAAC,CAAA;QACJ,CAAC,CAAC;YAEF,OAAO,IAAI,CAAA;KACd;IAED,OAAO,KAAK,CAAA;AACd,CAAC,CAAA;AAbY,QAAA,SAAS,aAarB"}

package/dist/aiTranslate.d.ts

@@ -1,7 +0,0 @@
-import type { CollectionAfterChangeHook } from 'payload/types';
-declare const aiTranslateHook: ({ collectionOptions, collection }: {
-    collectionOptions: any;
-    collection: object;
-}, fallback?: string) => CollectionAfterChangeHook;
-export default aiTranslateHook;
-export declare function translateCollection({ req, doc, collection, previousDoc, context, collectionOptions, onlyMissing, codes, }: any): Promise<void>;