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

blue_assistant/__init__.py

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

blue_assistant/chat/context.py

@@ -1,5 +1,8 @@
+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
@@ -10,7 +13,10 @@ class ChatContext:
         self,
         object_name: str,
         load_history: bool = True,
+        verbose: bool = False,
     ):
+        self.verbose = verbose
+
         self.ended: bool = False
 
         self.object_name = object_name
@@ -28,7 +34,6 @@ class ChatContext:
     def chat(
         self,
         interactive: bool = True,
-        verbose: bool = False,
     ) -> bool:
         logger.info('Type in "help" for help.')
 
@@ -45,7 +50,11 @@ class ChatContext:
 
         return True
 
-    def process_prompt(self, prompt: str) -> bool:
+    def process_prompt(
+        self,
+        prompt: str,
+        max_tokens: int = 2000,
+    ) -> bool:
         if prompt in ["help", "?", ""]:
             return self.show_help()
 
@@ -53,15 +62,36 @@ class ChatContext:
             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 True
+        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(

blue_assistant/chat/functions.py

@@ -25,11 +25,11 @@ def chat(
     context = ChatContext(
         object_name,
         load_history=load_history,
+        verbose=verbose,
     )
 
     if not context.chat(
         interactive=interactive,
-        verbose=verbose,
     ):
         return False
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: blue_assistant
-Version: 4.17.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.207.1`](https://github.com/kamangir/awesome-bash-cli), based on 🌀 [`blue_assistant-4.17.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.17.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=pcxl_W4PLmlpRZX3KNDCS1DC2tCbbMrbau0-2R9x2BY,310
+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
@@ -20,14 +20,14 @@ blue_assistant/.abcli/tests/help.sh,sha256=n2FZO78a93YeKSqqZfJoqYDBMHQRWQp2l96zE
 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/context.py,sha256=AjBkw0FK6CtEyFskDsAdyXvNYGs_4v-sFW6257ftA6o,1812
-blue_assistant/chat/functions.py,sha256=dzDuTxslopJogU8PqPiMLqeD490kWG-NvZUaOuROoWo,714
+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.17.1.dist-info/LICENSE,sha256=ogEPNDSH0_dhiv_lT3ifVIdgIzHAqNA_SemnxUfPBJk,7048
-blue_assistant-4.17.1.dist-info/METADATA,sha256=GpHzia7hgY2DWZ0zE5uUfGNq9QoRkrK6ZC6HTW0P7Qk,2263
-blue_assistant-4.17.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-blue_assistant-4.17.1.dist-info/top_level.txt,sha256=ud0BkBbdOVze13bNqHuhZj1rwCztaBtDf5ChEYzASOs,15
-blue_assistant-4.17.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,,