janito 2.5.0__tar.gz → 2.5.1__tar.gz

{janito-2.5.0 → janito-2.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: janito
-Version: 2.5.0
+Version: 2.5.1
 Summary: A new Python package called janito.
 Author-email: João Pinto <lamego.pinto@gmail.com>
 Project-URL: Homepage, https://github.com/janito-dev/janito

{janito-2.5.0 → janito-2.5.1}/janito/agent/setup_agent.py

@@ -156,6 +156,7 @@ def setup_agent(
     agent.template_vars["profile"] = profile
     # Store template path and context for dynamic prompt refresh
     agent.system_prompt_template = str(template_path)
+    
     agent._template_vars = context.copy()
     agent._original_template_vars = context.copy()
     return agent

{janito-2.5.0 → janito-2.5.1}/janito/cli/chat_mode/shell/commands/conversation_restart.py

@@ -17,8 +17,7 @@ def handle_restart(shell_state=None):
         # Reset system prompt to original template context if available
         if hasattr(shell_state.agent, "_original_template_vars"):
             shell_state.agent._template_vars = shell_state.agent._original_template_vars.copy()
-        if hasattr(shell_state.agent, "_refresh_system_prompt_from_template"):
-            shell_state.agent._refresh_system_prompt_from_template()
+        shell_state.agent.refresh_system_prompt_from_template()
         # No need to print the system prompt after restart
 
     # Reset tool use tracker
@@ -62,8 +61,8 @@ def handle_restart(shell_state=None):
             janito.tools.local_tools_adapter.set_allowed_permissions(ToolPermissions(read=False, write=False, execute=False))
             msg = "[green]All tool permissions have been set to OFF (read, write, execute = False).[/green]"
         # Refresh system prompt to reflect new permissions
-        if hasattr(shell_state, "agent") and shell_state.agent and hasattr(shell_state.agent, "_refresh_system_prompt_from_template"):
-            shell_state.agent._refresh_system_prompt_from_template()
+        if hasattr(shell_state, "agent") and shell_state.agent and hasattr(shell_state.agent, "refresh_system_prompt_from_template"):
+            shell_state.agent.refresh_system_prompt_from_template()
         if msg:
             shared_console.print(msg)
 

{janito-2.5.0 → janito-2.5.1}/janito/cli/chat_mode/shell/commands/execute.py

@@ -33,8 +33,7 @@ class ExecuteShellHandler(ShellCmdHandler):
         # Refresh system prompt if agent is available
         agent = getattr(self.shell_state, "agent", None)
         if agent:
-            if hasattr(agent, "_refresh_system_prompt_from_template"):
-                agent._refresh_system_prompt_from_template()
+            agent.refresh_system_prompt_from_template()
             # No need to print the system prompt after permission change
         if enable:
             shared_console.print("[green]Execution tools ENABLED. Tools can now execute code and commands.[/green]")

{janito-2.5.0 → janito-2.5.1}/janito/cli/chat_mode/shell/commands/prompt.py

@@ -41,14 +41,7 @@ class RoleShellHandler(ShellCmdHandler):
         if agent and hasattr(agent, "set_template_var"):
             agent.set_template_var("role", new_role)
             # Refresh the system prompt with the new role if possible
-            if (
-                hasattr(agent, "set_system_using_template")
-                and hasattr(agent, "_system_prompt_template_file")
-                and agent._system_prompt_template_file
-            ):
-                agent.set_system_using_template(
-                    agent._system_prompt_template_file, **agent.template_vars
-                )
+            agent.refresh_system_prompt_from_template()
         shared_console.print(
             f"[bold green]System role updated to:[/bold green] {new_role}"
         )

{janito-2.5.0 → janito-2.5.1}/janito/cli/chat_mode/shell/commands/read.py

@@ -28,8 +28,7 @@ class ReadShellHandler(ShellCmdHandler):
         # Refresh system prompt if agent is available
         agent = getattr(self.shell_state, "agent", None)
         if agent:
-            if hasattr(agent, "_refresh_system_prompt_from_template"):
-                agent._refresh_system_prompt_from_template()
+            agent.refresh_system_prompt_from_template()
             # No need to print the system prompt after permission change
         if enable:
             shared_console.print("[green]Read permissions ENABLED. Tools can now read files and data.[/green]")

{janito-2.5.0 → janito-2.5.1}/janito/cli/chat_mode/shell/commands/role.py

@@ -20,15 +20,8 @@ class RoleCommandShellHandler(ShellCmdHandler):
             new_role = " ".join(args)
             self.handler.agent.template_vars["role"] = new_role
             # Refresh the system prompt with the new role if possible
-            if (
-                hasattr(self.handler.agent, "set_system_using_template")
-                and hasattr(self.handler.agent, "_system_prompt_template_file")
-                and self.handler.agent._system_prompt_template_file
-            ):
-                self.handler.agent.set_system_using_template(
-                    self.handler.agent._system_prompt_template_file,
-                    **self.handler.agent.template_vars,
-                )
+            self.handler.agent.refresh_system_prompt_from_template()
+
             return f"Role set to: {new_role}"
 
     def get_completions(self, document, complete_event):

{janito-2.5.0 → janito-2.5.1}/janito/cli/chat_mode/shell/commands/write.py

@@ -28,8 +28,7 @@ class WriteShellHandler(ShellCmdHandler):
         # Refresh system prompt if agent is available
         agent = getattr(self.shell_state, "agent", None)
         if agent:
-            if hasattr(agent, "_refresh_system_prompt_from_template"):
-                agent._refresh_system_prompt_from_template()
+            agent.refresh_system_prompt_from_template()
             # No need to print the system prompt after permission change
         if enable:
             shared_console.print("[green]Write permissions ENABLED. Tools can now write files and data.[/green]")

{janito-2.5.0 → janito-2.5.1}/janito/llm/agent.py

@@ -87,7 +87,7 @@ class LLMAgent:
         template = env.get_template(Path(template_path).name)
         self.system_prompt = template.render(**kwargs)
 
-    def _refresh_system_prompt_from_template(self):
+    def refresh_system_prompt_from_template(self):
         if hasattr(self, "_template_vars") and hasattr(self, "system_prompt_template"):
             env = Environment(
                 loader=FileSystemLoader(Path(self.system_prompt_template).parent),

{janito-2.5.0 → janito-2.5.1}/janito.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: janito
-Version: 2.5.0
+Version: 2.5.1
 Summary: A new Python package called janito.
 Author-email: João Pinto <lamego.pinto@gmail.com>
 Project-URL: Homepage, https://github.com/janito-dev/janito

{janito-2.5.0 → janito-2.5.1}/janito.egg-info/SOURCES.txt

@@ -16,7 +16,6 @@ requirements.txt
 tox.ini
 .github/workflows/python-app.yml
 .vscode/settings.json
-benchmarks/test_chat_script_runner.py
 docs/DIV.md
 docs/Interfaces.txt
 docs/TOOLBAR-STYLING.md

janito-2.5.0/benchmarks/test_chat_script_runner.py

@@ -1,12 +0,0 @@
-"""Regression tests for the *ChatScriptRunner* utility."""
-from janito.cli.chat_mode.script_runner import ChatScriptRunner
-
-
-def test_script_runner_stubbed():
-    inputs = ["Hello there!", "/exit"]
-    runner = ChatScriptRunner(inputs)
-    transcript = runner.run()
-
-    # Basic sanity checks
-    assert "Hello there!" in transcript
-