blue-assistant 4.66.1__py3-none-any.whl → 4.87.1__py3-none-any.whl

blue_assistant/__init__.py

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

blue_assistant/script/__main__.py

@@ -28,7 +28,7 @@ parser.add_argument(
 parser.add_argument(
     "--verbose",
     type=int,
-    default=1,
+    default=0,
     help="0 | 1",
 )
 parser.add_argument(

blue_assistant/script/actions/functions.py

@@ -20,29 +20,3 @@ def get_action_class(
 
     logger.error(f"{action_name}: action not found.")
     return False, GenericAction
-
-
-def perform_action(
-    script: BaseScript,
-    node_name: str,
-) -> bool:
-    action_name = script.nodes[node_name].get("action", "unknown")
-
-    success, action_class = get_action_class(action_name=action_name)
-    if not success:
-        return success
-
-    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/script/actions/generate_text.py

@@ -29,9 +29,9 @@ class GenerateTextAction(GenericAction):
             return False
 
         messages: List = []
-        node_history = self.script.get_history(node_name)
-        logger.info("node history: {}".format(node_history))
-        for successor in reversed(node_history):
+        node_context = self.script.get_history(node_name)
+        logger.info("node context: {}".format(" <- ".join(node_context)))
+        for successor in reversed(node_context):
             messages += [
                 {
                     "role": "user",
@@ -46,7 +46,7 @@ class GenerateTextAction(GenericAction):
                 }
             ]
 
-            if self.script.G.nodes[successor]["completed"]:
+            if self.script.nodes[successor]["completed"]:
                 messages += [
                     {
                         "role": "assistant",
@@ -84,7 +84,7 @@ class GenerateTextAction(GenericAction):
         output = response.choices[0].message.content
         logger.info(f"🗣️ output: {output}")
 
-        self.script.G.nodes[node_name]["output"] = output
+        self.script.nodes[node_name]["output"] = output
 
         var_name = self.script.nodes[node_name].get("output", "")
         if var_name:

blue_assistant/script/repository/blue_amo/classes.py

@@ -1,5 +1,5 @@
 from blue_objects import file, path
-
+import copy
 
 from blue_assistant.script.repository.generic.classes import GenericScript
 from blue_assistant.logger import logger
@@ -7,3 +7,32 @@ from blue_assistant.logger import logger
 
 class BlueAmoScript(GenericScript):
     name = path.name(file.path(__file__))
+
+    def __init__(
+        self,
+        object_name: str,
+        verbose: bool = False,
+    ):
+        super().__init__(
+            object_name=object_name,
+            verbose=verbose,
+        )
+
+        holder_node_name = "generating-the-frames"
+
+        holder_node = self.nodes[holder_node_name]
+        del self.nodes[holder_node_name]
+        self.G.remove_node(holder_node_name)
+
+        for index in range(self.vars["frame_count"]):
+            node_name = f"generating-frame-{index+1:03d}"
+
+            self.nodes[node_name] = copy.deepcopy(holder_node)
+
+            self.G.add_node(node_name)
+            self.G.add_edge(
+                node_name,
+                "slicing-into-frames",
+            )
+
+        assert self.save_graph()

blue_assistant/script/repository/generic/classes.py

@@ -9,7 +9,7 @@ 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.script.actions.functions import get_action_class
 from blue_assistant.logger import logger
 
 
@@ -19,26 +19,45 @@ NAME = module.name(__file__, NAME)
 class GenericScript(BaseScript):
     name = path.name(file.path(__file__))
 
+    def perform_action(
+        self,
+        node_name: str,
+    ) -> bool:
+        action_name = self.nodes[node_name].get("action", "unknown")
+
+        success, action_class = get_action_class(action_name=action_name)
+        if not success:
+            return success
+
+        logger.info(
+            "{}.perform_action: {} == {} on {}".format(
+                NAME,
+                action_name,
+                action_class.__name__,
+                node_name,
+            )
+        )
+
+        action_object = action_class(script=self)
+
+        return action_object.perform(node_name=node_name)
+
     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"]:
+        while not all(self.nodes[node]["completed"] for node in self.nodes) and success:
+            for node_name in tqdm(self.nodes):
+                if self.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 not self.nodes[node_name_]["completed"]
                 ]
                 if pending_dependencies:
                     logger.info(
@@ -50,19 +69,16 @@ class GenericScript(BaseScript):
                     )
                     continue
 
-                if not perform_action(
-                    script=self,
-                    node_name=node_name,
-                ):
+                if not self.perform_action(node_name=node_name):
                     success = False
                     break
 
-                self.G.nodes[node_name]["completed"] = True
+                self.nodes[node_name]["completed"] = True
 
         if not post_to_object(
             self.object_name,
             "output",
-            metadata,
+            self.metadata,
         ):
             return False
 

{blue_assistant-4.66.1.dist-info → blue_assistant-4.87.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: blue_assistant
-Version: 4.66.1
+Version: 4.87.1
 Summary: 🧠 An AI Assistant.
 Home-page: https://github.com/kamangir/blue-assistant
 Author: Arash Abadpour (Kamangir)
@@ -72,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.66.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.87.1`](https://github.com/kamangir/blue-assistant).

{blue_assistant-4.66.1.dist-info → blue_assistant-4.87.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
 blue_assistant/README.py,sha256=Nzr3v-VSa8CISkqkLHd8ILyY_WiJ7x2igYPH-lYpzEY,795
-blue_assistant/__init__.py,sha256=0v_LdK6vN-oGFPitqPD_ntVE3VlcyYyrddqz4xbjDnM,310
+blue_assistant/__init__.py,sha256=bHmdRdS7buZRGyYFPR7olkJj3ivLxtWFzxIOMAGY3ZY,310
 blue_assistant/__main__.py,sha256=URtal70XZc0--3FDTYWcLtnGOqBYjMX9gt-L1k8hDXI,361
 blue_assistant/config.env,sha256=JV9zx17FflIPmV1hK-yuM0GlrUjkGuaQ_EzPFPTkoXM,66
 blue_assistant/env.py,sha256=rgIlKVsALrRD1sYJV3SkFfzIvWNavix2HN0sSy2cCi4,391
@@ -26,23 +26,23 @@ blue_assistant/help/__main__.py,sha256=cVejR7OpoWPg0qLbm-PZf5TuJS27x49jzfiyCLyzE
 blue_assistant/help/functions.py,sha256=9WsmXGMN-R7sqlkGLK0nY90Peg8Gah4rIu75QbLhImo,689
 blue_assistant/help/script.py,sha256=wvS_5FWu9LAsaDDq7UoxXEadRwGseYqwu2P-9dNc5Bo,1077
 blue_assistant/script/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-blue_assistant/script/__main__.py,sha256=D5APoSVe-l5H0Nrgyq5do4cSbmRyYmflq1x_Dd5d1fM,1529
+blue_assistant/script/__main__.py,sha256=immGrRfasPHdPDxZJweu0coi_V0jg8_2zfqTnRxNfaU,1529
 blue_assistant/script/load.py,sha256=Bd5L9HBL0Yh1MLGAyGrQX3XQLH-UiOWLsHJvzCmNs3g,773
 blue_assistant/script/actions/__init__.py,sha256=8Sp6avoJGDXVORvx5mvkxTNaxjz_KevKN4Wb8qLZHsQ,487
-blue_assistant/script/actions/functions.py,sha256=0fiihOWbFiwZEXV6t9oxJaxJ5E5X-k3weqGTOKTu3sA,1233
+blue_assistant/script/actions/functions.py,sha256=nazOu20plxxxUIcr1YCpNiWn4zmcGCCWcojWxLnZtl8,669
 blue_assistant/script/actions/generate_image.py,sha256=bGDdP9XfBtFFFNL4d7DA7_FJvsFKadPOgwVAXkFgyjE,477
-blue_assistant/script/actions/generate_text.py,sha256=rKDjY_keN_E_vdboQ_SqG2APyUanpAKRDVGCxzghO3k,2817
+blue_assistant/script/actions/generate_text.py,sha256=-RShuIAYvRVLtOsuAW9XRrxeaU9NPUYGYWrYUyJo_tc,2826
 blue_assistant/script/actions/generic.py,sha256=dIjqocIEpLEb1BI76kHivVpDUG5QpeZHLA2onsYn1SU,731
 blue_assistant/script/actions/wip.py,sha256=K4kJE4bn3u6o-QWTESPydO_eD54zOwZbU9WLAO94z1Q,171
 blue_assistant/script/repository/__init__.py,sha256=WFkbe-6yyljpmeVpXgLhOPt-YRc7BwkRNzPO-7Wz0Dg,573
 blue_assistant/script/repository/blue_amo/__init__.py,sha256=WjL9GIlN-DBnbUMJ8O_FxTp0rcVGlsIS3H9YtXEefTk,76
-blue_assistant/script/repository/blue_amo/classes.py,sha256=RY1gjZVPX8xu7nfeiZoTL8RTpu73RsoaNv4ZoGv9NNs,234
+blue_assistant/script/repository/blue_amo/classes.py,sha256=LlbdXlQfV3WRqooirmgSENa5NKjfllfCvi0JWlhR3SA,994
 blue_assistant/script/repository/generic/__init__.py,sha256=kLffGsQMQAFJTw6IZBE5eBxvshP1x9wwHHR4hsDJblo,75
-blue_assistant/script/repository/generic/classes.py,sha256=X6h-cWFleUQF9flyq6ClYSdi5ysGOTsy156w8jT1wvY,1964
+blue_assistant/script/repository/generic/classes.py,sha256=3E5OrRSvI98y51zfVFYZV8bnoaPXyu3bU_IruIX6G34,2413
 blue_assistant/script/repository/moon_datasets/__init__.py,sha256=aCtmP2avh3yKAJ668S3GsLR9vbBOm5zt9FSFCqy_tAs,86
 blue_assistant/script/repository/moon_datasets/classes.py,sha256=68zThDhjF9gGRnsw8EKNLGOMBFbCSljt0jGovuOzCAc,197
-blue_assistant-4.66.1.dist-info/LICENSE,sha256=ogEPNDSH0_dhiv_lT3ifVIdgIzHAqNA_SemnxUfPBJk,7048
-blue_assistant-4.66.1.dist-info/METADATA,sha256=o9HM9_F9s8xZ0lhJhtngCE2qqdbASHaRGOMzhuFw4XE,2625
-blue_assistant-4.66.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-blue_assistant-4.66.1.dist-info/top_level.txt,sha256=ud0BkBbdOVze13bNqHuhZj1rwCztaBtDf5ChEYzASOs,15
-blue_assistant-4.66.1.dist-info/RECORD,,
+blue_assistant-4.87.1.dist-info/LICENSE,sha256=ogEPNDSH0_dhiv_lT3ifVIdgIzHAqNA_SemnxUfPBJk,7048
+blue_assistant-4.87.1.dist-info/METADATA,sha256=iEwfKly_X4g_d-UOeGVnY4yQcaFuJ_YfIbqmRW5ZCeo,2625
+blue_assistant-4.87.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+blue_assistant-4.87.1.dist-info/top_level.txt,sha256=ud0BkBbdOVze13bNqHuhZj1rwCztaBtDf5ChEYzASOs,15
+blue_assistant-4.87.1.dist-info/RECORD,,