cleaning up

Author: aseembits93

myclient.py

@@ -1,25 +1,7 @@
-# import asyncio
-# from fastmcp import Client
-#
-# client = Client("myserver.py")
-#
-# async def call_tool(file: str, function: str)-> None:
-#     async with client:
-#         result = await client.call_tool("optimize_code", {"file": f"{file}", "function":f"{function}"})
-#         print(result)
-#
-# asyncio.run(call_tool("Ford","Mustang"))
-
-
 import anthropic
 from rich import print
-
-# Your server URL (replace with your actual URL)
 url = 'https://0de03d07f8b9.ngrok-free.app'
-
 client = anthropic.Anthropic()
-
-#file and fn names are hard coded for now
 response = client.beta.messages.create(
     model="claude-sonnet-4-20250514",
     max_tokens=1000,
@@ -28,39 +10,11 @@
         {
             "type": "url",
             "url": f"{url}/mcp/",
-            "name": "HelpfulAssistant",
+            "name": "Code Optimization Assistant",
         }
     ],
     extra_headers={
         "anthropic-beta": "mcp-client-2025-04-04"
     }
 )
-
-print(response.content)
-
-
-# import anthropic
-# from rich import print
-#
-# # Your server URL (replace with your actual URL)
-# url = 'https://0de03d07f8b9.ngrok-free.app'
-#
-# client = anthropic.Anthropic()
-#
-# response = client.beta.messages.create(
-#     model="claude-sonnet-4-20250514",
-#     max_tokens=1000,
-#     messages=[{"role": "user", "content": "Roll a few dice!"}],
-#     mcp_servers=[
-#         {
-#             "type": "url",
-#             "url": f"{url}/mcp/",
-#             "name": "Dice Roller",
-#         }
-#     ],
-#     extra_headers={
-#         "anthropic-beta": "mcp-client-2025-04-04"
-#     }
-# )
-#
-# print(response.content)
+print(response.content)

myserver.py

@@ -1,14 +1,11 @@
 import pathlib
-
 from fastmcp import FastMCP
-
 from tests.scripts.end_to_end_test_utilities import CoverageExpectation, TestConfig, run_codeflash_command
-
-mcp = FastMCP(name="HelpfulAssistant",
+mcp = FastMCP(name="Code Optimization Assistant",
     instructions="""
         This server provides code optimization tools.
         Call optimize_code(file, function) to optimize your code.
-    """,)
+    """)
 
 @mcp.tool
 def optimize_code(file: str, function: str) -> bool:
@@ -23,7 +20,7 @@ def optimize_code(file: str, function: str) -> bool:
             )
         ],
     )
-    cwd = (pathlib.Path(__file__).parent/ "code_to_optimize").resolve() # todo remove it
+    cwd = (pathlib.Path(__file__).parent/ "code_to_optimize").resolve() # TODO remove it
     return run_codeflash_command(
         cwd, config, 100, ['print("codeflash stdout: Sorting list")', 'print(f"result: {arr}")']
     )