oauth-codex 3.2.0__tar.gz → 3.2.2__tar.gz

{oauth_codex-3.2.0 → oauth_codex-3.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: oauth-codex
-Version: 3.2.0
+Version: 3.2.2
 Summary: Codex OAuth-based Python SDK with a single Client and generate-first API
 Author: Codex
 Requires-Python: >=3.11

{oauth_codex-3.2.0 → oauth_codex-3.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "oauth-codex"
-version = "3.2.0"
+version = "3.2.2"
 description = "Codex OAuth-based Python SDK with a single Client and generate-first API"
 readme = "README.md"
 requires-python = ">=3.11"

{oauth_codex-3.2.0 → oauth_codex-3.2.2}/src/oauth_codex/_sdk_client.py

@@ -59,14 +59,14 @@ class _SyncEngine:
             return self._stream_responses(payload)
 
         response = self._client.request("POST", "/responses", json_data=payload)
-        return _to_namespace(response.json())
+        return response.json()
 
     def responses_input_tokens_count(self, **kwargs: Any) -> Any:
         payload = _payload_without_none(kwargs)
         response = self._client.request(
             "POST", "/responses/input_tokens", json_data=payload
         )
-        return _to_namespace(response.json())
+        return response.json()
 
     def _stream_responses(self, payload: dict[str, Any]) -> Iterator[Any]:
         payload = dict(payload)
@@ -75,7 +75,7 @@ class _SyncEngine:
         events = response.json()
         if isinstance(events, list):
             for event in events:
-                yield _to_namespace(event)
+                yield event
 
 
 class _AsyncEngine:
@@ -88,14 +88,14 @@ class _AsyncEngine:
             return self._stream_responses(payload)
 
         response = await self._client.request("POST", "/responses", json_data=payload)
-        return _to_namespace(response.json())
+        return response.json()
 
     async def aresponses_input_tokens_count(self, **kwargs: Any) -> Any:
         payload = _payload_without_none(kwargs)
         response = await self._client.request(
             "POST", "/responses/input_tokens", json_data=payload
         )
-        return _to_namespace(response.json())
+        return response.json()
 
     async def _stream_responses(self, payload: dict[str, Any]) -> AsyncIterator[Any]:
         payload = dict(payload)
@@ -104,7 +104,7 @@ class _AsyncEngine:
         events = response.json()
         if isinstance(events, list):
             for event in events:
-                yield _to_namespace(event)
+                yield event
 
 
 class Client(SyncAPIClient):

{oauth_codex-3.2.0 → oauth_codex-3.2.2}/src/oauth_codex/_streaming.py

@@ -16,7 +16,10 @@ def _extract_event_data(raw_event: str) -> str | None:
     data_lines: list[str] = []
     for line in raw_event.split("\n"):
         if line.startswith("data:"):
-            data_lines.append(line[5:].lstrip())
+            content = line[5:].lstrip()
+            if content == "[DONE]":
+                return "[DONE]"
+            data_lines.append(content)
 
     if not data_lines:
         return None
@@ -41,12 +44,16 @@ class Stream(Generic[T]):
                 while "\n\n" in buffer:
                     raw_event, buffer = buffer.split("\n\n", 1)
                     data = _extract_event_data(raw_event)
+                    if data == "[DONE]":
+                        return
                     if data is None:
                         continue
                     yield self._cast_event(data)
 
             if buffer:
                 data = _extract_event_data(buffer)
+                if data == "[DONE]":
+                    return
                 if data is not None:
                     yield self._cast_event(data)
         finally:
@@ -71,12 +78,16 @@ class AsyncStream(Generic[T]):
                 while "\n\n" in buffer:
                     raw_event, buffer = buffer.split("\n\n", 1)
                     data = _extract_event_data(raw_event)
+                    if data == "[DONE]":
+                        return
                     if data is None:
                         continue
                     yield self._cast_event(data)
 
             if buffer:
                 data = _extract_event_data(buffer)
+                if data == "[DONE]":
+                    return
                 if data is not None:
                     yield self._cast_event(data)
         finally:

{oauth_codex-3.2.0 → oauth_codex-3.2.2}/src/oauth_codex/_version.py

@@ -1,2 +1,2 @@
 __title__ = "oauth-codex"
-__version__ = "3.2.0"
+__version__ = "3.2.2"

{oauth_codex-3.2.0 → oauth_codex-3.2.2}/src/oauth_codex/resources/chat/completions.py

@@ -104,9 +104,16 @@ def _extract_tool_calls(response_data: dict[str, Any]) -> list[dict[str, Any]]:
         if not isinstance(name, str) or not name:
             continue
 
-        arguments = item.get("arguments")
+        # In Codex, arguments are often in `arguments_json` or `arguments` as a dict
+        arguments = item.get("arguments_json")
         if not isinstance(arguments, str):
-            arguments = "{}"
+            args_dict = item.get("arguments")
+            if isinstance(args_dict, dict):
+                arguments = json.dumps(args_dict)
+            elif isinstance(args_dict, str):
+                arguments = args_dict
+            else:
+                arguments = "{}"
 
         call_id = item.get("call_id") or item.get("id")
         if not isinstance(call_id, str) or not call_id:
@@ -125,7 +132,6 @@ def _extract_tool_calls(response_data: dict[str, Any]) -> list[dict[str, Any]]:
 
     return tool_calls
 
-
 def _build_assistant_message(completion: ChatCompletion) -> dict[str, Any]:
     message = completion.choices[0].message
     out: dict[str, Any] = {"role": "assistant"}
@@ -145,34 +151,38 @@ def _coerce_tool_output_content(value: Any) -> str:
         return str(value)
 
 
-def _parse_tool_arguments(arguments: str) -> Any:
-    if not arguments.strip():
-        return {}
-    try:
-        return json.loads(arguments)
-    except json.JSONDecodeError:
-        return {}
-
-
-def _build_tool_function_map(tools: Any) -> dict[str, Any]:
-    if not isinstance(tools, list):
-        return {}
-
-    functions: dict[str, Any] = {}
+def _build_tool_function_map(tools: list[Any]) -> dict[str, Any]:
+    tool_map = {}
     for tool in tools:
         if callable(tool):
-            name = getattr(tool, "__name__", "")
-            if isinstance(name, str) and name:
-                functions[name] = tool
-    return functions
-
-
+            name = getattr(tool, "__name__", "tool")
+            tool_map[name] = tool
+        elif isinstance(tool, dict) and "function" in tool:
+            # Not a callable directly but maybe user wants it? 
+            # OpenAI SDK run_tools specifically requires callables.
+            pass
+    return tool_map
+
+def _parse_tool_arguments(arguments: Any) -> dict[str, Any]:
+    if isinstance(arguments, dict):
+        return arguments
+    if not isinstance(arguments, str) or not arguments.strip():
+        raise ValueError(f"Tool arguments must be a valid JSON object string, got: {arguments!r}")
+    
+    try:
+        parsed = json.loads(arguments)
+        if isinstance(parsed, str):
+            parsed = json.loads(parsed)
+        if isinstance(parsed, dict):
+            return parsed
+        raise ValueError("Parsed arguments is not a dictionary")
+    except Exception as e:
+        raise ValueError(f"Failed to parse tool arguments: {arguments!r}") from e
 def _run_tool_function(*, fn: Any, arguments: Any) -> Any:
     if isinstance(arguments, dict):
         return fn(**arguments)
     return fn(arguments)
 
-
 def _to_chat_completion(
     *,
     response_data: dict[str, Any],

{oauth_codex-3.2.0 → oauth_codex-3.2.2}/src/oauth_codex/resources/responses/_helpers.py

@@ -9,6 +9,8 @@ from ...types.shared import TokenUsage
 def usage_from_engine(usage: Any) -> TokenUsage | None:
     if usage is None:
         return None
+    if isinstance(usage, dict):
+        return TokenUsage(**usage)
     return TokenUsage(
         input_tokens=getattr(usage, "input_tokens", None),
         output_tokens=getattr(usage, "output_tokens", None),
@@ -21,14 +23,28 @@ def usage_from_engine(usage: Any) -> TokenUsage | None:
 
 
 def response_from_engine(resp: Any) -> Response:
+    if isinstance(resp, dict):
+        return Response(
+            id=resp.get("id", ""),
+            output=resp.get("output", []) or [],
+            output_text=resp.get("output_text", "") or "",
+            usage=usage_from_engine(resp.get("usage", None)),
+            error=resp.get("error", None),
+            reasoning_summary=resp.get("reasoning_summary", None),
+            reasoning_items=resp.get("reasoning_items", []) or [],
+            encrypted_reasoning_content=resp.get("encrypted_reasoning_content", None),
+            finish_reason=resp.get("finish_reason", None),
+            previous_response_id=resp.get("previous_response_id", None),
+            raw_response=resp.get("raw_response", None),
+        )
     return Response(
         id=getattr(resp, "id", ""),
-        output=list(getattr(resp, "output", []) or []),
+        output=getattr(resp, "output", []) or [],
         output_text=getattr(resp, "output_text", "") or "",
         usage=usage_from_engine(getattr(resp, "usage", None)),
         error=getattr(resp, "error", None),
         reasoning_summary=getattr(resp, "reasoning_summary", None),
-        reasoning_items=list(getattr(resp, "reasoning_items", []) or []),
+        reasoning_items=getattr(resp, "reasoning_items", []) or [],
         encrypted_reasoning_content=getattr(resp, "encrypted_reasoning_content", None),
         finish_reason=getattr(resp, "finish_reason", None),
         previous_response_id=getattr(resp, "previous_response_id", None),
@@ -37,6 +53,18 @@ def response_from_engine(resp: Any) -> Response:
 
 
 def event_from_engine(event: Any) -> ResponseStreamEvent:
+    if isinstance(event, dict):
+        return ResponseStreamEvent(
+            type=event.get("type", "event"),
+            delta=event.get("delta", None),
+            usage=usage_from_engine(event.get("usage", None)),
+            raw=event.get("raw", None),
+            error=event.get("error", None),
+            call_id=event.get("call_id", None),
+            response_id=event.get("response_id", None),
+            finish_reason=event.get("finish_reason", None),
+            schema_version=event.get("schema_version", "v1"),
+        )
     return ResponseStreamEvent(
         type=getattr(event, "type", "event"),
         delta=getattr(event, "delta", None),

{oauth_codex-3.2.0 → oauth_codex-3.2.2}/src/oauth_codex.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: oauth-codex
-Version: 3.2.0
+Version: 3.2.2
 Summary: Codex OAuth-based Python SDK with a single Client and generate-first API
 Author: Codex
 Requires-Python: >=3.11