smart_prompt 0.4.1 → 0.4.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 729a2b8524be928407172c7e12f14dc6bebc97f9fdf4e1f3ff35f9e7791a5f11
-  data.tar.gz: a8fa95d01d5c5372bb669d67f5dca337db77c7e8444ddb166ac579318cf38e8d
+  metadata.gz: 396c6097973289a34143e86b65f428d55919d0e755916992a5c714e289ebf5a2
+  data.tar.gz: 5d2c2d81b486e1fb05b53f116047ab1f90be77c9536fd6440a96b573bdad00c3
 SHA512:
-  metadata.gz: ab384181eba6b17a53505110cc3500c35d5774bbe08d4ea0d7cc367281dcd22c1f45ed0eff367e22aeac382d6f8d29efb20e328f6e07e0b8c64f5e2c42368029
-  data.tar.gz: 62cbb97e5794f77fb467fdef9b5e6403450e5ed7332ca6f0044c967c16a193c70a334a2bd95897af79605ae6d51be8b2317e3011b70cc923dd859cd8b951fd57
+  metadata.gz: c6880395149678a195ea6efc46a81623d61c14c56534936041a625616a9e6b716597c078ccee0ac0d47c3b2a8e67d272742ee36940fa64321426799db0b26e4d
+  data.tar.gz: 63f0b8ae0f6f62363443731cae6fed3eafe746c463d259c9b88eee8563d6ef0cdcd84e684d6daeedd7329c3dcab84748ab62358fa2935b9d0278ec347df45ffb

data/lib/smart_prompt/anthropic_adapter.rb

@@ -239,16 +239,60 @@ module SmartPrompt
     end
 
     def extract_content(response)
-      response.fetch("content", []).map do |block|
+      text_parts = []
+      tool_calls = []
+
+      response.fetch("content", []).each do |block|
         case block["type"]
         when "text"
-          block["text"].to_s
+          text_parts << block["text"].to_s
         when "tool_use"
-          block.to_s
+          tool_calls << openai_tool_call(block)
         else
-          block.to_s
+          text_parts << block.to_s
         end
-      end.join
+      end
+
+      content = text_parts.join
+      return content if tool_calls.empty?
+
+      openai_response(response, content, tool_calls)
+    end
+
+    def openai_response(response, content, tool_calls)
+      {
+        "id" => response["id"],
+        "object" => "chat.completion",
+        "created" => Time.now.to_i,
+        "model" => response["model"],
+        "choices" => [{
+          "index" => 0,
+          "message" => {
+            "role" => "assistant",
+            "content" => content,
+            "tool_calls" => tool_calls,
+          },
+          "finish_reason" => openai_finish_reason(response["stop_reason"]),
+        }],
+        "usage" => response["usage"],
+      }
+    end
+
+    def openai_tool_call(block)
+      {
+        "id" => block["id"],
+        "type" => "function",
+        "function" => {
+          "name" => block["name"],
+          "arguments" => JSON.generate(block["input"] || {}),
+        },
+      }
+    end
+
+    def openai_finish_reason(stop_reason)
+      return "tool_calls" if stop_reason == "tool_use"
+
+      stop_reason
     end
   end
 end

data/lib/smart_prompt/version.rb

@@ -1,3 +1,3 @@
 module SmartPrompt
-  VERSION = "0.4.1"
+  VERSION = "0.4.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: smart_prompt
 version: !ruby/object:Gem::Version
-  version: 0.4.1
+  version: 0.4.2
 platform: ruby
 authors:
 - zhuang biaowei