工具箱

import ktoolkits

ktoolkits.api_key="XXXX"

#当工具有多个参数时，可以使用参数字典为输入tool_input={"scan_target":"baidu.com"}
output = ktoolkits.Runner.call(
	tool_name="nmap",
	tool_input="www.baidu.com",
)

import autogen
import ktoolkits

user_proxy = autogen.UserProxyAgent(

	name="用户代理",
	
	#human_input_mode="NEVER",
	
	is_termination_msg=lambda x: x.get("content", "").rstrip().endswith("TERMINATE"),
	
	code_execution_config={
	
	"last_n_messages": 10,
	
	"work_dir": "code",
	
	"use_docker": False,

	}

)


tasks=[

"""请对www.baidu.com进行端口扫描""",

]

assistant = autogen.AssistantAgent(

name="渗透测试专家",

llm_config=llm_config

)

tool = ktool.FunctionCall.create_tool(tool_name="nmap")

register_function(

	f=tool.get("tool_func"),
	
	caller=assistant,
	
	executor=user_proxy,
	
	description=tool.get("tool_desc"),

)

user_proxy.initiate_chat(assistant,message=tasks[0])