naas-abi 1.0.0__py3-none-any.whl

naas_abi/pipelines/UpdatePersonPipeline.py

@@ -0,0 +1,184 @@
+from dataclasses import dataclass
+from enum import Enum
+from typing import Annotated, Optional
+
+from fastapi import APIRouter
+from langchain_core.tools import BaseTool, StructuredTool
+from naas_abi_core.pipeline import Pipeline, PipelineConfiguration, PipelineParameters
+from naas_abi_core.services.triple_store.TripleStorePorts import ITripleStoreService
+from naas_abi_core.utils.Graph import ABI, URI_REGEX
+from pydantic import Field
+from rdflib import XSD, Graph, Literal, URIRef
+
+
+@dataclass
+class UpdatePersonPipelineConfiguration(PipelineConfiguration):
+    """Configuration for UpdatePersonPipeline.
+
+    Attributes:
+        triple_store (ITripleStoreService): The triple store service to use
+    """
+
+    triple_store: ITripleStoreService
+
+
+class UpdatePersonPipelineParameters(PipelineParameters):
+    individual_uri: Annotated[
+        str,
+        Field(
+            description="URI of the person.",
+            pattern=URI_REGEX,
+            example="https://www.commoncoreontologies.org/ont00001262/Florent_Ravenel",
+        ),
+    ]
+    first_name: Annotated[
+        Optional[str], Field(None, description="First name of the person.")
+    ]
+    last_name: Annotated[
+        Optional[str], Field(None, description="Last name of the person.")
+    ]
+    date_of_birth: Annotated[
+        Optional[str],
+        Field(
+            None,
+            description="Date of birth of the person. It must be in the format 'YYYY-MM-DD' (e.g. '1990-01-01').",
+            pattern=r"^\d{4}-\d{2}-\d{2}$",
+        ),
+    ]
+    linkedin_page_uri: Annotated[
+        Optional[str],
+        Field(
+            None,
+            description="LinkedIn Page URI of the person from class: http://ontology.naas.ai/abi/LinkedInProfilePage or http://ontology.naas.ai/abi/LinkedInCompanyPage or http://ontology.naas.ai/abi/LinkedInSchoolPage.",
+            pattern=URI_REGEX,
+        ),
+    ]
+    skill_uri: Annotated[
+        str,
+        Field(
+            None,
+            description="Skill URI of the person from class: https://www.commoncoreontologies.org/ont00000089",
+            pattern=URI_REGEX,
+        ),
+    ]
+
+
+class UpdatePersonPipeline(Pipeline):
+    """Pipeline for updating a person in the ontology."""
+
+    __configuration: UpdatePersonPipelineConfiguration
+
+    def __init__(self, configuration: UpdatePersonPipelineConfiguration):
+        super().__init__(configuration)
+        self.__configuration = configuration
+
+    def run(self, parameters: PipelineParameters) -> Graph:
+        if not isinstance(parameters, UpdatePersonPipelineParameters):
+            raise ValueError(
+                "Parameters must be of type UpdatePersonPipelineParameters"
+            )
+
+        # Initialize graphs
+        graph_insert = Graph()
+
+        # Get subject URI & graph
+        individual_uri = URIRef(parameters.individual_uri)
+        graph = self.__configuration.triple_store.get_subject_graph(individual_uri)
+
+        # Update properties
+        if parameters.linkedin_page_uri:
+            check_linkedin_page = list(
+                graph.triples(
+                    (
+                        individual_uri,
+                        ABI.hasLinkedInPage,
+                        URIRef(parameters.linkedin_page_uri),
+                    )
+                )
+            )
+            if len(check_linkedin_page) == 0:
+                graph_insert.add(
+                    (
+                        individual_uri,
+                        ABI.hasLinkedInPage,
+                        URIRef(parameters.linkedin_page_uri),
+                    )
+                )
+        if parameters.skill_uri:
+            check_skill = list(
+                graph.triples(
+                    (individual_uri, ABI.hasSkill, URIRef(parameters.skill_uri))
+                )
+            )
+            if len(check_skill) == 0:
+                graph_insert.add(
+                    (individual_uri, ABI.hasSkill, URIRef(parameters.skill_uri))
+                )
+        if parameters.first_name:
+            check_first_name = list(
+                graph.triples(
+                    (individual_uri, ABI.first_name, Literal(parameters.first_name))
+                )
+            )
+            if len(check_first_name) == 0:
+                graph_insert.add(
+                    (individual_uri, ABI.first_name, Literal(parameters.first_name))
+                )
+        if parameters.last_name:
+            check_last_name = list(
+                graph.triples(
+                    (individual_uri, ABI.last_name, Literal(parameters.last_name))
+                )
+            )
+            if len(check_last_name) == 0:
+                graph_insert.add(
+                    (individual_uri, ABI.last_name, Literal(parameters.last_name))
+                )
+        if parameters.date_of_birth:
+            check_date_of_birth = list(
+                graph.triples(
+                    (
+                        individual_uri,
+                        ABI.date_of_birth,
+                        Literal(parameters.date_of_birth, datatype=XSD.date),
+                    )
+                )
+            )
+            if len(check_date_of_birth) == 0:
+                graph_insert.add(
+                    (
+                        individual_uri,
+                        ABI.date_of_birth,
+                        Literal(parameters.date_of_birth, datatype=XSD.date),
+                    )
+                )
+
+        # Save the graph
+        self.__configuration.triple_store.insert(graph_insert)
+        graph += graph_insert
+        return graph
+
+    def as_tools(self) -> list[BaseTool]:
+        return [
+            StructuredTool(
+                name="update_person",
+                description="Update a person in the ontology.",
+                func=lambda **kwargs: self.run(
+                    UpdatePersonPipelineParameters(**kwargs)
+                ),
+                args_schema=UpdatePersonPipelineParameters,
+            )
+        ]
+
+    def as_api(
+        self,
+        router: APIRouter,
+        route_name: str = "",
+        name: str = "",
+        description: str = "",
+        description_stream: str = "",
+        tags: list[str | Enum] | None = None,
+    ) -> None:
+        if tags is None:
+            tags = []
+        return None

naas_abi/pipelines/UpdateSkillPipeline.py

@@ -0,0 +1,118 @@
+from dataclasses import dataclass
+from enum import Enum
+from typing import Annotated, Optional
+
+from fastapi import APIRouter
+from langchain_core.tools import BaseTool, StructuredTool
+from naas_abi_core.pipeline import Pipeline, PipelineConfiguration, PipelineParameters
+from naas_abi_core.services.triple_store.TripleStorePorts import ITripleStoreService
+from naas_abi_core.utils.Graph import ABI, URI_REGEX
+from pydantic import Field
+from rdflib import Graph, Literal, URIRef
+
+
+@dataclass
+class UpdateSkillPipelineConfiguration(PipelineConfiguration):
+    """Configuration for UpdateSkillPipeline.
+
+    Attributes:
+        triple_store (ITripleStoreService): The triple store service to use
+    """
+
+    triple_store: ITripleStoreService
+
+
+class UpdateSkillPipelineParameters(PipelineParameters):
+    individual_uri: Annotated[
+        str, Field(description="URI of the skill.", pattern=URI_REGEX)
+    ]
+    description: Annotated[
+        Optional[str], Field(None, description="Description of the skill")
+    ] = None
+    person_uri: Annotated[
+        Optional[str],
+        Field(
+            None,
+            description="Person URI from class: https://www.commoncoreontologies.org/ont00001262.",
+            pattern=URI_REGEX,
+        ),
+    ] = None
+
+
+class UpdateSkillPipeline(Pipeline):
+    """Pipeline for updating a skill in the ontology."""
+
+    __configuration: UpdateSkillPipelineConfiguration
+
+    def __init__(self, configuration: UpdateSkillPipelineConfiguration):
+        super().__init__(configuration)
+        self.__configuration = configuration
+
+    def run(self, parameters: PipelineParameters) -> Graph:
+        if not isinstance(parameters, UpdateSkillPipelineParameters):
+            raise ValueError("Parameters must be of type UpdateSkillPipelineParameters")
+
+        # Initialize a new graph
+        graph_insert = Graph()
+
+        # Get subject URI & graph
+        individual_uri = URIRef(parameters.individual_uri)
+        graph = self.__configuration.triple_store.get_subject_graph(individual_uri)
+
+        # Update properties
+        if parameters.description:
+            check_description = list(
+                graph.triples(
+                    (
+                        individual_uri,
+                        ABI.hasDescription,
+                        Literal(parameters.description),
+                    )
+                )
+            )
+            if len(check_description) == 0:
+                graph_insert.add(
+                    (
+                        individual_uri,
+                        ABI.hasDescription,
+                        Literal(parameters.description),
+                    )
+                )
+        if parameters.person_uri:
+            check_person_uri = list(
+                graph.triples(
+                    (individual_uri, ABI.isSkillOf, URIRef(parameters.person_uri))
+                )
+            )
+            if len(check_person_uri) == 0:
+                graph_insert.add(
+                    (individual_uri, ABI.isSkillOf, URIRef(parameters.person_uri))
+                )
+
+        # Save the graph
+        self.__configuration.triple_store.insert(graph_insert)
+        graph += graph_insert
+        return graph
+
+    def as_tools(self) -> list[BaseTool]:
+        return [
+            StructuredTool(
+                name="update_skill",
+                description="Update a skill with a description in the ontology.",
+                func=lambda **kwargs: self.run(UpdateSkillPipelineParameters(**kwargs)),
+                args_schema=UpdateSkillPipelineParameters,
+            )
+        ]
+
+    def as_api(
+        self,
+        router: APIRouter,
+        route_name: str = "",
+        name: str = "",
+        description: str = "",
+        description_stream: str = "",
+        tags: list[str | Enum] | None = None,
+    ) -> None:
+        if tags is None:
+            tags = []
+        return None

naas_abi/pipelines/UpdateTickerPipeline.py

@@ -0,0 +1,104 @@
+from dataclasses import dataclass
+from enum import Enum
+from typing import Annotated, Optional
+
+from fastapi import APIRouter
+from langchain_core.tools import BaseTool, StructuredTool
+from naas_abi_core.pipeline import Pipeline, PipelineConfiguration, PipelineParameters
+from naas_abi_core.services.triple_store.TripleStorePorts import ITripleStoreService
+from naas_abi_core.utils.Graph import ABI, URI_REGEX
+from pydantic import Field
+from rdflib import Graph, URIRef
+
+
+@dataclass
+class UpdateTickerPipelineConfiguration(PipelineConfiguration):
+    triple_store: ITripleStoreService
+
+
+class UpdateTickerPipelineParameters(PipelineParameters):
+    individual_uri: Annotated[
+        str,
+        Field(
+            description="URI of the ticker. It must start with 'http://ontology.naas.ai/abi/'.",
+            pattern=URI_REGEX,
+        ),
+    ]
+    organization_uri: Annotated[
+        Optional[str],
+        Field(
+            None,
+            description="Organization URI from class: https://www.commoncoreontologies.org/ont00000443. It must start with 'http://ontology.naas.ai/abi/'.",
+            pattern=URI_REGEX,
+        ),
+    ]
+
+
+class UpdateTickerPipeline(Pipeline):
+    __configuration: UpdateTickerPipelineConfiguration
+
+    def __init__(self, configuration: UpdateTickerPipelineConfiguration):
+        super().__init__(configuration)
+        self.__configuration = configuration
+
+    def run(self, parameters: PipelineParameters) -> Graph:
+        if not isinstance(parameters, UpdateTickerPipelineParameters):
+            raise ValueError(
+                "Parameters must be of type UpdateTickerPipelineParameters"
+            )
+
+        # Initialize graphs
+        graph_insert = Graph()
+
+        # Get subject URI & graph
+        individual_uri = URIRef(parameters.individual_uri)
+        graph = self.__configuration.triple_store.get_subject_graph(individual_uri)
+
+        # Update properties
+        if parameters.organization_uri:
+            check_organization_uri = list(
+                graph.triples(
+                    (
+                        individual_uri,
+                        ABI.isTickerSymbolOf,
+                        URIRef(parameters.organization_uri),
+                    )
+                )
+            )
+            if len(check_organization_uri) == 0:
+                graph_insert.add(
+                    (
+                        individual_uri,
+                        ABI.isTickerSymbolOf,
+                        URIRef(parameters.organization_uri),
+                    )
+                )
+
+        self.__configuration.triple_store.insert(graph_insert)
+        graph += graph_insert
+        return graph
+
+    def as_tools(self) -> list[BaseTool]:
+        return [
+            StructuredTool(
+                name="update_ticker",
+                description="Update a ticker symbol in the ontology.",
+                func=lambda **kwargs: self.run(
+                    UpdateTickerPipelineParameters(**kwargs)
+                ),
+                args_schema=UpdateTickerPipelineParameters,
+            )
+        ]
+
+    def as_api(
+        self,
+        router: APIRouter,
+        route_name: str = "",
+        name: str = "",
+        description: str = "",
+        description_stream: str = "",
+        tags: list[str | Enum] | None = None,
+    ) -> None:
+        if tags is None:
+            tags = []
+        return None

naas_abi/pipelines/UpdateWebsitePipeline.py

@@ -0,0 +1,106 @@
+from dataclasses import dataclass
+from enum import Enum
+from typing import Annotated, Optional
+
+from fastapi import APIRouter
+from langchain_core.tools import BaseTool, StructuredTool
+from naas_abi_core.pipeline import Pipeline, PipelineConfiguration, PipelineParameters
+from naas_abi_core.services.triple_store.TripleStorePorts import ITripleStoreService
+from naas_abi_core.utils.Graph import ABI, URI_REGEX
+from pydantic import Field
+from rdflib import Graph, URIRef
+
+
+@dataclass
+class UpdateWebsitePipelineConfiguration(PipelineConfiguration):
+    """Configuration for UpdateWebsitePipeline.
+
+    Attributes:
+        triple_store (ITripleStoreService): The triple store service to use
+    """
+
+    triple_store: ITripleStoreService
+
+
+class UpdateWebsitePipelineParameters(PipelineParameters):
+    individual_uri: Annotated[
+        str,
+        Field(
+            description="URI of the website. It must start with 'http://ontology.naas.ai/abi/'.",
+            pattern=URI_REGEX,
+        ),
+    ]
+    owner_uri: Annotated[
+        Optional[str],
+        Field(
+            None,
+            description="Owner URI from class: https://www.commoncoreontologies.org/ont00001262 or https://www.commoncoreontologies.org/ont00000443.",
+            pattern=URI_REGEX,
+        ),
+    ] = None
+
+
+class UpdateWebsitePipeline(Pipeline):
+    """Pipeline for updating a Website in the ontology."""
+
+    __configuration: UpdateWebsitePipelineConfiguration
+
+    def __init__(self, configuration: UpdateWebsitePipelineConfiguration):
+        super().__init__(configuration)
+        self.__configuration = configuration
+
+    def run(self, parameters: PipelineParameters) -> Graph:
+        if not isinstance(parameters, UpdateWebsitePipelineParameters):
+            raise ValueError(
+                "Parameters must be of type UpdateWebsitePipelineParameters"
+            )
+
+        # Initialize a new graph
+        graph = Graph()
+
+        # Get website URI
+        website_uri = URIRef(parameters.individual_uri)
+        if not parameters.individual_uri.startswith("http://ontology.naas.ai/abi/"):
+            raise ValueError(
+                f"Invalid Website URI: {parameters.individual_uri}. It must start with 'http://ontology.naas.ai/abi/'."
+            )
+
+        # Add owners URI if provided
+        if parameters.owner_uri:
+            check_owner_uri = list(
+                graph.triples(
+                    (website_uri, ABI.isWebsiteOf, URIRef(parameters.owner_uri))
+                )
+            )
+            if len(check_owner_uri) == 0:
+                graph.add((website_uri, ABI.isWebsiteOf, URIRef(parameters.owner_uri)))
+                graph.add((URIRef(parameters.owner_uri), ABI.hasWebsite, website_uri))
+
+        # Save the graph
+        self.__configuration.triple_store.insert(graph)
+        return graph
+
+    def as_tools(self) -> list[BaseTool]:
+        return [
+            StructuredTool(
+                name="update_website",
+                description="Update a website in the ontology.",
+                func=lambda **kwargs: self.run(
+                    UpdateWebsitePipelineParameters(**kwargs)
+                ),
+                args_schema=UpdateWebsitePipelineParameters,
+            )
+        ]
+
+    def as_api(
+        self,
+        router: APIRouter,
+        route_name: str = "",
+        name: str = "",
+        description: str = "",
+        description_stream: str = "",
+        tags: list[str | Enum] | None = None,
+    ) -> None:
+        if tags is None:
+            tags = []
+        return None

naas_abi/triggers.py

@@ -0,0 +1,131 @@
+# This file defines triggers for the opendata module
+# Triggers allow the module to react automatically to changes in the ontology store
+# Each trigger consists of:
+#   1. A triple pattern (subject, predicate, object) where None acts as a wildcard
+#   2. An event type (OntologyEvent.INSERT or OntologyEvent.DELETE)
+#   3. A callback function to execute when the event occurs
+#
+# Example:
+# from naas_abi_core.services.triple_store.TripleStorePorts import OntologyEvent
+# from rdflib import URIRef, Namespace
+#
+# ex = Namespace("http://example.org/")
+#
+# def handle_new_dataset(event_type, ontology_name, triple):
+#     subject, predicate, obj = triple
+#     # Process the new dataset here
+#
+# triggers = [
+#     ((None, URIRef(ex + "hasOpenData"), None), OntologyEvent.INSERT, handle_new_dataset),
+# ]
+
+import os
+
+from naas_abi_core import logger, secret, services
+from naas_abi_core.services.triple_store.TripleStorePorts import OntologyEvent
+
+
+def is_production_mode():
+    """Check if the application is running in production mode"""
+    return os.environ.get("ENV") != "dev"
+
+
+def create_class_ontology_yaml():
+    """Create class ontology YAML trigger - only active in production"""
+    if not is_production_mode():
+        logger.debug("Skipping class ontology YAML trigger - not in production mode")
+        return None
+
+    from naas_abi_marketplace.applications.naas.integrations.NaasIntegration import (
+        NaasIntegrationConfiguration,
+    )
+
+    from naas_abi.workflows.ConvertOntologyGraphToYamlWorkflow import (
+        ConvertOntologyGraphToYamlWorkflowConfiguration,
+    )
+    from naas_abi.workflows.CreateClassOntologyYamlWorkflow import (
+        CreateClassOntologyYamlWorkflow,
+        CreateClassOntologyYamlWorkflowConfiguration,
+    )
+
+    # Get secrets
+    naas_api_key = secret.get("NAAS_API_KEY")
+    if naas_api_key is None:
+        logger.error("NAAS_API_KEY is not set")
+        return None
+
+    # Configure Naas integration
+    naas_integration_config = NaasIntegrationConfiguration(api_key=naas_api_key)
+
+    # Configure ConvertOntologyGraphToYaml workflow
+    convert_ontology_graph_config = ConvertOntologyGraphToYamlWorkflowConfiguration(
+        naas_integration_config
+    )
+    workflow = CreateClassOntologyYamlWorkflow(
+        CreateClassOntologyYamlWorkflowConfiguration(
+            services.triple_store_service, convert_ontology_graph_config
+        )
+    )
+
+    # Subscribe to the trigger
+    logger.info("Activating class ontology YAML trigger in production mode")
+    return ((None, None, None), OntologyEvent.INSERT, workflow.trigger, True)
+
+
+def create_individual_ontology_yaml():
+    """Create individual ontology YAML trigger - only active in production"""
+    if not is_production_mode():
+        logger.debug(
+            "Skipping individual ontology YAML trigger - not in production mode"
+        )
+        return None
+
+    from naas_abi_marketplace.applications.naas.integrations.NaasIntegration import (
+        NaasIntegrationConfiguration,
+    )
+
+    from naas_abi.workflows.ConvertOntologyGraphToYamlWorkflow import (
+        ConvertOntologyGraphToYamlWorkflowConfiguration,
+    )
+    from naas_abi.workflows.CreateIndividualOntologyYamlWorkflow import (
+        CreateIndividualOntologyYamlWorkflow,
+        CreateIndividualOntologyYamlWorkflowConfiguration,
+    )
+
+    # Get secrets
+    naas_api_key = secret.get("NAAS_API_KEY")
+    if naas_api_key is None:
+        logger.error("NAAS_API_KEY is not set")
+        return None
+
+    # Configure Naas integration
+    naas_integration_config = NaasIntegrationConfiguration(api_key=naas_api_key)
+
+    # Configure ConvertOntologyGraphToYaml workflow
+    convert_ontology_graph_config = ConvertOntologyGraphToYamlWorkflowConfiguration(
+        naas_integration_config
+    )
+    workflow = CreateIndividualOntologyYamlWorkflow(
+        CreateIndividualOntologyYamlWorkflowConfiguration(
+            services.triple_store_service, convert_ontology_graph_config
+        )
+    )
+
+    # Subscribe to the trigger
+    logger.info("Activating individual ontology YAML trigger in production mode")
+    return ((None, None, None), OntologyEvent.INSERT, workflow.trigger, True)
+
+
+# Filter out None values from triggers (when not in production mode)
+# Skip triggers in test environment to avoid SSH tunnel initialization
+if os.getenv("PYTEST_CURRENT_TEST") is not None or os.getenv("TESTING") == "true":
+    triggers = []  # Skip all triggers during testing
+else:
+    triggers = [
+        trigger
+        for trigger in [
+            create_class_ontology_yaml(),
+            create_individual_ontology_yaml(),
+        ]
+        if trigger is not None
+    ]