mojentic 0.8.4__py3-none-any.whl → 1.0.0__py3-none-any.whl

mojentic/llm/tools/file_manager.py

@@ -1,7 +1,6 @@
 import os
 import re
 import glob
-import difflib
 
 from mojentic.llm.tools.llm_tool import LLMTool
 
@@ -93,7 +92,6 @@ class FilesystemGateway:
 
         return matching_lines
 
-
     def edit_file_with_diff(self, path: str, file_name: str, diff: str) -> str:
         """Edit a file by applying a diff to it."""
         resolved_path = self._resolve_path(path)
@@ -150,14 +148,14 @@ class FilesystemGateway:
                     match = re.match(r'@@ -(\d+) \+(\d+) @@', line)
                     if match:
                         start_line = int(match.group(1))
-                        start_line_new = int(match.group(2))
+                        int(match.group(2))
                     else:
                         # Skip invalid hunk header
                         i += 1
                         continue
                 else:
                     start_line = int(match.group(1))
-                    start_line_new = int(match.group(3))
+                    int(match.group(3))
 
                 # Line numbers in diff are 1-based, but our array is 0-based
                 start_line -= 1
@@ -308,17 +306,29 @@ class ListFilesTool(LLMTool):
             "type": "function",
             "function": {
                 "name": "list_files",
-                "description": "List files in the specified directory (non-recursive), optionally filtered by extension. Use this when you need to see what files are available in a specific directory without including subdirectories.",
+                "description": (
+                    "List files in the specified directory (non-recursive), optionally filtered by extension. "
+                    "Use this when you need to see what files are available in a specific directory without "
+                    "including subdirectories."
+                ),
                 "parameters": {
                     "type": "object",
                     "properties": {
                         "path": {
                             "type": "string",
-                            "description": "The path relative to the sandbox root to list files from. For example, '.' for the root directory, 'src' for the src directory, or 'docs/images' for a nested directory."
+                            "description": (
+                                "The path relative to the sandbox root to list files from. For example, '.' for "
+                                "the root directory, 'src' for the src directory, or 'docs/images' for a nested "
+                                "directory."
+                            )
                         },
                         "extension": {
                             "type": "string",
-                            "description": "The file extension to filter by (e.g., '.py', '.txt', '.md'). If not provided, all files will be listed. For example, using '.py' will only list Python files in the directory."
+                            "description": (
+                                "The file extension to filter by (e.g., '.py', '.txt', '.md'). If not provided, "
+                                "all files will be listed. For example, using '.py' will only list Python files "
+                                "in the directory."
+                            )
                         }
                     },
                     "additionalProperties": False,
@@ -356,13 +366,20 @@ class ReadFileTool(LLMTool):
             "type": "function",
             "function": {
                 "name": "read_file",
-                "description": "Read the entire content of a file as a string. Use this when you need to access or analyze the complete contents of a file.",
+                "description": (
+                    "Read the entire content of a file as a string. Use this when you need to access or "
+                    "analyze the complete contents of a file."
+                ),
                 "parameters": {
                     "type": "object",
                     "properties": {
                         "path": {
                             "type": "string",
-                            "description": "The full relative path including the filename of the file to read. For example, 'README.md' for a file in the root directory, 'src/main.py' for a file in the src directory, or 'docs/images/diagram.png' for a file in a nested directory."
+                            "description": (
+                                "The full relative path including the filename of the file to read. For example, "
+                                "'README.md' for a file in the root directory, 'src/main.py' for a file in the "
+                                "src directory, or 'docs/images/diagram.png' for a file in a nested directory."
+                            )
                         }
                     },
                     "additionalProperties": False,
@@ -399,17 +416,29 @@ class WriteFileTool(LLMTool):
             "type": "function",
             "function": {
                 "name": "write_file",
-                "description": "Write content to a file, completely overwriting any existing content. Use this when you want to replace the entire contents of a file with new content.",
+                "description": (
+                    "Write content to a file, completely overwriting any existing content. Use this when you "
+                    "want to replace the entire contents of a file with new content."
+                ),
                 "parameters": {
                     "type": "object",
                     "properties": {
                         "path": {
                             "type": "string",
-                            "description": "The full relative path including the filename where the file should be written. For example, 'output.txt' for a file in the root directory, 'src/main.py' for a file in the src directory, or 'docs/images/diagram.png' for a file in a nested directory."
+                            "description": (
+                                "The full relative path including the filename where the file should be written. "
+                                "For example, 'output.txt' for a file in the root directory, 'src/main.py' for "
+                                "a file in the src directory, or 'docs/images/diagram.png' for a file in a "
+                                "nested directory."
+                            )
                         },
                         "content": {
                             "type": "string",
-                            "description": "The content to write to the file. This will completely replace any existing content in the file. For example, 'Hello, world!' for a simple text file, or a JSON string for a configuration file."
+                            "description": (
+                                "The content to write to the file. This will completely replace any existing "
+                                "content in the file. For example, 'Hello, world!' for a simple text file, or a "
+                                "JSON string for a configuration file."
+                            )
                         }
                     },
                     "additionalProperties": False,
@@ -441,13 +470,22 @@ class ListAllFilesTool(LLMTool):
             "type": "function",
             "function": {
                 "name": "list_all_files",
-                "description": "List all files recursively in the specified directory, including files in subdirectories. Use this when you need a complete inventory of all files in a directory and its subdirectories.",
+                "description": (
+                    "List all files recursively in the specified directory, including files in "
+                    "subdirectories. Use this when you need a complete inventory of all files in a "
+                    "directory and its subdirectories."
+                ),
                 "parameters": {
                     "type": "object",
                     "properties": {
                         "path": {
                             "type": "string",
-                            "description": "The path relative to the sandbox root to list files from recursively. For example, '.' for the root directory and all subdirectories, 'src' for the src directory and all its subdirectories, or 'docs/images' for a nested directory and its subdirectories."
+                            "description": (
+                                "The path relative to the sandbox root to list files from recursively. For "
+                                "example, '.' for the root directory and all subdirectories, 'src' for the src "
+                                "directory and all its subdirectories, or 'docs/images' for a nested directory "
+                                "and its subdirectories."
+                            )
                         }
                     },
                     "additionalProperties": False,
@@ -479,17 +517,30 @@ class FindFilesByGlobTool(LLMTool):
             "type": "function",
             "function": {
                 "name": "find_files_by_glob",
-                "description": "Find files matching a glob pattern in the specified directory. Use this when you need to locate files with specific patterns in their names or paths (e.g., all Python files with '*.py' or all text files in any subdirectory with '**/*.txt').",
+                "description": (
+                    "Find files matching a glob pattern in the specified directory. Use this when you need to "
+                    "locate files with specific patterns in their names or paths (e.g., all Python files with "
+                    "'*.py' or all text files in any subdirectory with '**/*.txt')."
+                ),
                 "parameters": {
                     "type": "object",
                     "properties": {
                         "path": {
                             "type": "string",
-                            "description": "The path relative to the sandbox root to search in. For example, '.' for the root directory, 'src' for the src directory, or 'docs/images' for a nested directory."
+                            "description": (
+                                "The path relative to the sandbox root to search for files from. For example, "
+                                "'.' for the root directory, 'src' for the src directory, or 'docs/images' for a "
+                                "nested directory."
+                            )
                         },
                         "pattern": {
                             "type": "string",
-                            "description": "The glob pattern to match files against. Examples: '*.py' for all Python files in the specified directory, '**/*.txt' for all text files in the specified directory and any subdirectory, or '**/*test*.py' for all Python files with 'test' in their name in the specified directory and any subdirectory."
+                            "description": (
+                                "The glob pattern to match files against. Examples: '*.py' for all Python files in "
+                                "the specified directory, '**/*.txt' for all text files in the specified directory "
+                                "and any subdirectory, or '**/*test*.py' for all Python files with 'test' in "
+                                "their name in the specified directory and any subdirectory."
+                            )
                         }
                     },
                     "additionalProperties": False,
@@ -523,17 +574,30 @@ class FindFilesContainingTool(LLMTool):
             "type": "function",
             "function": {
                 "name": "find_files_containing",
-                "description": "Find files containing text matching a regex pattern in the specified directory. Use this when you need to search for specific content across multiple files, such as finding all files that contain a particular function name or text string.",
+                "description": (
+                    "Find files containing text matching a regex pattern in the specified directory. Use "
+                    "this when you need to search for specific content across multiple files, such as "
+                    "finding all files that contain a particular function name or text string."
+                ),
                 "parameters": {
                     "type": "object",
                     "properties": {
                         "path": {
                             "type": "string",
-                            "description": "The path relative to the sandbox root to search in. For example, '.' for the root directory, 'src' for the src directory, or 'docs/images' for a nested directory."
+                            "description": (
+                                "The path relative to the sandbox root to search in. For example, '.' for the "
+                                "root directory, 'src' for the src directory, or 'docs/images' for a nested "
+                                "directory."
+                            )
                         },
                         "pattern": {
                             "type": "string",
-                            "description": "The regex pattern to search for in files. Examples: 'function\\s+main' to find files containing a main function, 'import\\s+os' to find files importing the os module, or 'TODO|FIXME' to find files containing TODO or FIXME comments. The pattern uses Python's re module syntax."
+                            "description": (
+                                "The regex pattern to search for in files. Examples: 'function\\s+main' to find "
+                                "files containing a main function, 'import\\s+os' to find files importing the os "
+                                "module, or 'TODO|FIXME' to find files containing TODO or FIXME comments. The "
+                                "pattern uses Python's re module syntax."
+                            )
                         }
                     },
                     "additionalProperties": False,
@@ -573,17 +637,31 @@ class FindLinesMatchingTool(LLMTool):
             "type": "function",
             "function": {
                 "name": "find_lines_matching",
-                "description": "Find all lines in a file matching a regex pattern, returning both line numbers and content. Use this when you need to locate specific patterns within a single file and need to know exactly where they appear.",
+                "description": (
+                    "Find all lines in a file matching a regex pattern, returning both line numbers and "
+                    "content. Use this when you need to locate specific patterns within a single file and "
+                    "need to know exactly where they appear."
+                ),
                 "parameters": {
                     "type": "object",
                     "properties": {
                         "path": {
                             "type": "string",
-                            "description": "The full relative path including the filename of the file to search in. For example, 'README.md' for a file in the root directory, 'src/main.py' for a file in the src directory, or 'docs/images/diagram.png' for a file in a nested directory."
+                            "description": (
+                                "The full relative path including the filename of the file to search. For "
+                                "example, 'README.md' for a file in the root directory, 'src/main.py' for a file "
+                                "in the src directory, or 'docs/images/diagram.png' for a file in a nested "
+                                "directory."
+                            )
                         },
                         "pattern": {
                             "type": "string",
-                            "description": "The regex pattern to match lines against. Examples: 'def\\s+\\w+' to find all function definitions, 'class\\s+\\w+' to find all class definitions, or 'TODO|FIXME' to find all TODO or FIXME comments. The pattern uses Python's re module syntax."
+                            "description": (
+                                "The regex pattern to match lines against. Examples: 'def\\s+\\w+' to find all "
+                                "function definitions, 'class\\s+\\w+' to find all class definitions, or "
+                                "'TODO|FIXME' to find all TODO or FIXME comments. The pattern uses Python's re "
+                                "module syntax."
+                            )
                         }
                     },
                     "additionalProperties": False,
@@ -621,17 +699,33 @@ class EditFileWithDiffTool(LLMTool):
             "type": "function",
             "function": {
                 "name": "edit_file_with_diff",
-                "description": "Edit a file by applying a diff to it. Use this for making selective changes to parts of a file while preserving the rest of the content, unlike write_file which completely replaces the file. The diff should be in a unified diff format with lines prefixed by '+' (add), '-' (remove), or ' ' (context).",
+                "description": (
+                    "Edit a file by applying a diff to it. Use this for making selective changes to parts "
+                    "of a file while preserving the rest of the content, unlike write_file which completely "
+                    "replaces the file. The diff should be in a unified diff format with lines prefixed by "
+                    "'+' (add), '-' (remove), or ' ' (context)."
+                ),
                 "parameters": {
                     "type": "object",
                     "properties": {
                         "path": {
                             "type": "string",
-                            "description": "The full relative path including the filename of the file to edit. For example, 'README.md' for a file in the root directory, 'src/main.py' for a file in the src directory, or 'docs/images/diagram.png' for a file in a nested directory."
+                            "description": (
+                                "The full relative path including the filename of the file to edit. For example, "
+                                "'README.md' for a file in the root directory, 'src/main.py' for a file in the "
+                                "src directory, or 'docs/images/diagram.png' for a file in a nested directory."
+                            )
                         },
                         "diff": {
                             "type": "string",
-                            "description": "The diff to apply to the file in unified diff format. Lines to add should be prefixed with '+', lines to remove with '-', and context lines with ' ' (space). Example:\n\n```\n This is a context line (unchanged)\n-This line will be removed\n+This line will be added\n This is another context line\n```\n\nThe diff should include enough context lines to uniquely identify the section of the file to modify."
+                            "description": (
+                                "The diff to apply to the file in unified diff format. Lines to add should be "
+                                "prefixed with '+', lines to remove with '-', and context lines with ' ' (space). "
+                                "Example:\n\n```\n This is a context line (unchanged)\n-This line will be "
+                                "removed\n+This line will be added\n This is another context line\n```\n\n"
+                                "The diff should include enough context lines to uniquely identify the section "
+                                "of the file to modify."
+                            )
                         }
                     },
                     "additionalProperties": False,
@@ -669,13 +763,22 @@ class CreateDirectoryTool(LLMTool):
             "type": "function",
             "function": {
                 "name": "create_directory",
-                "description": "Create a new directory at the specified path. If the directory already exists, this operation will succeed without error. Use this when you need to create a directory structure before writing files to it.",
+                "description": (
+                    "Create a new directory at the specified path. If the directory already exists, this "
+                    "operation will succeed without error. Use this when you need to create a directory "
+                    "structure before writing files to it."
+                ),
                 "parameters": {
                     "type": "object",
                     "properties": {
                         "path": {
                             "type": "string",
-                            "description": "The relative path where the directory should be created. For example, 'new_folder' for a directory in the root, 'src/new_folder' for a directory in the src directory, or 'docs/images/new_folder' for a nested directory. Parent directories will be created automatically if they don't exist."
+                            "description": (
+                                "The relative path where the directory should be created. For example, "
+                                "'new_folder' for a directory in the root, 'src/new_folder' for a directory in "
+                                "the src directory, or 'docs/images/new_folder' for a nested directory. Parent "
+                                "directories will be created automatically if they don't exist."
+                            )
                         }
                     },
                     "additionalProperties": False,

mojentic/llm/tools/file_manager_spec.py

@@ -1,9 +1,6 @@
 import os
-import re
-import glob
 import tempfile
 import shutil
-from unittest.mock import Mock, patch
 
 import pytest
 

mojentic/llm/tools/llm_tool.py

@@ -1,5 +1,5 @@
 import json
-from typing import Any, Dict, Optional
+from typing import Optional
 
 from mojentic.tracer.tracer_system import TracerSystem
 from mojentic.llm.gateways.models import TextContent

mojentic/llm/tools/llm_tool_spec.py

@@ -1,9 +1,7 @@
-import json
 from unittest.mock import Mock
 
 import pytest
 
-from mojentic.llm.gateways.models import TextContent
 from mojentic.llm.tools.llm_tool import LLMTool
 
 

mojentic/llm/tools/organic_web_search.py

@@ -21,8 +21,10 @@ class OrganicWebSearchTool(LLMTool):
             "type": "function",
             "function": {
                 "name": "organic_web_search",
-                "description":
-                    "Search the Internet for information matching the given query and return the organic search results.",
+                "description": (
+                    "Search the Internet for information matching the given query "
+                    "and return the organic search results."
+                ),
                 "parameters": {
                     "type": "object",
                     "properties": {

mojentic/llm/tools/tell_user_tool.py

@@ -12,7 +12,11 @@ class TellUserTool(LLMTool):
             "type": "function",
             "function": {
                 "name": "tell_user",
-                "description": "Display a message to the user without expecting a response. Use this to send important intermediate information to the user as you work on completing their request.",
+                "description": (
+                    "Display a message to the user without expecting a response. Use this to send "
+                    "important intermediate information to the user as you work on completing their "
+                    "request."
+                ),
                 "parameters": {
                     "type": "object",
                     "properties": {
@@ -24,4 +28,4 @@ class TellUserTool(LLMTool):
                     "required": ["message"]
                 },
             },
-        }
+        }

mojentic/llm/tools/tool_wrapper.py

@@ -1,10 +1,10 @@
 """
 Not sure about this module right now. There are a couple ways to do this.
 """
-from typing import Any, TYPE_CHECKING
+from typing import TYPE_CHECKING
 
 from mojentic.llm.tools.llm_tool import LLMTool
-from mojentic.llm.gateways.models import LLMMessage, MessageRole
+from mojentic.llm.gateways.models import LLMMessage
 
 # Use TYPE_CHECKING to avoid circular imports
 if TYPE_CHECKING:

mojentic/tracer/__init__.py

@@ -4,18 +4,9 @@ Mojentic tracer module for tracking system operations.
 """
 
 # Core tracer components
-from .tracer_system import TracerSystem
 from .null_tracer import NullTracer
-from .event_store import EventStore
 
 # Tracer event types
-from .tracer_events import (
-    TracerEvent,
-    LLMCallTracerEvent,
-    LLMResponseTracerEvent,
-    ToolCallTracerEvent,
-    AgentInteractionTracerEvent
-)
 
 # Create a singleton NullTracer instance for use throughout the application
-null_tracer = NullTracer()
+null_tracer = NullTracer()

mojentic/tracer/event_store.py

@@ -1,6 +1,4 @@
-from typing import Any, Callable, Dict, List, Optional, Type, Union
-import time
-from datetime import datetime
+from typing import Callable, List, Optional, Type
 
 from mojentic.event import Event
 from mojentic.tracer.tracer_events import TracerEvent
@@ -38,11 +36,12 @@ class EventStore:
         if self.on_store_callback is not None:
             self.on_store_callback(event)
 
-    def get_events(self, 
-                  event_type: Optional[Type[Event]] = None, 
-                  start_time: Optional[float] = None,
-                  end_time: Optional[float] = None,
-                  filter_func: Optional[Callable[[Event], bool]] = None) -> List[Event]:
+    def get_events(
+            self,
+            event_type: Optional[Type[Event]] = None,
+            start_time: Optional[float] = None,
+            end_time: Optional[float] = None,
+            filter_func: Optional[Callable[[Event], bool]] = None) -> List[Event]:
         """
         Get events from the store, optionally filtered by type, time range, and custom filter function.
 

mojentic/tracer/event_store_spec.py

@@ -1,8 +1,7 @@
 import time
-from typing import List
 
 from mojentic import Event
-from mojentic.tracer.tracer_events import TracerEvent, LLMCallTracerEvent, AgentInteractionTracerEvent
+from mojentic.tracer.tracer_events import TracerEvent
 from mojentic.tracer.event_store import EventStore
 
 

mojentic/tracer/null_tracer.py

@@ -32,15 +32,14 @@ class NullTracer:
             The tracer event to record (will be ignored).
         """
         # Do nothing
-        pass
-
-    def record_llm_call(self, 
-                      model: str, 
-                      messages: List[Dict], 
-                      temperature: float = 1.0,
-                      tools: Optional[List[Dict]] = None,
-                      source: Any = None,
-                      correlation_id: str = None) -> None:
+
+    def record_llm_call(self,
+                        model: str,
+                        messages: List[Dict],
+                        temperature: float = 1.0,
+                        tools: Optional[List[Dict]] = None,
+                        source: Any = None,
+                        correlation_id: str = None) -> None:
         """
         Do nothing implementation of record_llm_call.
 
@@ -60,15 +59,14 @@ class NullTracer:
             UUID string that is copied from cause-to-affect for tracing events.
         """
         # Do nothing
-        pass
 
-    def record_llm_response(self, 
-                         model: str,
-                         content: str,
-                         tool_calls: Optional[List[Dict]] = None,
-                         call_duration_ms: Optional[float] = None,
-                         source: Any = None,
-                         correlation_id: str = None) -> None:
+    def record_llm_response(self,
+                            model: str,
+                            content: str,
+                            tool_calls: Optional[List[Dict]] = None,
+                            call_duration_ms: Optional[float] = None,
+                            source: Any = None,
+                            correlation_id: str = None) -> None:
         """
         Do nothing implementation of record_llm_response.
 
@@ -88,16 +86,15 @@ class NullTracer:
             UUID string that is copied from cause-to-affect for tracing events.
         """
         # Do nothing
-        pass
 
     def record_tool_call(self,
-                       tool_name: str,
-                       arguments: Dict[str, Any],
-                       result: Any,
-                       caller: Optional[str] = None,
-                       call_duration_ms: Optional[float] = None,
-                       source: Any = None,
-                       correlation_id: str = None) -> None:
+                         tool_name: str,
+                         arguments: Dict[str, Any],
+                         result: Any,
+                         caller: Optional[str] = None,
+                         call_duration_ms: Optional[float] = None,
+                         source: Any = None,
+                         correlation_id: str = None) -> None:
         """
         Do nothing implementation of record_tool_call.
 
@@ -119,15 +116,15 @@ class NullTracer:
             UUID string that is copied from cause-to-affect for tracing events.
         """
         # Do nothing
-        pass
-
-    def record_agent_interaction(self,
-                               from_agent: str,
-                               to_agent: str,
-                               event_type: str,
-                               event_id: Optional[str] = None,
-                               source: Any = None,
-                               correlation_id: str = None) -> None:
+
+    def record_agent_interaction(
+            self,
+            from_agent: str,
+            to_agent: str,
+            event_type: str,
+            event_id: Optional[str] = None,
+            source: Any = None,
+            correlation_id: str = None) -> None:
         """
         Do nothing implementation of record_agent_interaction.
 
@@ -147,13 +144,13 @@ class NullTracer:
             UUID string that is copied from cause-to-affect for tracing events.
         """
         # Do nothing
-        pass
 
-    def get_events(self, 
-                  event_type: Optional[Type[TracerEvent]] = None, 
-                  start_time: Optional[float] = None,
-                  end_time: Optional[float] = None,
-                  filter_func: Optional[Callable[[TracerEvent], bool]] = None) -> List[TracerEvent]:
+    def get_events(
+            self,
+            event_type: Optional[Type[TracerEvent]] = None,
+            start_time: Optional[float] = None,
+            end_time: Optional[float] = None,
+            filter_func: Optional[Callable[[TracerEvent], bool]] = None) -> List[TracerEvent]:
         """
         Return an empty list for any get_events request.
 
@@ -195,12 +192,9 @@ class NullTracer:
 
     def clear(self) -> None:
         """Do nothing implementation of clear method."""
-        pass
 
     def enable(self) -> None:
         """No-op method for interface compatibility."""
-        pass
 
     def disable(self) -> None:
         """No-op method for interface compatibility."""
-        pass

mojentic/tracer/tracer_events.py

@@ -1,7 +1,7 @@
 """
 Defines tracer event types for tracking system interactions.
 """
-from typing import Any, Dict, List, Optional, Type
+from typing import Any, Dict, List, Optional
 from datetime import datetime
 import uuid
 
@@ -18,7 +18,13 @@ class TracerEvent(Event):
     They are distinct from regular events which are used for agent communication.
     """
     timestamp: float = Field(..., description="Timestamp when the event occurred")
-    correlation_id: str = Field(default_factory=lambda: str(uuid.uuid4()), description="UUID string that is copied from cause-to-affect for tracing events")
+    correlation_id: str = Field(
+        default_factory=lambda: str(uuid.uuid4()),
+        description=(
+            "UUID string that is copied from cause-to-affect "
+            "for tracing events"
+        )
+    )
 
     def printable_summary(self) -> str:
         """