feature(cohere): add clientName param to cohere endpoint

huggingface · Aug 6, 2024 · 28351df · 28351df
1 parent 01f4f3a
commit 28351df
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/lib/server/endpoints/cohere/endpointCohere.ts b/src/lib/server/endpoints/cohere/endpointCohere.ts
@@ -13,19 +13,21 @@ export const endpointCohereParametersSchema = z.object({
 	model: z.any(),
 	type: z.literal("cohere"),
 	apiKey: z.string().default(env.COHERE_API_TOKEN),
+	clientName: z.string().optional(),
 	raw: z.boolean().default(false),
 });
 
 export async function endpointCohere(
 	input: z.input<typeof endpointCohereParametersSchema>
 ): Promise<Endpoint> {
-	const { apiKey, model, raw } = endpointCohereParametersSchema.parse(input);
+	const { apiKey, clientName, model, raw } = endpointCohereParametersSchema.parse(input);
 
 	let cohere: CohereClient;
 
 	try {
 		cohere = new (await import("cohere-ai")).CohereClient({
 			token: apiKey,
+			clientName,
 		});
 	} catch (e) {
 		throw new Error("Failed to import cohere-ai", { cause: e });