54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
import asyncio
|
|
from src.services.agent_builder import AgentBuilder
|
|
from src.services.mcp_builder import MCPBuilder
|
|
|
|
async def main():
|
|
# Configuração dos servidores MCP
|
|
mcp_config = {
|
|
"brave-search": {
|
|
"url": "http://localhost:8000",
|
|
"headers": {
|
|
"Authorization": "Bearer seu_token_aqui"
|
|
}
|
|
},
|
|
"google-calendar-mcp": {
|
|
"url": "http://localhost:8001",
|
|
"headers": {
|
|
"Authorization": "Bearer seu_token_aqui"
|
|
}
|
|
}
|
|
}
|
|
|
|
# Configuração do agente
|
|
agent_config = {
|
|
"model": "gemini-pro",
|
|
"name": "Agente de Pesquisa",
|
|
"description": "Agente especializado em pesquisas e agendamentos",
|
|
"instruction": "Use as ferramentas disponíveis para realizar pesquisas e agendamentos",
|
|
"tools": {
|
|
"custom_tool": {
|
|
"type": "search",
|
|
"config": {
|
|
"api_key": "sua_chave_aqui"
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
# Cria o builder
|
|
builder = AgentBuilder(None) # None pois não estamos usando banco de dados neste exemplo
|
|
builder.set_mcp_config(mcp_config)
|
|
|
|
# Constrói o agente
|
|
agent, exit_stack = await builder.build_agent(agent_config)
|
|
|
|
try:
|
|
# Usa o agente
|
|
response = await agent.run("Pesquise sobre inteligência artificial")
|
|
print(response)
|
|
finally:
|
|
# Limpa os recursos
|
|
await exit_stack.aclose()
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main()) |