exaai-agent 2.2.0__tar.gz → 2.2.1__tar.gz

{exaai_agent-2.2.0 → exaai_agent-2.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: exaai-agent
-Version: 2.2.0
+Version: 2.2.1
 Summary: ExaAi - Advanced AI Security Agent for Comprehensive Penetration Testing
 License: Apache-2.0
 License-File: LICENSE
@@ -37,6 +37,7 @@ Requires-Dist: rich
 Requires-Dist: tenacity (>=9.0.0,<10.0.0)
 Requires-Dist: textual (>=4.0.0,<5.0.0)
 Requires-Dist: uvicorn
+Requires-Dist: websockets (>=12.0,<13.0)
 Requires-Dist: xmltodict (>=0.13.0,<0.14.0)
 Description-Content-Type: text/markdown
 

{exaai_agent-2.2.0 → exaai_agent-2.2.1}/exaaiagnt/interface/main.py

@@ -224,7 +224,17 @@ async def warm_up_llm() -> None:
         error_text.append("\n\n", style="white")
         error_text.append("Could not establish connection to the language model.\n", style="white")
         error_text.append("Please check your configuration and try again.\n", style="white")
-        error_text.append(f"\nError: {e}", style="dim white")
+        
+        # Enhanced error diagnosis
+        error_str = str(e)
+        if "AuthenticationError" in error_str:
+             error_text.append("🔍 Hint: Your API key seems invalid or expired.\n", style="bold yellow")
+        elif "NotFoundError" in error_str:
+             error_text.append(f"🔍 Hint: The model '{model_name}' was not found. Check if the model name is correct.\n", style="bold yellow")
+        elif "ConnectionError" in error_str:
+             error_text.append("🔍 Hint: Network connection failed. Check your internet or proxy settings.\n", style="bold yellow")
+             
+        error_text.append(f"\nError Details: {e}", style="dim white")
 
         panel = Panel(
             error_text,
@@ -242,7 +252,7 @@ async def warm_up_llm() -> None:
 
 def get_version() -> str:
     """Get the current ExaAi version."""
-    return "2.1.2"
+    return "2.2.1"
 
 
 def parse_arguments() -> argparse.Namespace:

{exaai_agent-2.2.0 → exaai_agent-2.2.1}/exaaiagnt/interface/tui.py

@@ -45,7 +45,7 @@ def get_package_version() -> str:
         return pkg_version("exaai-agent")
     except PackageNotFoundError:
         # Fallback version if package not installed
-        return "2.1.2"
+        return "2.2.1"
 
 
 class ChatTextArea(TextArea):  # type: ignore[misc]
@@ -80,7 +80,7 @@ class SplashScreen(Static):  # type: ignore[misc]
     NEON_ORANGE = "#ff8800"
     SOFT_WHITE = "#e0e0e0"
     
-    # Enhanced ASCII Logo - ExaAi v2.1.2
+    # Enhanced ASCII Logo - ExaAi v2.2.1
     BANNER = r"""
     ███████╗██╗  ██╗ █████╗      █████╗ ██╗
     ██╔════╝╚██╗██╔╝██╔══██╗    ██╔══██╗██║

{exaai_agent-2.2.0 → exaai_agent-2.2.1}/exaaiagnt/runtime/docker_runtime.py

@@ -104,6 +104,14 @@ class DockerRuntime(AbstractRuntime):
                 self._tool_server_port = tool_server_port
                 self._tool_server_token = tool_server_token
 
+                # Mount kubeconfig if available
+                volumes = {}
+                kube_config = os.path.expanduser("~/.kube")
+                if os.path.exists(kube_config):
+                    volumes[kube_config] = {'bind': '/root/.kube', 'mode': 'ro'}
+                    # Also mount for pentester user
+                    volumes[kube_config] = {'bind': '/home/pentester/.kube', 'mode': 'ro'}
+
                 container = self.client.containers.run(
                     EXAAI_IMAGE,
                     command="sleep infinity",
@@ -114,6 +122,7 @@ class DockerRuntime(AbstractRuntime):
                         f"{caido_port}/tcp": caido_port,
                         f"{tool_server_port}/tcp": tool_server_port,
                     },
+                    volumes=volumes,
                     cap_add=["NET_ADMIN", "NET_RAW"],
                     labels={"exaai-scan-id": scan_id},
                     environment={

{exaai_agent-2.2.0 → exaai_agent-2.2.1}/exaaiagnt/tools/k8s_scanner/k8s_actions.py

@@ -10,7 +10,7 @@ Comprehensive security testing for Kubernetes clusters including:
 - Container vulnerability scanning integration
 
 Author: ALhilali
-Version: 1.0.0
+Version: 2.2.1
 """
 
 import logging

{exaai_agent-2.2.0 → exaai_agent-2.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "exaai-agent"
-version = "2.2.0"
+version = "2.2.1"
 description = "ExaAi - Advanced AI Security Agent for Comprehensive Penetration Testing"
 authors = ["ExaAi Team <mahmedhleli@gmail.com>"]
 readme = "README.md"
@@ -63,6 +63,7 @@ xmltodict = "^0.13.0"
 pyte = "^0.8.1"
 requests = "^2.32.0"
 libtmux = "^0.46.2"
+websockets = "^12.0"
 
 [tool.poetry.group.dev.dependencies]
 # Type checking and static analysis