Skip to content

Commit 1cb2077

Browse files
committed
fix: build errors
1 parent 7841989 commit 1cb2077

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

src/providers/google-vertex-ai/utils.ts

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,9 @@ import {
22
GoogleBatchRecord,
33
GoogleErrorResponse,
44
GoogleFinetuneRecord,
5-
GoogleResponseCandidate,
5+
GoogleResponseCandidate as VertexResponseCandidate,
66
} from './types';
7+
import { GoogleResponseCandidate } from '../google/chatComplete';
78
import { generateErrorResponse } from '../utils';
89
import {
910
BatchEndpoints,
@@ -513,7 +514,7 @@ export const fetchGoogleCustomEndpoint = async ({
513514
};
514515

515516
export const transformVertexLogprobs = (
516-
generation: GoogleResponseCandidate
517+
generation: GoogleResponseCandidate | VertexResponseCandidate
517518
) => {
518519
const logprobsContent: Logprobs[] = [];
519520
if (!generation.logprobsResult) return null;

src/providers/google/chatComplete.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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;

0 commit comments

Comments
 (0)