84555dcb44
- Injected HTTPUpstream service into LanguageServerService - Implemented real upstream API requests via callUpstreamAPI() - Added SSE streaming response handler for streaming messages - Complete error handling and structured logging - Support for masquerading headers (User-Agent, Authorization) - Request/response body marshaling and streaming - Thread-safe session management with metadata storage Core implementation: - LanguageServerService now depends on HTTPUpstream for all HTTP operations - HTTP requests sent to configured Anthropic API endpoint - SSE event parsing and forwarding to clients via update channels - Proper context and timeout handling for streaming operations Phase 1 Status: 95% complete - Upstream API integration: ✅ DONE - Wire dependency injection: ⏳ TODO - Masquerading layer: ⏳ TODO (Phase 2) Next steps: 1. Add Wire provider for LanguageServerService 2. Register HTTP routes in application startup 3. Implement device fingerprinting and token refresh 4. End-to-end testing with real Anthropic API Co-Authored-By: Claude Haiku 4.5 <noreply@anthropic.com>
449 lines
13 KiB
Protocol Buffer
449 lines
13 KiB
Protocol Buffer
syntax = "proto3";
|
|
package exa.index_pb;
|
|
|
|
option go_package = "github.com/user/antigravity-client/gen/exa/index_pb";
|
|
|
|
import "exa/codeium_common_pb/codeium_common.proto";
|
|
import "google/protobuf/duration.proto";
|
|
import "google/protobuf/source_context.proto";
|
|
import "google/protobuf/timestamp.proto";
|
|
|
|
enum IndexMode {
|
|
INDEX_MODE_UNSPECIFIED = 0;
|
|
INDEX_MODE_HALFVEC = 1;
|
|
INDEX_MODE_BINARY = 2;
|
|
INDEX_MODE_BINARY_WITH_RERANK = 3;
|
|
INDEX_MODE_BRUTE_FORCE = 4;
|
|
INDEX_MODE_RANDOM_SEARCH = 5;
|
|
}
|
|
|
|
enum IndexingStatus {
|
|
INDEXING_STATUS_UNSPECIFIED = 0;
|
|
INDEXING_STATUS_ERROR = 1;
|
|
INDEXING_STATUS_QUEUED = 2;
|
|
INDEXING_STATUS_CLONING_REPO = 3;
|
|
INDEXING_STATUS_SCANNING_REPO = 4;
|
|
INDEXING_STATUS_GENERATING_EMBEDDINGS = 5;
|
|
INDEXING_STATUS_VECTOR_INDEXING = 6;
|
|
INDEXING_STATUS_DONE = 7;
|
|
INDEXING_STATUS_CANCELING = 8;
|
|
INDEXING_STATUS_CANCELED = 9;
|
|
}
|
|
|
|
message IndexDbVersion {
|
|
int32 version = 1;
|
|
int32 enterprise_version = 2;
|
|
}
|
|
|
|
message IndexBuildConfig {
|
|
exa.index_pb.IndexDbVersion db_version = 2;
|
|
int32 cci_timeout_secs = 3;
|
|
exa.index_pb.IndexMode index_mode = 4;
|
|
}
|
|
|
|
message RepositoryConfig {
|
|
message AutoIndexConfig {
|
|
string branch_name = 1;
|
|
google.protobuf.Duration interval = 2;
|
|
int32 max_num_auto_indexes = 3;
|
|
}
|
|
string git_url = 1;
|
|
exa.codeium_common_pb.ScmProvider scm_provider = 2;
|
|
exa.index_pb.RepositoryConfig.AutoIndexConfig auto_index_config = 3;
|
|
bool store_snippets = 4;
|
|
repeated string whitelisted_groups = 5;
|
|
bool use_github_app = 6;
|
|
string auth_uid = 7;
|
|
string email = 9;
|
|
string service_key_id = 8;
|
|
}
|
|
|
|
message IndexConfig {
|
|
google.protobuf.Timestamp prune_time = 1;
|
|
google.protobuf.Duration prune_interval = 2;
|
|
bool enable_prune = 3;
|
|
bool enable_smallest_repo_first = 4;
|
|
bool enable_round_robin = 5;
|
|
}
|
|
|
|
message VectorIndexStats {
|
|
int64 num_embeddings = 1;
|
|
int64 index_bytes_count = 2;
|
|
}
|
|
|
|
message ProgressBar {
|
|
float progress = 1;
|
|
string text = 2;
|
|
google.protobuf.Duration remaining_time = 3;
|
|
}
|
|
|
|
message Index {
|
|
message RepoStats {
|
|
int64 size = 1;
|
|
int64 file_count = 2;
|
|
int64 size_no_ignore = 3;
|
|
int64 file_count_no_ignore = 4;
|
|
}
|
|
string id = 1;
|
|
string repo_name = 2;
|
|
string workspace = 3;
|
|
exa.codeium_common_pb.GitRepoInfo repo_info = 4;
|
|
google.protobuf.Timestamp created_at = 5;
|
|
google.protobuf.Timestamp updated_at = 6;
|
|
google.protobuf.Timestamp scheduled_at = 13;
|
|
exa.index_pb.IndexingStatus status = 7;
|
|
string status_detail = 8;
|
|
bool auto_indexed = 9;
|
|
bool has_snippets = 12;
|
|
string auth_uid = 15;
|
|
string email = 16;
|
|
exa.index_pb.Index.RepoStats repo_stats = 14;
|
|
map<string, exa.index_pb.ProgressBar> indexing_progress = 10;
|
|
exa.index_pb.VectorIndexStats index_stats = 11;
|
|
}
|
|
|
|
message Repository {
|
|
string repo_name = 1;
|
|
exa.index_pb.RepositoryConfig config = 2;
|
|
google.protobuf.Timestamp created_at = 4;
|
|
google.protobuf.Timestamp updated_at = 5;
|
|
google.protobuf.Timestamp last_used_at = 6;
|
|
exa.index_pb.Index latest_index = 3;
|
|
}
|
|
|
|
message RequestIndexVersion {
|
|
oneof version {
|
|
string commit = 1;
|
|
string branch = 2;
|
|
}
|
|
string version_alias = 3;
|
|
}
|
|
|
|
message ManagementMetadata {
|
|
string auth_token = 1;
|
|
string auth_uid = 2;
|
|
string service_key = 3;
|
|
bool force_target_public_index = 4;
|
|
string force_team_id = 5;
|
|
string service_key_id = 6;
|
|
}
|
|
|
|
message AddRepositoryRequest {
|
|
exa.index_pb.ManagementMetadata metadata = 1;
|
|
exa.index_pb.RepositoryConfig config = 2;
|
|
exa.index_pb.RequestIndexVersion initial_index = 3;
|
|
}
|
|
|
|
message AddRepositoryResponse {
|
|
string repo_name = 1;
|
|
string index_id = 2;
|
|
}
|
|
|
|
message EnableIndexingRequest {
|
|
exa.index_pb.ManagementMetadata metadata = 1;
|
|
exa.index_pb.IndexBuildConfig config = 2;
|
|
}
|
|
|
|
message EnableIndexingResponse {
|
|
}
|
|
|
|
message DisableIndexingRequest {
|
|
exa.index_pb.ManagementMetadata metadata = 1;
|
|
}
|
|
|
|
message DisableIndexingResponse {
|
|
}
|
|
|
|
message EditRepositoryRequest {
|
|
exa.index_pb.ManagementMetadata metadata = 1;
|
|
string repo_name = 2;
|
|
exa.index_pb.RepositoryConfig config = 3;
|
|
}
|
|
|
|
message EditRepositoryResponse {
|
|
}
|
|
|
|
message DeleteRepositoryRequest {
|
|
exa.index_pb.ManagementMetadata metadata = 1;
|
|
string repo_name = 2;
|
|
repeated string repo_names = 3;
|
|
}
|
|
|
|
message DeleteRepositoryResponse {
|
|
}
|
|
|
|
message GetRepositoriesFilter {
|
|
string repo_name = 1;
|
|
string group_id = 2;
|
|
}
|
|
|
|
message GetRepositoriesRequest {
|
|
exa.index_pb.ManagementMetadata metadata = 1;
|
|
exa.index_pb.GetRepositoriesFilter filter = 2;
|
|
}
|
|
|
|
message GetRepositoriesResponse {
|
|
repeated exa.index_pb.Repository repositories = 1;
|
|
}
|
|
|
|
message GetIndexesRequest {
|
|
exa.index_pb.ManagementMetadata metadata = 1;
|
|
string repo_name = 2;
|
|
}
|
|
|
|
message GetIndexesResponse {
|
|
repeated exa.index_pb.Index indexes = 1;
|
|
}
|
|
|
|
message GetIndexRequest {
|
|
exa.index_pb.ManagementMetadata metadata = 1;
|
|
string index_id = 2;
|
|
}
|
|
|
|
message GetIndexResponse {
|
|
exa.index_pb.Index index = 1;
|
|
exa.index_pb.Repository repository = 2;
|
|
}
|
|
|
|
message RemoteIndexStats {
|
|
string index_id = 1;
|
|
int64 cci_count = 2;
|
|
int64 snippet_count = 3;
|
|
int64 embedding_count = 4;
|
|
}
|
|
|
|
message GetRemoteIndexStatsRequest {
|
|
exa.index_pb.ManagementMetadata metadata = 1;
|
|
repeated string index_ids = 2;
|
|
}
|
|
|
|
message GetRemoteIndexStatsResponse {
|
|
repeated exa.index_pb.RemoteIndexStats index_stats = 1;
|
|
}
|
|
|
|
message AddIndexRequest {
|
|
exa.index_pb.ManagementMetadata metadata = 1;
|
|
string repo_name = 2;
|
|
exa.index_pb.RequestIndexVersion version = 3;
|
|
}
|
|
|
|
message AddIndexResponse {
|
|
string index_id = 1;
|
|
}
|
|
|
|
message CancelIndexingRequest {
|
|
exa.index_pb.ManagementMetadata metadata = 1;
|
|
string index_id = 2;
|
|
repeated string index_ids = 3;
|
|
}
|
|
|
|
message CancelIndexingResponse {
|
|
}
|
|
|
|
message RetryIndexingRequest {
|
|
exa.index_pb.ManagementMetadata metadata = 1;
|
|
string index_id = 2;
|
|
repeated string index_ids = 3;
|
|
}
|
|
|
|
message RetryIndexingResponse {
|
|
}
|
|
|
|
message DeleteIndexRequest {
|
|
exa.index_pb.ManagementMetadata metadata = 1;
|
|
string index_id = 2;
|
|
repeated string index_ids = 3;
|
|
}
|
|
|
|
message DeleteIndexResponse {
|
|
}
|
|
|
|
message PruneDatabaseRequest {
|
|
exa.index_pb.ManagementMetadata metadata = 1;
|
|
}
|
|
|
|
message PruneDatabaseResponse {
|
|
}
|
|
|
|
message GetDatabaseStatsRequest {
|
|
exa.index_pb.ManagementMetadata metadata = 1;
|
|
}
|
|
|
|
message GetDatabaseStatsResponse {
|
|
int64 database_total_bytes_count = 1;
|
|
int64 table_total_bytes_count = 2;
|
|
int64 index_total_bytes_count = 3;
|
|
int64 estimate_prunable_bytes = 4;
|
|
bool is_pruning = 5;
|
|
string last_prune_error = 6;
|
|
int64 all_tables_bytes_count = 7;
|
|
}
|
|
|
|
message SetIndexConfigRequest {
|
|
exa.index_pb.ManagementMetadata metadata = 1;
|
|
exa.index_pb.IndexConfig index_config = 2;
|
|
}
|
|
|
|
message SetIndexConfigResponse {
|
|
}
|
|
|
|
message GetIndexConfigRequest {
|
|
exa.index_pb.ManagementMetadata metadata = 1;
|
|
}
|
|
|
|
message GetIndexConfigResponse {
|
|
exa.index_pb.IndexConfig index_config = 1;
|
|
}
|
|
|
|
message GetNumberConnectionsRequest {
|
|
exa.index_pb.ManagementMetadata metadata = 1;
|
|
}
|
|
|
|
message GetNumberConnectionsResponse {
|
|
map<string, uint32> connections_map = 1;
|
|
}
|
|
|
|
message GetConnectionsDebugInfoRequest {
|
|
exa.index_pb.ManagementMetadata metadata = 1;
|
|
}
|
|
|
|
message GetConnectionsDebugInfoResponse {
|
|
string debug_info = 1;
|
|
}
|
|
|
|
message GetIndexedRepositoriesRequest {
|
|
exa.codeium_common_pb.Metadata metadata = 1;
|
|
bool include_incomplete = 2;
|
|
repeated string group_ids_filter = 3;
|
|
}
|
|
|
|
message GetIndexedRepositoriesResponse {
|
|
repeated exa.codeium_common_pb.GitRepoInfo repositories = 1;
|
|
repeated exa.index_pb.Index indexes = 2;
|
|
}
|
|
|
|
message RepositoryFilter {
|
|
exa.codeium_common_pb.GitRepoInfo repository = 1;
|
|
repeated string excluded_files = 2;
|
|
repeated string filter_paths = 3;
|
|
}
|
|
|
|
message GetMatchingFilePathsRequest {
|
|
exa.codeium_common_pb.Metadata metadata = 1;
|
|
exa.codeium_common_pb.GitRepoInfo repository = 2;
|
|
string query = 3;
|
|
uint32 max_items = 4;
|
|
repeated string group_ids_filter = 5;
|
|
}
|
|
|
|
message GetMatchingFilePathsResponse {
|
|
repeated string relative_file_paths = 1;
|
|
}
|
|
|
|
message GetNearestCCIsFromEmbeddingRequest {
|
|
exa.codeium_common_pb.Metadata metadata = 1;
|
|
exa.codeium_common_pb.Embedding embedding = 2;
|
|
repeated exa.index_pb.RepositoryFilter repository_filters = 3;
|
|
int64 max_results = 4;
|
|
repeated string group_ids_filter = 5;
|
|
}
|
|
|
|
message ScoredContextItem {
|
|
exa.codeium_common_pb.CodeContextItem code_context_item = 1;
|
|
float score = 2;
|
|
}
|
|
|
|
message GetNearestCCIsFromEmbeddingResponse {
|
|
repeated exa.index_pb.ScoredContextItem scored_context_items = 1;
|
|
}
|
|
|
|
message GetEmbeddingsForCodeContextItemsRequest {
|
|
exa.codeium_common_pb.Metadata metadata = 1;
|
|
repeated exa.codeium_common_pb.CodeContextItem code_context_items = 2;
|
|
exa.codeium_common_pb.ContextSnippetType snippet_type = 3;
|
|
}
|
|
|
|
message GetEmbeddingsForCodeContextItemsResponse {
|
|
repeated exa.codeium_common_pb.Embedding embeddings = 1;
|
|
}
|
|
|
|
message IndexStats {
|
|
string repository_name = 1;
|
|
int64 file_count = 2;
|
|
int64 code_context_item_count = 3;
|
|
}
|
|
|
|
message IndexerEvent {
|
|
message Deletion {
|
|
string absolute_uri = 1;
|
|
}
|
|
message Untrack {
|
|
string absolute_uri = 1;
|
|
repeated exa.codeium_common_pb.WorkspacePath paths = 2;
|
|
string workspace_uri = 3;
|
|
}
|
|
message Update {
|
|
message AddWorkspaceInfo {
|
|
uint64 add_workspace_uid = 1;
|
|
uint64 add_workspace_queue_uid = 2;
|
|
}
|
|
string absolute_uri = 1;
|
|
repeated exa.codeium_common_pb.WorkspacePath paths = 2;
|
|
google.protobuf.Timestamp mod_time = 3;
|
|
exa.index_pb.IndexerEvent.Update.AddWorkspaceInfo add_workspace_info = 4;
|
|
}
|
|
message AddWorkspace {
|
|
uint64 add_workspace_uid = 1;
|
|
uint64 add_workspace_queue_uid = 2;
|
|
string workspace_uri = 3;
|
|
int64 num_files = 4;
|
|
int64 size = 5;
|
|
}
|
|
message RemoveWorkspace {
|
|
string workspace_uri = 1;
|
|
}
|
|
message IgnoreWorkspace {
|
|
string workspace_uri = 1;
|
|
}
|
|
message AddCommit {
|
|
string sha = 1;
|
|
}
|
|
uint64 uid = 1;
|
|
oneof event_oneof {
|
|
exa.index_pb.IndexerEvent.Deletion deletion = 2;
|
|
exa.index_pb.IndexerEvent.Untrack untrack = 3;
|
|
exa.index_pb.IndexerEvent.Update update = 4;
|
|
exa.index_pb.IndexerEvent.AddWorkspace add_workspace = 5;
|
|
exa.index_pb.IndexerEvent.RemoveWorkspace remove_workspace = 6;
|
|
exa.index_pb.IndexerEvent.IgnoreWorkspace ignore_workspace = 7;
|
|
exa.index_pb.IndexerEvent.AddCommit add_commit = 8;
|
|
}
|
|
}
|
|
|
|
service IndexManagementService {
|
|
rpc EnableIndexing(exa.index_pb.EnableIndexingRequest) returns (exa.index_pb.EnableIndexingResponse);
|
|
rpc DisableIndexing(exa.index_pb.DisableIndexingRequest) returns (exa.index_pb.DisableIndexingResponse);
|
|
rpc AddRepository(exa.index_pb.AddRepositoryRequest) returns (exa.index_pb.AddRepositoryResponse);
|
|
rpc EditRepository(exa.index_pb.EditRepositoryRequest) returns (exa.index_pb.EditRepositoryResponse);
|
|
rpc DeleteRepository(exa.index_pb.DeleteRepositoryRequest) returns (exa.index_pb.DeleteRepositoryResponse);
|
|
rpc GetRepositories(exa.index_pb.GetRepositoriesRequest) returns (exa.index_pb.GetRepositoriesResponse);
|
|
rpc AddIndex(exa.index_pb.AddIndexRequest) returns (exa.index_pb.AddIndexResponse);
|
|
rpc CancelIndexing(exa.index_pb.CancelIndexingRequest) returns (exa.index_pb.CancelIndexingResponse);
|
|
rpc RetryIndexing(exa.index_pb.RetryIndexingRequest) returns (exa.index_pb.RetryIndexingResponse);
|
|
rpc DeleteIndex(exa.index_pb.DeleteIndexRequest) returns (exa.index_pb.DeleteIndexResponse);
|
|
rpc GetIndexes(exa.index_pb.GetIndexesRequest) returns (exa.index_pb.GetIndexesResponse);
|
|
rpc GetIndex(exa.index_pb.GetIndexRequest) returns (exa.index_pb.GetIndexResponse);
|
|
rpc GetRemoteIndexStats(exa.index_pb.GetRemoteIndexStatsRequest) returns (exa.index_pb.GetRemoteIndexStatsResponse);
|
|
rpc PruneDatabase(exa.index_pb.PruneDatabaseRequest) returns (exa.index_pb.PruneDatabaseResponse);
|
|
rpc GetDatabaseStats(exa.index_pb.GetDatabaseStatsRequest) returns (exa.index_pb.GetDatabaseStatsResponse);
|
|
rpc SetIndexConfig(exa.index_pb.SetIndexConfigRequest) returns (exa.index_pb.SetIndexConfigResponse);
|
|
rpc GetIndexConfig(exa.index_pb.GetIndexConfigRequest) returns (exa.index_pb.GetIndexConfigResponse);
|
|
rpc GetNumberConnections(exa.index_pb.GetNumberConnectionsRequest) returns (exa.index_pb.GetNumberConnectionsResponse);
|
|
rpc GetConnectionsDebugInfo(exa.index_pb.GetConnectionsDebugInfoRequest) returns (exa.index_pb.GetConnectionsDebugInfoResponse);
|
|
}
|
|
service IndexService {
|
|
rpc GetIndexedRepositories(exa.index_pb.GetIndexedRepositoriesRequest) returns (exa.index_pb.GetIndexedRepositoriesResponse);
|
|
rpc GetNearestCCIsFromEmbedding(exa.index_pb.GetNearestCCIsFromEmbeddingRequest) returns (exa.index_pb.GetNearestCCIsFromEmbeddingResponse);
|
|
rpc GetEmbeddingsForCodeContextItems(exa.index_pb.GetEmbeddingsForCodeContextItemsRequest) returns (exa.index_pb.GetEmbeddingsForCodeContextItemsResponse);
|
|
rpc GetMatchingFilePaths(exa.index_pb.GetMatchingFilePathsRequest) returns (exa.index_pb.GetMatchingFilePathsResponse);
|
|
} |