@automattic/jetpack-ai-client 0.1.16 → 0.2.1

package/src/data-flow/with-ai-assistant-data.tsx

@@ -3,59 +3,62 @@
  */
 import { createHigherOrderComponent } from '@wordpress/compose';
 import { useMemo } from '@wordpress/element';
-import React, { ReactNode } from 'react';
+import React from 'react';
 /**
  * Internal Dependencies
  */
-import { useAiSuggestions } from '../../';
+import useAiSuggestions from '../hooks/use-ai-suggestions';
 import { AiDataContextProvider } from '.';
 
 /**
  * High Order Component that provides the
  * AI Assistant Data context to the wrapped component.
  *
- * @param {ReactNode} WrappedComponent - component to wrap.
- * @returns {ReactNode}          		 Wrapped component, with the AI Assistant Data context.
+ * @param {React.ReactElement} WrappedComponent - component to wrap.
+ * @returns {React.ReactElement}          		  Wrapped component, with the AI Assistant Data context.
  */
-const withAiDataProvider = createHigherOrderComponent( ( WrappedComponent: ReactNode ) => {
-	return props => {
-		// Connect with the AI Assistant communication layer.
-		const {
-			suggestion,
-			error: requestingError,
-			requestingState,
-			request: requestSuggestion,
-			stopSuggestion,
-			eventSource,
-		} = useAiSuggestions();
-
-		// Build the context value to pass to the ai assistant data provider.
-		const dataContextValue = useMemo(
-			() => ( {
+const withAiDataProvider = createHigherOrderComponent(
+	( WrappedComponent: React.ComponentType ) => {
+		return props => {
+			// Connect with the AI Assistant communication layer.
+			const {
 				suggestion,
-				requestingError,
+				error: requestingError,
 				requestingState,
-				eventSource,
-
-				requestSuggestion,
+				request: requestSuggestion,
 				stopSuggestion,
-			} ),
-			[
-				suggestion,
-				requestingError,
-				requestingState,
 				eventSource,
-				requestSuggestion,
-				stopSuggestion,
-			]
-		);
+			} = useAiSuggestions();
+
+			// Build the context value to pass to the ai assistant data provider.
+			const dataContextValue = useMemo(
+				() => ( {
+					suggestion,
+					requestingError,
+					requestingState,
+					eventSource,
+
+					requestSuggestion,
+					stopSuggestion,
+				} ),
+				[
+					suggestion,
+					requestingError,
+					requestingState,
+					eventSource,
+					requestSuggestion,
+					stopSuggestion,
+				]
+			);
 
-		return (
-			<AiDataContextProvider value={ dataContextValue }>
-				<WrappedComponent { ...props } />
-			</AiDataContextProvider>
-		);
-	};
-}, 'withAiDataProvider' );
+			return (
+				<AiDataContextProvider value={ dataContextValue }>
+					<WrappedComponent { ...props } />
+				</AiDataContextProvider>
+			);
+		};
+	},
+	'withAiDataProvider'
+);
 
 export default withAiDataProvider;

package/src/index.ts

@@ -0,0 +1,32 @@
+/*
+ * Core library exports
+ */
+export { default as requestJwt } from './jwt';
+export { default as SuggestionsEventSource } from './suggestions-event-source';
+export { default as askQuestion } from './ask-question';
+
+/*
+ * Hooks
+ */
+export { default as useAiSuggestions } from './hooks/use-ai-suggestions';
+export { default as useMediaRecording } from './hooks/use-media-recording';
+
+/*
+ * Components: Icons
+ */
+export * from './icons';
+
+/*
+ * Components
+ */
+export * from './components';
+
+/*
+ * Contexts
+ */
+export * from './data-flow';
+
+/*
+ * Types
+ */
+export * from './types';

package/src/types.ts

@@ -60,3 +60,20 @@ export const AI_MODEL_GPT_3_5_Turbo_16K = 'gpt-3.5-turbo-16k' as const;
 export const AI_MODEL_GPT_4 = 'gpt-4' as const;
 
 export type AiModelTypeProp = typeof AI_MODEL_GPT_3_5_Turbo_16K | typeof AI_MODEL_GPT_4;
+
+// Connection initial state
+// @todo: it should be provided by the connection package
+interface JPConnectionInitialState {
+	apiNonce: string;
+	siteSuffix: string;
+	connectionStatus: {
+		isActive: boolean;
+	};
+}
+
+// Global
+declare global {
+	interface Window {
+		JP_CONNECTION_INITIAL_STATE: JPConnectionInitialState;
+	}
+}

package/index.ts

@@ -1,32 +0,0 @@
-/*
- * Core library exports
- */
-export { default as requestJwt } from './src/jwt';
-export { default as SuggestionsEventSource } from './src/suggestions-event-source';
-export { default as askQuestion } from './src/ask-question';
-
-/*
- * Hooks
- */
-export { default as useAiSuggestions } from './src/hooks/use-ai-suggestions';
-export { default as useMediaRecording } from './src/hooks/use-media-recording';
-
-/*
- * Components: Icons
- */
-export * from './src/icons';
-
-/*
- * Components
- */
-export * from './src/components';
-
-/*
- * Contexts
- */
-export * from './src/data-flow';
-
-/*
- * Types
- */
-export * from './src/types';

package/src/global.d.ts

@@ -1,9 +0,0 @@
-export declare global {
-	interface Window {
-		JP_CONNECTION_INITIAL_STATE: {
-			apiNonce: string;
-			siteSuffix: string;
-			siteId: string;
-		};
-	}
-}