luo 0.1.8 → 0.1.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e9c2a48667ea14c7c9d58e171baa44342929cb9ef585867543ba545559623cb8
-  data.tar.gz: d586451833334193d7da2f08efc9b0ffa0a06e66dbf3e1de1d18c8c46cf53706
+  metadata.gz: 102af9d7257f2583c64ff49ad555fd6bfe6127e2d5ef5db2817ef88c0deff2dd
+  data.tar.gz: f407808206d508f2741a165fe45534d5d951dd3f064481ed75e683cee577a6a2
 SHA512:
-  metadata.gz: f45e76d27242650bf196acec577d910b7134ccfe693742eb5c2e200c9e9b68a91b8113d8a3c198231ae6fc4cf96caffea918fb6f1c40ad0db0ad5742eaf2b03c
-  data.tar.gz: ed3a621280d191459520076628630716b5e684a01623df426dd048d28df2eb55fbf2ff7dc3eaa5e235a6518a858bc8b0fd65e1bc7dec8acfc7f99e05d003c1d6
+  metadata.gz: 732fd0060a870fa7439d7699fad27929f21873c1051e7ec79dc70478501d5d66a7eff359184530b214624e01c3446f5ee75ead7eee6227d24cebdd074ee100dc
+  data.tar.gz: 9327870633acba5bcbea062e3bbee7487d6d91cb9168d645b641422742d25c95fb2574f1d55bbc569bb36538a631795b25217b4449314132326836338d08b88a

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    luo (0.1.7)
+    luo (0.1.8)
       dotenv (~> 2.8, >= 2.8.1)
       dry-cli (~> 1.0)
       dry-configurable (~> 1.0, >= 1.0.1)

data/lib/luo/agent.rb

@@ -2,6 +2,8 @@
 
 module Luo
   class Agent
+    extend Dry::Initializer
+
     attr_reader :context, :action_input, :client
     def initialize(context: nil, action_input: nil, client: nil)
       @context = context
@@ -27,7 +29,7 @@ module Luo
           result = instance_eval(&block)
           if result.nil?
             messages = context.messages.to_a[0...-1] + [{role: :user, content: context.user_input}]
-            context.final_result = client.chat(messages)
+            context.final_result = client&.chat(messages)
           else
             context.final_result = result
           end
@@ -35,7 +37,7 @@ module Luo
       end
 
       alias_method :on_call_with_fallback, :on_call_with_final_result
-      
+
       def self.create_parameter_method(method_name, not_provided = Object.new, &block)
         define_method(method_name.to_sym) do |content = not_provided|
           if content === not_provided

data/lib/luo/aiui.rb

@@ -23,7 +23,7 @@ module Luo
       client.post('/external/ls_log/aiui_request', params.to_h)
     end
 
-    def chat(messages)
+    def chat(messages, temperature: nil)
       if messages.is_a?(Messages)
         messages = messages.to_a
       end

data/lib/luo/open_ai.rb

@@ -48,13 +48,13 @@ module Luo
       embeddings(params).body.dig("data").map { |v| v["embedding"] }
     end
 
-    def chat(messages)
+    def chat(messages, temperature: nil)
       if messages.is_a?(Messages)
         messages = messages.to_a
       end
       params = PARAMS.call(
         model: 'gpt-3.5-turbo',
-        temperature: config.temperature,
+        temperature: temperature || config.temperature,
         messages: messages
       )
       return params.errors unless params.success?

data/lib/luo/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Luo
-  VERSION = "0.1.8"
+  VERSION = "0.1.9"
 end

data/lib/luo/xinghuo.rb

@@ -30,7 +30,7 @@ module Luo
       client.post('/external/ls_log/xf_completions', params.to_h)
     end
 
-    def chat(messages)
+    def chat(messages, temperature: nil)
       if messages.is_a?(Messages)
         messages = messages.to_a
       end
@@ -39,7 +39,7 @@ module Luo
         domain: config.domain,
         messages: messages,
         max_tokens: config.max_tokens,
-        random_threshold: config.temperature,
+        random_threshold: temperature || config.temperature,
         uid: config.uid.call
       )
       return params.errors unless params.success?

data/lib/luo.rb

@@ -13,6 +13,7 @@ require 'dry/cli'
 require 'yaml'
 require 'tty-markdown'
 require 'fileutils'
+require 'dry-initializer'
 
 require "zeitwerk"
 loader = Zeitwerk::Loader.for_gem

data/luo.gemspec

@@ -37,7 +37,6 @@ Gem::Specification.new do |spec|
   spec.add_dependency 'dry-configurable', '~> 1.0', '>= 1.0.1'
   spec.add_dependency 'dry-cli', '~> 1.0'
   spec.add_dependency 'tty-markdown', '~> 0.7.2'
-  # spec.add_dependency "example-gem", "~> 1.0"
 
   spec.add_development_dependency "rspec", '~> 3.12'
 

data/sig/luo/agent.rbs

@@ -1,7 +1,7 @@
 module Luo
   class Agent
     @action_input: untyped
-    @context: AgentRunnerContext
+    @context: AgentRunnerContext?
 
     attr_reader self.agent_desc: String
     attr_reader self.agent_name: String
@@ -9,6 +9,7 @@ module Luo
     def self.on_call_with_final_result: -> untyped
 
     attr_reader action_input: untyped
+    attr_reader client: _Client?
     attr_reader context: AgentRunnerContext
 
     def initialize: (AgentRunnerContext) -> untyped
@@ -19,5 +20,7 @@ module Luo
 
     attr_reader input: String
     attr_reader row_input: String?
+
+    def call: -> untyped
   end
 end

data/sig/luo/agent_runner_context.rbs

@@ -1,5 +1,6 @@
 module Luo
   class AgentRunnerContext
+    @client: untyped
     @histories: MemoryHistory
 
     @running_agents: Array[Agent]
@@ -14,6 +15,8 @@ module Luo
 
     def client: -> untyped
 
+    def client=: -> untyped
+
     def have_running_agents: -> Set[Agent]
 
     def histories: -> MemoryHistory

data/sig/luo/client.rbs

@@ -0,0 +1,5 @@
+module Luo
+  interface _Client
+    def chat: (untyped, temperature: Integer) -> String
+  end
+end

data/sig/luo/open_ai.rbs

@@ -11,6 +11,8 @@ module Luo
 
     def create_embedding: -> untyped
 
+    def create_embeddings: -> untyped
+
     def embeddings: -> untyped
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: luo
 version: !ruby/object:Gem::Version
-  version: 0.1.8
+  version: 0.1.9
 platform: ruby
 authors:
 - MJ
@@ -228,6 +228,7 @@ files:
 - sig/luo/agent_runner_base.rbs
 - sig/luo/agent_runner_context.rbs
 - sig/luo/aiui.rbs
+- sig/luo/client.rbs
 - sig/luo/configurable.rbs
 - sig/luo/env.rbs
 - sig/luo/helpers.rbs