@elyx-code/editor-ui 0.0.2

package/src/services/editor/modules/operations-implementations/external-integrations/google-gemini/generate-content.ts

@@ -0,0 +1,193 @@
+import {
+	ActionDescriptorState,
+	OperationState,
+	IPrimitiveExecutionResult,
+	ExecutionTerminationType,
+	DefinitionEntityState,
+	IDynamicValue,
+	BaseRegisteredExtension,
+	ExtensionContextType,
+	ExtensionManifest,
+	ExtensionPermissions,
+	ExtentionPriority,
+	IRegisteredExtension,
+	DefaultEditorContext,
+	DefaultProjectLogicContext,
+	getConnectionIdFromDefinitionEntity,
+	OperationsImplementation,
+	BUILT_IN_FUNCTION_IMPLEMENTATIONS__PREFIX,
+	ExternalIntegrationOperations,
+	ExternalIntegrationNames,
+	InputMapState,
+	BUILT_IN_BASE_ENTITY_IDS,
+} from '@elyx-code/project-logic-tree';
+import { EditorService } from '../../../../EditorService';
+import { Logger } from '@elyx-code/common-ts-utils';
+
+export class OperationsImplementationsGoogleGeminiGenerateContentModule
+	extends BaseRegisteredExtension
+	implements IRegisteredExtension<OperationsImplementation>
+{
+	manifest: ExtensionManifest = {
+		id:
+			BUILT_IN_FUNCTION_IMPLEMENTATIONS__PREFIX +
+			ExternalIntegrationOperations.GoogleGeminiGenerateContent,
+		version: '1.0.0',
+		description: 'Send Gmail email operation implementation module',
+		initPriority: ExtentionPriority.High,
+		permissions: [ExtensionPermissions.All],
+		contexts: [ExtensionContextType.Editor, ExtensionContextType.Logic],
+	};
+
+	module = {
+		init: (contexts: (DefaultEditorContext | DefaultProjectLogicContext)[]) => {
+			const projectState = contexts.find(
+				(context) => context.contextType === ExtensionContextType.Logic,
+			) as DefaultProjectLogicContext;
+
+			if (projectState) {
+				const operation = projectState.get(
+					ExternalIntegrationOperations.GoogleGeminiGenerateContent,
+				) as ActionDescriptorState;
+
+				if (operation) {
+					operation.implementation = this.module.main;
+				}
+			}
+		},
+
+		main: async (
+			entity: OperationState,
+			_inputs: IDynamicValue[],
+		): Promise<IPrimitiveExecutionResult> => {
+			try {
+				const connectionEntityDynamicValueInput = _inputs.find(
+					(input) =>
+						(input.valueOwner as InputMapState)?.declaration?.id ===
+						BUILT_IN_BASE_ENTITY_IDS.operation[
+							'external-integration--google-gemini--post--generate-content'
+						].inputs.connection.id,
+				);
+
+				const connectionEntity =
+					connectionEntityDynamicValueInput?.value?.valueAsType;
+
+				if (!connectionEntity) {
+					return {
+						value:
+							entity.catchesError || entity.errorCalls.length
+								? ExecutionTerminationType.CaughtError
+								: ExecutionTerminationType.UnhandledError,
+						error: 'No entity template provided',
+						result: null,
+					};
+				}
+
+				if (!(connectionEntity instanceof DefinitionEntityState)) {
+					return {
+						value:
+							entity.catchesError || entity.errorCalls.length
+								? ExecutionTerminationType.CaughtError
+								: ExecutionTerminationType.UnhandledError,
+						error: 'Invalid entity template value provided',
+						result: null,
+					};
+				}
+
+				const promptDynamicValueInput = _inputs.find(
+					(input) =>
+						(input.valueOwner as InputMapState)?.declaration?.id ===
+						BUILT_IN_BASE_ENTITY_IDS.operation[
+							'external-integration--google-gemini--post--generate-content'
+						].inputs.prompt.id,
+				);
+
+				const prompt = promptDynamicValueInput?.value?.value;
+
+				if (!prompt || typeof prompt !== 'string') {
+					return {
+						value:
+							entity.catchesError || entity.errorCalls.length
+								? ExecutionTerminationType.CaughtError
+								: ExecutionTerminationType.UnhandledError,
+						error: 'Invalid prompt value provided',
+						result: null,
+					};
+				}
+
+				const modelDynamicValueInput = _inputs.find(
+					(input) =>
+						(input.valueOwner as InputMapState)?.declaration?.id ===
+						BUILT_IN_BASE_ENTITY_IDS.operation[
+							'external-integration--google-gemini--post--generate-content'
+						].inputs.model.id,
+				);
+
+				const model = modelDynamicValueInput?.value?.value;
+
+				if (!model || typeof model !== 'string') {
+					return {
+						value:
+							entity.catchesError || entity.errorCalls.length
+								? ExecutionTerminationType.CaughtError
+								: ExecutionTerminationType.UnhandledError,
+						error: 'Invalid model value provided',
+						result: null,
+					};
+				}
+
+				const editor = this.extensionsRegistry.requestContexts(this, [
+					ExtensionContextType.Editor,
+				])[0] as EditorService;
+
+				const connectionId =
+					getConnectionIdFromDefinitionEntity(connectionEntity);
+
+				console.log(
+					'Triggering Google Gemini generate content action with connectionId:',
+					connectionId,
+				);
+
+				const actionResponse: {
+					content: string;
+				} = await editor.integrationsService.triggerAction(
+					ExternalIntegrationNames['google-gemini'],
+					connectionId as string,
+					'generate-content',
+					{
+						prompt,
+						model,
+					},
+				);
+
+				Logger.log(
+					'[Google Gemini] Generate content action response:',
+					actionResponse,
+				);
+
+				return {
+					value: ExecutionTerminationType.Success,
+					error: null,
+					result: {
+						content: actionResponse.content,
+					},
+				};
+			} catch (e: any) {
+				// Extract the juicy details from Axios if they exist
+				const errorData = e.response?.data || e.message || e;
+
+				Logger.error('[Google Gemini] Generate content failed:', errorData);
+
+				return {
+					value:
+						entity.catchesError || entity.errorCalls.length
+							? ExecutionTerminationType.CaughtError
+							: ExecutionTerminationType.UnhandledError,
+					// Pass the extracted error downstream instead of the raw Axios object
+					error: errorData,
+					result: null,
+				};
+			}
+		},
+	};
+}