blue-assistant 4.9.1__tar.gz → 4.21.1__tar.gz

{blue_assistant-4.9.1/blue_assistant.egg-info → blue_assistant-4.21.1}/PKG-INFO

@@ -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 → blue_assistant-4.21.1}/README.md

@@ -23,4 +23,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 → blue_assistant-4.21.1}/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-4.9.1 → blue_assistant-4.21.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-4.21.1/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-4.9.1 → blue_assistant-4.21.1}/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 → blue_assistant-4.21.1/blue_assistant.egg-info}/PKG-INFO

@@ -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 → blue_assistant-4.21.1}/blue_assistant.egg-info/SOURCES.txt

@@ -31,6 +31,7 @@ blue_assistant/.abcli/tests/help.sh
 blue_assistant/.abcli/tests/version.sh
 blue_assistant/chat/__init__.py
 blue_assistant/chat/__main__.py
+blue_assistant/chat/context.py
 blue_assistant/chat/functions.py
 blue_assistant/help/__init__.py
 blue_assistant/help/__main__.py

{blue_assistant-4.9.1 → blue_assistant-4.21.1}/blue_assistant.egg-info/requires.txt

@@ -1,6 +1,7 @@
 blueness
 blue-options
 abcli
+openai_commands
 boto3
 geojson
 geopandas

{blue_assistant-4.9.1 → blue_assistant-4.21.1}/requirements.txt

@@ -1,6 +1,7 @@
 blueness
 blue-options
 abcli
+openai_commands
 boto3
 geojson
 geopandas