blue-assistant 4.9.1__py3-none-any.whl → 4.21.1__py3-none-any.whl

blue_assistant/.abcli/chat.sh

@@ -7,7 +7,7 @@ function blue_assistant_chat() {
     local do_upload=$(abcli_option_int "$options" upload $(abcli_not $do_dryrun))
 
     local chat_options=$2
-    local interactive=$(abcli_option_int "$options" interact 1)
+    local interactive=$(abcli_option_int "$chat_options" interact 1)
 
     local object_name=$(abcli_clarify_object $3 chat-$(abcli_string_timestamp))
     [[ "$do_download" == 1 ]] &&

blue_assistant/__init__.py

@@ -4,7 +4,7 @@ ICON = "🌀"
 
 DESCRIPTION = f"{ICON} An AI Assistant."
 
-VERSION = "4.9.1"
+VERSION = "4.21.1"
 
 REPO_NAME = "blue-assistant"
 

blue_assistant/chat/context.py

@@ -0,0 +1,109 @@
+from blue_objects import file, objects
+from blue_options import string
+from blue_objects.metadata import post_to_object, get_from_object
+from blue_options.terminal.functions import hr
+from openai_commands.prompt_completion.api import complete_prompt
+
+from blue_assistant import ICON
+from blue_assistant.logger import logger
+
+
+class ChatContext:
+    def __init__(
+        self,
+        object_name: str,
+        load_history: bool = True,
+        verbose: bool = False,
+    ):
+        self.verbose = verbose
+
+        self.ended: bool = False
+
+        self.object_name = object_name
+
+        self.history = (
+            get_from_object(
+                self.object_name,
+                "history",
+                [],
+            )
+            if load_history
+            else []
+        )
+
+    def chat(
+        self,
+        interactive: bool = True,
+    ) -> bool:
+        logger.info('Type in "help" for help.')
+
+        while True:
+            prompt = input(f"{ICON}  > ") if interactive else ""
+
+            if not self.process_prompt(prompt):
+                return False
+
+            if self.ended or not interactive:
+                break
+
+            print(hr(width=21))
+
+        return True
+
+    def process_prompt(
+        self,
+        prompt: str,
+        max_tokens: int = 2000,
+    ) -> bool:
+        if prompt in ["help", "?", ""]:
+            return self.show_help()
+
+        if prompt in ["quit", "exit", "q"]:
+            self.ended = True
+            return True
+
+        success, response, metadata = complete_prompt(
+            prompt=prompt,
+            max_tokens=max_tokens,
+            verbose=self.verbose,
+        )
+        if not success:
+            return success
+        logger.info(response)
+
+        logger.info(f"🪄 {prompt}")
+
+        id_ = string.pretty_date(
+            as_filename=True,
+            unique=True,
+        )
+        self.history.append(
+            {
+                "id": id_,
+                "prompt": prompt,
+                "response": response,
+            }
+        )
+
+        return file.save_yaml(
+            objects.path_of(
+                object_name=self.object_name,
+                filename=f"{id_}.yaml",
+            ),
+            metadata,
+        )
+
+    def save(self) -> bool:
+        return post_to_object(
+            self.object_name,
+            "history",
+            self.history,
+        )
+
+    def show_help(self) -> bool:
+        for command, description in {
+            "help": "show this help.",
+            "quit": "quit the chat.",
+        }.items():
+            logger.info(f"{command}: {description}")
+        return True

blue_assistant/chat/functions.py

@@ -1,6 +1,7 @@
 from blueness import module
 
 from blue_assistant import NAME
+from blue_assistant.chat.context import ChatContext
 from blue_assistant.logger import logger
 
 
@@ -11,6 +12,7 @@ def chat(
     object_name: str,
     interactive: bool = True,
     verbose: bool = False,
+    load_history: bool = True,
 ) -> bool:
     logger.info(
         "{}.chat -{}> {}".format(
@@ -20,6 +22,15 @@ def chat(
         )
     )
 
-    logger.info("🪄")
+    context = ChatContext(
+        object_name,
+        load_history=load_history,
+        verbose=verbose,
+    )
+
+    if not context.chat(
+        interactive=interactive,
+    ):
+        return False
 
-    return True
+    return context.save()

{blue_assistant-4.9.1.dist-info → blue_assistant-4.21.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: blue_assistant
-Version: 4.9.1
+Version: 4.21.1
 Summary: 🌀 An AI Assistant.
 Home-page: https://github.com/kamangir/blue-assistant
 Author: Arash Abadpour (Kamangir)
@@ -15,6 +15,7 @@ License-File: LICENSE
 Requires-Dist: blueness
 Requires-Dist: blue-options
 Requires-Dist: abcli
+Requires-Dist: openai_commands
 Requires-Dist: boto3
 Requires-Dist: geojson
 Requires-Dist: geopandas
@@ -62,4 +63,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.205.1`](https://github.com/kamangir/awesome-bash-cli), based on 🌀 [`blue_assistant-4.9.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.21.1`](https://github.com/kamangir/blue-assistant).

{blue_assistant-4.9.1.dist-info → blue_assistant-4.21.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
 blue_assistant/README.py,sha256=q6EWCy7zojQESGDF-2xEoXS1Bl0wRJTe2jw_1QM_WYw,600
-blue_assistant/__init__.py,sha256=9LsmeDG8swE0pYICUhzxci43g_rP3ypSMITxMD2fNE0,309
+blue_assistant/__init__.py,sha256=1lvcHZ3f-cuu4kggGxsk0tHJqiiPFtcPSSzcpP8kG7Q,310
 blue_assistant/__main__.py,sha256=URtal70XZc0--3FDTYWcLtnGOqBYjMX9gt-L1k8hDXI,361
 blue_assistant/config.env,sha256=89ao50sGXYWiU-NMV_fUVXAXdTHKKfWV03BY3_p6EOE,25
 blue_assistant/env.py,sha256=Luc_ws9lWcgE6u-wjgBL-1WwsL7-kBpiN6bhO5MO96M,177
@@ -13,20 +13,21 @@ blue_assistant/.abcli/actions.sh,sha256=vW1hNMuhjghvqib0775kDzDwqGnqPo3mqLTUkPCd
 blue_assistant/.abcli/alias.sh,sha256=MCJzXaDnX1QMllWsZJJkDePBYt1nY9ZWa3o4msfGD28,57
 blue_assistant/.abcli/blue_assistant.sh,sha256=plLTQQerVmfb_SNlOkv0MEaQCF7YdsOHzCq0M3FWT4c,239
 blue_assistant/.abcli/browse.sh,sha256=qZ_RK_WnsjmF-hfWKiMEOnnv22QtZh9HQ0VFJUbP6aI,294
-blue_assistant/.abcli/chat.sh,sha256=1lrDwRrBjR_uhDc7_zeuhxjqgucaZMrrUaqw-o166bs,823
+blue_assistant/.abcli/chat.sh,sha256=8jMqlOm1Pdg-KSb1YSSC2tbW3-1fGwRGpZ6aDITvL2U,828
 blue_assistant/.abcli/tests/README.sh,sha256=Qs0YUxVB1OZZ70Nqw2kT1LKXeUnC5-XfQRMfqb8Cbwg,152
 blue_assistant/.abcli/tests/chat.sh,sha256=ptW6aLBp1xB7G4RligZOMnTDtpmdE0LxlkqC236TpCo,272
 blue_assistant/.abcli/tests/help.sh,sha256=n2FZO78a93YeKSqqZfJoqYDBMHQRWQp2l96zEvl2Ui0,653
 blue_assistant/.abcli/tests/version.sh,sha256=oR2rvYR8zi-0VDPIdPJsmsmWwYaamT8dmNTqUh3-8Gw,154
 blue_assistant/chat/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 blue_assistant/chat/__main__.py,sha256=n9fvhhcTHjg33EHvHdFnc2kf8X1yhmOJUahDFs82bW0,870
-blue_assistant/chat/functions.py,sha256=QHjrEoYbO95olWrRyOh9njuUwAwYy5NM5VzVNpZF4fM,444
+blue_assistant/chat/context.py,sha256=v1ZQYnK7-eVCRiX9vZt2gTcCwzrClWwvFiMtITpcQSE,2616
+blue_assistant/chat/functions.py,sha256=Al_JUUldshY191Q6rlpVPCExNH5oOFrtkh50419K_0I,714
 blue_assistant/help/__init__.py,sha256=ajz1GSNU9xYVrFEDSz6Xwg7amWQ_yvW75tQa1ZvRIWc,3
 blue_assistant/help/__main__.py,sha256=cVejR7OpoWPg0qLbm-PZf5TuJS27x49jzfiyCLyzEns,241
 blue_assistant/help/chat.py,sha256=fkcT14htkjEt7EOpgvHuUPzj7d8jQDxdgN2ZcU0lVEg,534
 blue_assistant/help/functions.py,sha256=e1S3oxVKSkLThMMyAsPotaovVFHQbiMN4yRWVZP6TB0,663
-blue_assistant-4.9.1.dist-info/LICENSE,sha256=ogEPNDSH0_dhiv_lT3ifVIdgIzHAqNA_SemnxUfPBJk,7048
-blue_assistant-4.9.1.dist-info/METADATA,sha256=IH3jq05dUAsaFIByH3z5q56tyqRKx5Dsphx4OoU5s6s,2261
-blue_assistant-4.9.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-blue_assistant-4.9.1.dist-info/top_level.txt,sha256=ud0BkBbdOVze13bNqHuhZj1rwCztaBtDf5ChEYzASOs,15
-blue_assistant-4.9.1.dist-info/RECORD,,
+blue_assistant-4.21.1.dist-info/LICENSE,sha256=ogEPNDSH0_dhiv_lT3ifVIdgIzHAqNA_SemnxUfPBJk,7048
+blue_assistant-4.21.1.dist-info/METADATA,sha256=x3wxmN4nVs5JwU-wuHGb42QItMl3lBm1e170Q5iy0Bw,2294
+blue_assistant-4.21.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+blue_assistant-4.21.1.dist-info/top_level.txt,sha256=ud0BkBbdOVze13bNqHuhZj1rwCztaBtDf5ChEYzASOs,15
+blue_assistant-4.21.1.dist-info/RECORD,,