Class: LLM::Function::Ractor::Task

Inherits:
Task
  • Object
show all
Defined in:
lib/llm/function/ractor/task.rb

Overview

The Task class wraps a ractor-backed function call and delegates mailbox coordination to Mailbox.

Instance Attribute Summary collapse

Attributes inherited from Task

#function

Instance Method Summary collapse

Constructor Details

#initialize(fn, options = {}) ⇒ LLM::Function::Ractor::Task

Parameters:

Options Hash (options):

  • :tracer (LLM::Tracer, nil)
  • :runner_class (Class)
  • :id (String, nil)
  • :name (String)
  • :arguments (Hash, Array, nil)
  • :model (String, nil)


23
24
25
26
27
28
29
30
31
# File 'lib/llm/function/ractor/task.rb', line 23

def initialize(fn, options = {})
  super
  @runner_class = options.fetch(:runner_class)
  @id = options.fetch(:id)
  @name = options.fetch(:name)
  @arguments = options.fetch(:arguments)
  @model = options.fetch(:model, nil)
  @tracer = options.fetch(:tracer, nil)
end

Instance Attribute Details

#mailboxLLM::Function::Ractor::Mailbox (readonly)



11
12
13
# File 'lib/llm/function/ractor/task.rb', line 11

def mailbox
  @mailbox
end

Instance Method Details

#alive?Boolean

Returns:

  • (Boolean)


46
47
48
# File 'lib/llm/function/ractor/task.rb', line 46

def alive?
  @mailbox&.alive? || false
end

#group_classClass

Returns:

  • (Class)


71
72
73
# File 'lib/llm/function/ractor/task.rb', line 71

def group_class
  LLM::Function::Ractor::Group
end

#interrupt!nil Also known as: cancel!

Returns:

  • (nil)


52
53
54
55
# File 'lib/llm/function/ractor/task.rb', line 52

def interrupt!
  mailbox&.interrupt!
  nil
end

#spawnLLM::Function::Ractor::Task



35
36
37
38
39
40
41
42
# File 'lib/llm/function/ractor/task.rb', line 35

def spawn
  @span = @tracer&.on_tool_start(
    id: @id, name: @name,
    arguments: @arguments, model: @model
  )
  @mailbox = Ractor::Mailbox.new(build_task)
  self
end

#waitLLM::Function::Return Also known as: value



60
61
62
63
64
65
66
# File 'lib/llm/function/ractor/task.rb', line 60

def wait
  spawn unless @mailbox
  id, name, value = mailbox.wait
  result = Return.new(id, name, value)
  @tracer&.on_tool_finish(result:, span: @span)
  result
end