braintrust 0.0.9 → 0.0.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: be2efc651c8e685179541cf2ade46f86e3ca66c408ed00707d2b9890a4d5fa72
-  data.tar.gz: ba9c9f993abf5ea64290c5510361297a6b7dca21ad06583634d7e4080d4f7531
+  metadata.gz: aa149154056d58474a362bf2fba5b27a73670208a278aae56abc193971616de3
+  data.tar.gz: 5aeb130d705670e0baab100fa5203782fb3feaa0c39d4b228b54ac77ce581458
 SHA512:
-  metadata.gz: a40a56eae61148496ee7c96775b9c6fef63106a5c943dbe1a43f66572cb245e5ec3fd5a2e98452926c8a326a0f06f9dd68bfb56a4b7f04081e65ec03ef3d7939
-  data.tar.gz: c1bed8505efca929e688538d293e4f329c8dd8c2d1152858bb6967fa0d9f54b88f4120fb3d1f7d591fc51cc0e05ec1577b8345524ddc45863c1eb2ff6537a334
+  metadata.gz: c1c23282dff68a81cf9fd7f7b8509ee73d7050c9ce00b2184b7b8801941c9099e3aac21fdbba144a5d1fcfe93aa78d3bed25f03bce84ca4fbebb0440f3f524ef
+  data.tar.gz: f64dda4cf4622f791daa91b3b1a35a83dbb2bb1ac5ad201d41038875a9a1e968d9efbf6f258dbb207ab598869b2b7edcbb66415adada3806ad7b56e7fd0b1ebe

data/lib/braintrust/api/functions.rb

@@ -10,6 +10,12 @@ module Braintrust
     # Functions API namespace
     # Provides methods for creating, invoking, and managing remote functions (prompts)
     class Functions
+      TYPE_LLM = "llm"
+      TYPE_PROMPT = "prompt"
+      TYPE_SCORER = "scorer"
+      TYPE_TASK = "task"
+      TYPE_TOOL = "tool"
+
       def initialize(api)
         @api = api
         @state = api.state
@@ -43,8 +49,12 @@ module Braintrust
       # @param prompt_data [Hash, nil] Prompt configuration (prompt, options, etc.)
       # @param name [String, nil] Optional display name (defaults to slug)
       # @param description [String, nil] Optional description
+      # @param function_type [String, nil] Function type ("llm", "scorer", "task", "tool", or nil)
+      # @param function_schema [Hash, nil] JSON schema for function parameters and return type
+      #   @option function_schema [Hash] :parameters JSON schema for input parameters
+      #   @option function_schema [Hash] :returns JSON schema for return value
       # @return [Hash] Function metadata
-      def create(project_name:, slug:, function_data:, prompt_data: nil, name: nil, description: nil)
+      def create(project_name:, slug:, function_data:, prompt_data: nil, name: nil, description: nil, function_type: nil, function_schema: nil)
         # Look up project ID
         projects_result = http_get("/v1/project", {"project_name" => project_name})
         project = projects_result["objects"]&.first
@@ -59,6 +69,8 @@ module Braintrust
         }
         payload[:prompt_data] = prompt_data if prompt_data
         payload[:description] = description if description
+        payload[:function_type] = function_type if function_type
+        payload[:function_schema] = function_schema if function_schema
 
         http_post_json("/v1/function", payload)
       end
@@ -81,8 +93,110 @@ module Braintrust
         http_delete("/v1/function/#{id}")
       end
 
+      # Create a tool function
+      # Tools are functions that LLMs can call during execution.
+      # @param project_name [String] Project name
+      # @param slug [String] Function slug (URL-friendly identifier)
+      # @param prompt_data [Hash] Prompt configuration (prompt, options, etc.)
+      # @param name [String, nil] Optional display name (defaults to slug)
+      # @param description [String, nil] Description of what the tool does (recommended for LLM understanding)
+      # @param function_schema [Hash, nil] JSON schema defining the tool's parameters and return type
+      # @return [Hash] Function metadata
+      def create_tool(project_name:, slug:, prompt_data:, name: nil, description: nil, function_schema: nil)
+        validate_prompt_data!(prompt_data)
+        create(
+          project_name: project_name,
+          slug: slug,
+          function_data: {type: TYPE_PROMPT},
+          prompt_data: prompt_data,
+          name: name,
+          description: description,
+          function_type: TYPE_TOOL,
+          function_schema: function_schema
+        )
+      end
+
+      # Create a scorer function
+      # Scorers evaluate task outputs and return scores (typically 0-1).
+      # @param project_name [String] Project name
+      # @param slug [String] Function slug (URL-friendly identifier)
+      # @param prompt_data [Hash] Prompt configuration for the scoring logic
+      # @param name [String, nil] Optional display name (defaults to slug)
+      # @param description [String, nil] Optional description
+      # @param function_schema [Hash, nil] JSON schema for parameters and return type
+      # @return [Hash] Function metadata
+      def create_scorer(project_name:, slug:, prompt_data:, name: nil, description: nil, function_schema: nil)
+        validate_prompt_data!(prompt_data)
+        create(
+          project_name: project_name,
+          slug: slug,
+          function_data: {type: TYPE_PROMPT},
+          prompt_data: prompt_data,
+          name: name,
+          description: description,
+          function_type: TYPE_SCORER,
+          function_schema: function_schema
+        )
+      end
+
+      # Create a task function
+      # Tasks are general-purpose prompt functions.
+      # @param project_name [String] Project name
+      # @param slug [String] Function slug (URL-friendly identifier)
+      # @param prompt_data [Hash] Prompt configuration (prompt, options, etc.)
+      # @param name [String, nil] Optional display name (defaults to slug)
+      # @param description [String, nil] Optional description
+      # @param function_schema [Hash, nil] JSON schema for parameters and return type
+      # @return [Hash] Function metadata
+      def create_task(project_name:, slug:, prompt_data:, name: nil, description: nil, function_schema: nil)
+        validate_prompt_data!(prompt_data)
+        create(
+          project_name: project_name,
+          slug: slug,
+          function_data: {type: TYPE_PROMPT},
+          prompt_data: prompt_data,
+          name: name,
+          description: description,
+          function_type: TYPE_TASK,
+          function_schema: function_schema
+        )
+      end
+
+      # Create an LLM function
+      # LLM functions are prompt-based functions categorized as LLM type.
+      # @param project_name [String] Project name
+      # @param slug [String] Function slug (URL-friendly identifier)
+      # @param prompt_data [Hash] Prompt configuration (prompt, options, etc.)
+      # @param name [String, nil] Optional display name (defaults to slug)
+      # @param description [String, nil] Optional description
+      # @param function_schema [Hash, nil] JSON schema for parameters and return type
+      # @return [Hash] Function metadata
+      def create_llm(project_name:, slug:, prompt_data:, name: nil, description: nil, function_schema: nil)
+        validate_prompt_data!(prompt_data)
+        create(
+          project_name: project_name,
+          slug: slug,
+          function_data: {type: TYPE_PROMPT},
+          prompt_data: prompt_data,
+          name: name,
+          description: description,
+          function_type: TYPE_LLM,
+          function_schema: function_schema
+        )
+      end
+
       private
 
+      # Validate prompt_data structure
+      # @param prompt_data [Hash] The prompt data to validate
+      # @raise [ArgumentError] If prompt_data is invalid
+      def validate_prompt_data!(prompt_data)
+        raise ArgumentError, "prompt_data must be a Hash" unless prompt_data.is_a?(Hash)
+
+        has_prompt = prompt_data.key?(:prompt) || prompt_data.key?(TYPE_PROMPT)
+        raise ArgumentError, "prompt_data must contain a :prompt key" unless has_prompt
+      end
+
       # Core HTTP request method with logging
       # @param method [Symbol] :get, :post, or :delete
       # @param path [String] API path

data/lib/braintrust/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Braintrust
-  VERSION = "0.0.9"
+  VERSION = "0.0.10"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: braintrust
 version: !ruby/object:Gem::Version
-  version: 0.0.9
+  version: 0.0.10
 platform: ruby
 authors:
 - Braintrust