o xi!@sddlZddlZddlZddlmZmZmZmZddlm Z ddl m Z ddl Z GdddZ Gdd d e ZGd d d e ZGd d d e ZGddde ZGddde ZGddde ZGddde Zddee deee ffddZdS)N)DictAnyListOptional)UserInterfaceAdapter)ui_statec@s<eZdZUdZeed<eed<deeefdefddZdS) BaseToolzCommon interface for all tools.name descriptionargsreturncCstN)NotImplementedError)selfr r5/var/www/apps/myagent/mysuperagent/agent_src/tools.pyrunsz BaseTool.runN) __name__ __module__ __qualname____doc__str__annotations__rrrrrrrr s r c@s`eZdZdZdZddeeDeee e dBZ de de fdd Z d ee efde fd d Zd S)CalcToolcalczEvaluate a simple arithmetic expression. Supported operators: + - * / ** and parentheses. Example args: {"expression": "2 * (3 + 4)"}cCs"i|] }|ds|tt|qS)__) startswithgetattrmath).0krrr s   zCalcTool.)absroundintfloatexprr cCs&td|s tdt|dii|jS)Nz[0-9\s\.\+\-\*\/\(\)\^\%]+z&Expression contains illegal characters __builtins__)re fullmatch ValueErroreval_allowed_names)rr'rrr _safe_eval#s zCalcTool._safe_evalr c CsX|d}|s dSz ||}d|WSty+}z d|WYd}~Sd}~ww)N expressionu"❌ Missing 'expression' argument.u ✅ Result = u❌ Calculator error: )getr. Exception)rr r'resulterrrr(s   z CalcTool.runN)rrrr r dirrr#r$r%r&r-rr.rrrrrrrrs rc@.eZdZdZdZdeeefdefddZdS) SearchToolsearchzpSearch the web for a short query and return the top result snippet. Example args: {"query": "capital of Norway"}r r cCs|d}|s dSd|dS)Nqueryu❌ Missing 'query' argument.u🔎 (stub) Top result for 'z&': )r0)rr r8rrrr<s  zSearchTool.runN rrrr r rrrrrrrrr65 r6c@r5) ReadFileTool read_fileuYRead a UTF‑8 text file from the local filesystem. Example args: {"path": "./notes.txt"}r r c Cs|d}|s dStj|}tjt}||sdStj|s)d|Sz(t|ddd}|d}d |d |WdWS1sJwYWdSt yh}z d |WYd}~Sd}~ww) Npath❌ Missing 'path' argument.u?❌ Access to paths outside the current directory is forbidden.❌ File not found: rutf-8encodingiu📄 Content of z (truncated): u❌ Could not read file: ) r0osr=abspathgetcwdrisfileopenreadr1)rr r= safe_pathcwdfcontentr3rrrrMs$      (zReadFileTool.runNr9rrrrr;Fr:r;c@r5) WriteFileTool write_fileuiCreate or overwrite a UTF‑8 text file. Example args: {"path": "./data.txt", "content": "Hello, world!"}r r c Cs|d}|dd}|sdStj|}tjt}||s$dSz't|ddd }||Wdn1su<❌ Refusing to write outside the current working directory.wrArBu ✅ Wrote z characters to u❌ Failed to write file: ) r0rDr=rErFrrHwritelenr1)rr r=rMrJrKrLr3rrrrgs"     zWriteFileTool.runNr9rrrrrN`r:rNc@r5)DeleteFileTool delete_filezLDelete a file from the local filesystem. Example args: {"path": "./old.txt"}r r c Cs|d}|s dStj|}tjt}||sdStj|s)d|Sz t|d|WStyK}z d|WYd}~Sd}~ww)Nr=r>u=❌ Refusing to delete outside the current working directory.r?u✅ Deleted file u❌ Could not delete file: ) r0rDr=rErFrrGremover1)rr r=rJrKr3rrrrs       zDeleteFileTool.runNr9rrrrrTyr:rTc@r5) ListFilesTool list_fileszrList files in a directory (optionally filter by extension). Example args: {"directory": "./", "extension": ".txt"}r r c s|dd}|dtj|}tjt}||s dStj|s+d|Sz$t|}r<fdd|D}|sHd|d d WSd d |WSt yf}z d |WYd}~Sd}~ww)N directory. extensionu;❌ Refusing to list outside the current working directory.u❌ Not a directory: csg|] }|r|qSr)endswith)r r3extrr sz%ListFilesTool.run..u🔍 No files found in z with extension 'z'.u 📂 Files:  u❌ Error while listing files: ) r0rDr=rErFrisdirlistdirjoinr1)rr rYsafe_dirrKentriesr3rr]rrs&       zListFilesTool.runNr9rrrrrWr:rWc@sXeZdZdZdZdZd deefddZd de e e fd ee e e fd e fd d Z dS) AskUserToolzPrompt the real user; supports both sync (CLI) and modal web via token. When used with a web adapter, returns {'awaiting': token, 'prompt': text}. ask_userzuAsk the person running the agent a question and return their answer. Example args: {"question": "What is your name?"}N ui_adaptercCs ||_dSrrh)rrhrrr__init__s zAskUserTool.__init__r contextr c Cs|dd}tt}|jrEt|jddrEz|j|}|d|d|WStyD}z| d|d|WYd}~Sd}~wwt }|d ||||d S) NquestionzPlease provide input: supports_syncFzAskUserTool sync answered: %su🗣️ User answered: z%Error during AskUserTool sync ask: %su❌ Error while asking user: z+AskUserTool created token %s for prompt: %s)awaitingprompt) r0logging getLoggerrrhrask_syncinfor1 exceptionr new_token)rr rkrlloggeransr3tokenrrrrs       zAskUserTool.runr) rrrrr r rrrjrrrrrrrrrfs0rfrhr cCs>tjttjttjttjttjttjttjt|iS)z Returns a dictionary mapping tool names to instantiated tool objects. Add any new tool class to this dictionary to make it visible to the agent. )rr r6r;rNrTrWrfrirrrbuild_tool_registrys ryr)rDrr)typingrrrr interfacesrrPrrpr rr6r;rNrTrWrfrryrrrrs     $"