diff options
| author | Tommaso Sciortino <[email protected]> | 2025-06-12 18:00:17 -0700 |
|---|---|---|
| committer | GitHub <[email protected]> | 2025-06-12 18:00:17 -0700 |
| commit | 431ee839a0ad3d1f97e0f437bacbab5108481b90 (patch) | |
| tree | d4bab9dbc2bffc8d843ff7656340e93cc63a0d62 /packages/core/src/code_assist/converter.ts | |
| parent | 9a11567f73b166eb9435f7f98877e7aba3ac4d06 (diff) | |
Code Assist cleanup and docs (#993)
Diffstat (limited to 'packages/core/src/code_assist/converter.ts')
| -rw-r--r-- | packages/core/src/code_assist/converter.ts | 32 |
1 files changed, 17 insertions, 15 deletions
diff --git a/packages/core/src/code_assist/converter.ts b/packages/core/src/code_assist/converter.ts index c7b0e7c7..495cbfae 100644 --- a/packages/core/src/code_assist/converter.ts +++ b/packages/core/src/code_assist/converter.ts @@ -27,13 +27,13 @@ import { ToolConfig, } from '@google/genai'; -export interface CcpaRequest { +export interface CodeAssistRequest { model: string; project?: string; - request: CcpaGenerateContentRequest; + request: CodeAssistGenerateContentRequest; } -interface CcpaGenerateContentRequest { +interface CodeAssistGenerateContentRequest { contents: Content[]; systemInstruction?: Content; cachedContent?: string; @@ -41,10 +41,10 @@ interface CcpaGenerateContentRequest { toolConfig?: ToolConfig; labels?: Record<string, string>; safetySettings?: SafetySetting[]; - generationConfig?: CcpaGenerationConfig; + generationConfig?: CodeAssistGenerationConfig; } -interface CcpaGenerationConfig { +interface CodeAssistGenerationConfig { temperature?: number; topP?: number; topK?: number; @@ -67,7 +67,7 @@ interface CcpaGenerationConfig { thinkingConfig?: ThinkingConfig; } -export interface CcpaResponse { +export interface CodeAssistResponse { response: VertexResponse; } @@ -78,18 +78,20 @@ interface VertexResponse { usageMetadata?: GenerateContentResponseUsageMetadata; } -export function toCcpaRequest( +export function toCodeAssistRequest( req: GenerateContentParameters, project?: string, -): CcpaRequest { +): CodeAssistRequest { return { model: req.model, project, - request: toCcpaGenerateContentRequest(req), + request: toCodeAssistGenerateContentRequest(req), }; } -export function fromCcpaResponse(res: CcpaResponse): GenerateContentResponse { +export function fromCodeAsistResponse( + res: CodeAssistResponse, +): GenerateContentResponse { const inres = res.response; const out = new GenerateContentResponse(); out.candidates = inres.candidates; @@ -99,9 +101,9 @@ export function fromCcpaResponse(res: CcpaResponse): GenerateContentResponse { return out; } -function toCcpaGenerateContentRequest( +function toCodeAssistGenerateContentRequest( req: GenerateContentParameters, -): CcpaGenerateContentRequest { +): CodeAssistGenerateContentRequest { return { contents: toContents(req.contents), systemInstruction: maybeToContent(req.config?.systemInstruction), @@ -110,7 +112,7 @@ function toCcpaGenerateContentRequest( toolConfig: req.config?.toolConfig, labels: req.config?.labels, safetySettings: req.config?.safetySettings, - generationConfig: toCcpaGenerationConfig(req.config), + generationConfig: toCodeAssistGenerationConfig(req.config), }; } @@ -168,9 +170,9 @@ function toPart(part: PartUnion): Part { return part; } -function toCcpaGenerationConfig( +function toCodeAssistGenerationConfig( config?: GenerateContentConfig, -): CcpaGenerationConfig | undefined { +): CodeAssistGenerationConfig | undefined { if (!config) { return undefined; } |
