Class: TurnKit::Agent

Inherits:
Object
  • Object
show all
Defined in:
lib/turnkit/agent.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(name:, description: "", model: nil, instructions: "", tools: [], skills: [], available_skills: [], sub_agents: [], system_prompt: nil, prompt_sections: nil, prompt_mode: nil, client: nil, store: nil, max_iterations: nil, timeout: nil, cost_limit: nil, max_depth: nil, max_tool_executions: nil) ⇒ Agent

Returns a new instance of Agent.

Raises:

  • (ArgumentError) 


9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
 
# File 'lib/turnkit/agent.rb', line 9

def initialize(name:, description: "", model: nil, instructions: "", tools: [], skills: [], available_skills: [], sub_agents: [],
  system_prompt: nil, prompt_sections: nil, prompt_mode: nil, client: nil, store: nil,
  max_iterations: nil, timeout: nil, cost_limit: nil, max_depth: nil, max_tool_executions: nil)
  @name = name.to_s
  @description = description.to_s
  @model = model
  @instructions = instructions.to_s
  @tools = Array(tools)
  @skills = Array(skills)
  @available_skills = Array(available_skills)
  @sub_agents = Array(sub_agents)
  @system_prompt = system_prompt
  @prompt_sections = prompt_sections
  @prompt_mode = prompt_mode&.to_sym
  @client = client
  @store = store
  @max_iterations = max_iterations
  @timeout = timeout
  @cost_limit = cost_limit
  @max_depth = max_depth
  @max_tool_executions = max_tool_executions
  raise ArgumentError, "name is required" if @name.empty?
end

Instance Attribute Details

#available_skills ⇒ Object (readonly)

Returns the value of attribute available_skills.



5
6
7
 
# File 'lib/turnkit/agent.rb', line 5

def available_skills
  @available_skills
end

#client ⇒ Object (readonly)

Returns the value of attribute client.



6
7
8
 
# File 'lib/turnkit/agent.rb', line 6

def client
  @client
end

#cost_limit ⇒ Object (readonly)

Returns the value of attribute cost_limit.



6
7
8
 
# File 'lib/turnkit/agent.rb', line 6

def cost_limit
  @cost_limit
end

#description ⇒ Object (readonly)

Returns the value of attribute description.



5
6
7
 
# File 'lib/turnkit/agent.rb', line 5

def description
  @description
end

#instructions ⇒ Object (readonly)

Returns the value of attribute instructions.



5
6
7
 
# File 'lib/turnkit/agent.rb', line 5

def instructions
  @instructions
end

#max_depth ⇒ Object (readonly)

Returns the value of attribute max_depth.



6
7
8
 
# File 'lib/turnkit/agent.rb', line 6

def max_depth
  @max_depth
end

#max_iterations ⇒ Object (readonly)

Returns the value of attribute max_iterations.



6
7
8
 
# File 'lib/turnkit/agent.rb', line 6

def max_iterations
  @max_iterations
end

#max_tool_executions ⇒ Object (readonly)

Returns the value of attribute max_tool_executions.



6
7
8
 
# File 'lib/turnkit/agent.rb', line 6

def max_tool_executions
  @max_tool_executions
end

#model ⇒ Object (readonly)

Returns the value of attribute model.



5
6
7
 
# File 'lib/turnkit/agent.rb', line 5

def model
  @model
end

#name ⇒ Object (readonly)

Returns the value of attribute name.



5
6
7
 
# File 'lib/turnkit/agent.rb', line 5

def name
  @name
end

#prompt_mode ⇒ Object (readonly)

Returns the value of attribute prompt_mode.



7
8
9
 
# File 'lib/turnkit/agent.rb', line 7

def prompt_mode
  @prompt_mode
end

#prompt_sections ⇒ Object (readonly)

Returns the value of attribute prompt_sections.



7
8
9
 
# File 'lib/turnkit/agent.rb', line 7

def prompt_sections
  @prompt_sections
end

#skills ⇒ Object (readonly)

Returns the value of attribute skills.



5
6
7
 
# File 'lib/turnkit/agent.rb', line 5

def skills
  @skills
end

#store ⇒ Object (readonly)

Returns the value of attribute store.



6
7
8
 
# File 'lib/turnkit/agent.rb', line 6

def store
  @store
end

#sub_agents ⇒ Object (readonly)

Returns the value of attribute sub_agents.



5
6
7
 
# File 'lib/turnkit/agent.rb', line 5

def sub_agents
  @sub_agents
end

#system_prompt ⇒ Object (readonly)

Returns the value of attribute system_prompt.



7
8
9
 
# File 'lib/turnkit/agent.rb', line 7

def system_prompt
  @system_prompt
end

#timeout ⇒ Object (readonly)

Returns the value of attribute timeout.



6
7
8
 
# File 'lib/turnkit/agent.rb', line 6

def timeout
  @timeout
end

#tools ⇒ Object (readonly)

Returns the value of attribute tools.



5
6
7
 
# File 'lib/turnkit/agent.rb', line 5

def tools
  @tools
end

Instance Method Details

#build_budget(root_started_at: Clock.now) ⇒ Object 



87
88
89
90
91
92
93
94
95
96
 
# File 'lib/turnkit/agent.rb', line 87

def build_budget(root_started_at: Clock.now)
  Budget.new(
    max_iterations: max_iterations || TurnKit.max_iterations,
    timeout: timeout || TurnKit.timeout,
    max_depth: max_depth || TurnKit.max_depth,
    max_tool_executions: max_tool_executions || TurnKit.max_tool_executions,
    cost_limit: cost_limit || TurnKit.cost_limit,
    root_started_at: root_started_at
  )
end

#conversation(model: nil, subject: nil, metadata: {}) ⇒ Object 



33
34
35
36
37
38
39
40
41
42
 
# File 'lib/turnkit/agent.rb', line 33

def conversation(model: nil, subject: nil, metadata: {})
  store = effective_store
  record = store.create_conversation(
    "agent_name" => name,
    "model" => model || effective_model,
    "subject" => subject,
    "metadata" => metadata
  )
  Conversation.new(agent: self, record: record, store: store, model: model || effective_model, subject: subject, metadata: metadata)
end

#effective_available_skills ⇒ Object 



60
61
62
 
# File 'lib/turnkit/agent.rb', line 60

def effective_available_skills
  (Array(TurnKit.available_skills) + available_skills).uniq { |skill| skill.key }
end

#effective_client ⇒ Object 



48
49
50
 
# File 'lib/turnkit/agent.rb', line 48

def effective_client
  client || TurnKit.client
end

#effective_model ⇒ Object 



44
45
46
 
# File 'lib/turnkit/agent.rb', line 44

def effective_model
  model || TurnKit.default_model
end

#effective_prompt_mode(turn: nil) ⇒ Object 



68
69
70
71
72
 
# File 'lib/turnkit/agent.rb', line 68

def effective_prompt_mode(turn: nil)
  return prompt_mode if prompt_mode

  turn&.depth.to_i.positive? ? :minimal : :full
end

#effective_prompt_sections ⇒ Object 



64
65
66
 
# File 'lib/turnkit/agent.rb', line 64

def effective_prompt_sections
  prompt_sections || TurnKit.prompt_sections
end

#effective_store ⇒ Object 



52
53
54
 
# File 'lib/turnkit/agent.rb', line 52

def effective_store
  store || TurnKit.store
end

#effective_tools ⇒ Object 



56
57
58
 
# File 'lib/turnkit/agent.rb', line 56

def effective_tools
  tools + sub_agents.map { |agent| SubAgentTool.for(agent) }
end

#instructions_with_skills ⇒ Object 



98
99
100
101
102
 
# File 'lib/turnkit/agent.rb', line 98

def instructions_with_skills
  parts = [ instructions ]
  parts << SystemPrompt.loaded_skills_text(skills)
  parts.reject(&:empty?).join("\n\n")
end

#system_prompt_for(turn:, conversation:) ⇒ Object 



74
75
76
77
78
79
80
81
82
83
84
85
 
# File 'lib/turnkit/agent.rb', line 74

def system_prompt_for(turn:, conversation:)
  prompt = SystemPrompt.new(agent: self, turn: turn, conversation: conversation, mode: effective_prompt_mode(turn: turn))

  case system_prompt
  when nil
    prompt.to_s
  when String
    system_prompt
  else
    system_prompt.call(prompt).to_s
  end
end