写了在 Go 里计算 GPT3 token 的包

2023-03-17 15:34:59 +08:00
 WendellSun

https://github.com/pandodao/tokenizer-go

最近在用 Go 做基于 OpenAI API 的应用,需要在请求前对 token 做检查,当请求太大时可以自动的缩减 (Chat completions) 或者分批次请求 (Embeddings),github 没找到可以在 Go 里用的包,所以打算自己写一个。

从头实现的难度太大,取巧了一下,embed 了现成的 npm 包,然后通过 JavaScript 函数调用来获取结果,这样在 Go 里就跟原生包一样直接用了。得益于使用的 goja 库是纯 Go 实现的 JavaScript engine ,所以也不需要 CGO 那些,从 Benchmark 上看执行效率也不错。有类似需求的话可以直接使用。

3126 次点击
所在节点    Go 编程语言
9 条回复
taowen
2023-03-17 16:17:36 +08:00
https://bellard.org/ts_server/ fabrice bellard is all you need
unco020511
2023-03-17 16:25:58 +08:00
有一些免费或者按次数调用的自然语言处理接口,可以用来归纳总结,然后当 token 过长时(特别是连续对话),就可以先归纳总结,然后再发送给 gpt,这样既保证了对话的连续性,又不至于 token 爆炸
Rehtt
2023-03-17 16:41:11 +08:00
@unco020511 怎么进行归纳总结?
WendellSun
2023-03-17 16:49:33 +08:00
@taowen 这个不错唉,做的挺全面,谢谢推荐。
unco020511
2023-03-17 17:44:45 +08:00
@Rehtt "有一些免费或者按次数调用的自然语言处理接口"
WendellSun
2023-03-17 17:49:36 +08:00
@unco020511 这样优化也可以。
hamsterbase
2023-03-17 19:22:43 +08:00
你不如把 npm 代码发给 chatgpg ,让他帮你转 golang
WendellSun
2023-03-17 20:29:02 +08:00
@hamsterbase 单纯的函数应该可以,这种有上下文依赖的估计不好搞。
xiaoz
2023-03-18 09:52:42 +08:00
@unco020511 #5 ,发出来看看呢

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/924883

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX