luo 0.1.22 → 0.1.23

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 88bde636d015458883a5e90dc2e1303b47843740c9c46aec01fc4805ed324a47
-  data.tar.gz: 1a6342aa5c4fc95af0334a393a86ff096209d605c2b4c26a039754a2ca37c2ac
+  metadata.gz: 86dbccd9a0a001b6dde132666ad444d748417dd4eeedf1a9b150160a0609e93c
+  data.tar.gz: 638296ce0a26c8471b6a005e74b897221415aa392fb65f47585a75898801992e
 SHA512:
-  metadata.gz: 4d6c1b65328aab12bb181dbe4aa9c075469a721b2b402b1a11323a2e21271452052da3f88423627f0ce37a01928bab5a1086d0eb11e1704ef0d362903791610a
-  data.tar.gz: 1722fa1125d04b9e0ac977465099d932a79347a39467c52f809a1194b9d11954448fcee9cfd8d782544bcc406e41bd590d3837e7116a216482edfdeac3abea43
+  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/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.22"
+  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.22
+  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