langchain-trigger-server 0.1.1__tar.gz → 0.1.3__tar.gz

{langchain_trigger_server-0.1.1 → langchain_trigger_server-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-trigger-server
-Version: 0.1.1
+Version: 0.1.3
 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
@@ -18,6 +18,7 @@ Classifier: Programming Language :: Python :: 3.12
 Requires-Python: >=3.9
 Requires-Dist: fastapi>=0.100.0
 Requires-Dist: httpx>=0.24.0
+Requires-Dist: langgraph-sdk>=0.2.6
 Requires-Dist: pydantic>=2.0.0
 Requires-Dist: python-jose[cryptography]>=3.3.0
 Requires-Dist: python-multipart>=0.0.6

{langchain_trigger_server-0.1.1 → langchain_trigger_server-0.1.3}/langchain_triggers/app.py

@@ -9,7 +9,7 @@ from typing import Any, Callable, Dict, List, Optional
 import httpx
 
 from fastapi import FastAPI, HTTPException, Request, Depends
-from fastapi.middleware.cors import CORSMiddleware
+from langgraph_sdk import get_client
 from starlette.middleware.base import BaseHTTPMiddleware
 from starlette.responses import Response
 
@@ -72,8 +72,7 @@ class TriggerServer:
     def __init__(
         self,
         auth_handler: Callable,
-        cors_origins: Optional[List[str]] = None,
-        database: Optional[TriggerDatabaseInterface] = None,
+        langgraph_headers_builder: Callable,
     ):
         self.app = FastAPI(
             title="Triggers Server",
@@ -81,8 +80,9 @@ class TriggerServer:
             version="0.1.0"
         )
         
-        self.database = database or create_database()  # Default to Supabase
+        self.database = create_database()
         self.auth_handler = auth_handler
+        self.langgraph_headers_builder = langgraph_headers_builder
         
         # LangGraph configuration
         self.langgraph_api_url = os.getenv("LANGGRAPH_API_URL")
@@ -93,6 +93,9 @@ 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)
+
         self.langchain_auth_client = None
         try:
             from langchain_auth import Client
@@ -107,18 +110,6 @@ class TriggerServer:
         
         self.triggers: List[TriggerTemplate] = []
         
-        # Setup CORS
-        if cors_origins is None:
-            cors_origins = ["*"]  # Allow all origins by default
-            
-        self.app.add_middleware(
-            CORSMiddleware,
-            allow_origins=cors_origins,
-            allow_credentials=True,
-            allow_methods=["*"],
-            allow_headers=["*"],
-        )
-        
         # Setup authentication middleware
         self.app.add_middleware(AuthenticationMiddleware, auth_handler=auth_handler)
         
@@ -282,7 +273,7 @@ class TriggerServer:
                 except Exception as e:
                     raise HTTPException(
                         status_code=400,
-                        detail=f"Invalid payload for trigger {trigger_type}: {str(e)}"
+                        detail=f"Invalid payload for trigger: {str(e)}"
                     )
                 
                 # Call the trigger's registration handler with parsed registration model
@@ -568,54 +559,35 @@ class TriggerServer:
         user_id: str,
         input_data: Dict[str, Any],
     ) -> Dict[str, Any]:
-        """Invoke LangGraph agent directly."""
-        headers = {
-            "Content-Type": "application/json",
-            "x-auth-scheme": "langchain-trigger-server-event",
-        }
-        
-        # Add API key if provided
-        if self.langgraph_api_key:
-            headers["Authorization"] = f"Bearer {self.langgraph_api_key}"
-        
-        # Add user-specific headers
-        if user_id:
-            headers["x-supabase-user-id"] = user_id
-        
-        payload = {
-            "input": input_data,
-            "assistant_id": agent_id,
-            "metadata": {
-                "triggered_by": "langchain-triggers",
-                "user_id": user_id,
-            },
-        }
-        
-        # Let LangGraph create a new thread automatically
-        url = f"{self.langgraph_api_url}/runs"
-        
+        """Invoke LangGraph agent using the SDK."""
         logger.info(f"Invoking LangGraph agent {agent_id} for user {user_id}")
         
-        async with httpx.AsyncClient() as client:
-            try:
-                response = await client.post(
-                    url,
-                    json=payload,
-                    headers=headers,
-                    timeout=30.0,
-                )
-                response.raise_for_status()
-                
-                result = response.json()
-                logger.info(f"Successfully invoked agent {agent_id}")
-                return result
-                
-            except httpx.HTTPStatusError as e:
-                logger.error(f"HTTP error invoking agent: {e.response.status_code} - {e.response.text}")
-                raise
-            except Exception as e:
-                logger.error(f"Error invoking agent {agent_id}: {e}")
-                raise
+        try:
+            # Build headers using the custom function
+            headers = await self.langgraph_headers_builder(
+                user_id=user_id,
+                api_key=self.langgraph_api_key,
+                agent_id=agent_id
+            )
+            
+            # Create a background run using the SDK
+            run = await self.langgraph_client.runs.create(
+                thread_id=None,  # Let LangGraph create a new thread
+                assistant_id=agent_id,
+                input=input_data,
+                metadata={
+                    "triggered_by": "langchain-triggers",
+                    "user_id": user_id,
+                },
+                headers=headers,
+            )
+            
+            logger.info(f"Successfully invoked agent {agent_id}, run_id: {run['run_id']}")
+            return run
+            
+        except Exception as e:
+            logger.error(f"Error invoking agent {agent_id}: {e}")
+            raise
     
     async def _get_authenticated_user(self, trigger: TriggerTemplate, user_id: str) -> UserAuthInfo:
         """Get authenticated user with OAuth tokens for the trigger's required providers."""

{langchain_trigger_server-0.1.1 → langchain_trigger_server-0.1.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "langchain-trigger-server"
-version = "0.1.1"
+version = "0.1.3"
 description = "Generic event-driven triggers framework"
 readme = "README.md"
 requires-python = ">=3.9"
@@ -30,6 +30,7 @@ dependencies = [
     "httpx>=0.24.0",
     "python-multipart>=0.0.6",
     "python-jose[cryptography]>=3.3.0",
+    "langgraph-sdk>=0.2.6",
 ]
 
 [project.optional-dependencies]