@@ -11,7 +11,7 @@ import (
1111
1212 "github.com/imtaebin/code-context-graph/internal/model"
1313 "github.com/imtaebin/code-context-graph/internal/parse"
14- "github.com/imtaebin/code-context-graph/internal/store/agestore "
14+ "github.com/imtaebin/code-context-graph/internal/store/pgstore "
1515)
1616
1717var skipDirs = map [string ]bool {
@@ -175,30 +175,30 @@ func newBuildCmd(deps *Deps) *cobra.Command {
175175 }
176176
177177 // Sync to Apache AGE graph + pgvector if --graph flag is set
178- if syncGraph && deps .AgeStore != nil {
178+ if syncGraph && deps .PGStore != nil {
179179 deps .Logger .Info ("syncing to AGE graph" )
180- if err := deps .AgeStore .ClearGraph (ctx ); err != nil {
180+ if err := deps .PGStore .ClearGraph (ctx ); err != nil {
181181 deps .Logger .Warn ("clear graph failed" , "error" , err )
182182 }
183- if err := deps .AgeStore .SyncNodes (ctx , indexNodes ); err != nil {
183+ if err := deps .PGStore .SyncNodes (ctx , indexNodes ); err != nil {
184184 deps .Logger .Warn ("sync nodes to AGE failed" , "error" , err )
185185 } else {
186186 var edges []model.Edge
187187 deps .DB .Find (& edges )
188- if err := deps .AgeStore .SyncEdges (ctx , edges ); err != nil {
188+ if err := deps .PGStore .SyncEdges (ctx , edges ); err != nil {
189189 deps .Logger .Warn ("sync edges to AGE failed" , "error" , err )
190190 } else {
191191 deps .Logger .Info ("AGE graph synced" , "nodes" , len (indexNodes ), "edges" , len (edges ))
192192 }
193193 }
194194
195195 // Sync pgvector documents for semantic search
196- if err := deps .AgeStore .InitPGVector (ctx ); err != nil {
196+ if err := deps .PGStore .InitPGVector (ctx ); err != nil {
197197 deps .Logger .Warn ("pgvector init failed" , "error" , err )
198198 } else {
199- var pvDocs []agestore .PGVectorDocument
199+ var pvDocs []pgstore .PGVectorDocument
200200 for _ , n := range indexNodes {
201- pvDocs = append (pvDocs , agestore .PGVectorDocument {
201+ pvDocs = append (pvDocs , pgstore .PGVectorDocument {
202202 NodeID : n .ID ,
203203 Content : buildContent (n ),
204204 Metadata : map [string ]string {
@@ -209,7 +209,7 @@ func newBuildCmd(deps *Deps) *cobra.Command {
209209 },
210210 })
211211 }
212- if err := deps .AgeStore .SyncPGVectorDocuments (ctx , pvDocs ); err != nil {
212+ if err := deps .PGStore .SyncPGVectorDocuments (ctx , pvDocs ); err != nil {
213213 deps .Logger .Warn ("pgvector sync failed" , "error" , err )
214214 } else {
215215 deps .Logger .Info ("pgvector documents synced" , "documents" , len (pvDocs ))
0 commit comments