V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
The Go Programming Language
http://golang.org/
Go Playground
Go Projects
Revel Web Framework
WendellSun
V2EX  ›  Go 编程语言

写了在 Go 里计算 GPT3 token 的包

  •  
  •   WendellSun · 4 天前 · 1379 次点击

    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 上看执行效率也不错。有类似需求的话可以直接使用。

    9 条回复    2023-03-18 09:52:42 +08:00
    taowen
        1
    taowen  
       4 天前
    https://bellard.org/ts_server/ fabrice bellard is all you need
    unco020511
        2
    unco020511  
       4 天前
    有一些免费或者按次数调用的自然语言处理接口,可以用来归纳总结,然后当 token 过长时(特别是连续对话),就可以先归纳总结,然后再发送给 gpt,这样既保证了对话的连续性,又不至于 token 爆炸
    Rehtt
        3
    Rehtt  
       4 天前 via Android
    @unco020511 怎么进行归纳总结?
    WendellSun
        4
    WendellSun  
    OP
       4 天前
    @taowen 这个不错唉,做的挺全面,谢谢推荐。
    unco020511
        5
    unco020511  
       4 天前
    @Rehtt "有一些免费或者按次数调用的自然语言处理接口"
    WendellSun
        6
    WendellSun  
    OP
       4 天前
    @unco020511 这样优化也可以。
    hamsterbase
        7
    hamsterbase  
       4 天前 via Android
    你不如把 npm 代码发给 chatgpg ,让他帮你转 golang
    WendellSun
        8
    WendellSun  
    OP
       4 天前
    @hamsterbase 单纯的函数应该可以,这种有上下文依赖的估计不好搞。
    xiaoz
        9
    xiaoz  
       3 天前 via Android
    @unco020511 #5 ,发出来看看呢
    关于   ·   帮助文档   ·   博客   ·   nftychat   ·   API   ·   FAQ   ·   我们的愿景   ·   广告投放   ·   实用小工具   ·   1019 人在线   最高记录 5556   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 49ms · UTC 20:41 · PVG 04:41 · LAX 13:41 · JFK 16:41
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.