blue-assistant 4.46.1__tar.gz → 4.62.1__tar.gz

{blue_assistant-4.46.1/blue_assistant.egg-info → blue_assistant-4.62.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: blue_assistant
-Version: 4.46.1
+Version: 4.62.1
 Summary: 🧠 An AI Assistant.
 Home-page: https://github.com/kamangir/blue-assistant
 Author: Arash Abadpour (Kamangir)
@@ -16,7 +16,9 @@ Requires-Dist: blueness
 Requires-Dist: blue-options
 Requires-Dist: abcli
 Requires-Dist: openai_commands
+Requires-Dist: blueflow
 Requires-Dist: boto3
+Requires-Dist: openai
 Requires-Dist: geojson
 Requires-Dist: geopandas
 Requires-Dist: matplotlib
@@ -28,6 +30,8 @@ Requires-Dist: pylint
 Requires-Dist: pytest
 Requires-Dist: python-dotenv[cli]
 Requires-Dist: tqdm
+Requires-Dist: networkx
+Requires-Dist: pydot
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier
@@ -48,11 +52,14 @@ pip install blue-assistant
 
 ```mermaid
 graph LR
+    assistant_script_list["@assistant<br>script<br>list"]
     assistant_script_run["@assistant<br>script<br>run -<br>&lt;script&gt;<br>&lt;object-name&gt;"]
 
     script["📜 script"]:::folder
     object["📂 object"]:::folder
 
+    script --> assistant_script_list
+
     script --> assistant_script_run
     object --> assistant_script_run
     assistant_script_run --> object
@@ -65,4 +72,4 @@ graph LR
 
 [![pylint](https://github.com/kamangir/blue-assistant/actions/workflows/pylint.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/pylint.yml) [![pytest](https://github.com/kamangir/blue-assistant/actions/workflows/pytest.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/pytest.yml) [![bashtest](https://github.com/kamangir/blue-assistant/actions/workflows/bashtest.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/bashtest.yml) [![PyPI version](https://img.shields.io/pypi/v/blue-assistant.svg)](https://pypi.org/project/blue-assistant/) [![PyPI - Downloads](https://img.shields.io/pypi/dd/blue-assistant)](https://pypistats.org/packages/blue-assistant)
 
-built by 🌀 [`blue_options-4.207.1`](https://github.com/kamangir/awesome-bash-cli), based on 🧠 [`blue_assistant-4.46.1`](https://github.com/kamangir/blue-assistant).
+built by 🌀 [`blue_options-4.207.1`](https://github.com/kamangir/awesome-bash-cli), based on 🧠 [`blue_assistant-4.62.1`](https://github.com/kamangir/blue-assistant).

{blue_assistant-4.46.1 → blue_assistant-4.62.1}/README.md

@@ -8,11 +8,14 @@ pip install blue-assistant
 
 ```mermaid
 graph LR
+    assistant_script_list["@assistant<br>script<br>list"]
     assistant_script_run["@assistant<br>script<br>run -<br>&lt;script&gt;<br>&lt;object-name&gt;"]
 
     script["📜 script"]:::folder
     object["📂 object"]:::folder
 
+    script --> assistant_script_list
+
     script --> assistant_script_run
     object --> assistant_script_run
     assistant_script_run --> object
@@ -25,4 +28,4 @@ graph LR
 
 [![pylint](https://github.com/kamangir/blue-assistant/actions/workflows/pylint.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/pylint.yml) [![pytest](https://github.com/kamangir/blue-assistant/actions/workflows/pytest.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/pytest.yml) [![bashtest](https://github.com/kamangir/blue-assistant/actions/workflows/bashtest.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/bashtest.yml) [![PyPI version](https://img.shields.io/pypi/v/blue-assistant.svg)](https://pypi.org/project/blue-assistant/) [![PyPI - Downloads](https://img.shields.io/pypi/dd/blue-assistant)](https://pypistats.org/packages/blue-assistant)
 
-built by 🌀 [`blue_options-4.207.1`](https://github.com/kamangir/awesome-bash-cli), based on 🧠 [`blue_assistant-4.46.1`](https://github.com/kamangir/blue-assistant).
+built by 🌀 [`blue_options-4.207.1`](https://github.com/kamangir/awesome-bash-cli), based on 🧠 [`blue_assistant-4.62.1`](https://github.com/kamangir/blue-assistant).

blue_assistant-4.62.1/blue_assistant/.abcli/script/list.sh

@@ -0,0 +1,7 @@
+#! /usr/bin/env bash
+
+function blue_assistant_script_list() {
+    python3 -m blue_assistant.script \
+        list \
+        "$@"
+}

{blue_assistant-4.46.1 → blue_assistant-4.62.1}/blue_assistant/.abcli/tests/help.sh

@@ -20,6 +20,7 @@ function test_blue_assistant_help() {
         "@assistant browse" \
         \
         "@assistant script" \
+        "@assistant script list" \
         "@assistant script run" \
         \
         "blue_assistant"; do

blue_assistant-4.62.1/blue_assistant/.abcli/tests/script_list.sh

@@ -0,0 +1,8 @@
+#! /usr/bin/env bash
+
+function test_blue_assistant_script_list() {
+    abcli_assert "$(blue_assistant_script_list \
+        --delim + \
+        --log 0)" \
+        - non-empty
+}

{blue_assistant-4.46.1 → blue_assistant-4.62.1}/blue_assistant/.abcli/tests/script_run.sh

@@ -2,9 +2,9 @@
 
 function test_blue_assistant_script_run() {
     local options=$1
-    local list_of_script_name=$(python3 -m blue_assistant.script \
-        get_list_of \
-        --delim +)
+    local list_of_script_name=$(blue_assistant_script_list \
+        --delim + \
+        --log 0)
     list_of_script_name=$(abcli_option "$options" script $list_of_script_name)
 
     local script_name

{blue_assistant-4.46.1 → blue_assistant-4.62.1}/blue_assistant/__init__.py

@@ -4,7 +4,7 @@ ICON = "🧠"
 
 DESCRIPTION = f"{ICON} An AI Assistant."
 
-VERSION = "4.46.1"
+VERSION = "4.62.1"
 
 REPO_NAME = "blue-assistant"
 

blue_assistant-4.62.1/blue_assistant/config.env

@@ -0,0 +1,3 @@
+BLUE_ASSISTANT_DEFAULT_MODEL=gpt-4o
+
+BLUE_ASSISTANT_MAX_TOKEN=2000

blue_assistant-4.62.1/blue_assistant/env.py

@@ -0,0 +1,17 @@
+import os
+from blue_options.env import load_config, load_env
+
+load_env(__name__)
+load_config(__name__)
+
+
+BLUE_ASSISTANT_DEFAULT_MODEL = os.getenv(
+    "BLUE_ASSISTANT_DEFAULT_MODEL",
+    "",
+)
+
+BLUE_ASSISTANT_MAX_TOKEN_str = os.getenv("BLUE_ASSISTANT_MAX_TOKEN", "")
+try:
+    BLUE_ASSISTANT_MAX_TOKEN = int(BLUE_ASSISTANT_MAX_TOKEN_str)
+except Exception:
+    BLUE_ASSISTANT_MAX_TOKEN = 2000

{blue_assistant-4.46.1 → blue_assistant-4.62.1}/blue_assistant/help/script.py

@@ -5,6 +5,27 @@ from blue_options.terminal import show_usage, xtra
 from blue_assistant.script.repository import list_of_script_names
 
 
+def help_list(
+    tokens: List[str],
+    mono: bool,
+) -> str:
+    args = [
+        "[--delim +]",
+        "[--log 0]",
+    ]
+
+    return show_usage(
+        [
+            "@assistant",
+            "script",
+            "list",
+        ]
+        + args,
+        "list scripts.",
+        mono=mono,
+    )
+
+
 def help_run(
     tokens: List[str],
     mono: bool,
@@ -33,4 +54,7 @@ def help_run(
     )
 
 
-help_functions = {"run": help_run}
+help_functions = {
+    "list": help_list,
+    "run": help_run,
+}

{blue_assistant-4.46.1 → blue_assistant-4.62.1}/blue_assistant/script/__main__.py

@@ -14,7 +14,7 @@ parser = argparse.ArgumentParser(NAME)
 parser.add_argument(
     "task",
     type=str,
-    help="get_list_of | run",
+    help="list | run",
 )
 parser.add_argument(
     "--script_name",
@@ -34,24 +34,36 @@ parser.add_argument(
 parser.add_argument(
     "--delim",
     type=str,
-    default="+",
+    default=", ",
+)
+parser.add_argument(
+    "--log",
+    type=int,
+    default=1,
+    help="0 | 1",
 )
 args = parser.parse_args()
 
 delim = " " if args.delim == "space" else args.delim
 
 success = False
-if args.task == "get_list_of":
+if args.task == "list":
     success = True
-    print(delim.join(list_of_script_names))
+    if args.log:
+        logger.info(f"{len(list_of_script_names)} script(s)")
+        for index, script_name in enumerate(list_of_script_names):
+            logger.info(f"#{index + 1: 3d}: {script_name}")
+    else:
+        print(delim.join(list_of_script_names))
 elif args.task == "run":
     success, script = load_script(
         script_name=args.script_name,
+        object_name=args.object_name,
         verbose=args.verbose == 1,
     )
 
     if success:
-        success = script.run(object_name=args.object_name)
+        success = script.run()
 else:
     success = None
 

blue_assistant-4.62.1/blue_assistant/script/actions/__init__.py

@@ -0,0 +1,14 @@
+from typing import List
+
+from blue_assistant.script.actions.generic import GenericAction
+from blue_assistant.script.actions.generate_image import GenerateImageAction
+from blue_assistant.script.actions.generate_text import GenerateTextAction
+from blue_assistant.script.actions.wip import WorkInProgressAction
+from blue_assistant.logger import logger
+
+list_of_actions: List[GenericAction] = [
+    GenericAction,
+    GenerateImageAction,
+    GenerateTextAction,
+    WorkInProgressAction,
+]

blue_assistant-4.62.1/blue_assistant/script/actions/functions.py

@@ -0,0 +1,52 @@
+from typing import List, Dict, Tuple, Type
+
+from blueness import module
+from blue_assistant.script.actions.generic import GenericAction
+from blue_assistant.script.repository.base.classes import BaseScript
+from blue_assistant.script.actions import list_of_actions
+
+from blue_assistant import NAME
+from blue_assistant.logger import logger
+
+NAME = module.name(__file__, NAME)
+
+
+def get_action_class(
+    action_name: str,
+) -> Tuple[bool, Type[GenericAction]]:
+    for action_class in list_of_actions:
+        if action_class.name == action_name:
+            return True, action_class
+
+    logger.error(f"{action_name}: action not found.")
+    return False, GenericAction
+
+
+def perform_action(
+    script: BaseScript,
+    node_name: str,
+) -> Tuple[bool, Dict]:
+    action_name = script.nodes[node_name].get("action", "unknown")
+
+    success, action_class = get_action_class(action_name=action_name)
+    if not success:
+        return False, {
+            "error": f"{action_name}: action not found.",
+        }
+
+    logger.info(
+        "{}.perform_action: {} == {} on {}".format(
+            NAME,
+            action_name,
+            action_class.__name__,
+            node_name,
+        )
+    )
+
+    action_object = action_class(
+        script=script,
+    )
+
+    return action_object.perform(
+        node_name=node_name,
+    )

blue_assistant-4.62.1/blue_assistant/script/actions/generate_image.py

@@ -0,0 +1,24 @@
+from typing import Dict, Tuple
+
+from blue_objects import file
+
+from blue_assistant.script.actions.generic import GenericAction
+from blue_assistant.logger import logger
+
+
+class GenerateImageAction(GenericAction):
+    name = file.name(__file__)
+
+    def perform(
+        self,
+        node_name: str,
+    ) -> Tuple[bool, Dict]:
+        success, generic_metadata = super().perform(node_name=node_name)
+        if not success:
+            return success, generic_metadata
+
+        logger.info(f"🪄 generating image ...: {node_name}")
+        metadata = {}
+
+        metadata.update(generic_metadata)
+        return True, metadata

blue_assistant-4.62.1/blue_assistant/script/actions/generate_text.py

@@ -0,0 +1,66 @@
+from typing import Dict, Tuple
+from openai import OpenAI
+
+from blueness import module
+from blue_objects import file
+from openai_commands.env import OPENAI_API_KEY
+
+from blue_assistant import NAME
+from blue_assistant.env import BLUE_ASSISTANT_DEFAULT_MODEL, BLUE_ASSISTANT_MAX_TOKEN
+from blue_assistant.script.actions.generic import GenericAction
+from blue_assistant.logger import logger
+
+NAME = module.name(__file__, NAME)
+
+
+class GenerateTextAction(GenericAction):
+    name = file.name(__file__)
+
+    def perform(
+        self,
+        node_name: str,
+    ) -> Tuple[bool, Dict]:
+        metadata = {}
+
+        if not OPENAI_API_KEY:
+            logger.error("OPENAI_API_KEY is not set.")
+            return False, {}
+
+        success, generic_metadata = super().perform(node_name=node_name)
+        if not success:
+            return success, generic_metadata
+
+        client = OpenAI(api_key=OPENAI_API_KEY)
+
+        try:
+            response = client.chat.completions.create(
+                messages=[
+                    {
+                        "role": "user",
+                        "content": [
+                            {
+                                "type": "text",
+                                "text": self.script.nodes[node_name]["prompt"],
+                            }
+                        ],
+                    }
+                ],
+                model=BLUE_ASSISTANT_DEFAULT_MODEL,
+                max_tokens=BLUE_ASSISTANT_MAX_TOKEN,
+            )
+        except Exception as e:
+            logger.error(str(e))
+            return False, {"error": str(e)}
+
+        if self.script.verbose:
+            logger.info("response: {}".format(response))
+
+        if not response.choices:
+            logger.error("no choice.")
+            return False, {}
+
+        metadata["reply"] = response.choices[0].message.content
+        logger.info("🗣️ reply: {}".format(metadata["reply"]))
+
+        metadata.update(generic_metadata)
+        return True, metadata

blue_assistant-4.62.1/blue_assistant/script/actions/generic.py

@@ -0,0 +1,34 @@
+from typing import Dict, Tuple
+
+from blueness import module
+from blue_objects import file
+
+from blue_assistant import NAME
+from blue_assistant.script.repository.base.classes import BaseScript
+from blue_assistant.logger import logger
+
+NAME = module.name(__file__, NAME)
+
+
+class GenericAction:
+    name = file.name(__file__)
+
+    def __init__(
+        self,
+        script: BaseScript,
+    ):
+        self.script = script
+
+    def perform(
+        self,
+        node_name: str,
+    ) -> Tuple[bool, Dict]:
+        logger.info(
+            "{}.perform({}) on {}.{} ...".format(
+                NAME,
+                self.__class__.__name__,
+                self.script.name,
+                node_name,
+            ),
+        )
+        return True, {}

blue_assistant-4.62.1/blue_assistant/script/actions/wip.py

@@ -0,0 +1,7 @@
+from blue_objects import file
+
+from blue_assistant.script.actions.generic import GenericAction
+
+
+class WorkInProgressAction(GenericAction):
+    name = file.name(__file__)

blue_assistant-4.62.1/blue_assistant/script/load.py

@@ -0,0 +1,27 @@
+from typing import Tuple, Type
+
+from blue_assistant.script.repository import list_of_script_classes
+from blue_assistant.script.repository.base.classes import BaseScript
+from blue_assistant.logger import logger
+
+
+def load_script(
+    script_name: str,
+    object_name: str,
+    verbose: bool = False,
+) -> Tuple[bool, BaseScript]:
+    found: bool = False
+    script_class: Type[BaseScript] = BaseScript
+    for script_class_option in list_of_script_classes:
+        if script_class_option.name == script_name:
+            found = True
+            script_class = script_class_option
+            break
+
+    if not found:
+        logger.error(f"{script_name}: script not found.")
+
+    return found, script_class(
+        object_name=object_name,
+        verbose=verbose,
+    )

{blue_assistant-4.46.1 → blue_assistant-4.62.1}/blue_assistant/script/repository/__init__.py

@@ -1,12 +1,14 @@
 from typing import List, Type
 
-from blue_assistant.script.repository.generic import GenericScript
+from blue_assistant.script.repository.generic.classes import GenericScript
 from blue_assistant.script.repository.blue_amo.classes import BlueAmoScript
+from blue_assistant.script.repository.hue.classes import HueScript
 from blue_assistant.script.repository.moon_datasets.classes import MiningOnMoonScript
 
 list_of_script_classes: List[Type[GenericScript]] = [
     GenericScript,
     BlueAmoScript,
+    HueScript,
     MiningOnMoonScript,
 ]
 

blue_assistant-4.62.1/blue_assistant/script/repository/blue_amo/classes.py

@@ -0,0 +1,9 @@
+from blue_objects import file, path
+
+
+from blue_assistant.script.repository.generic.classes import GenericScript
+from blue_assistant.logger import logger
+
+
+class BlueAmoScript(GenericScript):
+    name = path.name(file.path(__file__))

blue_assistant-4.62.1/blue_assistant/script/repository/generic/classes.py

@@ -0,0 +1,72 @@
+from typing import Dict, List
+import os
+from tqdm import tqdm
+
+
+from blueness import module
+from blue_objects import file, path
+from blue_objects.metadata import post_to_object
+
+from blue_assistant import NAME
+from blue_assistant.script.repository.base.classes import BaseScript
+from blue_assistant.script.actions.functions import perform_action
+from blue_assistant.logger import logger
+
+
+NAME = module.name(__file__, NAME)
+
+
+class GenericScript(BaseScript):
+    name = path.name(file.path(__file__))
+
+    def run(
+        self,
+    ) -> bool:
+        if not super().run():
+            return False
+
+        metadata: Dict[Dict] = {"nodes": {}}
+        success: bool = True
+        while (
+            not all(self.G.nodes[node]["completed"] for node in self.G.nodes)
+            and success
+        ):
+            for node_name in tqdm(self.G.nodes):
+                if self.G.nodes[node_name]["completed"]:
+                    continue
+
+                pending_dependencies = [
+                    node_name_
+                    for node_name_ in self.G.successors(node_name)
+                    if not self.G.nodes[node_name_]["completed"]
+                ]
+                if pending_dependencies:
+                    logger.info(
+                        'node "{}": {} pending dependenci(es): {}'.format(
+                            node_name,
+                            len(pending_dependencies),
+                            ", ".join(pending_dependencies),
+                        )
+                    )
+                    continue
+
+                success, output = perform_action(
+                    script=self,
+                    node_name=node_name,
+                )
+                self.G.nodes[node_name]["completed"] = success
+                metadata["nodes"][node_name] = {
+                    "success": success,
+                    "output": output,
+                }
+                if not success:
+                    break
+
+        if not post_to_object(
+            self.object_name,
+            "output",
+            metadata,
+        ):
+            return False
+
+        return success

{blue_assistant-4.46.1 → blue_assistant-4.62.1}/blue_assistant/script/repository/moon_datasets/classes.py

@@ -1,6 +1,3 @@
-from typing import Dict
-from tqdm import tqdm
-
 from blue_objects import file, path
 
 from blue_assistant.script.repository.generic.classes import GenericScript
@@ -8,9 +5,3 @@ from blue_assistant.script.repository.generic.classes import GenericScript
 
 class MiningOnMoonScript(GenericScript):
     name = path.name(file.path(__file__))
-
-    def __init__(
-        self,
-        verbose: bool = False,
-    ):
-        super().__init__(verbose=verbose)

{blue_assistant-4.46.1 → blue_assistant-4.62.1/blue_assistant.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: blue_assistant
-Version: 4.46.1
+Version: 4.62.1
 Summary: 🧠 An AI Assistant.
 Home-page: https://github.com/kamangir/blue-assistant
 Author: Arash Abadpour (Kamangir)
@@ -16,7 +16,9 @@ Requires-Dist: blueness
 Requires-Dist: blue-options
 Requires-Dist: abcli
 Requires-Dist: openai_commands
+Requires-Dist: blueflow
 Requires-Dist: boto3
+Requires-Dist: openai
 Requires-Dist: geojson
 Requires-Dist: geopandas
 Requires-Dist: matplotlib
@@ -28,6 +30,8 @@ Requires-Dist: pylint
 Requires-Dist: pytest
 Requires-Dist: python-dotenv[cli]
 Requires-Dist: tqdm
+Requires-Dist: networkx
+Requires-Dist: pydot
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier
@@ -48,11 +52,14 @@ pip install blue-assistant
 
 ```mermaid
 graph LR
+    assistant_script_list["@assistant<br>script<br>list"]
     assistant_script_run["@assistant<br>script<br>run -<br>&lt;script&gt;<br>&lt;object-name&gt;"]
 
     script["📜 script"]:::folder
     object["📂 object"]:::folder
 
+    script --> assistant_script_list
+
     script --> assistant_script_run
     object --> assistant_script_run
     assistant_script_run --> object
@@ -65,4 +72,4 @@ graph LR
 
 [![pylint](https://github.com/kamangir/blue-assistant/actions/workflows/pylint.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/pylint.yml) [![pytest](https://github.com/kamangir/blue-assistant/actions/workflows/pytest.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/pytest.yml) [![bashtest](https://github.com/kamangir/blue-assistant/actions/workflows/bashtest.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/bashtest.yml) [![PyPI version](https://img.shields.io/pypi/v/blue-assistant.svg)](https://pypi.org/project/blue-assistant/) [![PyPI - Downloads](https://img.shields.io/pypi/dd/blue-assistant)](https://pypistats.org/packages/blue-assistant)
 
-built by 🌀 [`blue_options-4.207.1`](https://github.com/kamangir/awesome-bash-cli), based on 🧠 [`blue_assistant-4.46.1`](https://github.com/kamangir/blue-assistant).
+built by 🌀 [`blue_options-4.207.1`](https://github.com/kamangir/awesome-bash-cli), based on 🧠 [`blue_assistant-4.62.1`](https://github.com/kamangir/blue-assistant).

{blue_assistant-4.46.1 → blue_assistant-4.62.1}/blue_assistant.egg-info/SOURCES.txt

@@ -25,9 +25,11 @@ blue_assistant/.abcli/alias.sh
 blue_assistant/.abcli/blue_assistant.sh
 blue_assistant/.abcli/browse.sh
 blue_assistant/.abcli/script.sh
+blue_assistant/.abcli/script/list.sh
 blue_assistant/.abcli/script/run.sh
 blue_assistant/.abcli/tests/README.sh
 blue_assistant/.abcli/tests/help.sh
+blue_assistant/.abcli/tests/script_list.sh
 blue_assistant/.abcli/tests/script_run.sh
 blue_assistant/.abcli/tests/version.sh
 blue_assistant/help/__init__.py
@@ -37,6 +39,12 @@ blue_assistant/help/script.py
 blue_assistant/script/__init__.py
 blue_assistant/script/__main__.py
 blue_assistant/script/load.py
+blue_assistant/script/actions/__init__.py
+blue_assistant/script/actions/functions.py
+blue_assistant/script/actions/generate_image.py
+blue_assistant/script/actions/generate_text.py
+blue_assistant/script/actions/generic.py
+blue_assistant/script/actions/wip.py
 blue_assistant/script/repository/__init__.py
 blue_assistant/script/repository/blue_amo/__init__.py
 blue_assistant/script/repository/blue_amo/classes.py

{blue_assistant-4.46.1 → blue_assistant-4.62.1}/blue_assistant.egg-info/requires.txt

@@ -2,7 +2,9 @@ blueness
 blue-options
 abcli
 openai_commands
+blueflow
 boto3
+openai
 geojson
 geopandas
 matplotlib
@@ -14,3 +16,5 @@ pylint
 pytest
 python-dotenv[cli]
 tqdm
+networkx
+pydot

{blue_assistant-4.46.1 → blue_assistant-4.62.1}/requirements.txt

@@ -2,7 +2,10 @@ blueness
 blue-options
 abcli
 openai_commands
+blueflow
+
 boto3
+openai
 geojson
 geopandas
 matplotlib
@@ -14,3 +17,6 @@ pylint
 pytest
 python-dotenv[cli]
 tqdm
+
+networkx
+pydot

{blue_assistant-4.46.1 → blue_assistant-4.62.1}/setup.py

@@ -11,6 +11,7 @@ setup(
         NAME,
         f"{NAME}.help",
         f"{NAME}.script",
+        f"{NAME}.script.actions",
         f"{NAME}.script.repository",
         f"{NAME}.script.repository.blue_amo",
         f"{NAME}.script.repository.generic",

blue_assistant-4.46.1/blue_assistant/config.env

@@ -1 +0,0 @@
-BLUE_ASSISTANT_TEST_OBJECT=blue-amo

blue_assistant-4.46.1/blue_assistant/env.py

@@ -1,11 +0,0 @@
-import os
-from blue_options.env import load_config, load_env
-
-load_env(__name__)
-load_config(__name__)
-
-
-BLUE_ASSISTANT_TEST_OBJECT = os.getenv(
-    "BLUE_ASSISTANT_TEST_OBJECT",
-    "",
-)

blue_assistant-4.46.1/blue_assistant/script/load.py

@@ -1,17 +0,0 @@
-from typing import Tuple
-
-from blue_assistant.script.repository import list_of_script_classes
-from blue_assistant.script.repository.generic import GenericScript
-from blue_assistant.logger import logger
-
-
-def load_script(
-    script_name: str,
-    verbose: bool = False,
-) -> Tuple[bool, GenericScript]:
-    for script_class in list_of_script_classes:
-        if script_class.name == script_name:
-            return True, script_class(verbose=verbose)
-
-    logger.error(f"{script_name}: script not found.")
-    return False, GenericScript(verbose=verbose)

blue_assistant-4.46.1/blue_assistant/script/repository/blue_amo/classes.py

@@ -1,42 +0,0 @@
-from typing import Dict
-from tqdm import tqdm
-
-from blue_objects import file, path
-from blue_objects.metadata import post_to_object
-
-from blue_assistant.script.repository.generic.classes import GenericScript
-from blue_assistant.logger import logger
-
-
-class BlueAmoScript(GenericScript):
-    name = path.name(file.path(__file__))
-
-    def __init__(
-        self,
-        verbose: bool = False,
-    ):
-        super().__init__(verbose=verbose)
-
-    def run(
-        self,
-        object_name: str,
-    ) -> bool:
-        if not super().run(object_name=object_name):
-            return False
-
-        metadata: Dict[Dict] = {"nodes": {}}
-        for node_name, node in tqdm(self.nodes.items()):
-            logger.info(
-                "{}{}".format(
-                    node_name,
-                    f": {node}" if self.verbose else " ...",
-                )
-            )
-
-            metadata["nodes"][node_name] = "..."
-
-        return post_to_object(
-            object_name,
-            "output",
-            metadata,
-        )

blue_assistant-4.46.1/blue_assistant/script/repository/generic/classes.py

@@ -1,69 +0,0 @@
-from typing import Dict, List
-import os
-
-from blueness import module
-from blue_objects import file, path
-from blue_objects.metadata import post_to_object
-
-from blue_assistant import NAME
-from blue_assistant.logger import logger
-
-
-NAME = module.name(__file__, NAME)
-
-
-class GenericScript:
-    name = path.name(file.path(__file__))
-
-    def __init__(
-        self,
-        verbose: bool = False,
-    ):
-        self.verbose = verbose
-
-        metadata_filename = os.path.join(
-            file.path(__file__),
-            f"../{self.name}",
-            "metadata.yaml",
-        )
-        self.metadata: Dict
-        success, self.metadata = file.load_yaml(metadata_filename)
-        assert success, f"cannot load {self.name}/metadata.yaml"
-
-        logger.info("loaded {} node(s)".format(len(self.nodes)))
-
-        logger.info("loaded {} variable(s)".format(len(self.vars)))
-        if verbose:
-            for var_name, var_value in self.vars.items():
-                logger.info("{}: {}".format(var_name, var_value))
-
-    @property
-    def script(self) -> str:
-        return self.metadata.get("script", {})
-
-    @property
-    def nodes(self) -> str:
-        return self.metadata.get("script", {}).get("nodes", [])
-
-    @property
-    def vars(self) -> str:
-        return self.metadata.get("script", {}).get("vars", {})
-
-    def run(
-        self,
-        object_name: str,
-    ) -> bool:
-        logger.info(
-            "{}.run: {}:{} -> {}".format(
-                NAME,
-                self.__class__.__name__,
-                self.name,
-                object_name,
-            )
-        )
-
-        return post_to_object(
-            object_name,
-            "script",
-            self.script,
-        )