naas-abi 1.0.7__py3-none-any.whl → 1.0.9__py3-none-any.whl

naas_abi/pipelines/RemoveIndividualPipeline_test.py

@@ -1,18 +1,20 @@
 import pytest
-from naas_abi import services
+from naas_abi import ABIModule
 from naas_abi.pipelines.RemoveIndividualPipeline import (
     RemoveIndividualPipeline,
     RemoveIndividualPipelineConfiguration,
     RemoveIndividualPipelineParameters,
 )
+from naas_abi_core.utils.SPARQL import SPARQLUtils
+
+triple_store_service = ABIModule.get_instance().engine.services.triple_store
+sparql_utils = SPARQLUtils(triple_store_service)
 
 
 @pytest.fixture
 def pipeline() -> RemoveIndividualPipeline:
     return RemoveIndividualPipeline(
-        RemoveIndividualPipelineConfiguration(
-            triple_store=services.triple_store_service
-        )
+        RemoveIndividualPipelineConfiguration(triple_store=triple_store_service)
     )
 
 
@@ -20,8 +22,6 @@ def test_remove_individual_pipeline(pipeline: RemoveIndividualPipeline):
     import time
     from uuid import uuid4
 
-    from naas_abi import services
-    from naas_abi_core.utils.SPARQL import get_subject_graph
     from rdflib import OWL, RDF, RDFS, Graph, Literal, Namespace, URIRef
 
     ABI = Namespace("http://ontology.naas.ai/abi/")
@@ -42,7 +42,7 @@ def test_remove_individual_pipeline(pipeline: RemoveIndividualPipeline):
             ),
         )
     )
-    services.triple_store_service.insert(graph)
+    triple_store_service.insert(graph)
     time.sleep(3)
 
     # Run pipeline to remove triples
@@ -54,5 +54,5 @@ def test_remove_individual_pipeline(pipeline: RemoveIndividualPipeline):
     assert graph is not None, graph.serialize(format="turtle")
 
     # Check if uri is removed in triplestore
-    graph = get_subject_graph(str(uri), 1)
+    graph = sparql_utils.get_subject_graph(str(uri), 1)
     assert len(graph) == 0, graph.serialize(format="turtle")

naas_abi/pipelines/UpdateDataPropertyPipeline.py

@@ -152,7 +152,11 @@ class UpdateDataPropertyPipeline(Pipeline):
 
 
 if __name__ == "__main__":
-    from naas_abi import services
+    from naas_abi_core.engine.Engine import Engine
+
+    engine = Engine()
+    engine.load(module_names=["naas_abi"])
+    triple_store_service = engine.services.triple_store
 
     subject_uri = "http://ontology.naas.ai/abi/1b765700-9fa1-4e1d-9496-108445aafb34"
     predicate_uri = str(RDFS.label)
@@ -160,7 +164,7 @@ if __name__ == "__main__":
     language = "en"
 
     configuration = UpdateDataPropertyPipelineConfiguration(
-        triple_store=services.triple_store_service
+        triple_store=triple_store_service
     )
 
     pipeline = UpdateDataPropertyPipeline(configuration)

naas_abi/workflows/GetObjectPropertiesFromClassWorkflow.py

@@ -40,7 +40,6 @@ class GetObjectPropertiesFromClassWorkflowParameters(WorkflowParameters):
         Field(
             ...,
             description="URI of the class to get object properties for",
-            example="http://purl.obolibrary.org/obo/BFO_0000040",
             pattern="^http.*",
         ),
     ]
@@ -237,6 +236,7 @@ class GetObjectPropertiesFromClassWorkflow(Workflow):
         """Check if the given URI is a class in the ontology."""
         uri_ref = URIRef(uri)
         # Check if it's declared as an OWL Class
+        print(self.graph.serialize(format="turtle"))
         return (uri_ref, RDF.type, OWL.Class) in self.graph
 
     def _class_matches_domain(self, class_uri, domains):
@@ -371,11 +371,18 @@ class GetObjectPropertiesFromClassWorkflow(Workflow):
 
 
 if __name__ == "__main__":
-    from naas_abi_core import logger, services
+    from naas_abi_core import logger
+    from naas_abi import ABIModule
+    from naas_abi_core.engine.Engine import Engine
+
+    engine = Engine()
+    engine.load(module_names=[
+        "naas_abi"
+    ])
 
     workflow = GetObjectPropertiesFromClassWorkflow(
         GetObjectPropertiesFromClassWorkflowConfiguration(
-            triple_store=services.triple_store_service
+            triple_store=ABIModule.get_instance().engine.services.triple_store
         )
     )
     class_uri = "http://purl.obolibrary.org/obo/BFO_0000015"

naas_abi/workflows/GetObjectPropertiesFromClassWorkflow_test.py

@@ -1,17 +1,22 @@
 import pytest
+from naas_abi import ABIModule
 from naas_abi.workflows.GetObjectPropertiesFromClassWorkflow import (
     GetObjectPropertiesFromClassWorkflow,
     GetObjectPropertiesFromClassWorkflowConfiguration,
     GetObjectPropertiesFromClassWorkflowParameters,
 )
-from naas_abi_core import logger, services
+from naas_abi_core import logger
+from naas_abi_core.utils.SPARQL import SPARQLUtils
+
+triple_store_service = ABIModule.get_instance().engine.services.triple_store
+sparql_utils = SPARQLUtils(triple_store_service)
 
 
 @pytest.fixture
 def workflow() -> GetObjectPropertiesFromClassWorkflow:
     return GetObjectPropertiesFromClassWorkflow(
         GetObjectPropertiesFromClassWorkflowConfiguration(
-            triple_store=services.triple_store_service
+            triple_store=triple_store_service
         )
     )
 

naas_abi/workflows/GetSubjectGraphWorkflow_test.py

@@ -1,9 +1,14 @@
 import pytest
+from naas_abi import ABIModule
 from naas_abi.workflows.GetSubjectGraphWorkflow import (
     GetSubjectGraphWorkflow,
     GetSubjectGraphWorkflowConfiguration,
     GetSubjectGraphWorkflowParameters,
 )
+from naas_abi_core.utils.SPARQL import SPARQLUtils
+
+triple_store_service = ABIModule.get_instance().engine.services.triple_store
+sparql_utils = SPARQLUtils(triple_store_service)
 
 
 @pytest.fixture
@@ -24,7 +29,7 @@ def test_get_subject_graph_workflow(
 ):
     from uuid import uuid4
 
-    from naas_abi_core import logger, services
+    from naas_abi_core import logger
     from naas_abi_core.utils.Graph import TEST
     from rdflib import OWL, RDFS, Graph, Literal, URIRef
 
@@ -48,7 +53,7 @@ def test_get_subject_graph_workflow(
     )
     graph.add((uri, RDFS.label, Literal(node_id)))
 
-    services.triple_store_service.insert(graph)
+    triple_store_service.insert(graph)
 
     result = get_subject_graph_workflow.get_subject_graph(
         GetSubjectGraphWorkflowParameters(
@@ -59,7 +64,7 @@ def test_get_subject_graph_workflow(
     assert isinstance(result, str), result
     assert result != "", result
 
-    services.triple_store_service.remove(graph)
+    triple_store_service.remove(graph)
 
     # Test
     result = get_subject_graph_workflow.get_subject_graph(

naas_abi/workflows/SearchIndividualWorkflow_test.py

@@ -1,18 +1,19 @@
 import pytest
+from naas_abi import ABIModule
 from naas_abi.workflows.SearchIndividualWorkflow import (
     SearchIndividualWorkflow,
     SearchIndividualWorkflowConfiguration,
     SearchIndividualWorkflowParameters,
 )
 
+triple_store_service = ABIModule.get_instance().engine.services.triple_store
+
 
 @pytest.fixture
 def search_individual_workflow() -> SearchIndividualWorkflow:
-    from naas_abi import services
-
     # Configurations
     search_individual_workflow_configuration = SearchIndividualWorkflowConfiguration(
-        triple_store=services.triple_store_service
+        triple_store=triple_store_service
     )
 
     # Workflow
@@ -28,7 +29,6 @@ def test_search_individual_workflow(
 ):
     from uuid import uuid4
 
-    from naas_abi import services
     from naas_abi_core.utils.Graph import TEST
     from rdflib import OWL, RDFS, Graph, Literal, URIRef
 
@@ -50,7 +50,7 @@ def test_search_individual_workflow(
     )
     graph.add((TEST[node_id], RDFS.label, Literal(node_id)))
 
-    services.triple_store_service.insert(graph)
+    triple_store_service.insert(graph)
 
     # Test
     result = search_individual_workflow.search_individual(
@@ -66,7 +66,7 @@ def test_search_individual_workflow(
     assert len(result) == 1, result
     assert result[0]["label"] == node_id, result[0]
 
-    services.triple_store_service.remove(graph)
+    triple_store_service.remove(graph)
 
     # Test
     result = search_individual_workflow.search_individual(

{naas_abi-1.0.7.dist-info → naas_abi-1.0.9.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: naas-abi
-Version: 1.0.7
-Summary: Add your description here
+Version: 1.0.9
+Summary: Multi-agent orchestrator and knowledge graph management system for AI orchestration, providing comprehensive coordination of specialized AI agents and semantic data management capabilities
 Author-email: Maxime Jublou <maxime@naas.ai>, Florent Ravenel <florent@naas.ai>, Jeremy Ravenel <jeremy@naas.ai>
 Requires-Python: <4,>=3.10
 Description-Content-Type: text/markdown

naas_abi-1.0.9.dist-info/RECORD

@@ -0,0 +1,49 @@
+naas_abi/__init__.py,sha256=y0syv4DSaZv87UaJEfNjN4VBvAF7aXqm_Y9EoVpyJAk,1538
+naas_abi/cli.py,sha256=bfCcxGoQ_iSMhQDDX0-dRCyhc53Nawsy3WAruCNCL7M,27549
+naas_abi/agents/AbiAgent.py,sha256=aTXDPJtJiIfl8R8U37XF3cBLuO2xVBNOr3ITfwXwvVM,18141
+naas_abi/agents/AbiAgent_test.py,sha256=AKggG5RS3zDWnql1DFE7cF7riAHeOdKTZp44WN65Arg,5906
+naas_abi/agents/EntitytoSPARQLAgent.py,sha256=WO77rMoHF8Iq6jqDWEsvDHz-tup6SHeDIBLiy5wvGe8,57173
+naas_abi/agents/EntitytoSPARQLAgent_test.py,sha256=kx0gkbJoQZqt2_f_RINajwQU5EnHR1PsQx53rh7wJvA,2267
+naas_abi/agents/KnowledgeGraphBuilderAgent.py,sha256=ve1NXFnzU1DNmmq7VD7N9BviYy8-sSnMuEhGzTKe_qY,13008
+naas_abi/agents/KnowledgeGraphBuilderAgent_test.py,sha256=pKgakywjYCkjEngbaG-D9q9gwNMQEgXeD1C2xb9irTo,2718
+naas_abi/agents/OntologyEngineerAgent.py,sha256=ssfX0s7Ath5tmAQJzfB5PzU5UsVEuTxEBYwWNp1RKo0,4209
+naas_abi/agents/OntologyEngineerAgent_test.py,sha256=Z5A9u-GKPMlaaUQmI5Yvm9N9mwZTOuyyrqIyrSwrCEI,1319
+naas_abi/apps/oxigraph_admin/main.py,sha256=dAls9K3ZcOh_f6E2EFOF2eAlJKn5dfGyLDmVLzEVdS8,14196
+naas_abi/apps/oxigraph_admin/terminal_style.py,sha256=ByiWVYM5Ju8iGMKbs1YwfEwLfLIpaQbB5nPHzb9zdUY,5152
+naas_abi/apps/sparql_terminal/main.py,sha256=Wgm3iWNiyTjZ5Nas0DAeHAUcWeU_HGNxEdWP15vjYik,2050
+naas_abi/apps/sparql_terminal/terminal_style.py,sha256=TYBM2ChnOLSgVw3TA33obBSc_5gAC5kJCuD2dqMOmF4,6560
+naas_abi/models/airgap_gemma.py,sha256=I9pWN_DSgy4h6rT7OiYWDVijyTMoEgONm4xfZYWvp6I,8469
+naas_abi/models/airgap_qwen.py,sha256=EQZ-0tAKZCTCqnovLiGCx4ss0yh2Jjk0xh7wCwDomG8,682
+naas_abi/models/default.py,sha256=XZ4--FmFjvO99fQdV0I7niJtHNgSFZuCKPb95pyhLns,721
+naas_abi/pipelines/AIAgentOntologyGenerationPipeline.py,sha256=QWrooZ_xJy0uX92Tzw7fBOs_uME2nLhsjeWi0EGTUBA,24640
+naas_abi/pipelines/AIAgentOntologyGenerationPipeline_test.py,sha256=QwYv5uhqyVIs_UmFi25PSKMOpgRj12kjkWH-Jlu7nbY,4203
+naas_abi/pipelines/AddIndividualPipeline.py,sha256=EKrJX4ODXBMSFUl0R2UsaSmLaj0cUJ3AdIhhtU0Fhso,8445
+naas_abi/pipelines/AddIndividualPipeline_test.py,sha256=3aOW1EWDJfMAQN7HXZGCgbj3KpqkJHE9NBgQyyzVm-I,2377
+naas_abi/pipelines/InsertDataSPARQLPipeline.py,sha256=pJaFywvsRP27yEaT40biYKXNafAyKvo6KSLE-WyvurI,6906
+naas_abi/pipelines/InsertDataSPARQLPipeline_test.py,sha256=SxsEXY7n-QDZNLb7PelDwh11zstsIYrAVD6zoN0X77w,3356
+naas_abi/pipelines/MergeIndividualsPipeline.py,sha256=RjjiEHBJxSJ9ferdQTz2fYwZszDZ6jrM3VAxVgism6Y,9361
+naas_abi/pipelines/MergeIndividualsPipeline_test.py,sha256=PJx7wZISvFyyHk2KVJuKin5Mmb57XOcaw6LG-drNiZw,3067
+naas_abi/pipelines/RemoveIndividualPipeline.py,sha256=5gY-oxNy69aRmVtLbig4zcWQ_Iy5YhV22nM94-_3VLc,5713
+naas_abi/pipelines/RemoveIndividualPipeline_test.py,sha256=kxYq5SqKenfzC4vkDKgXA2H2f7yPjRg1mErv6GK4wsw,1759
+naas_abi/pipelines/UpdateCommercialOrganizationPipeline.py,sha256=TDboVTlOpsbrixRkp-b9dA6Cm194D4ZVUOeQX_uSMe4,6768
+naas_abi/pipelines/UpdateDataPropertyPipeline.py,sha256=zgN1RdpDaODEBStvo02eSiZaKbPQLFzKPCS6tzmIhHw,6853
+naas_abi/pipelines/UpdateLegalNamePipeline.py,sha256=jlN7uZANhoMDmZUpn7Hv4_eVLL-UdI2Ik8orRKagr0k,3451
+naas_abi/pipelines/UpdateLinkedInPagePipeline.py,sha256=TGkGdQGlXUL1LfHQzfJEoSk5eY0Kl93cAnOYgHVyNDg,6180
+naas_abi/pipelines/UpdatePersonPipeline.py,sha256=IAJROMvcsSPeDO5hkXabrH08TYHyhqKN2deq4kABCgI,6263
+naas_abi/pipelines/UpdateSkillPipeline.py,sha256=XqOs5ssrT32ARCilg1Q2Dr-p7bPPbcp3sO2I1Wt1Dao,3907
+naas_abi/pipelines/UpdateTickerPipeline.py,sha256=h1vf5b5vR1okwM94UBDlHu43G2IhynwpkasDf9VnMhg,3397
+naas_abi/pipelines/UpdateWebsitePipeline.py,sha256=GfuqqbciT4yocYTuPGEgZq40KdsfxndbGgYiVT4eg9c,3556
+naas_abi/workflows/AgentRecommendationWorkflow.py,sha256=5R89Fhhe8qMo3IEqoYAx37rWc8sTboDYPbwAuhZVoAw,12993
+naas_abi/workflows/AgentRecommendationWorkflow_test.py,sha256=9LaYrn3qNbPtKFOBxxL9o6pFKZ3aSxlU3IkKmvh2ok8,5205
+naas_abi/workflows/ArtificialAnalysisWorkflow.py,sha256=Jop3puoCQgJAyAQIJDC4kWB8Oo_oF4DcuiC4UHgm-aM,12133
+naas_abi/workflows/ArtificialAnalysisWorkflow_test.py,sha256=glb0WJa19EpyuTqg7SCDkSI5oo_7UCTiDSeHdpaDDUY,1928
+naas_abi/workflows/GetObjectPropertiesFromClassWorkflow.py,sha256=f6YuVWi6TncXh_6hXoLysKCecsQhBX-eyHBKUTuVDeE,14691
+naas_abi/workflows/GetObjectPropertiesFromClassWorkflow_test.py,sha256=4ICPgW7aoL5CprwJitSTWX0VdrrtbZ-3rr8UD6NXAak,2155
+naas_abi/workflows/GetSubjectGraphWorkflow.py,sha256=EkgrFQPyaUVeC8sd7oVcxT9LUkaafkhA1B2UqJDPR8k,2844
+naas_abi/workflows/GetSubjectGraphWorkflow_test.py,sha256=5B2r3QsdBIKwYOuHkEhnfEPKV6xpWZzwRQyhBintgpo,2067
+naas_abi/workflows/SearchIndividualWorkflow.py,sha256=6vWwywKNBlaosopPxiw0OpQnrn99wuiji0uA7jwKOn4,6572
+naas_abi/workflows/SearchIndividualWorkflow_test.py,sha256=ySEgDQFEbevAxkjX9W8No-7JG9A6bH6Y8bCi86aenVA,2762
+naas_abi-1.0.9.dist-info/METADATA,sha256=866XfEsrtWXw2worQ-kA4_C24ihLTsQtx-xHnKL9ibI,552
+naas_abi-1.0.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+naas_abi-1.0.9.dist-info/top_level.txt,sha256=Dhf3mhATB9-JuvBR8Y55OabvI-PCeQUqb5ae3EMmbeA,9
+naas_abi-1.0.9.dist-info/RECORD,,