janito 2.6.0__py3-none-any.whl → 2.7.0__py3-none-any.whl

janito/cli/single_shot_mode/handler.py

@@ -1,95 +1,95 @@
-"""
-PromptHandler: Handles prompt submission and response formatting for janito CLI (one-shot prompt execution).
-"""
-
-from __future__ import annotations
-
-from janito.cli.prompt_setup import setup_agent_and_prompt_handler
-import janito.tools  # Ensure all tools are registered
-from janito.cli.console import shared_console
-import time
-
-
-class PromptHandler:
-    def __init__(self, args, provider_instance, llm_driver_config, role=None,  allowed_permissions=None):
-        self.args = args
-        self.provider_instance = provider_instance
-        self.llm_driver_config = llm_driver_config
-        self.role = role
-                # Instantiate agent together with prompt handler using the shared helper
-        self.agent, self.generic_handler = setup_agent_and_prompt_handler(
-            args=args,
-            provider_instance=provider_instance,
-            llm_driver_config=llm_driver_config,
-            role=role,
-            verbose_tools=getattr(args, "verbose_tools", False),
-            verbose_agent=getattr(args, "verbose_agent", False),
-            
-            allowed_permissions=allowed_permissions,
-            profile=getattr(args, "profile", None),
-        )
-
-    def handle(self) -> None:
-        import traceback
-
-        user_prompt = " ".join(getattr(self.args, "user_prompt", [])).strip()
-        # UTF-8 sanitize user_prompt
-        sanitized = user_prompt
-        try:
-            sanitized.encode("utf-8")
-        except UnicodeEncodeError:
-            sanitized = sanitized.encode("utf-8", errors="replace").decode("utf-8")
-            shared_console.print(
-                "[yellow]Warning: Some characters in your input were not valid UTF-8 and have been replaced.[/yellow]"
-            )
-        import time
-        try:
-            start_time = time.time()
-            self.generic_handler.handle_prompt(
-                sanitized,
-                args=self.args,
-                print_header=True,
-                raw=getattr(self.args, "raw", False),
-            )
-            end_time = time.time()
-            elapsed = end_time - start_time
-            self._post_prompt_actions(elapsed=elapsed)
-            if hasattr(self.args, "verbose_agent") and self.args.verbose_agent:
-                print("[debug] handle_prompt() completed without exception.")
-        except Exception as e:
-            print(
-                f"[error] Exception occurred in handle_prompt: {type(e).__name__}: {e}"
-            )
-            traceback.print_exc()
-
-    def _post_prompt_actions(self, elapsed=None):
-        # Align with chat mode: only print token usage summary
-        import sys
-        from janito.formatting_token import print_token_message_summary
-        from janito.perf_singleton import performance_collector
-        usage = performance_collector.get_last_request_usage()
-        # If running in stdin mode, do not print token usage
-        if sys.stdin.isatty():
-            print_token_message_summary(shared_console, msg_count=1, usage=usage, elapsed=elapsed)
-        self._cleanup_driver_and_console()
-
-
-    def _cleanup_driver_and_console(self):
-        if hasattr(self.agent, "join_driver"):
-            if (
-                hasattr(self.agent, "input_queue")
-                and self.agent.input_queue is not None
-            ):
-                self.agent.input_queue.put(None)
-            self.agent.join_driver()
-        try:
-            shared_console.file.flush()
-        except Exception:
-            pass
-        try:
-            import sys
-
-            sys.stdout.flush()
-        except Exception:
-            pass
-        # If event logger is active, flush event log
+"""
+PromptHandler: Handles prompt submission and response formatting for janito CLI (one-shot prompt execution).
+"""
+
+from __future__ import annotations
+
+from janito.cli.prompt_setup import setup_agent_and_prompt_handler
+import janito.tools  # Ensure all tools are registered
+from janito.cli.console import shared_console
+import time
+
+
+class PromptHandler:
+    def __init__(self, args, provider_instance, llm_driver_config, role=None,  allowed_permissions=None):
+        self.args = args
+        self.provider_instance = provider_instance
+        self.llm_driver_config = llm_driver_config
+        self.role = role
+                # Instantiate agent together with prompt handler using the shared helper
+        self.agent, self.generic_handler = setup_agent_and_prompt_handler(
+            args=args,
+            provider_instance=provider_instance,
+            llm_driver_config=llm_driver_config,
+            role=role,
+            verbose_tools=getattr(args, "verbose_tools", False),
+            verbose_agent=getattr(args, "verbose_agent", False),
+            
+            allowed_permissions=allowed_permissions,
+            profile=getattr(args, "profile", None),
+        )
+
+    def handle(self) -> None:
+        import traceback
+
+        user_prompt = " ".join(getattr(self.args, "user_prompt", [])).strip()
+        # UTF-8 sanitize user_prompt
+        sanitized = user_prompt
+        try:
+            sanitized.encode("utf-8")
+        except UnicodeEncodeError:
+            sanitized = sanitized.encode("utf-8", errors="replace").decode("utf-8")
+            shared_console.print(
+                "[yellow]Warning: Some characters in your input were not valid UTF-8 and have been replaced.[/yellow]"
+            )
+        import time
+        try:
+            start_time = time.time()
+            self.generic_handler.handle_prompt(
+                sanitized,
+                args=self.args,
+                print_header=True,
+                raw=getattr(self.args, "raw", False),
+            )
+            end_time = time.time()
+            elapsed = end_time - start_time
+            self._post_prompt_actions(elapsed=elapsed)
+            if hasattr(self.args, "verbose_agent") and self.args.verbose_agent:
+                print("[debug] handle_prompt() completed without exception.")
+        except Exception as e:
+            print(
+                f"[error] Exception occurred in handle_prompt: {type(e).__name__}: {e}"
+            )
+            traceback.print_exc()
+
+    def _post_prompt_actions(self, elapsed=None):
+        # Align with chat mode: only print token usage summary
+        import sys
+        from janito.formatting_token import print_token_message_summary
+        from janito.perf_singleton import performance_collector
+        usage = performance_collector.get_last_request_usage()
+        # If running in stdin mode, do not print token usage
+        if sys.stdin.isatty():
+            print_token_message_summary(shared_console, msg_count=1, usage=usage, elapsed=elapsed)
+        self._cleanup_driver_and_console()
+
+
+    def _cleanup_driver_and_console(self):
+        if hasattr(self.agent, "join_driver"):
+            if (
+                hasattr(self.agent, "input_queue")
+                and self.agent.input_queue is not None
+            ):
+                self.agent.input_queue.put(None)
+            self.agent.join_driver()
+        try:
+            shared_console.file.flush()
+        except Exception:
+            pass
+        try:
+            import sys
+
+            sys.stdout.flush()
+        except Exception:
+            pass
+        # If event logger is active, flush event log

janito/drivers/driver_registry.py

@@ -1,27 +1,27 @@
-# janito/drivers/driver_registry.py
-"""
-DriverRegistry: Maps driver string names to class objects for use by providers.
-"""
-
-from typing import Dict, Type
-
-# --- Import driver classes ---
-from janito.drivers.azure_openai.driver import AzureOpenAIModelDriver
-from janito.drivers.openai.driver import OpenAIModelDriver
-
-_DRIVER_REGISTRY: Dict[str, Type] = {
-    "AzureOpenAIModelDriver": AzureOpenAIModelDriver,
-    "OpenAIModelDriver": OpenAIModelDriver,
-}
-
-
-def get_driver_class(name: str):
-    """Get the driver class by string name."""
-    try:
-        return _DRIVER_REGISTRY[name]
-    except KeyError:
-        raise ValueError(f"No driver found for name: {name}")
-
-
-def register_driver(name: str, cls: type):
-    _DRIVER_REGISTRY[name] = cls
+# janito/drivers/driver_registry.py
+"""
+DriverRegistry: Maps driver string names to class objects for use by providers.
+"""
+
+from typing import Dict, Type
+
+# --- Import driver classes ---
+from janito.drivers.azure_openai.driver import AzureOpenAIModelDriver
+from janito.drivers.openai.driver import OpenAIModelDriver
+
+_DRIVER_REGISTRY: Dict[str, Type] = {
+    "AzureOpenAIModelDriver": AzureOpenAIModelDriver,
+    "OpenAIModelDriver": OpenAIModelDriver,
+}
+
+
+def get_driver_class(name: str):
+    """Get the driver class by string name."""
+    try:
+        return _DRIVER_REGISTRY[name]
+    except KeyError:
+        raise ValueError(f"No driver found for name: {name}")
+
+
+def register_driver(name: str, cls: type):
+    _DRIVER_REGISTRY[name] = cls