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

{blue_assistant-4.53.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.53.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
@@ -68,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.53.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.53.1 → blue_assistant-4.62.1}/README.md

@@ -28,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.53.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.53.1 → blue_assistant-4.62.1}/blue_assistant/__init__.py

@@ -4,7 +4,7 @@ ICON = "🧠"
 
 DESCRIPTION = f"{ICON} An AI Assistant."
 
-VERSION = "4.53.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.53.1 → blue_assistant-4.62.1}/blue_assistant/script/__main__.py

@@ -58,11 +58,12 @@ if args.task == "list":
 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.53.1/blue_assistant/script/actions/__init__.py → blue_assistant-4.62.1/blue_assistant/script/actions/functions.py

@@ -1,17 +1,14 @@
 from typing import List, Dict, Tuple, Type
 
+from blueness import module
 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.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
 
-list_of_actions: List[GenericAction] = [
-    GenericAction,
-    GenerateImageAction,
-    GenerateTextAction,
-    WorkInProgressAction,
-]
+NAME = module.name(__file__, NAME)
 
 
 def get_action_class(
@@ -26,15 +23,30 @@ def get_action_class(
 
 
 def perform_action(
-    action_name: str,
-    node: Dict,
+    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.",
         }
 
-    action_object = action_class(node=node)
-
-    return action_object.perform()
+    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.53.1 → blue_assistant-4.62.1}/blue_assistant/script/actions/generate_image.py

@@ -9,12 +9,15 @@ from blue_assistant.logger import logger
 class GenerateImageAction(GenericAction):
     name = file.name(__file__)
 
-    def perform(self) -> Tuple[bool, Dict]:
-        success, generic_metadata = super().perform()
+    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 ...: {self.node}")
+        logger.info(f"🪄 generating image ...: {node_name}")
         metadata = {}
 
         metadata.update(generic_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.53.1 → blue_assistant-4.62.1}/blue_assistant/script/actions/generic.py

@@ -4,6 +4,7 @@ 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)
@@ -12,14 +13,22 @@ NAME = module.name(__file__, NAME)
 class GenericAction:
     name = file.name(__file__)
 
-    def __init__(self, node: Dict):
-        self.node = node
+    def __init__(
+        self,
+        script: BaseScript,
+    ):
+        self.script = script
 
-    def perform(self) -> Tuple[bool, Dict]:
+    def perform(
+        self,
+        node_name: str,
+    ) -> Tuple[bool, Dict]:
         logger.info(
-            "{}.perform({}) ...".format(
+            "{}.perform({}) on {}.{} ...".format(
                 NAME,
                 self.__class__.__name__,
+                self.script.name,
+                node_name,
             ),
         )
         return True, {}

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.53.1 → blue_assistant-4.62.1}/blue_assistant/script/repository/__init__.py

@@ -1,6 +1,6 @@
 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

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.53.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.53.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
@@ -68,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.53.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.53.1 → blue_assistant-4.62.1}/blue_assistant.egg-info/SOURCES.txt

@@ -40,6 +40,7 @@ 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-4.53.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.53.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.53.1/blue_assistant/config.env

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

blue_assistant-4.53.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.53.1/blue_assistant/script/actions/generate_text.py

@@ -1,21 +0,0 @@
-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 GenerateTextAction(GenericAction):
-    name = file.name(__file__)
-
-    def perform(self) -> Tuple[bool, Dict]:
-        success, generic_metadata = super().perform()
-        if not success:
-            return success, generic_metadata
-
-        logger.info(f"🪄 generating text ...: {self.node}")
-        metadata = {}
-
-        metadata.update(generic_metadata)
-        return True, metadata

blue_assistant-4.53.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.53.1/blue_assistant/script/repository/generic/classes.py

@@ -1,104 +0,0 @@
-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.actions import perform_action
-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,
-            )
-        )
-
-        if not post_to_object(
-            object_name,
-            "script",
-            self.script,
-        ):
-            return False
-
-        metadata: Dict[Dict] = {"nodes": {}}
-        success: bool = True
-        for node_name, node in tqdm(self.nodes.items()):
-            logger.info(
-                "{}{}".format(
-                    node_name,
-                    f": {node}" if self.verbose else " ...",
-                )
-            )
-
-            assert isinstance(node, dict)
-            success, output = perform_action(
-                action_name=node.get("action", "unknown"),
-                node=node,
-            )
-            metadata["nodes"][node_name] = {
-                "success": success,
-                "output": output,
-            }
-            if not success:
-                break
-
-        if not post_to_object(
-            object_name,
-            "output",
-            metadata,
-        ):
-            return False
-
-        return success