Usa un agente de LangGraph

Además de las instrucciones generales para usar un agente, en esta página, se describen las funciones específicas de LanggraphAgent.

Antes de comenzar

En este instructivo, se supone que leíste y seguiste las instrucciones que se indican en los siguientes documentos:

• Desarrolla un agente de LangGraph: Para desarrollar agent como una instancia de LanggraphAgent.
• Autenticación del usuario para autenticarse como usuario y consultar al agente
• Importa e inicializa el SDK para inicializar el cliente y obtener una instancia implementada (si es necesario).

Operaciones admitidas

Se admiten las siguientes operaciones para LanggraphAgent:

• query: Para obtener una respuesta a una búsqueda de forma síncrona
• stream_query: Para transmitir una respuesta a una búsqueda.
• get_state: Para obtener un punto de control específico.
• get_state_history: Para enumerar los puntos de control de un subproceso.
• update_state: Para crear ramas correspondientes a diferentes situaciones

Cómo transmitir una respuesta a una búsqueda

LangGraph admite varios modos de transmisión. Las principales son las siguientes:

• values: Este modo transmite el estado completo del gráfico después de que se llama a cada nodo.
• updates: En este modo, se transmiten actualizaciones al estado del gráfico después de que se llama a cada nodo.

Para transmitir values (que corresponde al estado completo del gráfico):

for state_values in agent.stream_query(
    input=inputs,
    stream_mode="values",
    config={"configurable": {"thread_id": "streaming-thread-values"}},
):
    print(state_values)

Para transmitir updates (que corresponde a las actualizaciones del estado del gráfico):

for state_updates in agent.stream_query(
    input=inputs,
    stream_mode="updates",
    config={"configurable": {"thread_id": "streaming-thread-updates"}},
):
    print(state_updates)

Con interacción humana

En LangGraph, un aspecto común de la intervención humana es agregar puntos de interrupción para interrumpir la secuencia de acciones del agente y hacer que un humano reanude el flujo en un momento posterior.

Nota: En este paso, se supone que usas un agente configurado para almacenar puntos de control.

Revisar

Puedes establecer puntos de interrupción con los argumentos interrupt_before= o interrupt_after= cuando llames a .query o .stream_query:

response = agent.query(
    input=inputs,
    interrupt_before=["tools"], # after generating the function call, before invoking the function
    interrupt_after=["tools"], # after getting a function response, before moving on
    config={"configurable": {"thread_id": "human-in-the-loop-deepdive"}},
)

langchain_load(response['messages'][-1]).pretty_print()

El resultado se verá similar al siguiente código:

================================== Ai Message ==================================
Tool Calls:
  get_exchange_rate (12610c50-4465-4296-b1f3-d751ec959fd5)
 Call ID: 12610c50-4465-4296-b1f3-d751ec959fd5
  Args:
    currency_from: USD
    currency_to: SEK

Aprobación

Para aprobar la llamada a la herramienta generada y reanudar el resto de la ejecución, pasa None a la entrada y especifica el subproceso o el punto de control dentro de config:

response = agent.query(
    input=None,  # Continue with the function call
    interrupt_before=["tools"], # after generating the function call, before invoking the function
    interrupt_after=["tools"], # after getting a function response, before moving on
    config={"configurable": {"thread_id": "human-in-the-loop-deepdive"}},
)

langchain_load(response['messages'][-1]).pretty_print()

El resultado se verá similar al siguiente código:

================================= Tool Message =================================
Name: get_exchange_rate

{"amount": 1.0, "base": "USD", "date": "2024-11-14", "rates": {"SEK": 11.0159}}

Historial

Para enumerar todos los puntos de control de un subproceso determinado, usa el método .get_state_history:

for state_snapshot in agent.get_state_history(
    config={"configurable": {"thread_id": "human-in-the-loop-deepdive"}},
):
    if state_snapshot["metadata"]["step"] >= 0:
        print(f'step {state_snapshot["metadata"]["step"]}: {state_snapshot["config"]}')
        state_snapshot["values"]["messages"][-1].pretty_print()
        print("\n")

La respuesta será similar a la siguiente secuencia de resultados:

step 3: {'configurable': {'thread_id': 'human-in-the-loop-deepdive', 'checkpoint_ns': '', 'checkpoint_id': '1efa2e95-ded5-67e0-8003-2d34e04507f5'}}
================================== Ai Message ==================================

The exchange rate from US dollars to Swedish krona is 1 USD to 11.0159 SEK.

step 2: {'configurable': {'thread_id': 'human-in-the-loop-deepdive', 'checkpoint_ns': '', 'checkpoint_id': '1efa2e95-d189-6a77-8002-5dbe79e2ce58'}}
================================= Tool Message =================================
Name: get_exchange_rate

{"amount": 1.0, "base": "USD", "date": "2024-11-14", "rates": {"SEK": 11.0159}}

step 1: {'configurable': {'thread_id': 'human-in-the-loop-deepdive', 'checkpoint_ns': '', 'checkpoint_id': '1efa2e95-cc7f-6d68-8001-1f6b5e57c456'}}
================================== Ai Message ==================================
Tool Calls:
  get_exchange_rate (12610c50-4465-4296-b1f3-d751ec959fd5)
 Call ID: 12610c50-4465-4296-b1f3-d751ec959fd5
  Args:
    currency_from: USD
    currency_to: SEK

step 0: {'configurable': {'thread_id': 'human-in-the-loop-deepdive', 'checkpoint_ns': '', 'checkpoint_id': '1efa2e95-c2e4-6f3c-8000-477fd654cb53'}}
================================ Human Message =================================

What is the exchange rate from US dollars to Swedish currency?

Obtén la configuración de un paso

Para obtener un punto de control anterior, especifica checkpoint_id (y checkpoint_ns). Primero, vuelve al paso 1, cuando se generó la llamada a la herramienta:

snapshot_config = {}
for state_snapshot in agent.get_state_history(
    config={"configurable": {"thread_id": "human-in-the-loop-deepdive"}},
):
    if state_snapshot["metadata"]["step"] == 1:
        snapshot_config = state_snapshot["config"]
        break

print(snapshot_config)

El resultado se verá similar al siguiente código:

{'configurable': {'thread_id': 'human-in-the-loop-deepdive',
  'checkpoint_ns': '',
  'checkpoint_id': '1efa2e95-cc7f-6d68-8001-1f6b5e57c456'}}

Viaje en el tiempo

Para obtener un punto de control, se puede usar el método .get_state:

# By default, it gets the latest state [unless (checkpoint_ns, checkpoint_id) is specified]
state = agent.get_state(config={"configurable": {
    "thread_id": "human-in-the-loop-deepdive",
}})

print(f'step {state["metadata"]["step"]}: {state["config"]}')
state["values"]["messages"][-1].pretty_print()

De forma predeterminada, obtiene el punto de control más reciente (por marca de tiempo). El resultado se verá similar al siguiente:

step 3: {'configurable': {'thread_id': 'human-in-the-loop-deepdive', 'checkpoint_ns': '', 'checkpoint_id': '1efa2e95-ded5-67e0-8003-2d34e04507f5'}}
================================== Ai Message ==================================

The exchange rate from US dollars to Swedish krona is 1 USD to 11.0159 SEK.

Obtén el punto de control de una configuración

Para una configuración determinada (p.ej., snapshot_config de la configuración de un paso), puedes obtener el punto de control correspondiente:

state = agent.get_state(config=snapshot_config)
print(f'step {state["metadata"]["step"]}: {state["config"]}')
state["values"]["messages"][-1].pretty_print()

El resultado se verá similar al siguiente código:

step 1: {'configurable': {'thread_id': 'human-in-the-loop-deepdive', 'checkpoint_ns': '', 'checkpoint_id': '1efa2e95-cc7f-6d68-8001-1f6b5e57c456'}}
================================== Ai Message ==================================
Tool Calls:
  get_exchange_rate (12610c50-4465-4296-b1f3-d751ec959fd5)
 Call ID: 12610c50-4465-4296-b1f3-d751ec959fd5
  Args:
    currency_from: USD
    currency_to: SEK

Volver a reproducir

Para reproducir desde un estado determinado, pasa la configuración del estado (es decir, state["config"]) al agente. La configuración del estado es un diccionario que se ve de la siguiente manera:

{'configurable': {'thread_id': 'human-in-the-loop-deepdive',
  'checkpoint_ns': '',
  'checkpoint_id': '1efa2e95-cc7f-6d68-8001-1f6b5e57c456'}}

Para reproducir desde state["config"] (donde se generó una llamada a herramienta), especifica None en la entrada:

for state_values in agent.stream_query(
    input=None, # resume
    stream_mode="values",
    config=state["config"],
):
    langchain_load(state_values["messages"][-1]).pretty_print()

El resultado será similar a la siguiente secuencia de resultados:

================================== Ai Message ==================================
Tool Calls:
  get_exchange_rate (12610c50-4465-4296-b1f3-d751ec959fd5)
 Call ID: 12610c50-4465-4296-b1f3-d751ec959fd5
  Args:
    currency_from: USD
    currency_to: SEK

================================= Tool Message =================================
Name: get_exchange_rate

{"amount": 1.0, "base": "USD", "date": "2024-11-14", "rates": {"SEK": 11.0159}}

================================== Ai Message ==================================

The exchange rate from US dollars to Swedish krona is 1 USD to 11.0159 SEK.

Ramificación

Puedes crear bifurcaciones a partir de puntos de control anteriores para probar situaciones alternativas con el método .update_state:

branch_config = agent.update_state(
    config=state["config"],
    values={"messages": [last_message]}, # the update we want to make
)

print(branch_config)

El resultado se verá similar al siguiente código:

{'configurable': {'thread_id': 'human-in-the-loop-deepdive',
  'checkpoint_ns': '',
  'checkpoint_id': '1efa2e96-0560-62ce-8002-d1bb48a337bc'}}

Nota: branch_config tiene un checkpoint_id diferente de state["config"] debido a .update_state

Podemos consultar al agente con branch_config para reanudar el proceso desde el punto de control con el estado actualizado:

for state_values in agent.stream_query(
    input=None, # resume
    stream_mode="values",
    config=branch_config,
):
    langchain_load(state_values["messages"][-1]).pretty_print()

El resultado será similar a la siguiente secuencia de resultados:

================================== Ai Message ==================================
Tool Calls:
  get_exchange_rate (12610c50-4465-4296-b1f3-d751ec959fd5)
 Call ID: 12610c50-4465-4296-b1f3-d751ec959fd5
  Args:
    currency_date: 2024-09-01
    currency_from: USD
    currency_to: SEK

================================= Tool Message =================================
Name: get_exchange_rate

{"amount": 1.0, "base": "USD", "date": "2024-08-30", "rates": {"SEK": 10.2241}}

================================== Ai Message ==================================

The exchange rate from US dollars to Swedish krona on 2024-08-30 was 1 USD to 10.2241 SEK.

¿Qué sigue?

• Usa un agente.
• Evalúa un agente.
• Administra los agentes implementados.
• Obtén asistencia.