@memberjunction/server 2.55.0 → 2.57.0

package/src/index.ts

@@ -37,6 +37,9 @@ LoadGeneratedActions(); // prevent tree shaking for this dynamic module
 import { LoadCoreEntitiesServerSubClasses } from '@memberjunction/core-entities-server';
 LoadCoreEntitiesServerSubClasses(); // prevent tree shaking for this dynamic module
 
+import { LoadAgentManagementActions } from '@memberjunction/ai-agent-manager-actions';
+LoadAgentManagementActions();
+
 import { ExternalChangeDetectorEngine } from '@memberjunction/external-change-detection';
 
 const cacheRefreshInterval = configInfo.databaseSettings.metadataCacheRefreshInterval;

package/src/resolvers/FileResolver.ts

@@ -1,4 +1,4 @@
-import { EntityPermissionType, Metadata } from '@memberjunction/core';
+import { EntityPermissionType, Metadata, FieldValueCollection } from '@memberjunction/core';
 import { FileEntity, FileStorageProviderEntity } from '@memberjunction/core-entities';
 import {
   AppContext,
@@ -57,23 +57,27 @@ export class FileResolver extends FileResolverBase {
     fileEntity.CheckPermissions(EntityPermissionType.Create, true);
 
     // Check to see if there's already an object with that name
-    const [sameName] = await this.findBy(context.dataSource, 'Files', { Name: input.Name, ProviderID: input.ProviderID }, context.userPayload.userRecord);
+    const [sameName] = await this.findBy(
+      context.dataSource,
+      'Files',
+      { Name: input.Name, ProviderID: input.ProviderID },
+      context.userPayload.userRecord
+    );
     const NameExists = Boolean(sameName);
 
-    const fileRecord = (await super.CreateFile({ ...input, Status: 'Pending' }, context, pubSub)) as File_;
+    const success = fileEntity.NewRecord(FieldValueCollection.FromObject({ ...input, Status: 'Pending' }));
 
     // If there's a problem creating the file record, the base resolver will return null
-    if (!fileRecord) {
+    if (!success) {
       return null;
     }
 
     // Create the upload URL and get the record updates (provider key, content type, etc)
-    const { updatedInput, UploadUrl } = await createUploadUrl(providerEntity, fileRecord);
+    const { updatedInput, UploadUrl } = await createUploadUrl(providerEntity, fileEntity);
 
     // Save the file record with the updated input
     const mapper = new FieldMapper();
-    await fileEntity.LoadFromData(mapper.ReverseMapFields({ ...input }));
-    fileEntity.SetMany(mapper.ReverseMapFields({ ...updatedInput }));
+    fileEntity.SetMany(mapper.ReverseMapFields({ ...updatedInput }), true, true);
     await fileEntity.Save();
     const File = mapper.MapFields({ ...fileEntity.GetAll() });
 

package/src/resolvers/RunAIPromptResolver.ts

@@ -45,8 +45,8 @@ export class RunAIPromptResolver extends ResolverBase {
         @Arg('promptId') promptId: string,
         @Ctx() { userPayload }: { userPayload: UserPayload },
         @Arg('data', { nullable: true }) data?: string,
-        @Arg('modelId', { nullable: true }) modelId?: string,
-        @Arg('vendorId', { nullable: true }) vendorId?: string,
+        @Arg('overrideModelId', { nullable: true }) overrideModelId?: string,
+        @Arg('overrideVendorId', { nullable: true }) overrideVendorId?: string,
         @Arg('configurationId', { nullable: true }) configurationId?: string,
         @Arg('skipValidation', { nullable: true }) skipValidation?: boolean,
         @Arg('templateData', { nullable: true }) templateData?: string,
@@ -137,12 +137,18 @@ export class RunAIPromptResolver extends ResolverBase {
             promptParams.prompt = promptEntity;
             promptParams.data = parsedData;
             promptParams.templateData = parsedTemplateData;
-            promptParams.modelId = modelId;
-            promptParams.vendorId = vendorId;
             promptParams.configurationId = configurationId;
             promptParams.contextUser = currentUser;
             promptParams.skipValidation = skipValidation || false;
             
+            // Set override if model or vendor ID provided
+            if (overrideModelId || overrideVendorId) {
+                promptParams.override = {
+                    modelId: overrideModelId,
+                    vendorId: overrideVendorId
+                };
+            }
+            
             // Parse and set conversation messages if provided
             if (messages) {
                 try {