Skip to content

Commit 70de78b

Browse files
committed
feat(chat): add new api AskCustom
1 parent 867cbba commit 70de78b

File tree

3 files changed

+7
-3
lines changed

3 files changed

+7
-3
lines changed

chat/init.go

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,10 @@ func Ask(p model.Protocol, grp int64, sysp string) deepinfra.Model {
4242
return lst.Modelize(p, grp, sysp)
4343
}
4444

45+
func AskCustom[T any](grp int64, f func(int, string) T) []T {
46+
return chat.Modelize(&lst, grp, f)
47+
}
48+
4549
func Sanitize(msg string) string {
4650
_, s, ok := strings.Cut(msg, "】")
4751
if ok {

go.mod

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ require (
1212
github.com/RomiChan/syncx v0.0.0-20240418144900-b7402ffdebc7
1313
github.com/RomiChan/websocket v1.4.3-0.20220227141055-9b2c6168c9c5
1414
github.com/fumiama/cron v1.3.0
15-
github.com/fumiama/deepinfra v0.0.0-20250222055014-e969fc5b4ccf
15+
github.com/fumiama/deepinfra v0.0.0-20250329122448-bf0f65bcaa63
1616
github.com/fumiama/go-base16384 v1.7.0
1717
github.com/gin-gonic/gin v1.8.2
1818
github.com/go-playground/assert/v2 v2.2.0

go.sum

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,8 +39,8 @@ github.com/ericpauley/go-quantize v0.0.0-20200331213906-ae555eb2afa4 h1:BBade+Jl
3939
github.com/ericpauley/go-quantize v0.0.0-20200331213906-ae555eb2afa4/go.mod h1:H7chHJglrhPPzetLdzBleF8d22WYOv7UM/lEKYiwlKM=
4040
github.com/fumiama/cron v1.3.0 h1:ZWlwuexF+HQHl3cYytEE5HNwD99q+3vNZF1GrEiXCFo=
4141
github.com/fumiama/cron v1.3.0/go.mod h1:bz5Izvgi/xEUI8tlBN8BI2jr9Moo8N4or0KV8xXuPDY=
42-
github.com/fumiama/deepinfra v0.0.0-20250222055014-e969fc5b4ccf h1:xi3K9hukyF34JTLZTNCwM42gxcWKGJXSVou/U0pTYKg=
43-
github.com/fumiama/deepinfra v0.0.0-20250222055014-e969fc5b4ccf/go.mod h1:wW05PQSn8mo1mZIoa6LBUE+3xIBjkoONvnfPTV5ZOhY=
42+
github.com/fumiama/deepinfra v0.0.0-20250329122448-bf0f65bcaa63 h1:Lc3CwBjr+Sg71rmsHA7Q3mksPPibWk7tSlcXpgC0V0k=
43+
github.com/fumiama/deepinfra v0.0.0-20250329122448-bf0f65bcaa63/go.mod h1:wW05PQSn8mo1mZIoa6LBUE+3xIBjkoONvnfPTV5ZOhY=
4444
github.com/fumiama/go-base16384 v1.7.0 h1:6fep7XPQWxRlh4Hu+KsdH+6+YdUp+w6CwRXtMWSsXCA=
4545
github.com/fumiama/go-base16384 v1.7.0/go.mod h1:OEn+947GV5gsbTAnyuUW/SrfxJYUdYupSIQXOuGOcXM=
4646
github.com/fumiama/go-registry v0.2.7 h1:tLEqgEpsiybQMqBv0dLHm5leia/z1DhajMupwnOHeNs=

0 commit comments

Comments
 (0)