262 lines
11 KiB
Go
262 lines
11 KiB
Go
// Package knowledge is the Code Intelligence (RAG) bounded context:
|
|
// chunking, embedding (via the knowledge service's vLLM-backed embed client) and
|
|
// cosine-similarity search over workspace source files. System of
|
|
// record for the pgvector-backed `documents` table (see
|
|
// services/knowledge/migrations/0001_init.sql).
|
|
//
|
|
// Mirrors internal/rag/*.go. IndexDirectory (cmd/indexer) now walks the
|
|
// workspace through the workspace service instead of the local
|
|
// filesystem — see tools/indexer.
|
|
|
|
// Code generated by protoc-gen-go-grpc. DO NOT EDIT.
|
|
// versions:
|
|
// - protoc-gen-go-grpc v1.5.1
|
|
// - protoc v3.21.12
|
|
// source: knowledge/v1/knowledge.proto
|
|
|
|
package knowledgev1
|
|
|
|
import (
|
|
context "context"
|
|
grpc "google.golang.org/grpc"
|
|
codes "google.golang.org/grpc/codes"
|
|
status "google.golang.org/grpc/status"
|
|
)
|
|
|
|
// This is a compile-time assertion to ensure that this generated file
|
|
// is compatible with the grpc package it is being compiled against.
|
|
// Requires gRPC-Go v1.64.0 or later.
|
|
const _ = grpc.SupportPackageIsVersion9
|
|
|
|
const (
|
|
KnowledgeService_IndexFile_FullMethodName = "/enkly.knowledge.v1.KnowledgeService/IndexFile"
|
|
KnowledgeService_PruneMissing_FullMethodName = "/enkly.knowledge.v1.KnowledgeService/PruneMissing"
|
|
KnowledgeService_Search_FullMethodName = "/enkly.knowledge.v1.KnowledgeService/Search"
|
|
KnowledgeService_ListIndexedFiles_FullMethodName = "/enkly.knowledge.v1.KnowledgeService/ListIndexedFiles"
|
|
)
|
|
|
|
// KnowledgeServiceClient is the client API for KnowledgeService service.
|
|
//
|
|
// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.
|
|
type KnowledgeServiceClient interface {
|
|
// IndexFile replaces rag.Store.IndexFile: chunks the given file
|
|
// content, embeds each new/changed chunk, and upserts it.
|
|
IndexFile(ctx context.Context, in *IndexFileRequest, opts ...grpc.CallOption) (*IndexFileResponse, error)
|
|
// PruneMissing replaces rag.Store.pruneMissing, called by the indexer
|
|
// after a full directory walk to drop rows for files no longer
|
|
// present.
|
|
PruneMissing(ctx context.Context, in *PruneMissingRequest, opts ...grpc.CallOption) (*PruneMissingResponse, error)
|
|
// Search replaces rag.Store.Search — the search_codebase agent tool.
|
|
Search(ctx context.Context, in *SearchRequest, opts ...grpc.CallOption) (*SearchResponse, error)
|
|
// ListIndexedFiles replaces db.ListIndexedFiles, used by the gateway's
|
|
// sidebar "indexed files" panel (Handlers.renderConversation).
|
|
ListIndexedFiles(ctx context.Context, in *ListIndexedFilesRequest, opts ...grpc.CallOption) (*ListIndexedFilesResponse, error)
|
|
}
|
|
|
|
type knowledgeServiceClient struct {
|
|
cc grpc.ClientConnInterface
|
|
}
|
|
|
|
func NewKnowledgeServiceClient(cc grpc.ClientConnInterface) KnowledgeServiceClient {
|
|
return &knowledgeServiceClient{cc}
|
|
}
|
|
|
|
func (c *knowledgeServiceClient) IndexFile(ctx context.Context, in *IndexFileRequest, opts ...grpc.CallOption) (*IndexFileResponse, error) {
|
|
cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
|
|
out := new(IndexFileResponse)
|
|
err := c.cc.Invoke(ctx, KnowledgeService_IndexFile_FullMethodName, in, out, cOpts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return out, nil
|
|
}
|
|
|
|
func (c *knowledgeServiceClient) PruneMissing(ctx context.Context, in *PruneMissingRequest, opts ...grpc.CallOption) (*PruneMissingResponse, error) {
|
|
cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
|
|
out := new(PruneMissingResponse)
|
|
err := c.cc.Invoke(ctx, KnowledgeService_PruneMissing_FullMethodName, in, out, cOpts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return out, nil
|
|
}
|
|
|
|
func (c *knowledgeServiceClient) Search(ctx context.Context, in *SearchRequest, opts ...grpc.CallOption) (*SearchResponse, error) {
|
|
cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
|
|
out := new(SearchResponse)
|
|
err := c.cc.Invoke(ctx, KnowledgeService_Search_FullMethodName, in, out, cOpts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return out, nil
|
|
}
|
|
|
|
func (c *knowledgeServiceClient) ListIndexedFiles(ctx context.Context, in *ListIndexedFilesRequest, opts ...grpc.CallOption) (*ListIndexedFilesResponse, error) {
|
|
cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
|
|
out := new(ListIndexedFilesResponse)
|
|
err := c.cc.Invoke(ctx, KnowledgeService_ListIndexedFiles_FullMethodName, in, out, cOpts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return out, nil
|
|
}
|
|
|
|
// KnowledgeServiceServer is the server API for KnowledgeService service.
|
|
// All implementations must embed UnimplementedKnowledgeServiceServer
|
|
// for forward compatibility.
|
|
type KnowledgeServiceServer interface {
|
|
// IndexFile replaces rag.Store.IndexFile: chunks the given file
|
|
// content, embeds each new/changed chunk, and upserts it.
|
|
IndexFile(context.Context, *IndexFileRequest) (*IndexFileResponse, error)
|
|
// PruneMissing replaces rag.Store.pruneMissing, called by the indexer
|
|
// after a full directory walk to drop rows for files no longer
|
|
// present.
|
|
PruneMissing(context.Context, *PruneMissingRequest) (*PruneMissingResponse, error)
|
|
// Search replaces rag.Store.Search — the search_codebase agent tool.
|
|
Search(context.Context, *SearchRequest) (*SearchResponse, error)
|
|
// ListIndexedFiles replaces db.ListIndexedFiles, used by the gateway's
|
|
// sidebar "indexed files" panel (Handlers.renderConversation).
|
|
ListIndexedFiles(context.Context, *ListIndexedFilesRequest) (*ListIndexedFilesResponse, error)
|
|
mustEmbedUnimplementedKnowledgeServiceServer()
|
|
}
|
|
|
|
// UnimplementedKnowledgeServiceServer must be embedded to have
|
|
// forward compatible implementations.
|
|
//
|
|
// NOTE: this should be embedded by value instead of pointer to avoid a nil
|
|
// pointer dereference when methods are called.
|
|
type UnimplementedKnowledgeServiceServer struct{}
|
|
|
|
func (UnimplementedKnowledgeServiceServer) IndexFile(context.Context, *IndexFileRequest) (*IndexFileResponse, error) {
|
|
return nil, status.Errorf(codes.Unimplemented, "method IndexFile not implemented")
|
|
}
|
|
func (UnimplementedKnowledgeServiceServer) PruneMissing(context.Context, *PruneMissingRequest) (*PruneMissingResponse, error) {
|
|
return nil, status.Errorf(codes.Unimplemented, "method PruneMissing not implemented")
|
|
}
|
|
func (UnimplementedKnowledgeServiceServer) Search(context.Context, *SearchRequest) (*SearchResponse, error) {
|
|
return nil, status.Errorf(codes.Unimplemented, "method Search not implemented")
|
|
}
|
|
func (UnimplementedKnowledgeServiceServer) ListIndexedFiles(context.Context, *ListIndexedFilesRequest) (*ListIndexedFilesResponse, error) {
|
|
return nil, status.Errorf(codes.Unimplemented, "method ListIndexedFiles not implemented")
|
|
}
|
|
func (UnimplementedKnowledgeServiceServer) mustEmbedUnimplementedKnowledgeServiceServer() {}
|
|
func (UnimplementedKnowledgeServiceServer) testEmbeddedByValue() {}
|
|
|
|
// UnsafeKnowledgeServiceServer may be embedded to opt out of forward compatibility for this service.
|
|
// Use of this interface is not recommended, as added methods to KnowledgeServiceServer will
|
|
// result in compilation errors.
|
|
type UnsafeKnowledgeServiceServer interface {
|
|
mustEmbedUnimplementedKnowledgeServiceServer()
|
|
}
|
|
|
|
func RegisterKnowledgeServiceServer(s grpc.ServiceRegistrar, srv KnowledgeServiceServer) {
|
|
// If the following call pancis, it indicates UnimplementedKnowledgeServiceServer was
|
|
// embedded by pointer and is nil. This will cause panics if an
|
|
// unimplemented method is ever invoked, so we test this at initialization
|
|
// time to prevent it from happening at runtime later due to I/O.
|
|
if t, ok := srv.(interface{ testEmbeddedByValue() }); ok {
|
|
t.testEmbeddedByValue()
|
|
}
|
|
s.RegisterService(&KnowledgeService_ServiceDesc, srv)
|
|
}
|
|
|
|
func _KnowledgeService_IndexFile_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
|
|
in := new(IndexFileRequest)
|
|
if err := dec(in); err != nil {
|
|
return nil, err
|
|
}
|
|
if interceptor == nil {
|
|
return srv.(KnowledgeServiceServer).IndexFile(ctx, in)
|
|
}
|
|
info := &grpc.UnaryServerInfo{
|
|
Server: srv,
|
|
FullMethod: KnowledgeService_IndexFile_FullMethodName,
|
|
}
|
|
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
|
|
return srv.(KnowledgeServiceServer).IndexFile(ctx, req.(*IndexFileRequest))
|
|
}
|
|
return interceptor(ctx, in, info, handler)
|
|
}
|
|
|
|
func _KnowledgeService_PruneMissing_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
|
|
in := new(PruneMissingRequest)
|
|
if err := dec(in); err != nil {
|
|
return nil, err
|
|
}
|
|
if interceptor == nil {
|
|
return srv.(KnowledgeServiceServer).PruneMissing(ctx, in)
|
|
}
|
|
info := &grpc.UnaryServerInfo{
|
|
Server: srv,
|
|
FullMethod: KnowledgeService_PruneMissing_FullMethodName,
|
|
}
|
|
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
|
|
return srv.(KnowledgeServiceServer).PruneMissing(ctx, req.(*PruneMissingRequest))
|
|
}
|
|
return interceptor(ctx, in, info, handler)
|
|
}
|
|
|
|
func _KnowledgeService_Search_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
|
|
in := new(SearchRequest)
|
|
if err := dec(in); err != nil {
|
|
return nil, err
|
|
}
|
|
if interceptor == nil {
|
|
return srv.(KnowledgeServiceServer).Search(ctx, in)
|
|
}
|
|
info := &grpc.UnaryServerInfo{
|
|
Server: srv,
|
|
FullMethod: KnowledgeService_Search_FullMethodName,
|
|
}
|
|
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
|
|
return srv.(KnowledgeServiceServer).Search(ctx, req.(*SearchRequest))
|
|
}
|
|
return interceptor(ctx, in, info, handler)
|
|
}
|
|
|
|
func _KnowledgeService_ListIndexedFiles_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
|
|
in := new(ListIndexedFilesRequest)
|
|
if err := dec(in); err != nil {
|
|
return nil, err
|
|
}
|
|
if interceptor == nil {
|
|
return srv.(KnowledgeServiceServer).ListIndexedFiles(ctx, in)
|
|
}
|
|
info := &grpc.UnaryServerInfo{
|
|
Server: srv,
|
|
FullMethod: KnowledgeService_ListIndexedFiles_FullMethodName,
|
|
}
|
|
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
|
|
return srv.(KnowledgeServiceServer).ListIndexedFiles(ctx, req.(*ListIndexedFilesRequest))
|
|
}
|
|
return interceptor(ctx, in, info, handler)
|
|
}
|
|
|
|
// KnowledgeService_ServiceDesc is the grpc.ServiceDesc for KnowledgeService service.
|
|
// It's only intended for direct use with grpc.RegisterService,
|
|
// and not to be introspected or modified (even as a copy)
|
|
var KnowledgeService_ServiceDesc = grpc.ServiceDesc{
|
|
ServiceName: "enkly.knowledge.v1.KnowledgeService",
|
|
HandlerType: (*KnowledgeServiceServer)(nil),
|
|
Methods: []grpc.MethodDesc{
|
|
{
|
|
MethodName: "IndexFile",
|
|
Handler: _KnowledgeService_IndexFile_Handler,
|
|
},
|
|
{
|
|
MethodName: "PruneMissing",
|
|
Handler: _KnowledgeService_PruneMissing_Handler,
|
|
},
|
|
{
|
|
MethodName: "Search",
|
|
Handler: _KnowledgeService_Search_Handler,
|
|
},
|
|
{
|
|
MethodName: "ListIndexedFiles",
|
|
Handler: _KnowledgeService_ListIndexedFiles_Handler,
|
|
},
|
|
},
|
|
Streams: []grpc.StreamDesc{},
|
|
Metadata: "knowledge/v1/knowledge.proto",
|
|
}
|