Rename tools and add titles search functionality

- Rename pia_search -> pia_search_content for clarity
- Rename pia_search_facets -> pia_search_content_facets
- Add new pia_search_titles tool for title-only searches
- Add new pia_search_titles_facets tool for title search facets
- Update tool schemas to match remote server implementation
- Synchronize prompts with examples/prompts structure
- Update all tests to reflect new tool names (14 tests passing)
- Update README documentation to describe all four tools
- Update imports and exports throughout codebase

Now provides comprehensive search capabilities:
- Content search: Full-text search in document content
- Title search: Fast document discovery by title
- Facets for both modes: Filter value discovery
- All tools support OData filtering with proper field names

Author: dividor

README.md

@@ -123,22 +123,57 @@ For Docker:
 
 ## 💡 Available Tools
 
-The server provides two main tools:
+The server provides four main tools:
 
-### 1. PIA Search
-Comprehensive search with OData filtering and faceting. The `filter` parameter uses standard [OData query syntax](https://docs.oasis-open.org/odata/odata/v4.01/odata-v4.01-part2-url-conventions.html).
+### 1. `pia_search_content`
+
+Comprehensive search tool for querying document content and recommendations in the PIA database.
 
-**Tool Name:** `pia_search`
+**Tool Name:** `pia_search_content`
 
 **Parameters:**
 - `query` (required): Search query text
 - `filter` (optional): OData filter expression
 - `page` (optional): Page number (default: 1)
 - `page_size` (optional): Results per page (default: 10)
 - `search_mode` (optional): Search mode (default: "content")
+
+### 2. `pia_search_content_facets`
+
+Tool for discovering available filter values before performing content searches.
+
+**Parameters:**
+- `query` (optional): Query to get facets for (if empty, gets all facets)
+- `filter` (optional): OData filter expression
+
+### 3. `pia_search_titles`
+
+Search document titles only for faster discovery of available documents.
+
+**Parameters:**
+- `query` (required): Search query text (searches document titles only)
+- `filter` (optional): OData filter expression
+- `page` (optional): Page number (default: 1)
+- `page_size` (optional): Results per page (default: 10)
+- `search_mode` (optional): Search mode (default: "titles")
 - `limit` (optional): Maximum results limit
 - `include_facets` (optional): Include facets in results (default: false)
 
+### 4. `pia_search_titles_facets`
+
+Tool for discovering available filter values before performing title searches.
+
+**Parameters:**
+- `query` (optional): Query to get facets for (if empty, gets all facets)
+- `filter` (optional): OData filter expression
+
+## Search Modes
+
+Comprehensive search with OData filtering and faceting. The `filter` parameter uses standard [OData query syntax](https://docs.oasis-open.org/odata/odata/v4.01/odata-v4.01-part2-url-conventions.html).
+
+- **Content Search** (`pia_search_content`): Searches within document content and recommendations for comprehensive results
+- **Title Search** (`pia_search_titles`): Searches document titles only - faster and useful for document discovery
+
 **Example Filter Expressions:**
 - Basic filter: `"SourceDocumentDataSource eq 'GAO'"`
 - Multiple conditions: `"SourceDocumentDataSource eq 'GAO' or SourceDocumentDataSource eq 'OIG'"`
@@ -183,7 +218,7 @@ Discover available field names and values for filtering.
 - Find possible field values (e.g., "OIG", "GAO", "audit_report")
 - Understand data types for each field (string, date, number)
 
-This information helps you construct proper `filter` expressions for the `pia_search` tool.
+This information helps you construct proper `filter` expressions for the search tools.
 
 ## 🔍 Filter Discovery Workflow
 

src/pia_mcp_server/prompts/handlers.py

@@ -9,26 +9,14 @@
 # Available prompts for PIA MCP server
 AVAILABLE_PROMPTS = [
     {
-        "name": "summary_prompt",
-        "description": "Instructions for LLM to summarize information only from provided search results with proper citations",
-        "arguments": [
-            {
-                "name": "search_results",
-                "description": "The search results to summarize (will be inserted into <SEARCH_RESULTS> tags)",
-                "required": True,
-            },
-        ],
+        "name": "summarization_guidance",
+        "description": "Provides guidance on how to summarize information from PIA search results with proper citations",
+        "arguments": [],
     },
     {
-        "name": "search_prompt",
-        "description": "Instructions for LLM on how to effectively use PIA search tools with proper filtering",
-        "arguments": [
-            {
-                "name": "user_query",
-                "description": "The user's search query to analyze for filter criteria",
-                "required": True,
-            },
-        ],
+        "name": "search_guidance",
+        "description": "Provides guidance on how to perform PIA searches with or without filters",
+        "arguments": [],
     },
 ]
 
@@ -76,10 +64,10 @@ async def get_prompt(
     arguments = arguments or {}
 
     # Generate prompt content based on the type
-    if name == "summary_prompt":
-        content = _generate_summary_prompt(arguments)
-    elif name == "search_prompt":
-        content = _generate_search_prompt(arguments)
+    if name == "summarization_guidance":
+        content = _generate_summarization_guidance()
+    elif name == "search_guidance":
+        content = _generate_search_guidance()
     else:
         content = f"Prompt template for {name} - implement specific logic based on arguments: {arguments}"
 
@@ -93,11 +81,9 @@ async def get_prompt(
     )
 
 
-def _generate_summary_prompt(arguments: Dict[str, str]) -> str:
-    """Generate summary prompt for LLM."""
-    search_results = arguments.get("search_results", "")
-
-    return f"""You are an assistant that summarizes information **only** from the provided search results.
+def _generate_summarization_guidance() -> str:
+    """Generate summarization guidance prompt."""
+    return """You are an assistant that summarizes information **only** from the provided search results.
 
 Your task:
 1. **Only include facts that appear in the provided search results.**
@@ -130,8 +116,6 @@ def _generate_summary_prompt(arguments: Dict[str, str]) -> str:
 
 Input to use:
 <SEARCH_RESULTS>
-{search_results}
-</SEARCH_RESULTS>
 
 Output template:
 ---
@@ -151,33 +135,37 @@ def _generate_summary_prompt(arguments: Dict[str, str]) -> str:
 - If you can't find enough information for a point, omit it entirely."""
 
 
-def _generate_search_prompt(arguments: Dict[str, str]) -> str:
-    """Generate search prompt for LLM."""
-    user_query = arguments.get("user_query", "")
-
-    return f"""You can perform searches using the PIA Search tools with or without filters.
+def _generate_search_guidance() -> str:
+    """Generate search guidance prompt."""
+    return """You can perform searches using the PIA Search tools with or without filters.
 
 General rules:
 - Run an **unfiltered search** by default if no filter criteria are mentioned in the user's request.
 - If the user's request includes any specific filter criteria (e.g., agency name, year, category):
-  1. Call `pia_search_facets` once to discover the filterable field names and allowed values.
+  1. Call `pia_search_content_facets` or `pia_search_titles_facets` once to discover the filterable field names and allowed values.
   2. Use these to build a valid OData filter expression.
   3. Call the search tool with the filter applied.
 
+Available search tools:
+- `pia_search_content`: Search within document content and recommendations
+- `pia_search_content_facets`: Get available filter values for content search
+- `pia_search_titles`: Search document titles only (faster, good for discovery)
+- `pia_search_titles_facets`: Get available filter values for title search
+
 Process for applying filters:
 
 1. **Detect filter intent**:
    - Examine the user's query for references to agencies, dates, categories, or other filterable attributes.
    - If such references are present, treat the search as **filtered**.
 
 2. **Discover filterable fields and values** (only if filtering):
-   - Call the `pia_search_facets` tool (one time per session unless filters change).
+   - Call the appropriate facets tool (one time per session unless filters change).
    - Review the output to see:
      - Field names that support filtering.
      - Possible values for each field.
 
 3. **Build the filter expression**:
-   - Use **only field names and values** returned by `pia_search_facets`.
+   - Use **only field names and values** returned by the facets tools.
    - Construct the filter in **OData syntax**:
      - `SourceDocumentDataSource eq 'GAO'`
      - `(SourceDocumentDataSource eq 'GAO' or SourceDocumentDataSource eq 'OIG') and SourceDocumentPublishDate ge '2020-01-01'`
@@ -200,12 +188,12 @@ def _generate_search_prompt(arguments: Dict[str, str]) -> str:
      - Inform the user you are showing unfiltered results because no matches were found with the filter.
 
 6. **Validation**:
-   - Never use a field or value not provided by `pia_search_facets`.
-   - If the user requests a filter that doesn't exist in `pia_search_facets`, explain it's not available and offer an unfiltered search instead.
+   - Never use a field or value not provided by the facets tools.
+   - If the user requests a filter that doesn't exist in facets, explain it's not available and offer an unfiltered search instead.
 
 Goal:
 - Default to unfiltered search unless filter criteria are clearly present in the query.
 - Always validate filter fields/values before applying them.
 - Fall back to unfiltered if filtering produces zero results and it hasn't already been run.
 
-User query to analyze: "{user_query}\""""
+"""

src/pia_mcp_server/server.py

@@ -14,12 +14,16 @@
 from mcp.server.stdio import stdio_server
 from .config import Settings
 from .tools import (
-    handle_pia_search,
-    handle_pia_search_facets,
+    handle_pia_search_content,
+    handle_pia_search_content_facets,
+    handle_pia_search_titles,
+    handle_pia_search_titles_facets,
 )
 from .tools import (
-    pia_search_tool,
-    pia_search_facets_tool,
+    pia_search_content_tool,
+    pia_search_content_facets_tool,
+    pia_search_titles_tool,
+    pia_search_titles_facets_tool,
 )
 from .prompts.handlers import list_prompts as handler_list_prompts
 from .prompts.handlers import get_prompt as handler_get_prompt
@@ -48,8 +52,10 @@ async def get_prompt(
 async def list_tools() -> List[types.Tool]:
     """List available PIA research tools."""
     return [
-        pia_search_tool,
-        pia_search_facets_tool,
+        pia_search_content_tool,
+        pia_search_content_facets_tool,
+        pia_search_titles_tool,
+        pia_search_titles_facets_tool,
     ]
 
 
@@ -58,10 +64,14 @@ async def call_tool(name: str, arguments: Dict[str, Any]) -> List[types.TextCont
     """Handle tool calls for PIA research functionality."""
     logger.debug(f"Calling tool {name} with arguments {arguments}")
     try:
-        if name == "pia_search":
-            return await handle_pia_search(arguments)
-        elif name == "pia_search_facets":
-            return await handle_pia_search_facets(arguments)
+        if name == "pia_search_content":
+            return await handle_pia_search_content(arguments)
+        elif name == "pia_search_content_facets":
+            return await handle_pia_search_content_facets(arguments)
+        elif name == "pia_search_titles":
+            return await handle_pia_search_titles(arguments)
+        elif name == "pia_search_titles_facets":
+            return await handle_pia_search_titles_facets(arguments)
         else:
             return [types.TextContent(type="text", text=f"Error: Unknown tool {name}")]
     except Exception as e:

src/pia_mcp_server/tools/__init__.py

@@ -6,15 +6,23 @@
 """
 
 from .search_tools import (
-    handle_pia_search,
-    pia_search_tool,
-    handle_pia_search_facets,
-    pia_search_facets_tool,
+    handle_pia_search_content,
+    pia_search_content_tool,
+    handle_pia_search_content_facets,
+    pia_search_content_facets_tool,
+    handle_pia_search_titles,
+    pia_search_titles_tool,
+    handle_pia_search_titles_facets,
+    pia_search_titles_facets_tool,
 )
 
 __all__ = [
-    "handle_pia_search",
-    "pia_search_tool",
-    "handle_pia_search_facets",
-    "pia_search_facets_tool",
+    "handle_pia_search_content",
+    "pia_search_content_tool",
+    "handle_pia_search_content_facets",
+    "pia_search_content_facets_tool",
+    "handle_pia_search_titles",
+    "pia_search_titles_tool",
+    "handle_pia_search_titles_facets",
+    "pia_search_titles_facets_tool",
 ]