Packages
fnord
0.8.30
0.9.40
0.9.39
0.9.38
0.9.37
0.9.36
0.9.35
0.9.34
0.9.33
0.9.32
0.9.31
0.9.30
0.9.29
0.9.28
0.9.27
0.9.26
0.9.25
0.9.24
0.9.23
0.9.22
0.9.21
0.9.20
0.9.19
0.9.18
0.9.17
0.9.16
0.9.15
0.9.14
0.9.13
0.9.12
0.9.11
0.9.10
0.9.9
0.9.8
0.9.7
0.9.6
0.9.5
0.9.4
0.9.3
0.9.2
0.9.1
0.9.0
0.8.99
0.8.98
0.8.97
0.8.96
0.8.95
0.8.94
0.8.93
0.8.92
0.8.91
0.8.90
0.8.89
0.8.88
0.8.87
0.8.86
0.8.85
0.8.84
0.8.83
0.8.82
0.8.81
0.8.80
0.8.79
0.8.78
0.8.77
0.8.76
0.8.75
0.8.74
0.8.73
0.8.72
0.8.71
0.8.70
0.8.69
0.8.68
0.8.67
0.8.66
0.8.65
0.8.64
0.8.63
0.8.62
0.8.61
0.8.60
0.8.59
0.8.58
0.8.57
0.8.56
0.8.55
0.8.54
0.8.53
0.8.52
0.8.51
0.8.50
0.8.49
0.8.48
0.8.47
0.8.46
0.8.45
0.8.44
0.8.43
0.8.42
0.8.41
0.8.40
0.8.39
0.8.38
0.8.37
0.8.36
0.8.35
0.8.34
0.8.33
0.8.32
0.8.31
0.8.30
0.8.29
0.8.27
0.8.26
0.8.25
0.8.24
0.8.23
0.8.22
0.8.21
0.8.20
0.8.19
0.8.18
0.8.17
0.8.16
0.8.15
0.8.14
0.8.13
0.8.12
0.8.11
0.8.1
0.8.0
0.7.24
0.7.23
0.7.22
0.7.21
0.7.20
0.7.19
0.7.18
0.7.17
0.7.16
0.7.15
0.7.14
0.7.13
0.7.12
0.7.11
0.7.10
0.7.9
0.7.8
0.7.7
0.7.6
0.7.5
0.7.3
0.7.2
0.7.1
0.7.0
0.6.9
0.6.8
0.6.7
0.6.6
0.6.5
0.6.4
0.6.3
0.6.1
0.6.0
0.5.9
0.5.8
0.5.7
0.5.6
0.5.5
0.5.4
0.5.3
0.5.2
0.5.1
0.5.0
0.4.44
0.4.43
0.4.42
0.4.41
0.4.40
0.4.39
0.4.38
0.4.37
0.4.36
0.4.35
0.4.34
0.4.33
0.4.32
0.4.30
0.4.29
0.4.28
0.4.27
0.4.26
0.4.25
0.4.24
0.4.23
0.4.22
0.4.21
0.4.20
0.4.19
0.4.18
0.4.17
0.4.16
0.4.15
0.4.14
0.4.13
0.4.12
0.4.11
0.4.10
0.4.9
0.4.8
0.4.7
0.4.6
0.4.5
0.4.4
0.4.3
0.4.2
0.4.1
0.4.0
0.3.0
0.2.0
0.1.0
AI code archaeology
Current section
Files
Jump to
Current section
Files
lib/ai/agent/researcher.ex
defmodule AI.Agent.Researcher do
use Agent
@model AI.Model.balanced()
@research_tool_prompt """
If your research results in more questions, spin off research_tool tool calls
as necessary, rather than going down those rabbit holes yourself. That helps
to ensure that your own task stays focused and you don't fill your context
window with information that may not be pertinent to *your* task, which can
result in hallucinations. This will help you to provide the most effective
response.
"""
@prompt """
You are an AI agent who performs research on behalf of the Coordinating Agent.
They will provide you with a research task.
You are to proactively use your tools to gather information about the code base.
Many code bases are complex and contain legacy code, so you will need to make
clever use of your tools to gather the information you need. Take special
care to identify ambiguities in terminology, unexpected dependencies,
oddities of code structure, and other potential pitfalls, and identify them
to the Coordinating Agent. Cite any files that you find that are relevant to
the research task in your response.
You are responding to another LLM; your response should be concise to save
tokens, using abbreviations that another LLM will understand. Attempt to
communicate the necessary information as tersely as possible to preserve the
coordinating agent's context window.
"""
# ----------------------------------------------------------------------------
# AI.Agent Behaviour implementation
# ----------------------------------------------------------------------------
@behaviour AI.Agent
@impl AI.Agent
def get_response(opts) do
with {:ok, agent} <- Map.fetch(opts, :agent),
{:ok, prompt} <- Map.fetch(opts, :prompt) do
can_recurse? =
inc_depth()
|> case do
{:ok, _} -> true
{:error, :max_depth_reached} -> false
end
tools =
if can_recurse? do
AI.Tools.all_tools()
else
AI.Tools.all_tools()
|> Map.drop(["research_tool"])
end
system_prompt =
if can_recurse? do
"#{@prompt}\n\n#{@research_tool_prompt}"
else
@prompt
end
try do
AI.Agent.get_completion(agent,
model: @model,
toolbox: tools,
messages: [
AI.Util.system_msg(system_prompt),
AI.Util.user_msg(prompt)
]
)
|> case do
{:ok, %{response: response}} -> {:ok, response}
{:error, %{response: response}} -> {:error, response}
{:error, reason} -> {:error, reason}
end
after
dec_depth()
end
end
end
# ----------------------------------------------------------------------------
# Agent: controls recursion, where the researcher can call itself as a tool
# call to investigate multiple lines of inquiry.
# ----------------------------------------------------------------------------
@max_researchers 10
def start_link(), do: Agent.start_link(fn -> 0 end, name: __MODULE__)
def depth, do: Agent.get(__MODULE__, fn depth -> depth end)
def inc_depth do
Agent.get_and_update(__MODULE__, fn
depth when depth >= @max_researchers ->
{{:error, :max_depth_reached}, depth}
depth ->
{{:ok, depth + 1}, depth + 1}
end)
end
def dec_depth do
Agent.get_and_update(__MODULE__, fn
0 ->
{{:ok, 0}, 0}
depth ->
{{:ok, depth - 1}, depth - 1}
end)
end
end