Create a notion of preferredAIProvider in state.

Action creators and reducers, not yet used for anything. Part of #5.
jkomoros · Jan 7, 2024 · 9a030d2 · 9a030d2
1 parent db17ba8
commit 9a030d2
Show file tree

Hide file tree

Showing 5 changed files with 32 additions and 2 deletions.
diff --git a/app/actions.ts b/app/actions.ts
@@ -14,7 +14,8 @@ import {
 	promptSchema,
 	sproutNameSchema,
 	sproutStateSchema,
-	uncompiledPackagedSproutSchema
+	uncompiledPackagedSproutSchema,
+	modelProvider
 } from '../src/types.js';
 
 export const UPDATE_PAGE = 'UPDATE_PAGE';
@@ -40,6 +41,7 @@ export const FORCE_OPEN_API_KEYS_DIALOG = 'FORCE_OPEN_API_KEYS_DIALOG';
 //This is actually poorly named, because it doesn't force the editor to close,
 //it just lowers the forcedOpen flag.
 export const FORCE_CLOSE_API_KEYS_DIALOG = 'FORCE_CLOSE_API_KEYS_DIALOG';
+export const SET_PREFERRED_AI_PROVIDER = 'SET_PREFERRED_AI_PROVIDER';
 
 const actionUpdatePage = z.object({
 	type: z.literal(UPDATE_PAGE),
@@ -139,6 +141,11 @@ const actionForceCloseAPIKeysDialog = z.object({
 	type: z.literal(FORCE_CLOSE_API_KEYS_DIALOG)
 }).strict();
 
+const actionSetPreferredAIProvider = z.object({
+	type: z.literal(SET_PREFERRED_AI_PROVIDER),
+	provider: modelProvider
+}).strict();
+
 const someAction = z.discriminatedUnion('type', [
 	actionUpdatePage,
 	actionUpdateOffline,
@@ -159,7 +166,8 @@ const someAction = z.discriminatedUnion('type', [
 	actionEditingModifySprout,
 	actionWriteSprout,
 	actionForceCloseAPIKeysDialog,
-	actionForceOpenAPIKeysDialog
+	actionForceOpenAPIKeysDialog,
+	actionSetPreferredAIProvider
 ]);
 
 export type SomeAction = z.infer<typeof someAction>;
diff --git a/app/actions/data.ts b/app/actions/data.ts
@@ -27,6 +27,7 @@ import {
 	selectDraftMessage,
 	selectEditorOpen,
 	selectIsEditing,
+	selectPreferredAIProvider,
 	selectSproutData,
 	selectSproutSnapshot,
 	selectSproutStreaming
@@ -46,6 +47,7 @@ import {
 	APIKeys,
 	DirectoryInfo,
 	ImageURL,
+	ModelProvider,
 	Path,
 	Prompt,
 	SproutName,
@@ -369,4 +371,14 @@ export const forceClosedAPIKeysDialog = () : SomeAction => {
 	return {
 		type: FORCE_CLOSE_API_KEYS_DIALOG
 	};
+};
+
+export const setPreferredAIProvider = (provider : ModelProvider) : ThunkSomeAction => (dispatch, getState) => {
+	const state = getState();
+	const currentProvider = selectPreferredAIProvider(state);
+	if (currentProvider === provider) return;
+	dispatch({
+		type: 'SET_PREFERRED_AI_PROVIDER',
+		provider
+	});
 };
diff --git a/app/reducers/data.ts b/app/reducers/data.ts
@@ -8,6 +8,7 @@ import {
 	OPEN_EDITOR,
 	SELECT_SPROUT,
 	SET_API_KEYS,
+	SET_PREFERRED_AI_PROVIDER,
 	SPROUT_PROVIDED_USER_RESPONSE,
 	SPROUT_STOPPED_STREAMING,
 	START_EDITING,
@@ -28,6 +29,7 @@ const INITIAL_STATE : DataState = {
 		'anthropic.com': ''
 	},
 	apiKeysEditorForcedOpen: false,
+	preferredAIProvider: 'openai.com',
 	sprouts: {},
 	currentSproutName: null,
 	sproutStreaming: false,
@@ -141,6 +143,11 @@ const data = (state : DataState = INITIAL_STATE, action : SomeAction) : DataStat
 			...state,
 			apiKeysEditorForcedOpen: false
 		};
+	case SET_PREFERRED_AI_PROVIDER:
+		return {
+			...state,
+			preferredAIProvider: action.provider
+		};
 	default:
 		return state;
 	}

diff --git a/app/selectors.ts b/app/selectors.ts
@@ -39,6 +39,7 @@ export const selectIsEditing = (state : RootState) => state.data ? state.data.ed
 export const selectChangesMade = (state : RootState) => state.data ? state.data.changesMade : false;
 export const selectSproutSnapshot = (state : RootState) => state.data ? state.data.sproutSnapshot : null;
 export const selectWrittenSprouts = (state : RootState) => state.data ? state.data.writtenSprouts : {};
+export const selectPreferredAIProvider = (state : RootState) => state.data ? state.data.preferredAIProvider : 'openai.com';
 const selectAPIKeysEditorForcedOpen = (state : RootState) => state.data ? state.data.apiKeysEditorForcedOpen : false;
 
 export const selectHashForCurrentState = (_state : RootState) => '';

diff --git a/app/types_store.ts b/app/types_store.ts
@@ -1,6 +1,7 @@
 import {
 	APIKeys,
 	ImageURL,
+	ModelProvider,
 	PackagedSprout,
 	SproutName,
 	UncompiledPackagedSprout
@@ -25,6 +26,7 @@ export type DataState = {
 	sprouts: SproutDataMap,
 	currentSproutName: SproutLocation | null,
 	sproutStreaming: boolean,
+	preferredAIProvider : ModelProvider,
 	//The conversation is managed by the currentSprout, but we need to tell the
 	//selector machinery once we have reason to believe its state has changed.
 	streamCounter: number,