xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-08 02:03:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

完成 Token 点卡功能

Browse Source
This commit is contained in:
RockYang
2023-03-28 10:17:36 +08:00
parent 5f702d92dc
commit d85e91a8da
7 changed files with 59 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
utils/config.go
View File

@@ -11,8 +11,10 @@ import (
func NewDefaultConfig() *types.Config {
return &types.Config{
Listen: "0.0.0.0:5678",
ProxyURL: make([]string, 0),
Listen: "0.0.0.0:5678",
ProxyURL: make([]string, 0),
EnableAuth: true,
AccessKey: "yangjian102621@gmail.com",
Session: types.Session{
SecretKey: RandString(64),
@@ -32,7 +34,6 @@ func NewDefaultConfig() *types.Config {
Temperature: 0.9,
EnableContext: true,
},
EnableAuth: true,
}
}

10
utils/leveldb.go
View File

@@ -28,19 +28,13 @@ func (db *LevelDB) Put(key string, value interface{}) error {
return db.driver.Put([]byte(key), bytes, nil)
}
func (db *LevelDB) Get(key string) (interface{}, error) {
func (db *LevelDB) Get(key string) ([]byte, error) {
bytes, err := db.driver.Get([]byte(key), nil)
if err != nil {
return nil, err
}
var value interface{}
err = json.Unmarshal(bytes, &value)
if err != nil {
return nil, err
}
return value, nil
return bytes, nil
}
func (db *LevelDB) Search(prefix string) []string {