misc: small fix or general refactoring i did not bother commenting

Author: evilsocket

nerve/generation/litellm.py

@@ -44,6 +44,8 @@ def __init__(
         self.is_ollama = "ollama" in self.generator_id
         self.reduced_window_size = 25
 
+        litellm.drop_params = True
+
         if not self.is_ollama:
             if self.tools and not litellm.supports_function_calling(model=self.generator_id):  # type: ignore
                 logger.warning(

nerve/runtime/logging.py

@@ -24,7 +24,7 @@ def init(
         log_path: The path to the log file.
         level: The log level to use.
         litellm_debug: Whether to enable litellm debug logging.
-        litellm_tracing: The callback to set for litellm tracing.        
+        litellm_tracing: The callback to set for litellm tracing.
     """
 
     if level != "DEBUG":
@@ -43,7 +43,7 @@ def init(
         litellm.suppress_debug_info = True
 
     if litellm_tracing:
-        litellm.callbacks = [litellm_tracing]
+        litellm.callbacks = [litellm_tracing]  # type: ignore
 
     if log_path:
         logger.add(log_path, format=format, level=level)

nerve/server/mcp_test.py

@@ -31,6 +31,7 @@ async def test_mcp_create_server_for_agent_adds_agent_as_tool(self, mock_runner:
             debug=False,
             log_path=None,
             litellm_debug=False,
+            litellm_tracing=None,
             start_state={},
             trace=None,
             task=None,
@@ -95,6 +96,7 @@ async def test_create_mcp_server_for_agent_uses_task_as_description_when_descrip
             debug=False,
             log_path=None,
             litellm_debug=False,
+            litellm_tracing=None,
             start_state={},
             trace=None,
             task=None,