diff --git a/go.mod b/go.mod
index 55c5281..cf61d7e 100644
--- a/go.mod
+++ b/go.mod
@@ -8,7 +8,7 @@ require (
 	github.com/gorilla/websocket v1.5.1
 	github.com/mitchellh/mapstructure v1.5.0
 	github.com/opentdp/go-helper v0.6.2
-	github.com/rehiy/one-llm v0.0.0-20240312034842-6b111aabeed3
+	github.com/rehiy/one-llm v0.0.0-20240312044550-e53258310885
 	github.com/sashabaranov/go-openai v1.19.4
 	go.nanomsg.org/mangos/v3 v3.4.2
 	google.golang.org/protobuf v1.32.0
diff --git a/go.sum b/go.sum
index d41924f..76e879e 100644
--- a/go.sum
+++ b/go.sum
@@ -98,8 +98,8 @@ github.com/pelletier/go-toml/v2 v2.1.1 h1:LWAJwfNvjQZCFIDKWYQaM62NcYeYViCmWIwmOS
 github.com/pelletier/go-toml/v2 v2.1.1/go.mod h1:tJU2Z3ZkXwnxa4DPO899bsyIoywizdUvyaeZurnPPDc=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
-github.com/rehiy/one-llm v0.0.0-20240312034842-6b111aabeed3 h1:0z4Hh56K73JWaOdlB5N6vdh+eLbWx/yku0/3bP/7Us0=
-github.com/rehiy/one-llm v0.0.0-20240312034842-6b111aabeed3/go.mod h1:0sQpMABZfqgZFsEFvW7jCkX6CkD+zTiS8TLhDxdatqs=
+github.com/rehiy/one-llm v0.0.0-20240312044550-e53258310885 h1:2wt5WHnLORtt8+7dTItScLb5opyd1YoLI96aIE4GfzY=
+github.com/rehiy/one-llm v0.0.0-20240312044550-e53258310885/go.mod h1:0sQpMABZfqgZFsEFvW7jCkX6CkD+zTiS8TLhDxdatqs=
 github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec h1:W09IVJc94icq4NjY3clb7Lk8O1qJ8BdBEF8z0ibU0rE=
 github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo=
 github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=