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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-trigger-server
-Version: 0.1.2
+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.2 → langchain_trigger_server-0.1.3}/langchain_triggers/app.py

@@ -9,6 +9,7 @@ from typing import Any, Callable, Dict, List, Optional
 import httpx
 
 from fastapi import FastAPI, HTTPException, Request, Depends
+from langgraph_sdk import get_client
 from starlette.middleware.base import BaseHTTPMiddleware
 from starlette.responses import Response
 
@@ -92,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
@@ -555,48 +559,35 @@ class TriggerServer:
         user_id: str,
         input_data: Dict[str, Any],
     ) -> Dict[str, Any]:
-        """Invoke LangGraph agent directly."""
-        # 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
-        )
-        
-        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.2 → langchain_trigger_server-0.1.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "langchain-trigger-server"
-version = "0.1.2"
+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]