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 的包

  •  1
     
  •   WendellSun · 2023-03-17 15:34:59 +08:00 · 3185 次点击
    这是一个创建于 623 天前的主题,其中的信息可能已经有所发展或是发生改变。

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