gitlab-gkg-proto 0.0.1 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b8bbd58de583e48c78d701b9e4ee0eae518b8a520958f4102b68be0ebe1c784f
-  data.tar.gz: 2b9af543c4b80021690afa37255155d29243117643a50f4b2995a4a035f5ff58
+  metadata.gz: 71e2d2e969a2fff28ea9d1a6e5514c406c4b4e976d7ef0e2c4ddbae60ec3b2ed
+  data.tar.gz: 5b002ba4d2f024efe84597d8e021e97084dfa58a0f0d6705200a4531981b351c
 SHA512:
-  metadata.gz: 12858c09308498da5737df1ad9edce12e95dc0ea61bfb9a32dfa613fa3994d58498b2bda19c65a76489d242dbdce3344392574447ff53eae825334777ed0968a
-  data.tar.gz: 3f9a4d568d28dd96df5bc49246242f21e9b4d887fc69f9e947449df114f98fb15c52ddf8bd7548c436eb7c89800939d4338c6a547a464401bb04b4d326e4b511
+  metadata.gz: bec0502a0e7e51cd2c64883837340e9bb103fcacf774c511d2ff945eac71100a43ec4d967b2c905ac7d354051e461ae3573d30827dbe4c2dde356f34a8701d30
+  data.tar.gz: 9d05ceb0ceb09f1b18d80fd0f28c2a07e3a636c5911bd4dfea85c439c6200ae8e3f4acaf7ebfd252c38c6c4fdd257f90491bb2d0372554cdcf9ed6ba0f47379d

data/lib/gkg/v1/gkg_pb.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: gkg.proto
+
+require 'google/protobuf'
+
+
+descriptor_data = "\n\tgkg.proto\x12\x06gkg.v1\"\x12\n\x10ListToolsRequest\":\n\x11ListToolsResponse\x12%\n\x05tools\x18\x01 \x03(\x0b\x32\x16.gkg.v1.ToolDefinition\"S\n\x0eToolDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12\x1e\n\x16parameters_json_schema\x18\x03 \x01(\t\"&\n\x05\x45rror\x12\x0c\n\x04\x63ode\x18\x01 \x01(\t\x12\x0f\n\x07message\x18\x02 \x01(\t\"}\n\x11RedactionExchange\x12-\n\x08required\x18\x01 \x01(\x0b\x32\x19.gkg.v1.RedactionRequiredH\x00\x12-\n\x08response\x18\x02 \x01(\x0b\x32\x19.gkg.v1.RedactionResponseH\x00\x42\n\n\x08\x65xchange\"P\n\x11RedactionRequired\x12\x11\n\tresult_id\x18\x01 \x01(\t\x12(\n\tresources\x18\x02 \x03(\x0b\x32\x15.gkg.v1.ResourceCheck\"D\n\rResourceCheck\x12\x15\n\rresource_type\x18\x01 \x01(\t\x12\x0b\n\x03ids\x18\x02 \x03(\x03\x12\x0f\n\x07\x61\x62ility\x18\x03 \x01(\t\"]\n\x11RedactionResponse\x12\x11\n\tresult_id\x18\x01 \x01(\t\x12\x35\n\x0e\x61uthorizations\x18\x02 \x03(\x0b\x32\x1d.gkg.v1.ResourceAuthorization\"\xa4\x01\n\x15ResourceAuthorization\x12\x15\n\rresource_type\x18\x01 \x01(\t\x12\x41\n\nauthorized\x18\x02 \x03(\x0b\x32-.gkg.v1.ResourceAuthorization.AuthorizedEntry\x1a\x31\n\x0f\x41uthorizedEntry\x12\x0b\n\x03key\x18\x01 \x01(\x03\x12\r\n\x05value\x18\x02 \x01(\x08:\x02\x38\x01\"\xc4\x01\n\x12\x45xecuteToolMessage\x12-\n\x07request\x18\x01 \x01(\x0b\x32\x1a.gkg.v1.ExecuteToolRequestH\x00\x12.\n\tredaction\x18\x02 \x01(\x0b\x32\x19.gkg.v1.RedactionExchangeH\x00\x12$\n\x06result\x18\x03 \x01(\x0b\x32\x12.gkg.v1.ToolResultH\x00\x12\x1e\n\x05\x65rror\x18\x04 \x01(\x0b\x32\r.gkg.v1.ErrorH\x00\x42\t\n\x07message\"?\n\x12\x45xecuteToolRequest\x12\x11\n\ttool_name\x18\x01 \x01(\t\x12\x16\n\x0e\x61rguments_json\x18\x02 \x01(\t\"!\n\nToolResult\x12\x13\n\x0bresult_json\x18\x01 \x01(\t\"\xc7\x01\n\x13\x45xecuteQueryMessage\x12.\n\x07request\x18\x01 \x01(\x0b\x32\x1b.gkg.v1.ExecuteQueryRequestH\x00\x12.\n\tredaction\x18\x02 \x01(\x0b\x32\x19.gkg.v1.RedactionExchangeH\x00\x12%\n\x06result\x18\x03 \x01(\x0b\x32\x13.gkg.v1.QueryResultH\x00\x12\x1e\n\x05\x65rror\x18\x04 \x01(\x0b\x32\r.gkg.v1.ErrorH\x00\x42\t\n\x07message\")\n\x13\x45xecuteQueryRequest\x12\x12\n\nquery_json\x18\x01 \x01(\t\"9\n\x0bQueryResult\x12\x13\n\x0bresult_json\x18\x01 \x01(\t\x12\x15\n\rgenerated_sql\x18\x02 \x01(\t\"\x14\n\x12GetOntologyRequest\"\xa6\x01\n\x13GetOntologyResponse\x12\x16\n\x0eschema_version\x18\x01 \x01(\t\x12%\n\x05nodes\x18\x02 \x03(\x0b\x32\x16.gkg.v1.NodeDefinition\x12%\n\x05\x65\x64ges\x18\x03 \x03(\x0b\x32\x16.gkg.v1.EdgeDefinition\x12)\n\x07\x64omains\x18\x04 \x03(\x0b\x32\x18.gkg.v1.DomainDefinition\"I\n\x10\x44omainDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12\x12\n\nnode_names\x18\x03 \x03(\t\"\xbf\x01\n\x0eNodeDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0e\n\x06\x64omain\x18\x02 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12\x13\n\x0bprimary_key\x18\x04 \x01(\t\x12\x13\n\x0blabel_field\x18\x05 \x01(\t\x12.\n\nproperties\x18\x06 \x03(\x0b\x32\x1a.gkg.v1.PropertyDefinition\x12 \n\x05style\x18\x07 \x01(\x0b\x32\x11.gkg.v1.NodeStyle\"\\\n\x12PropertyDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x11\n\tdata_type\x18\x02 \x01(\t\x12\x10\n\x08nullable\x18\x03 \x01(\x08\x12\x13\n\x0b\x65num_values\x18\x04 \x03(\t\"(\n\tNodeStyle\x12\x0c\n\x04size\x18\x01 \x01(\x05\x12\r\n\x05\x63olor\x18\x02 \x01(\t\"Z\n\x0e\x45\x64geDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12%\n\x08variants\x18\x03 \x03(\x0b\x32\x13.gkg.v1.EdgeVariant\"7\n\x0b\x45\x64geVariant\x12\x13\n\x0bsource_type\x18\x01 \x01(\t\x12\x13\n\x0btarget_type\x18\x02 \x01(\t\"\x19\n\x17GetClusterHealthRequest\"\x92\x01\n\x18GetClusterHealthResponse\x12%\n\x06status\x18\x01 \x01(\x0e\x32\x15.gkg.v1.ClusterStatus\x12\x11\n\ttimestamp\x18\x02 \x01(\t\x12\x0f\n\x07version\x18\x03 \x01(\t\x12+\n\ncomponents\x18\x04 \x03(\x0b\x32\x17.gkg.v1.ComponentHealth\"\xd6\x01\n\x0f\x43omponentHealth\x12\x0c\n\x04name\x18\x01 \x01(\t\x12%\n\x06status\x18\x02 \x01(\x0e\x32\x15.gkg.v1.ClusterStatus\x12\'\n\x08replicas\x18\x03 \x01(\x0b\x32\x15.gkg.v1.ReplicaStatus\x12\x35\n\x07metrics\x18\x04 \x03(\x0b\x32$.gkg.v1.ComponentHealth.MetricsEntry\x1a.\n\x0cMetricsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"/\n\rReplicaStatus\x12\r\n\x05ready\x18\x01 \x01(\x05\x12\x0f\n\x07\x64\x65sired\x18\x02 \x01(\x05*\x86\x01\n\rClusterStatus\x12\x1e\n\x1a\x43LUSTER_STATUS_UNSPECIFIED\x10\x00\x12\x1a\n\x16\x43LUSTER_STATUS_HEALTHY\x10\x01\x12\x1b\n\x17\x43LUSTER_STATUS_DEGRADED\x10\x02\x12\x1c\n\x18\x43LUSTER_STATUS_UNHEALTHY\x10\x03\x32\x91\x03\n\x15KnowledgeGraphService\x12@\n\tListTools\x12\x18.gkg.v1.ListToolsRequest\x1a\x19.gkg.v1.ListToolsResponse\x12I\n\x0b\x45xecuteTool\x12\x1a.gkg.v1.ExecuteToolMessage\x1a\x1a.gkg.v1.ExecuteToolMessage(\x01\x30\x01\x12L\n\x0c\x45xecuteQuery\x12\x1b.gkg.v1.ExecuteQueryMessage\x1a\x1b.gkg.v1.ExecuteQueryMessage(\x01\x30\x01\x12\x46\n\x0bGetOntology\x12\x1a.gkg.v1.GetOntologyRequest\x1a\x1b.gkg.v1.GetOntologyResponse\x12U\n\x10GetClusterHealth\x12\x1f.gkg.v1.GetClusterHealthRequest\x1a .gkg.v1.GetClusterHealthResponseb\x06proto3"
+
+pool = Google::Protobuf::DescriptorPool.generated_pool
+pool.add_serialized_file(descriptor_data)
+
+module Gkg
+  module V1
+    ListToolsRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.ListToolsRequest").msgclass
+    ListToolsResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.ListToolsResponse").msgclass
+    ToolDefinition = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.ToolDefinition").msgclass
+    Error = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.Error").msgclass
+    RedactionExchange = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.RedactionExchange").msgclass
+    RedactionRequired = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.RedactionRequired").msgclass
+    ResourceCheck = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.ResourceCheck").msgclass
+    RedactionResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.RedactionResponse").msgclass
+    ResourceAuthorization = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.ResourceAuthorization").msgclass
+    ExecuteToolMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.ExecuteToolMessage").msgclass
+    ExecuteToolRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.ExecuteToolRequest").msgclass
+    ToolResult = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.ToolResult").msgclass
+    ExecuteQueryMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.ExecuteQueryMessage").msgclass
+    ExecuteQueryRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.ExecuteQueryRequest").msgclass
+    QueryResult = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.QueryResult").msgclass
+    GetOntologyRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.GetOntologyRequest").msgclass
+    GetOntologyResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.GetOntologyResponse").msgclass
+    DomainDefinition = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.DomainDefinition").msgclass
+    NodeDefinition = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.NodeDefinition").msgclass
+    PropertyDefinition = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.PropertyDefinition").msgclass
+    NodeStyle = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.NodeStyle").msgclass
+    EdgeDefinition = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.EdgeDefinition").msgclass
+    EdgeVariant = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.EdgeVariant").msgclass
+    GetClusterHealthRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.GetClusterHealthRequest").msgclass
+    GetClusterHealthResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.GetClusterHealthResponse").msgclass
+    ComponentHealth = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.ComponentHealth").msgclass
+    ReplicaStatus = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.ReplicaStatus").msgclass
+    ClusterStatus = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gkg.v1.ClusterStatus").enummodule
+  end
+end

data/lib/gkg/v1/gkg_services_pb.rb

@@ -0,0 +1,44 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# Source: gkg.proto for package 'gkg.v1'
+# Original file comments:
+# Knowledge Graph gRPC service definition
+# Supports bidirectional streaming for tool execution with redaction flow
+#
+
+require 'grpc'
+require 'gkg_pb'
+
+module Gkg
+  module V1
+    module KnowledgeGraphService
+      # Knowledge Graph service for tool execution and raw queries
+      class Service
+
+        include ::GRPC::GenericService
+
+        self.marshal_class_method = :encode
+        self.unmarshal_class_method = :decode
+        self.service_name = 'gkg.v1.KnowledgeGraphService'
+
+        # List all available tools with their schemas
+        rpc :ListTools, ::Gkg::V1::ListToolsRequest, ::Gkg::V1::ListToolsResponse
+        # Execute a tool with bidirectional streaming for redaction:
+        # 1. Client sends ExecuteToolRequest
+        # 2. Server sends RedactionExchange.required with resources to check
+        # 3. Client sends RedactionExchange.response with authorized resources
+        # 4. Server sends ToolResult with redacted, context-engineered response
+        rpc :ExecuteTool, stream(::Gkg::V1::ExecuteToolMessage), stream(::Gkg::V1::ExecuteToolMessage)
+        # Execute a raw query with bidirectional streaming for redaction:
+        # Same redaction flow as ExecuteTool but returns raw filtered result
+        # (no context-engine processing) for the query playground
+        rpc :ExecuteQuery, stream(::Gkg::V1::ExecuteQueryMessage), stream(::Gkg::V1::ExecuteQueryMessage)
+        # Returns the complete ontology schema (nodes, edges, domains)
+        rpc :GetOntology, ::Gkg::V1::GetOntologyRequest, ::Gkg::V1::GetOntologyResponse
+        # Returns cluster health status for the Orbit dashboard
+        rpc :GetClusterHealth, ::Gkg::V1::GetClusterHealthRequest, ::Gkg::V1::GetClusterHealthResponse
+      end
+
+      Stub = Service.rpc_stub_class
+    end
+  end
+end

data/lib/gkg/version.rb

@@ -0,0 +1,3 @@
+module Gkg
+  VERSION = '0.4.0'
+end

data/lib/gkg.rb

@@ -0,0 +1,2 @@
+require_relative 'gkg/version'
+require_relative 'gkg/v1/gkg_services_pb'

metadata

@@ -1,11 +1,11 @@
 --- !ruby/object:Gem::Specification
 name: gitlab-gkg-proto
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.4.0
 platform: ruby
 authors:
 - GitLab
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
 date: 2026-02-22 00:00:00.000000000 Z
@@ -24,23 +24,24 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1.0'
-description: Placeholder gem to reserve the name. Will contain auto-generated protobuf
-  and gRPC stubs for the GitLab Knowledge Graph (GKG) service.
+description:
 email:
 - engineering@gitlab.com
 executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- lib/gitlab-gkg-proto.rb
-- lib/gitlab/gkg/proto/version.rb
+- lib/gkg.rb
+- lib/gkg/v1/gkg_pb.rb
+- lib/gkg/v1/gkg_services_pb.rb
+- lib/gkg/version.rb
 homepage: https://gitlab.com/gitlab-org/orbit/knowledge-graph
 licenses:
 - MIT
 metadata:
   allowed_push_host: https://rubygems.org
   source_code_uri: https://gitlab.com/gitlab-org/orbit/knowledge-graph
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -55,8 +56,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3.1
-signing_key: 
+rubygems_version: 3.5.22
+signing_key:
 specification_version: 4
 summary: Auto-generated gRPC client for the GitLab Knowledge Graph service
 test_files: []

data/lib/gitlab/gkg/proto/version.rb

@@ -1,7 +0,0 @@
-module Gitlab
-  module Gkg
-    module Proto
-      VERSION = '0.0.1'
-    end
-  end
-end

data/lib/gitlab-gkg-proto.rb

@@ -1 +0,0 @@
-require_relative 'gitlab/gkg/proto/version'