lm-deluge 0.0.84__tar.gz → 0.0.86__tar.gz

{lm_deluge-0.0.84/src/lm_deluge.egg-info → lm_deluge-0.0.86}/PKG-INFO

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

{lm_deluge-0.0.84 → lm_deluge-0.0.86}/pyproject.toml

@@ -3,7 +3,7 @@ requires = ["setuptools", "wheel"]
 
 [project]
 name = "lm_deluge"
-version = "0.0.84"
+version = "0.0.86"
 authors = [{ name = "Benjamin Anderson", email = "ben@trytaylor.ai" }]
 description = "Python utility for using LLM API models."
 readme = "README.md"

{lm_deluge-0.0.84 → lm_deluge-0.0.86}/src/lm_deluge/client.py

@@ -918,7 +918,7 @@ class _LLMClient(BaseModel):
                 if not isinstance(result, (str, dict, list)):
                     result = str(result)
 
-                conversation.with_tool_result(call.id, result)  # type: ignore
+                conversation = conversation.with_tool_result(call.id, result)  # type: ignore
 
         if response is None:
             raise RuntimeError("model did not return a response")

{lm_deluge-0.0.84 → lm_deluge-0.0.86}/src/lm_deluge/models/openai.py

@@ -149,6 +149,34 @@ OPENAI_MODELS = {
         "output_cost": 6.0,
         "reasoning_model": True,
     },
+    "o4-mini-deep-research": {
+        "id": "o4-mini-deep-research",
+        "name": "o4-mini-deep-research",
+        "api_base": "https://api.openai.com/v1",
+        "api_key_env_var": "OPENAI_API_KEY",
+        "supports_json": True,
+        "supports_logprobs": False,
+        "supports_responses": True,
+        "api_spec": "openai",
+        "input_cost": 2,
+        "cached_input_cost": 0.5,
+        "output_cost": 8.0,
+        "reasoning_model": True,
+    },
+    "o3-deep-research": {
+        "id": "o3-deep-research",
+        "name": "o3-deep-research",
+        "api_base": "https://api.openai.com/v1",
+        "api_key_env_var": "OPENAI_API_KEY",
+        "supports_json": True,
+        "supports_logprobs": False,
+        "supports_responses": True,
+        "api_spec": "openai",
+        "input_cost": 10,
+        "cached_input_cost": 2.50,
+        "output_cost": 40.0,
+        "reasoning_model": True,
+    },
     "o3": {
         "id": "o3",
         "name": "o3-2025-04-16",

lm_deluge-0.0.86/src/lm_deluge/tool/builtin/openai.py

@@ -0,0 +1,74 @@
+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(
+    preview: bool = False,
+    user_location: dict | None = None,
+    allowed_domains: list[str] | None = None,
+    search_context_size: str | None = None,
+):
+    """OpenAI's built-in web search tool for the Responses API.
+
+    Args:
+        preview: If True, use web_search_preview. If False (default), use
+            the GA web_search tool.
+        user_location: Optional approximate user location to refine search results.
+            Should be a dict with "type": "approximate" and an "approximate" key
+            containing any of: country (ISO code), city, region, timezone.
+            Note: Not supported for deep research models.
+        allowed_domains: Optional list of domains to restrict search results to.
+            Up to 100 URLs, without http/https prefix (e.g. "openai.com").
+            Only available with web_search (not preview).
+        search_context_size: Controls how much context from web search results
+            is provided to the model. Options: "low", "medium" (default), "high".
+            Higher values use more tokens but may improve response quality.
+
+    Returns:
+        A dict representing the web search tool configuration.
+    """
+    tool: dict = {}
+    if preview:
+        tool["type"] = "web_search_preview"
+        if user_location:
+            tool["user_location"] = user_location
+        if search_context_size:
+            tool["search_context_size"] = search_context_size
+        return tool
+
+    # GA web_search tool
+    tool["type"] = "web_search"
+
+    if user_location:
+        tool["user_location"] = user_location
+
+    if search_context_size:
+        tool["search_context_size"] = search_context_size
+
+    # Domain filtering uses a nested filters structure
+    if allowed_domains:
+        tool["filters"] = {"allowed_domains": allowed_domains}
+
+    return tool
+
+
+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-0.0.84 → lm_deluge-0.0.86}/src/lm_deluge/tool/prefab/web_search.py

@@ -38,8 +38,8 @@ class AbstractWebSearchManager(abc.ABC):
             return self._tools
 
         self._tools = [
-            Tool.from_function(self._search),
-            Tool.from_function(self._fetch),
+            Tool.from_function(self._search, name=self.search_tool_name),
+            Tool.from_function(self._fetch, name=self.fetch_tool_name),
         ]
 
         return self._tools
@@ -76,12 +76,14 @@ class ExaWebSearchManager(AbstractWebSearchManager):
         search_tool_name: str = "web_search",
         fetch_tool_name: str = "web_fetch",
         timeout: int = 30,
+        max_contents_chars: int = 20_000,
     ):
         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,
@@ -146,7 +148,9 @@ class ExaWebSearchManager(AbstractWebSearchManager):
                 raise ValueError("EXA_API_KEY environment variable not set")
             data = {
                 "urls": [url],
-                "text": True,
+                "text": {
+                    "maxCharacters": self.max_contents_chars,
+                },
             }
 
             headers = {

{lm_deluge-0.0.84 → lm_deluge-0.0.86/src/lm_deluge.egg-info}/PKG-INFO

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

lm_deluge-0.0.84/src/lm_deluge/tool/builtin/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,
-    }