fix type

gmpetrov · May 14, 2024 · 1a060d6 · 1a060d6
1 parent 3fbfd38
commit 1a060d6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/packages/lib/tokens-estimate/index.ts b/packages/lib/tokens-estimate/index.ts
@@ -90,7 +90,8 @@ export function promptTokensEstimate({
       tool_choice === 'none'
         ? 1
         : stringTokens(
-            tool_choice?.function?.name || '',
+            (tool_choice as ChatCompletionNamedToolChoice)?.function?.name ||
+              '',
             useFastApproximation
           ) + 4;
   }