blue-assistant 4.78.1__tar.gz → 4.97.1__tar.gz

{blue_assistant-4.78.1/blue_assistant.egg-info → blue_assistant-4.97.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: blue_assistant
-Version: 4.78.1
+Version: 4.97.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.78.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.97.1`](https://github.com/kamangir/blue-assistant).

{blue_assistant-4.78.1 → blue_assistant-4.97.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.78.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.97.1`](https://github.com/kamangir/blue-assistant).

{blue_assistant-4.78.1 → blue_assistant-4.97.1}/blue_assistant/.abcli/tests/script_run.sh

@@ -16,7 +16,8 @@ function test_blue_assistant_script_run() {
             ~upload,$options \
             script=$script_name \
             test_blue_assistant_script_run-$(abcli_string_timestamp_short) \
-            "${@:2}"
+            "${@:2}" \
+            --test_mode 1
 
         [[ $? -ne 0 ]] && return 1
 

{blue_assistant-4.78.1 → blue_assistant-4.97.1}/blue_assistant/__init__.py

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

blue_assistant-4.97.1/blue_assistant/config.env

@@ -0,0 +1,6 @@
+BLUE_ASSISTANT_TEXT_DEFAULT_MODEL=gpt-4o
+BLUE_ASSISTANT_TEXT_MAX_TOKEN=2000
+
+BLUE_ASSISTANT_IMAGE_DEFAULT_MODEL=dall-e-3
+BLUE_ASSISTANT_IMAGE_DEFAULT_QUALITY=standard
+BLUE_ASSISTANT_IMAGE_DEFAULT_SIZE=1024x1024

blue_assistant-4.97.1/blue_assistant/env.py

@@ -0,0 +1,33 @@
+import os
+from blue_options.env import load_config, load_env
+
+load_env(__name__)
+load_config(__name__)
+
+
+BLUE_ASSISTANT_TEXT_DEFAULT_MODEL = os.getenv(
+    "BLUE_ASSISTANT_TEXT_DEFAULT_MODEL",
+    "",
+)
+
+BLUE_ASSISTANT_TEXT_MAX_TOKEN_str = os.getenv("BLUE_ASSISTANT_TEXT_MAX_TOKEN", "")
+try:
+    BLUE_ASSISTANT_TEXT_MAX_TOKEN = int(BLUE_ASSISTANT_TEXT_MAX_TOKEN_str)
+except Exception:
+    BLUE_ASSISTANT_TEXT_MAX_TOKEN = 2000
+
+
+BLUE_ASSISTANT_IMAGE_DEFAULT_MODEL = os.getenv(
+    "BLUE_ASSISTANT_IMAGE_DEFAULT_MODEL",
+    "",
+)
+
+BLUE_ASSISTANT_IMAGE_DEFAULT_QUALITY = os.getenv(
+    "BLUE_ASSISTANT_IMAGE_DEFAULT_QUALITY",
+    "",
+)
+
+BLUE_ASSISTANT_IMAGE_DEFAULT_SIZE = os.getenv(
+    "BLUE_ASSISTANT_IMAGE_DEFAULT_SIZE",
+    "",
+)

{blue_assistant-4.78.1 → blue_assistant-4.97.1}/blue_assistant/help/script.py

@@ -34,7 +34,10 @@ def help_run(
 
     script_options = "script=<script>"
 
-    args = ["--verbose 1"]
+    args = [
+        "[--test_mode 1]",
+        "[--verbose 1]",
+    ]
 
     return show_usage(
         [

{blue_assistant-4.78.1 → blue_assistant-4.97.1}/blue_assistant/script/__main__.py

@@ -31,6 +31,12 @@ parser.add_argument(
     default=0,
     help="0 | 1",
 )
+parser.add_argument(
+    "--test_mode",
+    type=int,
+    default=0,
+    help="0 | 1",
+)
 parser.add_argument(
     "--delim",
     type=str,
@@ -59,6 +65,7 @@ elif args.task == "run":
     success, script = load_script(
         script_name=args.script_name,
         object_name=args.object_name,
+        test_mode=args.test_mode == 1,
         verbose=args.verbose == 1,
     )
 

{blue_assistant-4.78.1 → blue_assistant-4.97.1}/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-4.97.1/blue_assistant/script/actions/generate_image.py

@@ -0,0 +1,58 @@
+from blue_objects import file, objects
+from openai_commands.image_generation.api import OpenAIImageGenerator
+
+from blue_assistant.env import (
+    BLUE_ASSISTANT_IMAGE_DEFAULT_MODEL,
+    BLUE_ASSISTANT_IMAGE_DEFAULT_SIZE,
+    BLUE_ASSISTANT_IMAGE_DEFAULT_QUALITY,
+)
+from blue_assistant.script.actions.generic import GenericAction
+from blue_assistant.script.repository.base.classes import BaseScript
+from blue_assistant.logger import logger
+
+
+class GenerateImageAction(GenericAction):
+    name = file.name(__file__)
+
+    def __init__(
+        self,
+        script: BaseScript,
+    ):
+        super().__init__(script=script)
+
+        self.generator = OpenAIImageGenerator(
+            model=BLUE_ASSISTANT_IMAGE_DEFAULT_MODEL,
+            verbose=self.script.verbose,
+        )
+
+    # https://platform.openai.com/docs/guides/images
+    def perform(
+        self,
+        node_name: str,
+    ) -> bool:
+        if not super().perform(node_name=node_name):
+            return False
+
+        filename = f"{node_name}.png"
+
+        success = self.generator.generate(
+            prompt=self.script.nodes[node_name]["prompt"],
+            filename=objects.path_of(
+                filename=filename,
+                object_name=self.script.object_name,
+                create=True,
+            ),
+            quality=(
+                BLUE_ASSISTANT_IMAGE_DEFAULT_QUALITY if self.script.test_mode else "hd"
+            ),
+            size=(
+                BLUE_ASSISTANT_IMAGE_DEFAULT_SIZE
+                if self.script.test_mode
+                else "1792x1024"
+            ),
+        )[0]
+
+        if success:
+            self.script.nodes[node_name]["filename"] = filename
+
+        return success

{blue_assistant-4.78.1 → blue_assistant-4.97.1}/blue_assistant/script/actions/generate_text.py

@@ -7,7 +7,10 @@ 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.env import (
+    BLUE_ASSISTANT_TEXT_DEFAULT_MODEL,
+    BLUE_ASSISTANT_TEXT_MAX_TOKEN,
+)
 from blue_assistant.script.actions.generic import GenericAction
 from blue_assistant.logger import logger
 
@@ -17,6 +20,7 @@ NAME = module.name(__file__, NAME)
 class GenerateTextAction(GenericAction):
     name = file.name(__file__)
 
+    # https://platform.openai.com/docs/guides/text-generation
     def perform(
         self,
         node_name: str,
@@ -29,9 +33,9 @@ class GenerateTextAction(GenericAction):
             return False
 
         messages: List = []
-        node_context = self.script.get_history(node_name)
-        logger.info("node context: {}".format(" -> ".join(node_context)))
-        for successor in reversed(node_context):
+        list_of_context_nodes = self.script.get_history(node_name)
+        logger.info("node context: {}".format(" <- ".join(list_of_context_nodes)))
+        for context_node in reversed(list_of_context_nodes):
             messages += [
                 {
                     "role": "user",
@@ -39,21 +43,21 @@ class GenerateTextAction(GenericAction):
                         {
                             "type": "text",
                             "text": self.script.apply_vars(
-                                self.script.nodes[successor]["prompt"]
+                                self.script.nodes[context_node]["prompt"]
                             ),
                         }
                     ],
                 }
             ]
 
-            if self.script.nodes[successor]["completed"]:
+            if self.script.nodes[context_node].get("completed", False):
                 messages += [
                     {
                         "role": "assistant",
                         "content": [
                             {
                                 "type": "text",
-                                "text": self.script.nodes[successor]["output"],
+                                "text": self.script.nodes[context_node]["output"],
                             }
                         ],
                     }
@@ -67,8 +71,8 @@ class GenerateTextAction(GenericAction):
         try:
             response = client.chat.completions.create(
                 messages=messages,
-                model=BLUE_ASSISTANT_DEFAULT_MODEL,
-                max_tokens=BLUE_ASSISTANT_MAX_TOKEN,
+                model=BLUE_ASSISTANT_TEXT_DEFAULT_MODEL,
+                max_tokens=BLUE_ASSISTANT_TEXT_MAX_TOKEN,
             )
         except Exception as e:
             logger.error(str(e))

{blue_assistant-4.78.1 → blue_assistant-4.97.1}/blue_assistant/script/actions/generic.py

@@ -1,14 +1,8 @@
-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__)
@@ -24,8 +18,7 @@ class GenericAction:
         node_name: str,
     ) -> bool:
         logger.info(
-            "{}.perform({}) on {}.{} ...".format(
-                NAME,
+            "{}.perform on {}.{} ...".format(
                 self.__class__.__name__,
                 self.script.name,
                 node_name,

{blue_assistant-4.78.1 → blue_assistant-4.97.1}/blue_assistant/script/load.py

@@ -8,6 +8,7 @@ from blue_assistant.logger import logger
 def load_script(
     script_name: str,
     object_name: str,
+    test_mode: bool = False,
     verbose: bool = False,
 ) -> Tuple[bool, BaseScript]:
     found: bool = False
@@ -23,5 +24,6 @@ def load_script(
 
     return found, script_class(
         object_name=object_name,
+        test_mode=test_mode,
         verbose=verbose,
     )

blue_assistant-4.97.1/blue_assistant/script/repository/blue_amo/actions/slice_into_frames.py

@@ -0,0 +1,35 @@
+from blueness import module
+
+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)
+
+
+def slice_into_frames(
+    script: BaseScript,
+    node_name: str,
+) -> bool:
+    logger.info(f"{NAME}: processing the output...")
+
+    list_of_frame_prompts = script.nodes[node_name]["output"].split("---")
+    if len(list_of_frame_prompts) != script.vars["frame_count"]:
+        logger.warning(
+            "{} != {}, frame count doesn't match, bad AI! 😁".format(
+                len(list_of_frame_prompts),
+                script.vars["frame_count"],
+            )
+        )
+
+    list_of_frame_prompts += script.vars["frame_count"] * [""]
+
+    for index in range(script.vars["frame_count"]):
+        node_name = f"generating-frame-{index+1:03d}"
+
+        script.nodes[node_name]["prompt"] = script.nodes[node_name]["prompt"].replace(
+            ":::input",
+            list_of_frame_prompts[index],
+        )
+
+    return True

blue_assistant-4.97.1/blue_assistant/script/repository/blue_amo/actions/stitch_the_frames.py

@@ -0,0 +1,81 @@
+from typing import List
+import numpy as np
+import cv2
+from tqdm import trange
+
+from blueness import module
+from blue_objects import file, objects
+from blue_options import string
+
+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)
+
+
+def stitch_the_frames(
+    script: BaseScript,
+    node_name: str,
+) -> bool:
+    list_of_frames_filenames: List[str] = [
+        filename
+        for filename in [
+            script.nodes[node_name_].get("filename", "")
+            for node_name_ in [
+                f"generating-frame-{index+1:03d}"
+                for index in range(script.vars["frame_count"])
+            ]
+        ]
+        if filename
+    ]
+    if not list_of_frames_filenames:
+        return True
+
+    logger.info(
+        "{} frames to stitch: {}".format(
+            len(list_of_frames_filenames),
+            ", ".join(list_of_frames_filenames),
+        )
+    )
+
+    list_of_frames: List[np.ndarray] = []
+    for filename in list_of_frames_filenames:
+        success, frame = file.load_image(
+            objects.path_of(
+                filename=filename,
+                object_name=script.object_name,
+            )
+        )
+
+        if success:
+            list_of_frames += [frame]
+
+    if not list_of_frames:
+        return True
+
+    common_height = list_of_frames[0].shape[0]
+    for index in trange(len(list_of_frames)):
+        if list_of_frames[index].shape[0] != common_height:
+            aspect_ratio = (
+                list_of_frames[index].shape[1] / list_of_frames[index].shape[0]
+            )
+            new_width = int(common_height * aspect_ratio)
+
+            list_of_frames[index] = cv2.resize(
+                list_of_frames[index],
+                (new_width, common_height),
+                interpolation=cv2.INTER_AREA,
+            )
+
+    full_frame = np.concatenate(list_of_frames, axis=1)
+    logger.info(f"full_frame: {string.pretty_shape_of_matrix(full_frame)}")
+
+    return file.save_image(
+        objects.path_of(
+            filename=f"{node_name}.png",
+            object_name=script.object_name,
+        ),
+        full_frame,
+        log=True,
+    )

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

@@ -0,0 +1,90 @@
+import copy
+
+from blueness import module
+from blue_objects import file, path
+
+from blue_assistant import NAME
+from blue_assistant.script.repository.generic.classes import GenericScript
+from blue_assistant.script.repository.blue_amo.actions.slice_into_frames import (
+    slice_into_frames,
+)
+from blue_assistant.script.repository.blue_amo.actions.stitch_the_frames import (
+    stitch_the_frames,
+)
+from blue_assistant.logger import logger
+
+NAME = module.name(__file__, NAME)
+
+
+class BlueAmoScript(GenericScript):
+    name = path.name(file.path(__file__))
+
+    def __init__(
+        self,
+        object_name: str,
+        test_mode: bool = False,
+        verbose: bool = False,
+    ):
+        super().__init__(
+            object_name=object_name,
+            test_mode=test_mode,
+            verbose=verbose,
+        )
+
+        if self.test_mode:
+            self.vars["frame_count"] = 1
+
+        holder_node_name = "generating-the-frames"
+        logger.info(
+            "{}: expanding {} X {}...".format(
+                NAME,
+                holder_node_name,
+                self.vars["frame_count"],
+            )
+        )
+
+        holder_node = self.nodes[holder_node_name]
+        del self.nodes[holder_node_name]
+        self.G.remove_node(holder_node_name)
+
+        reduce_node = "stitching-the-frames"
+        self.G.add_node(reduce_node)
+        self.nodes[reduce_node] = {"action": "skip"}
+
+        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",
+            )
+            self.G.add_edge(
+                reduce_node,
+                node_name,
+            )
+
+        assert self.save_graph()
+
+    def perform_action(
+        self,
+        node_name: str,
+    ) -> bool:
+        if not super().perform_action(node_name=node_name):
+            return False
+
+        if node_name == "slicing-into-frames":
+            return slice_into_frames(
+                script=self,
+                node_name=node_name,
+            )
+
+        if node_name == "stitching-the-frames":
+            return stitch_the_frames(
+                script=self,
+                node_name=node_name,
+            )
+
+        return True

{blue_assistant-4.78.1 → blue_assistant-4.97.1}/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,6 +19,31 @@ 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")
+        if action_name == "skip":
+            return True
+
+        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:
@@ -26,15 +51,18 @@ class GenericScript(BaseScript):
             return False
 
         success: bool = True
-        while not all(self.nodes[node]["completed"] for node in self.nodes) and success:
+        while (
+            not all(self.nodes[node].get("completed", False) for node in self.nodes)
+            and success
+        ):
             for node_name in tqdm(self.nodes):
-                if self.nodes[node_name]["completed"]:
+                if self.nodes[node_name].get("completed", False):
                     continue
 
                 pending_dependencies = [
                     node_name_
                     for node_name_ in self.G.successors(node_name)
-                    if not self.nodes[node_name_]["completed"]
+                    if not self.nodes[node_name_].get("completed", False)
                 ]
                 if pending_dependencies:
                     logger.info(
@@ -46,10 +74,7 @@ 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
 

{blue_assistant-4.78.1 → blue_assistant-4.97.1/blue_assistant.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: blue_assistant
-Version: 4.78.1
+Version: 4.97.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.78.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.97.1`](https://github.com/kamangir/blue-assistant).

{blue_assistant-4.78.1 → blue_assistant-4.97.1}/blue_assistant.egg-info/SOURCES.txt

@@ -48,6 +48,9 @@ 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/script/repository/blue_amo/actions/__init__.py
+blue_assistant/script/repository/blue_amo/actions/slice_into_frames.py
+blue_assistant/script/repository/blue_amo/actions/stitch_the_frames.py
 blue_assistant/script/repository/generic/__init__.py
 blue_assistant/script/repository/generic/classes.py
 blue_assistant/script/repository/moon_datasets/__init__.py

{blue_assistant-4.78.1 → blue_assistant-4.97.1}/setup.py

@@ -14,6 +14,7 @@ setup(
         f"{NAME}.script.actions",
         f"{NAME}.script.repository",
         f"{NAME}.script.repository.blue_amo",
+        f"{NAME}.script.repository.blue_amo.actions",
         f"{NAME}.script.repository.generic",
         f"{NAME}.script.repository.moon_datasets",
     ],

blue_assistant-4.78.1/blue_assistant/config.env

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

blue_assistant-4.78.1/blue_assistant/env.py

@@ -1,17 +0,0 @@
-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.78.1/blue_assistant/script/actions/generate_image.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 GenerateImageAction(GenericAction):
-    name = file.name(__file__)
-
-    def perform(
-        self,
-        node_name: str,
-    ) -> bool:
-        if not super().perform(node_name=node_name):
-            return False
-
-        logger.info(f"🪄 generating image ...: {node_name}")
-
-        return True

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

@@ -1,9 +0,0 @@
-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__))