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

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

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

@@ -181,7 +181,6 @@ class TriggerServer:
                     name=trigger.name,
                     description=trigger.description,
                     registration_schema=trigger.registration_model.model_json_schema(),
-                    require_display_name=getattr(trigger.registration_model, 'require_display_name', False)
                 )
                 logger.info(f"✓ Successfully created trigger template: {trigger.name} ({trigger.id})")
             else:
@@ -217,7 +216,6 @@ class TriggerServer:
                     "path": "/api/triggers/registrations",
                     "method": "POST",
                     "payloadSchema": template.get("registration_schema", {}),
-                    "requireDisplayName": template.get("require_display_name", False),
                 })
             
             return {
@@ -231,21 +229,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"]
                     })
                 
@@ -314,12 +309,7 @@ class TriggerServer:
                         media_type="application/json"
                     )
                 
-                resource_dict = registration_instance.model_dump(exclude={'require_display_name'})
-                
-                # Remove display_name if it's null/None and not required
-                if (not getattr(trigger.registration_model, 'require_display_name', False) and 
-                    resource_dict.get('display_name') is None):
-                    resource_dict.pop('display_name', None)
+                resource_dict = registration_instance.model_dump()
 
                 registration = await self.database.create_trigger_registration(
                     user_id=user_id,

{langchain_trigger_server-0.2.0 → langchain_trigger_server-0.2.2}/langchain_triggers/core.py

@@ -75,11 +75,4 @@ class TriggerRegistrationResult(BaseModel):
 
 class TriggerRegistrationModel(BaseModel):
     """Base class for trigger resource models that define how webhooks are matched to registrations."""
-    
-    require_display_name: bool = False  # Class attribute to indicate if display_name is required
-    display_name: Optional[str] = Field(default=None, description="Optional display name for this registration")
-    
-    def model_post_init(self, __context) -> None:
-        """Validate that display_name is provided when require_display_name is True."""
-        if self.require_display_name and not self.display_name:
-            raise ValueError(f"{self.__class__.__name__} requires display_name to be set")
+    pass

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

@@ -16,8 +16,7 @@ class TriggerDatabaseInterface(ABC):
         provider: str,
         name: str, 
         description: str = None,
-        registration_schema: Dict = None,
-        require_display_name: bool = False
+        registration_schema: Dict = None
     ) -> Optional[Dict[str, Any]]:
         """Create a new trigger template."""
         pass
@@ -50,6 +49,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.2}/langchain_triggers/database/supabase.py

@@ -75,8 +75,7 @@ class SupabaseTriggerDatabase(TriggerDatabaseInterface):
         provider: str,
         name: str, 
         description: str = None,
-        registration_schema: Dict = None,
-        require_display_name: bool = False
+        registration_schema: Dict = None
     ) -> Optional[Dict[str, Any]]:
         """Create a new trigger template."""
         try:
@@ -85,8 +84,7 @@ class SupabaseTriggerDatabase(TriggerDatabaseInterface):
                 "provider": provider,
                 "name": name,
                 "description": description,
-                "registration_schema": registration_schema or {},
-                "require_display_name": require_display_name
+                "registration_schema": registration_schema or {}
             }
             
             response = self.client.table("trigger_templates").insert(data).execute()
@@ -162,6 +160,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.2}/pyproject.toml

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