python-codex 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

pycodex/tools/read_file_tool.py

@@ -9,13 +9,12 @@ Expected behavior:
   mode used to inspect code structure without shelling out to `sed` or `cat`.
 """
 
-from __future__ import annotations
-
 from collections import deque
 from pathlib import Path
 
 from ..protocol import JSONDict, JSONValue
 from .base_tool import BaseTool, ToolContext
+import typing
 
 MAX_LINE_LENGTH = 500
 TAB_WIDTH = 4
@@ -49,7 +48,7 @@ class ReadFileTool(BaseTool):
         "required": ["file_path"],
     }
 
-    async def run(self, context: ToolContext, args: JSONDict) -> JSONValue:
+    async def run(self, context: 'ToolContext', args: 'JSONDict') -> 'JSONValue':
         del context
         file_path = Path(str(args.get("file_path", "")))
         offset = int(args.get("offset", 1))
@@ -72,7 +71,7 @@ class ReadFileTool(BaseTool):
             return self._read_indentation(file_path, offset, limit, indentation)
         return self._read_slice(file_path, offset, limit)
 
-    def _read_slice(self, file_path: Path, offset: int, limit: int) -> str:
+    def _read_slice(self, file_path: 'Path', offset: 'int', limit: 'int') -> 'str':
         lines = file_path.read_text(errors="replace").splitlines()
         if offset > len(lines):
             return "Error: `offset` exceeds file length."
@@ -85,11 +84,11 @@ class ReadFileTool(BaseTool):
 
     def _read_indentation(
         self,
-        file_path: Path,
-        offset: int,
-        limit: int,
-        indentation: JSONDict,
-    ) -> str:
+        file_path: 'Path',
+        offset: 'int',
+        limit: 'int',
+        indentation: 'JSONDict',
+    ) -> 'str':
         lines = self._collect_line_records(file_path)
         anchor_line = int(indentation.get("anchor_line", offset))
         max_levels = int(indentation.get("max_levels", 0))
@@ -171,7 +170,7 @@ class ReadFileTool(BaseTool):
             f"L{record['number']}: {record['display']}" for record in selected
         )
 
-    def _collect_line_records(self, file_path: Path) -> list[dict[str, object]]:
+    def _collect_line_records(self, file_path: 'Path') -> 'typing.List[typing.Dict[str, object]]':
         records = []
         for number, raw in enumerate(file_path.read_text(errors="replace").splitlines(), start=1):
             records.append(
@@ -185,7 +184,7 @@ class ReadFileTool(BaseTool):
             )
         return records
 
-    def _compute_effective_indents(self, records: list[dict[str, object]]) -> list[int]:
+    def _compute_effective_indents(self, records: 'typing.List[typing.Dict[str, object]]') -> 'typing.List[int]':
         effective = []
         previous_indent = 0
         for record in records:
@@ -196,7 +195,7 @@ class ReadFileTool(BaseTool):
                 effective.append(previous_indent)
         return effective
 
-    def _measure_indent(self, line: str) -> int:
+    def _measure_indent(self, line: 'str') -> 'int':
         total = 0
         for character in line:
             if character == " ":
@@ -207,10 +206,10 @@ class ReadFileTool(BaseTool):
                 break
         return total
 
-    def _format_line(self, text: str) -> str:
+    def _format_line(self, text: 'str') -> 'str':
         return text[:MAX_LINE_LENGTH]
 
-    def _trim_empty_lines(self, records: deque[dict[str, object]]) -> None:
+    def _trim_empty_lines(self, records: 'deque[typing.Dict[str, object]]') -> 'None':
         while records and not str(records[0]["raw"]).strip():
             records.popleft()
         while records and not str(records[-1]["raw"]).strip():

pycodex/tools/request_permissions_tool.py

@@ -9,8 +9,6 @@ Expected behavior:
 - Return the granted permission profile and scope so later tool calls can use it.
 """
 
-from __future__ import annotations
-
 from ..protocol import JSONDict, JSONValue
 from ..runtime_services import RequestPermissionsManager
 from .base_tool import BaseTool, ToolContext
@@ -73,10 +71,10 @@ class RequestPermissionsTool(BaseTool):
     }
     supports_parallel = False
 
-    def __init__(self, request_manager: RequestPermissionsManager) -> None:
+    def __init__(self, request_manager: 'RequestPermissionsManager') -> 'None':
         self._request_manager = request_manager
 
-    async def run(self, context: ToolContext, args: JSONDict) -> JSONValue:
+    async def run(self, context: 'ToolContext', args: 'JSONDict') -> 'JSONValue':
         del context
         permissions = args.get("permissions")
         if not isinstance(permissions, dict):

pycodex/tools/request_user_input_tool.py

@@ -12,14 +12,13 @@ Expected behavior:
   `success=true`.
 """
 
-from __future__ import annotations
-
 import json
 
 from ..collaboration import collaboration_mode_display_name
 from ..protocol import JSONDict, JSONValue
 from ..runtime_services import RequestUserInputManager
 from .base_tool import BaseTool, StructuredToolOutput, ToolContext
+import typing
 
 REQUEST_USER_INPUT_QUESTION_SCHEMA = {
     "type": "object",
@@ -67,9 +66,9 @@ REQUEST_USER_INPUT_QUESTION_SCHEMA = {
 
 
 def request_user_input_is_available(
-    mode: str,
-    default_mode_request_user_input: bool = False,
-) -> bool:
+    mode: 'str',
+    default_mode_request_user_input: 'bool' = False,
+) -> 'bool':
     normalized = mode.strip().lower()
     return normalized == "plan" or (
         default_mode_request_user_input and normalized == "default"
@@ -77,9 +76,9 @@ def request_user_input_is_available(
 
 
 def request_user_input_unavailable_message(
-    mode: str,
-    default_mode_request_user_input: bool = False,
-) -> str | None:
+    mode: 'str',
+    default_mode_request_user_input: 'bool' = False,
+) -> 'typing.Union[str, None]':
     if request_user_input_is_available(mode, default_mode_request_user_input):
         return None
     return (
@@ -89,8 +88,8 @@ def request_user_input_unavailable_message(
 
 
 def request_user_input_tool_description(
-    default_mode_request_user_input: bool = False,
-) -> str:
+    default_mode_request_user_input: 'bool' = False,
+) -> 'str':
     if default_mode_request_user_input:
         allowed_modes = "Default or Plan mode"
     else:
@@ -120,16 +119,16 @@ class RequestUserInputTool(BaseTool):
 
     def __init__(
         self,
-        request_manager: RequestUserInputManager,
-        default_mode_request_user_input: bool = False,
-    ) -> None:
+        request_manager: 'RequestUserInputManager',
+        default_mode_request_user_input: 'bool' = False,
+    ) -> 'None':
         self._request_manager = request_manager
         self._default_mode_request_user_input = default_mode_request_user_input
         self.description = request_user_input_tool_description(
             default_mode_request_user_input
         )
 
-    async def run(self, context: ToolContext, args: JSONDict) -> JSONValue:
+    async def run(self, context: 'ToolContext', args: 'JSONDict') -> 'JSONValue':
         unavailable = request_user_input_unavailable_message(
             context.collaboration_mode,
             self._default_mode_request_user_input,

pycodex/tools/resume_agent_tool.py

@@ -8,8 +8,6 @@ Expected behavior:
 - Return the agent's current status payload.
 """
 
-from __future__ import annotations
-
 from ..protocol import JSONDict, JSONValue
 from ..runtime_services import SubAgentManager
 from .agent_tool_schemas import AGENT_STATUS_SCHEMA
@@ -45,10 +43,10 @@ class ResumeAgentTool(BaseTool):
     output_schema = RESUME_AGENT_OUTPUT_SCHEMA
     supports_parallel = False
 
-    def __init__(self, subagent_manager: SubAgentManager) -> None:
+    def __init__(self, subagent_manager: 'SubAgentManager') -> 'None':
         self._subagent_manager = subagent_manager
 
-    async def run(self, context: ToolContext, args: JSONDict) -> JSONValue:
+    async def run(self, context: 'ToolContext', args: 'JSONDict') -> 'JSONValue':
         del context
         agent_id = str(args.get("id", "")).strip()
         if not agent_id:

pycodex/tools/send_input_tool.py

@@ -9,12 +9,11 @@ Expected behavior:
 - Return the submission id for the queued input.
 """
 
-from __future__ import annotations
-
 from ..protocol import JSONDict, JSONValue
 from ..runtime_services import SubAgentManager
 from .agent_tool_schemas import COLLAB_INPUT_ITEMS_SCHEMA
 from .base_tool import BaseTool, ToolContext
+import typing
 
 SEND_INPUT_OUTPUT_SCHEMA = {
     "type": "object",
@@ -58,10 +57,10 @@ class SendInputTool(BaseTool):
     output_schema = SEND_INPUT_OUTPUT_SCHEMA
     supports_parallel = False
 
-    def __init__(self, subagent_manager: SubAgentManager) -> None:
+    def __init__(self, subagent_manager: 'SubAgentManager') -> 'None':
         self._subagent_manager = subagent_manager
 
-    async def run(self, context: ToolContext, args: JSONDict) -> JSONValue:
+    async def run(self, context: 'ToolContext', args: 'JSONDict') -> 'JSONValue':
         del context
         agent_id = str(args.get("id", "")).strip()
         if not agent_id:
@@ -83,10 +82,10 @@ class SendInputTool(BaseTool):
 
     def _compose_prompt(
         self,
-        message: str | None,
-        items: list[dict[str, object]] | None,
-    ) -> str:
-        parts: list[str] = []
+        message: 'typing.Union[str, None]',
+        items: 'typing.Union[typing.List[typing.Dict[str, object]], None]',
+    ) -> 'str':
+        parts: 'typing.List[str]' = []
         if message:
             parts.append(message.strip())
         for item in items or []:
@@ -99,7 +98,7 @@ class SendInputTool(BaseTool):
                 parts.append(str(item))
         return "\n\n".join(part for part in parts if part)
 
-    def _optional_string(self, args: JSONDict, key: str) -> str | None:
+    def _optional_string(self, args: 'JSONDict', key: 'str') -> 'typing.Union[str, None]':
         value = args.get(key)
         if value in (None, ""):
             return None

pycodex/tools/shell_command_tool.py

@@ -11,13 +11,12 @@ Expected behavior:
   stdout, and stderr.
 """
 
-from __future__ import annotations
-
 import asyncio
 from pathlib import Path
 
 from ..protocol import JSONDict, JSONValue
 from .base_tool import BaseTool, ToolContext
+import typing
 
 DEFAULT_SHELL_TIMEOUT_MS = 30_000
 MAX_OUTPUT_CHARS = 12_000
@@ -38,10 +37,10 @@ class ShellCommandTool(BaseTool):
     }
     supports_parallel = False
 
-    def __init__(self, cwd: str | Path | None = None) -> None:
+    def __init__(self, cwd: 'typing.Union[typing.Union[str, Path], None]' = None) -> 'None':
         self._working_directory = Path(cwd or Path.cwd()).resolve()
 
-    async def run(self, context: ToolContext, args: JSONDict) -> JSONValue:
+    async def run(self, context: 'ToolContext', args: 'JSONDict') -> 'JSONValue':
         del context
         command = str(args.get("command", "")).strip()
         timeout_ms = int(args.get("timeout_ms", DEFAULT_SHELL_TIMEOUT_MS))
@@ -93,7 +92,7 @@ class ShellCommandTool(BaseTool):
 
         return "\n".join(pieces)
 
-    def _resolve_workdir(self, workdir_arg) -> Path:
+    def _resolve_workdir(self, workdir_arg) -> 'Path':
         if workdir_arg in (None, ""):
             return self._working_directory
         workdir = Path(str(workdir_arg))
@@ -101,7 +100,7 @@ class ShellCommandTool(BaseTool):
             workdir = self._working_directory / workdir
         return workdir.resolve()
 
-    def _clip_output(self, text: str) -> str:
+    def _clip_output(self, text: 'str') -> 'str':
         if len(text) <= MAX_OUTPUT_CHARS:
             return text
         return text[:MAX_OUTPUT_CHARS] + "\n...[truncated]..."

pycodex/tools/shell_tool.py

@@ -11,13 +11,12 @@ Expected behavior:
   stdout, and stderr.
 """
 
-from __future__ import annotations
-
 import asyncio
 from pathlib import Path
 
 from ..protocol import JSONDict, JSONValue
 from .base_tool import BaseTool, ToolContext
+import typing
 
 DEFAULT_SHELL_TIMEOUT_MS = 30_000
 MAX_OUTPUT_CHARS = 12_000
@@ -43,10 +42,10 @@ class ShellTool(BaseTool):
     }
     supports_parallel = False
 
-    def __init__(self, cwd: str | Path | None = None) -> None:
+    def __init__(self, cwd: 'typing.Union[typing.Union[str, Path], None]' = None) -> 'None':
         self._working_directory = Path(cwd or Path.cwd()).resolve()
 
-    async def run(self, context: ToolContext, args: JSONDict) -> JSONValue:
+    async def run(self, context: 'ToolContext', args: 'JSONDict') -> 'JSONValue':
         del context
         command = args.get("command")
         timeout_ms = int(args.get("timeout_ms", DEFAULT_SHELL_TIMEOUT_MS))
@@ -98,7 +97,7 @@ class ShellTool(BaseTool):
 
         return "\n".join(pieces)
 
-    def _resolve_workdir(self, workdir_arg) -> Path:
+    def _resolve_workdir(self, workdir_arg) -> 'Path':
         if workdir_arg in (None, ""):
             return self._working_directory
         workdir = Path(str(workdir_arg))
@@ -106,7 +105,7 @@ class ShellTool(BaseTool):
             workdir = self._working_directory / workdir
         return workdir.resolve()
 
-    def _clip_output(self, text: str) -> str:
+    def _clip_output(self, text: 'str') -> 'str':
         if len(text) <= MAX_OUTPUT_CHARS:
             return text
         return text[:MAX_OUTPUT_CHARS] + "\n...[truncated]..."

pycodex/tools/spawn_agent_tool.py

@@ -9,12 +9,11 @@ Expected behavior:
 - Return the new agent identifier plus any user-facing nickname.
 """
 
-from __future__ import annotations
-
 from ..protocol import JSONDict, JSONValue
 from ..runtime_services import SubAgentManager
 from .agent_tool_schemas import COLLAB_INPUT_ITEMS_SCHEMA
 from .base_tool import BaseTool, ToolContext
+import typing
 
 SPAWN_AGENT_OUTPUT_SCHEMA = {
     "type": "object",
@@ -70,10 +69,10 @@ class SpawnAgentTool(BaseTool):
     output_schema = SPAWN_AGENT_OUTPUT_SCHEMA
     supports_parallel = False
 
-    def __init__(self, subagent_manager: SubAgentManager) -> None:
+    def __init__(self, subagent_manager: 'SubAgentManager') -> 'None':
         self._subagent_manager = subagent_manager
 
-    async def run(self, context: ToolContext, args: JSONDict) -> JSONValue:
+    async def run(self, context: 'ToolContext', args: 'JSONDict') -> 'JSONValue':
         message = self._optional_string(args, "message")
         items = args.get("items")
         if items is not None and not isinstance(items, list):
@@ -90,7 +89,7 @@ class SpawnAgentTool(BaseTool):
             history=context.history,
         )
 
-    def _optional_string(self, args: JSONDict, key: str) -> str | None:
+    def _optional_string(self, args: 'JSONDict', key: 'str') -> 'typing.Union[str, None]':
         value = args.get(key)
         if value in (None, ""):
             return None