blue-assistant 4.17.1__tar.gz → 4.24.1__tar.gz

{blue_assistant-4.17.1/blue_assistant.egg-info → blue_assistant-4.24.1}/PKG-INFO

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

{blue_assistant-4.17.1 → blue_assistant-4.24.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.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.24.1`](https://github.com/kamangir/blue-assistant).

{blue_assistant-4.17.1 → blue_assistant-4.24.1}/blue_assistant/__init__.py

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

{blue_assistant-4.17.1 → blue_assistant-4.24.1}/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,34 @@ class ChatContext:
             self.ended = True
             return True
 
-        logger.info(f"🪄 {prompt}")
+        success, response, metadata = complete_prompt(
+            prompt=prompt,
+            max_tokens=max_tokens,
+            verbose=self.verbose,
+        )
+        if not success:
+            return success
+        logger.info(response)
 
+        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-4.17.1 → blue_assistant-4.24.1/blue_assistant.egg-info}/PKG-INFO

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

{blue_assistant-4.17.1 → blue_assistant-4.24.1}/blue_assistant.egg-info/requires.txt

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

{blue_assistant-4.17.1 → blue_assistant-4.24.1}/requirements.txt

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

{blue_assistant-4.17.1 → blue_assistant-4.24.1}/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