Enha: add (loaded) suffix if model is loaded
This commit is contained in:
24
bot.go
24
bot.go
@@ -403,6 +403,23 @@ func fetchLCPModels() ([]string, error) {
|
||||
return localModels, nil
|
||||
}
|
||||
|
||||
// fetchLCPModelsWithLoadStatus returns models with "(loaded)" indicator for loaded models
|
||||
func fetchLCPModelsWithLoadStatus() ([]string, error) {
|
||||
models, err := fetchLCPModelsWithStatus()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
result := make([]string, 0, len(models.Data))
|
||||
for _, m := range models.Data {
|
||||
modelName := m.ID
|
||||
if m.Status.Value == "loaded" {
|
||||
modelName = modelName + " (loaded)"
|
||||
}
|
||||
result = append(result, modelName)
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
|
||||
// fetchLCPModelsWithStatus returns the full LCPModels struct including status information.
|
||||
func fetchLCPModelsWithStatus() (*models.LCPModels, error) {
|
||||
resp, err := http.Get(cfg.FetchModelNameAPI)
|
||||
@@ -832,6 +849,7 @@ func chatRound(r *models.ChatRoundReq) error {
|
||||
// Variables for handling thinking blocks during streaming
|
||||
inThinkingBlock := false
|
||||
thinkingBuffer := strings.Builder{}
|
||||
justExitedThinkingCollapsed := false
|
||||
out:
|
||||
for {
|
||||
select {
|
||||
@@ -859,6 +877,7 @@ out:
|
||||
if thinkingCollapsed {
|
||||
// Thinking already displayed as placeholder, just update respText
|
||||
respText.WriteString(chunk)
|
||||
justExitedThinkingCollapsed = true
|
||||
if scrollToEndEnabled {
|
||||
textView.ScrollToEnd()
|
||||
}
|
||||
@@ -872,6 +891,11 @@ out:
|
||||
}
|
||||
// If not collapsed, fall through to normal display
|
||||
}
|
||||
// Add spacing after collapsed thinking block before real response
|
||||
if justExitedThinkingCollapsed {
|
||||
chunk = "\n\n" + chunk
|
||||
justExitedThinkingCollapsed = false
|
||||
}
|
||||
fmt.Fprint(textView, chunk)
|
||||
respText.WriteString(chunk)
|
||||
// Update the message in chatBody.Messages so it persists during Alt+T
|
||||
|
||||
Reference in New Issue
Block a user