smart_prompt 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5405e5a08476567f1737276d32ac24849eaa693c9ed302ca1c633a372e230996
-  data.tar.gz: f5720ede56fc9eff6afa79990e25538dd6749b67bf85638f23ed94527e302401
+  metadata.gz: 48814b17b9e142698d6cb49efd4ab15195bddce4fca7ac5f111b52e5e8b3e04b
+  data.tar.gz: c0e23945a97549403ab6b5d3225e3b281b0d2f6f6f6097f5158ae98edb70d722
 SHA512:
-  metadata.gz: a2cff6947e229493681d555ca407563d6aff0e9543d8fecb843fb1ee5c2f2534ff9e9994eb44a1ea4f9ddacdfc12a95da7b119aa2c2cd55ee638a65682a250c6
-  data.tar.gz: 96db920d1c87b839dadeecf3b2552e6e9179d6013d3867f72e5e23370d63294261d134b9d9eb13ad6aa51743e9282d3dceac4743333d35e2554dbb8138e50eca
+  metadata.gz: 1dac2c898d593345d1f1d8c9508faabda41c2954938171f99aa9999fbe4f34532124e441ad2ccbb265bc6ceb65fb188fd7b60d8e8f1439ad478f6d2d62a38018
+  data.tar.gz: 851707ebcac1db79aa0fc29a8a92dd9e1571dd0d1207d4cd9ac75fb5992ecfac1bac09eec8599594e247066e586d4bd48b6fddf3da073e168f2039823dcce747

data/lib/smart_prompt/conversation.rb

@@ -8,14 +8,15 @@ module SmartPrompt
       @messages = []
       @engine = engine
       @adapters = engine.adapters
+      @llms = engine.llms
       @templates = engine.templates
       @current_adapter = engine.current_adapter
       @last_response = nil
     end
 
-    def use(adapter_name)
-      raise "Adapter #{adapter_name} not configured" unless @adapters.key?(adapter_name)
-      @current_adapter = adapter_name
+    def use(llm_name)
+      raise "Adapter #{adapter_name} not configured" unless @llms.key?(llm_name)
+      @current_llm = @llms[llm_name]
       self
     end
 
@@ -38,8 +39,8 @@ module SmartPrompt
     end
 
     def send_msg
-      raise "No adapter selected" if @current_adapter.nil?
-      @last_response = @adapters[@current_adapter].send_request(@messages, @model_name)
+      raise "No LLM selected" if @current_llm.nil?
+      @last_response = @current_llm.send_request(@messages, @model_name)
       @messages=[]
       @messages << { role: 'system', content: @sys_msg }
       @last_response

data/lib/smart_prompt/engine.rb

@@ -1,9 +1,10 @@
 module SmartPrompt
     class Engine
-      attr_reader :config_file, :config, :adapters, :current_adapter, :templates
+      attr_reader :config_file, :config, :adapters, :current_adapter, :llms, :templates
       def initialize(config_file)
         @config_file = config_file
         @adapters={}
+        @llms={}
         @templates={}
         load_config(config_file)
       end
@@ -11,13 +12,18 @@ module SmartPrompt
       def load_config(config_file)
         @config_file = config_file
         @config = YAML.load_file(config_file)
-        @config['adapters'].each do |adapter_name, adapter_config|
-          adapter_class = SmartPrompt.const_get("#{adapter_name.capitalize}Adapter")          
-          @adapters[adapter_name] = adapter_class.new(adapter_config)
+        @config['adapters'].each do |adapter_name, adapter_class|
+          adapter_class = SmartPrompt.const_get(adapter_class)
+          @adapters[adapter_name] = adapter_class
         end
-        @current_adapter = @config['default_adapter'] if @config['default_adapter']        
-        @config['templates'].each do |template_name, template_file|
-          @templates[template_name] = PromptTemplate.new(template_file)
+        @config['llms'].each do |llm_name,llm_config|
+          adapter_class = @adapters[llm_config['adapter']]
+          @llms[llm_name]=adapter_class.new(llm_config)
+        end
+        @current_llm = @config['default_llm'] if @config['default_llm']
+        Dir.glob(File.join(@config['template_path'], '*.erb')).each do |file|
+          template_name = file.gsub(@config['template_path']+"/","").gsub("\.erb","")
+          @templates[template_name] = PromptTemplate.new(file)
         end
         load_workers
       end

data/lib/smart_prompt/llm_adapter.rb

@@ -15,7 +15,7 @@ module SmartPrompt
     end
   end
 
-  class OpenaiAdapter < LLMAdapter
+  class OpenAIAdapter < LLMAdapter
     def initialize(config)
       super
       @client = OpenAI::Client.new(

data/lib/smart_prompt/version.rb

@@ -1,3 +1,3 @@
 module SmartPrompt
-  VERSION = "0.1.2"
+  VERSION = "0.1.3"
 end

metadata

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