Merge pull request #12 from singnet/senna-proto-11-1

[#11] Added rpc for get_importance()
singnet · Oct 10, 2024 · 8b455b2 · 8b455b2
2 parents ca45a52 + 5df12eb
commit 8b455b2
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 14 deletions.
diff --git a/attention_broker.proto b/attention_broker.proto
@@ -2,8 +2,27 @@ syntax = "proto3";
 import "common.proto";
 package dasproto;
 
+message HandleList {
+    repeated string list = 1;
+    int32 request_type = 2;
+    int64 hebbian_network = 3;
+    string context = 4;
+}
+
+message HandleCount {
+    map<string, uint32> map = 1;
+    int32 request_type = 2;
+    int64 hebbian_network = 3;
+    string context = 4;
+}
+
+message ImportanceList {
+    repeated float list = 1;
+}
+
 service AttentionBroker {
     rpc ping (Empty) returns (Ack) {}
     rpc stimulate (HandleCount) returns (Ack) {}
     rpc correlate (HandleList) returns (Ack) {}
+    rpc get_importance (HandleList) returns (ImportanceList) {}
 }
diff --git a/common.proto b/common.proto
@@ -7,17 +7,3 @@ message Ack {
     bool error = 1;
     string msg = 2;
 }
-
-message HandleList {
-    repeated string handle_list = 1;
-    int32 request_type = 2;
-    int64 hebbian_network = 3;
-    string context = 4;
-}
-
-message HandleCount {
-    map<string, uint32> handle_count = 1;
-    int32 request_type = 2;
-    int64 hebbian_network = 3;
-    string context = 4;
-}