llm_memory_gmail_loader 0.1.5 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f5488bc6c4a2535d24b1d3e8f3649ecb14aaf9f4624cfb27731e9f1a1f20330a
-  data.tar.gz: db11e4bd9ab7f43c1d3be98ef3ac02003150befecd269bcc946da9d091fc2427
+  metadata.gz: c37a2b38b54c844f82c37d863d7732e530d7a1e0d0ed6e04f996bf03dc148015
+  data.tar.gz: c49fa2520349e85383c17be41d3b89ac2c381cfaf164b99c0392894f1def0385
 SHA512:
-  metadata.gz: 351b811870a323dfead2744273d34af66596bb6745b407c72d311fc30efe93317c2191f12d7af6805e55485a7ad3dad71245e114b9b3104cf76f2c434545e5a7
-  data.tar.gz: 8583d952e71f223f9823e9458922cc0d12b528e23d77e3c1f16ef9ee5ec455d448cdf3f8ca67f1842f3d903ea4518bb8986d7d5b10964857bdd8a5bd41a3df14
+  metadata.gz: 2468d9f5250cf3b5aa62a654bfd0a448b35eba4489594997a2c4096a2bdbaab2bce2d6bb5ec58ea8e64cd9cc05cb922e4e6ba797c4f953f48ce5de66094114fe
+  data.tar.gz: 07c76830cbc0623569976bd4795ec9a682ac39dd06793d23ac910761eb3b6c9ca4a3d704eeb8d3ba0867a42d1cd2b87cf73609976804d1e5fbdc3c939515c749

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    llm_memory_gmail_loader (0.1.5)
+    llm_memory_gmail_loader (0.1.6)
       google-apis-gmail_v1 (~> 0.27.0)
       googleauth (>= 0.17)
       nokogiri (~> 1.14.0)

data/lib/llm_memory_gmail_loader/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module LlmMemoryGmailLoader
-  VERSION = "0.1.5"
+  VERSION = "0.1.6"
 end

data/lib/llm_memory_gmail_loader.rb

@@ -50,7 +50,9 @@ module LlmMemoryGmailLoader
       # authorizer.sub = "user@example.com" # Replace with the user's email address
     end
 
-    def list_emails(email:, limit: 100, query: "label:sent")
+    def list_emails(email:, limit: nil, query: nil)
+      limit ||= 100
+      query ||= "label:sent"
       @service.authorization.sub = email
       next_page_token = nil
       sent_emails = []
@@ -58,6 +60,7 @@ module LlmMemoryGmailLoader
       max_results = (limit < 100) ? limit : 100
       loop do
         result = @service.list_user_messages("me", q: query, page_token: next_page_token, max_results: max_results)
+        break if result.messages.nil?
         sent_emails.concat(result.messages) if result.messages
         count += result.messages.length
         next_page_token = result.next_page_token
@@ -164,6 +167,7 @@ module LlmMemoryGmailLoader
     def load(args)
       emails = args[:emails]
       limit = args[:limit]
+      query = args[:query]
       raise "emails is required" if emails.nil? || emails.length == 0
 
       @service ||= Google::Apis::GmailV1::GmailService.new
@@ -171,11 +175,7 @@ module LlmMemoryGmailLoader
 
       results = []
       emails.each do |email|
-        results += if limit.nil?
-          list_emails(email: email)
-        else
-          list_emails(email: email, limit: limit)
-        end
+        results += list_emails(email: email, limit: limit, query: query)
       end
 
       create_documents(results)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: llm_memory_gmail_loader
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.1.6
 platform: ruby
 authors:
 - Shohei Kameda
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-07-27 00:00:00.000000000 Z
+date: 2023-07-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: googleauth