eva-exploit 3.4.2__tar.gz → 3.4.21__tar.gz

{eva_exploit-3.4.2 → eva_exploit-3.4.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: eva-exploit
-Version: 3.4.2
+Version: 3.4.21
 Summary: Exploit Vector Agent
 Author: ARCANGEL0
 License: MIT

{eva_exploit-3.4.2 → eva_exploit-3.4.21}/config.py

@@ -10,7 +10,7 @@ from pathlib import Path
 
 # ================= CONFIG =================
 APP_NAME = "EVA"
-APP_VERSION = "3.4.2"
+APP_VERSION = "3.4.21"
 GITHUB_REPO = "arcangel0/EVA"
 PYPI_PACKAGE = "eva-exploit"
 API_ENDPOINT = "NOT_SET"

{eva_exploit-3.4.2 → eva_exploit-3.4.21}/eva_exploit.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: eva-exploit
-Version: 3.4.2
+Version: 3.4.21
 Summary: Exploit Vector Agent
 Author: ARCANGEL0
 License: MIT

{eva_exploit-3.4.2 → eva_exploit-3.4.21}/modules/llm.py

@@ -703,10 +703,59 @@ def _query_g4f(history):
     return ""
 
 
-def _query_custom_api(history, prompt_text=""):
+def normalize_util(session_name):
+    suffix = re.sub(r"[^A-Za-z0-9_]", "_", str(session_name or "").strip())
+    suffix = re.sub(r"_+", "_", suffix).strip("_")
+    return suffix
+
+
+def custom_enpdoint_func(session_name=""):
     endpoint = str(getattr(config_module, "API_ENDPOINT", API_ENDPOINT) or "").strip()
     if endpoint == "NOT_SET":
         endpoint = ""
+
+    session_name = str(session_name or "").strip()
+    if session_name:
+        session_key = f"EVA_API_CUSTOMAPI_{session_name}"
+        session_endpoint = os.getenv(session_key, "").strip()
+        if session_endpoint:
+            return session_endpoint
+
+        normalized_suffix = normalize_util(session_name)
+        if normalized_suffix and normalized_suffix != session_name:
+            normalized_key = f"EVA_API_CUSTOMAPI_{normalized_suffix}"
+            normalized_endpoint = os.getenv(normalized_key, "").strip()
+            if normalized_endpoint:
+                return normalized_endpoint
+
+    return endpoint
+
+
+def _call_custom_api_handler(query_fn, history, endpoint, prompt_text, session_name):
+    attempts = [
+        lambda: query_fn(history=history, endpoint=endpoint, prompt=prompt_text, session=session_name),
+        lambda: query_fn(history=history, endpoint=endpoint, session=session_name, prompt=prompt_text),
+        lambda: query_fn(history=history, endpoint=endpoint, prompt=prompt_text),
+        lambda: query_fn(history=history, endpoint=endpoint),
+        lambda: query_fn(history, endpoint, prompt_text, session_name),
+        lambda: query_fn(history, endpoint, prompt_text),
+        lambda: query_fn(history, endpoint),
+        lambda: query_fn(history),
+    ]
+    last_type_error = None
+    for call in attempts:
+        try:
+            return call()
+        except TypeError as e:
+            last_type_error = e
+
+    if last_type_error is not None:
+        raise last_type_error
+    return ""
+
+
+def _query_custom_api(history, prompt_text="", session_name=""):
+    endpoint = custom_enpdoint_func(session_name=session_name)
     handler_path = str(getattr(config_module, "CUSTOM_API_HANDLER", CUSTOM_API_HANDLER) or "").strip()
 
     if handler_path and handler_path != "NOT_SET":
@@ -720,19 +769,13 @@ def _query_custom_api(history, prompt_text=""):
                     spec.loader.exec_module(module)
                     query_fn = getattr(module, "query_custom_api", None)
                     if callable(query_fn):
-                        try:
-                            result = query_fn(history=history, endpoint=endpoint, prompt=prompt_text)
-                        except TypeError:
-                            try:
-                                result = query_fn(history=history, endpoint=endpoint)
-                            except TypeError:
-                                try:
-                                    result = query_fn(history, endpoint, prompt_text)
-                                except TypeError:
-                                    try:
-                                        result = query_fn(history, endpoint)
-                                    except TypeError:
-                                        result = query_fn(history)
+                        result = _call_custom_api_handler(
+                            query_fn,
+                            history=history,
+                            endpoint=endpoint,
+                            prompt_text=prompt_text,
+                            session_name=session_name,
+                        )
                         if isinstance(result, (dict, list)):
                             return json.dumps(result)
                         if result is None:
@@ -745,7 +788,7 @@ def _query_custom_api(history, prompt_text=""):
         print(Fore.RED + "⚠️ Custom API endpoint is not configured.")
         return ""
 
-    payload = {"conversation": history, "prompt": prompt_text}
+    payload = {"conversation": history, "prompt": prompt_text, "session": session_name}
     r = requests.post(endpoint, json=payload, timeout=None)
     return r.text
 
@@ -855,10 +898,14 @@ def _query_gemini(history):
 
 
 class LLM:
-    def __init__(self, backend):
+    def __init__(self, backend, session_name=""):
         self.backend = backend
+        self.session_name = str(session_name or "").strip()
         self.history = []
 
+    def set_session_name(self, session_name):
+        self.session_name = str(session_name or "").strip()
+
     def query(self, user_msg, last_output="", on_stream_start=None, workflow_context=""):
         followup = _is_followup_analysis_request(user_msg)
         if followup and (
@@ -917,7 +964,11 @@ class LLM:
             elif self.backend == "g4f":
                 raw = _query_g4f(request_messages)
             elif self.backend == "api":
-                raw = _query_custom_api(request_messages, prompt_text=prompt)
+                raw = _query_custom_api(
+                    request_messages,
+                    prompt_text=prompt,
+                    session_name=self.session_name,
+                )
             elif self.backend == "gpt":
                 raw = _query_openai(request_messages)
             elif self.backend == "anthropic":

{eva_exploit-3.4.2 → eva_exploit-3.4.21}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "eva-exploit"
-version = "3.4.2"
+version = "3.4.21"
 description = "Exploit Vector Agent"
 readme = "README.md"
 requires-python = ">=3.10"

{eva_exploit-3.4.2 → eva_exploit-3.4.21}/sessions/eva_session.py

@@ -95,7 +95,7 @@ class Eva:
             self.memory = json.loads(session_path.read_text())
             self.backend = self.memory.get("backend", backend)
 
-        self.model = LLM(self.backend)
+        self.model = LLM(self.backend, self.sessionName)
         self._hydrate_model_context()
 
     def _hydrate_model_context(self):
@@ -164,7 +164,7 @@ class Eva:
             checkAPI()
             self.memory["backend"] = self.backend
             self.save()
-        self.model = LLM(self.backend)
+        self.model = LLM(self.backend, self.sessionName)
         self._hydrate_model_context()
 
     def rename_session(self):
@@ -189,6 +189,7 @@ class Eva:
         self.session_path.rename(new_path)
         self.session_path = new_path
         self.sessionName = new_name
+        self.model.set_session_name(self.sessionName)
         self.save()
         cyber(f"Session renamed to {new_name}", color=Fore.GREEN)
 

{eva_exploit-3.4.2 → eva_exploit-3.4.21}/utils/system.py

@@ -70,7 +70,7 @@ def _custom_api_template(endpoint):
         "#!/usr/bin/env python3\n"
         "import requests\n\n"
         f"API_ENDPOINT = {json.dumps(target)}\n\n"
-        "def query_custom_api(history, endpoint=None, prompt=None):\n"
+        "def query_custom_api(history, endpoint=None, session=None, prompt=None):\n"
         "    target = endpoint or API_ENDPOINT\n"
         "    compiled_prompt = str(prompt or \"\").strip()\n"
         "    if not compiled_prompt:\n"
@@ -81,7 +81,7 @@ def _custom_api_template(endpoint):
         "    payload = {\n"
         "        \"prompt\": compiled_prompt,\n"
         "        \"conversation\": history,\n"
-        "        \"session\": \"eva-session\",\n"
+        "        \"session\": session,\n"
         "    }\n"
         "    r = requests.post(target, json=payload, timeout=None)\n"
         "    try:\n"