diff --git a/src/token_estimator.js b/src/token_estimator.js index 481913c..60ced5b 100644 --- a/src/token_estimator.js +++ b/src/token_estimator.js @@ -307,7 +307,7 @@ export async function countTokensWithUpstream(text, model) { const { getSelectedModel, getSelectedModelEndpoint } = await import("./models.js"); try { - const endpoint = getSelectedModelEndpoint(); + const endpoint = getSelectedModelEndpoint().replace('/v1', ''); console.log(`Trying tokenization at: ${endpoint}/tokenize`); const response = await fetch(`${endpoint}/tokenize`, { @@ -359,7 +359,7 @@ export async function getTokenCount(text, model) { const { getSelectedModel, getSelectedModelEndpoint } = await import("./models.js"); try { - const endpoint = getSelectedModelEndpoint(); + const endpoint = getSelectedModelEndpoint().replace('/v1', ''); console.log(`Trying tokenization at: ${endpoint}/tokenize`); const response = await fetch(`${endpoint}/tokenize`, {