Class: SmartAgent::AgentContext

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

Instance Method Summary collapse

Constructor Details

#initialize(agent) ⇒ AgentContext

Returns a new instance of AgentContext.



68
69
70
 
# File 'lib/smart_agent/agent.rb', line 68

def initialize(agent)
  @agent = agent
end

Instance Method Details

#call_tool(name, params = {}) ⇒ Object 



177
178
179
180
181
182
183
184
 
# File 'lib/smart_agent/agent.rb', line 177

def call_tool(name, params = {})
  if Tool.find_tool(name)
    return Tool.find_tool(name).call(params, @agent)
  end
  if server_name = MCPClient.find_server_by_tool_name(name)
    return MCPClient.new(server_name).call(name, params, @agent)
  end
end

#call_tools(result) ⇒ Object 



145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
 
# File 'lib/smart_agent/agent.rb', line 145

def call_tools(result)
  @agent.processor(:tool).call({ :status => :start }) if @agent.processor(:tool)
  SmartAgent.logger.info("call tools: " + result.to_s)
  results = []
  result.call_tools.each do |tool|
    tool_call_id = tool["id"]
    tool_name = tool["function"]["name"].to_sym
    params = safe_parse(tool["function"]["arguments"])
    if Tool.find_tool(tool_name)
      tool_result = Tool.find_tool(tool_name).call(params, @agent)
      if tool_result
        @agent.processor(:tool).call({ :content => tool_result })
        SmartAgent.prompt_engine.history_messages << { "role" => "assistant", "content" => "", "tool_calls" => [tool] } #result.response.dig("choices", 0, "message")
        SmartAgent.prompt_engine.history_messages << { "role" => "tool", "tool_call_id" => tool_call_id, "content" => tool_result.to_s.force_encoding("UTF-8") }
        results << tool_result
      end
    end
    if server_name = MCPClient.find_server_by_tool_name(tool_name)
      tool_result = MCPClient.new(server_name).call(tool_name, params, @agent)
      if tool_result
        @agent.processor(:tool).call({ :content => tool_result })
        SmartAgent.prompt_engine.history_messages << { "role" => "assistant", "content" => "", "tool_calls" => [tool] } # result.response.dig("choices", 0, "message")
        SmartAgent.prompt_engine.history_messages << { "role" => "tool", "tool_call_id" => tool_call_id, "content" => tool_result.to_s }
        results << tool_result
      end
    end
    @agent.processor(:tool).call({ :content => " ... done\n" }) if @agent.processor(:tool)
  end
  @agent.processor(:tool).call({ :status => :end }) if @agent.processor(:tool)
  return results
end

#call_worker(name, params, with_tools: true, with_history: false) ⇒ Object 



72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
 
# File 'lib/smart_agent/agent.rb', line 72

def call_worker(name, params, with_tools: true, with_history: false)
  SmartAgent.logger.info("Call Worker name is: #{name}")
  SmartAgent.logger.info("Call Worker params is: #{params}")
  if with_tools
    simple_tools = []
    if @agent.tools
      simple_tools = @agent.tools.map { |tool_name| Tool.find_tool(tool_name).to_json }
    end
    if @agent.servers
      mcp_tools = @agent.servers.map { |mcp_name| MCPClient.new(mcp_name).to_json }
      mcp_tools.each do |tools|
        tools["tools"].each do |tool|
          simple_tools << tool
        end
      end
    end
    params[:tools] = simple_tools
  end
  params[:with_history] = with_history
  ret = nil
  if @agent.on_event
    SmartAgent.prompt_engine.call_worker_by_stream(name, params) do |chunk, _bytesize|
      if chunk.dig("choices", 0, "delta", "reasoning_content")
        @agent.processor(:reasoning).call(chunk) if @agent.processor(:reasoning)
      end
      if chunk.dig("choices", 0, "delta", "content")
        @agent.processor(:content).call(chunk) if @agent.processor(:content)
      end
    end
    result = SmartAgent.prompt_engine.stream_response
  else
    result = SmartAgent.prompt_engine.call_worker(name, params)
  end
  ret = Result.new(result)
  return ret
end

#params ⇒ Object 



186
187
188
 
# File 'lib/smart_agent/agent.rb', line 186

def params
  @params ||= {}
end

#safe_parse(input) ⇒ Object 



109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
 
# File 'lib/smart_agent/agent.rb', line 109

def safe_parse(input)
  # 保存原始输入用于调试
  original_input = input.dup

  # 步骤1: 清理输入
  cleaned = input.strip

  # 步骤2: 处理外层引号（如果存在）
  if cleaned.start_with?('"') && cleaned.end_with?('"')
    cleaned = cleaned[1...-1]
  end

  # 步骤3: 反转义双重转义字符
  # 关键：只处理需要反转义的字符，保持JSON合法性
  cleaned = cleaned
    .gsub(/\\"/, '"') # 反转义引号
    .gsub(/\\\\/, '\\')    # 反转义反斜杠
  # 不反转义\n, \t, \r等，因为它们是JSON合法的转义序列

  # 步骤4: 尝试解析
  begin
    return JSON.parse(cleaned)
  rescue JSON::ParserError => e
    # 如果清理后失败，尝试原始输入
    begin
      return JSON.parse(original_input)
    rescue JSON::ParserError
      puts "Failed to parse JSON: #{e.message}"
      puts "Original: #{original_input}"
      puts "Cleaned: #{cleaned}"
      # 返回原始字符串以便后续处理
      return original_input
    end
  end
end