small fixes

Author: zainhas

examples/code_interpreter_demo.py

@@ -103,7 +103,7 @@
 response = code_interpreter.run(
     code=code_to_run_script,
     language="python",
-    files=[script_file], # Pass the script dictionary in a list
+    files=[script_file],  # Pass the script dictionary in a list
 )
 
 # Print results
@@ -124,12 +124,13 @@
 
 image_file = {
     "name": "tiny.png",
-    "encoding": "base64", # Use base64 encoding for binary files
+    "encoding": "base64",  # Use base64 encoding for binary files
     "content": tiny_png_base64,
 }
 
 # Code to check if the file exists and its size (Python doesn't inherently know image dimensions from bytes alone)
-code_to_check_file = """
+code_to_check_file = (
+    """
 import os
 import base64
 
@@ -138,14 +139,17 @@
     # Read the raw bytes back
     with open(file_path, 'rb') as f:
         raw_bytes = f.read()
-    original_bytes = base64.b64decode('""" + tiny_png_base64 + """')
+    original_bytes = base64.b64decode('"""
+    + tiny_png_base64
+    + """')
     print(f"File '{file_path}' exists.")
     print(f"Size on disk: {os.path.getsize(file_path)} bytes.")
     print(f"Size of original decoded base64 data: {len(original_bytes)} bytes.")
 
 else:
     print(f"File '{file_path}' does not exist.")
 """
+)
 
 response = code_interpreter.run(
     code=code_to_check_file,

src/together/resources/code_interpreter.py

@@ -41,7 +41,7 @@ def run(
             client=self._client,
         )
 
-        data: Dict[str, str] = {
+        data: Dict[str, Any] = {
             "code": code,
             "language": language,
         }
@@ -60,7 +60,9 @@ def run(
             except ValidationError as e:
                 raise ValueError(f"Invalid file input format: {e}") from e
             except TypeError as e:
-                 raise ValueError(f"Invalid file input: Each item in 'files' must be a dictionary. Error: {e}") from e
+                raise ValueError(
+                    f"Invalid file input: Each item in 'files' must be a dictionary. Error: {e}"
+                ) from e
 
             data["files"] = serialized_files
 

src/together/types/code_interpreter.py

@@ -6,6 +6,7 @@
 
 from together.types.endpoints import TogetherJSONModel
 
+
 class FileInput(TogetherJSONModel):
     """File input to be uploaded to the code interpreter session."""
 

tests/unit/test_code_interpreter.py

@@ -366,7 +366,7 @@ def test_code_interpreter_run_with_files(mocker):
     response = interpreter.run(
         code='with open("test.txt") as f: print(f.read())',
         language="python",
-        files=files_to_upload, # Pass the list of dictionaries directly
+        files=files_to_upload,  # Pass the list of dictionaries directly
     )
 
     # Verify the response
@@ -394,13 +394,14 @@ def test_code_interpreter_run_with_files(mocker):
         "files": expected_files_payload,
     }
 
+
 def test_code_interpreter_run_with_invalid_file_dict_structure(mocker):
     """Test that run raises ValueError for missing keys in file dict."""
     client = mocker.MagicMock()
     interpreter = CodeInterpreter(client)
 
     invalid_files = [
-        {"name": "test.txt", "content": "Missing encoding"} # Missing 'encoding'
+        {"name": "test.txt", "content": "Missing encoding"}  # Missing 'encoding'
     ]
 
     with pytest.raises(ValueError, match="Invalid file input format"):
@@ -410,13 +411,18 @@ def test_code_interpreter_run_with_invalid_file_dict_structure(mocker):
             files=invalid_files,
         )
 
+
 def test_code_interpreter_run_with_invalid_file_dict_encoding(mocker):
     """Test that run raises ValueError for invalid encoding value."""
     client = mocker.MagicMock()
     interpreter = CodeInterpreter(client)
 
     invalid_files = [
-        {"name": "test.txt", "encoding": "utf-8", "content": "Invalid encoding"} # Invalid 'encoding' value
+        {
+            "name": "test.txt",
+            "encoding": "utf-8",
+            "content": "Invalid encoding",
+        }  # Invalid 'encoding' value
     ]
 
     with pytest.raises(ValueError, match="Invalid file input format"):
@@ -426,17 +432,21 @@ def test_code_interpreter_run_with_invalid_file_dict_encoding(mocker):
             files=invalid_files,
         )
 
+
 def test_code_interpreter_run_with_invalid_file_list_item(mocker):
     """Test that run raises ValueError for non-dict item in files list."""
     client = mocker.MagicMock()
     interpreter = CodeInterpreter(client)
 
     invalid_files = [
         {"name": "good.txt", "encoding": "string", "content": "Good"},
-        "not a dictionary" # Invalid item type
+        "not a dictionary",  # Invalid item type
     ]
 
-    with pytest.raises(ValueError, match="Invalid file input: Each item in 'files' must be a dictionary"):
+    with pytest.raises(
+        ValueError,
+        match="Invalid file input: Each item in 'files' must be a dictionary",
+    ):
         interpreter.run(
             code="print('test')",
             language="python",