Planificación De Viajes

En este ejemplo, recorreremos el proceso de creación de un sistema sofisticado de planificación de viajes usando Saptiva-Agents. Nuestro planificador de viajes utilizará múltiples agentes de IA, cada uno con un rol específico, para crear colaborativamente un itinerario de viaje completo.

Primero, importamos los módulos necesarios:

from saptiva_agents import SAPTIVA_LEGACY
from saptiva_agents.agents import AssistantAgent
from saptiva_agents.conditions import TextMentionTermination
from saptiva_agents.teams import RoundRobinGroupChat
from saptiva_agents.ui import Console
from saptiva_agents.base import SaptivaAIChatCompletionClient

Definiendo Los Agentes

En la siguiente sección definiremos los agentes que se utilizarán en el equipo de planificación de viajes:

model_client = SaptivaAIChatCompletionClient(
    model=SAPTIVA_LEGACY,
    api_key="TU_SAPTIVA_API_KEY",
)

planner_agent = AssistantAgent(
    "planner_agent",
    model_client=model_client,
    description="Un asistente útil que puede planificar viajes.",
    system_message="Eres un asistente útil que puede sugerir un plan de viaje para un usuario según su solicitud.",
)

local_agent = AssistantAgent(
    "local_agent",
    model_client=model_client,
    description="Un asistente local que puede sugerir actividades o lugares para visitar.",
    system_message="Eres un asistente útil que puede sugerir actividades locales auténticas e interesantes o lugares para visitar, y puedes utilizar cualquier información contextual proporcionada.",
)

language_agent = AssistantAgent(
    "language_agent",
    model_client=model_client,
    description="Un asistente útil que puede proporcionar consejos lingüísticos para un destino dado.",
    system_message="Eres un asistente útil que puede revisar planes de viaje y proporcionar comentarios sobre consejos importantes/críticos sobre cómo abordar los desafíos lingüísticos o de comunicación para el destino dado. Si el plan ya incluye consejos de idioma, puedes mencionar que el plan es satisfactorio, con su justificación.",
)

travel_summary_agent = AssistantAgent(
    "travel_summary_agent",
    model_client=model_client,
    description="Un asistente útil que puede resumir el plan de viaje.",
    system_message="Eres un asistente útil que puede tomar todas las sugerencias y consejos de los otros agentes y proporcionar un plan de viaje final detallado. Debes asegurarte de que el plan final esté integrado y completo. TU RESPUESTA FINAL DEBE SER EL PLAN COMPLETO. Cuando el plan esté completo y todas las perspectivas estén integradas, puedes responder con TERMINATE.",
)

Condición de terminación y ejecución del equipo

Esto genera una conversación colaborativa entre los agentes para desarrollar un itinerario de 3 días a Nepal, con recomendaciones culturales, actividades locales, consejos de idioma y un resumen completo del viaje.

Última actualización