32 lines
1.0 KiB
Elixir
32 lines
1.0 KiB
Elixir
defmodule ElixirAi.ChatRunner.ToolConfig do
|
|
alias ElixirAi.{AiProvider, AiTools, Conversation}
|
|
|
|
def handle_call({:set_tool_choice, tool_choice}, _from, state) do
|
|
Conversation.update_tool_choice(state.name, tool_choice)
|
|
{:reply, :ok, %{state | tool_choice: tool_choice}}
|
|
end
|
|
|
|
def handle_call({:set_allowed_tools, tool_names}, _from, state) do
|
|
Conversation.update_allowed_tools(state.name, tool_names)
|
|
server_tools = AiTools.build_server_tools(self(), tool_names)
|
|
liveview_tools = AiTools.build_liveview_tools(self(), tool_names)
|
|
|
|
{:reply, :ok,
|
|
%{
|
|
state
|
|
| allowed_tools: tool_names,
|
|
server_tools: server_tools,
|
|
liveview_tools: liveview_tools
|
|
}}
|
|
end
|
|
|
|
def handle_call({:set_provider, provider_id}, _from, state) do
|
|
with :ok <- Conversation.update_provider(state.name, provider_id),
|
|
{:ok, provider} <- AiProvider.find_by_id(provider_id) do
|
|
{:reply, {:ok, provider}, %{state | provider: provider}}
|
|
else
|
|
error -> {:reply, error, state}
|
|
end
|
|
end
|
|
end
|