Skip to content

Commit 9d8805e

Browse files
authored
fix(openai): switched to pure js tiktoken (#248)
1 parent 8ee046e commit 9d8805e

File tree

3 files changed

+7
-12
lines changed

3 files changed

+7
-12
lines changed

package-lock.json

+4-9
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

packages/instrumentation-openai/package.json

+1-1
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@
4040
"@opentelemetry/instrumentation": "^0.49.0",
4141
"@opentelemetry/semantic-conventions": "^1.22.0",
4242
"@traceloop/ai-semantic-conventions": "^0.8.0",
43-
"tiktoken": "^1.0.13"
43+
"js-tiktoken": "^1.0.11"
4444
},
4545
"devDependencies": {
4646
"@pollyjs/adapter-node-http": "^6.0.6",

packages/instrumentation-openai/src/instrumentation.ts

+2-2
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ import type {
4545
} from "openai/resources";
4646
import type { Stream } from "openai/streaming";
4747
import { version } from "../package.json";
48-
import { encoding_for_model, TiktokenModel, Tiktoken } from "tiktoken";
48+
import { encodingForModel, TiktokenModel, Tiktoken } from "js-tiktoken";
4949

5050
export class OpenAIInstrumentation extends InstrumentationBase<any> {
5151
protected declare _config: OpenAIInstrumentationConfig;
@@ -726,7 +726,7 @@ export class OpenAIInstrumentation extends InstrumentationBase<any> {
726726

727727
if (!encoding) {
728728
try {
729-
encoding = encoding_for_model(model as TiktokenModel);
729+
encoding = encodingForModel(model as TiktokenModel);
730730
this._encodingCache.set(model, encoding);
731731
} catch (e) {
732732
this._diag.warn(e);

0 commit comments

Comments
 (0)