naas-abi 1.0.8__py3-none-any.whl → 1.0.10__py3-none-any.whl

naas_abi/apps/terminal_agent/terminal_style.py

@@ -1,175 +0,0 @@
-from rich.console import Console
-from rich.markdown import Markdown
-from rich.panel import Panel
-from rich.syntax import Syntax
-from rich.text import Text
-from rich.box import ROUNDED
-from PIL import Image
-import os
-import platform
-import subprocess
-
-console = Console()
-
-
-def set_terminal_title():
-    """Set the terminal title to 'ABI'"""
-    if platform.system() == "Windows":
-        os.system("title ABI")
-    else:  # For Unix-like systems (Linux, macOS)
-        print("\33]0;ABI\a", end="", flush=True)
-
-
-def print_agent_response(text, agent_label):
-    console.print()  # Add a blank line before the assistant's response
-    console.print(f"[bold green]{agent_label}:[/bold green] ", end="")
-    md = Markdown(text)
-    console.print(md)
-    console.print()  # Add a blank line after the assistant's response
-
-
-def print_system_message(text):
-    console.print()  # Add a blank line before the system message
-    system_text = Text(text, style="yellow")
-    console.print(
-        Panel(
-            system_text,
-            border_style="yellow",
-            box=ROUNDED,
-            expand=False,
-            title="System",
-            title_align="left",
-        )
-    )
-    console.print()  # Add a blank line after the system message
-
-
-def print_code(code, language="python"):
-    syntax = Syntax(code, language, theme="monokai", line_numbers=True)
-    console.print(
-        Panel(
-            syntax,
-            border_style="red",
-            box=ROUNDED,
-            expand=False,
-            title=f"Code ({language})",
-            title_align="left",
-        )
-    )
-
-
-def dict_to_equal_string(d: dict) -> str:
-    return "\n".join([f'-{key}="{value}"' for key, value in d.items()])
-
-
-def print_tool_usage(message):
-    print_message = ""
-    tool_name = message.tool_calls[0]["name"]
-    arguments = ""
-    if (
-        "args" in message.tool_calls[0]
-        and len(message.tool_calls[0]["args"].values()) > 0
-    ):
-        arguments += dict_to_equal_string(message.tool_calls[0]["args"])
-
-    if tool_name.startswith("transfer_to_"):
-        tool_label = " ".join(
-            word.capitalize()
-            for word in tool_name.split("transfer_to_")[1].replace("_", " ").split()
-        )
-        print_message = f"\n🧞 [bold blue]Delegated to [/bold blue]{tool_label}"
-    else:
-        tool_label = tool_name.capitalize().replace("_", " ")
-        print_message = f"\n[bold blue]Tool Used:[/bold blue] {tool_label}\n{arguments}"
-    console.print(print_message)
-
-
-def print_tool_response(response):
-    console.print(f"\n[bold blue]Response:[/bold blue] {response}")
-
-
-def clear_screen():
-    console.clear()
-
-
-def print_welcome_message(agent):
-    # Set terminal title
-    set_terminal_title()
-    
-    # Skip the welcome - we already said hello in the CLI startup
-    # Just quietly start the conversation
-    pass
-
-
-def print_divider():
-    console.print("─" * console.width, style="dim")
-
-
-def get_user_input(agent_label):
-    try:
-        user_input = console.input(
-            "\n[bold cyan]You:[/bold cyan] "
-        )  # Add a newline before the prompt
-        console.print()  # Add a blank line after the user's input
-        return user_input
-    except EOFError:
-        console.print(
-            f"\n[bold red]{agent_label}:[/bold red] Conversation ended by user."
-        )
-        return "exit"
-    except KeyboardInterrupt:
-        console.print(
-            f"\n[bold red]{agent_label}:[/bold red] Conversation ended by user."
-        )
-        return "exit"
-
-
-def print_image(image_path: str):
-    """Display an image in the terminal."""
-    try:
-        console.print()  # Add some spacing
-
-        # Display the file path and viewing instructions
-        message = (
-            f"[yellow]Image saved at: {image_path}[/yellow]\n\n"
-            "[dim]To view the image:[/dim]\n"
-            f"[cyan]• Local system: Open {image_path} with your image viewer[/cyan]\n"
-            "[cyan]• Remote/SSH: Download the file to your local machine to view[/cyan]"
-        )
-
-        console.print(
-            Panel(
-                message,
-                border_style="yellow",
-                box=ROUNDED,
-                expand=False,
-                title="Graph Output",
-                title_align="left",
-            )
-        )
-
-        # Only try to show the image if we're in a GUI environment
-        if os.environ.get("DISPLAY") and platform.system() != "Windows":
-            try:
-                img = Image.open(image_path)
-                img.show()
-            except Exception:
-                pass  # Silently fail if we can't display the image
-        elif platform.system() == "Windows":
-            try:
-                subprocess.run(['start', '', image_path], shell=True)  # Windows-specific file opening
-            except Exception:
-                pass
-
-        console.print()  # Add some spacing after
-    except Exception as e:
-        console.print(
-            Panel(
-                f"[yellow]Unable to process image. File saved at: {image_path}[/yellow]\nError: {str(e)}",
-                border_style="yellow",
-                box=ROUNDED,
-                expand=False,
-                title="Graph Output",
-                title_align="left",
-            )
-        )

naas_abi/mappings.py

@@ -1,83 +0,0 @@
-COLORS_NODES = {
-    # Role
-    "http://ontology.naas.ai/abi/Offering": "#f61685",
-    "http://ontology.naas.ai/abi/Product": "#C4116A",
-    "http://ontology.naas.ai/abi/Service": "#F973B5",
-    "http://ontology.naas.ai/abi/Positioning": "#FDD0E6",
-    # Capability
-    "http://ontology.naas.ai/abi/HumanCapabilities": "#D397F8",
-    "http://ontology.naas.ai/abi/TechnologicalCapabilities": "#bd87df",
-    "https://www.commoncoreontologies.org/ont00000089": "#bd87df",  # Skill
-    # Temporal Region
-    "https://www.commoncoreontologies.org/ont00000832": "#47DD82",
-    "http://ontology.naas.ai/abi/ISO8601UTCDateTime": "#47DD82",
-    # Agent Ontology
-    'https://www.commoncoreontologies.org/ont00001180': 'white', # Organization
-    'https://www.commoncoreontologies.org/ont00000443': 'white', # Commercial Organization
-    'http://ontology.naas.ai/abi/StockMarket': 'white', # Stock Market Institution
-    'https://www.commoncoreontologies.org/ont00001302': 'white', # Civil Organization
-    'https://www.commoncoreontologies.org/ont00000564': 'white', # Educational Organization
-    'https://www.commoncoreontologies.org/ont00000408': 'white', # Government Organization
-    'https://www.commoncoreontologies.org/ont00001239': '#f6f6f6', # Group Organization
-    'http://ontology.naas.ai/abi/Industry': '#f6f6f6', # Group of organizations
-    'http://ontology.naas.ai/abi/LinkedInIndustry': '#f6f6f6', # Group of organizations
-    'https://www.commoncoreontologies.org/ont00001262': '#919191', # Group
-    'https://www.commoncoreontologies.org/ont00000647': '#919191', # Member
-    'https://www.commoncoreontologies.org/ont00000914': 'white', # Group Member
-    'https://www.commoncoreontologies.org/ont00000175': '#3e3e3e', # Member Role
-    'http://ontology.naas.ai/abi/leadership#Position': '#905e0e', # Position
-    # Event Ontology
-    "http://ontology.naas.ai/abi/ActOfPartnership": "black",
-    "http://ontology.naas.ai/abi/ActOfJointVenture": "black",
-    "http://ontology.naas.ai/abi/ActOfMarketingAlliance": "black",
-    "http://ontology.naas.ai/abi/ActOfResearchCollaboration": "black",
-    "http://ontology.naas.ai/abi/ActOfTechnologyLicensing": "black",
-    "http://ontology.naas.ai/abi/ActOfDistributionAgreement": "black",
-    "http://ontology.naas.ai/abi/ActOfOrganizationalMerger": "black",
-    "http://ontology.naas.ai/abi/ActOfOrganizationalAcquisition": "black",
-    "http://ontology.naas.ai/abi/ActOfSubsidiaryEstablishment": "black",
-    # Information Content
-    "https://www.commoncoreontologies.org/ont00001331": "#AECCE4",  # Legal Name
-    "http://ontology.naas.ai/abi/OrganizationSize": "#9ABDDC",
-    "http://ontology.naas.ai/abi/Ticker": "#6C849A",
-    "http://ontology.naas.ai/abi/Address": "#4D5F6E",
-    "http://ontology.naas.ai/abi/Market": "#4169E1",
-    "http://ontology.naas.ai/abi/MarketSegment": "#6495ED",
-    "http://ontology.naas.ai/abi/LinkedInProfilePage": "#0A66C2",
-    "http://ontology.naas.ai/abi/IncomeStatement": "#847c35",
-    "http://ontology.naas.ai/abi/BalanceSheet": "#9d965d",
-    "http://ontology.naas.ai/abi/CashFlowStatement": "#c1bd9a",
-    # Finance Ontology
-    "http://ontology.naas.ai/abi/Revenue": "#bf9b30",
-    "http://ontology.naas.ai/abi/Expenses": "#bf9b30",
-    "http://ontology.naas.ai/abi/Profit": "#bf9b30",
-    "http://ontology.naas.ai/abi/EBITDA": "#bf9b30",
-    "http://ontology.naas.ai/abi/KPI": "#d4bf79",
-    # Facility Ontology
-    "http://ontology.naas.ai/abi/GlobalHeadquarters": "#9D7153",
-    "http://ontology.naas.ai/abi/RegionalHeadquarters": "#b0917c",
-    # Geospatial Ontology
-    "http://ontology.naas.ai/abi/Country": "#8e7010",
-    "http://ontology.naas.ai/abi/CountryISO3166-1": "#8e7010",
-    "http://ontology.naas.ai/abi/State": "#c6a84f",
-    "https://www.commoncoreontologies.org/ont00000887": "#aa8b26",
-    # Euronext Index
-    "http://ontology.naas.ai/abi/DataSource": "#5b1f00",
-    "http://ontology.naas.ai/abi/DataSourceComponent": "#713112",
-    "http://ontology.naas.ai/abi/TradingLocation": "#09a12d",
-    "http://ontology.naas.ai/abi/euronext#Index": "#7fed0e",
-    # LinkedIn
-    "http://ontology.naas.ai/abi/linkedin#LinkedInProfile": "white",
-    "http://ontology.naas.ai/abi/linkedin#PositionGroup": "white",
-    "http://ontology.naas.ai/abi/linkedin#Position": "grey",
-    "http://ontology.naas.ai/abi/linkedin#Language": "#D397F8",
-    "http://ontology.naas.ai/abi/linkedin#Skill": "#bd87df",
-    "http://ontology.naas.ai/abi/linkedin#Certification": "#76b3ec",
-    "http://ontology.naas.ai/abi/linkedin#Course": "#b9986d",
-    
-    # "http://ontology.naas.ai/abi/linkedin#Skill": "white",
-    # "http://ontology.naas.ai/abi/linkedin#Education": "white",
-    # "http://ontology.naas.ai/abi/linkedin#Certification": "white",
-    # "http://ontology.naas.ai/abi/linkedin#Company": "white",
-    # "http://ontology.naas.ai/abi/linkedin#Industry": "white",
-}

naas_abi/triggers.py

@@ -1,131 +0,0 @@
-# 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
-    ]

naas_abi/workflows/ConvertOntologyGraphToYamlWorkflow.py

@@ -1,210 +0,0 @@
-from dataclasses import dataclass
-from enum import Enum
-from typing import Annotated, Dict, Optional
-
-import pydash as _
-import yaml
-from langchain_core.tools import BaseTool, StructuredTool
-from naas_abi import config, logger
-from naas_abi.mappings import COLORS_NODES
-from naas_abi_core.utils.Expose import APIRouter
-from naas_abi_core.utils.OntologyYaml import OntologyYaml
-from naas_abi_core.workflow import Workflow, WorkflowConfiguration
-from naas_abi_core.workflow.workflow import WorkflowParameters
-from naas_abi_marketplace.applications.naas.integrations.NaasIntegration import (
-    NaasIntegration,
-    NaasIntegrationConfiguration,
-)
-from pydantic import Field
-from rdflib import Graph
-from yaml import Dumper
-
-
-@dataclass
-class ConvertOntologyGraphToYamlWorkflowConfiguration(WorkflowConfiguration):
-    """Configuration for ConvertOntologyGraphToYaml workflow.
-
-    Attributes:
-        naas_integration_config (NaasIntegrationConfiguration): Configuration for the Naas integration
-    """
-
-    naas_integration_config: NaasIntegrationConfiguration
-
-
-class ConvertOntologyGraphToYamlWorkflowParameters(WorkflowParameters):
-    """Parameters for ConvertOntologyGraphToYaml workflow execution.
-
-    Attributes:
-        ontology_name (str): The name of the ontology store to use
-        label (str): The label of the ontology
-        description (str): The description of the ontology
-        logo_url (str): The URL of the ontology logo
-        level (str): The level of the ontology (e.g., 'TOP_LEVEL', 'MID_LEVEL', 'DOMAIN', 'USE_CASE')
-        display_relations_names (bool): Whether to display relation names in the visualization
-    """
-
-    graph: Annotated[
-        str, Field(..., description="The graph serialized as turtle format")
-    ]
-    ontology_id: Annotated[
-        Optional[str], Field(..., description="The ID of the ontology")
-    ] = None
-    label: Annotated[str, Field(..., description="The label of the ontology")] = (
-        "New Ontology"
-    )
-    description: Annotated[
-        str,
-        Field(
-            ...,
-            description="The description of the ontology. Example: 'Represents ABI Ontology with agents, workflows, ontologies, pipelines and integrations.'",
-        ),
-    ] = "New Ontology Description"
-    logo_url: Annotated[
-        Optional[str], Field(..., description="The logo URL of the ontology")
-    ] = "https://naasai-public.s3.eu-west-3.amazonaws.com/abi-demo/ontology_ULO.png"
-    level: Annotated[str, Field(..., description="The level of the ontology")] = (
-        "USE_CASE"
-    )
-    display_relations_names: Annotated[
-        bool,
-        Field(
-            ..., description="Whether to display relation names in the visualization"
-        ),
-    ] = True
-    class_colors_mapping: Annotated[
-        Dict, Field(..., description="The mapping of class colors")
-    ] = COLORS_NODES
-
-
-class ConvertOntologyGraphToYamlWorkflow(Workflow):
-    """Workflow for converting ontology graph to YAML."""
-
-    __configuration: ConvertOntologyGraphToYamlWorkflowConfiguration
-
-    def __init__(self, configuration: ConvertOntologyGraphToYamlWorkflowConfiguration):
-        self.__configuration = configuration
-        self.__naas_integration = NaasIntegration(
-            self.__configuration.naas_integration_config
-        )
-
-    def graph_to_yaml(
-        self, parameters: ConvertOntologyGraphToYamlWorkflowParameters
-    ) -> str:
-        # Initialize parameters
-        logger.debug(f"==> Converting ontology graph to YAML: {parameters.label}")
-        yaml_data = None
-        ontology_id = parameters.ontology_id
-
-        # Create Graph from turtle string
-        g = Graph()
-        g.parse(data=parameters.graph, format="turtle")
-
-        # Upload asset to Naas
-        asset = self.__naas_integration.upload_asset(
-            data=parameters.graph.encode("utf-8"),  # Use the original turtle string
-            workspace_id=config.workspace_id,
-            storage_name=config.storage_name,
-            prefix="assets",
-            object_name=str(parameters.label + ".ttl"),
-            visibility="public",
-        )
-        # Save asset URL to JSON
-        if asset is None:
-            raise ValueError("Failed to upload asset to Naas")
-
-        asset_url = asset.get("asset", {}).get("url")
-        if not asset_url:
-            raise ValueError("Asset URL not found in response")
-
-        if asset_url.endswith("/"):
-            asset_url = asset_url[:-1]
-
-        # Convert to YAML
-        try:
-            yaml_data = OntologyYaml.rdf_to_yaml(
-                g,
-                display_relations_names=parameters.display_relations_names,
-                class_colors_mapping=parameters.class_colors_mapping,
-            )
-        except Exception as e:
-            message = f"Error converting ontology to YAML: {e}"
-            raise e
-
-        # Initialize parameters
-        if yaml_data is not None:
-            workspace_id = config.workspace_id
-            onto_label = parameters.label
-            onto_description = parameters.description
-            onto_logo_url = parameters.logo_url
-            onto_level = parameters.level
-
-            # Get ontology ID if it exists
-            ontologies = self.__naas_integration.get_ontologies(workspace_id).get(
-                "ontologies", []
-            )
-            for ontology in ontologies:
-                if (
-                    ontology
-                    and isinstance(ontology, dict)
-                    and ontology.get("label") == onto_label
-                ):
-                    ontology_id = ontology.get("id")
-                    break
-
-            if ontology_id is None:
-                # Create new ontology
-                res = self.__naas_integration.create_ontology(
-                    workspace_id=workspace_id,
-                    label=onto_label,
-                    source=yaml.dump(yaml_data, Dumper=Dumper),
-                    level=onto_level,
-                    description=onto_description,
-                    download_url=asset_url,
-                    logo_url=onto_logo_url,
-                )
-                ontology_id = str(_.get(res, "ontology.id", ""))
-                message = (
-                    f"✅ Ontology '{onto_label}' ({ontology_id}) successfully created."
-                )
-            else:
-                # Update existing ontology
-                res = self.__naas_integration.update_ontology(
-                    workspace_id=workspace_id,
-                    ontology_id=ontology_id,
-                    source=yaml.dump(yaml_data, Dumper=Dumper),
-                    level=onto_level,
-                    description=onto_description,
-                    download_url=asset_url,
-                    logo_url=onto_logo_url,
-                )
-                message = (
-                    f"✅ Ontology '{onto_label}' ({ontology_id}) successfully updated."
-                )
-        logger.info(message)
-        if ontology_id is None:
-            raise ValueError("Failed to create or update ontology")
-        return ontology_id
-
-    def as_tools(self) -> list[BaseTool]:
-        """Returns a list of LangChain tools for this workflow."""
-        return [
-            StructuredTool(
-                name="convert_graph_to_yaml",
-                description="Convert an ontology graph to YAML.",
-                func=lambda **kwargs: self.graph_to_yaml(
-                    ConvertOntologyGraphToYamlWorkflowParameters(**kwargs)
-                ),
-                args_schema=ConvertOntologyGraphToYamlWorkflowParameters,
-            )
-        ]
-
-    def as_api(
-        self,
-        router: APIRouter,
-        route_name: str = "",
-        name: str = "",
-        description: str = "",
-        description_stream: str = "",
-        tags: list[str | Enum] | None = None,
-    ) -> None:
-        pass

naas_abi/workflows/ConvertOntologyGraphToYamlWorkflow_test.py

@@ -1,78 +0,0 @@
-import pytest
-from naas_abi import secret
-from naas_abi.workflows.ConvertOntologyGraphToYamlWorkflow import (
-    ConvertOntologyGraphToYamlWorkflow,
-    ConvertOntologyGraphToYamlWorkflowConfiguration,
-    ConvertOntologyGraphToYamlWorkflowParameters,
-)
-from naas_abi_marketplace.applications.naas.integrations.NaasIntegration import (
-    NaasIntegration,
-    NaasIntegrationConfiguration,
-)
-
-
-@pytest.fixture
-def workflow() -> ConvertOntologyGraphToYamlWorkflow:
-    return ConvertOntologyGraphToYamlWorkflow(
-        ConvertOntologyGraphToYamlWorkflowConfiguration(
-            NaasIntegrationConfiguration(api_key=secret.get("NAAS_API_KEY"))
-        )
-    )
-
-
-def test_convert_ontology_graph_to_yaml_workflow(
-    workflow: ConvertOntologyGraphToYamlWorkflow,
-):
-    import time
-    from uuid import uuid4
-
-    from naas_abi import config, logger, services
-    from rdflib import OWL, RDF, RDFS, Graph, Literal, Namespace, URIRef
-
-    ABI = Namespace("http://ontology.naas.ai/abi/")
-
-    graph = Graph()
-    uri = ABI[str(uuid4())]
-    graph.add(
-        (uri, RDF.type, URIRef("https://www.commoncoreontologies.org/ont00000443"))
-    )
-    graph.add((uri, RDF.type, OWL.NamedIndividual))
-    graph.add((uri, RDFS.label, Literal("Naas.ai")))
-    graph.add(
-        (
-            uri,
-            ABI.logo,
-            Literal(
-                "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcQ9gXMaBLQZ39W6Pk53PRuzFjUvv_6lLRWPoQ&s"
-            ),
-        )
-    )
-    services.triple_store_service.insert(graph)
-    time.sleep(3)
-
-    # Run workflow
-    ontology_id = workflow.graph_to_yaml(
-        ConvertOntologyGraphToYamlWorkflowParameters(
-            graph=graph.serialize(format="turtle"),
-            ontology_id=None,
-            label="Naas.ai",
-            description="Naas.ai Ontology",
-            logo_url="https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcQ9gXMaBLQZ39W6Pk53PRuzFjUvv_6lLRWPoQ&s",
-            level="USE_CASE",
-            display_relations_names=True,
-            class_colors_mapping={},
-        )
-    )
-
-    # Remove graph
-    services.triple_store_service.remove(graph)
-
-    # Remove ontology
-    naas_integration = NaasIntegration(
-        NaasIntegrationConfiguration(api_key=secret.get("NAAS_API_KEY"))
-    )
-    result = naas_integration.delete_ontology(
-        workspace_id=config.workspace_id, ontology_id=ontology_id
-    )
-    logger.info(f"Removed ontology: {result}")
-    assert result is not None, result