983dbd3cee
- Updated the BuildContent method to utilize content parsing with improved error handling and context injection, enhancing the processing of user input. - Refactored the loadMap function to support multiple search configuration types, improving flexibility in handling search settings. - Enhanced the shouldAutoSearch method to include a check for search disabling via context metadata, allowing for more granular control over search behavior. - Updated the getMergedSearchUses method to prioritize options.Uses, ensuring that search configurations can be dynamically adjusted based on provided options. - Removed deprecated audio and excel handling code, streamlining the content processing package and improving maintainability.
99 lines
2 KiB
Go
99 lines
2 KiB
Go
package assistant
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"strconv"
|
|
"time"
|
|
|
|
jsoniter "github.com/json-iterator/go"
|
|
"github.com/kaptinlin/jsonrepair"
|
|
)
|
|
|
|
func getTimestamp(v interface{}) (int64, error) {
|
|
switch v := v.(type) {
|
|
case int64:
|
|
return v, nil
|
|
case int:
|
|
return int64(v), nil
|
|
|
|
case string:
|
|
if ts, err := time.Parse(time.RFC3339, v); err == nil {
|
|
return ts.UnixNano(), nil
|
|
}
|
|
|
|
// MySQL format
|
|
if ts, err := time.Parse("2006-01-02 15:04:05", v); err == nil {
|
|
return ts.UnixNano(), nil
|
|
}
|
|
|
|
// UnixNano format
|
|
if ts, err := strconv.ParseInt(v, 10, 64); err == nil {
|
|
return ts, nil
|
|
}
|
|
|
|
case time.Time:
|
|
return v.UnixNano(), nil
|
|
|
|
case nil:
|
|
return 0, nil
|
|
}
|
|
|
|
return 0, fmt.Errorf("invalid timestamp type %T", v)
|
|
}
|
|
|
|
// getBool gets bool from data map[string]interface{}, key string
|
|
func getBool(data map[string]interface{}, key string) bool {
|
|
switch v := data[key].(type) {
|
|
case bool:
|
|
return v
|
|
case int64:
|
|
return v != 0
|
|
case int:
|
|
return v != 0
|
|
case float64:
|
|
return v != 0
|
|
case string:
|
|
return v == "true" || v == "1" || v == "enabled" || v == "yes" || v == "on"
|
|
case nil:
|
|
return false
|
|
}
|
|
return false
|
|
}
|
|
|
|
// stringHash returns the sha256 hash of the string
|
|
func stringHash(v string) string {
|
|
h := sha256.New()
|
|
h.Write([]byte(v))
|
|
return hex.EncodeToString(h.Sum(nil))
|
|
}
|
|
|
|
// ParseJSON attempts to parse a potentially malformed JSON string
|
|
func ParseJSON(jsonStr string, v interface{}) error {
|
|
// Try parsing as-is first
|
|
err := jsoniter.UnmarshalFromString(jsonStr, v)
|
|
if err == nil {
|
|
return nil
|
|
}
|
|
originalErr := err
|
|
|
|
// Try adding a closing brace
|
|
if err := jsoniter.UnmarshalFromString(jsonStr+"}", v); err == nil {
|
|
return nil
|
|
}
|
|
|
|
// Try repairing the JSON
|
|
repaired, err := jsonrepair.JSONRepair(jsonStr)
|
|
if err != nil {
|
|
return originalErr
|
|
}
|
|
|
|
// Try parsing the repaired JSON
|
|
if err := jsoniter.UnmarshalFromString(repaired, v); err == nil {
|
|
return nil
|
|
}
|
|
|
|
// If all attempts fail, return the original error
|
|
return originalErr
|
|
}
|