luo 0.1.21 → 0.1.23

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0f0041ac90d5a47bfaa129bc877e38a350e0bdf757a6ea8d1da945ec2af83864
-  data.tar.gz: fafb2c21415dbe6a5fbcf1ab21477e9ef580b3122ea35c3f333d2bb1a370555b
+  metadata.gz: 86dbccd9a0a001b6dde132666ad444d748417dd4eeedf1a9b150160a0609e93c
+  data.tar.gz: 638296ce0a26c8471b6a005e74b897221415aa392fb65f47585a75898801992e
 SHA512:
-  metadata.gz: db9272f29f87859e3c364240b7ec586966ffe230ae951c56d6457f64c929e1eedacd4176cde01c6fc33489bea169bad26756d39640802ce41441c86629e02db5
-  data.tar.gz: 6b279cdfb131e7042ec94ea9b12f09c7bf3012ad782d0cd8a84b98ec197470ef78bb03776c046f3ab54d7f6382cd4e5a3825288224ef6f80fc9ac8dd720a16b8
+  metadata.gz: 8838f0489d34c03b8927d60fb9d527ab0653d1ce00d35b2f8fd5020f4183d6abfda24e7ef8fa81517dff8b31410817d9e0b813ac5507bff4d1aaa63809583e2d
+  data.tar.gz: 045d43cc7a1f0d2bb41e68ea36d5e07ce2b5da84c46daadef05054fb9a73269bbbe13b458ac9242c53075ca42f37bc0717a37fe792165c87eca60da71c1d9596

data/.idea/dictionaries/mj.xml

@@ -0,0 +1,7 @@
+<component name="ProjectDictionaryState">
+  <dictionary name="mj">
+    <words>
+      <w>marqo</w>
+    </words>
+  </dictionary>
+</component>

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    luo (0.1.19)
+    luo (0.1.22)
       dotenv (~> 2.8, >= 2.8.1)
       dry-cli (~> 1.0)
       dry-configurable (~> 1.0, >= 1.0.1)

data/lib/luo/http_client.rb

@@ -2,10 +2,10 @@
 
 module Luo
   module HttpClient
-    def self.init_client
+    def self.init_client(options={})
       Module.new do
         define_method :client do
-          @client ||= Faraday.new(self.class.config.host) { |conn|
+          @client ||= Faraday.new(self.class.config.host, options) { |conn|
             conn.request :authorization, 'Bearer', self.class.config.access_token if self.class.config.respond_to?(:access_token) && !self.class.config.access_token.nil?
             conn.request :retry, max: (self.class.config.retries || 3), interval: 0.05,
                          interval_randomness: 0.5, backoff_factor: 2,

data/lib/luo/marqo.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+module Luo
+  class Marqo
+    include Configurable
+    setting :host, default: "http://localhost:8882"
+    setting :x_api_key, default: nil
+    setting :retries, default: ENV.fetch('OPENAI_REQUEST_RETRIES', 3).to_i
+
+    include HttpClient.init_client(headers: { 'X-API-KEY' => config.x_api_key })
+
+    SEARCH_PARAMS = Dry::Schema.Params do
+      required(:q).filled(:string)
+      optional(:limit).maybe(:integer)
+      optional(:offset).maybe(:integer)
+      optional(:filter).maybe(:string)
+      optional(:searchableAttributes).maybe(:array)
+      optional(:showHighlights).maybe(:bool)
+      optional(:searchMethod).value(included_in?: %w[TENSOR LEXICAL])
+      optional(:attributesToRetrieve).maybe(:array)
+      optional(:reRanker).maybe(:string)
+      optional(:boost).maybe(:hash)
+      optional(:image_download_headers).maybe(:hash)
+      optional(:context).maybe(:hash)
+      optional(:scoreModifiers).maybe(:hash)
+      optional(:modelAuth).maybe(:hash)
+    end
+
+    def index(index_name)
+      @index_name = index_name
+      self
+    end
+
+    ##
+    # 添加文档到向量数据库
+    def add_documents(hash, non_tensor_fields: nil)
+      client.post("/indexes/#{@index_name}/documents") do |req|
+        req.params[:non_tensor_fields] = non_tensor_fields unless non_tensor_fields.nil?
+        req.body = hash
+      end
+    end
+
+    def refresh
+      client.post("/indexes/#{@index_name}/refresh")
+    end
+
+    def search(params)
+      params = SEARCH_PARAMS.call(params)
+      return params.errors unless params.success?
+      client.post("/indexes/#{@index_name}/search", params.to_h)
+    end
+
+    def delete(id_or_ids)
+      id_or_ids = [id_or_ids] if id_or_ids.is_a?(String)
+      client.post("/indexes/#{@index_name}/documents/delete-batch", id_or_ids)
+    end
+
+    def remove
+      client.delete("/indexes/#{@index_name}")
+    end
+  end
+end

data/lib/luo/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Luo
-  VERSION = "0.1.21"
+  VERSION = "0.1.23"
 end

data/lib/luo.rb

@@ -14,6 +14,7 @@ require 'yaml'
 require 'tty-markdown'
 require 'fileutils'
 require 'dry-initializer'
+require 'uri'
 
 require "zeitwerk"
 loader = Zeitwerk::Loader.for_gem

data/luo.ps1

@@ -0,0 +1,31 @@
+function Get-ScriptRoot {
+    $result = $PSScriptRoot
+    if ($result.Length -eq 0) {
+        $result = split-path -parent $MyInvocation.MyCommand.Definition
+    }
+    return $result
+}
+
+function Set-LuoPath {
+    $test_path = ([System.Environment]::GetEnvironmentVariable('PATH') | findstr (Get-ScriptRoot)).Length
+    if ($test_path -eq 0) {
+        [System.Environment]::SetEnvironmentVariable('PATH', "${Env:PATH};" + (Get-ScriptRoot))
+        $Env:PATH += ";" + (Get-ScriptRoot)
+        Write-Host "=== PATH set for Luo ==="
+    }
+}
+
+Set-LuoPath
+
+if ($args[0] -eq "update") {
+    Write-Host "Pulling the latest Docker image..."
+    docker pull ghcr.io/mjason/luo:latest
+}
+
+Write-Host "Running the Docker container..."
+
+docker run --rm -it `
+    -p 8888:8888 `
+    -v "${PWD}:/workdir" `
+    ghcr.io/mjason/luo:latest `
+    $args

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: luo
 version: !ruby/object:Gem::Version
-  version: 0.1.21
+  version: 0.1.23
 platform: ruby
 authors:
 - MJ
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-05-12 00:00:00.000000000 Z
+date: 2023-05-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: zeitwerk
@@ -175,6 +175,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - ".idea/.gitignore"
+- ".idea/dictionaries/mj.xml"
 - ".idea/inspectionProfiles/Project_Default.xml"
 - ".idea/luo.iml"
 - ".idea/misc.xml"
@@ -201,6 +202,7 @@ files:
 - lib/luo/http_client.rb
 - lib/luo/init_project.rb
 - lib/luo/loader.rb
+- lib/luo/marqo.rb
 - lib/luo/memory_history.rb
 - lib/luo/messages.rb
 - lib/luo/open_ai.rb
@@ -233,6 +235,7 @@ files:
 - luo
 - luo.gemspec
 - luo.ipynb
+- luo.ps1
 - sig/luo.rbs
 - sig/luo/agent.rbs
 - sig/luo/agent_runner_base.rbs