Press n or j to go to the next uncovered block, b, p or k for the previous block.
| 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 | import type {
GoogleErrorResponse,
PalmEmbedResponse,
} from '@api/ai-providers/google/types';
import type { ErrorResponseBody } from '@shared/types/api/response/body';
import type { ChatCompletionResponseBody } from '@shared/types/api/routes/chat-completions-api';
import type {
CompletionRequestBody,
CompletionResponseBody,
} from '@shared/types/api/routes/completions-api';
import type { CreateEmbeddingsResponseBody } from '@shared/types/api/routes/embeddings-api';
export interface PalmMessage {
content?: string;
author?: string;
citation_metadata?: unknown;
}
export interface PalmChatCompleteResponse {
candidates: PalmMessage[];
messages: PalmMessage[];
filters: unknown[];
error?: PalmError;
}
export interface PalmTextOutput {
output: string;
safetyRatings: unknown[];
}
export interface PalmFilter {
reason: 'OTHER';
}
export interface PalmError {
code: number;
message: string;
status: string;
}
export interface PalmCompleteResponse {
candidates: PalmTextOutput[];
filters: PalmFilter[];
error?: PalmError;
}
// -----------------------Vertex Response Transforms-----------------------
export type PalmCompleteResponseTransformFunction = (
response: PalmCompleteResponse | GoogleErrorResponse,
status: number,
) => CompletionResponseBody | ErrorResponseBody;
export type PalmChatCompleteResponseTransformFunction = (
response: PalmChatCompleteResponse | GoogleErrorResponse,
status: number,
) => ChatCompletionResponseBody | ErrorResponseBody;
export type PalmEmbedResponseTransformFunction = (
response: PalmEmbedResponse | GoogleErrorResponse,
status: number,
responseHeaders: Headers,
strictOpenAiCompliance: boolean,
gatewayRequestUrl: string,
gatewayRequest: CompletionRequestBody,
) => CreateEmbeddingsResponseBody | ErrorResponseBody;
export type PalmResponseTransformFunctions =
| PalmCompleteResponseTransformFunction
| PalmChatCompleteResponseTransformFunction
| PalmEmbedResponseTransformFunction;
|