langchain-trigger-server 0.2.8__py3-none-any.whl → 0.2.10__py3-none-any.whl

{langchain_trigger_server-0.2.8.dist-info → langchain_trigger_server-0.2.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-trigger-server
-Version: 0.2.8
+Version: 0.2.10
 Summary: Generic event-driven triggers framework
 Project-URL: Homepage, https://github.com/langchain-ai/open-agent-platform
 Project-URL: Repository, https://github.com/langchain-ai/open-agent-platform

{langchain_trigger_server-0.2.8.dist-info → langchain_trigger_server-0.2.10.dist-info}/RECORD

@@ -1,7 +1,7 @@
 langchain_triggers/__init__.py,sha256=WoW9LC_FJRs42mLWq2iuM-jjPow2Rue50q2zm56Oul0,536
-langchain_triggers/app.py,sha256=_PWD8qB4wE98OCkIGeC4mx2D1TUrFW50Rm2z1kMXhTU,34532
+langchain_triggers/app.py,sha256=rVl_Yc2n0bz37mPb3E1MnPG2bCXH-bZkuuleaQIII8U,38569
 langchain_triggers/core.py,sha256=_CNZRyj78yCHG8FACwhCmZ0zvRoWls924OIFYMOC27Q,3772
-langchain_triggers/cron_manager.py,sha256=RYtaco1mMYRI0BnB-u9NKoI55mTinn485h8HwWNoxJU,10268
+langchain_triggers/cron_manager.py,sha256=ISo-P2gw7eQ6y7xWQOfojqcJr_K-zagZt9Ocy8nX0fw,10477
 langchain_triggers/decorators.py,sha256=zsfgf171qkEDdIiSn8LUr--3dz6bxBBKZO6dpRM2ILs,3711
 langchain_triggers/auth/__init__.py,sha256=RtDKuBoKYuyHzLNpKr74cmALO0PhHlWO9Ho7k3CUYFE,349
 langchain_triggers/auth/slack_hmac.py,sha256=kiwjhTXITgQvLAtEcOv8BnnWJRJcxaQ9dXkQm3JJDQ4,2948
@@ -10,6 +10,6 @@ langchain_triggers/database/interface.py,sha256=jpADOOwcBQo1ZichgiZVaOvfZqEqVVo8
 langchain_triggers/database/supabase.py,sha256=zi_75GbqRvzzlXd5EgfYof4h6vHWOLS4I1759wvY9kQ,17009
 langchain_triggers/triggers/__init__.py,sha256=Uw1544gxzN4XDRn2RzpZ5EAG6EAF38ZYQtVvlciEsMs,146
 langchain_triggers/triggers/cron_trigger.py,sha256=SeWz_ETBCBO1_r96tzTZgsvn4BdF4yMKabygjmHoGwY,3174
-langchain_trigger_server-0.2.8.dist-info/METADATA,sha256=_RdZIPeGJI6fGKsJUXtaAekSj4gXd9SY-jGDp9l2qYk,1486
-langchain_trigger_server-0.2.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langchain_trigger_server-0.2.8.dist-info/RECORD,,
+langchain_trigger_server-0.2.10.dist-info/METADATA,sha256=EcKCDT6DqADexKESf7YhP8mMkZImpBDRaGHnvl4XAdU,1487
+langchain_trigger_server-0.2.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langchain_trigger_server-0.2.10.dist-info/RECORD,,

langchain_triggers/app.py

@@ -5,8 +5,10 @@ from __future__ import annotations
 import logging
 import os
 from collections.abc import Callable
+from datetime import UTC, datetime, timedelta
 from typing import Any
 
+import jwt
 from fastapi import Depends, FastAPI, HTTPException, Request
 from langchain_auth.client import Client
 from langgraph_sdk import get_client
@@ -27,6 +29,28 @@ from .triggers.cron_trigger import CRON_TRIGGER_ID
 logger = logging.getLogger(__name__)
 
 
+def get_x_service_jwt_token(
+    payload: dict[str, Any] | None = None, expiration_seconds: int = 60 * 60
+) -> str:
+    exp_datetime = datetime.now(tz=UTC) + timedelta(seconds=expiration_seconds)
+    exp = int(exp_datetime.timestamp())
+
+    payload = payload or {}
+    payload = {
+        "sub": "unspecified",
+        "exp": exp,
+        **payload,
+    }
+
+    secret = os.environ["X_SERVICE_AUTH_JWT_SECRET"]
+
+    return jwt.encode(
+        payload,
+        secret,
+        algorithm="HS256",
+    )
+
+
 class AuthenticationMiddleware(BaseHTTPMiddleware):
     """Middleware to handle authentication for API endpoints."""
 
@@ -106,7 +130,6 @@ class TriggerServer:
 
         # LangGraph configuration
         self.langgraph_api_url = os.getenv("LANGGRAPH_API_URL")
-        self.langsmith_api_key = os.getenv("LANGCHAIN_API_KEY")
         self.trigger_server_auth_api_url = os.getenv("TRIGGER_SERVER_HOST_API_URL")
 
         if not self.langgraph_api_url:
@@ -115,16 +138,10 @@ class TriggerServer:
         self.langgraph_api_url = self.langgraph_api_url.rstrip("/")
 
         # Initialize LangGraph SDK client
-        self.langgraph_client = get_client(
-            url=self.langgraph_api_url, api_key=self.langsmith_api_key
-        )
+        self.langgraph_client = get_client(url=self.langgraph_api_url, api_key=None)
         logger.info(
             f"✓ LangGraph client initialized with URL: {self.langgraph_api_url}"
         )
-        if self.langsmith_api_key:
-            logger.info("✓ LangGraph client initialized with API key.")
-        else:
-            logger.warning("⚠ LangGraph client initialized without API key")
 
         # Initialize LangChain auth client
         langchain_api_key = os.getenv("LANGCHAIN_API_KEY")
@@ -577,6 +594,14 @@ class TriggerServer:
     ) -> dict[str, Any]:
         """Handle an incoming request with a handler function."""
         try:
+            logger.info(
+                "incoming_trigger_request method=%s path=%s trigger_id=%s provider=%s content_type=%s",
+                request.method,
+                request.url.path,
+                getattr(trigger, "id", "<unknown>"),
+                getattr(trigger, "provider", "<unknown>"),
+                request.headers.get("content-type", ""),
+            )
             if request.method == "POST":
                 if request.headers.get("content-type", "").startswith(
                     "application/json"
@@ -584,15 +609,34 @@ class TriggerServer:
                     # Read body once for both auth and parsing
                     body_bytes = await request.body()
                     body_str = body_bytes.decode("utf-8")
+                    logger.debug(
+                        "request_body_read bytes=%s",
+                        len(body_bytes) if body_bytes is not None else 0,
+                    )
 
+                    # TODO(sam/palash): We should not have API specific things in this framework repo. We should clean this up.
                     if self._is_slack_trigger(trigger):
+                        logger.info(
+                            "slack_request_detected trigger_id=%s verifying_signature=true",
+                            getattr(trigger, "id", "<unknown>"),
+                        )
                         await self._verify_slack_webhook_auth_with_body(
                             request, body_str
                         )
 
                     import json
 
-                    payload = json.loads(body_str)
+                    try:
+                        payload = json.loads(body_str)
+                    except Exception as e:
+                        logger.error("json_parse_error error=%s", e)
+                        raise
+                    logger.info(
+                        "payload_parsed kind=json keys=%s",
+                        sorted(payload.keys())
+                        if isinstance(payload, dict)
+                        else "<non-dict>",
+                    )
 
                     if (
                         payload.get("type") == "url_verification"
@@ -604,22 +648,55 @@ class TriggerServer:
                     # Handle form data or other content types
                     body = await request.body()
                     payload = {"raw_body": body.decode("utf-8") if body else ""}
+                    logger.info(
+                        "payload_parsed kind=raw length=%s",
+                        len(payload.get("raw_body", "")),
+                    )
             else:
                 payload = dict(request.query_params)
+                logger.info(
+                    "payload_parsed kind=query keys=%s",
+                    sorted(payload.keys()),
+                )
 
             query_params = dict(request.query_params)
+            logger.info(
+                "invoking_trigger_handler trigger_id=%s provider=%s query_keys=%s",
+                getattr(trigger, "id", "<unknown>"),
+                getattr(trigger, "provider", "<unknown>"),
+                sorted(query_params.keys()),
+            )
             result = await trigger.trigger_handler(
                 payload, query_params, self.database, self.langchain_auth_client
             )
+            logger.info(
+                "trigger_handler_completed invoke_agent=%s messages=%s has_registration=%s",
+                getattr(result, "invoke_agent", None),
+                len(getattr(result, "agent_messages", []) or []),
+                bool(getattr(result, "registration", None)),
+            )
             if not result.invoke_agent:
+                logger.info(
+                    "no_agent_invocation trigger_id=%s returning_handler_response=true",
+                    getattr(trigger, "id", "<unknown>"),
+                )
                 return result.response_body
 
             registration_id = result.registration["id"]
             agent_links = await self.database.get_agents_for_trigger(registration_id)
+            logger.info(
+                "agent_links_fetched registration_id=%s count=%s",
+                registration_id,
+                len(agent_links) if agent_links is not None else 0,
+            )
 
             agents_invoked = 0
             # Iterate through each message and invoke agents for each
             for message in result.agent_messages:
+                logger.debug(
+                    "processing_agent_message message_len=%s",
+                    len(str(message)) if message is not None else 0,
+                )
                 for agent_link in agent_links:
                     agent_id = (
                         agent_link
@@ -629,17 +706,34 @@ class TriggerServer:
                     # Ensure agent_id and user_id are strings for JSON serialization
                     agent_id_str = str(agent_id)
                     user_id_str = str(result.registration["user_id"])
+                    tenant_id_str = str(
+                        result.registration.get("metadata", {})
+                        .get("client_metadata", {})
+                        .get("tenant_id")
+                    )
 
                     agent_input = {"messages": [{"role": "human", "content": message}]}
 
                     try:
+                        logger.info(
+                            "invoking_agent assistant_id=%s user_id=%s tenant_id=%s",
+                            agent_id_str,
+                            user_id_str,
+                            tenant_id_str,
+                        )
                         success = await self._invoke_agent(
                             agent_id=agent_id_str,
                             user_id=user_id_str,
+                            tenant_id=tenant_id_str,
                             input_data=agent_input,
                         )
                         if success:
                             agents_invoked += 1
+                            logger.info(
+                                "agent_invocation_success assistant_id=%s registration_id=%s",
+                                agent_id_str,
+                                registration_id,
+                            )
                     except Exception as e:
                         logger.error(
                             f"Error invoking agent {agent_id_str}: {e}", exc_info=True
@@ -662,6 +756,7 @@ class TriggerServer:
         self,
         agent_id: str,
         user_id: str,
+        tenant_id: str,
         input_data: dict[str, Any],
     ) -> bool:
         """Invoke LangGraph agent using the SDK."""
@@ -671,20 +766,23 @@ class TriggerServer:
 
         try:
             headers = {
-                "x-auth-scheme": "agent-builder-trigger",
+                "x-api-key": "",
+                "x-auth-scheme": "langsmith-agent",
                 "x-user-id": user_id_str,
+                "x-tenant-id": tenant_id,
+                "x-service-key": get_x_service_jwt_token(
+                    payload={
+                        "tenant_id": tenant_id,
+                        "user_id": user_id_str,
+                    }
+                ),
             }
 
-            # Note: API key is already set in client initialization, no need to add to headers
-            if not self.langsmith_api_key:
-                logger.warning(
-                    "No LANGSMITH_API_KEY available - authentication may fail"
-                )
-
             thread = await self.langgraph_client.threads.create(
                 metadata={
                     "triggered_by": "langchain-triggers",
                     "user_id": user_id_str,
+                    "tenant_id": tenant_id,
                 },
                 headers=headers,
             )
@@ -697,6 +795,7 @@ class TriggerServer:
                 metadata={
                     "triggered_by": "langchain-triggers",
                     "user_id": user_id_str,
+                    "tenant_id": tenant_id,
                 },
                 headers=headers,
             )

langchain_triggers/cron_manager.py

@@ -202,6 +202,9 @@ class CronTriggerManager:
         """Execute a cron job - invoke agents. Can be called manually or by scheduler."""
         registration_id = registration["id"]
         user_id = registration["user_id"]
+        tenant_id = (
+            registration.get("metadata", {}).get("client_metadata", {}).get("tenant_id")
+        )
 
         # Get agent links
         agent_links = await self.trigger_server.database.get_agents_for_trigger(
@@ -222,6 +225,7 @@ class CronTriggerManager:
             # Ensure agent_id and user_id are strings for JSON serialization
             agent_id_str = str(agent_id)
             user_id_str = str(user_id)
+            tenant_id_str = str(tenant_id)
 
             current_time = datetime.utcnow()
             current_time_str = current_time.strftime("%A, %B %d, %Y at %H:%M UTC")
@@ -239,6 +243,7 @@ class CronTriggerManager:
                 success = await self.trigger_server._invoke_agent(
                     agent_id=agent_id_str,
                     user_id=user_id_str,
+                    tenant_id=tenant_id_str,
                     input_data=agent_input,
                 )
                 if success: