OpenSwarm: el framework open-source de enjambre para agentes IA

"""
Bot de atencion al cliente con OpenSwarm.
Tres agentes: triaje, ventas y soporte tecnico.
Compatible con Ollama (local) y la API de OpenAI.
"""
from swarm import Swarm, Agent, Result
from openai import OpenAI


# ─── CONFIGURACION DEL CLIENTE ───────────────────────────────────────────────

# Para Ollama: sustituye por tu servidor local si es distinto
cliente = Swarm(
    client=OpenAI(
        base_url="http://localhost:11434/v1",
        api_key="ollama",
    )
)
MODELO = "llama3.1:8b"


# ─── AGENTES ESPECIALISTAS ───────────────────────────────────────────────────

agente_ventas = Agent(
    name="Especialista en ventas",
    instructions="""
Eres el especialista en ventas de TiendaDemo, una tienda online de electronica.
Tu objetivo es ayudar al cliente a encontrar el producto que mejor se adapta
a sus necesidades y presupuesto. Sé amable, proporciona especificaciones tecnicas
cuando se pidan y sugiere alternativas si el producto solicitado no esta disponible.
Si el cliente tiene un problema con un pedido ya realizado, transfierelo a soporte.
""",
    functions=[],  # Se rellenara tras definir transferir_a_soporte
)

agente_soporte = Agent(
    name="Tecnico de soporte",
    instructions="""
Eres el tecnico de soporte de TiendaDemo. Tu trabajo es resolver incidencias
con pedidos, devoluciones, productos defectuosos y problemas de envio.
Sé empatico con el cliente, documenta el problema con precision y proporciona
un numero de ticket de soporte al final de cada interaccion. Si el cliente
quiere hacer una compra nueva, transfierelo al equipo de ventas.
""",
    functions=[],  # Se rellenara tras definir transferir_a_ventas
)


# ─── FUNCIONES DE HANDOFF ─────────────────────────────────────────────────────

def transferir_a_ventas(context_variables: dict) -> Result:
    """
    Transfiere al cliente al equipo de ventas.
    Usa esta funcion cuando el usuario quiere comprar, consultar precios
    o conocer disponibilidad de productos.
    """
    nombre = context_variables.get("nombre_cliente", "el cliente")
    return Result(
        value=f"Transfiriendo a {nombre} al equipo de ventas. Un momento.",
        agent=agente_ventas,
        context_variables={**context_variables, "ultimo_departamento": "ventas"},
    )


def transferir_a_soporte(context_variables: dict) -> Result:
    """
    Transfiere al cliente al soporte tecnico.
    Usa esta funcion cuando el usuario tiene un problema con un pedido,
    quiere hacer una devolucion o reporta un producto defectuoso.
    """
    nombre = context_variables.get("nombre_cliente", "el cliente")
    return Result(
        value=f"Transfiriendo a {nombre} al soporte tecnico. Un momento.",
        agent=agente_soporte,
        context_variables={**context_variables, "ultimo_departamento": "soporte"},
    )


# Asignar las funciones de transferencia cruzada
agente_ventas.functions = [transferir_a_soporte]
agente_soporte.functions = [transferir_a_ventas]


# ─── AGENTE DE TRIAJE ─────────────────────────────────────────────────────────

def registrar_nombre(nombre: str, context_variables: dict) -> Result:
    """
    Registra el nombre del cliente para personalizar la conversacion.
    Usa esta funcion cuando el cliente se presente o diga su nombre.
    """
    return Result(
        value=f"Encantado, {nombre}. En que puedo ayudarte hoy?",
        context_variables={**context_variables, "nombre_cliente": nombre},
    )


agente_triaje = Agent(
    name="Recepcion",
    instructions="""
Eres la recepcion virtual de TiendaDemo. Bienvenida al cliente, identifica
su necesidad y dirigelo al departamento correcto.

Reglas de triaje:
- Si el cliente quiere comprar, consultar precios o disponibilidad → transferir_a_ventas
- Si hay un problema con pedido, devolucion o producto defectuoso → transferir_a_soporte
- Si el cliente se presenta con su nombre → registrar_nombre
- Si la intencion no esta clara → pide mas informacion antes de transferir

Siempre saluda al cliente al inicio si no lo has hecho aun.
""",
    functions=[registrar_nombre, transferir_a_ventas, transferir_a_soporte],
)


# ─── BUCLE DE CONVERSACION MULTI-VUELTA ──────────────────────────────────────

def main():
    historial = []
    agente_activo = agente_triaje
    contexto = {}

    print("=== Bot de atencion al cliente (OpenSwarm) ===")
    print("Escribe 'salir' para terminar la conversacion.\n")

    while True:
        entrada = input("Usuario: ").strip()
        if entrada.lower() in ("salir", "exit", "quit"):
            print("Hasta pronto.")
            break

        historial.append({"role": "user", "content": entrada})

        respuesta = cliente.run(
            agent=agente_activo,
            messages=historial,
            context_variables=contexto,
            model_override=MODELO,
        )

        # Actualizar estado para la siguiente vuelta
        agente_activo = respuesta.agent
        historial = respuesta.messages
        contexto = respuesta.context_variables

        # Mostrar el ultimo mensaje del agente
        ultimo_mensaje = next(
            (m for m in reversed(respuesta.messages) if m["role"] == "assistant"),
            None,
        )
        if ultimo_mensaje:
            print(f"\n{agente_activo.name}: {ultimo_mensaje['content']}\n")


if __name__ == "__main__":
    main()