fix: move api to internal

Author: astandrik

src/components/ComponentsProvider/chatPlaceholderTypes.ts

@@ -1,7 +1,5 @@
 import type React from 'react';
 
-import type {ChatConfig, ChatContext} from '../../types/chat';
-
 // Placeholder types for chat components
 // These will be properly typed when implemented in external packages
 
@@ -10,9 +8,9 @@ export interface AIAssistantButtonProps {
 }
 
 export interface ChatPanelProps {
-    config?: ChatConfig;
-    context?: ChatContext;
-    formatContext?: (context: ChatContext) => string;
+    config?: any; // Will be properly typed in the implementation
+    context?: any; // Will be properly typed in the implementation
+    formatContext?: (context: any) => string;
 }
 
 // Placeholder components that do nothing

src/containers/Heatmap/Heatmap.tsx

@@ -88,7 +88,9 @@ export const Heatmap = ({path, database}: HeatmapProps) => {
         const {min, max} = getCurrentMetricLimits(currentMetric, tablets);
 
         const preparedTablets = tablets.map((tablet) => {
-            const value = currentMetric && Number(tablet.metrics?.[currentMetric]);
+            const value =
+                currentMetric &&
+                Number(tablet.metrics?.[currentMetric as keyof typeof tablet.metrics]);
             const colorIndex = getColorIndex(value, min, max);
             const color = COLORS_RANGE[colorIndex];
 

src/lib.ts

@@ -37,23 +37,5 @@ export type {GetMonitoringLink, GetMonitoringClusterLink} from './utils/monitori
 
 export {configureUIFactory} from './uiFactory/uiFactory';
 
-// Export chat types
-export type {
-    ChatMessage,
-    ChatState,
-    ChatDelta,
-    ChatRequest,
-    ChatConfig,
-    ChatContext,
-    ChatCustomAction,
-    QuotaInfo,
-    ToolCall,
-    UsageBreakdown,
-} from './types/chat';
-
 // Re-export chat component types from ComponentsProvider
 export type {AIAssistantButtonProps, ChatPanelProps} from './components/ComponentsProvider';
-
-// Export placeholder chat reducer and actions
-// These can be overridden by external implementations
-export {chatReducer, chatActions} from './store/reducers/chat/chatPlaceholder';

src/services/api/aiAssist.ts

@@ -1,9 +1,7 @@
 import type {AxiosRequestConfig} from 'axios';
 
 import {codeAssistBackend} from '../../store';
-import {aiAssistBackend} from '../../store/configureStore';
 
-import {AIAssistAPI} from './aiAssist';
 import {AuthAPI} from './auth';
 import {CodeAssistAPI} from './codeAssist';
 import {MetaAPI} from './meta';
@@ -28,7 +26,6 @@ export class YdbEmbeddedAPI {
     viewer: ViewerAPI;
     meta?: MetaAPI;
     codeAssist?: CodeAssistAPI;
-    aiAssist?: AIAssistAPI;
 
     constructor({webVersion = false, withCredentials = false} = {}) {
         const config: AxiosRequestConfig = {withCredentials};
@@ -42,10 +39,6 @@ export class YdbEmbeddedAPI {
             this.codeAssist = new CodeAssistAPI({config});
         }
 
-        if (aiAssistBackend) {
-            this.aiAssist = new AIAssistAPI({config});
-        }
-
         this.operation = new OperationAPI({config});
         this.pdisk = new PDiskAPI({config});
         this.scheme = new SchemeAPI({config});

src/services/api/index.ts

@@ -1,4 +1,4 @@
-import {configureStore as configureReduxStore} from '@reduxjs/toolkit';
+import {combineReducers, configureStore as configureReduxStore} from '@reduxjs/toolkit';
 import type {Action, Dispatch, Middleware, Reducer, UnknownAction} from '@reduxjs/toolkit';
 import type {History} from 'history';
 import {createBrowserHistory} from 'history';
@@ -7,7 +7,7 @@ import {listenForHistoryChange} from 'redux-location-state';
 import {YdbEmbeddedAPI} from '../services/api';
 
 import {getUrlData} from './getUrlData';
-import rootReducer from './reducers';
+import combinedRootReducer, {rootReducer} from './reducers';
 import {api as storeApi} from './reducers/api';
 import {syncUserSettingsFromLS} from './reducers/settings/settings';
 import {UPDATE_REF} from './reducers/tooltip';
@@ -51,19 +51,18 @@ export const webVersion = window.web_version;
 export const customBackend = window.custom_backend;
 export const metaBackend = window.meta_backend;
 export const codeAssistBackend = window.code_assist_backend;
-export const aiAssistBackend = window.ai_assist_backend;
 
 const isSingleClusterMode = `${metaBackend}` === 'undefined';
 
 export interface ConfigureStoreOptions {
-    aRootReducer?: typeof rootReducer;
+    aRootReducer?: Reducer;
     singleClusterMode?: boolean;
     api?: YdbEmbeddedAPI;
     additionalReducers?: Record<string, Reducer>;
 }
 
 export function configureStore({
-    aRootReducer = rootReducer,
+    aRootReducer,
     singleClusterMode = isSingleClusterMode,
     api = new YdbEmbeddedAPI({webVersion, withCredentials: !customBackend}),
     additionalReducers = {},
@@ -74,21 +73,26 @@ export function configureStore({
     }));
     const history = createBrowserHistory({basename});
 
-    // Merge root reducer with additional reducers
-    const mergedReducer =
-        Object.keys(additionalReducers).length > 0
-            ? {
-                  ...aRootReducer,
-                  ...additionalReducers,
-              }
-            : aRootReducer;
-
-    const store = _configureStore(
-        mergedReducer as typeof rootReducer,
-        history,
-        {singleClusterMode},
-        [storeApi.middleware],
-    );
+    // Create the final reducer
+    let finalReducer: Reducer;
+
+    if (aRootReducer) {
+        // If custom root reducer is provided, use it
+        finalReducer = aRootReducer;
+    } else if (Object.keys(additionalReducers).length > 0) {
+        // If additional reducers are provided, combine them with the default ones
+        finalReducer = combineReducers({
+            ...rootReducer,
+            ...additionalReducers,
+        });
+    } else {
+        // Otherwise use the default combined reducer
+        finalReducer = combinedRootReducer;
+    }
+
+    const store = _configureStore(finalReducer, history, {singleClusterMode}, [
+        storeApi.middleware,
+    ]);
     listenForHistoryChange(store, history);
 
     window.api = api;

src/store/configureStore.ts

@@ -5,7 +5,6 @@ export {
     configureStore,
     customBackend,
     metaBackend,
-    aiAssistBackend,
     codeAssistBackend,
     webVersion,
 } from './configureStore';

src/store/index.ts

@@ -2,7 +2,6 @@ import {combineReducers} from '@reduxjs/toolkit';
 
 import {api} from './api';
 import authentication from './authentication/authentication';
-import {chatReducer} from './chat/chatPlaceholder';
 import cluster from './cluster/cluster';
 import clusters from './clusters/clusters';
 import executeTopQueries from './executeTopQueries/executeTopQueries';
@@ -39,7 +38,6 @@ export const rootReducer = {
     queryActions,
     fullscreen,
     clusters,
-    chat: chatReducer,
 };
 
 const combinedReducer = combineReducers({