refactor: consolodate codeium types with the codeium request code

Author: 1egoman

packages/shared/index.mts

@@ -7,6 +7,5 @@ export * from './src/types/cells.mjs';
 export * from './src/types/tsserver.mjs';
 export * from './src/types/websockets.mjs';
 export * from './src/types/secrets.mjs';
-export * from './src/types/ai-autocomplete.mjs';
 export * from './src/utils.mjs';
 export * from './src/ai.mjs';

packages/shared/src/types/ai-autocomplete.mts

@@ -1,6 +1,5 @@
 import protobuf from 'protobufjs';
 import Long from 'long';
-import { type CodiumCompletionResult } from '@srcbook/shared';
 
 import languageServerProto from './language-server-proto';
 
@@ -72,3 +71,50 @@ export async function runCodiumAiAutocomplete(
 
   return responseBody.toJSON() as CodiumCompletionResult;
 }
+
+type CodiumCompletionItem = {
+  completion: {
+    completionId: string;
+    text: string;
+    prefix: string;
+    stop: string;
+    score: number;
+    tokens: Array<string>;
+    decoded_tokens: Array<string>;
+    probabilities: Array<number>;
+    adjustedProbabilities: Array<number>;
+    generatedLength: string;
+  };
+  completionParts: Array<{
+    text: string;
+    offset: string;
+    prefix: string;
+    type:
+      | 'COMPLETION_PART_TYPE_UNSPECIFIED'
+      // Single-line completion parts that appear within an existing line of text.
+      | 'COMPLETION_PART_TYPE_INLINE'
+      // Possibly multi-line completion parts that appear below an existing line of text.
+      | 'COMPLETION_PART_TYPE_BLOCK'
+      // Like COMPLETION_PART_TYPE_INLINE, but overwrites the existing text.
+      | 'COMPLETION_PART_TYPE_INLINE_MASK';
+  }>;
+  range: {
+    endOffset: string;
+    endPosition: { row?: string; col?: string };
+    startPosition: { row?: string; col?: string };
+  };
+};
+
+type CodiumCompletionResult = {
+  completionItems?: Array<CodiumCompletionItem>;
+  state: {
+    state:
+      | 'CODEIUM_STATE_UNSPECIFIED'
+      | 'CODEIUM_STATE_INACTIVE'
+      | 'CODEIUM_STATE_PROCESSING'
+      | 'CODEIUM_STATE_SUCCESS'
+      | 'CODEIUM_STATE_WARNING'
+      | 'CODEIUM_STATE_ERROR';
+    status: string;
+  };
+};

packages/web/src/lib/ai-autocomplete/index.ts

@@ -4,7 +4,6 @@ import type {
   MarkdownCellType,
   CodeCellType,
   SecretWithAssociatedSessions,
-  CodiumCompletionResult,
 } from '@srcbook/shared';
 import { SessionType, ExampleSrcbookType } from '@/types';
 import SRCBOOK_CONFIG from '@/config';