langchain-trigger-server 0.2.0__tar.gz → 0.2.1__tar.gz

{langchain_trigger_server-0.2.0 → langchain_trigger_server-0.2.1}/PKG-INFO

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

@@ -231,21 +231,18 @@ class TriggerServer:
             try:
                 user_id = current_user["identity"]
                 
-                # Get user's trigger registrations using new schema
-                user_registrations = await self.database.get_user_trigger_registrations(user_id)
+                # Get user's trigger registrations with linked agents in a single query
+                user_registrations = await self.database.get_user_trigger_registrations_with_agents(user_id)
                 
                 # Format response to match expected structure
                 registrations = []
                 for reg in user_registrations:
-                    # Get linked agent IDs
-                    linked_agent_ids = await self.database.get_agents_for_trigger(reg["id"])
-                    
                     registrations.append({
                         "id": reg["id"],
                         "user_id": reg["user_id"],
                         "template_id": reg.get("trigger_templates", {}).get("id"),
                         "resource": reg["resource"],
-                        "linked_assistant_ids": linked_agent_ids,  # For backward compatibility
+                        "linked_agent_ids": reg.get("linked_agent_ids", []),
                         "created_at": reg["created_at"]
                     })
                 

{langchain_trigger_server-0.2.0 → langchain_trigger_server-0.2.1}/langchain_triggers/database/interface.py

@@ -50,6 +50,11 @@ class TriggerDatabaseInterface(ABC):
         """Get all trigger registrations for a user."""
         pass
     
+    @abstractmethod
+    async def get_user_trigger_registrations_with_agents(self, user_id: str) -> List[Dict[str, Any]]:
+        """Get all trigger registrations for a user with linked agents in a single query."""
+        pass
+    
     @abstractmethod
     async def get_trigger_registration(
         self, 

{langchain_trigger_server-0.2.0 → langchain_trigger_server-0.2.1}/langchain_triggers/database/supabase.py

@@ -162,6 +162,31 @@ class SupabaseTriggerDatabase(TriggerDatabaseInterface):
         except Exception as e:
             logger.error(f"Error getting user trigger registrations: {e}")
             return []
+
+    async def get_user_trigger_registrations_with_agents(self, user_id: str) -> List[Dict[str, Any]]:
+        """Get all trigger registrations for a user with linked agents in a single query."""
+        try:
+            response = self.client.table("trigger_registrations").select("""
+                *,
+                trigger_templates(id, name, description),
+                agent_trigger_links(agent_id)
+            """).eq("user_id", user_id).order("created_at", desc=True).execute()
+            
+            # Process the results to extract linked agent IDs
+            if response.data:
+                for registration in response.data:
+                    # Extract agent IDs from the agent_trigger_links
+                    agent_links = registration.get("agent_trigger_links", [])
+                    linked_agent_ids = [link.get("agent_id") for link in agent_links if link.get("agent_id")]
+                    registration["linked_agent_ids"] = linked_agent_ids
+                    
+                    # Clean up the raw agent_trigger_links data as it's no longer needed
+                    registration.pop("agent_trigger_links", None)
+            
+            return response.data or []
+        except Exception as e:
+            logger.error(f"Error getting user trigger registrations with agents: {e}")
+            return []
     
     async def get_trigger_registration(self, registration_id: str, user_id: str = None) -> Optional[Dict[str, Any]]:
         """Get a specific trigger registration."""

{langchain_trigger_server-0.2.0 → langchain_trigger_server-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "langchain-trigger-server"
-version = "0.2.0"
+version = "0.2.1"
 description = "Generic event-driven triggers framework"
 readme = "README.md"
 requires-python = ">=3.9"