o +i+@sdZddlmZddlZddlZddlmZmZddlm Z m Z m Z e r*ddlm Z ddl mZddlmZmZdd lmZdddZdddZdddZGdddeZdgZdS)zFile search middleware for Anthropic text editor and memory tools. This module provides Glob and Grep search tools that operate on files stored in state or filesystem. ) annotationsN)Path PurePosixPath) TYPE_CHECKINGLiteralcast)Any)AgentMiddleware) ToolRuntimetool)AnthropicToolsStatepatternstrreturnlist[str] | NonecsLd|vr d|vr dSgd fdd z|WSty%YdSw) z?Expand brace patterns like ``*.{py,pyi}`` into a list of globs.}{NcurrentrrNonecs|d}|dkr|dS|d|}|dkrt|d|}||dd}||d|}|s6t|dD] }|||q;dS)Nrr,)findappend ValueErrorsplit)rstartendprefixsuffixinneroption_expandexpandedg/var/www/html/psymed-ai/venv/lib/python3.10/site-packages/langchain_anthropic/middleware/file_search.pyr#s    z)_expand_include_patterns.._expand)rrrr)rr r%r"r&_expand_include_patternss  r(boolcsnsdStfdddDrdSt}|durdSz|D] }tt|qWdStjy6YdSw)z/Validate glob pattern used for include filters.Fc3s|]}|vVqdSNr%).0charr'r%r& >sz,_is_valid_include_pattern..)  NT)anyr(recompilefnmatch translateerror)r r$ candidater%r'r&_is_valid_include_pattern9sr8basenamecs&t|}|sdStfdd|DS)z8Return True if the basename matches the include pattern.Fc3s|] }t|VqdSr*)r4)r+r7r9r%r&r-Tsz)_match_include_pattern..)r(r1)r9r r$r%r:r&_match_include_patternNsr;c@sBeZdZdZeZddddd ZdddZdddZdddZ dS)StateFileSearchMiddlewareaProvides Glob and Grep search over state-based files. This middleware adds two tools that search through virtual files in state: - Glob: Fast file pattern matching by file path - Grep: Fast content search using regular expressions Example: ```python from langchain.agents import create_agent from langchain.agents.middleware import ( StateTextEditorToolMiddleware, StateFileSearchMiddleware, ) agent = create_agent( model=model, tools=[], middleware=[ StateTextEditorToolMiddleware(), StateFileSearchMiddleware(), ], ) ``` text_editor_files) state_keyr>rrrcsP|_t ddfdd }t  ddfdd }|_|_||g_d S)zInitialize the search middleware. Args: state_key: State key to search (default: "text_editor_files"). Use "memory_files" to search memory tool files. /runtime&ToolRuntime[None, AnthropicToolsState]r rpathrcs|||jS)aFast file pattern matching tool that works with any codebase size. Supports glob patterns like **/*.js or src/**/*.ts. Returns matching file paths sorted by modification time. Use this tool when you need to find files by name patterns. Args: pattern: The glob pattern to match files against. path: The directory to search in. If not specified, searches from root. Returns: Newline-separated list of matching file paths, sorted by modification time (most recently modified first). Returns "No files found" if no matches. )_handle_glob_searchstate)r@r rBselfr%r& glob_searchsz7StateFileSearchMiddleware.__init__..glob_searchNfiles_with_matchesinclude str | None output_mode1Literal['files_with_matches', 'content', 'count']cs|||||jS)aFast content search tool that works with any codebase size. Searches file contents using regular expressions. Supports full regex syntax and filters files by pattern with the include parameter. Args: pattern: The regular expression pattern to search for in file contents. path: The directory to search in. If not specified, searches from root. include: File pattern to filter (e.g., "*.js", "*.{ts,tsx}"). output_mode: Output format: - "files_with_matches": Only file paths containing matches (default) - "content": Matching lines with file:line:content format - "count": Count of matches per file Returns: Search results formatted according to output_mode. Returns "No matches found" if no results. )_handle_grep_searchrD)r@r rBrIrKrEr%r& grep_searchs z7StateFileSearchMiddleware.__init__..grep_search)r?)r@rAr rrBrrr)r?NrH) r@rAr rrBrrIrJrKrLrr)r>r rGrNtools)rFr>rGrNr%rEr&__init__ss z"StateFileSearchMiddleware.__init__r rBrDr c Cs|dr|nd|}td||ji}g}|D]U\}}||rp|dkr/|dd} n||kr9t|j} n||drK|t|dd} nqt|  |} | se|dret|  |dd} | rp| ||dfq|sudS|j d d d d d d|D} d | S)aHandle glob search operation. Args: pattern: The glob pattern to match files against. path: The directory to search in. state: The current agent state. Returns: Newline-separated list of matching file paths, sorted by modification time (most recently modified first). Returns "No files found" if no matches. r?dict[str, Any]rNz**/ modified_atzNo files foundcSs|dS)Nrr%)xr%r%r&sz?StateFileSearchMiddleware._handle_glob_search..T)keyreversecSsg|]\}}|qSr%r%)r+rB_r%r%r& szAStateFileSearchMiddleware._handle_glob_search..r/) startswithrgetr>itemsrnamelenrmatchrsortjoin) rFr rBrD base_pathfilesmatches file_path file_datarelativeis_match file_pathsr%r%r&rCs.   z-StateFileSearchMiddleware._handle_glob_searchrIrJrKc Cs|dr|nd|}zt|}Wntjy*}z d|WYd}~Sd}~ww|r3t|s3dStd||ji} i} | D]9\} } | |sMqC|rZt | j } t | |sZqCt | ddD]\}}| |r{| | vrrg| | <| | ||fqaqC| sdS|| |S) aHandle grep search operation. Args: pattern: The regular expression pattern to search for in file contents. path: The directory to search in. include: File pattern to filter (e.g., "*.js", "*.{ts,tsx}"). output_mode: Output format. state: The current agent state. Returns: Search results formatted according to output_mode. Returns "No matches found" if no results. r?zInvalid regex pattern: NzInvalid include patternrQcontentrzNo matches found)rZr2r3r6r8rr[r>r\rr]r; enumeratesearchr_format_grep_results)rFr rBrIrKrDrbregexercresultsrerfr9line_numliner%r%r&rMs8      z-StateFileSearchMiddleware._handle_grep_searchrp dict[str, list[tuple[int, str]]]c Cs|dkr dt|S|dkr7g}t|D]}||D]\}}||d|d|qqd|S|dkr[g}t|D]}t||}||d|qCd|Sdt|S)z)Format grep results based on output mode.rHr/rj:count)rasortedkeysrr^)rFrprKlinesrerqrrrur%r%r&rm3s    z.StateFileSearchMiddleware._format_grep_resultsN)r>rrr)r rrBrrDr rr) r rrBrrIrJrKrrDr rr)rprsrKrrr) __name__ __module__ __qualname____doc__r state_schemarPrCrMrmr%r%r%r&r<Ws I ;s$      "  |