llm_memory_gmail_loader 0.1.4 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 961d839a0a7637f69ad1fe8e2077f3d474bdfb6d1fc86db52e040c612ce5205c
-  data.tar.gz: 6ed95dbd86936ef59301c39c4a1a52225f2efd2775708503bdcdc20baf6e9919
+  metadata.gz: c37a2b38b54c844f82c37d863d7732e530d7a1e0d0ed6e04f996bf03dc148015
+  data.tar.gz: c49fa2520349e85383c17be41d3b89ac2c381cfaf164b99c0392894f1def0385
 SHA512:
-  metadata.gz: 8e9e1a9518d135f37aec7cb072e0c0d4762548118ff683e7a9f29d3741a545cdd42c6805ae0c81f145819eb949d51430794afd8f3a7e6beeca112a0e4ec537b7
-  data.tar.gz: bd9020e40a85d84ede0a8ef0e07c6f896b3246ff0253189cc8f7c355acb8ca487f39e1f20e7a864b91940b3e10cd2a8e9aa226b6f16f51d5ee11361bd5fb9822
+  metadata.gz: 2468d9f5250cf3b5aa62a654bfd0a448b35eba4489594997a2c4096a2bdbaab2bce2d6bb5ec58ea8e64cd9cc05cb922e4e6ba797c4f953f48ce5de66094114fe
+  data.tar.gz: 07c76830cbc0623569976bd4795ec9a682ac39dd06793d23ac910761eb3b6c9ca4a3d704eeb8d3ba0867a42d1cd2b87cf73609976804d1e5fbdc3c939515c749

data/Gemfile.lock

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

data/lib/llm_memory_gmail_loader/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module LlmMemoryGmailLoader
-  VERSION = "0.1.4"
+  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)

data/llm_memory_gmail_loader.gemspec

@@ -31,7 +31,7 @@ Gem::Specification.new do |spec|
 
   # Uncomment to register a new dependency of your gem
   # spec.add_dependency "example-gem", "~> 1.0"
-  spec.add_dependency "googleauth", "~> 1.5.2"
+  spec.add_dependency "googleauth", ">= 0.17"
   spec.add_dependency "google-apis-gmail_v1", "~> 0.27.0"
   spec.add_dependency "nokogiri", "~> 1.14.0"
 

metadata

@@ -1,29 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: llm_memory_gmail_loader
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  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
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 1.5.2
+        version: '0.17'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 1.5.2
+        version: '0.17'
 - !ruby/object:Gem::Dependency
   name: google-apis-gmail_v1
   requirement: !ruby/object:Gem::Requirement