File tree Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Original file line number Diff line number Diff line change 22 GoogleBatchRecord ,
33 GoogleErrorResponse ,
44 GoogleFinetuneRecord ,
5- GoogleResponseCandidate ,
5+ GoogleResponseCandidate as VertexResponseCandidate ,
66} from './types' ;
7+ import { GoogleResponseCandidate } from '../google/chatComplete' ;
78import { generateErrorResponse } from '../utils' ;
89import {
910 BatchEndpoints ,
@@ -513,7 +514,7 @@ export const fetchGoogleCustomEndpoint = async ({
513514} ;
514515
515516export const transformVertexLogprobs = (
516- generation : GoogleResponseCandidate
517+ generation : GoogleResponseCandidate | VertexResponseCandidate
517518) => {
518519 const logprobsContent : Logprobs [ ] = [ ] ;
519520 if ( ! generation . logprobsResult ) return null ;
Original file line number Diff line number Diff line change @@ -450,7 +450,7 @@ interface GoogleGenerateFunctionCall {
450450 args : Record < string , any > ;
451451}
452452
453- interface GoogleResponseCandidate {
453+ export interface GoogleResponseCandidate {
454454 content : {
455455 parts : {
456456 text ?: string ;
You can’t perform that action at this time.
0 commit comments