mirror of
https://github.com/NoFxAiOS/nofx.git
synced 2026-07-16 17:12:25 +08:00
change v1
This commit is contained in:
lky-spec
@@ -11,6 +11,7 @@ import (
|
||||
)
|
||||
|
||||
var titleCaser = cases.Title(language.English)
|
||||
|
||||
const setupExchangeAccountName = "Default"
|
||||
|
||||
// Onboard handles first-time setup through natural language.
|
||||
@@ -41,6 +42,11 @@ func (a *Agent) needsSetup() bool {
|
||||
|
||||
// getSetupState loads the current setup state from user preferences.
|
||||
func (a *Agent) getSetupState(userID int64) *SetupState {
|
||||
if cached, ok := a.setupStates.Load(userID); ok {
|
||||
if state, ok := cached.(*SetupState); ok && state != nil {
|
||||
return cloneSetupState(state)
|
||||
}
|
||||
}
|
||||
step, _ := a.store.GetSystemConfig(fmt.Sprintf("setup_step_%d", userID))
|
||||
if step == "" {
|
||||
return &SetupState{}
|
||||
@@ -49,35 +55,30 @@ func (a *Agent) getSetupState(userID int64) *SetupState {
|
||||
Step: step,
|
||||
Exchange: getConfig(a.store, userID, "exchange"),
|
||||
ExchangeID: getConfig(a.store, userID, "exchange_id"),
|
||||
APIKey: getConfig(a.store, userID, "api_key"),
|
||||
APISecret: getConfig(a.store, userID, "api_secret"),
|
||||
Passphrase: getConfig(a.store, userID, "passphrase"),
|
||||
AIProvider: getConfig(a.store, userID, "ai_provider"),
|
||||
AIModel: getConfig(a.store, userID, "ai_model"),
|
||||
AIModelID: getConfig(a.store, userID, "ai_model_id"),
|
||||
AIKey: getConfig(a.store, userID, "ai_key"),
|
||||
AIBaseURL: getConfig(a.store, userID, "ai_base_url"),
|
||||
}
|
||||
}
|
||||
|
||||
func (a *Agent) saveSetupState(userID int64, s *SetupState) {
|
||||
a.setupStates.Store(userID, cloneSetupState(s))
|
||||
a.store.SetSystemConfig(fmt.Sprintf("setup_step_%d", userID), s.Step)
|
||||
setConfig(a.store, userID, "exchange", s.Exchange)
|
||||
setConfig(a.store, userID, "exchange_id", s.ExchangeID)
|
||||
setConfig(a.store, userID, "api_key", s.APIKey)
|
||||
setConfig(a.store, userID, "api_secret", s.APISecret)
|
||||
setConfig(a.store, userID, "passphrase", s.Passphrase)
|
||||
setConfig(a.store, userID, "ai_provider", s.AIProvider)
|
||||
setConfig(a.store, userID, "ai_model", s.AIModel)
|
||||
setConfig(a.store, userID, "ai_model_id", s.AIModelID)
|
||||
setConfig(a.store, userID, "ai_key", s.AIKey)
|
||||
setConfig(a.store, userID, "ai_base_url", s.AIBaseURL)
|
||||
}
|
||||
|
||||
func (a *Agent) clearSetupState(userID int64) {
|
||||
for _, k := range []string{"step", "exchange", "exchange_id", "api_key", "api_secret", "passphrase", "ai_provider", "ai_model", "ai_model_id", "ai_key", "ai_base_url"} {
|
||||
a.setupStates.Delete(userID)
|
||||
for _, k := range []string{"step", "exchange", "exchange_id", "ai_provider", "ai_model", "ai_model_id", "ai_base_url"} {
|
||||
a.store.SetSystemConfig(fmt.Sprintf("setup_%s_%d", k, userID), "")
|
||||
}
|
||||
a.store.SetSystemConfig(fmt.Sprintf("setup_step_%d", userID), "")
|
||||
}
|
||||
|
||||
func getConfig(st *store.Store, uid int64, key string) string {
|
||||
@@ -89,6 +90,14 @@ func setConfig(st *store.Store, uid int64, key, val string) {
|
||||
st.SetSystemConfig(fmt.Sprintf("setup_%s_%d", key, uid), val)
|
||||
}
|
||||
|
||||
func cloneSetupState(s *SetupState) *SetupState {
|
||||
if s == nil {
|
||||
return &SetupState{}
|
||||
}
|
||||
copy := *s
|
||||
return ©
|
||||
}
|
||||
|
||||
// handleSetupFlow processes the setup conversation.
|
||||
// Returns (response, handled). If handled=false, continue to normal routing.
|
||||
func (a *Agent) handleSetupFlow(userID int64, text string, L string) (string, bool) {
|
||||
@@ -152,7 +161,7 @@ func (a *Agent) handleSetupFlowForStoreUser(storeUserID string, userID int64, te
|
||||
if L == "zh" {
|
||||
return fmt.Sprintf("⚠️ 交易所配置保存失败: %v\n请再试一次，或稍后去 Web UI 继续。", err), true
|
||||
}
|
||||
return fmt.Sprintf("⚠️ Failed to save exchange config: %v\nPlease try again, or continue later in the Web UI.", err), true
|
||||
return fmt.Sprintf("⚠️ I could not save the exchange settings just now: %v\nPlease try again, or continue later on the web page.", err), true
|
||||
}
|
||||
state.ExchangeID = exchangeID
|
||||
state.Step = "await_ai_model"
|
||||
@@ -169,7 +178,7 @@ func (a *Agent) handleSetupFlowForStoreUser(storeUserID string, userID int64, te
|
||||
if L == "zh" {
|
||||
return fmt.Sprintf("⚠️ 交易所配置保存失败: %v\n请再试一次，或稍后去 Web UI 继续。", err), true
|
||||
}
|
||||
return fmt.Sprintf("⚠️ Failed to save exchange config: %v\nPlease try again, or continue later in the Web UI.", err), true
|
||||
return fmt.Sprintf("⚠️ I could not save the exchange settings just now: %v\nPlease try again, or continue later on the web page.", err), true
|
||||
}
|
||||
state.ExchangeID = exchangeID
|
||||
state.Step = "await_ai_model"
|
||||
@@ -188,7 +197,7 @@ func (a *Agent) handleSetupFlowForStoreUser(storeUserID string, userID int64, te
|
||||
if L == "zh" {
|
||||
return fmt.Sprintf("⚠️ AI 模型配置保存失败: %v\n请再试一次，或稍后去 Web UI 继续。", err), true
|
||||
}
|
||||
return fmt.Sprintf("⚠️ Failed to save AI model config: %v\nPlease try again, or continue later in the Web UI.", err), true
|
||||
return fmt.Sprintf("⚠️ I could not save the AI model settings just now: %v\nPlease try again, or continue later on the web page.", err), true
|
||||
}
|
||||
state.AIModelID = aiModelID
|
||||
return a.finishSetup(storeUserID, userID, state, L)
|
||||
@@ -252,19 +261,19 @@ func (a *Agent) handleAIChoice(storeUserID string, userID int64, text string, st
|
||||
lower := strings.ToLower(strings.TrimSpace(text))
|
||||
|
||||
models := map[string]struct{ provider, model, url string }{
|
||||
"deepseek": {"deepseek", "deepseek-chat", "https://api.deepseek.com/v1"},
|
||||
"1": {"deepseek", "deepseek-chat", "https://api.deepseek.com/v1"},
|
||||
"qwen": {"qwen", "qwen-plus", "https://dashscope.aliyuncs.com/compatible-mode/v1"},
|
||||
"deepseek": {"deepseek", "deepseek-chat", "https://api.deepseek.com/v1"},
|
||||
"1": {"deepseek", "deepseek-chat", "https://api.deepseek.com/v1"},
|
||||
"qwen": {"qwen", "qwen-plus", "https://dashscope.aliyuncs.com/compatible-mode/v1"},
|
||||
"通义": {"qwen", "qwen-plus", "https://dashscope.aliyuncs.com/compatible-mode/v1"},
|
||||
"2": {"qwen", "qwen-plus", "https://dashscope.aliyuncs.com/compatible-mode/v1"},
|
||||
"openai": {"openai", "gpt-4o", "https://api.openai.com/v1"},
|
||||
"gpt": {"openai", "gpt-4o", "https://api.openai.com/v1"},
|
||||
"3": {"openai", "gpt-4o", "https://api.openai.com/v1"},
|
||||
"claude": {"claude", "claude-3-5-sonnet-20241022", "https://api.anthropic.com/v1"},
|
||||
"4": {"claude", "claude-3-5-sonnet-20241022", "https://api.anthropic.com/v1"},
|
||||
"skip": {"", "", ""},
|
||||
"2": {"qwen", "qwen-plus", "https://dashscope.aliyuncs.com/compatible-mode/v1"},
|
||||
"openai": {"openai", "gpt-4o", "https://api.openai.com/v1"},
|
||||
"gpt": {"openai", "gpt-4o", "https://api.openai.com/v1"},
|
||||
"3": {"openai", "gpt-4o", "https://api.openai.com/v1"},
|
||||
"claude": {"claude", "claude-3-5-sonnet-20241022", "https://api.anthropic.com/v1"},
|
||||
"4": {"claude", "claude-3-5-sonnet-20241022", "https://api.anthropic.com/v1"},
|
||||
"skip": {"", "", ""},
|
||||
"跳过": {"", "", ""},
|
||||
"5": {"", "", ""},
|
||||
"5": {"", "", ""},
|
||||
}
|
||||
|
||||
choice, ok := models[lower]
|
||||
|
||||
Reference in New Issue
Block a user