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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-trigger-server
-Version: 0.2.8
+Version: 0.2.9
 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.9.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=MARlj82Ena79n0MG23m-Vu9w6YDQWqp0Ao24TuEKTzs,35325
 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.9.dist-info/METADATA,sha256=8ER3hixfKxyvOIn308bFj64yjZkscTxotQRkah3pS60,1486
+langchain_trigger_server-0.2.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langchain_trigger_server-0.2.9.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")
@@ -585,6 +602,7 @@ class TriggerServer:
                     body_bytes = await request.body()
                     body_str = body_bytes.decode("utf-8")
 
+                    # 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):
                         await self._verify_slack_webhook_auth_with_body(
                             request, body_str
@@ -629,6 +647,11 @@ 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}]}
 
@@ -636,6 +659,7 @@ class TriggerServer:
                         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:
@@ -662,6 +686,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 +696,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 +725,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: