Improve MCP tool descriptions

Author: juntao

app/mcp_tools.py

@@ -18,35 +18,116 @@
 
 @mcp.tool()
 async def generate(description: str, requirements: str) -> str:
-    """Generate a new Rust cargo project from the description and requirements"""
+    """
+      Generate a new Rust cargo project from the description and requirements. The input arguments are
 
-    async with httpx.AsyncClient() as client:
-        response = await client.post(f"{API_BASE_URL}/generate-sync", json={'description': description, 'requirements': requirements})
-        return response.text
+        * description: a text string description of the generated Rust project.
+        * requiremenets: functional requirements on what the generated Rust project.
+
+      The return value is a text string that contains all files in the project. Each file is seperated by a [filename: path_to_file] line. For example, a project that contains a Cargo.toml file and a src/main.rs file will be returned as the following.
+
+[filename: Cargo.toml]
+[package]
+name = "a_command_line_calcu"
+version = "0.1.0"
+edition = "2021"
+
+[dependencies]
+
+
+[filename: src/main.rs]
+fn main() {
+    println!("Hello, world!");
+}
+
+    """
+
+    async with httpx.AsyncClient(timeout=60.0) as client:
+        try:
+            response = await client.post(f"{API_BASE_URL}/generate-sync", json={'description': description, 'requirements': requirements})
+            response.raise_for_status()
+            return response.text
+        except httpx.HTTPError as e:
+            print(f"HTTP error occurred: {e}")
+            return f"Error trying to generate a Rust project: {str(e)}"
 
 @mcp.tool()
 async def compile_and_fix(code: str, description: str = "A Rust project", max_attempts: int = 3) -> str:
-    """Compile a Rust cargo project and fix any compiler errors"""
+    """
+        Compile a Rust cargo project and fix any compiler errors.
+
+        The argument `code` is a text string that contains all files in the project. Each file is seperated by a [filename: path_to_file] line. For example, a project that contains a Cargo.toml file and a src/main.rs file will be returned as the following.
+
+[filename: Cargo.toml]
+[package]
+name = "a_command_line_calcu"
+version = "0.1.0"
+edition = "2021"
 
+[dependencies]
+
+
+[filename: src/main.rs]
+fn main() {
+    println!("Hello, world!");
+}
+
+        The return value is also a text string that contains all files in the project. It is in the same format as the input `code` argument.
+    """
     async with httpx.AsyncClient(timeout=60.0) as client:
         try:
             response = await client.post(
                 f"{API_BASE_URL}/compile-and-fix", 
                 json={'code': code, 'description': description, 'max_attempts': max_attempts}
             )
             response.raise_for_status()
-            return response.text
+            
+            resp_json = json.loads(response.text)
+            if "combined_text" in resp_json:
+                return ["combined_text"]
+            else:
+                return "Cannot fix the Rust compiler error."
+            # return response.text
         except httpx.HTTPError as e:
             print(f"HTTP error occurred: {e}")
-            return f"Error calling compile-and-fix API: {str(e)}"
+            return f"Error trying to fixing the Rust compiler error: {str(e)}"
 
 @mcp.tool()
 async def compile(code: str) -> str:
-    """Compile a Rust cargo project"""
+    """
+        Compile a Rust cargo project and return the compiler output.
+
+        The argument `code` is a text string that contains all files in the project. Each file is seperated by a [filename: path_to_file] line. For example, a project that contains a Cargo.toml file and a src/main.rs file will be returned as the following.
+
+[filename: Cargo.toml]
+[package]
+name = "a_command_line_calcu"
+version = "0.1.0"
+edition = "2021"
+
+[dependencies]
 
-    async with httpx.AsyncClient() as client:
-        response = await client.post(f"{API_BASE_URL}/compile", json={'code': code})
-        return response.text
+
+[filename: src/main.rs]
+fn main() {
+    println!("Hello, world!");
+}
+
+        The return value is a text string that contains the Rust compiler output.
+    """
+    async with httpx.AsyncClient(timeout=60.0) as client:
+        try:
+            response = await client.post(f"{API_BASE_URL}/compile", json={'code': code})
+            response.raise_for_status()
+            
+            resp_json = json.loads(response.text)
+            if "build_output" in resp_json:
+                return ["build_output"]
+            else:
+                return "Rust compiler error."
+        except httpx.HTTPError as e:
+            print(f"HTTP error occurred: {e}")
+            return f"Rust compiler error: {str(e)}"
 
 if __name__ == "__main__":
     # Use transport from environment variable or default to stdio