Fix pyright configuration and type errors in loop.py

- Add working-directory parameter to pyright-action to scope checks to computer-use-demo
- Fix type errors in loop.py by using proper isinstance checks and type casting
- Use .get() for safe dictionary access and cast to BetaToolUseBlockParam
- Replace .pop() with conditional del for cache_control removal

Author: zealoushacker

.github/workflows/tests.yaml

@@ -38,6 +38,8 @@ jobs:
           pip install -r dev-requirements.txt
       - run: echo "$PWD/.venv/bin" >> $GITHUB_PATH
       - uses: jakebailey/pyright-action@v1
+        with:
+          working-directory: computer-use-demo
   pytest:
     runs-on: ubuntu-latest
     defaults:

computer-use-demo/computer_use_demo/loop.py

@@ -169,11 +169,12 @@ async def sampling_loop(
         tool_result_content: list[BetaToolResultBlockParam] = []
         for content_block in response_params:
             output_callback(content_block)
-            if content_block.get("type") == "tool_use":
+            if isinstance(content_block, dict) and content_block.get("type") == "tool_use":
+                # Type narrowing for tool use blocks
                 tool_use_block = cast(BetaToolUseBlockParam, content_block)
                 result = await tool_collection.run(
                     name=tool_use_block["name"],
-                    tool_input=cast(dict[str, Any], tool_use_block["input"]),
+                    tool_input=cast(dict[str, Any], tool_use_block.get("input", {})),
                 )
                 tool_result_content.append(
                     _make_api_tool_result(result, tool_use_block["id"])
@@ -278,7 +279,8 @@ def _inject_prompt_caching(
                     {"type": "ephemeral"}
                 )
             else:
-                content[-1].pop("cache_control", None)
+                if isinstance(content[-1], dict) and "cache_control" in content[-1]:
+                    del content[-1]["cache_control"]  # type: ignore
                 # we'll only every have one extra turn per loop
                 break