odin-engine 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

retrieval/writers/__init__.py

@@ -1,9 +1,9 @@
-from .base import PersistenceWriter
-from .arango_writer import ArangoWriter
-from .janus_writer import JanusGraphWriter
-
-__all__ = [
-    'PersistenceWriter',
-    'ArangoWriter',
-    'JanusGraphWriter',
-]
+from .base import PersistenceWriter
+from .arango_writer import ArangoWriter
+from .janus_writer import JanusGraphWriter
+
+__all__ = [
+    'PersistenceWriter',
+    'ArangoWriter',
+    'JanusGraphWriter',
+]

retrieval/writers/arango_writer.py

@@ -1,28 +1,28 @@
-from __future__ import annotations
-from typing import Dict, Any, Optional
-
-from .base import PersistenceWriter
-
-class ArangoWriter(PersistenceWriter):
-    """
-    Implementation of the PersistenceWriter protocol for ArangoDB.
-    """
-    def __init__(self, arango_client, database: str, persist_threshold: float = 0.8):
-        self.client = arango_client
-        self.db = self.client.db(database)
-        self.persist_threshold = persist_threshold
-
-    def maybe_write_link(self, src_entity: str, rel: str, dst_entity: str, confidence: float, metadata: Optional[Dict[str, Any]] = None) -> bool:
-        if confidence < self.persist_threshold:
-            return False
-        
-        edges = self.db.collection('kg_edges')
-        doc = {
-            '_from': f'entities/{src_entity}',
-            '_to': f'entities/{dst_entity}',
-            'relation': rel,
-            'confidence': float(confidence),
-            'metadata': metadata or {},
-        }
-        edges.insert(doc)
-        return True
+from __future__ import annotations
+from typing import Dict, Any, Optional
+
+from .base import PersistenceWriter
+
+class ArangoWriter(PersistenceWriter):
+    """
+    Implementation of the PersistenceWriter protocol for ArangoDB.
+    """
+    def __init__(self, arango_client, database: str, persist_threshold: float = 0.8):
+        self.client = arango_client
+        self.db = self.client.db(database)
+        self.persist_threshold = persist_threshold
+
+    def maybe_write_link(self, src_entity: str, rel: str, dst_entity: str, confidence: float, metadata: Optional[Dict[str, Any]] = None) -> bool:
+        if confidence < self.persist_threshold:
+            return False
+        
+        edges = self.db.collection('kg_edges')
+        doc = {
+            '_from': f'entities/{src_entity}',
+            '_to': f'entities/{dst_entity}',
+            'relation': rel,
+            'confidence': float(confidence),
+            'metadata': metadata or {},
+        }
+        edges.insert(doc)
+        return True

retrieval/writers/base.py

@@ -1,21 +1,21 @@
-from __future__ import annotations
-from typing import Protocol, Dict, Any, Optional
-
-class PersistenceWriter(Protocol):
-    """
-    Protocol for a generic persistence writer.
-    Defines the interface for writing links and relations to a database.
-    """
-    def maybe_write_link(
-        self, 
-        src_entity: str, 
-        rel: str, 
-        dst_entity: str, 
-        confidence: float, 
-        metadata: Optional[Dict[str, Any]] = None
-    ) -> bool:
-        """
-        Writes a link to the database if the confidence meets the threshold.
-        Returns True if the link was written, False otherwise.
-        """
-        ...
+from __future__ import annotations
+from typing import Protocol, Dict, Any, Optional
+
+class PersistenceWriter(Protocol):
+    """
+    Protocol for a generic persistence writer.
+    Defines the interface for writing links and relations to a database.
+    """
+    def maybe_write_link(
+        self, 
+        src_entity: str, 
+        rel: str, 
+        dst_entity: str, 
+        confidence: float, 
+        metadata: Optional[Dict[str, Any]] = None
+    ) -> bool:
+        """
+        Writes a link to the database if the confidence meets the threshold.
+        Returns True if the link was written, False otherwise.
+        """
+        ...

retrieval/writers/janus_writer.py

@@ -1,36 +1,36 @@
-from __future__ import annotations
-from typing import Dict, Any, Optional
-from gremlin_python.process.graph_traversal import __
-from gremlin_python.structure.graph import Graph
-
-from .base import PersistenceWriter
-
-class JanusGraphWriter(PersistenceWriter):
-    """
-    Implementation of the PersistenceWriter protocol for JanusGraph.
-    Assumes connection to a JanusGraph instance via a Gremlin client.
-    """
-    def __init__(self, graph: Graph, persist_threshold: float = 0.8):
-        self.g = graph.traversal()
-        self.persist_threshold = persist_threshold
-
-    def maybe_write_link(self, src_entity: str, rel: str, dst_entity: str, confidence: float, metadata: Optional[Dict[str, Any]] = None) -> bool:
-        if confidence < self.persist_threshold:
-            return False
-        
-        # Conceptual Gremlin query to add an edge
-        # Note: This is a simplified example. You may need to handle vertex existence checks,
-        # property updates, and transaction management in a production setting.
-        try:
-            (self.g.V(src_entity).as_('a')
-             .V(dst_entity).as_('b')
-             .addE(rel)
-             .from_('a').to('b')
-             .property('confidence', confidence)
-             # Add any other metadata as properties
-             .iterate())
-            return True
-        except Exception as e:
-            # In a real implementation, you would log this error
-            print(f"Failed to write link to JanusGraph: {e}")
-            return False
+from __future__ import annotations
+from typing import Dict, Any, Optional
+from gremlin_python.process.graph_traversal import __
+from gremlin_python.structure.graph import Graph
+
+from .base import PersistenceWriter
+
+class JanusGraphWriter(PersistenceWriter):
+    """
+    Implementation of the PersistenceWriter protocol for JanusGraph.
+    Assumes connection to a JanusGraph instance via a Gremlin client.
+    """
+    def __init__(self, graph: Graph, persist_threshold: float = 0.8):
+        self.g = graph.traversal()
+        self.persist_threshold = persist_threshold
+
+    def maybe_write_link(self, src_entity: str, rel: str, dst_entity: str, confidence: float, metadata: Optional[Dict[str, Any]] = None) -> bool:
+        if confidence < self.persist_threshold:
+            return False
+        
+        # Conceptual Gremlin query to add an edge
+        # Note: This is a simplified example. You may need to handle vertex existence checks,
+        # property updates, and transaction management in a production setting.
+        try:
+            (self.g.V(src_entity).as_('a')
+             .V(dst_entity).as_('b')
+             .addE(rel)
+             .from_('a').to('b')
+             .property('confidence', confidence)
+             # Add any other metadata as properties
+             .iterate())
+            return True
+        except Exception as e:
+            # In a real implementation, you would log this error
+            print(f"Failed to write link to JanusGraph: {e}")
+            return False

odin_engine-0.1.0.dist-info/RECORD

@@ -1,62 +0,0 @@
-benchmarks/__init__.py,sha256=aSfiClviEKA-lk2R3A4mxJ4RExLERWbsSwOeZ07jw6I,501
-benchmarks/datasets.py,sha256=t-9Ci_qUdvUP_yeaOKUk0ZagsGJUDIFQhK_N-riA0b4,8455
-benchmarks/metrics.py,sha256=vTpudmq7wo-IWiyZ-wLOMIO8gyO2pd76V9cvUs53lxQ,8026
-benchmarks/run_ablation.py,sha256=DINZFBVXXbda0dM_kC7t3LmN0NWVCBnGxo8WJji4eFo,8763
-benchmarks/run_npll_benchmark.py,sha256=NoQo-rq0U4yp-mf-R3bSnqEPPpiYOkQHdCFUwEu34bg,8016
-npll/__init__.py,sha256=PR9xpeKkRiP8PDhAyW7edNm6J9xdFaUiBkWNvedHvsk,241
-npll/bootstrap.py,sha256=ciBBuiKPGKl0pu9gp7GNZLaAsFZJGsPYz3AYDZ-7pv0,18849
-npll/npll_model.py,sha256=YWW1meVusR7Kwe25NNBprTtM_L3XopekbnfDk5fBjhM,27171
-npll/core/__init__.py,sha256=P0V9CEKyuEhumNkBQfBG4aUlnsmzUakvPnc8ujWiCFg,865
-npll/core/knowledge_graph.py,sha256=MOUBvLQGLb79g-mnNJcQncEw8Nh_eH0Slt-Aj2KLllg,11491
-npll/core/logical_rules.py,sha256=Pc3-zNxB0q0LAhDBWV_O_ZTuT3hN11tNbJUapyz1uCM,18267
-npll/core/mln.py,sha256=j5_LP5pufV4fXA4Vw1M0MSejVZCXyV8nv6gUO664uWY,19059
-npll/inference/__init__.py,sha256=ybppVpzmMABSFjk74818TietK1qc-ps-nmmuFxSGDDI,1079
-npll/inference/e_step.py,sha256=nqRtL82KQbR5hQ0PzVN0Tjl9VM1rcLD_wut55v5t4Nk,17348
-npll/inference/elbo.py,sha256=hr-kxQho44dSNmBdoBIvNyhQ5G3Ic4YSjpliukJGBjY,17320
-npll/inference/m_step.py,sha256=XafLUEHhcn8B-D0pbv4ur1ITPSRHmOQ3liXGy74LPVY,23306
-npll/scoring/__init__.py,sha256=MqtlaCj9iJA31Zo6iGuZsbAprU6PpIcOn2ugQAjIQew,1338
-npll/scoring/embeddings.py,sha256=mtbvFDb4A8RhjoaAC_iFGIRP_UpBmGDxcGBecfhF4Eo,19919
-npll/scoring/probability.py,sha256=_k-Dpus6cZ9MgFbKQXwvJVk7z0y6suWBjwzmkLxgxD0,17201
-npll/scoring/scoring_module.py,sha256=NNHnV0777c3JbNb_HD8fagqqxBB6pC1lmgOOKAU6AMI,14219
-npll/training/__init__.py,sha256=dg9BFwlNbhENC51AKEq_Xd2W2PWEJp_DsVKHPIwmJ2k,665
-npll/training/evaluation.py,sha256=AKWstFidiAZgM436pkZN3ejpwZVLTs1HoGRDy3bbUVw,17651
-npll/training/npll_trainer.py,sha256=jVVSzfZGvvn2BPVF-Xm7Qb7-853lRKRMG8gujuQnf1Q,18400
-npll/utils/__init__.py,sha256=DETz3sU0Vl1M3JCtZDSyDyf4TF2-eg3Q-rxA2_9cmj4,1384
-npll/utils/batch_utils.py,sha256=arhPJJQMmM9LZwWOXRYz5yOjeqirMJ5_1ikod4I4FTc,18990
-npll/utils/config.py,sha256=uLT8f6-lW84SriQKJhAgn8XgBYX2tnu2_DwWghngEk4,5278
-npll/utils/math_utils.py,sha256=O126BegQUCxthlfSrwigy7PAhYi-47gtA6jQcwCI0WA,11736
-odin/__init__.py,sha256=zcAH0eVte4K9waL-ncShb3wCtLyg5xvAAVk-rcMnbGI,587
-odin/engine.py,sha256=iExs4yfkQuJ2TSXR9tgtcu2y7LE7sjD5StqHNEnSgx4,9366
-odin_engine-0.1.0.dist-info/licenses/LICENSE,sha256=qpoLOuIV3ib39skSw-G12n-yAz9XSZkkHqU1zwN79fs,1091
-retrieval/__init__.py,sha256=_R7G42rukMXgu-z-lriAB7oG5vEOCITxNKPYPR3QYEI,2334
-retrieval/adapters.py,sha256=Ge1iVjOSuyQCPmdxEMqbHvWFUKooTn_YCg2RF7ehSBk,5161
-retrieval/adapters_arango.py,sha256=96QKq-PmJMma4D2jY30mmZkRsXY3yrqpu0JzbMYFcu8,55182
-retrieval/aggregators.py,sha256=3_l22YMwGk_-2VPhAgnwUcBAt4EY2Nh7E4v2KnHmpdM,24641
-retrieval/beam.py,sha256=-bLw4icAK21zVS6ViiOh4CwunbGDwFcgYQY5_9fuvUU,4983
-retrieval/budget.py,sha256=hezUkGDWUBVZb3uRF71c3zzP2nK3pICGUCE3pMYb70c,1534
-retrieval/cache.py,sha256=2TPZ5WEM3JO5LjnGouYPvAv1amTQdda41POIOAKuQnA,6266
-retrieval/confidence.py,sha256=QvfX-fR1rp3AFPZEMtgT6nc_01v-L5glgdjgIHKTafo,3360
-retrieval/eval.py,sha256=sPxxsNRIrH0IiRlwykIw0Ltt0qLFL3WDZVRNEKWJyW8,1532
-retrieval/linker.py,sha256=uE7IdOglkxpEH3mWeZIpgbGvl3nz89DGpfhWKXAzduU,3297
-retrieval/metrics.py,sha256=97aq5EYkBVZakWqjUy0ejtlhQx1uKl_BJXyJeRRSekE,3393
-retrieval/metrics_motifs.py,sha256=1i7Uz44JMaCVQAWviBgh45Pb0W6qWAP5oGkp2YsPDwU,1382
-retrieval/orchestrator.py,sha256=TLoUFgpiOdTlrsArh-53EMBVXoX4s-_6Rd84SZIGZY4,23064
-retrieval/ppr.py,sha256=1pi3jVxQOT2-0EPWu45139aZHNubb8u6jAp-VIZqSpM,5532
-retrieval/ppr_cache.py,sha256=JXZwfNmCqLghpfhOqufKNe7WOQn4yLFkQpV3Kechf_I,785
-retrieval/scoring.py,sha256=KUV7-vR4uhCoWvhPpsabK6FkaIHL55Ii8orr_OyqkeU,11316
-retrieval/ppr/__init__.py,sha256=rHZe08Pow9K82vslLtBB-hCrrd7YsQqiA-QmeqSEh7c,465
-retrieval/ppr/anchors.py,sha256=N4k-E-HPGV7J5GFqbbofkz8zF7jjaXoLts5oFjbwKaQ,1262
-retrieval/ppr/bippr.py,sha256=iIdCXzCBqsU87uu5MZnJDWvvCs6Y1uanmT2uaNT9QMg,2236
-retrieval/ppr/engines.py,sha256=R9BKMIB9pvDPsSoTwNymjPPY4vO1jQW0vpSP09YwByc,10389
-retrieval/ppr/global_pr.py,sha256=GpaZdJnwo-XuBDX363qoKXZ6WD2dnOg_8LlJHZUpXnQ,2533
-retrieval/ppr/indexes.py,sha256=3sbfpOHgiyPbb3A7D4UwNS2yrYAslzQQy960sDYxDLg,2558
-retrieval/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-retrieval/utils/pii_redaction.py,sha256=gfNK3Vd281P19CmMpXaiiOpFygMA_Vr2iMXxOtxerNs,1260
-retrieval/writers/__init__.py,sha256=Qohu0qUUSLV9scSoUS_ut98prFX1G9Xzm-vslf6wTUQ,212
-retrieval/writers/arango_writer.py,sha256=sv5FOc7y63eMyogXLLjKpApUVbRFjvIQhn8Vu7sx0U4,1032
-retrieval/writers/base.py,sha256=sE0Z6yKYtYhOWOe9XfC9KjkyJ0dfZSY-jgNvmqxgupQ,658
-retrieval/writers/janus_writer.py,sha256=SK4e1C9icKsRVUo_FpDF6LfnaftQU4o-pmwJf6_cBEY,1541
-odin_engine-0.1.0.dist-info/METADATA,sha256=x5e-9_Pxa83cPJ_lYkbE6mJLNiiUQlpwKcVFpR9JkAI,16098
-odin_engine-0.1.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-odin_engine-0.1.0.dist-info/top_level.txt,sha256=U53RwIYgQIJ0DBf_OaThVdHosqy06lkB4Zz4crmFadQ,31
-odin_engine-0.1.0.dist-info/RECORD,,