Skip to content

Commit 4829443

Browse files
committed
feat(storage): implement milvus vector store
Signed-off-by: joey <[email protected]>
1 parent eb1d374 commit 4829443

File tree

19 files changed

+7408
-60
lines changed

19 files changed

+7408
-60
lines changed

examples/knowledge/go.mod

Lines changed: 99 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -1,75 +1,147 @@
11
module trpc.group/trpc-go/trpc-agent-go/examples/knowledge
22

3-
go 1.24.1
3+
go 1.24.6
44

55
replace (
66
trpc.group/trpc-go/trpc-agent-go => ../../
77
trpc.group/trpc-go/trpc-agent-go/knowledge/document/reader/pdf => ../../knowledge/document/reader/pdf
88
trpc.group/trpc-go/trpc-agent-go/knowledge/embedder/gemini => ../../knowledge/embedder/gemini
99
trpc.group/trpc-go/trpc-agent-go/knowledge/vectorstore/elasticsearch => ../../knowledge/vectorstore/elasticsearch
10+
trpc.group/trpc-go/trpc-agent-go/knowledge/vectorstore/milvus => ../../knowledge/vectorstore/milvus
1011
trpc.group/trpc-go/trpc-agent-go/knowledge/vectorstore/pgvector => ../../knowledge/vectorstore/pgvector
1112
trpc.group/trpc-go/trpc-agent-go/knowledge/vectorstore/tcvector => ../../knowledge/vectorstore/tcvector
13+
trpc.group/trpc-go/trpc-agent-go/storage/milvus => ../../storage/milvus
1214
trpc.group/trpc-go/trpc-agent-go/storage/postgres => ../../storage/postgres
1315
)
1416

1517
require (
1618
github.com/tencent/vectordatabase-sdk-go v1.8.0
17-
trpc.group/trpc-go/trpc-agent-go v0.5.0
19+
trpc.group/trpc-go/trpc-agent-go v0.6.0
1820
trpc.group/trpc-go/trpc-agent-go/knowledge/document/reader/pdf v0.0.0-20250917031858-f0ddbd5b2cb4
1921
trpc.group/trpc-go/trpc-agent-go/knowledge/embedder/gemini v0.0.0-20250917031858-f0ddbd5b2cb4
2022
trpc.group/trpc-go/trpc-agent-go/knowledge/embedder/huggingface v0.0.0-20251119113046-0cbdb93921df
2123
trpc.group/trpc-go/trpc-agent-go/knowledge/embedder/ollama v0.0.0-20251111070215-8fe58a4f2ffa
2224
trpc.group/trpc-go/trpc-agent-go/knowledge/vectorstore/elasticsearch v0.2.1
25+
trpc.group/trpc-go/trpc-agent-go/knowledge/vectorstore/milvus v0.0.0-00010101000000-000000000000
2326
trpc.group/trpc-go/trpc-agent-go/knowledge/vectorstore/pgvector v0.2.0
2427
trpc.group/trpc-go/trpc-agent-go/knowledge/vectorstore/tcvector v0.2.0
2528
)
2629

2730
require (
28-
cloud.google.com/go v0.116.0 // indirect
29-
cloud.google.com/go/auth v0.9.3 // indirect
31+
cloud.google.com/go v0.118.3 // indirect
32+
cloud.google.com/go/auth v0.15.0 // indirect
3033
cloud.google.com/go/compute/metadata v0.6.0 // indirect
34+
github.com/beorn7/perks v1.0.1 // indirect
35+
github.com/blang/semver/v4 v4.0.0 // indirect
3136
github.com/bmatcuk/doublestar/v4 v4.9.1 // indirect
3237
github.com/cenkalti/backoff/v4 v4.3.0 // indirect
38+
github.com/cespare/xxhash/v2 v2.3.0 // indirect
39+
github.com/cilium/ebpf v0.11.0 // indirect
3340
github.com/clbanning/mxj v1.8.4 // indirect
41+
github.com/cockroachdb/errors v1.9.1 // indirect
42+
github.com/cockroachdb/logtags v0.0.0-20211118104740-dabe8e521a4f // indirect
43+
github.com/cockroachdb/redact v1.1.3 // indirect
44+
github.com/containerd/cgroups/v3 v3.0.3 // indirect
45+
github.com/coreos/go-semver v0.3.0 // indirect
46+
github.com/coreos/go-systemd/v22 v22.3.2 // indirect
47+
github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect
48+
github.com/docker/go-units v0.5.0 // indirect
49+
github.com/dustin/go-humanize v1.0.1 // indirect
3450
github.com/elastic/elastic-transport-go/v8 v8.7.0 // indirect
3551
github.com/elastic/go-elasticsearch/v7 v7.17.10 // indirect
3652
github.com/elastic/go-elasticsearch/v8 v8.19.0 // indirect
3753
github.com/elastic/go-elasticsearch/v9 v9.1.0 // indirect
54+
github.com/felixge/httpsnoop v1.0.4 // indirect
55+
github.com/form3tech-oss/jwt-go v3.2.3+incompatible // indirect
56+
github.com/fxamacker/cbor/v2 v2.7.0 // indirect
57+
github.com/getsentry/sentry-go v0.12.0 // indirect
3858
github.com/go-ego/gse v0.80.3 // indirect
3959
github.com/go-logr/logr v1.4.3 // indirect
4060
github.com/go-logr/stdr v1.2.2 // indirect
61+
github.com/go-ole/go-ole v1.2.6 // indirect
62+
github.com/godbus/dbus/v5 v5.0.4 // indirect
63+
github.com/gogo/protobuf v1.3.2 // indirect
4164
github.com/golang/glog v1.2.4 // indirect
42-
github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da // indirect
65+
github.com/golang/protobuf v1.5.4 // indirect
4366
github.com/gonfva/docxlib v0.0.0-20210517191039-d8f39cecf1ad // indirect
67+
github.com/google/btree v1.1.2 // indirect
4468
github.com/google/go-cmp v0.7.0 // indirect
4569
github.com/google/go-querystring v1.1.0 // indirect
46-
github.com/google/s2a-go v0.1.8 // indirect
70+
github.com/google/s2a-go v0.1.9 // indirect
4771
github.com/google/uuid v1.6.0 // indirect
48-
github.com/googleapis/enterprise-certificate-proxy v0.3.4 // indirect
72+
github.com/googleapis/enterprise-certificate-proxy v0.3.5 // indirect
73+
github.com/googleapis/gax-go/v2 v2.14.1 // indirect
4974
github.com/gorilla/websocket v1.5.3 // indirect
75+
github.com/grpc-ecosystem/go-grpc-middleware v1.3.0 // indirect
76+
github.com/grpc-ecosystem/go-grpc-prometheus v1.2.0 // indirect
77+
github.com/grpc-ecosystem/grpc-gateway v1.16.0 // indirect
5078
github.com/grpc-ecosystem/grpc-gateway/v2 v2.22.0 // indirect
5179
github.com/jackc/pgpassfile v1.0.0 // indirect
5280
github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 // indirect
5381
github.com/jackc/pgx/v5 v5.7.2 // indirect
5482
github.com/jackc/puddle/v2 v2.2.2 // indirect
83+
github.com/jonboulle/clockwork v0.2.2 // indirect
84+
github.com/json-iterator/go v1.1.12 // indirect
85+
github.com/klauspost/compress v1.17.9 // indirect
86+
github.com/kr/pretty v0.3.1 // indirect
87+
github.com/kr/text v0.2.0 // indirect
5588
github.com/ledongthuc/pdf v0.0.0-20250511090121-5959a4027728 // indirect
89+
github.com/lufia/plan9stats v0.0.0-20211012122336-39d0f177ccd0 // indirect
90+
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.3 // indirect
91+
github.com/milvus-io/milvus/client/v2 v2.6.1 // indirect
92+
github.com/milvus-io/milvus/pkg/v2 v2.6.3 // indirect
5693
github.com/mitchellh/mapstructure v1.5.0 // indirect
94+
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
95+
github.com/modern-go/reflect2 v1.0.2 // indirect
5796
github.com/mozillazg/go-httpheader v0.4.0 // indirect
97+
github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 // indirect
5898
github.com/ollama/ollama v0.12.9 // indirect
5999
github.com/openai/openai-go v1.12.0 // indirect
100+
github.com/opencontainers/runtime-spec v1.0.2 // indirect
60101
github.com/panjf2000/ants/v2 v2.11.3 // indirect
61102
github.com/pgvector/pgvector-go v0.3.0 // indirect
62103
github.com/pkg/errors v0.9.1 // indirect
104+
github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect
105+
github.com/power-devops/perfstat v0.0.0-20210106213030-5aafc221ea8c // indirect
106+
github.com/prometheus/client_golang v1.20.5 // indirect
107+
github.com/prometheus/client_model v0.6.1 // indirect
108+
github.com/prometheus/common v0.55.0 // indirect
109+
github.com/prometheus/procfs v0.15.1 // indirect
110+
github.com/rogpeppe/go-internal v1.13.1 // indirect
111+
github.com/samber/lo v1.27.0 // indirect
112+
github.com/shirou/gopsutil/v3 v3.23.12 // indirect
113+
github.com/shoenig/go-m1cpu v0.1.6 // indirect
114+
github.com/sirupsen/logrus v1.9.3 // indirect
115+
github.com/soheilhy/cmux v0.1.5 // indirect
63116
github.com/spaolacci/murmur3 v1.1.0 // indirect
117+
github.com/spf13/cast v1.3.1 // indirect
118+
github.com/spf13/pflag v1.0.5 // indirect
119+
github.com/stretchr/testify v1.11.1 // indirect
64120
github.com/tencentyun/cos-go-sdk-v5 v0.7.69 // indirect
65-
github.com/tidwall/gjson v1.14.4 // indirect
121+
github.com/tidwall/gjson v1.17.1 // indirect
66122
github.com/tidwall/match v1.1.1 // indirect
67123
github.com/tidwall/pretty v1.2.1 // indirect
68124
github.com/tidwall/sjson v1.2.5 // indirect
125+
github.com/tklauser/go-sysconf v0.3.12 // indirect
126+
github.com/tklauser/numcpus v0.6.1 // indirect
127+
github.com/tmc/grpc-websocket-proxy v0.0.0-20201229170055-e5319fda7802 // indirect
128+
github.com/uber/jaeger-client-go v2.30.0+incompatible // indirect
69129
github.com/vcaesar/cedar v0.20.2 // indirect
130+
github.com/x448/float16 v0.8.4 // indirect
131+
github.com/xiang90/probing v0.0.0-20190116061207-43a291ad63a2 // indirect
70132
github.com/yuin/goldmark v1.4.13 // indirect
71-
go.opencensus.io v0.24.0 // indirect
133+
github.com/yusufpapurcu/wmi v1.2.3 // indirect
134+
go.etcd.io/bbolt v1.3.6 // indirect
135+
go.etcd.io/etcd/api/v3 v3.5.5 // indirect
136+
go.etcd.io/etcd/client/pkg/v3 v3.5.5 // indirect
137+
go.etcd.io/etcd/client/v2 v2.305.5 // indirect
138+
go.etcd.io/etcd/client/v3 v3.5.5 // indirect
139+
go.etcd.io/etcd/pkg/v3 v3.5.5 // indirect
140+
go.etcd.io/etcd/raft/v3 v3.5.5 // indirect
141+
go.etcd.io/etcd/server/v3 v3.5.5 // indirect
72142
go.opentelemetry.io/auto/sdk v1.1.0 // indirect
143+
go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.59.0 // indirect
144+
go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.59.0 // indirect
73145
go.opentelemetry.io/otel v1.37.0 // indirect
74146
go.opentelemetry.io/otel/exporters/otlp/otlptrace v1.29.0 // indirect
75147
go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracegrpc v1.29.0 // indirect
@@ -78,20 +150,32 @@ require (
78150
go.opentelemetry.io/otel/sdk v1.35.0 // indirect
79151
go.opentelemetry.io/otel/trace v1.37.0 // indirect
80152
go.opentelemetry.io/proto/otlp v1.3.1 // indirect
81-
go.uber.org/multierr v1.10.0 // indirect
153+
go.uber.org/atomic v1.11.0 // indirect
154+
go.uber.org/automaxprocs v1.5.3 // indirect
155+
go.uber.org/multierr v1.11.0 // indirect
82156
go.uber.org/zap v1.27.0 // indirect
83-
golang.org/x/crypto v0.39.0 // indirect
84-
golang.org/x/net v0.41.0 // indirect
85-
golang.org/x/sync v0.15.0 // indirect
86-
golang.org/x/sys v0.33.0 // indirect
87-
golang.org/x/text v0.26.0 // indirect
157+
golang.org/x/crypto v0.41.0 // indirect
158+
golang.org/x/exp v0.0.0-20250218142911-aa4b98e5adaa // indirect
159+
golang.org/x/net v0.43.0 // indirect
160+
golang.org/x/sync v0.16.0 // indirect
161+
golang.org/x/sys v0.35.0 // indirect
162+
golang.org/x/text v0.28.0 // indirect
163+
golang.org/x/time v0.10.0 // indirect
88164
google.golang.org/genai v1.0.0 // indirect
165+
google.golang.org/genproto v0.0.0-20250303144028-a0af3efb3deb // indirect
89166
google.golang.org/genproto/googleapis/api v0.0.0-20250324211829-b45e905df463 // indirect
90167
google.golang.org/genproto/googleapis/rpc v0.0.0-20250707201910-8d1bb00bc6a7 // indirect
91168
google.golang.org/grpc v1.73.0 // indirect
92169
google.golang.org/protobuf v1.36.6 // indirect
170+
gopkg.in/inf.v0 v0.9.1 // indirect
171+
gopkg.in/natefinch/lumberjack.v2 v2.2.1 // indirect
172+
gopkg.in/yaml.v2 v2.4.0 // indirect
173+
gopkg.in/yaml.v3 v3.0.1 // indirect
174+
k8s.io/apimachinery v0.32.3 // indirect
175+
sigs.k8s.io/yaml v1.4.0 // indirect
93176
trpc.group/trpc-go/trpc-a2a-go v0.2.5-0.20251023030722-7f02b57fd14a // indirect
94177
trpc.group/trpc-go/trpc-agent-go/storage/elasticsearch v0.2.0 // indirect
178+
trpc.group/trpc-go/trpc-agent-go/storage/milvus v0.0.4 // indirect
95179
trpc.group/trpc-go/trpc-agent-go/storage/postgres v0.0.0 // indirect
96180
trpc.group/trpc-go/trpc-agent-go/storage/tcvector v0.0.4 // indirect
97181
)

0 commit comments

Comments
 (0)