@drone1/alt/src/providers/anthropic.mjs

An AI-powered localization tool

export function name() {
	return 'Anthropic'
}

export async function listModels(apiKey) {
	let allModels = []
	let hasMore = true
	let lastId = null

	while (hasMore) {
		const url = new URL('https://api.anthropic.com/v1/models')
		if (lastId) {
			url.searchParams.append('after', lastId)
		}

		const response = await fetch(url, {
			method: 'GET',
			headers: {
				'x-api-key': apiKey,
				'anthropic-version': '2023-06-01'
			}
		})

		const result = await response.json()

		if (result.data && result.data.length > 0) {
			allModels = [
				...allModels,
				...result.data
			]
			lastId = result.last_id
			hasMore = result.has_more
		} else {
			hasMore = false
		}
	}

	return allModels
}

export function getTranslationRequestDetails({ model, messages, apiKey, log }) {
	return {
		url: 'https://api.anthropic.com/v1/messages',
		params: {
			model,
			max_tokens: 1024,
			messages: messages.map(m => ({ role: 'user', content: m })),
		},
		config: {
			headers: {
				'Content-Type': 'application/json',
				'x-api-key': apiKey,
				'anthropic-version': '2023-06-01',
			},
		},
	}
}

export function getResult(response, log) {
	return response.data.content[0].text.trim()
}

function getHeader(headers, name) {
	return headers[name] || headers.get?.(name)
}

export function getSleepInterval(headers, log) {
	log.T(headers)
	if (getHeader(headers, 'x-should-retry') !== 'true')
		return 0

	const retryAfter = parseInt(getHeader(headers, 'retry-after'))
	log.D('retryAfter', retryAfter)
	return 1000 * retryAfter + 200
}