ticuna 0.1.2 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e437b292297ce9fb31d37e3ad84e7d314f673bd0dab8a511ae8df35b737f17cf
-  data.tar.gz: 19fd66d1964c4655fc8e7cab7d8a461de5a90fc4e4dd31a0a28bc81c5864f4d9
+  metadata.gz: 25afb92c6ab2ba62fe78323b2c25e8a6f87d08d0297139ea4433e52de1156426
+  data.tar.gz: 90d7976a68c9ae75c972bda54765e11d4dd9050c32d2f6433852b15a607f2b11
 SHA512:
-  metadata.gz: 0ef912de56ab1e96259632020d089808548e9734663a0950eebbb918815c74429a4003b078db2183407bf0be1cdcf9dbfbadd332384c0f3958dbe7af0fe4c92c
-  data.tar.gz: 8bbac24142de3cb268ac0b5968d2cbabca5adb4dd0357478758a7e3002f4628426a4dc10cda3af21889f4dafcc77a158c97ce31143ea9c55df046f9f98e24cb9
+  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

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.2"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ticuna
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  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