lm-deluge 0.0.83__py3-none-any.whl → 0.0.85__py3-none-any.whl

lm_deluge/tool/prefab/web_search.py

@@ -1,5 +1,6 @@
 """Web search prefab tool using Exa API."""
 
+import abc
 import json
 import os
 from typing import Literal
@@ -9,7 +10,42 @@ from aiohttp import ClientSession, ClientTimeout
 from .. import Tool
 
 
-class WebSearchManager:
+class AbstractWebSearchManager(abc.ABC):
+    def __init__(
+        self,
+        search_tool_name: str = "web_search",
+        fetch_tool_name: str = "web_fetch",
+        timeout: int = 30,
+    ):
+        self.search_tool_name = search_tool_name
+        self.fetch_tool_name = fetch_tool_name
+        self.timeout = ClientTimeout(total=timeout)
+        self._tools: list[Tool] | None = None
+
+    @abc.abstractmethod
+    async def _search(self, query: str, limit: int) -> list[dict]:
+        """Search the web and get results with content."""
+        pass
+
+    @abc.abstractmethod
+    async def _fetch(self, url: str) -> str:
+        """Get the contents of a specific URL as markdown."""
+        pass
+
+    def get_tools(self) -> list[Tool]:
+        """Return the web search tools."""
+        if self._tools is not None:
+            return self._tools
+
+        self._tools = [
+            Tool.from_function(self._search),
+            Tool.from_function(self._fetch),
+        ]
+
+        return self._tools
+
+
+class ExaWebSearchManager(AbstractWebSearchManager):
     """
     Simple web search tools using the Exa API.
 
@@ -18,14 +54,16 @@ class WebSearchManager:
     - fetch: Get the contents of a specific URL as markdown
 
     Args:
-        api_key: Exa API key. If not provided, uses EXA_API_KEY env variable.
         search_tool_name: Name for the search tool (default: "web_search")
         fetch_tool_name: Name for the fetch tool (default: "web_fetch")
         timeout: Request timeout in seconds (default: 30)
 
+    Environment variables:
+        EXA_API_KEY: Your Exa API key (required)
+
     Example:
         ```python
-        manager = WebSearchManager()
+        manager = ExaWebSearchManager()
         tools = manager.get_tools()
         ```
     """
@@ -34,30 +72,20 @@ class WebSearchManager:
 
     def __init__(
         self,
-        api_key: str | None = None,
         *,
         search_tool_name: str = "web_search",
         fetch_tool_name: str = "web_fetch",
         timeout: int = 30,
+        max_contents_chars: int = 20_000,
     ):
-        self.search_tool_name = search_tool_name
-        self.fetch_tool_name = fetch_tool_name
-        self.timeout = ClientTimeout(total=timeout)
-
-        if api_key is not None:
-            self.api_key = api_key
-        else:
-            env_key = os.environ.get("EXA_API_KEY")
-            if env_key:
-                self.api_key = env_key
-            else:
-                raise ValueError(
-                    "No API key provided. Set api_key parameter or EXA_API_KEY env variable."
-                )
-
-        self._tools: list[Tool] | None = None
-
-    async def _search(
+        super().__init__(
+            search_tool_name=search_tool_name,
+            fetch_tool_name=fetch_tool_name,
+            timeout=timeout,
+        )
+        self.max_contents_chars = max_contents_chars
+
+    async def _search(  # type: ignore
         self,
         query: str,
         limit: int = 5,
@@ -65,6 +93,9 @@ class WebSearchManager:
     ) -> str:
         """Search the web and return results with content."""
         try:
+            key = os.getenv("EXA_API_KEY")
+            if not key:
+                raise ValueError("EXA_API_KEY environment variable not set")
             data = {
                 "query": query,
                 "numResults": limit,
@@ -74,7 +105,7 @@ class WebSearchManager:
 
             headers = {
                 "Content-Type": "application/json",
-                "x-api-key": self.api_key,
+                "x-api-key": key,
             }
 
             async with ClientSession() as session:
@@ -112,14 +143,19 @@ class WebSearchManager:
     async def _fetch(self, url: str) -> str:
         """Fetch the contents of a URL as markdown."""
         try:
+            key = os.getenv("EXA_API_KEY")
+            if not key:
+                raise ValueError("EXA_API_KEY environment variable not set")
             data = {
                 "urls": [url],
-                "text": True,
+                "text": {
+                    "maxCharacters": self.max_contents_chars,
+                },
             }
 
             headers = {
                 "Content-Type": "application/json",
-                "x-api-key": self.api_key,
+                "x-api-key": key,
             }
 
             async with ClientSession() as session:
@@ -159,48 +195,5 @@ class WebSearchManager:
         except Exception as e:
             return json.dumps({"status": "error", "error": str(e)})
 
-    def get_tools(self) -> list[Tool]:
-        """Return the web search tools."""
-        if self._tools is not None:
-            return self._tools
-
-        self._tools = [
-            Tool(
-                name=self.search_tool_name,
-                description="Search the web and get results with their content.",
-                run=self._search,
-                parameters={
-                    "query": {
-                        "type": "string",
-                        "description": "The search query",
-                    },
-                    "limit": {
-                        "type": "integer",
-                        "description": "Number of results (default: 5, max: 10)",
-                    },
-                    "search_type": {
-                        "type": "string",
-                        "enum": ["auto", "deep"],
-                        "description": "Search type: 'auto' (default) or 'deep' for more thorough search",
-                    },
-                },
-                required=["query"],
-            ),
-            Tool(
-                name=self.fetch_tool_name,
-                description="Fetch the contents of a specific URL as text.",
-                run=self._fetch,
-                parameters={
-                    "url": {
-                        "type": "string",
-                        "description": "The URL to fetch content from",
-                    },
-                },
-                required=["url"],
-            ),
-        ]
-
-        return self._tools
-
 
-__all__ = ["WebSearchManager"]
+__all__ = ["ExaWebSearchManager", "AbstractWebSearchManager"]

{lm_deluge-0.0.83.dist-info → lm_deluge-0.0.85.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lm_deluge
-Version: 0.0.83
+Version: 0.0.85
 Summary: Python utility for using LLM API models.
 Author-email: Benjamin Anderson <ben@trytaylor.ai>
 Requires-Python: >=3.10

{lm_deluge-0.0.83.dist-info → lm_deluge-0.0.85.dist-info}/RECORD

@@ -2,41 +2,34 @@ lm_deluge/__init__.py,sha256=ye2mm-8r9bveEAMWyV13F6APAu2cNzHROU0LOULyPfY,792
 lm_deluge/batches.py,sha256=Km6QM5_7BlF2qEyo4WPlhkaZkpzrLqf50AaveHXQOoY,25127
 lm_deluge/cache.py,sha256=xO2AIYvP3tUpTMKQjwQQYfGRJSRi6e7sMlRhLjsS-u4,4873
 lm_deluge/cli.py,sha256=Ilww5gOw3J5v0NReq_Ra4hhxU4BCIJBl1oTGxJZKedc,12065
-lm_deluge/client.py,sha256=SD829zTZf1CjA_Yv0X6xuhI1QF4JJYZaK8oBaBIENs4,45715
+lm_deluge/client.py,sha256=HZ_frrki94g65kMuy1RjRK_oBouCXoYvXLadTPXsQ-U,49216
 lm_deluge/config.py,sha256=C-_rVwAFL5sivLfKSkaa2ANMqqxKbyDCW86KfQB_Lck,1357
 lm_deluge/embed.py,sha256=CO-TOlC5kOTAM8lcnicoG4u4K664vCBwHF1vHa-nAGg,13382
 lm_deluge/errors.py,sha256=oHjt7YnxWbh-eXMScIzov4NvpJMo0-2r5J6Wh5DQ1tk,209
 lm_deluge/file.py,sha256=PTmlJQ-IaYcYUFun9V0bJ1NPVP84edJrR0hvCMWFylY,19697
 lm_deluge/image.py,sha256=5AMXmn2x47yXeYNfMSMAOWcnlrOxxOel-4L8QCJwU70,8928
 lm_deluge/mock_openai.py,sha256=-u4kxSzwoxDt_2fLh5LaiqETnu0Jg_VDL7TWAAYHGNw,21762
-lm_deluge/prompt.py,sha256=vgI4cvk5gl6-x4uLm4X67eSANyrolJbpF5HXTz88550,70365
+lm_deluge/prompt.py,sha256=aOHNDGQv9UEpyxTLFkMvnpAOef_Yh1XDmgB__vQeH6Y,71763
 lm_deluge/request_context.py,sha256=CX15dT4Jxz77C-w5EKNyJCfYEa69wNKHbfNi47iG8W4,2771
 lm_deluge/rerank.py,sha256=-NBAJdHz9OB-SWWJnHzkFmeVO4wR6lFV7Vw-SxG7aVo,11457
 lm_deluge/tracker.py,sha256=B53KIsrK10L9L73cYbVB2pNSC0-FdvJGpIfw735CvaA,14808
 lm_deluge/usage.py,sha256=xz9tAw2hqaJvv9aAVhnQ6N1Arn7fS8Shb28VwCW26wI,5136
 lm_deluge/warnings.py,sha256=3_lWpR20b5WEfchqvbWVIc-vo8afU0Jg6S9FkbA5pZQ,2479
 lm_deluge/api_requests/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-lm_deluge/api_requests/anthropic.py,sha256=KaIJAspFiTESuYy0uqzDTXXVhGN9raIjJBWF-YG2BOo,11594
+lm_deluge/api_requests/anthropic.py,sha256=b9mr8oXJIX8AOYAOlOc0gTtmf2Ob_eVjWnuq4YvazUQ,11801
 lm_deluge/api_requests/base.py,sha256=05j5nrZhgNon2YRFXT_L-yVXKlvdodwOJan6Z6WpSp8,10911
 lm_deluge/api_requests/bedrock.py,sha256=mY1xTvgfCLyqLlfFFmu_baKgkVq1Df1_MJXeN_G1jWQ,15597
 lm_deluge/api_requests/chat_reasoning.py,sha256=sJvstvKFqsSBUjYcwxzGt2_FH4cEp3Z6gKcBPyPjGwk,236
 lm_deluge/api_requests/common.py,sha256=BZ3vRO5TB669_UsNKugkkuFSzoLHOYJIKt4nV4sf4vc,422
-lm_deluge/api_requests/gemini.py,sha256=q1Muf1LnWKxJA1x5braXSvAdtmxQrOkAaYYVh1K_EEM,11115
+lm_deluge/api_requests/gemini.py,sha256=w42YdpBDcgPOCUwZT2voJbKLFV5GTT5skCX93RyYPLc,12388
 lm_deluge/api_requests/mistral.py,sha256=8JZP2CDf1XZfaPcTk0WS4q-VfYYj58ptpoH8LD3MQG4,4528
-lm_deluge/api_requests/openai.py,sha256=dhzFw-FCZ52ZsZ0dVwOdP0rX81YvyIDOK945wFfAOco,28898
+lm_deluge/api_requests/openai.py,sha256=KjPu5z5rkinN0DHQs-_GczJZzHimLd7vADHKPYMeUzI,28891
 lm_deluge/api_requests/response.py,sha256=vG194gAH5p7ulpNy4qy5Pryfb1p3ZV21-YGoj__ru3E,7436
 lm_deluge/api_requests/deprecated/bedrock.py,sha256=WrcIShCoO8JCUSlFOCHxg6KQCNTZfw3TpYTvSpYk4mA,11320
 lm_deluge/api_requests/deprecated/cohere.py,sha256=KgDScD6_bWhAzOY5BHZQKSA3kurt4KGENqC4wLsGmcU,5142
 lm_deluge/api_requests/deprecated/deepseek.py,sha256=FEApI93VAWDwuaqTooIyKMgONYqRhdUmiAPBRme-IYs,4582
 lm_deluge/api_requests/deprecated/mistral.py,sha256=pOfOZUM4U35I3Plch84SnAFpDAzouHcSNNMtgxRvjy4,4709
 lm_deluge/api_requests/deprecated/vertex.py,sha256=ygXz2RjdXErPCSBbiHLEWbf5_sSTIi31WoX0UaoYzRI,15275
-lm_deluge/built_in_tools/base.py,sha256=FLYdKVAqlffA6WOu4j8wQVRd0iHMsyBW_T3vfl--aXo,276
-lm_deluge/built_in_tools/openai.py,sha256=aLuJdXbANvXVIU38Vo2zsir7zlwWgX0d8oDPT7Ql64A,721
-lm_deluge/built_in_tools/anthropic/__init__.py,sha256=Dxm8MJTUwMhyXT_78uIdRe8dJsLHZcHThJ9UEurTw18,4526
-lm_deluge/built_in_tools/anthropic/bash.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lm_deluge/built_in_tools/anthropic/computer_use.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lm_deluge/built_in_tools/anthropic/editor.py,sha256=DyC_DrHVTm1khU9QDL39vBuhu4tO5mS5H7xMRIT0Ng4,23327
-lm_deluge/llm_tools/__init__.py,sha256=c_S5wiBkSZcObPLYaMrsykKAJ302YF5QIoNUOQyDfsE,527
 lm_deluge/models/__init__.py,sha256=_c-gxqAaNO4xy4dtsqIwG1odpwcCa2J02_YQnuXYtc0,4669
 lm_deluge/models/anthropic.py,sha256=X92EYIapos-8LXnIYiypPJcFhI0tqmXja_w8e9H4CF8,6781
 lm_deluge/models/arcee.py,sha256=4OI8eA8RoA-zYww4fWwhVZDFWB2Kd4-KQTTPl9r3Ay4,465
@@ -45,14 +38,14 @@ lm_deluge/models/cerebras.py,sha256=u2FMXJF6xMr0euDRKLKMo_NVTOcvSrrEpehbHr8sSeE,
 lm_deluge/models/cohere.py,sha256=iXjYtM6jy_YL73Op8OfNsrMNopwae9y-Sw-4vF9cEBw,3406
 lm_deluge/models/deepseek.py,sha256=b5t_ep6fE-2cKD2mmImBaLcJUbYrfizYnjG96sfKNTk,2072
 lm_deluge/models/fireworks.py,sha256=yvt2Ggzye4aUqCqY74ta67Vu7FrQaLFjdFtN4P7D-dc,638
-lm_deluge/models/google.py,sha256=fARfBMHDwhPJ48SGcYg3sHQDQ0Mm0yPQ-9s6iVYne8M,6011
+lm_deluge/models/google.py,sha256=IDePlNOvF0lvpv3UhkUD8g30TUJqoaJHQGzTglyGg80,6560
 lm_deluge/models/grok.py,sha256=TDzr8yfTaHbdJhwMA-Du6L-efaKFJhjTQViuVElCCHI,2566
 lm_deluge/models/groq.py,sha256=Mi5WE1xOBGoZlymD0UN6kzhH_NOmfJYU4N2l-TO0Z8Q,2552
 lm_deluge/models/kimi.py,sha256=B_ZL4_0q6hS1VVskBWlBR569nNSjC8RgA2lj1eCjRRE,1183
 lm_deluge/models/meta.py,sha256=BBgnscL1gMcIdPbRqrlDl_q9YAYGSrkw9JkAIabXtLs,1883
 lm_deluge/models/minimax.py,sha256=rwW9gNotAYfDVtMlqmSYegN6GoZM_9DSNNZU2yPOmaU,275
 lm_deluge/models/mistral.py,sha256=x67o5gckBGmPcIGdVbS26XZAYFKBYM4tsxEAahGp8bk,4323
-lm_deluge/models/openai.py,sha256=t6fcXo0YXgPQ6YiftZJP8gPw8FOBqoVapSavMVmtaOw,12411
+lm_deluge/models/openai.py,sha256=UDmPqvMaBjqky2Z6yNV4bG4LqxUZmbWEJJO4pq0Mqzc,13329
 lm_deluge/models/openrouter.py,sha256=AHQTvnXM96_70XR0eqb3Wu2rAj2SHBZCWQeEKJ307LY,2816
 lm_deluge/models/together.py,sha256=wrGs4wO65on-dSlU9AARAA-rc4GDuWkidPjRQ7GScNg,4749
 lm_deluge/models/zai.py,sha256=BIde8TwjvmkfEi-6bSSBSFIh7KVnlJ7_aNdlqNZRGGI,16
@@ -63,7 +56,21 @@ lm_deluge/pipelines/locate.py,sha256=lYNbKTmy9dTvj0lEQkOQ7yrxyqsgYzjD0C_byJKI_4w
 lm_deluge/pipelines/ocr.py,sha256=7fDlvs6uUOvbxMasvGGNJx5Fj6biM6z3lijKZaGN26k,23
 lm_deluge/pipelines/score.py,sha256=hkLMroJMfQ92HPlTBNOHrDRtvdYUBWK0MBlhOfvFTMk,2582
 lm_deluge/pipelines/translate.py,sha256=v_OvBQA2RB-QcWf0aopKHpYc2PDmckxzJGSmSuUX3Sw,1461
-lm_deluge/tool/__init__.py,sha256=uEqutw31_XtzydvK1LB_yNnFOyQg5XPR7V8STxs6B5I,39889
+lm_deluge/tool/__init__.py,sha256=_GD2RAK0wYnKsAPdrM2w8X4xKtugBEm-eSZTboGRL9s,40260
+lm_deluge/tool/builtin/base.py,sha256=FLYdKVAqlffA6WOu4j8wQVRd0iHMsyBW_T3vfl--aXo,276
+lm_deluge/tool/builtin/gemini.py,sha256=uKrzzEZ0RO5EHddYYFvRKoMk7O6YOSWFDojhzbpQSfs,1724
+lm_deluge/tool/builtin/openai.py,sha256=a6Y4TX_iqghDTnbL-HeFeVGq1nRdBosYJcLZzfCTy9g,2499
+lm_deluge/tool/builtin/anthropic/__init__.py,sha256=U-5VdcRS_ITjTBCoFoeABZOoG4nfBTgRvhT5eo1UY1g,10675
+lm_deluge/tool/builtin/anthropic/bash.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lm_deluge/tool/builtin/anthropic/computer_use.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lm_deluge/tool/builtin/anthropic/editor.py,sha256=DyC_DrHVTm1khU9QDL39vBuhu4tO5mS5H7xMRIT0Ng4,23327
+lm_deluge/tool/cua/__init__.py,sha256=HDXXwCqPRNm_-Wh6JCsIT89CYwioHdRSZcu2WQWYaCo,4756
+lm_deluge/tool/cua/actions.py,sha256=ZeQheCJ9F5nHLu3teL14_qr8mr4UuaR78EhgrvYw_G8,3191
+lm_deluge/tool/cua/base.py,sha256=DwOTHp8AGu5-ZRzg_6RdOu9Bh4cgAbK5EMPEusJlDrI,799
+lm_deluge/tool/cua/batch.py,sha256=ddYHbg72y4e62G6-B6BMM1SNq5zI9WYsnC6jzKNJiOU,7312
+lm_deluge/tool/cua/converters.py,sha256=O1wO4u_78ZZNBrXuyjh4vZreYckhh-e3J304C-d6uOA,13741
+lm_deluge/tool/cua/kernel.py,sha256=8klzaCTBB-4yYvZTzroi-t5ABNX81jAjc5uCCKG3Kvc,25654
+lm_deluge/tool/cua/trycua.py,sha256=SvdA5DkAVweegQ3r_SI_AbKKX242mFOxsBDf5CNiRkY,36954
 lm_deluge/tool/prefab/__init__.py,sha256=JXwyAkBvKqOYC8us80ClxZ31vKECAJ8FsrZKiEK_PT8,909
 lm_deluge/tool/prefab/batch_tool.py,sha256=VTMoW7ImQYmeDHrbB4BKSw3WeHXhsjp9XqFHz4ytjcc,5584
 lm_deluge/tool/prefab/docs.py,sha256=5X3EVUH4ws72og4TxDpgIjN3clroqUsR1Wx8fIuQWd0,41566
@@ -76,7 +83,7 @@ lm_deluge/tool/prefab/sheets.py,sha256=RhH4PgRI4E6WYKfJpScflT7HtAULvp88ZA94NmJyE
 lm_deluge/tool/prefab/subagents.py,sha256=srJ7On7YR0Y8WuNvf5TJl_7IUfEtG3zlxZeLgmn_-NI,8484
 lm_deluge/tool/prefab/todos.py,sha256=mrtv68uRc-grc0xKP6xKpfPSA2yXYU7FLNy6fDzEbG8,14902
 lm_deluge/tool/prefab/tool_search.py,sha256=EJL5R5BgnO6owspk7F01Yzr8C9q9oJqmfMrWBjLE_bA,6151
-lm_deluge/tool/prefab/web_search.py,sha256=oc1sQloAdybKlOJBPgbwPuBew0degyj9hqTK1dV9Woc,6615
+lm_deluge/tool/prefab/web_search.py,sha256=i_FFBB2TAZwgN4-_9a6cD10Z9bgFGk4qNUbnOI3EnJA,6200
 lm_deluge/tool/prefab/otc/__init__.py,sha256=33AcwAt9ycECxonnvkcyh13w9Sr2Cbs6OOlBjzBvl54,6373
 lm_deluge/tool/prefab/otc/executor.py,sha256=4IROA_0un3HaV4GK7r4vQiVxJvcoHkqVii-4asGH-Cw,10508
 lm_deluge/tool/prefab/otc/parse.py,sha256=lSAtez-pBFcJVQMW2evRvV9KlduRtPevzGCEB1fmUMo,4517
@@ -87,8 +94,8 @@ lm_deluge/util/schema.py,sha256=q6uwhA4s1lM2dHT1Kwc46E7OY1VecMOtTEI0PTFn6tA,1320
 lm_deluge/util/spatial.py,sha256=BsF_UKhE-x0xBirc-bV1xSKZRTUhsOBdGqsMKme20C8,4099
 lm_deluge/util/validation.py,sha256=hz5dDb3ebvZrZhnaWxOxbNSVMI6nmaOODBkk0htAUhs,1575
 lm_deluge/util/xml.py,sha256=Ft4zajoYBJR3HHCt2oHwGfymGLdvp_gegVmJ-Wqk4Ck,10547
-lm_deluge-0.0.83.dist-info/licenses/LICENSE,sha256=uNNXGXPCw2TC7CUs7SEBkA-Mz6QBQFWUUEWDMgEs1dU,1058
-lm_deluge-0.0.83.dist-info/METADATA,sha256=TRA97JDBxqqO8DmVIZZ8p7FynREzdunRpCnaVK7A_oQ,13595
-lm_deluge-0.0.83.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lm_deluge-0.0.83.dist-info/top_level.txt,sha256=hqU-TJX93yBwpgkDtYcXyLr3t7TLSCCZ_reytJjwBaE,10
-lm_deluge-0.0.83.dist-info/RECORD,,
+lm_deluge-0.0.85.dist-info/licenses/LICENSE,sha256=uNNXGXPCw2TC7CUs7SEBkA-Mz6QBQFWUUEWDMgEs1dU,1058
+lm_deluge-0.0.85.dist-info/METADATA,sha256=MgU3mDxSvV_NBnmoacgJYXQZ-dAUdSI0JZHgXz7-BxY,13595
+lm_deluge-0.0.85.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lm_deluge-0.0.85.dist-info/top_level.txt,sha256=hqU-TJX93yBwpgkDtYcXyLr3t7TLSCCZ_reytJjwBaE,10
+lm_deluge-0.0.85.dist-info/RECORD,,

lm_deluge/built_in_tools/anthropic/__init__.py

@@ -1,128 +0,0 @@
-from typing import Literal
-
-# from lm_deluge.prompt import ToolCall
-
-ToolVersion = Literal["2024-10-22", "2025-01-24", "2025-04-29"]
-ToolType = Literal["bash", "computer", "editor"]
-
-
-def model_to_version(model: str) -> ToolVersion:
-    if "opus" not in model and "sonnet" not in model:
-        raise ValueError("cannot use computer tools with incompatible model")
-    if "claude-4" in model:
-        return "2025-04-29"
-    elif "3.7" in model:
-        return "2025-01-24"
-    elif "3.6" in model:
-        return "2024-10-22"
-    else:
-        raise ValueError("unsupported model for anthropic CUA")
-
-
-def get_anthropic_cu_tools(
-    model: str,
-    display_width: int = 1024,
-    display_height: int = 768,
-    exclude_tools: list[ToolType] | None = None,
-):
-    version = model_to_version(model)
-    if version == "2024-10-22":
-        result = [
-            {
-                "name": "computer",
-                "type": "computer_20241022",
-                "display_width_px": display_width,
-                "display_height_px": display_height,
-                "display_number": None,
-            },
-            {"name": "str_replace_editor", "type": "text_editor_20241022"},
-            {"name": "bash", "type": "bash_20241022"},
-        ]
-    elif version == "2025-01-24":
-        result = [
-            {
-                "name": "computer",
-                "type": "computer_20250124",
-                "display_width_px": display_width,
-                "display_height_px": display_height,
-                "display_number": None,
-            },
-            {"name": "str_replace_editor", "type": "text_editor_20250124"},
-            {"type": "bash_20250124", "name": "bash"},
-        ]
-    elif version == "2025-04-29":
-        result = [
-            {
-                "name": "computer",
-                "type": "computer_20250124",
-                "display_width_px": display_width,
-                "display_height_px": display_height,
-                "display_number": None,
-            },
-            {"name": "str_replace_based_edit_tool", "type": "text_editor_20250429"},
-            {
-                "name": "bash",
-                "type": "bash_20250124",
-            },
-        ]
-    else:
-        raise ValueError("invalid tool version")
-
-    if exclude_tools is None:
-        return result
-    if "bash" in exclude_tools:
-        result = [x for x in result if x["name"] != "bash"]
-    if "editor" in exclude_tools:
-        result = [x for x in result if "edit" not in x["name"]]
-    if "computer" in exclude_tools:
-        result = [x for x in result if "computer" not in x["name"]]
-    return result
-
-
-def bash_tool(model: str = "claude-4-sonnet"):
-    # Claude Sonnet 3.5 requires the computer-use-2024-10-22 beta header when using the bash tool.
-    # The bash tool is generally available in Claude 4 and Sonnet 3.7.
-    if "claude-4" in model:
-        return {"type": "text_editor_20250429", "name": "str_replace_based_edit_tool"}
-    elif "3.7" in model:
-        return {"type": "text_editor_20250124", "name": "str_replace_editor"}
-    else:
-        return {"type": "text_editor_20241022", "name": "str_replace_editor"}
-
-
-def text_editor_tool(model: str = "claude-4-sonnet"):
-    if "claude-4" in model:
-        return {"type": "bash_20250124", "name": "bash"}
-    elif "3.7" in model:
-        return {"type": "bash_20250124", "name": "bash"}
-    else:
-        return {"type": "bash_20241022", "name": "bash"}
-
-
-def web_search_tool(max_uses: int = 5):
-    res = {
-        "type": "web_search_20250305",
-        "name": "web_search",
-        # Optional: Limit the number of searches per request
-        "max_uses": max_uses,
-        # You can use either allowed_domains or blocked_domains, but not both in the same request.
-        # Optional: Only include results from these domains
-        # "allowed_domains": ["example.com", "trusteddomain.org"],
-        #  Optional: Never include results from these domains
-        # "blocked_domains": ["untrustedsource.com"],
-        # Optional: Localize search results
-        # "user_location": {
-        #   "type": "approximate",
-        #   "city": "San Francisco",
-        #   "region": "California",
-        #   "country": "US",
-        #   "timezone": "America/Los_Angeles"
-        # }
-    }
-    return res
-
-
-def code_execution_tool():
-    # The code execution tool is currently in beta.
-    # This feature requires the beta header: "anthropic-beta": "code-execution-2025-05-22"
-    return {"type": "code_execution_20250522", "name": "code_execution"}

lm_deluge/built_in_tools/openai.py

@@ -1,28 +0,0 @@
-def image_generation_openai():
-    # TODO: handle result properly
-    return {"type": "image_generation"}
-
-
-def code_interpreter_openai(container: dict | None = None):
-    if container is None:
-        container = {"type": "auto"}
-    return {"type": "code_interpreter", "container": container}
-
-
-def local_shell_openai():
-    return {"type": "local_shell"}
-
-
-def web_search_openai():
-    return {"type": "web_search_preview"}
-
-
-def computer_use_openai(
-    display_width: int = 1024, display_height: int = 768, environment: str = "browser"
-):
-    return {
-        "type": "computer_use_preview",
-        "display_width": display_width,
-        "display_height": display_height,
-        "environment": environment,
-    }

lm_deluge/llm_tools/__init__.py

@@ -1,25 +0,0 @@
-# Backward compatibility - re-export from new locations
-# Pipelines (workflow functions)
-from ..pipelines import extract, extract_async, score_llm, translate, translate_async
-
-# Prefab tools (Tool managers)
-from ..tool.prefab import (
-    SubAgentManager,
-    TodoItem,
-    TodoManager,
-    TodoPriority,
-    TodoStatus,
-)
-
-__all__ = [
-    "extract",
-    "extract_async",
-    "TodoItem",
-    "TodoManager",
-    "TodoPriority",
-    "TodoStatus",
-    "translate",
-    "translate_async",
-    "score_llm",
-    "SubAgentManager",
-]