naas-abi-cli 1.9.1__tar.gz → 1.10.0__tar.gz

naas_abi_cli-1.10.0/=0.9.13

@@ -0,0 +1,6 @@
+Collecting uv
+  Downloading uv-0.9.28-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (11 kB)
+Downloading uv-0.9.28-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (22.7 MB)
+   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 22.7/22.7 MB 103.6 MB/s  0:00:00
+Installing collected packages: uv
+Successfully installed uv-0.9.28

{naas_abi_cli-1.9.1 → naas_abi_cli-1.10.0}/CHANGELOG.md

@@ -2,6 +2,14 @@
 
 <!-- version list -->
 
+## v1.10.0 (2026-02-03)
+
+### Features
+
+- Working on Dagster orchestration
+  ([`b970d21`](https://github.com/jupyter-naas/abi/commit/b970d21e135ab32c344122be50f292c0b44ff039))
+
+
 ## v1.9.1 (2026-01-26)
 
 ### Bug Fixes

{naas_abi_cli-1.9.1 → naas_abi_cli-1.10.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: naas-abi-cli
-Version: 1.9.1
+Version: 1.10.0
 Summary: Abi cli allowing you to build your AI system.
 Project-URL: Homepage, https://github.com/jupyter-naas/abi
 Project-URL: Repository, https://github.com/jupyter-naas/abi/tree/main/libs/naas-abi-cli

naas_abi_cli-1.10.0/naas_abi_cli/cli/new/__init__.py

@@ -0,0 +1,9 @@
+import naas_abi_cli.cli.new.agent as agent  # noqa: F401
+import naas_abi_cli.cli.new.integration as integration  # noqa: F401
+import naas_abi_cli.cli.new.module as module  # noqa: F401
+import naas_abi_cli.cli.new.orchestration as orchestration  # noqa: F401
+import naas_abi_cli.cli.new.pipeline as pipeline  # noqa: F401
+import naas_abi_cli.cli.new.project as project  # noqa: F401
+import naas_abi_cli.cli.new.workflow as workflow  # noqa: F401
+
+from .new import new  # noqa: F401

naas_abi_cli-1.10.0/naas_abi_cli/cli/new/agent.py

@@ -0,0 +1,42 @@
+import os
+
+import click
+
+import naas_abi_cli
+from naas_abi_cli.cli.utils.Copier import Copier
+
+from .new import new
+from .utils import to_pascal_case
+
+
+@new.command("agent")
+@click.argument("agent_name", required=True)
+@click.argument("agent_path", required=False, default=".")
+def _new_agent(agent_name: str, agent_path: str = "."):
+    new_agent(agent_name, agent_path)
+
+
+def new_agent(
+    agent_name: str,
+    agent_path: str = ".",
+    extra_values: dict = {},
+):
+    agent_name = to_pascal_case(agent_name)
+
+    if agent_path == ".":
+        agent_path = os.getcwd()
+
+    if not os.path.exists(agent_path):
+        os.makedirs(agent_path, exist_ok=True)
+
+    copier = Copier(
+        templates_path=os.path.join(
+            os.path.dirname(naas_abi_cli.__file__), "cli/new/templates/agent"
+        ),
+        destination_path=agent_path,
+    )
+
+    if "module_name_snake" not in extra_values:
+        extra_values["module_name_snake"] = "your_module_name"
+
+    copier.copy(values={"agent_name_pascal": to_pascal_case(agent_name)} | extra_values)

naas_abi_cli-1.10.0/naas_abi_cli/cli/new/integration.py

@@ -0,0 +1,36 @@
+import os
+
+import click
+import naas_abi_cli
+from naas_abi_cli.cli.utils.Copier import Copier
+
+from .new import new
+from .utils import to_pascal_case
+
+
+@new.command("integration")
+@click.argument("integration_name", required=True)
+@click.argument("integration_path", required=False, default=".")
+def _new_integration(integration_name: str, integration_path: str = "."):
+    new_integration(integration_name, integration_path)
+
+
+def new_integration(integration_name: str, integration_path: str = ".", extra_values: dict = {}):
+    integration_name = to_pascal_case(integration_name)
+
+    if integration_path == ".":
+        integration_path = os.getcwd()
+
+    if not os.path.exists(integration_path):
+        os.makedirs(integration_path, exist_ok=True)
+
+    copier = Copier(
+        templates_path=os.path.join(
+            os.path.dirname(naas_abi_cli.__file__), "cli/new/templates/integration"
+        ),
+        destination_path=integration_path,
+    )
+
+    copier.copy(
+        values={"integration_name_pascal": to_pascal_case(integration_name)} | extra_values
+    )

{naas_abi_cli-1.9.1 → naas_abi_cli-1.10.0}/naas_abi_cli/cli/new/module.py

@@ -4,8 +4,13 @@ import click
 import naas_abi_cli
 from naas_abi_cli.cli.utils.Copier import Copier
 
+from .agent import new_agent
+from .integration import new_integration
 from .new import new
+from .orchestration import new_orchestration
+from .pipeline import new_pipeline
 from .utils import to_kebab_case, to_pascal_case, to_snake_case
+from .workflow import new_workflow
 
 
 @new.command("module")
@@ -44,6 +49,36 @@ def new_module(module_name: str, module_path: str = ".", quiet: bool = False):
         }
     )
 
+    new_agent(
+        module_name,
+        os.path.join(module_path, "agents"),
+        extra_values={"module_name_snake": to_snake_case(module_name)},
+    )
+
+    new_integration(
+        module_name,
+        os.path.join(module_path, "integrations"),
+        extra_values={"module_name_snake": to_snake_case(module_name)},
+    )
+    
+    new_pipeline(
+        module_name,
+        os.path.join(module_path, "pipelines"),
+        extra_values={"module_name_snake": to_snake_case(module_name)},
+    )
+    
+    new_workflow(
+        module_name,
+        os.path.join(module_path, "workflows"),
+        extra_values={"module_name_snake": to_snake_case(module_name)},
+    )
+    
+    new_orchestration(
+        module_name,
+        os.path.join(module_path, "orchestrations"),
+        extra_values={"module_name_snake": to_snake_case(module_name)},
+    )
+
     if not quiet:
         print(f"\nModule '{module_name}' has been created at:\n  {module_path}\n")
         print("To enable this module, add the following to your config.yaml:\n")

naas_abi_cli-1.10.0/naas_abi_cli/cli/new/orchestration.py

@@ -0,0 +1,36 @@
+import os
+
+import click
+import naas_abi_cli
+from naas_abi_cli.cli.utils.Copier import Copier
+
+from .new import new
+from .utils import to_pascal_case
+
+
+@new.command("orchestration")
+@click.argument("orchestration_name", required=True)
+@click.argument("orchestration_path", required=False, default=".")
+def _new_orchestration(orchestration_name: str, orchestration_path: str = "."):
+    new_orchestration(orchestration_name, orchestration_path)
+
+
+def new_orchestration(orchestration_name: str, orchestration_path: str = ".", extra_values: dict = {}):
+    orchestration_name = to_pascal_case(orchestration_name)
+
+    if orchestration_path == ".":
+        orchestration_path = os.getcwd()
+
+    if not os.path.exists(orchestration_path):
+        os.makedirs(orchestration_path, exist_ok=True)
+
+    copier = Copier(
+        templates_path=os.path.join(
+            os.path.dirname(naas_abi_cli.__file__), "cli/new/templates/orchestration"
+        ),
+        destination_path=orchestration_path,
+    )
+
+    copier.copy(
+        values={"orchestration_name_pascal": to_pascal_case(orchestration_name)} | extra_values
+    )

naas_abi_cli-1.10.0/naas_abi_cli/cli/new/pipeline.py

@@ -0,0 +1,36 @@
+import os
+
+import click
+import naas_abi_cli
+from naas_abi_cli.cli.utils.Copier import Copier
+
+from .new import new
+from .utils import to_pascal_case
+
+
+@new.command("pipeline")
+@click.argument("pipeline_name", required=True)
+@click.argument("pipeline_path", required=False, default=".")
+def _new_pipeline(pipeline_name: str, pipeline_path: str = "."):
+    new_pipeline(pipeline_name, pipeline_path)
+
+
+def new_pipeline(pipeline_name: str, pipeline_path: str = ".", extra_values: dict = {}):
+    pipeline_name = to_pascal_case(pipeline_name)
+
+    if pipeline_path == ".":
+        pipeline_path = os.getcwd()
+
+    if not os.path.exists(pipeline_path):
+        os.makedirs(pipeline_path, exist_ok=True)
+
+    copier = Copier(
+        templates_path=os.path.join(
+            os.path.dirname(naas_abi_cli.__file__), "cli/new/templates/pipeline"
+        ),
+        destination_path=pipeline_path,
+    )
+
+    copier.copy(
+        values={"pipeline_name_pascal": to_pascal_case(pipeline_name)} | extra_values
+    )

naas_abi_cli-1.10.0/naas_abi_cli/cli/new/templates/agent/{{agent_name_pascal}}Agent.py

@@ -0,0 +1,47 @@
+from typing import Optional
+
+from naas_abi_core.services.agent.Agent import (
+    Agent,
+    AgentConfiguration,
+    AgentSharedState,
+)
+
+NAME = "{{agent_name_pascal}}Agent"
+DESCRIPTION = "An helpful agent that can help you with your tasks."
+SYSTEM_PROMPT = """
+You are {{agent_name_pascal}}Agent.
+"""
+
+class {{agent_name_pascal}}Agent(Agent):
+
+    @staticmethod
+    def New(cls, agent_shared_state: Optional[AgentSharedState] = None, agent_configuration: Optional[AgentConfiguration] = None) -> "{{agent_name_pascal}}Agent":
+        #from {{module_name_snake}} import ABIModule
+        
+        # Set model
+        from naas_abi_marketplace.ai.chatgpt.models.gpt_5 import model as chatgpt_model
+
+        model = chatgpt_model.model
+
+        # Use provided configuration or create default one
+        if agent_configuration is None:
+            agent_configuration = AgentConfiguration(system_prompt=SYSTEM_PROMPT)
+
+        # Use provided shared state or create new one
+        if agent_shared_state is None:
+            agent_shared_state = AgentSharedState()
+
+        tools: list = []
+
+        agents: list = []
+
+        return cls(
+            name=NAME,
+            description=DESCRIPTION,
+            chat_model=model,
+            tools=tools,
+            agents=agents,
+            memory=None,
+            state=agent_shared_state,
+            configuration=agent_configuration,
+        )

naas_abi_cli-1.10.0/naas_abi_cli/cli/new/templates/integration/{{integration_name_pascal}}Integration.py

@@ -0,0 +1,30 @@
+import hashlib
+import re
+from dataclasses import dataclass
+from datetime import date, datetime, timedelta
+from typing import Dict, List, Optional
+
+import requests
+from naas_abi_core import logger
+from naas_abi_core.integration import Integration, IntegrationConfiguration
+from naas_abi_core.integration.integration import IntegrationConnectionError
+from naas_abi_core.services.cache.CacheFactory import CacheFactory
+from naas_abi_core.services.cache.CachePort import DataType
+from naas_abi_core.services.cache.CacheService import CacheService
+
+
+@dataclass
+class {{integration_name_pascal}}IntegrationConfiguration(IntegrationConfiguration):
+    """Configuration for the {{integration_name_pascal}} integration."""
+    pass
+
+
+class {{integration_name_pascal}}Integration(Integration):
+    """{{integration_name_pascal}} integration."""
+
+    __configuration: {{integration_name_pascal}}IntegrationConfiguration
+
+    def __init__(self, configuration: {{integration_name_pascal}}IntegrationConfiguration):
+        super().__init__(configuration)
+        self.__configuration = configuration
+

naas_abi_cli-1.10.0/naas_abi_cli/cli/new/templates/orchestration/{{orchestration_name_pascal}}Orchestration.py

@@ -0,0 +1,16 @@
+import dagster as dg
+from naas_abi_core.orchestrations.DagsterOrchestration import DagsterOrchestration
+
+
+class {{orchestration_name_pascal}}Orchestration(DagsterOrchestration):
+
+    @classmethod
+    def New(cls) -> "{{orchestration_name_pascal}}Orchestration":
+        return cls(
+            definitions=dg.Definitions(
+                assets=[],
+                schedules=[],
+                jobs=[],
+                sensors=[],
+            )
+        )

naas_abi_cli-1.10.0/naas_abi_cli/cli/new/templates/pipeline/{{pipeline_name_pascal}}Pipeline.py

@@ -0,0 +1,69 @@
+import uuid
+from dataclasses import dataclass
+from enum import Enum
+from typing import Annotated, Optional
+
+from langchain_core.tools import BaseTool, StructuredTool
+from naas_abi_core import logger
+from naas_abi_core.pipeline import (Pipeline, PipelineConfiguration,
+                                    PipelineParameters)
+from naas_abi_core.utils.Expose import APIRouter
+from pydantic import Field
+from rdflib import DCTERMS, OWL, RDF, RDFS, Graph, Literal, Namespace, URIRef
+
+
+@dataclass
+class {{pipeline_name_pascal}}PipelineConfiguration(PipelineConfiguration):
+    """Configuration for {{pipeline_name_pascal}}Pipeline."""
+
+    pass
+
+
+class {{pipeline_name_pascal}}PipelineParameters(PipelineParameters):
+    # example_parameter: Annotated[
+    #     str,
+    #     Field(
+    #         description="Description of the example parameter",
+    #         example="Example value",
+    #     ),
+    # ]
+    pass
+
+
+class {{pipeline_name_pascal}}Pipeline(Pipeline[{{pipeline_name_pascal}}PipelineParameters]):
+    """Pipeline for adding a named individual."""
+
+    __configuration: {{pipeline_name_pascal}}PipelineConfiguration
+
+    def __init__(self, configuration: {{pipeline_name_pascal}}PipelineConfiguration):
+        super().__init__(configuration)
+        self.__configuration = configuration
+
+    def run(self, parameters: {{pipeline_name_pascal}}PipelineParameters) -> Graph:
+        # Implement the pipeline logic here.
+        pass
+
+    def as_tools(self) -> list[BaseTool]:
+        return [
+            StructuredTool(
+                name="{{pipeline_name_pascal}}",
+                description="Description of what the pipeline does",
+                func=lambda **kwargs: self.run(
+                    {{pipeline_name_pascal}}PipelineParameters(**kwargs)
+                ),
+                args_schema={{pipeline_name_pascal}}PipelineParameters,
+            )
+        ]
+
+    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_cli-1.10.0/naas_abi_cli/cli/new/templates/workflow/{{workflow_name_pascal}}Workflow.py

@@ -0,0 +1,73 @@
+from dataclasses import dataclass
+from enum import Enum
+from typing import Annotated, Optional
+
+from langchain_core.tools import BaseTool, StructuredTool
+from naas_abi_core.services.triple_store.TripleStorePorts import ITripleStoreService
+from naas_abi_core.utils.Expose import APIRouter
+from naas_abi_core.workflow import Workflow, WorkflowConfiguration
+from naas_abi_core.workflow.workflow import WorkflowParameters
+from pydantic import Field
+
+@dataclass
+class {{workflow_name_pascal}}WorkflowConfiguration(WorkflowConfiguration):
+    """Configuration for {{workflow_name_pascal}} workflow."""
+
+
+    pass
+
+class {{workflow_name_pascal}}WorkflowParameters(WorkflowParameters):
+    """Parameters for {{workflow_name_pascal}} workflow."""
+
+    pass
+
+    # example_parameter: Annotated[
+    #     str,
+    #     Field(
+    #         ...,
+    #         description="Description of the example parameter",
+    #         example="Example value",
+    #     ),
+    # ]
+
+
+class {{workflow_name_pascal}}Workflow(Workflow[{{workflow_name_pascal}}WorkflowParameters]):
+    """Workflow for {{workflow_name_pascal}}."""
+
+    __configuration: {{workflow_name_pascal}}WorkflowConfiguration
+
+    def __init__(self, configuration: {{workflow_name_pascal}}WorkflowConfiguration):
+        super().__init__(configuration)
+        self.__configuration = configuration
+
+    def run(
+        self, parameters: {{workflow_name_pascal}}WorkflowParameters
+    ) -> dict | list[dict]:
+        # Implement the workflow logic here
+        pass
+
+
+    def as_tools(self) -> list[BaseTool]:
+        return [
+            StructuredTool(
+                name="{{workflow_name_pascal}}",
+                description="Description of what the workflow does",
+                func=lambda **kwargs: self.run(
+                    {{workflow_name_pascal}}WorkflowParameters(**kwargs)
+                ),
+                args_schema={{workflow_name_pascal}}WorkflowParameters,
+            )
+        ]
+
+    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_cli-1.10.0/naas_abi_cli/cli/new/workflow.py

@@ -0,0 +1,36 @@
+import os
+
+import click
+import naas_abi_cli
+from naas_abi_cli.cli.utils.Copier import Copier
+
+from .new import new
+from .utils import to_pascal_case
+
+
+@new.command("workflow")
+@click.argument("workflow_name", required=True)
+@click.argument("workflow_path", required=False, default=".")
+def _new_workflow(workflow_name: str, workflow_path: str = "."):
+    new_workflow(workflow_name, workflow_path)
+
+
+def new_workflow(workflow_name: str, workflow_path: str = ".", extra_values: dict = {}):
+    workflow_name = to_pascal_case(workflow_name)
+
+    if workflow_path == ".":
+        workflow_path = os.getcwd()
+
+    if not os.path.exists(workflow_path):
+        os.makedirs(workflow_path, exist_ok=True)
+
+    copier = Copier(
+        templates_path=os.path.join(
+            os.path.dirname(naas_abi_cli.__file__), "cli/new/templates/workflow"
+        ),
+        destination_path=workflow_path,
+    )
+
+    copier.copy(
+        values={"workflow_name_pascal": to_pascal_case(workflow_name)} | extra_values
+    )

{naas_abi_cli-1.9.1 → naas_abi_cli-1.10.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "naas-abi-cli"
-version = "1.9.1"
+version = "1.10.0"
 description = "Abi cli allowing you to build your AI system."
 authors = [{ name = "Maxime Jublou", email = "maxime@naas.ai" },{ name = "Florent Ravenel", email = "florent@naas.ai" }, { name = "Jeremy Ravenel", email = "jeremy@naas.ai" }]
 requires-python = ">=3.10,<4"

naas_abi_cli-1.9.1/=0.9.13

@@ -1,5 +0,0 @@
-Collecting uv
-  Using cached uv-0.9.26-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (11 kB)
-Using cached uv-0.9.26-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (23.3 MB)
-Installing collected packages: uv
-Successfully installed uv-0.9.26

naas_abi_cli-1.9.1/naas_abi_cli/cli/new/__init__.py

@@ -1,4 +0,0 @@
-import naas_abi_cli.cli.new.module as module  # noqa: F401
-import naas_abi_cli.cli.new.project as project  # noqa: F401
-
-from .new import new  # noqa: F401

naas_abi_cli-1.9.1/naas_abi_cli/cli/new/templates/module/agents/{{module_name_pascal}}Agent.py

@@ -1,52 +0,0 @@
-from typing import Optional
-
-from naas_abi_core.services.agent.Agent import (
-    Agent,
-    AgentConfiguration,
-    AgentSharedState,
-)
-
-NAME = "{{module_name_pascal}}Agent"
-DESCRIPTION = "An helpful agent that can help you with your tasks."
-SYSTEM_PROMPT = """
-You are {{module_name_pascal}}Agent.
-"""
-
-
-def create_agent(
-    agent_shared_state: Optional[AgentSharedState] = None,
-    agent_configuration: Optional[AgentConfiguration] = None,
-) -> Optional[Agent]:
-    #from {{module_name_snake}} import ABIModule
-    
-    # Set model
-    from naas_abi_marketplace.ai.chatgpt.models.gpt_5 import model as chatgpt_model
-
-    model = chatgpt_model.model
-
-    # Use provided configuration or create default one
-    if agent_configuration is None:
-        agent_configuration = AgentConfiguration(system_prompt=SYSTEM_PROMPT)
-
-    # Use provided shared state or create new one
-    if agent_shared_state is None:
-        agent_shared_state = AgentSharedState()
-
-    tools: list = []
-
-    agents: list = []
-
-    return {{module_name_pascal}}Agent(
-        name=NAME,
-        description=DESCRIPTION,
-        chat_model=model,
-        tools=tools,
-        agents=agents,
-        memory=None,
-        state=agent_shared_state,
-        configuration=agent_configuration,
-    )
-
-
-class {{module_name_pascal}}Agent(Agent):
-    pass