smart_prompt 0.1.8 → 0.1.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ab0d62158bbd98af9d2f57d9708426cceb95c2d3b2797abc0094de3caf96adf8
-  data.tar.gz: d2a50c4ede575eb70caeda1893b60412b04afdec665690b8fd2d4d0f3bdd410e
+  metadata.gz: dce823cedfccc6a2288e6a37b2a91571d4bc784d26440d11503697c43d5558de
+  data.tar.gz: b8f4cbb2104d640f423f727afcb38962a7cfa663887ada0d32ec3a61d2ce513c
 SHA512:
-  metadata.gz: 6d61c7f8db100a715b16a8bd08ef27bf49f3e38127d68cb37a5fc55c3d5ee781e505d8c4acbc6ac98206bd2aeb22bcdd4dc6d749ed65293fecb5abbc361ffac9
-  data.tar.gz: 7f8ff17971d9979a95400bb1e4497737fab2c503d69a1c415a0003ff514400b55ade390d0f007a146e43569978b98e29f5e308091d74e25cd8b405689b73b136
+  metadata.gz: 70c6bd8c45f433cbc4a60bd4d3964f694b836178ff09a84873aeaedc62cc9064e5c4d043d45b7f864d95b5f60f3a8eb6571230abb9c45385a8525014417493db
+  data.tar.gz: 43eca2651ae52dab6e8ca1fff67acbc1099e3adc839c8035111148aae45b3bdcaf64f8b1f5daed8e45c4b2962449eef86c19a16d7e69fe59f874af2e4b66b4d1

data/lib/smart_prompt/api_handler.rb

@@ -0,0 +1,17 @@
+module SmartPrompt
+    module APIHandler
+        MAX_RETRIES = 3
+        RETRY_OPTIONS = {
+          tries: MAX_RETRIES,
+          base_interval: 1,
+          max_interval: 10,
+          rand_factor: 0.5,
+          on: [
+            Errno::ECONNRESET,
+            Errno::ECONNABORTED,
+            Errno::EPIPE,
+            Errno::ETIMEDOUT
+          ]
+        }
+    end
+end

data/lib/smart_prompt/conversation.rb

@@ -28,6 +28,10 @@ module SmartPrompt
       @model_name = model_name
     end
 
+    def temperature(temperature)
+      @temperature = temperature
+    end
+
     def prompt(template_name, params = {})
       if template_name.class == Symbol
         template_name = template_name.to_s
@@ -50,7 +54,7 @@ module SmartPrompt
 
     def send_msg
       raise "No LLM selected" if @current_llm.nil?
-      @last_response = @current_llm.send_request(@messages, @model_name)
+      @last_response = @current_llm.send_request(@messages, @model_name, @temperature)
       @messages=[]
       @messages << { role: 'system', content: @sys_msg }
       @last_response

data/lib/smart_prompt/db_adapter.rb

@@ -0,0 +1,45 @@
+require "sequel"
+require "json"
+
+module SmartPrompt
+  class DBAdapter
+    attr_reader :db, :tables
+    def initialize(config)
+      db_uri = config[:database]
+      @db = Sequel.connect(db_uri)
+      @tables = {}
+      @db.tables.each do |table_name|
+        define_table(table_name)
+      end
+    end
+
+    def define_table(table_name, class_name=table_name.to_s.capitalize)
+      class_define = <<-EOT
+        class #{class_name} < Sequel::Model(:#{table_name})
+        end
+      EOT
+      eval(class_define)
+      @tables[table_name] = eval(class_name)
+    end
+
+    def get_table_schema(table_name)
+      @tables[table_name].db_schema
+    end
+
+    def get_table_schema_str(table_name)
+      JSON.pretty_generate(get_table_schema(table_name))
+    end
+
+    def get_db_schema
+      schema = {}
+      @db.tables.each do |table_name|
+        schema[table_name] = get_table_schema(table_name)
+      end
+      schema
+    end
+
+    def get_db_schema_str
+      JSON.pretty_generate(get_db_schema)
+    end
+  end
+end

data/lib/smart_prompt/openai_adapter.rb

@@ -31,7 +31,7 @@ module SmartPrompt
       end
     end
 
-    def send_request(messages, model=nil)
+    def send_request(messages, model=nil, send_request=0.7)
       SmartPrompt.logger.info "OpenAIAdapter: Sending request to OpenAI"
       if model
         model_name = model
@@ -44,7 +44,7 @@ module SmartPrompt
           parameters: {
             model: model_name,
             messages: messages,
-            temperature: @config['temperature'] || 0.7
+            temperature: @config['temperature'] || send_request
           }
         )
       rescue OpenAI::Error => e

data/lib/smart_prompt/version.rb

@@ -1,3 +1,3 @@
 module SmartPrompt
-  VERSION = "0.1.8"
+  VERSION = "0.1.9"
 end

metadata

@@ -1,14 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: smart_prompt
 version: !ruby/object:Gem::Version
-  version: 0.1.8
+  version: 0.1.9
 platform: ruby
 authors:
 - zhuang biaowei
-autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-11-28 00:00:00.000000000 Z
+date: 2025-01-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: yaml
@@ -39,19 +38,19 @@ dependencies:
       - !ruby/object:Gem::Version
         version: 7.1.0
 - !ruby/object:Gem::Dependency
-  name: ollama-ai
+  name: ollama-ruby
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.3.0
+        version: 0.13.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.3.0
+        version: 0.13.0
 - !ruby/object:Gem::Dependency
   name: json
   requirement: !ruby/object:Gem::Requirement
@@ -107,7 +106,9 @@ files:
 - README.md
 - Rakefile
 - lib/smart_prompt.rb
+- lib/smart_prompt/api_handler.rb
 - lib/smart_prompt/conversation.rb
+- lib/smart_prompt/db_adapter.rb
 - lib/smart_prompt/engine.rb
 - lib/smart_prompt/llamacpp_adapter.rb
 - lib/smart_prompt/llm_adapter.rb
@@ -124,7 +125,6 @@ metadata:
   homepage_uri: https://github.com/zhuangbiaowei/smart_prompt
   source_code_uri: https://github.com/zhuangbiaowei/smart_prompt
   changelog_uri: https://github.com/zhuangbiaowei/smart_prompt/blob/master/CHANGELOG.md
-post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -139,8 +139,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.23
-signing_key: 
+rubygems_version: 3.6.2
 specification_version: 4
 summary: A smart prompt management and LLM interaction gem
 test_files: []