Updated API contract

src/sdk/client.ts

@@ -31,7 +31,7 @@ import {
     RegisterContextResponse,
     RequiredSourceContext,
     SourceContext,
-    Vote,
+    Vote, LanguageCode,
 } from "./model/model";
 import {ProcessRequest as CodemakerProcessRequest} from "./proto/ai/codemaker/service/ProcessRequest";
 import {PredictRequest as CodemakerPredictRequest} from "./proto/ai/codemaker/service/PredictRequest";
@@ -71,6 +71,7 @@ import {Output__Output as CodemakerOutput} from "./proto/ai/codemaker/service/Ou
 import {Encoding as CodemakerEncoding} from "./proto/ai/codemaker/service/Encoding";
 import {Modify as CodemakerModify} from "./proto/ai/codemaker/service/Modify";
 import {Vote as CodemakerVote} from "./proto/ai/codemaker/service/Vote";
+import {LanguageCode as CodemakerLanguageCode} from "./proto/ai/codemaker/service/LanguageCode";
 import {
     AssistantCompletionRequest as CodemakerAssistantCompletionRequest
 } from "./proto/ai/codemaker/service/AssistantCompletionRequest";
@@ -324,6 +325,9 @@ export class Client {
     private createAssistantCompletionRequest(request: AssistantCompletionRequest): CodemakerAssistantCompletionRequest {
         return {
             message: request.message,
+            options: {
+                language: this.mapLanguage(request.options?.language)
+            }
         };
     }
 
@@ -476,6 +480,14 @@ export class Client {
         return vote === Vote.upVote ? "UP_VOTE" : "DOWN_VOTE";
     }
 
+    private mapLanguage(language: LanguageCode | undefined): CodemakerLanguageCode {
+        if (!language) {
+            return "UNSPECIFIED";
+        }
+
+        return language;
+    }
+
     private mapCodeSnippetContexts(codeSnippetContexts: CodeSnippetContext[] | undefined): CodemakerCodeSnippetContext[] | undefined {
         if (!codeSnippetContexts) {
             return undefined;

src/sdk/model/model.ts

@@ -54,6 +54,7 @@ export type RegisterContextResponse = {};
 
 export type AssistantCompletionRequest = {
     message: string;
+    options?: Options;
 };
 
 export type AssistantCompletionResponse = {
@@ -134,6 +135,7 @@ export type Options = {
     readonly codeSnippetContexts?: CodeSnippetContext[];
     readonly contextId?: string;
     readonly model?: string;
+    readonly language?: LanguageCode;
 };
 
 export enum Mode {
@@ -170,4 +172,18 @@ export enum Modify {
 export enum Vote {
     upVote = "UP_VOTE",
     downVote = "DOWN_VOTE"
+}
+
+export enum LanguageCode {
+    en = "EN",
+    es = "ES",
+    pt = "PT",
+    jp = "JP",
+    vi = "VI",
+    tr = "TR",
+    ko = "KO",
+    de = "DE",
+    fr = "FR",
+    pl = "PL",
+    zn = "ZN"
 }

src/sdk/proto/ai/codemaker/service/AssistantCompletionOptions.ts

@@ -0,0 +1,11 @@
+// Original file: src/sdk/proto/codemakerai.proto
+
+import type { LanguageCode as _ai_codemaker_service_LanguageCode, LanguageCode__Output as _ai_codemaker_service_LanguageCode__Output } from '../../../ai/codemaker/service/LanguageCode';
+
+export interface AssistantCompletionOptions {
+  'language'?: (_ai_codemaker_service_LanguageCode);
+}
+
+export interface AssistantCompletionOptions__Output {
+  'language': (_ai_codemaker_service_LanguageCode__Output);
+}

src/sdk/proto/ai/codemaker/service/AssistantCompletionRequest.ts

@@ -1,10 +1,13 @@
 // Original file: src/sdk/proto/codemakerai.proto
 
+import type { AssistantCompletionOptions as _ai_codemaker_service_AssistantCompletionOptions, AssistantCompletionOptions__Output as _ai_codemaker_service_AssistantCompletionOptions__Output } from '../../../ai/codemaker/service/AssistantCompletionOptions';
 
 export interface AssistantCompletionRequest {
   'message'?: (string);
+  'options'?: (_ai_codemaker_service_AssistantCompletionOptions | null);
 }
 
 export interface AssistantCompletionRequest__Output {
   'message': (string);
+  'options': (_ai_codemaker_service_AssistantCompletionOptions__Output | null);
 }

src/sdk/proto/ai/codemaker/service/LanguageCode.ts

@@ -0,0 +1,44 @@
+// Original file: src/sdk/proto/codemakerai.proto
+
+export const LanguageCode = {
+  UNSPECIFIED: 'UNSPECIFIED',
+  EN: 'EN',
+  ES: 'ES',
+  PT: 'PT',
+  JP: 'JP',
+  VI: 'VI',
+  TR: 'TR',
+  KO: 'KO',
+  DE: 'DE',
+  FR: 'FR',
+  PL: 'PL',
+  ZN: 'ZN',
+} as const;
+
+export type LanguageCode =
+  | 'UNSPECIFIED'
+  | 0
+  | 'EN'
+  | 1
+  | 'ES'
+  | 2
+  | 'PT'
+  | 3
+  | 'JP'
+  | 4
+  | 'VI'
+  | 5
+  | 'TR'
+  | 6
+  | 'KO'
+  | 7
+  | 'DE'
+  | 8
+  | 'FR'
+  | 9
+  | 'PL'
+  | 10
+  | 'ZN'
+  | 11
+
+export type LanguageCode__Output = typeof LanguageCode[keyof typeof LanguageCode]

src/sdk/proto/codemakerai.proto

@@ -76,6 +76,21 @@ enum Vote {
   DOWN_VOTE = 1;
 }
 
+enum LanguageCode {
+  UNSPECIFIED = 0;
+  EN = 1;
+  ES = 2;
+  PT = 3;
+  JP = 4;
+  VI = 5;
+  TR = 6;
+  KO = 7;
+  DE = 8;
+  FR = 9;
+  PL = 10;
+  ZN = 11;
+}
+
 message Source {
   bytes content = 1;
   Encoding encoding = 2;
@@ -137,6 +152,10 @@ message CompletionOptions {
   string model = 5;
 }
 
+message AssistantCompletionOptions {
+  LanguageCode language = 1;
+}
+
 message AssistantCodeCompletionOptions {
   string contextId = 1;
   string model = 2;
@@ -149,6 +168,7 @@ message PredictionOptions {
 
 message AssistantCompletionRequest {
   string message = 1;
+  AssistantCompletionOptions options = 2;
 }
 
 message AssistantCompletionResponse {

src/sdk/proto/codemakerai.ts

@@ -14,6 +14,7 @@ export interface ProtoGrpcType {
         AssistantCodeCompletionOptions: MessageTypeDefinition
         AssistantCodeCompletionRequest: MessageTypeDefinition
         AssistantCodeCompletionResponse: MessageTypeDefinition
+        AssistantCompletionOptions: MessageTypeDefinition
         AssistantCompletionRequest: MessageTypeDefinition
         AssistantCompletionResponse: MessageTypeDefinition
         AssistantSpeechRequest: MessageTypeDefinition
@@ -31,6 +32,7 @@ export interface ProtoGrpcType {
         IndentationCharacter: EnumTypeDefinition
         Input: MessageTypeDefinition
         Language: EnumTypeDefinition
+        LanguageCode: EnumTypeDefinition
         ListModelsRequest: MessageTypeDefinition
         ListModelsResponse: MessageTypeDefinition
         Metadata: MessageTypeDefinition