Better logs messages!

The Jekyll logger supports a topic, we should use it!
mkasberg · May 16, 2024 · de39128 · de39128
1 parent 03cfcd4
commit de39128
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/lib/jekyll_ai_related_posts/generator.rb b/lib/jekyll_ai_related_posts/generator.rb
@@ -9,6 +9,8 @@
 module JekyllAiRelatedPosts
   class Generator < Jekyll::Generator
     def generate(site)
+      Jekyll.logger.info "AI Related Posts:", "Generating related posts..."
+
       @site = site
       setup_database
 
@@ -18,7 +20,6 @@ def generate(site)
       end
 
       if fetch_enabled?
-        Jekyll.logger.info "[ai_related_posts] Generating related posts..."
         @embeddings_fetcher = new_fetcher
 
         @site.posts.docs.each do |p|
@@ -29,7 +30,7 @@ def generate(site)
           find_related(p)
         end
       else
-        Jekyll.logger.info "[ai_related_posts] Using cached related posts data..."
+        Jekyll.logger.info "AI Related Posts:", "Fetch disabled. Using cached related posts data."
 
         @site.posts.docs.each do |p|
           fallback_generate_related(p)
@@ -139,7 +140,7 @@ def embedding_text(post)
     end
 
     def embedding_for(post)
-      Jekyll.logger.info "[ai_related_posts] Fetching embedding for #{post.relative_path}"
+      Jekyll.logger.info "AI Related Posts:", "Fetching embedding for #{post.relative_path}"
       input = embedding_text(post)
 
       @embeddings_fetcher.embedding_for(input)
@@ -175,7 +176,7 @@ def setup_database
       SQL
       ActiveRecord::Base.connection.execute(create_vss_posts)
 
-      Jekyll.logger.debug("ai_related_posts db setup complete")
+      Jekyll.logger.debug "AI Related Posts:", "DB setup complete"
     end
   end
 end
diff --git a/lib/jekyll_ai_related_posts/open_ai_embeddings.rb b/lib/jekyll_ai_related_posts/open_ai_embeddings.rb
@@ -29,8 +29,8 @@ def embedding_for(text)
 
       res.body["data"].first["embedding"]
     rescue Faraday::Error => e
-      Jekyll.logger.error "Error response from OpanAI API!"
-      Jekyll.logger.error e.inspect
+      Jekyll.logger.error "AI Related Posts:", "Error response from OpenAI API!"
+      Jekyll.logger.error "AI Related Posts:", e.inspect
 
       raise
     end