ticuna 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 265d5920dcf993177bbac8fbea70aff8dedadc946d21dfc2ab3ae8df6c56048d
-  data.tar.gz: 12355c8e356b369df4c59cb3b8ac8f9502f5ca1a06928cccd7b6a54e08fce65c
+  metadata.gz: 25afb92c6ab2ba62fe78323b2c25e8a6f87d08d0297139ea4433e52de1156426
+  data.tar.gz: 90d7976a68c9ae75c972bda54765e11d4dd9050c32d2f6433852b15a607f2b11
 SHA512:
-  metadata.gz: 970adeae5f453b7e5143375867ad5868e9a6c925b73cba14e73409a2bbc8bc9fa05271d7a2570b029cbb42219da6fb9f64a92d86157bc5e6370639153e44ee7e
-  data.tar.gz: 9b98707287ad15dc2568e054892d23088b278fc458f427199cf74345bb4597661918ecaac61d0630753f1c35e322075b0fde3193abe45b38d73150cefc59a2a7
+  metadata.gz: 368e1eefe0937908dbbaf2dec05aaafc5ebd2f2e818d7fd5f445a3631b644d05e1ce22426861f9754b7e10e2177db4fb7dd3badebf70050bd1a9e81a6a674897
+  data.tar.gz: 13e403ca060db22f7427a33df32cddb4b255a15c27865a6080c714292bec45a307c39c632e403aa2483e5581ee5dc75fcda3be090d29438fe6531ef31ffaae68

data/lib/ticuna/llm.rb

@@ -2,6 +2,7 @@
 
 require "ticuna/providers/openai"
 require "ticuna/config"
+require "ticuna/response"
 
 module Ticuna
   class LLM
@@ -54,7 +55,8 @@ module Ticuna
                  else
                    [{ role: "user", content: message }]
                  end
-      @provider.ask_with_messages(messages, stream: stream, model: model, &block)
+
+      Ticuna::Response.new(@provider.ask_with_messages(messages, stream: stream, model: model, &block))
     end
 
     private
@@ -66,11 +68,11 @@ module Ticuna
 
       case valid.size
       when 0
-        raise "Provider not found. Define at least one ENV with token."
+        raise "Provider not found. Define at least one in config/initializers/ticuna.rb"
       when 1
-        valid.keys.firstraise
+        valid.keys.first
       else
-        raise "Multiple providers detected: #{valid.keys.join(", ")}. Define one in Ticuna::LLM.new(provider: :provider_name)."
+        raise "Multiple LLM APIs providers detected: #{valid.keys.join(", ")}. Define one in Ticuna::LLM.new(provider: :provider_name)."
       end
     end
   end

data/lib/ticuna/response.rb

@@ -0,0 +1,60 @@
+# frozen_string_literal: true
+
+module Ticuna
+  class Response
+    attr_reader :data
+
+    def initialize(data)
+      @data = deep_symbolize_keys(data)
+    end
+
+    def [](key)
+      @data[key.to_sym]
+    end
+
+    def to_h
+      @data
+    end
+
+    def inspect
+      "#<Ticuna::Response #{@data.inspect}>"
+    end
+
+    def method_missing(name, *args, &block)
+      value = @data[name]
+      return wrap(value) if @data.key?(name)
+
+      super
+    end
+
+    def respond_to_missing?(name, include_private = false)
+      @data.key?(name) || super
+    end
+
+    private
+
+    def deep_symbolize_keys(obj)
+      case obj
+      when Hash
+        obj.each_with_object({}) do |(k, v), h|
+          h[k.to_sym] = deep_symbolize_keys(v)
+        end
+      when Array
+        obj.map { |v| deep_symbolize_keys(v) }
+      else
+        obj
+      end
+    end
+
+    def wrap(value)
+      case value
+      when Hash
+        self.class.new(value)
+      when Array
+        value.map { |v| wrap(v) }
+      else
+        value
+      end
+    end
+  end
+end

data/lib/ticuna/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Ticuna
-  VERSION = "0.1.1"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ticuna
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - Chirana
@@ -46,6 +46,7 @@ files:
 - lib/ticuna/config.rb
 - lib/ticuna/llm.rb
 - lib/ticuna/providers/openai.rb
+- lib/ticuna/response.rb
 - lib/ticuna/tool.rb
 - lib/ticuna/version.rb
 - sig/ticuna.rbs