smart_prompt 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 846477d8ddb503119a9233f9ce85d2094c1755be52adcfc0515b0a7209524a3d
-  data.tar.gz: 4f165cedebc54e9d8ae292e3fd03329ec04150a8e0804fcb90234b7837135e6d
+  metadata.gz: 844c87d47dccd945bedcab69c58377ae144238ac9087b49219041382db539146
+  data.tar.gz: 3e70f88fbe8d26a0a15408e73ae9060b9f711a18e6b44eb561910b90156ca14b
 SHA512:
-  metadata.gz: 7f3521861b4690b233c9b82d6f08123b9859c0fed28e40274fb4eaa014aa93b2df63121aff77451c9b1f5f93f9eeb683c5b044c94cef5e7d11a8e2151c5e97f8
-  data.tar.gz: 9c78f290101fd158403d2feb9e84312d8ffb90a22a32a57e91d96f5c287a38cfd8cb8c318817e860d7fb09bd5584310e6d397fce490deaa29ca52bafb3db7fba
+  metadata.gz: 398854ce070f96f794ae944cc656a10d2f4752268cb35b4f9a857fcd8a722489b3ee112de97e5cdfe1cc1efb75cf92e4cd4a9b36cb338cbd8de6e22f1d6cc7dc
+  data.tar.gz: babac8b7d0479eb376df22b00cd693e6e657293e6b28b654aaa6b2ef4f9fcfdccb9709d6f1dd0b066cb6f22c07eed0173426816121061a80c91fc9afdd62ba06

data/lib/smart_prompt/conversation.rb

@@ -5,6 +5,7 @@ module SmartPrompt
     attr_reader :messages, :last_response, :config_file
 
     def initialize(engine)
+      SmartPrompt.logger.info "Create Conversation"
       @messages = []
       @engine = engine
       @adapters = engine.adapters
@@ -26,6 +27,7 @@ module SmartPrompt
 
     def prompt(template_name, params = {})
       template_name = template_name.to_s
+      SmartPrompt.logger.info "Use template #{template_name}"
       raise "Template #{template_name} not found" unless @templates.key?(template_name)
       content = @templates[template_name].render(params)
       @messages << { role: 'user', content: content }

data/lib/smart_prompt/engine.rb

@@ -2,6 +2,7 @@ module SmartPrompt
     class Engine
       attr_reader :config_file, :config, :adapters, :current_adapter, :llms, :templates
       def initialize(config_file)
+        SmartPrompt.logger.info "Start create the SmartPrompt engine."
         @config_file = config_file
         @adapters={}
         @llms={}
@@ -10,6 +11,7 @@ module SmartPrompt
       end
 
       def load_config(config_file)
+        SmartPrompt.logger.info "Loading configuration from file: #{config_file}"
         @config_file = config_file
         @config = YAML.load_file(config_file)
         @config['adapters'].each do |adapter_name, adapter_class|
@@ -35,14 +37,25 @@ module SmartPrompt
       end
         
       def call_worker(worker_name, params = {})
+        SmartPrompt.logger.info "Calling worker: #{worker_name} with params: #{params}"
         worker = get_worker(worker_name)
-        worker.execute(params)
+        
+        begin
+          result = worker.execute(params)
+          SmartPrompt.logger.info "Worker #{worker_name} executed successfully"
+          result
+        rescue => e
+          SmartPrompt.logger.error "Error executing worker #{worker_name}: #{e.message}"
+          SmartPrompt.logger.debug e.backtrace.join("\n")
+          raise
+        end
       end
   
       private
   
       def get_worker(worker_name)
-        worker = Worker.new(worker_name, self)
+        SmartPrompt.logger.info "Creating worker instance for: #{worker_name}"
+        Worker.new(worker_name, self)
       end
     end
   end

data/lib/smart_prompt/llm_adapter.rb

@@ -7,10 +7,12 @@ require 'ollama-ai'
 module SmartPrompt
   class LLMAdapter
     def initialize(config)
+      SmartPrompt.logger.info "Start create the SmartPrompt LLMAdapter."
       @config = config
     end
 
     def send_request(messages)
+      SmartPrompt.logger.error "LLMAdapter: Subclasses must implement send_request"
       raise NotImplementedError, "Subclasses must implement send_request"
     end
   end
@@ -30,11 +32,13 @@ module SmartPrompt
     end
 
     def send_request(messages, model=nil)
+      SmartPrompt.logger.info "OpenAIAdapter: Sending request to OpenAI"
       if model
         model_name = model
       else
         model_name = @config['model']        
       end
+      SmartPrompt.logger.info "OpenAIAdapter: Using model #{model_name}"
       response = @client.chat(
         parameters: {
           model: model_name,
@@ -42,6 +46,7 @@ module SmartPrompt
           temperature: @config['temperature'] || 0.7
         }
       )
+      SmartPrompt.logger.info "OpenAIAdapter: Received response from OpenAI"
       response.dig("choices", 0, "message", "content")
     end
   end
@@ -54,12 +59,14 @@ module SmartPrompt
       )
     end
     def send_request(messages, model=nil)
+      SmartPrompt.logger.info "LlamacppAdapter: Sending request to Llamacpp"
       response = @client.chat(
         parameters: {
           messages: messages,
           temperature: @config['temperature'] || 0.7
         }
       )
+      SmartPrompt.logger.info "LlamacppAdapter: Received response from Llamacpp"
       response.dig("choices", 0, "message", "content")
     end
   end
@@ -71,11 +78,13 @@ module SmartPrompt
     end
 
     def send_request(messages, model=nil)
+      SmartPrompt.logger.info "OllamaAdapter: Sending request to Ollama"
       if model
         model_name = model
       else
         model_name = @config['model']        
       end
+      SmartPrompt.logger.info "OllamaAdapter: Using model #{model_name}"
       response = @client.generate(
         {
           model: model_name,
@@ -83,6 +92,7 @@ module SmartPrompt
           stream: false
         }
       )
+      SmartPrompt.logger.info "OllamaAdapter: Received response from Ollama"
       return response[0]["response"]
     end
   end

data/lib/smart_prompt/version.rb

@@ -1,3 +1,3 @@
 module SmartPrompt
-  VERSION = "0.1.4"
+  VERSION = "0.1.5"
 end

data/lib/smart_prompt/worker.rb

@@ -3,6 +3,7 @@ module SmartPrompt
     attr_reader :name, :config_file
 
     def initialize(name, engine)
+      SmartPrompt.logger.info "Create worker's name is #{name}"
       @name = name
       @engine = engine
       @config = engine.config

data/lib/smart_prompt.rb

@@ -7,6 +7,7 @@ require File.expand_path('../smart_prompt/worker', __FILE__)
 
 module SmartPrompt
   class Error < StandardError; end
+  attr_writer :logger
 
   def self.define_worker(name, &block)
     Worker.define(name, &block)
@@ -16,4 +17,10 @@ module SmartPrompt
     worker = Worker.new(name, config_file)
     worker.execute(params)
   end
+  
+  def self.logger
+    @logger ||= Logger.new($stdout).tap do |log|
+      log.progname = self.name
+    end
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: smart_prompt
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - zhuang biaowei
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-10-02 00:00:00.000000000 Z
+date: 2024-10-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: yaml