ticuna 0.1.2 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e437b292297ce9fb31d37e3ad84e7d314f673bd0dab8a511ae8df35b737f17cf
-  data.tar.gz: 19fd66d1964c4655fc8e7cab7d8a461de5a90fc4e4dd31a0a28bc81c5864f4d9
+  metadata.gz: 91f1d0f3d86d36f73ee9127c48209b950a0a9e03d234ec0cf397015794797dda
+  data.tar.gz: 3db96fa9f07cc35f3baf5fe3af99da54d8e41e7ed954d3add56ee488f4a15de1
 SHA512:
-  metadata.gz: 0ef912de56ab1e96259632020d089808548e9734663a0950eebbb918815c74429a4003b078db2183407bf0be1cdcf9dbfbadd332384c0f3958dbe7af0fe4c92c
-  data.tar.gz: 8bbac24142de3cb268ac0b5968d2cbabca5adb4dd0357478758a7e3002f4628426a4dc10cda3af21889f4dafcc77a158c97ce31143ea9c55df046f9f98e24cb9
+  metadata.gz: e33ed0a919eda4ba474d3fad6993640b01ddf8b09c754cf5a8492673fbd914a4db3cd387db618ff85dc3a04afad70bc9e8d8de4925eb4ac47d43c78d44b392e7
+  data.tar.gz: aa329d1ad68be5fa446acbafd41eb3d6b99caf6a3d81f990950243439fbe5e359778ae375ff6f10d3133f9f919983c32d612f3183cdcb4cd34470663582b3b95

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,75 @@
+# frozen_string_literal: true
+
+require "json"
+
+module Ticuna
+  class Response
+    attr_reader :data
+
+    def initialize(raw)
+      parsed =
+        case raw
+        when String
+          begin
+            JSON.parse(raw)
+          rescue JSON::ParserError
+            { content: raw }
+          end
+        when Hash
+          raw
+        else
+          raise ArgumentError, "Ticuna::Response only accepts Hash or String, received: #{raw.class}"
+        end
+
+      @data = deep_symbolize_keys(parsed)
+    end
+
+    def [](key)
+      @data[key.to_sym]
+    end
+
+    def to_h
+      @data
+    end
+
+    def to_s
+      @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.1"
 end

metadata

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