fix: move configs to api

Author: astandrik

package-lock.json

@@ -28,7 +28,7 @@
     "@hookform/resolvers": "^3.10.0",
     "@reduxjs/toolkit": "^2.5.0",
     "@tanstack/react-table": "^8.20.6",
-    "@ydb-platform/monaco-ghost": "^0.3.0",
+    "@ydb-platform/monaco-ghost": "^0.4.0",
     "axios": "^1.7.9",
     "axios-retry": "^4.5.0",
     "colord": "^2.9.3",

package.json

@@ -81,7 +81,9 @@ export function YqlEditor({
             onCompletionDecline: codeAssist.onCompletionDecline,
             onCompletionIgnore: codeAssist.onCompletionIgnore,
         },
-        config: codeAssist.config,
+        config: {
+            language: YQL_LANGUAGE_ID,
+        },
     });
 
     const editorDidMount = (editor: Monaco.editor.IStandaloneCodeEditor, monaco: typeof Monaco) => {

src/containers/Tenant/Query/QueryEditor/YqlEditor.tsx

@@ -7,11 +7,6 @@ import {codeAssistApi} from '../../../../store/reducers/codeAssist/codeAssist';
 import type {TelemetryOpenTabs} from '../../../../types/api/codeAssist';
 import {AUTOCOMPLETE_ON_ENTER, ENABLE_AUTOCOMPLETE} from '../../../../utils/constants';
 import {useSetting} from '../../../../utils/hooks';
-import {YQL_LANGUAGE_ID} from '../../../../utils/monaco/constats';
-
-const limitForTab = 10_000;
-const limitBeforeCursor = 8_000;
-const limitAfterCursor = 1_000;
 
 export type EditorOptions = Monaco.editor.IEditorOptions & Monaco.editor.IGlobalEditorOptions;
 
@@ -72,7 +67,7 @@ export function useCodeAssist() {
         async (queries: {text: string; name?: string}[]) => {
             const preparedData: TelemetryOpenTabs = queries.map((query, index) => ({
                 FileName: query.name || `query${index}.yql`,
-                Text: query.text.slice(0, limitForTab),
+                Text: query.text,
             }));
             try {
                 return sendUserQueriesData(preparedData).unwrap();
@@ -83,16 +78,7 @@ export function useCodeAssist() {
         [sendUserQueriesData],
     );
 
-    const config = {
-        language: YQL_LANGUAGE_ID,
-        textLimits: {
-            beforeCursor: limitBeforeCursor,
-            afterCursor: limitAfterCursor,
-        },
-    };
-
     return {
-        config,
         getCodeAssistSuggestions,
         onCompletionAccept,
         onCompletionDecline,

src/containers/Tenant/Query/QueryEditor/helpers.ts

@@ -16,25 +16,64 @@ const ideInfo = {
     PluginVersion: '0.2',
 };
 
+const limitForTab = 10_000;
+const limitBeforeCursor = 8_000;
+const limitAfterCursor = 1_000;
+
+function prepareCodeAssistTabs(tabs: TelemetryOpenTabs): TelemetryOpenTabs {
+    return tabs.map((tab) => {
+        const text = tab.Text;
+        if (text.length > limitForTab) {
+            return {
+                ...tab,
+                Text: text.slice(0, limitForTab),
+            };
+        }
+
+        return tab;
+    });
+}
+
 function prepareCodeAssistPrompt(promptFiles: PromptFile[]): CodeAssistSuggestionsFiles {
-    return promptFiles.map((file) => ({
-        Fragments: file.fragments.map((fragment) => ({
-            Text: fragment.text,
-            Start: {
-                Ln: fragment.start.lineNumber,
-                Col: fragment.start.column,
-            },
-            End: {
-                Ln: fragment.end.lineNumber,
-                Col: fragment.end.column,
+    return promptFiles.map((file) => {
+        const cursorLine = file.cursorPosition.lineNumber;
+        const cursorCol = file.cursorPosition.column;
+
+        return {
+            Fragments: file.fragments.map((fragment) => {
+                let text = fragment.text;
+                const isBeforeCursor =
+                    fragment.end.lineNumber < cursorLine ||
+                    (fragment.end.lineNumber === cursorLine && fragment.end.column <= cursorCol);
+                const isAfterCursor =
+                    fragment.start.lineNumber > cursorLine ||
+                    (fragment.start.lineNumber === cursorLine && fragment.start.column > cursorCol);
+
+                if (isBeforeCursor) {
+                    text = text.slice(-limitBeforeCursor);
+                } else if (isAfterCursor) {
+                    text = text.slice(0, limitAfterCursor);
+                }
+
+                return {
+                    Text: text,
+                    Start: {
+                        Ln: fragment.start.lineNumber,
+                        Col: fragment.start.column,
+                    },
+                    End: {
+                        Ln: fragment.end.lineNumber,
+                        Col: fragment.end.column,
+                    },
+                };
+            }),
+            Cursor: {
+                Ln: cursorLine,
+                Col: cursorCol,
             },
-        })),
-        Cursor: {
-            Ln: file.cursorPosition.lineNumber,
-            Col: file.cursorPosition.column,
-        },
-        Path: `${file.path}.yql`,
-    }));
+            Path: `${file.path}.yql`,
+        };
+    });
 }
 
 // In the future code assist api will be provided to ydb-ui component explicitly by consumer service.
@@ -77,7 +116,7 @@ export class CodeAssistAPI extends BaseYdbAPI {
     sendCodeAssistOpenTabs(data: TelemetryOpenTabs) {
         return this.post(
             '/code-assist-telemetry',
-            {OpenTabs: {Tabs: data, IdeInfo: ideInfo}},
+            {OpenTabs: {Tabs: prepareCodeAssistTabs(data), IdeInfo: ideInfo}},
             null,
             {
                 concurrentId: 'code-assist-telemetry',