enh: logging

Author: tjbck

src/mcpo/main.py

@@ -222,7 +222,7 @@ async def run(
         try:
             headers = json.loads(headers)
         except json.JSONDecodeError:
-            print("Warning: Invalid JSON format for headers. Headers will be ignored.")
+            logger.warning("Invalid JSON format for headers. Headers will be ignored.")
             headers = None
 
     if server_type == "sse":

src/mcpo/utils/main.py

@@ -1,7 +1,7 @@
 import json
 import traceback
 from typing import Any, Dict, ForwardRef, List, Optional, Type, Union
-
+import logging
 from fastapi import HTTPException
 
 from mcp import ClientSession, types
@@ -27,6 +27,8 @@
     INTERNAL_ERROR: 500,
 }
 
+logger = logging.getLogger(__name__)
+
 
 def process_tool_response(result: CallToolResult) -> list:
     """Universal response processor for all tool endpoints"""
@@ -51,7 +53,7 @@ def process_tool_response(result: CallToolResult) -> list:
 
 def name_needs_alias(name: str) -> bool:
     """Check if a field name needs aliasing (for now if it starts with '__')."""
-    return name.startswith('__')
+    return name.startswith("__")
 
 
 def generate_alias_name(original_name: str, existing_names: set) -> str:
@@ -65,7 +67,7 @@ def generate_alias_name(original_name: str, existing_names: set) -> str:
     Returns:
         An alias name that doesn't conflict with existing names
     """
-    alias_name = original_name.lstrip('_')
+    alias_name = original_name.lstrip("_")
     # Handle potential naming conflicts
     original_alias_name = alias_name
     suffix_counter = 1
@@ -171,12 +173,14 @@ def _process_schema_property(
             )
 
             if name_needs_alias(name):
-                other_names = set().union(nested_properties, nested_fields, _model_cache)
+                other_names = set().union(
+                    nested_properties, nested_fields, _model_cache
+                )
                 alias_name = generate_alias_name(name, other_names)
                 aliased_field = Field(
                     default=nested_pydantic_field.default,
                     description=nested_pydantic_field.description,
-                    alias=name
+                    alias=name,
                 )
                 nested_fields[alias_name] = (nested_type_hint, aliased_field)
             else:
@@ -245,7 +249,7 @@ def get_model_fields(form_model_name, properties, required_fields, schema_defs=N
             aliased_field = Field(
                 default=pydantic_field_info.default,
                 description=pydantic_field_info.description,
-                alias=param_name
+                alias=param_name,
             )
             # Use the generated type hint and Field info
             model_fields[alias_name] = (python_type_hint, aliased_field)
@@ -274,7 +278,7 @@ def make_endpoint_func(
         ):  # Parameterized endpoint
             async def tool(form_data: FormModel) -> ResponseModel:
                 args = form_data.model_dump(exclude_none=True, by_alias=True)
-                print(f"Calling endpoint: {endpoint_name}, with args: {args}")
+                logger.info(f"Calling endpoint: {endpoint_name}, with args: {args}")
                 try:
                     result = await session.call_tool(endpoint_name, arguments=args)
 
@@ -299,7 +303,7 @@ async def tool(form_data: FormModel) -> ResponseModel:
                     return final_response
 
                 except McpError as e:
-                    print(
+                    logger.info(
                         f"MCP Error calling {endpoint_name}: {traceback.format_exc()}"
                     )
                     status_code = MCP_ERROR_TO_HTTP_STATUS.get(e.error.code, 500)
@@ -312,7 +316,7 @@ async def tool(form_data: FormModel) -> ResponseModel:
                         ),
                     )
                 except Exception as e:
-                    print(
+                    logger.info(
                         f"Unexpected error calling {endpoint_name}: {traceback.format_exc()}"
                     )
                     raise HTTPException(
@@ -329,7 +333,7 @@ def make_endpoint_func_no_args(
             endpoint_name: str, session: ClientSession
         ):  # Parameterless endpoint
             async def tool():  # No parameters
-                print(f"Calling endpoint: {endpoint_name}, with no args")
+                logger.info(f"Calling endpoint: {endpoint_name}, with no args")
                 try:
                     result = await session.call_tool(
                         endpoint_name, arguments={}
@@ -353,7 +357,7 @@ async def tool():  # No parameters
                     return final_response
 
                 except McpError as e:
-                    print(
+                    logger.info(
                         f"MCP Error calling {endpoint_name}: {traceback.format_exc()}"
                     )
                     status_code = MCP_ERROR_TO_HTTP_STATUS.get(e.error.code, 500)
@@ -367,7 +371,7 @@ async def tool():  # No parameters
                         ),
                     )
                 except Exception as e:
-                    print(
+                    logger.info(
                         f"Unexpected error calling {endpoint_name}: {traceback.format_exc()}"
                     )
                     raise HTTPException(