wcgw 5.0.1__py3-none-any.whl → 5.1.0__py3-none-any.whl

wcgw/client/tools.py

@@ -55,6 +55,7 @@ from ..types_ import (
     WriteIfEmpty,
 )
 from .encoder import EncoderDecoder, get_default_encoder
+from .file_ops.extensions import select_max_tokens
 from .file_ops.search_replace import (
     DIVIDER_MARKER,
     REPLACE_MARKER,
@@ -99,7 +100,8 @@ def initialize(
     any_workspace_path: str,
     read_files_: list[str],
     task_id_to_resume: str,
-    max_tokens: Optional[int],
+    coding_max_tokens: Optional[int],
+    noncoding_max_tokens: Optional[int],
     mode: ModesConfig,
     chat_id: str,
 ) -> tuple[str, Context, dict[str, list[tuple[int, int]]]]:
@@ -115,7 +117,7 @@ def initialize(
         # Try to load state from the chat ID
         if not context.bash_state.load_state_from_chat_id(chat_id):
             return (
-                f"Error: No saved bash state found for chat ID {chat_id}",
+                f"Error: No saved bash state found for chat ID {chat_id}. Please re-initialize to get a new id or use correct id.",
                 context,
                 {},
             )
@@ -128,7 +130,8 @@ def initialize(
         try:
             project_root_path, task_mem, loaded_state = load_memory(
                 task_id_to_resume,
-                max_tokens,
+                coding_max_tokens,
+                noncoding_max_tokens,
                 lambda x: default_enc.encoder(x),
                 lambda x: default_enc.decoder(x),
             )
@@ -248,7 +251,7 @@ def initialize(
                 for f in read_files_
             ]
         initial_files, initial_paths_with_ranges, _ = read_files(
-            read_files_, max_tokens, context
+            read_files_, coding_max_tokens, noncoding_max_tokens, context
         )
         initial_files_context = f"---\n# Requested files\n{initial_files}\n---\n"
 
@@ -316,7 +319,7 @@ def reset_wcgw(
     if chat_id != context.bash_state.current_chat_id:
         # Try to load state from the chat ID
         if not context.bash_state.load_state_from_chat_id(chat_id):
-            return f"Error: No saved bash state found for chat ID {chat_id}"
+            return f"Error: No saved bash state found for chat ID {chat_id}. Please re-initialize to get a new id or use correct id."
     if mode_name:
         # update modes if they're relative
         if isinstance(change_mode, CodeWriterMode):
@@ -455,7 +458,11 @@ def read_image_from_shell(file_path: str, context: Context) -> ImageData:
 
 
 def get_context_for_errors(
-    errors: list[tuple[int, int]], file_content: str, max_tokens: Optional[int]
+    errors: list[tuple[int, int]],
+    file_content: str,
+    filename: str,
+    coding_max_tokens: Optional[int],
+    noncoding_max_tokens: Optional[int],
 ) -> str:
     file_lines = file_content.split("\n")
     min_line_num = max(0, min([error[0] for error in errors]) - 10)
@@ -463,6 +470,7 @@ def get_context_for_errors(
     context_lines = file_lines[min_line_num:max_line_num]
     context = "\n".join(context_lines)
 
+    max_tokens = select_max_tokens(filename, coding_max_tokens, noncoding_max_tokens)
     if max_tokens is not None and max_tokens > 0:
         ntokens = len(default_enc.encoder(context))
         if ntokens > max_tokens:
@@ -473,7 +481,8 @@ def get_context_for_errors(
 def write_file(
     writefile: WriteIfEmpty,
     error_on_exist: bool,
-    max_tokens: Optional[int],
+    coding_max_tokens: Optional[int],
+    noncoding_max_tokens: Optional[int],
     context: Context,
 ) -> tuple[
     str, dict[str, list[tuple[int, int]]]
@@ -536,7 +545,7 @@ def write_file(
                     msg = f"Error: you need to read existing file {path_} at least once before it can be overwritten.\n\n"
                     # Read the entire file
                     file_content_str, truncated, _, _, line_range = read_file(
-                        path_, max_tokens, context, False
+                        path_, coding_max_tokens, noncoding_max_tokens, context, False
                     )
                     file_ranges = [line_range]
 
@@ -558,7 +567,7 @@ def write_file(
                     msg = "Error: the file has changed since last read.\n\n"
                     # Read the entire file again
                     file_content_str, truncated, _, _, line_range = read_file(
-                        path_, max_tokens, context, False
+                        path_, coding_max_tokens, noncoding_max_tokens, context, False
                     )
                     file_ranges = [line_range]
 
@@ -591,7 +600,8 @@ def write_file(
                     )
                     readfiles, file_ranges_dict, truncated = read_files(
                         paths_readfiles.file_paths,
-                        max_tokens,
+                        coding_max_tokens,
+                        noncoding_max_tokens,
                         context,
                         show_line_numbers=False,
                         start_line_nums=paths_readfiles.start_line_nums,
@@ -631,7 +641,11 @@ def write_file(
                 syntax_errors += "\nNote: Ignore if 'tagged template literals' are used, they may raise false positive errors in tree-sitter."
 
             context_for_errors = get_context_for_errors(
-                check.errors, writefile.file_content, max_tokens
+                check.errors,
+                writefile.file_content,
+                path_,
+                coding_max_tokens,
+                noncoding_max_tokens,
             )
             context.console.print(f"W: Syntax errors encountered: {syntax_errors}")
             warnings.append(f"""
@@ -656,10 +670,13 @@ Syntax errors:
 
 
 def do_diff_edit(
-    fedit: FileEdit, max_tokens: Optional[int], context: Context
+    fedit: FileEdit,
+    coding_max_tokens: Optional[int],
+    noncoding_max_tokens: Optional[int],
+    context: Context,
 ) -> tuple[str, dict[str, list[tuple[int, int]]]]:
     try:
-        return _do_diff_edit(fedit, max_tokens, context)
+        return _do_diff_edit(fedit, coding_max_tokens, noncoding_max_tokens, context)
     except Exception as e:
         # Try replacing \"
         try:
@@ -669,14 +686,19 @@ def do_diff_edit(
                     '\\"', '"'
                 ),
             )
-            return _do_diff_edit(fedit, max_tokens, context)
+            return _do_diff_edit(
+                fedit, coding_max_tokens, noncoding_max_tokens, context
+            )
         except Exception:
             pass
         raise e
 
 
 def _do_diff_edit(
-    fedit: FileEdit, max_tokens: Optional[int], context: Context
+    fedit: FileEdit,
+    coding_max_tokens: Optional[int],
+    noncoding_max_tokens: Optional[int],
+    context: Context,
 ) -> tuple[str, dict[str, list[tuple[int, int]]]]:
     context.console.log(f"Editing file: {fedit.file_path}")
 
@@ -736,7 +758,11 @@ def _do_diff_edit(
         syntax_errors = check.description
         if syntax_errors:
             context_for_errors = get_context_for_errors(
-                check.errors, apply_diff_to, max_tokens
+                check.errors,
+                apply_diff_to,
+                path_,
+                coding_max_tokens,
+                noncoding_max_tokens,
             )
             if extension in {"tsx", "ts"}:
                 syntax_errors += "\nNote: Ignore if 'tagged template literals' are used, they may raise false positive errors in tree-sitter."
@@ -782,7 +808,8 @@ def _is_edit(content: str, percentage: int) -> bool:
 
 def file_writing(
     file_writing_args: FileWriteOrEdit,
-    max_tokens: Optional[int],
+    coding_max_tokens: Optional[int],
+    noncoding_max_tokens: Optional[int],
     context: Context,
 ) -> tuple[
     str, dict[str, list[tuple[int, int]]]
@@ -797,7 +824,7 @@ def file_writing(
         # Try to load state from the chat ID
         if not context.bash_state.load_state_from_chat_id(file_writing_args.chat_id):
             return (
-                f"Error: No saved bash state found for chat ID {file_writing_args.chat_id}. Please initialize first with this ID.",
+                f"Error: No saved bash state found for chat ID {file_writing_args.chat_id}. Please re-initialize to get a new id or use correct id.",
                 {},
             )
 
@@ -820,7 +847,8 @@ def file_writing(
                 file_content=file_writing_args.file_content_or_search_replace_blocks,
             ),
             True,
-            max_tokens,
+            coding_max_tokens,
+            noncoding_max_tokens,
             context,
         )
         return result, paths
@@ -831,7 +859,8 @@ def file_writing(
                 file_path=path_,
                 file_edit_using_search_replace_blocks=file_writing_args.file_content_or_search_replace_blocks,
             ),
-            max_tokens,
+            coding_max_tokens,
+            noncoding_max_tokens,
             context,
         )
         return result, paths
@@ -888,7 +917,8 @@ def get_tool_output(
     enc: EncoderDecoder[int],
     limit: float,
     loop_call: Callable[[str, float], tuple[str, float]],
-    max_tokens: Optional[int],
+    coding_max_tokens: Optional[int],
+    noncoding_max_tokens: Optional[int],
 ) -> tuple[list[str | ImageData], float]:
     global TOOL_CALLS
     if isinstance(args, dict):
@@ -906,13 +936,15 @@ def get_tool_output(
         context.console.print("Calling execute bash tool")
 
         output_str, cost = execute_bash(
-            context.bash_state, enc, arg, max_tokens, arg.wait_for_seconds
+            context.bash_state, enc, arg, noncoding_max_tokens, arg.wait_for_seconds
         )
         output = output_str, cost
     elif isinstance(arg, WriteIfEmpty):
         context.console.print("Calling write file tool")
 
-        result, write_paths = write_file(arg, True, max_tokens, context)
+        result, write_paths = write_file(
+            arg, True, coding_max_tokens, noncoding_max_tokens, context
+        )
         output = result, 0
         # Add write paths with their ranges to our tracking dictionary
         for path, ranges in write_paths.items():
@@ -923,7 +955,9 @@ def get_tool_output(
     elif isinstance(arg, FileEdit):
         context.console.print("Calling full file edit tool")
 
-        result, edit_paths = do_diff_edit(arg, max_tokens, context)
+        result, edit_paths = do_diff_edit(
+            arg, coding_max_tokens, noncoding_max_tokens, context
+        )
         output = result, 0.0
         # Add edit paths with their ranges to our tracking dictionary
         for path, ranges in edit_paths.items():
@@ -934,7 +968,9 @@ def get_tool_output(
     elif isinstance(arg, FileWriteOrEdit):
         context.console.print("Calling file writing tool")
 
-        result, write_edit_paths = file_writing(arg, max_tokens, context)
+        result, write_edit_paths = file_writing(
+            arg, coding_max_tokens, noncoding_max_tokens, context
+        )
         output = result, 0.0
         # Add write/edit paths with their ranges to our tracking dictionary
         for path, ranges in write_edit_paths.items():
@@ -951,7 +987,8 @@ def get_tool_output(
         # Access line numbers through properties
         result, file_ranges_dict, _ = read_files(
             arg.file_paths,
-            max_tokens,
+            coding_max_tokens,
+            noncoding_max_tokens,
             context,
             bool(arg.show_line_numbers_reason),
             arg.start_line_nums,
@@ -995,7 +1032,8 @@ def get_tool_output(
                 arg.any_workspace_path,
                 arg.initial_files_to_read,
                 arg.task_id_to_resume,
-                max_tokens,
+                coding_max_tokens,
+                noncoding_max_tokens,
                 arg.mode,
                 arg.chat_id,
             )
@@ -1024,7 +1062,9 @@ def get_tool_output(
             relevant_files.extend(globs[:1000])
             if not globs:
                 warnings += f"Warning: No files found for the glob: {fglob}\n"
-        relevant_files_data, _, _ = read_files(relevant_files[:10_000], None, context)
+        relevant_files_data, _, _ = read_files(
+            relevant_files[:10_000], None, None, context
+        )
         save_path = save_memory(
             arg, relevant_files_data, context.bash_state.serialize()
         )
@@ -1069,7 +1109,8 @@ def range_format(start_line_num: Optional[int], end_line_num: Optional[int]) ->
 
 def read_files(
     file_paths: list[str],
-    max_tokens: Optional[int],
+    coding_max_tokens: Optional[int],
+    noncoding_max_tokens: Optional[int],
     context: Context,
     show_line_numbers: bool = False,
     start_line_nums: Optional[list[Optional[int]]] = None,
@@ -1105,7 +1146,8 @@ def read_files(
             # if they weren't provided as parameters
             content, truncated, tokens, path, line_range = read_file(
                 file,
-                max_tokens,
+                coding_max_tokens,
+                noncoding_max_tokens,
                 context,
                 show_line_numbers,
                 start_line_num,
@@ -1121,13 +1163,20 @@ def read_files(
             message += f"\n{file}: {str(e)}\n"
             continue
 
-        if max_tokens:
-            max_tokens = max_tokens - tokens
+        if coding_max_tokens:
+            coding_max_tokens = coding_max_tokens - tokens
+        if noncoding_max_tokens:
+            noncoding_max_tokens = noncoding_max_tokens - tokens
 
         range_formatted = range_format(start_line_num, end_line_num)
         message += f"\n{file}{range_formatted}\n```\n{content}\n"
 
-        if truncated or (max_tokens and max_tokens <= 0):
+        # Check if we've hit either token limit
+        if (
+            truncated
+            or (coding_max_tokens is not None and coding_max_tokens <= 0)
+            or (noncoding_max_tokens is not None and noncoding_max_tokens <= 0)
+        ):
             not_reading = file_paths[i + 1 :]
             if not_reading:
                 message += f"\nNot reading the rest of the files: {', '.join(not_reading)} due to token limit, please call again"
@@ -1139,7 +1188,8 @@ def read_files(
 
 def read_file(
     file_path: str,
-    max_tokens: Optional[int],
+    coding_max_tokens: Optional[int],
+    noncoding_max_tokens: Optional[int],
     context: Context,
     show_line_numbers: bool = False,
     start_line_num: Optional[int] = None,
@@ -1201,6 +1251,9 @@ def read_file(
     truncated = False
     tokens_counts = 0
 
+    # Select the appropriate max_tokens based on file type
+    max_tokens = select_max_tokens(file_path, coding_max_tokens, noncoding_max_tokens)
+
     # Handle token limit if specified
     if max_tokens is not None:
         tokens = default_enc.encoder(content)
@@ -1220,7 +1273,10 @@ def read_file(
             content = truncated_content
             # Add informative message about truncation with total line count
             total_lines = len(all_lines)
-            content += f"\n(...truncated) Only showing till line number {last_line_shown} of {total_lines} total lines due to the token limit, please continue reading from {last_line_shown + 1} if required"
+            content += (
+                f"\n(...truncated) Only showing till line number {last_line_shown} of {total_lines} total lines due to the token limit, please continue reading from {last_line_shown + 1} if required"
+                f" using syntax {file_path}:{last_line_shown + 1}-{total_lines}"
+            )
             truncated = True
 
             # Update effective_end if truncated
@@ -1262,7 +1318,8 @@ if __name__ == "__main__":
                 default_enc,
                 0,
                 lambda x, y: ("", 0),
-                None,
+                24000,  # coding_max_tokens
+                8000,  # noncoding_max_tokens
             )
         )
         print(
@@ -1275,7 +1332,8 @@ if __name__ == "__main__":
                 default_enc,
                 0,
                 lambda x, y: ("", 0),
-                None,
+                24000,  # coding_max_tokens
+                8000,  # noncoding_max_tokens
             )
         )
 
@@ -1289,7 +1347,8 @@ if __name__ == "__main__":
                 default_enc,
                 0,
                 lambda x, y: ("", 0),
-                15000,
+                24000,  # coding_max_tokens
+                8000,  # noncoding_max_tokens
             )[0][0]
         )
 
@@ -1305,6 +1364,7 @@ if __name__ == "__main__":
                 default_enc,
                 0,
                 lambda x, y: ("", 0),
-                800,
+                24000,  # coding_max_tokens
+                8000,  # noncoding_max_tokens
             )[0][0]
         )

{wcgw-5.0.1.dist-info → wcgw-5.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wcgw
-Version: 5.0.1
+Version: 5.1.0
 Summary: Shell and coding agent for Claude and other mcp clients
 Project-URL: Homepage, https://github.com/rusiaaman/wcgw
 Author-email: Aman Rusia <gapypi@arcfu.com>
@@ -8,6 +8,7 @@ License-File: LICENSE
 Requires-Python: >=3.11
 Requires-Dist: anthropic>=0.39.0
 Requires-Dist: fastapi>=0.115.0
+Requires-Dist: mcp>=1.7.0
 Requires-Dist: openai>=1.46.0
 Requires-Dist: petname>=2.6
 Requires-Dist: pexpect>=4.9.0

wcgw-5.1.0.dist-info/RECORD

@@ -0,0 +1,37 @@
+wcgw/__init__.py,sha256=JgAY25VsA208v8E7QTIU0E50nsk-TCJ4FWTEHmnssYU,127
+wcgw/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wcgw/types_.py,sha256=y60Lv_uUA1_sGIfADLUKy7rFPTax8jxor5GGCDKBfZ0,7533
+wcgw/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wcgw/client/common.py,sha256=OCH7Tx64jojz3M3iONUrGMadE07W21DiZs5sOxWX1Qc,1456
+wcgw/client/diff-instructions.txt,sha256=eKRFA86yXWIGwNxIDaegTgTzIrFIBDWWiN1yP8Hf3i4,1685
+wcgw/client/memory.py,sha256=U2Nw2si3Zg7n_RhNAuaYcmrrDtZ_Mooi-kfAOKflT-I,3079
+wcgw/client/modes.py,sha256=roH6SPBokJMr5IzAlccdI-vJyvyS5vqSMMyth7TE86A,10315
+wcgw/client/tool_prompts.py,sha256=Qgj8aqEFeKilF9otDu6ZQkV0Ig4l6JaeM99uWu4ecDQ,4802
+wcgw/client/tools.py,sha256=lKGvkHBEBTrm80o6sFTNUv9VtMHEs-y4cT_mdkYjJNM,48276
+wcgw/client/bash_state/bash_state.py,sha256=GXBh9zcCqSgUCKn9ZtVLosfqOkJRwrzO-qzyETNHTeE,41744
+wcgw/client/bash_state/parser/__init__.py,sha256=AnlNSmoQTSoqqlLOLX4P1uXfzc5VGeCGJsGgtisq2zE,207
+wcgw/client/bash_state/parser/bash_statement_parser.py,sha256=9a8vPO1r3_tXmaAcubTQ5UY-NseWlalgm8LZA17LXuY,6058
+wcgw/client/encoder/__init__.py,sha256=Y-8f43I6gMssUCWpX5rLYiAFv3D-JPRs4uNEejPlke8,1514
+wcgw/client/file_ops/diff_edit.py,sha256=AwLq6-pY7czv1y-JA5O2Q4rgbvn82YmSL9jD8XB3Vo4,19019
+wcgw/client/file_ops/extensions.py,sha256=CmfD7ON6SY24Prh2tRZdV9KbhuOrWqqk8qL1VtshzB8,3608
+wcgw/client/file_ops/search_replace.py,sha256=bB1S-xkDdOX4h7UHHxfCHaQiS9KyYkIdU6UGvIrgwQM,6820
+wcgw/client/mcp_server/Readme.md,sha256=2Z88jj1mf9daYGW1CWaldcJ0moy8owDumhR2glBY3A8,109
+wcgw/client/mcp_server/__init__.py,sha256=mm7xhBIPwJpRT3u-Qsj4cKVMpVyucJoKRlbMP_gRRB0,343
+wcgw/client/mcp_server/server.py,sha256=jjwrmZZ8X0tXD0rsPZ9fKjEpdXpXCfdhEsN3Ho_tC8I,4989
+wcgw/client/repo_ops/display_tree.py,sha256=uOGX2IbXTKXwtXT2wdDszuH4ODmSYsHm0toU55e1vYI,4021
+wcgw/client/repo_ops/file_stats.py,sha256=AUA0Br7zFRpylWFYZPGMeGPJy3nWp9e2haKi34JptHE,4887
+wcgw/client/repo_ops/path_prob.py,sha256=SWf0CDn37rtlsYRQ51ufSxay-heaQoVIhr1alB9tZ4M,2144
+wcgw/client/repo_ops/paths_model.vocab,sha256=M1pXycYDQehMXtpp-qAgU7rtzeBbCOiJo4qcYFY0kqk,315087
+wcgw/client/repo_ops/paths_tokens.model,sha256=jiwwE4ae8ADKuTZISutXuM5Wfyc_FBmN5rxTjoNnCos,1569052
+wcgw/client/repo_ops/repo_context.py,sha256=e_w-1VfxWQiZT3r66N13nlmPt6AGm0uvG3A7aYSgaCI,9632
+wcgw_cli/__init__.py,sha256=TNxXsTPgb52OhakIda9wTRh91cqoBqgQRx5TxjzQQFU,21
+wcgw_cli/__main__.py,sha256=wcCrL4PjG51r5wVKqJhcoJPTLfHW0wNbD31DrUN0MWI,28
+wcgw_cli/anthropic_client.py,sha256=8bjDY59-aioyTJgpB-NBHZNhZaq6rqcTJcOf81kzCyA,19886
+wcgw_cli/cli.py,sha256=-7FBe_lahKyUOhf65iurTA1M1gXXXAiT0OVKQVcZKKo,948
+wcgw_cli/openai_client.py,sha256=GOqoSFazTV-cFjpdZGPM0DIwec8Up2TEcKUbsN40AGY,15990
+wcgw_cli/openai_utils.py,sha256=xGOb3W5ALrIozV7oszfGYztpj0FnXdD7jAxm5lEIVKY,2439
+wcgw-5.1.0.dist-info/METADATA,sha256=VUNkqbng5Y1O7WtFaJkjuCkTtUSSZBpCjv6aHGZcdq0,14866
+wcgw-5.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+wcgw-5.1.0.dist-info/entry_points.txt,sha256=UnjK-MAH4Qssh0tGJDMeij1oi-oRKokItkknP_BwShE,94
+wcgw-5.1.0.dist-info/licenses/LICENSE,sha256=BvY8xqjOfc3X2qZpGpX3MZEmF-4Dp0LqgKBbT6L_8oI,11142
+wcgw-5.1.0.dist-info/RECORD,,

wcgw_cli/anthropic_client.py

@@ -130,7 +130,8 @@ def loop(
         try:
             _, memory, _ = load_memory(
                 resume,
-                8000,
+                24000,  # coding_max_tokens
+                8000,   # noncoding_max_tokens
                 lambda x: default_enc.encoder(x),
                 lambda x: default_enc.decoder(x),
             )
@@ -197,7 +198,7 @@ def loop(
     tools = [
         ToolParam(
             name=tool.name,
-            description=tool.description,
+            description=tool.description or "",  # Ensure it's not None
             input_schema=tool.inputSchema,
         )
         for tool in TOOL_PROMPTS
@@ -223,8 +224,10 @@ def loop(
             os.getcwd(),
             [],
             resume if (memory and resume) else "",
-            max_tokens=8000,
+            24000,  # coding_max_tokens
+            8000,   # noncoding_max_tokens
             mode="wcgw",
+            chat_id="",
         )
 
         if history:
@@ -412,7 +415,8 @@ def loop(
                                         default_enc,
                                         limit - cost,
                                         loop,
-                                        max_tokens=8000,
+                                        24000,  # coding_max_tokens
+                                        8000,   # noncoding_max_tokens
                                     )
                                 except Exception as e:
                                     output_or_dones = [

wcgw_cli/openai_client.py

@@ -120,7 +120,8 @@ def loop(
         try:
             _, memory, _ = load_memory(
                 resume,
-                8000,
+                24000,  # coding_max_tokens
+                8000,   # noncoding_max_tokens
                 lambda x: default_enc.encoder(x),
                 lambda x: default_enc.decoder(x),
             )
@@ -187,8 +188,10 @@ def loop(
             os.getcwd(),
             [],
             resume if (memory and resume) else "",
-            max_tokens=8000,
+            24000,  # coding_max_tokens
+            8000,   # noncoding_max_tokens
             mode="wcgw",
+            chat_id="",
         )
 
         if not history:
@@ -283,7 +286,8 @@ def loop(
                                         enc,
                                         limit - cost,
                                         loop,
-                                        max_tokens=8000,
+                                        24000,  # coding_max_tokens
+                                        8000,   # noncoding_max_tokens
                                     )
                                     output_or_done = output_or_dones[0]
                                 except Exception as e:

mcp_wcgw/__init__.py

@@ -1,114 +0,0 @@
-from .client.session import ClientSession
-from .client.stdio import StdioServerParameters, stdio_client
-from .server.session import ServerSession
-from .server.stdio import stdio_server
-from .shared.exceptions import McpError
-from .types import (
-    CallToolRequest,
-    ClientCapabilities,
-    ClientNotification,
-    ClientRequest,
-    ClientResult,
-    CompleteRequest,
-    CreateMessageRequest,
-    CreateMessageResult,
-    ErrorData,
-    GetPromptRequest,
-    GetPromptResult,
-    Implementation,
-    IncludeContext,
-    InitializedNotification,
-    InitializeRequest,
-    InitializeResult,
-    JSONRPCError,
-    JSONRPCRequest,
-    JSONRPCResponse,
-    ListPromptsRequest,
-    ListPromptsResult,
-    ListResourcesRequest,
-    ListResourcesResult,
-    ListToolsResult,
-    LoggingLevel,
-    LoggingMessageNotification,
-    Notification,
-    PingRequest,
-    ProgressNotification,
-    PromptsCapability,
-    ReadResourceRequest,
-    ReadResourceResult,
-    Resource,
-    ResourcesCapability,
-    ResourceUpdatedNotification,
-    RootsCapability,
-    SamplingMessage,
-    ServerCapabilities,
-    ServerNotification,
-    ServerRequest,
-    ServerResult,
-    SetLevelRequest,
-    StopReason,
-    SubscribeRequest,
-    Tool,
-    ToolsCapability,
-    UnsubscribeRequest,
-)
-from .types import (
-    Role as SamplingRole,
-)
-
-__all__ = [
-    "CallToolRequest",
-    "ClientCapabilities",
-    "ClientNotification",
-    "ClientRequest",
-    "ClientResult",
-    "ClientSession",
-    "CreateMessageRequest",
-    "CreateMessageResult",
-    "ErrorData",
-    "GetPromptRequest",
-    "GetPromptResult",
-    "Implementation",
-    "IncludeContext",
-    "InitializeRequest",
-    "InitializeResult",
-    "InitializedNotification",
-    "JSONRPCError",
-    "JSONRPCRequest",
-    "ListPromptsRequest",
-    "ListPromptsResult",
-    "ListResourcesRequest",
-    "ListResourcesResult",
-    "ListToolsResult",
-    "LoggingLevel",
-    "LoggingMessageNotification",
-    "McpError",
-    "Notification",
-    "PingRequest",
-    "ProgressNotification",
-    "PromptsCapability",
-    "ReadResourceRequest",
-    "ReadResourceResult",
-    "ResourcesCapability",
-    "ResourceUpdatedNotification",
-    "Resource",
-    "RootsCapability",
-    "SamplingMessage",
-    "SamplingRole",
-    "ServerCapabilities",
-    "ServerNotification",
-    "ServerRequest",
-    "ServerResult",
-    "ServerSession",
-    "SetLevelRequest",
-    "StdioServerParameters",
-    "StopReason",
-    "SubscribeRequest",
-    "Tool",
-    "ToolsCapability",
-    "UnsubscribeRequest",
-    "stdio_client",
-    "stdio_server",
-    "CompleteRequest",
-    "JSONRPCResponse",
-]