uptriever 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f44598b07e4b7c130e49c2308587dcc675eda01710f9ad24ce2fb4284fcfea63
-  data.tar.gz: 98ba686f341d7cdb66c06db19b2c0fba4c871a9c8b92c760177a6fd8f51d673e
+  metadata.gz: d560762067e20107bf3f7e01850924035f533da723f49a0d83d076cbb05a6dfe
+  data.tar.gz: 535736316a8248d66cf6cc14723b98129de2c83a29298deac10b6de3094d2b93
 SHA512:
-  metadata.gz: 7627037babef9668fd456b79bd09e34686e543871e51eff72e8126ebf08113f3cc79517fa2a62846f192651e34d2e36bebb30a8d8d574aeeb9eef67685fa1f3a
-  data.tar.gz: a74448f80ab208acbf768c31464f730ea9402e73a68630fc4f71f372b93f26b573890d329640213dedc2afe977b9b9e5acd78fc59929e9285614ee16201677dc
+  metadata.gz: 4c8bc654e0c71835078513c631959bbaecdb5f792dcb51ec59b5e162677416790bfb4208f92baa28491abcd2485d2e2bd3fe938a83a7666bcb024dc2c22599c4
+  data.tar.gz: cc1830d5f0365bd834e3fc5374e21e3c20c2a31643658679abfb37186c0131c4891bd11e4fed9c05d4c59cf9f3daffd0825b2ee8b1ac257e4fd9c651191a5b96

data/CHANGELOG.md

@@ -2,6 +2,10 @@
 
 ## master
 
+## 0.1.2 (2025-01-03)
+
+- Fix the bug with links including PWD. ([@palkan][])
+
 ## 0.1.1 (2024-07-25)
 
 - Add ERB support to allow include dynamic fragments to configs. ([@palkan][])

data/lib/uptriever/client.rb

@@ -8,9 +8,10 @@ module Uptriever
     BASE_URL = "https://api.trieve.ai/api"
 
     attr_reader :headers
-    private attr_reader :dry_run
+    private attr_reader :dry_run, :dataset_id
 
-    def initialize(api_key, dataset, dry_run: false)
+    def initialize(api_key = ENV["TRIEVE_API_KEY"], dataset = ENV["TRIEVE_DATASET"], dry_run: false)
+      @dataset_id = dataset
       @dry_run = dry_run
       @headers = {
         "Authorization" => api_key,
@@ -28,19 +29,50 @@ module Uptriever
       perform_request("/chunk", chunk.to_json)
     end
 
+    def scroll_chunks(per_page: 100)
+      data = {
+        filters: {must: nil},
+        page_size: per_page
+      }
+
+      offset_id = nil
+
+      loop do
+        data[:offset_chunk_id] = offset_id if offset_id
+        data = perform_request("/chunks/scroll", data.to_json)
+
+        chunks = data.fetch("chunks")
+        chunks = chunks.select { _1["id"] != offset_id } if offset_id
+
+        break if chunks.empty?
+
+        chunks.each { yield _1 }
+
+        offset_id = chunks.last["id"]
+      end
+    end
+
+    def delete_chunk(id)
+      perform_request("/chunk/#{id}", method: :delete, expected_code: 204)
+    end
+
+    def usage
+      perform_request("/dataset/usage/#{dataset_id}", method: :get)
+    end
+
     private
 
-    def perform_request(path, data)
+    def perform_request(path, data = nil, method: :post, expected_code: 200)
       uri = URI.parse(BASE_URL + path)
 
       http = Net::HTTP.new(uri.host, uri.port)
       http.use_ssl = true if uri.scheme == "https"
 
-      request = Net::HTTP::Post.new(
+      request = Net::HTTP.const_get(method.to_s.capitalize).new(
         uri.request_uri,
         headers.merge("Content-Type" => "application/json")
       )
-      request.body = data
+      request.body = data if data
 
       if dry_run
         puts "[DRY RUN] Perform POST #{path}: #{data}"
@@ -49,11 +81,11 @@ module Uptriever
 
       response = http.request(request)
 
-      if response.code.to_i != 200
+      if response.code.to_i != expected_code
         raise "Invalid response code: #{response.code} (#{response.body[100...]})"
       end
 
-      JSON.parse(response.body)
+      JSON.parse(response.body) if response.body
     end
   end
 end

data/lib/uptriever/config.rb

@@ -10,7 +10,7 @@ module Uptriever
     attr_reader :config_path, :root_dir
 
     def initialize(root_dir)
-      @root_dir = root_dir
+      @root_dir = File.expand_path(root_dir)
       @config_path = File.join(root_dir, ".trieve.yml")
       raise ArgumentError, ".trieve.yml is missing in the #{root_dir}" unless File.file?(config_path)
     end

data/lib/uptriever/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Uptriever # :nodoc:
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: uptriever
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Vladimir Dementyev
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-07-25 00:00:00.000000000 Z
+date: 2025-01-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: redcarpet