Merge pull request #97 from sloganking/codex/continue-porting-v2-features

Port screen brightness feature to v2

Author: sloganking

assistant_v2/FEATURE_PROGRESS.md

@@ -4,7 +4,7 @@ This document tracks which features from the original assistant have been implem
 
 | Feature | Status |
 | --- | --- |
-| Screen brightness control | Pending |
+| Screen brightness control | Done |
 | System volume adjustment (Windows only) | Pending |
 | Media playback commands | Pending |
 | Launch applications from voice | Pending |

assistant_v2/src/main.rs

@@ -100,6 +100,19 @@ async fn main() -> Result<(), Box<dyn Error>> {
                 strict: None,
             }
             .into(),
+            FunctionObject {
+                name: "set_screen_brightness".into(),
+                description: Some(
+                    "Sets the screen brightness from 0 to 100 using the `luster` utility.".into(),
+                ),
+                parameters: Some(serde_json::json!({
+                    "type": "object",
+                    "properties": {"brightness": {"type": "integer"}},
+                    "required": ["brightness"],
+                })),
+                strict: None,
+            }
+            .into(),
             FunctionObject {
                 name: "set_clipboard".into(),
                 description: Some("Sets the clipboard to the given text.".into()),
@@ -299,6 +312,25 @@ async fn handle_requires_action(
                 });
             }
 
+            if tool.function.name == "set_screen_brightness" {
+                let brightness = match serde_json::from_str::<serde_json::Value>(&tool.function.arguments) {
+                    Ok(v) => v["brightness"].as_i64().unwrap_or(0) as u32,
+                    Err(_) => 0,
+                };
+
+                let result = std::process::Command::new("luster")
+                    .arg(brightness.to_string())
+                    .output();
+                let msg = match result {
+                    Ok(_) => "Brightness set".to_string(),
+                    Err(e) => format!("Failed to set brightness: {}", e),
+                };
+                tool_outputs.push(ToolsOutputs {
+                    tool_call_id: Some(tool.id.clone()),
+                    output: Some(msg.into()),
+                });
+            }
+
             if tool.function.name == "open_openai_billing" {
                 let result = open::that("https://platform.openai.com/usage");
                 let msg = match result {
@@ -401,4 +433,32 @@ mod tests {
             _ => false,
         }));
     }
+
+    #[test]
+    fn includes_set_screen_brightness_function() {
+        let req = CreateAssistantRequestArgs::default()
+            .instructions("test")
+            .model("gpt-4o")
+            .tools(vec![
+                FunctionObject {
+                    name: "set_screen_brightness".into(),
+                    description: Some("Sets the screen brightness from 0 to 100 using the `luster` utility.".into()),
+                    parameters: Some(serde_json::json!({
+                        "type": "object",
+                        "properties": {"brightness": {"type": "integer"}},
+                        "required": ["brightness"],
+                    })),
+                    strict: None,
+                }
+                .into(),
+            ])
+            .build()
+            .unwrap();
+
+        let tools = req.tools.unwrap();
+        assert!(tools.iter().any(|t| match t {
+            async_openai::types::AssistantTools::Function(f) => f.function.name == "set_screen_brightness",
+            _ => false,
+        }));
+    }
 }