Skip to content

Commit 137f586

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

File tree

19 files changed

+7214
-43
lines changed

19 files changed

+7214
-43
lines changed

examples/knowledge/go.mod

Lines changed: 93 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,53 +1,81 @@
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/ocr/tesseract => ../../knowledge/ocr/tesseract
1010
trpc.group/trpc-go/trpc-agent-go/knowledge/vectorstore/elasticsearch => ../../knowledge/vectorstore/elasticsearch
11+
trpc.group/trpc-go/trpc-agent-go/knowledge/vectorstore/milvus => ../../knowledge/vectorstore/milvus
1112
trpc.group/trpc-go/trpc-agent-go/knowledge/vectorstore/pgvector => ../../knowledge/vectorstore/pgvector
1213
trpc.group/trpc-go/trpc-agent-go/knowledge/vectorstore/tcvector => ../../knowledge/vectorstore/tcvector
14+
trpc.group/trpc-go/trpc-agent-go/storage/milvus => ../../storage/milvus
1315
trpc.group/trpc-go/trpc-agent-go/storage/postgres => ../../storage/postgres
1416
)
1517

1618
require (
1719
github.com/tencent/vectordatabase-sdk-go v1.8.0
18-
trpc.group/trpc-go/trpc-agent-go v0.5.0
20+
trpc.group/trpc-go/trpc-agent-go v0.6.0
1921
trpc.group/trpc-go/trpc-agent-go/knowledge/document/reader/pdf v0.5.0
2022
trpc.group/trpc-go/trpc-agent-go/knowledge/embedder/gemini v0.0.0-20250917031858-f0ddbd5b2cb4
2123
trpc.group/trpc-go/trpc-agent-go/knowledge/embedder/huggingface v0.0.0-20251119113046-0cbdb93921df
2224
trpc.group/trpc-go/trpc-agent-go/knowledge/embedder/ollama v0.0.0-20251111070215-8fe58a4f2ffa
2325
trpc.group/trpc-go/trpc-agent-go/knowledge/ocr/tesseract v0.0.0-00010101000000-000000000000
2426
trpc.group/trpc-go/trpc-agent-go/knowledge/vectorstore/elasticsearch v0.2.1
27+
trpc.group/trpc-go/trpc-agent-go/knowledge/vectorstore/milvus v0.0.0-00010101000000-000000000000
2528
trpc.group/trpc-go/trpc-agent-go/knowledge/vectorstore/pgvector v0.2.0
2629
trpc.group/trpc-go/trpc-agent-go/knowledge/vectorstore/tcvector v0.2.0
2730
)
2831

2932
require (
30-
cloud.google.com/go v0.116.0 // indirect
31-
cloud.google.com/go/auth v0.9.3 // indirect
33+
cloud.google.com/go v0.118.3 // indirect
34+
cloud.google.com/go/auth v0.15.0 // indirect
3235
cloud.google.com/go/compute/metadata v0.6.0 // indirect
36+
github.com/beorn7/perks v1.0.1 // indirect
37+
github.com/blang/semver/v4 v4.0.0 // indirect
3338
github.com/bmatcuk/doublestar/v4 v4.9.1 // indirect
3439
github.com/cenkalti/backoff/v4 v4.3.0 // indirect
40+
github.com/cespare/xxhash/v2 v2.3.0 // indirect
41+
github.com/cilium/ebpf v0.11.0 // indirect
3542
github.com/clbanning/mxj v1.8.4 // indirect
3643
github.com/clipperhouse/uax29/v2 v2.2.0 // indirect
44+
github.com/cockroachdb/errors v1.9.1 // indirect
45+
github.com/cockroachdb/logtags v0.0.0-20211118104740-dabe8e521a4f // indirect
46+
github.com/cockroachdb/redact v1.1.3 // indirect
47+
github.com/containerd/cgroups/v3 v3.0.3 // indirect
48+
github.com/coreos/go-semver v0.3.0 // indirect
49+
github.com/coreos/go-systemd/v22 v22.3.2 // indirect
50+
github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect
51+
github.com/docker/go-units v0.5.0 // indirect
52+
github.com/dustin/go-humanize v1.0.1 // indirect
3753
github.com/elastic/elastic-transport-go/v8 v8.7.0 // indirect
3854
github.com/elastic/go-elasticsearch/v7 v7.17.10 // indirect
3955
github.com/elastic/go-elasticsearch/v8 v8.19.0 // indirect
4056
github.com/elastic/go-elasticsearch/v9 v9.1.0 // indirect
57+
github.com/felixge/httpsnoop v1.0.4 // indirect
58+
github.com/form3tech-oss/jwt-go v3.2.3+incompatible // indirect
59+
github.com/fxamacker/cbor/v2 v2.7.0 // indirect
60+
github.com/getsentry/sentry-go v0.12.0 // indirect
4161
github.com/go-ego/gse v0.80.3 // indirect
4262
github.com/go-logr/logr v1.4.3 // indirect
4363
github.com/go-logr/stdr v1.2.2 // indirect
44-
github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da // indirect
64+
github.com/go-ole/go-ole v1.2.6 // indirect
65+
github.com/godbus/dbus/v5 v5.0.4 // indirect
66+
github.com/gogo/protobuf v1.3.2 // indirect
67+
github.com/golang/protobuf v1.5.4 // indirect
68+
github.com/google/btree v1.1.2 // indirect
4569
github.com/google/go-cmp v0.7.0 // indirect
4670
github.com/google/go-querystring v1.1.0 // indirect
47-
github.com/google/s2a-go v0.1.8 // indirect
71+
github.com/google/s2a-go v0.1.9 // indirect
4872
github.com/google/uuid v1.6.0 // indirect
49-
github.com/googleapis/enterprise-certificate-proxy v0.3.4 // indirect
73+
github.com/googleapis/enterprise-certificate-proxy v0.3.5 // indirect
74+
github.com/googleapis/gax-go/v2 v2.14.1 // indirect
5075
github.com/gorilla/websocket v1.5.3 // indirect
76+
github.com/grpc-ecosystem/go-grpc-middleware v1.3.0 // indirect
77+
github.com/grpc-ecosystem/go-grpc-prometheus v1.2.0 // indirect
78+
github.com/grpc-ecosystem/grpc-gateway v1.16.0 // indirect
5179
github.com/grpc-ecosystem/grpc-gateway/v2 v2.22.0 // indirect
5280
github.com/hhrutter/lzw v1.0.0 // indirect
5381
github.com/hhrutter/pkcs7 v0.2.0 // indirect
@@ -56,27 +84,71 @@ require (
5684
github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 // indirect
5785
github.com/jackc/pgx/v5 v5.7.2 // indirect
5886
github.com/jackc/puddle/v2 v2.2.2 // indirect
87+
github.com/jonboulle/clockwork v0.2.2 // indirect
88+
github.com/json-iterator/go v1.1.12 // indirect
89+
github.com/klauspost/compress v1.17.9 // indirect
90+
github.com/kr/pretty v0.3.1 // indirect
91+
github.com/kr/text v0.2.0 // indirect
5992
github.com/ledongthuc/pdf v0.0.0-20250511090121-5959a4027728 // indirect
93+
github.com/lufia/plan9stats v0.0.0-20211012122336-39d0f177ccd0 // indirect
6094
github.com/mattn/go-runewidth v0.0.19 // indirect
95+
github.com/milvus-io/milvus-proto/go-api/v2 v2.6.3 // indirect
96+
github.com/milvus-io/milvus/client/v2 v2.6.1 // indirect
97+
github.com/milvus-io/milvus/pkg/v2 v2.6.3 // indirect
6198
github.com/mitchellh/mapstructure v1.5.0 // indirect
99+
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
100+
github.com/modern-go/reflect2 v1.0.2 // indirect
62101
github.com/mozillazg/go-httpheader v0.4.0 // indirect
102+
github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 // indirect
63103
github.com/ollama/ollama v0.12.9 // indirect
64104
github.com/openai/openai-go v1.12.0 // indirect
105+
github.com/opencontainers/runtime-spec v1.0.2 // indirect
65106
github.com/otiai10/gosseract/v2 v2.4.1 // indirect
66107
github.com/panjf2000/ants/v2 v2.11.3 // indirect
67108
github.com/pdfcpu/pdfcpu v0.11.1 // indirect
68109
github.com/pgvector/pgvector-go v0.3.0 // indirect
69110
github.com/pkg/errors v0.9.1 // indirect
111+
github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect
112+
github.com/power-devops/perfstat v0.0.0-20210106213030-5aafc221ea8c // indirect
113+
github.com/prometheus/client_golang v1.20.5 // indirect
114+
github.com/prometheus/client_model v0.6.1 // indirect
115+
github.com/prometheus/common v0.55.0 // indirect
116+
github.com/prometheus/procfs v0.15.1 // indirect
117+
github.com/rogpeppe/go-internal v1.13.1 // indirect
118+
github.com/samber/lo v1.27.0 // indirect
119+
github.com/shirou/gopsutil/v3 v3.23.12 // indirect
120+
github.com/shoenig/go-m1cpu v0.1.6 // indirect
121+
github.com/sirupsen/logrus v1.9.3 // indirect
122+
github.com/soheilhy/cmux v0.1.5 // indirect
70123
github.com/spaolacci/murmur3 v1.1.0 // indirect
124+
github.com/spf13/cast v1.3.1 // indirect
125+
github.com/spf13/pflag v1.0.5 // indirect
126+
github.com/stretchr/testify v1.11.1 // indirect
71127
github.com/tencentyun/cos-go-sdk-v5 v0.7.69 // indirect
72-
github.com/tidwall/gjson v1.14.4 // indirect
128+
github.com/tidwall/gjson v1.17.1 // indirect
73129
github.com/tidwall/match v1.1.1 // indirect
74130
github.com/tidwall/pretty v1.2.1 // indirect
75131
github.com/tidwall/sjson v1.2.5 // indirect
132+
github.com/tklauser/go-sysconf v0.3.12 // indirect
133+
github.com/tklauser/numcpus v0.6.1 // indirect
134+
github.com/tmc/grpc-websocket-proxy v0.0.0-20201229170055-e5319fda7802 // indirect
135+
github.com/uber/jaeger-client-go v2.30.0+incompatible // indirect
76136
github.com/vcaesar/cedar v0.20.2 // indirect
137+
github.com/x448/float16 v0.8.4 // indirect
138+
github.com/xiang90/probing v0.0.0-20190116061207-43a291ad63a2 // indirect
77139
github.com/yuin/goldmark v1.4.13 // indirect
78-
go.opencensus.io v0.24.0 // indirect
140+
github.com/yusufpapurcu/wmi v1.2.3 // indirect
141+
go.etcd.io/bbolt v1.3.6 // indirect
142+
go.etcd.io/etcd/api/v3 v3.5.5 // indirect
143+
go.etcd.io/etcd/client/pkg/v3 v3.5.5 // indirect
144+
go.etcd.io/etcd/client/v2 v2.305.5 // indirect
145+
go.etcd.io/etcd/client/v3 v3.5.5 // indirect
146+
go.etcd.io/etcd/pkg/v3 v3.5.5 // indirect
147+
go.etcd.io/etcd/raft/v3 v3.5.5 // indirect
148+
go.etcd.io/etcd/server/v3 v3.5.5 // indirect
79149
go.opentelemetry.io/auto/sdk v1.1.0 // indirect
150+
go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.59.0 // indirect
151+
go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.59.0 // indirect
80152
go.opentelemetry.io/otel v1.37.0 // indirect
81153
go.opentelemetry.io/otel/exporters/otlp/otlptrace v1.29.0 // indirect
82154
go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracegrpc v1.29.0 // indirect
@@ -85,22 +157,33 @@ require (
85157
go.opentelemetry.io/otel/sdk v1.35.0 // indirect
86158
go.opentelemetry.io/otel/trace v1.37.0 // indirect
87159
go.opentelemetry.io/proto/otlp v1.3.1 // indirect
88-
go.uber.org/multierr v1.10.0 // indirect
160+
go.uber.org/atomic v1.11.0 // indirect
161+
go.uber.org/automaxprocs v1.5.3 // indirect
162+
go.uber.org/multierr v1.11.0 // indirect
89163
go.uber.org/zap v1.27.0 // indirect
90164
golang.org/x/crypto v0.43.0 // indirect
165+
golang.org/x/exp v0.0.0-20250218142911-aa4b98e5adaa // indirect
91166
golang.org/x/image v0.32.0 // indirect
92167
golang.org/x/net v0.45.0 // indirect
93168
golang.org/x/sync v0.17.0 // indirect
94169
golang.org/x/sys v0.37.0 // indirect
95170
golang.org/x/text v0.30.0 // indirect
171+
golang.org/x/time v0.10.0 // indirect
96172
google.golang.org/genai v1.0.0 // indirect
173+
google.golang.org/genproto v0.0.0-20250303144028-a0af3efb3deb // indirect
97174
google.golang.org/genproto/googleapis/api v0.0.0-20250324211829-b45e905df463 // indirect
98175
google.golang.org/genproto/googleapis/rpc v0.0.0-20250707201910-8d1bb00bc6a7 // indirect
99176
google.golang.org/grpc v1.73.0 // indirect
100177
google.golang.org/protobuf v1.36.6 // indirect
178+
gopkg.in/inf.v0 v0.9.1 // indirect
179+
gopkg.in/natefinch/lumberjack.v2 v2.2.1 // indirect
101180
gopkg.in/yaml.v2 v2.4.0 // indirect
181+
gopkg.in/yaml.v3 v3.0.1 // indirect
182+
k8s.io/apimachinery v0.32.3 // indirect
183+
sigs.k8s.io/yaml v1.4.0 // indirect
102184
trpc.group/trpc-go/trpc-a2a-go v0.2.5-0.20251023030722-7f02b57fd14a // indirect
103185
trpc.group/trpc-go/trpc-agent-go/storage/elasticsearch v0.2.0 // indirect
186+
trpc.group/trpc-go/trpc-agent-go/storage/milvus v0.0.4 // indirect
104187
trpc.group/trpc-go/trpc-agent-go/storage/postgres v0.0.0-20251126064502-c8c2594d2519 // indirect
105188
trpc.group/trpc-go/trpc-agent-go/storage/tcvector v0.0.4 // indirect
106189
)

0 commit comments

Comments
 (0)