Patch up ContextualLogger (#785)

* Ollama logging uses the current Langchain.logger.level * ContextualLogger outputs everything passed in
patterns-ai-core · Sep 18, 2024 · 77a8229 · 77a8229
1 parent bbb9510
commit 77a8229
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/lib/langchain/contextual_logger.rb b/lib/langchain/contextual_logger.rb
@@ -49,6 +49,8 @@ def method_missing(method, *args, **kwargs, &block)
         "[#{for_class_name}]:"
       end
       log_line_parts << colorize(args.first, MESSAGE_COLOR_OPTIONS[method])
+      log_line_parts << kwargs if !!kwargs && kwargs.any?
+      log_line_parts << block.call if block
       log_line = log_line_parts.compact.join(" ")
 
       @logger.send(

diff --git a/lib/langchain/llm/ollama.rb b/lib/langchain/llm/ollama.rb
@@ -270,7 +270,7 @@ def client
         conn.request :json
         conn.response :json
         conn.response :raise_error
-        conn.response :logger, nil, {headers: true, bodies: true, errors: true}
+        conn.response :logger, Langchain.logger, {headers: true, bodies: true, errors: true}
       end
     end