Add assignee_id parameter to linear_search_issues tool

src/codegen/extensions/langchain/tools.py

@@ -66,11 +66,11 @@
 class ViewFileTool(BaseTool):
     """Tool for viewing file contents and metadata."""
 
     name: ClassVar[str] = "view_file"
     description: ClassVar[str] = """View the contents and metadata of a file in the codebase.
 For large files (>500 lines), content will be paginated. Use start_line and end_line to navigate through the file.
 The response will indicate if there are more lines available to view."""
     args_schema: ClassVar[type[BaseModel]] = ViewFileInput
     codebase: Codebase = Field(exclude=True)
 
     def __init__(self, codebase: Codebase) -> None:
@@ -108,9 +108,9 @@
 class ListDirectoryTool(BaseTool):
     """Tool for listing directory contents."""
 
     name: ClassVar[str] = "list_directory"
     description: ClassVar[str] = "List contents of a directory in the codebase"
     args_schema: ClassVar[type[BaseModel]] = ListDirectoryInput
     codebase: Codebase = Field(exclude=True)
 
     def __init__(self, codebase: Codebase) -> None:
@@ -138,9 +138,9 @@
 class RipGrepTool(BaseTool):
     """Tool for searching the codebase via RipGrep."""
 
     name: ClassVar[str] = "search"
     description: ClassVar[str] = "Search the codebase using `ripgrep` or regex pattern matching"
     args_schema: ClassVar[type[BaseModel]] = SearchInput
     codebase: Codebase = Field(exclude=True)
 
     def __init__(self, codebase: Codebase) -> None:
@@ -162,7 +162,7 @@
 class EditFileTool(BaseTool):
     """Tool for editing files."""
 
     name: ClassVar[str] = "edit_file"
     description: ClassVar[str] = r"""
 Edit a file by replacing its entire content. This tool should only be used for replacing entire file contents.
 Input for searching the codebase.
@@ -744,6 +744,7 @@
 
     query: str = Field(..., description="Search query string")
     limit: int = Field(default=10, description="Maximum number of issues to return")
+    assignee_id: str | None = Field(default=None, description="Optional assignee ID to filter issues by")
 
 
 class LinearSearchIssuesTool(BaseTool):
@@ -757,8 +758,8 @@
     def __init__(self, client: LinearClient) -> None:
         super().__init__(client=client)
 
-    def _run(self, query: str, limit: int = 10) -> str:
-        result = linear_search_issues_tool(self.client, query, limit)
+    def _run(self, query: str, limit: int = 10, assignee_id: str | None = None) -> str:
+        result = linear_search_issues_tool(self.client, query, limit, assignee_id)
         return result.render()
 
 

src/codegen/extensions/tools/linear/linear.py

@@ -273,8 +273,19 @@ def linear_register_webhook_tool(
         )
 
 
-def linear_search_issues_tool(client: LinearClient, query: str, limit: int = 10) -> LinearSearchObservation:
-    """Search for issues using a query string."""
+def linear_search_issues_tool(client: LinearClient, query: str, limit: int = 10, assignee_id: str | None = None) -> LinearSearchObservation:
+    """Search for issues using a query string.
+
+    Args:
+        client: LinearClient instance
+        query: Search query string
+        limit: Maximum number of issues to return
+        assignee_id: Optional assignee ID to filter issues by
+    """
+    # If assignee_id is provided and not already in the query, add it
+    if assignee_id and "assignee:" not in query:
+        query = f"{query} assignee:{assignee_id}"
+
     try:
         issues = client.search_issues(query, limit)
         return LinearSearchObservation(