koi-net 1.0.0b14__py3-none-any.whl → 1.0.0b16__py3-none-any.whl

koi_net/config.py

@@ -40,7 +40,7 @@ class EnvConfig(BaseModel):
             return env_val
         return value
 
-class Config(BaseModel):
+class NodeConfig(BaseModel):
     server: ServerConfig | None = Field(default_factory=ServerConfig)
     koi_net: KoiNetConfig
     _file_path: str = PrivateAttr(default="config.yaml")
@@ -49,7 +49,7 @@ class Config(BaseModel):
     @classmethod
     def load_from_yaml(
         cls, 
-        file_path: str | None = None, 
+        file_path: str = "config.yaml", 
         generate_missing: bool = True
     ):
         yaml = YAML()
@@ -62,8 +62,14 @@ class Config(BaseModel):
             config._file_content = file_content
             
         except FileNotFoundError:
+            # empty_fields = {}
+            # for name, field in cls.model_fields.items():
+                
+            #     if field.default is None or field.default_factory is None:
+            # print(empty_fields)
             config = cls()
             
+            
         config._file_path = file_path
         
         if generate_missing:            
@@ -92,4 +98,4 @@ class Config(BaseModel):
                     f.write(self._file_content)
                 raise e
                 
-ConfigType = TypeVar("ConfigType", bound=Config)
+ConfigType = TypeVar("ConfigType", bound=NodeConfig)

koi_net/identity.py

@@ -3,7 +3,7 @@ from rid_lib.ext.bundle import Bundle
 from rid_lib.ext.cache import Cache
 from rid_lib.types.koi_net_node import KoiNetNode
 
-from .config import Config
+from .config import NodeConfig
 from .protocol.node import NodeProfile
 
 logger = logging.getLogger(__name__)
@@ -12,12 +12,12 @@ logger = logging.getLogger(__name__)
 class NodeIdentity:
     """Represents a node's identity (RID, profile, bundle)."""
     
-    config: Config    
+    config: NodeConfig    
     cache: Cache
     
     def __init__(
         self,
-        config: Config,
+        config: NodeConfig,
         cache: Cache
     ):
         """Initializes node identity from a name and profile.

koi_net/network/interface.py

@@ -15,7 +15,7 @@ from ..protocol.node import NodeType
 from ..protocol.edge import EdgeType
 from ..protocol.event import Event
 from ..identity import NodeIdentity
-from ..config import Config, ConfigType
+from ..config import ConfigType
 
 logger = logging.getLogger(__name__)
 

koi_net/processor/default_handlers.py

@@ -9,6 +9,8 @@ from .handler import KnowledgeHandler, HandlerType, STOP_CHAIN
 from .knowledge_object import KnowledgeObject, KnowledgeSource
 from ..protocol.event import Event, EventType
 from ..protocol.edge import EdgeProfile, EdgeStatus, EdgeType
+from ..protocol.node import NodeProfile
+from ..protocol.helpers import generate_edge_bundle
 
 logger = logging.getLogger(__name__)
 
@@ -124,6 +126,57 @@ def edge_negotiation_handler(processor: ProcessorInterface, kobj: KnowledgeObjec
 
 # Network handlers
 
+@KnowledgeHandler.create(HandlerType.Network, rid_types=[KoiNetNode])
+def coordinator_contact(processor: ProcessorInterface, kobj: KnowledgeObject):
+    node_profile = kobj.bundle.validate_contents(NodeProfile)
+    
+    # looking for event provider of nodes
+    if KoiNetNode not in node_profile.provides.event:
+        return
+    
+    # already have an edge established
+    if processor.network.graph.get_edge_profile(
+        source=kobj.rid,
+        target=processor.identity.rid,
+    ) is not None:
+        return
+    
+    logger.info("Identified a coordinator!")
+    logger.info("Proposing new edge")
+    
+    if processor.identity.profile.node_type == NodeType.FULL:
+        edge_type = EdgeType.WEBHOOK
+    else:
+        edge_type = EdgeType.POLL
+    
+    # queued for processing
+    processor.handle(bundle=generate_edge_bundle(
+        source=kobj.rid,
+        target=processor.identity.rid,
+        edge_type=edge_type,
+        rid_types=[KoiNetNode]
+    ))
+    
+    logger.info("Catching up on network state")
+    
+    payload = processor.network.request_handler.fetch_rids(
+        node=kobj.rid, 
+        rid_types=[KoiNetNode]
+    )
+    for rid in payload.rids:
+        if rid == processor.identity.rid:
+            logger.info("Skipping myself")
+            continue
+        if processor.cache.exists(rid):
+            logger.info(f"Skipping known RID '{rid}'")
+            continue
+        
+        # marked as external since we are handling RIDs from another node
+        # will fetch remotely instead of checking local cache
+        processor.handle(rid=rid, source=KnowledgeSource.External)
+    logger.info("Done")
+    
+
 @KnowledgeHandler.create(HandlerType.Network)
 def basic_network_output_filter(processor: ProcessorInterface, kobj: KnowledgeObject):
     """Default network handler.

koi_net/processor/interface.py

@@ -9,7 +9,7 @@ from rid_lib.types.koi_net_node import KoiNetNode
 from ..identity import NodeIdentity
 from ..network import NetworkInterface
 from ..protocol.event import Event, EventType
-from ..config import Config
+from ..config import NodeConfig
 from .handler import (
     KnowledgeHandler, 
     HandlerType, 
@@ -28,7 +28,7 @@ logger = logging.getLogger(__name__)
 class ProcessorInterface():
     """Provides access to this node's knowledge processing pipeline."""
     
-    config: Config
+    config: NodeConfig
     cache: Cache
     network: NetworkInterface
     identity: NodeIdentity
@@ -39,7 +39,7 @@ class ProcessorInterface():
     
     def __init__(
         self,
-        config: Config,
+        config: NodeConfig,
         cache: Cache, 
         network: NetworkInterface,
         identity: NodeIdentity,

{koi_net-1.0.0b14.dist-info → koi_net-1.0.0b16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: koi-net
-Version: 1.0.0b14
+Version: 1.0.0b16
 Summary: Implementation of KOI-net protocol in Python
 Project-URL: Homepage, https://github.com/BlockScience/koi-net/
 Author-email: Luke Miller <luke@block.science>

{koi_net-1.0.0b14.dist-info → koi_net-1.0.0b16.dist-info}/RECORD

@@ -1,16 +1,16 @@
 koi_net/__init__.py,sha256=b0Ze0pZmJAuygpWUFHM6Kvqo3DkU_uzmkptv1EpAArw,31
-koi_net/config.py,sha256=4Lglj9mz9i4ratUXypdrLTuI1RHUOYeal9n8RKdVP7A,2911
+koi_net/config.py,sha256=35RNSOcWxnJzasPAkYabFWcKyDR-2kM1NTrq_fYOkm0,3153
 koi_net/core.py,sha256=IO8kqiNMYVeuNzilq7eHBA7IulsxRjrCbWnIAx6_abA,4406
-koi_net/identity.py,sha256=Ro6iGw9DMc4KrLJyqejtdh9l872nCoCGc5ioF9OptJc,1276
+koi_net/identity.py,sha256=muc5vuQ8zUOebhwAB3-ql6W2pgQETiYXXQAFBv8bLyg,1288
 koi_net/network/__init__.py,sha256=r_RN-q_mDYC-2RAkN-lJoMUX76TXyfEUc_MVKW87z0g,39
 koi_net/network/graph.py,sha256=dsfPuHUTkCzlj0QeL0e7dgp7-FR5_AGP7eE8EpBPhC0,4710
-koi_net/network/interface.py,sha256=DhMbVjKp9xdZReKldHWh72jkhd3RS_Q4XeNqg42mwyM,10474
+koi_net/network/interface.py,sha256=CKZACkUxG6Y4GV8XR16lhFLzRpXOw59hx7sFewZ_35w,10466
 koi_net/network/request_handler.py,sha256=66gjX2x4UnBWZYwKLjp_3WkhL-ekhR3VAyfGviHTcUs,4790
 koi_net/network/response_handler.py,sha256=CAwici2Etj9ESndERXdtYkMlc4gWHz_xc7jHgY2Qjcg,1830
 koi_net/processor/__init__.py,sha256=x4fAY0hvQEDcpfdTB3POIzxBQjYAtn0qQazPo1Xm0m4,41
-koi_net/processor/default_handlers.py,sha256=vfQNvucQR5zANyYhdLnc3kOXGFrgNtq5swd-vH7n0Uo,6810
+koi_net/processor/default_handlers.py,sha256=f8Yl21lLGMa_oW51bf8sBuunfP9_rmiSu32T-rZ4kUY,8542
 koi_net/processor/handler.py,sha256=7X6M6PP8m6-xdtsP1y4QO83g_MN5VSszNNikprITK80,2523
-koi_net/processor/interface.py,sha256=Ecy0XWW7KfuDBFt7fXEmag4CzHP768mtO4uRZWNZN4E,13000
+koi_net/processor/interface.py,sha256=3ofjBFz3FVjESABXJ1aC_4ESTuLFMHyl5d8uWz5QsFA,13012
 koi_net/processor/knowledge_object.py,sha256=RCgzkILsWm1Jw_NkSu4jTRYA9Ugga6mJ4jqKWwketQs,4090
 koi_net/protocol/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 koi_net/protocol/api_models.py,sha256=DYDKCRD2Uja633bBAyTsaxyb1oF9pX9yQ9NpNAbkczo,1070
@@ -19,7 +19,7 @@ koi_net/protocol/edge.py,sha256=CcmvIY4P1HEBdKNJ4wFRDmwYMRMss24Besmbi7ZRFxQ,427
 koi_net/protocol/event.py,sha256=HxzLN-iCXPyr2YzrswMIkgZYeUdFbBpa5v98dAB06lQ,1328
 koi_net/protocol/helpers.py,sha256=8ZkQrjb_G0QEaMIKe9wkFOBonl1bkmemx_pwKMwIiLg,695
 koi_net/protocol/node.py,sha256=2HhCh3LdBLlY2Z_kXNmKHzpVLKbP_ODob3HjHayFQtM,375
-koi_net-1.0.0b14.dist-info/METADATA,sha256=98NmBhai3c88c5UJf459u3GnyAdDTJ0jlzUwO3iVZsc,34203
-koi_net-1.0.0b14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-koi_net-1.0.0b14.dist-info/licenses/LICENSE,sha256=03mgCL5qth2aD9C3F3qNVs4sFJSpK9kjtYCyOwdSp7s,1069
-koi_net-1.0.0b14.dist-info/RECORD,,
+koi_net-1.0.0b16.dist-info/METADATA,sha256=YQ_kvSI7441neMoeU6uy5SMpt1Lo2ZwKx_KsIAbu0WE,34203
+koi_net-1.0.0b16.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+koi_net-1.0.0b16.dist-info/licenses/LICENSE,sha256=03mgCL5qth2aD9C3F3qNVs4sFJSpK9kjtYCyOwdSp7s,1069
+koi_net-1.0.0b16.dist-info/RECORD,,