|
|
@@ -120,10 +120,12 @@ func (k *Kimi) NewChat(ctx context.Context) (llm.Chat, error) {
|
|
|
if err != nil {
|
|
|
return nil, err
|
|
|
}
|
|
|
- localStorage, ok := k.userContext.LocalStorage[k.GetName()]
|
|
|
- if ok {
|
|
|
- for key, val := range localStorage {
|
|
|
- c.Page.MustEval(fmt.Sprintf("() => {window.localStorage.setItem('%s','%s')}", key, val))
|
|
|
+ if k.userContext != nil {
|
|
|
+ localStorage, ok := k.userContext.LocalStorage[k.GetName()]
|
|
|
+ if ok {
|
|
|
+ for key, val := range localStorage {
|
|
|
+ c.Page.MustEval(fmt.Sprintf("() => {window.localStorage.setItem('%s','%s')}", key, val))
|
|
|
+ }
|
|
|
}
|
|
|
}
|
|
|
wait()
|
|
|
@@ -135,16 +137,16 @@ func (k *Kimi) Close(ctx context.Context) error {
|
|
|
return nil
|
|
|
}
|
|
|
|
|
|
-func (k *Kimi) Init(ctx context.Context) (llm.AI, error) {
|
|
|
+func (k *Kimi) Init(ctx context.Context, debug bool) (llm.AI, error) {
|
|
|
path, ok := launcher.LookPath()
|
|
|
if !ok {
|
|
|
return nil, errors.New("浏览器未找到")
|
|
|
}
|
|
|
launcherHandler := launcher.New().
|
|
|
Bin(path).
|
|
|
- HeadlessNew(true).
|
|
|
+ HeadlessNew(!debug).
|
|
|
Set("disable-gpu").
|
|
|
- Devtools(false)
|
|
|
+ Devtools(debug)
|
|
|
browser := rod.New().ControlURL(launcherHandler.MustLaunch()).MustConnect()
|
|
|
return &Kimi{
|
|
|
browser: browser,
|
