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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-trigger-server
-Version: 0.1.2
+Version: 0.1.5
 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.5}/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
@@ -523,12 +527,13 @@ class TriggerServer:
                 }
 
                 try:
-                    await self._invoke_agent(
+                    success = await self._invoke_agent(
                         agent_id=agent_id,
                         user_id=registration["user_id"],
                         input_data=agent_input,
                     )
-                    agents_invoked += 1
+                    if success:
+                        agents_invoked += 1
                 except Exception as e:
                     logger.error(f"Error invoking agent {agent_id}: {e}", exc_info=True)
             
@@ -554,47 +559,40 @@ class TriggerServer:
         agent_id: str,
         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"
-        
+    ) -> bool:
+        """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:
+        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,
+                assistant_id=agent_id,
+                input=input_data,
+                metadata={
+                    "triggered_by": "langchain-triggers",
+                    "user_id": user_id,
+                },
+                headers=headers,
+                if_not_exists="create",
+            )
+            
+            logger.info(f"Successfully invoked agent {agent_id}, run_id: {run['run_id']}, thread_id: {run['thread_id']}")
+            return True
+            
+        except Exception as e:
+            # Handle 404s (agent not found) as warnings, not errors
+            if hasattr(e, 'response') and getattr(e.response, 'status_code', None) == 404:
+                logger.warning(f"Agent {agent_id} not found (404) - agent may have been deleted or moved")
+                return False
+            else:
                 logger.error(f"Error invoking agent {agent_id}: {e}")
                 raise
     

{langchain_trigger_server-0.1.2 → langchain_trigger_server-0.1.5}/langchain_triggers/database/interface.py

@@ -116,27 +116,7 @@ class TriggerDatabaseInterface(ABC):
     async def get_triggers_for_agent(self, agent_id: str) -> List[Dict[str, Any]]:
         """Get all trigger registrations linked to an agent."""
         pass
-    
-    @abstractmethod
-    async def set_agent_trigger_links(
-        self,
-        agent_id: str, 
-        registration_ids: List[str],
-        created_by: str
-    ) -> bool:
-        """Replace all trigger links for an agent (atomic operation)."""
-        pass
-    
-    @abstractmethod
-    async def replace_trigger_agent_links(
-        self,
-        registration_id: str,
-        agent_ids: List[str], 
-        created_by: str
-    ) -> bool:
-        """Replace all agent links for a trigger (atomic operation)."""
-        pass
-    
+
     # ========== Helper Methods ==========
     
     @abstractmethod

{langchain_trigger_server-0.1.2 → langchain_trigger_server-0.1.5}/langchain_triggers/database/supabase.py

@@ -277,68 +277,6 @@ class SupabaseTriggerDatabase(TriggerDatabaseInterface):
             logger.error(f"Error getting triggers for agent: {e}")
             return []
     
-    async def set_agent_trigger_links(
-        self,
-        agent_id: str, 
-        registration_ids: List[str],
-        created_by: str
-    ) -> bool:
-        """Replace all trigger links for an agent (atomic operation)."""
-        try:
-            # Delete existing links
-            await self.client.table("agent_trigger_links").delete().eq("agent_id", agent_id).execute()
-            
-            # Create new links
-            if registration_ids:
-                links = [
-                    {
-                        "agent_id": agent_id,
-                        "registration_id": reg_id,
-                        "created_by": created_by
-                    }
-                    for reg_id in registration_ids
-                ]
-                
-                response = self.client.table("agent_trigger_links").insert(links).execute()
-                return bool(response.data)
-            
-            return True  # Successfully cleared all links
-            
-        except Exception as e:
-            logger.error(f"Error setting agent trigger links: {e}")
-            return False
-    
-    async def replace_trigger_agent_links(
-        self,
-        registration_id: str,
-        agent_ids: List[str], 
-        created_by: str
-    ) -> bool:
-        """Replace all agent links for a trigger (atomic operation)."""
-        try:
-            # Delete existing links
-            await self.client.table("agent_trigger_links").delete().eq("registration_id", registration_id).execute()
-            
-            # Create new links
-            if agent_ids:
-                links = [
-                    {
-                        "agent_id": agent_id,
-                        "registration_id": registration_id,
-                        "created_by": created_by
-                    }
-                    for agent_id in agent_ids
-                ]
-                
-                response = self.client.table("agent_trigger_links").insert(links).execute()
-                return bool(response.data)
-            
-            return True  # Successfully cleared all links
-            
-        except Exception as e:
-            logger.error(f"Error replacing trigger agent links: {e}")
-            return False
-    
     # ========== Helper Methods ==========
     
     async def get_user_from_token(self, token: str) -> Optional[str]:

{langchain_trigger_server-0.1.2 → langchain_trigger_server-0.1.5}/pyproject.toml

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