nano-bots 0.0.10 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b042b76220123246b7456f35a892292b4df837d7174870ec8f5b9afed97a17cf
-  data.tar.gz: 3088c0a1f4b1744ee8827d7eaa9aaefdc6a35568dd0eff5963b22034ae3955e9
+  metadata.gz: 4632306b0fe1fc9af4ee96c7624f163d80c70eb259c80f3efad158e95e76414a
+  data.tar.gz: a2156c47648e3b8b118a4698d23031ca4bd4bf139db6f5f19fa5561f1591cf7b
 SHA512:
-  metadata.gz: d8abc4664db7e12421519eea694c8d42267cce0264d0501e3469a452854b6b1e9de27a0d5a939afb8b6903d32114d05d3f6392d0f45bbfe4f399de0d99b3a6f6
-  data.tar.gz: a12beec15362c517d5d0cbe3d114a74c9cd95b5bd44d6d66346e7180a399f96521e8508719a501c0a62817bf9783bf54687e9ac2eb5b18cc5e92cc1c5fe39948
+  metadata.gz: d8271268413e87793b147385c999edddbb59dc70203dfca243adb7836ee680dc430a57dc625d038394b6b86e3bb0a9a4e45f12291f9e1ef9574da8ee69ed93e5
+  data.tar.gz: 1d6d1118162ed3dbc3438205aa7b4e858499d154e7cc6bdba963a53ddeeff7c071398e9eebc1f3d273c7aa128c39fbcb452a8f6f9bfff8d41b2dac81fb04d9ce

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    nano-bots (0.0.10)
+    nano-bots (0.1.0)
       babosa (~> 2.0)
       dotenv (~> 2.8, >= 2.8.1)
       faraday (~> 2.7, >= 2.7.5)

data/README.md

@@ -28,13 +28,13 @@ https://user-images.githubusercontent.com/113217272/238141567-c58a240c-7b67-4b3b
 For a system usage:
 
 ```sh
-gem install nano-bots -v 0.0.10
+gem install nano-bots -v 0.1.0
 ```
 
 To use it in a project, add it to your `Gemfile`:
 
 ```ruby
-gem 'nano-bots', '~> 0.0.10'
+gem 'nano-bots', '~> 0.1.0'
 ```
 
 ```sh
@@ -44,10 +44,11 @@ bundle install
 For credentials and configurations, relevant environment variables can be set in your `.bashrc`, `.zshrc`, or equivalent files, as well as in your Docker Container or System Environment. Example:
 
 ```sh
-export NANO_BOTS_ENCRYPTION_PASSWORD="UNSAFE"
 export OPENAI_API_ADDRESS=https://api.openai.com
-export OPENAI_API_ACCESS_TOKEN=your-token
-export OPENAI_API_USER_IDENTIFIER=your-user
+export OPENAI_API_KEY=your-access-token
+
+export NANO_BOTS_ENCRYPTION_PASSWORD=UNSAFE
+export NANO_BOTS_END_USER=your-user
 
 # export NANO_BOTS_STATE_DIRECTORY=/home/user/.local/state/nano-bots
 # export NANO_BOTS_CARTRIDGES_DIRECTORY=/home/user/.local/share/nano-bots/cartridges
@@ -56,10 +57,11 @@ export OPENAI_API_USER_IDENTIFIER=your-user
 Alternatively, if your current directory has a `.env` file with the environment variables, they will be automatically loaded:
 
 ```sh
-NANO_BOTS_ENCRYPTION_PASSWORD="UNSAFE"
 OPENAI_API_ADDRESS=https://api.openai.com
-OPENAI_API_ACCESS_TOKEN=your-token
-OPENAI_API_USER_IDENTIFIER=your-user
+OPENAI_API_KEY=your-access-token
+
+NANO_BOTS_ENCRYPTION_PASSWORD=UNSAFE
+NANO_BOTS_END_USER=your-user
 
 # NANO_BOTS_STATE_DIRECTORY=/home/user/.local/state/nano-bots
 # NANO_BOTS_CARTRIDGES_DIRECTORY=/home/user/.local/share/nano-bots/cartridges
@@ -83,15 +85,15 @@ version: '3.7'
 services:
   nano-bots:
     image: ruby:3.2.2-slim-bullseye
-    command: sh -c "gem install nano-bots -v 0.0.10 && bash"
+    command: sh -c "gem install nano-bots -v 0.1.0 && bash"
     environment:
-      NANO_BOTS_ENCRYPTION_PASSWORD: UNSAFE
       OPENAI_API_ADDRESS: https://api.openai.com
-      OPENAI_API_ACCESS_TOKEN: your-token
-      OPENAI_API_USER_IDENTIFIER: your-user
+      OPENAI_API_KEY: your-access-token
+      NANO_BOTS_ENCRYPTION_PASSWORD: UNSAFE
+      NANO_BOTS_END_USER: your-user
     volumes:
-      - ./your-cartridges:/cartridges
-      # - ./your-data:/data
+      - ./your-cartridges:/.local/share/nano-bots/cartridges
+      - ./your-state:/.local/state/nano-bots
 ```
 
 Enter the container:
@@ -104,8 +106,8 @@ Start playing:
 nb - - eval "hello"
 nb - - repl
 
-nb cartridges/assistant.yml - eval "hello"
-nb cartridges/assistant.yml - repl
+nb assistant.yml - eval "hello"
+nb assistant.yml - repl
 ```
 
 ## Usage
@@ -256,13 +258,13 @@ behaviors:
     directive: You are a helpful assistant.
 
 provider:
-  name: openai
+  id: openai
+  credentials:
+    address: ENV/OPENAI_API_ADDRESS
+    access-token: ENV/OPENAI_API_KEY
   settings:
+    user: ENV/NANO_BOTS_END_USER
     model: gpt-3.5-turbo
-    credentials:
-      address: ENV/OPENAI_API_ADDRESS
-      access-token: ENV/OPENAI_API_ACCESS_TOKEN
-      user-identifier: ENV/OPENAI_API_USER_IDENTIFIER
 ```
 
 Check the Nano Bots specification to learn more about [how to build cartridges](https://spec.nbots.io/#/README?id=cartridges).
@@ -314,33 +316,48 @@ A common strategy for deploying Nano Bots to multiple users through APIs or auto
 You can define custom end-user identifiers in the following way:
 
 ```ruby
-NanoBot.new(environment: { NANO_BOTS_USER_IDENTIFIER: 'user-a' })
-NanoBot.new(environment: { NANO_BOTS_USER_IDENTIFIER: 'user-b' })
+NanoBot.new(environment: { NANO_BOTS_END_USER: 'custom-user-a' })
+NanoBot.new(environment: { NANO_BOTS_END_USER: 'custom-user-b' })
 ```
 
-Consider that you have have the following OpenAI user identifier:
+Consider that you have the following end-user identifier in your environment:
 ```sh
-OPENAI_API_USER_IDENTIFIER=your-name
+NANO_BOTS_END_USER=your-name
+```
+
+Or a configuration in your Cartridge:
+```yml
+---
+provider:
+  id: openai
+  settings:
+    user: your-name
 ```
 
 The requests will be performed as follows:
 
 ```ruby
-NanoBot.new(environment: {NANO_BOTS_USER_IDENTIFIER: 'user-a'})
-# { user: 'your-name/user-a' }
+NanoBot.new(cartridge: '-')
+# { user: 'your-name' }
 
-NanoBot.new(environment: {NANO_BOTS_USER_IDENTIFIER: 'user-b'})
-# { user: 'your-name/user-b' }
+NanoBot.new(cartridge: '-', environment: { NANO_BOTS_END_USER: 'custom-user-a' })
+# { user: 'custom-user-a' }
+
+NanoBot.new(cartridge: '-', environment: { NANO_BOTS_END_USER: 'custom-user-b' })
+# { user: 'custom-user-b' }
 ```
 
 Actually, to enhance privacy, neither your user nor your users' identifiers will be shared in this way. Instead, they will be encrypted before being shared with the provider:
 
 ```ruby
-'your-name/user-a'
-# -onBK9GWafYz-JM8-cydhn4jd4Bkfkec5FtJ1ReCrtHCDPjkhCqUjRobG1zLnAz3BLo1kFhRW3w=
+'your-name'
+# _O7OjYUESagb46YSeUeSfSMzoO1Yg0BZqpsAkPg4j62SeNYlgwq3kn51Ob2wmIehoA==
+
+'custom-user-a'
+# _O7OjYUESagb46YSeUeSfSMzoO1Yg0BZJgIXHCBHyADW-rn4IQr-s2RvP7vym8u5tnzYMIs=
 
-'your-name/user-a'
-# RldW4_xxktCksEAR9G8aORuq3skPAc9ivWj3eye2ICCHQy8gG_R5qLMS3Fg-0lY6LwxKGQur5Ww=
+'custom-user-b'
+# _O7OjYUESagb46YSeUeSfSMzoO1Yg0BZkjUwCcsh9sVppKvYMhd2qGRvP7vym8u5tnzYMIg=
 ```
 
 In this manner, you possess identifiers if required, however, their actual content can only be decrypted by you via your secure password (`NANO_BOTS_ENCRYPTION_PASSWORD`).
@@ -352,11 +369,14 @@ To decrypt your encrypted data, once you have properly configured your password,
 ```ruby
 require 'nano-bots'
 
-NanoBot.security.decrypt('-onBK9GWafYz-JM8-cydhn4jd4Bkfkec5FtJ1ReCrtHCDPjkhCqUjRobG1zLnAz3BLo1kFhRW3w=')
-# your-name/user-b
+NanoBot.security.decrypt('_O7OjYUESagb46YSeUeSfSMzoO1Yg0BZqpsAkPg4j62SeNYlgwq3kn51Ob2wmIehoA==')
+# your-name
+
+NanoBot.security.decrypt('_O7OjYUESagb46YSeUeSfSMzoO1Yg0BZJgIXHCBHyADW-rn4IQr-s2RvP7vym8u5tnzYMIs=')
+# custom-user-a
 
-NanoBot.security.decrypt('RldW4_xxktCksEAR9G8aORuq3skPAc9ivWj3eye2ICCHQy8gG_R5qLMS3Fg-0lY6LwxKGQur5Ww=')
-# your-name/user-b
+NanoBot.security.decrypt('_O7OjYUESagb46YSeUeSfSMzoO1Yg0BZkjUwCcsh9sVppKvYMhd2qGRvP7vym8u5tnzYMIg=')
+# custom-user-b
 ```
 
 If you lose your password, you lose your data. It is not possible to recover it at all. For real.
@@ -388,5 +408,5 @@ gem build nano-bots.gemspec
 
 gem signin
 
-gem push nano-bots-0.0.10.gem
+gem push nano-bots-0.1.0.gem
 ```

data/components/provider.rb

@@ -8,11 +8,11 @@ module NanoBot
   module Components
     class Provider
       def self.new(provider, environment: {})
-        case provider[:name]
+        case provider[:id]
         when 'openai'
-          Providers::OpenAI.new(provider[:settings], environment:)
+          Providers::OpenAI.new(provider[:settings], provider[:credentials], environment:)
         else
-          raise "Unsupported provider #{provider[:name]}"
+          raise "Unsupported provider \"#{provider[:id]}\""
         end
       end
     end

data/components/providers/base.rb

@@ -6,6 +6,10 @@ module NanoBot
   module Components
     module Providers
       class Base
+        def initialize(_settings, _credentials, _environment: {})
+          raise NoMethodError, "The 'initialize' method is not implemented for the current provider."
+        end
+
         def evaluate(_payload)
           raise NoMethodError, "The 'evaluate' method is not implemented for the current provider."
         end

data/components/providers/openai.rb

@@ -9,6 +9,8 @@ module NanoBot
   module Components
     module Providers
       class OpenAI < Base
+        DEFAULT_ADDRESS = 'https://api.openai.com'
+
         CHAT_SETTINGS = %i[
           model stream temperature top_p n stop max_tokens
           presence_penalty frequency_penalty logit_bias
@@ -16,14 +18,18 @@ module NanoBot
 
         attr_reader :settings
 
-        def initialize(settings, environment: {})
+        def initialize(settings, credentials, environment: {})
           @settings = settings
+          @credentials = credentials
           @environment = environment
 
-          @client = ::OpenAI::Client.new(
-            uri_base: "#{@settings[:credentials][:address].sub(%r{/$}, '')}/",
-            access_token: @settings[:credentials][:'access-token']
-          )
+          uri_base = if @credentials[:address].nil? || @credentials[:address].to_s.strip.empty?
+                       "#{DEFAULT_ADDRESS}/"
+                     else
+                       "#{@credentials[:address].to_s.sub(%r{/$}, '')}/"
+                     end
+
+          @client = ::OpenAI::Client.new(uri_base:, access_token: @credentials[:'access-token'])
         end
 
         def stream(input)
@@ -48,16 +54,7 @@ module NanoBot
             )
           end
 
-          user = @settings[:credentials][:'user-identifier']
-
-          user_suffix = @environment && (
-            @environment['NANO_BOTS_USER_IDENTIFIER'] ||
-            @environment[:NANO_BOTS_USER_IDENTIFIER]
-          )
-
-          user = "#{user}/#{user_suffix}" if user_suffix && user_suffix != ''
-
-          payload = { model: @settings[:model], user: Crypto.encrypt(user, soft: true), messages: }
+          payload = { user: OpenAI.end_user(@settings, @environment), messages: }
 
           CHAT_SETTINGS.each do |key|
             payload[key] = @settings[key] if @settings.key?(key)
@@ -87,6 +84,27 @@ module NanoBot
             block.call({ who: 'AI', message: result.dig('choices', 0, 'message', 'content') }, true)
           end
         end
+
+        def self.end_user(settings, environment)
+          user = ENV.fetch('NANO_BOTS_END_USER', nil)
+
+          user = settings[:user] if !settings[:user].nil? && !settings[:user].to_s.strip.empty?
+
+          candidate = environment && (
+            environment['NANO_BOTS_END_USER'] ||
+            environment[:NANO_BOTS_END_USER]
+          )
+
+          user = candidate if !candidate.nil? && !candidate.to_s.strip.empty?
+
+          user = if user.nil? || user.to_s.strip.empty?
+                   'unknown'
+                 else
+                   user.to_s.strip
+                 end
+
+          Crypto.encrypt(user, soft: true)
+        end
       end
     end
   end

data/components/storage.rb

@@ -8,6 +8,31 @@ require_relative './crypto'
 module NanoBot
   module Components
     class Storage
+      def self.end_user(cartridge, environment)
+        user = ENV.fetch('NANO_BOTS_END_USER', nil)
+
+        if cartridge[:provider][:id] == 'openai' &&
+           !cartridge[:provider][:settings][:user].nil? &&
+           !cartridge[:provider][:settings][:user].to_s.strip.empty?
+          user = cartridge[:provider][:settings][:user]
+        end
+
+        candidate = environment && (
+          environment['NANO_BOTS_END_USER'] ||
+          environment[:NANO_BOTS_END_USER]
+        )
+
+        user = candidate if !candidate.nil? && !candidate.to_s.strip.empty?
+
+        user = if user.nil? || user.to_s.strip.empty?
+                 'unknown'
+               else
+                 user.to_s.strip
+               end
+
+        Crypto.encrypt(user, soft: true)
+      end
+
       def self.build_path_and_ensure_state_file!(key, cartridge, environment: {})
         path = [
           Logic::Helpers::Hash.fetch(cartridge, %i[state directory]),
@@ -18,27 +43,12 @@ module NanoBot
 
         path = "#{user_home!.sub(%r{/$}, '')}/.local/state/nano-bots" if path.nil?
 
-        prefix = environment && (
-          environment['NANO_BOTS_USER_IDENTIFIER'] ||
-          environment[:NANO_BOTS_USER_IDENTIFIER]
-        )
-
-        path = "#{path.sub(%r{/$}, '')}/ruby-nano-bots/vault"
-
-        if prefix
-          normalized = prefix.split('/').map do |part|
-            Crypto.encrypt(
-              part.to_s.gsub('.', '-').force_encoding('UTF-8').to_slug.normalize,
-              soft: true
-            )
-          end.join('/')
-
-          path = "#{path}/#{normalized}"
-        end
+        path = "#{path.sub(%r{/$}, '')}/ruby-nano-bots"
 
         path = "#{path}/#{cartridge[:meta][:author].to_slug.normalize}"
         path = "#{path}/#{cartridge[:meta][:name].to_slug.normalize}"
         path = "#{path}/#{cartridge[:meta][:version].to_s.gsub('.', '-').to_slug.normalize}"
+        path = "#{path}/#{end_user(cartridge, environment)}"
         path = "#{path}/#{Crypto.encrypt(key, soft: true)}"
         path = "#{path}/state.json"
 

data/controllers/security.rb

@@ -9,6 +9,10 @@ module NanoBot
         Components::Crypto.decrypt(content)
       end
 
+      def self.encrypt(content, soft: false)
+        Components::Crypto.encrypt(content, soft:)
+      end
+
       def self.check
         password = ENV.fetch('NANO_BOTS_ENCRYPTION_PASSWORD', nil)
         password = 'UNSAFE' unless password && password != ''

data/docker-compose.example.yml

@@ -3,12 +3,12 @@ version: '3.7'
 services:
   nano-bots:
     image: ruby:3.2.2-slim-bullseye
-    command: sh -c "apt-get update && apt-get install -y --no-install-recommends build-essential libffi-dev libsodium-dev lua5.4-dev && gem install nano-bots -v 0.0.10 && bash"
+    command: sh -c "apt-get update && apt-get install -y --no-install-recommends build-essential libffi-dev libsodium-dev lua5.4-dev && gem install nano-bots -v 0.1.0 && bash"
     environment:
-      NANO_BOTS_ENCRYPTION_PASSWORD: UNSAFE
       OPENAI_API_ADDRESS: https://api.openai.com
-      OPENAI_API_ACCESS_TOKEN: your-token
-      OPENAI_API_USER_IDENTIFIER: your-user
+      OPENAI_API_KEY: your-access-token
+      NANO_BOTS_ENCRYPTION_PASSWORD: UNSAFE
+      NANO_BOTS_END_USER: your-user
     volumes:
-      - ./your-cartridges:/cartridges
-      # - ./your-data:/data
+      - ./your-cartridges:/.local/share/nano-bots/cartridges
+      - ./your-state:/.local/state/nano-bots

data/static/cartridges/baseline.yml

@@ -1,14 +1,17 @@
 ---
 meta:
+  symbol: 🤖
   name: Unknown
-  author: Nobody
+  author: None
   version: 0.0.0
+  license: CC0-1.0
+  description: Unknown
 
 provider:
-  name: openai
+  id: openai
+  credentials:
+    address: ENV/OPENAI_API_ADDRESS
+    access-token: ENV/OPENAI_API_KEY
   settings:
+    user: ENV/NANO_BOTS_END_USER
     model: gpt-3.5-turbo
-    credentials:
-      address: ENV/OPENAI_API_ADDRESS
-      access-token: ENV/OPENAI_API_ACCESS_TOKEN
-      user-identifier: ENV/OPENAI_API_USER_IDENTIFIER

data/static/cartridges/default.yml

@@ -8,7 +8,6 @@ interfaces:
     prompt:
       - text: '🤖'
       - text: '> '
-        color: blue
   eval:
     output:
       stream: true

data/static/gem.rb

@@ -3,7 +3,7 @@
 module NanoBot
   GEM = {
     name: 'nano-bots',
-    version: '0.0.10',
+    version: '0.1.0',
     author: 'icebaker',
     summary: 'Ruby Implementation of Nano Bots: small, AI-powered bots',
     description: 'Ruby Implementation of Nano Bots: small, AI-powered bots easily shared as a single file, designed to support multiple providers such as Vicuna, OpenAI ChatGPT, Google PaLM, Alpaca, and LLaMA.',

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: nano-bots
 version: !ruby/object:Gem::Version
-  version: 0.0.10
+  version: 0.1.0
 platform: ruby
 authors:
 - icebaker
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-06-03 00:00:00.000000000 Z
+date: 2023-06-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: babosa