Añade soporte para el cliente LangSmith en `app.py` y mejora la gestión de trazas. Se implementa la función `_flush_langsmith` para asegurar que las trazas se envían antes de la salida del proceso o entre ejecuciones. Además, se registra esta función para su ejecución al finalizar el programa, optimizando el manejo de trazas y garantizando que no se pierdan datos importantes. Se actualiza la función `respond` para llamar a `_flush_langsmith` entre solicitudes, mejorando la fiabilidad del sistema.

app.py

@@ -1,10 +1,14 @@
 import os
+import atexit
+import asyncio
+import inspect
 import base64
 import mimetypes
 import gradio as gr
 from openai import OpenAI
 from dotenv import load_dotenv
 from langsmith import traceable
+from langsmith import Client as LangSmithClient
 from langsmith.run_trees import RunTree
 
 load_dotenv()
@@ -15,6 +19,35 @@ GEMINI_MODEL = "gemini-2.5-flash"
 _api_key = os.getenv("GEMINI_API_KEY")
 _client = OpenAI(api_key=_api_key, base_url=GEMINI_BASE_URL) if _api_key else None
 
+# Optional LangSmith client for guaranteed flush
+_ls_api_key_env = os.getenv("LANGSMITH_API_KEY")
+_ls_client = LangSmithClient() if _ls_api_key_env else None
+
+
+def _flush_langsmith():
+    """Ensure LangSmith traces are sent before process exit or between runs."""
+    if not _ls_client:
+        return
+    try:
+        result = _ls_client.flush()
+        if inspect.isawaitable(result):
+            try:
+                asyncio.run(result)
+            except RuntimeError:
+                # If an event loop is already running (e.g., in some servers), fallback
+                loop = asyncio.get_event_loop()
+                loop.create_task(result)
+    except Exception:
+        # Best-effort flush; do not break the app
+        pass
+
+
+if _ls_client:
+    try:
+        atexit.register(_flush_langsmith)
+    except Exception:
+        pass
+
 system_prompt = """
 Eres un asistente experto que guía a personas no técnicas para crear:
 - Credenciales de Gmail (Google Cloud) o
@@ -294,6 +327,8 @@ def respond(message, history: list[tuple[str, str]]):
                 pipeline.patch()
             except Exception:
                 pass
+        # Ensure traces are flushed between requests
+        _flush_langsmith()
     except Exception as e:
         if child_llm:
             try:
@@ -308,6 +343,7 @@ def respond(message, history: list[tuple[str, str]]):
             except Exception:
                 pass
         yield f"Ocurrió un error al llamar a Gemini: {e}"
+        _flush_langsmith()
 
 
 chat = gr.ChatInterface(