v0.1.1: Added flexfible agent settings and smart db

Alexx-coder · Alexx-coder · commit 938ce29186ee · 2026-07-10T21:01:23.000+03:00
diff --git a/README.md b/README.md
@@ -3,7 +3,7 @@
 # **Cortex**
 Secure Multi-Agent AI Framework
 
-[![Cortex: 0.1.0](https://img.shields.io/badge/Version-v0.1.0-blue?style=flat-square)](https://github.com/Alexx-coder/Cortex.git)
+[![Cortex: 0.1.1](https://img.shields.io/badge/Version-v0.1.1-blue?style=flat-square)](https://github.com/Alexx-coder/Cortex.git)
 [![License: MIT](https://img.shields.io/badge/License-MIT-yellow?style=flat-square)](https://opensource.org/licenses/MIT)
 [![Python 3.8+](https://img.shields.io/badge/Python-3.8+-green?style=flat-square&logo=python)](https://www.python.org/downloads/)
 [![Encryption: Fernet AES](https://img.shields.io/badge/Encryption-Fernet_AES-red?style=flat-square&logo=datadog)](https://cryptography.io/en/latest/fernet/)
diff --git a/__pycache__/commands.cpython-314.pyc b/__pycache__/commands.cpython-314.pyc
diff --git a/commands.py b/commands.py
@@ -92,7 +92,11 @@ def message(self, full_input):
             print(f"[ERROR] Agent '{agent_name}' does not exist. Available: {', '.join(self.agent_map.keys())}")
             return
 
-        provider_name = self.agent_map[agent_name]
+        agent_settings = self.agent_map[agent_name]
+        provider_name = agent_settings.get("provider")
+        temperature = agent_settings.get("temperature", 0.7)
+        max_tokens = agent_settings.get("max_tokens", 4096)
+
         agent_provider = self.providers.get(provider_name)
 
         if not agent_provider:
@@ -104,15 +108,18 @@ def message(self, full_input):
         
         print(f"\n[{agent_name.upper()}] Thinking...")
         
-        # Сохраняем запрос пользователя в историю ПЕРЕД отправкой
         self.db.add_message("user", prompt_text)
 
         try:
-            # Отправляем историю в ИИ
-            response = agent_provider.chat(prompt_text, system=system_prompt, history=history)
+            response = agent_provider.chat(
+                prompt_text, 
+                system=system_prompt, 
+                history=history,
+                temperature=temperature,
+                max_tokens=max_tokens
+            )
             print(f"\n[{agent_name.upper()}]:\n{response}\n")
             
-            # Сохраняем ответ ИИ в историю
             self.db.add_message("assistant", response)
             
         except Exception as e:
diff --git a/database.py b/database.py
@@ -19,7 +19,6 @@ def __init__(self, password: str, db_path: str = None):
         self.data = self._load()
 
     def _load(self):
-        # Используем self.db_path вместо старого DB_PATH
         if not os.path.exists(self.db_path) or os.path.getsize(self.db_path) == 0:
             return {"active_chat": None, "chats": {}}
 
@@ -30,24 +29,29 @@ def _load(self):
             return {"active_chat": None, "chats": {}}
 
         try:
-            # Дешифруем сырые байты из файла
             decrypted_bytes = self.cipher.decrypt(encrypted_data)
             json_string = decrypted_bytes.decode('utf-8')
             return json.loads(json_string)
         except Exception as e:
-            print(f"[ERROR] Failed to decrypt database. Wrong password? Details: {e}")
-            exit(1)
+            print(f"\n[CRITICAL ERROR] Failed to decrypt database.")
+            print(f"Reason: {str(e) or 'Database file is corrupted or wrong password.'}")
+            
+            choice = input("Do you want to DELETE the corrupted database and start fresh? (y/n): ").strip().lower()
+            if choice == 'y':
+                os.remove(self.db_path)
+                print("[SYSTEM] Corrupted database deleted. Starting with a clean slate.")
+                return {"active_chat": None, "chats": {}}
+            else:
+                print("[SYSTEM] Exiting. Please check your password or database file manually.")
+                exit(1)
 
     def _save(self):
         json_string = json.dumps(self.data, indent=4, ensure_ascii=False)
         
         try:
-            # Шифруем строку JSON, получаем байты
             encrypted_data = self.cipher.encrypt(json_string.encode('utf-8'))
             
-            # Используем self.db_path!
             os.makedirs(os.path.dirname(self.db_path), exist_ok=True)
-            # Записываем зашифрованные байты
             with open(self.db_path, "wb") as f:
                 f.write(encrypted_data)
         except Exception as e:
@@ -77,7 +81,7 @@ def add_message(self, role: str, content: str):
         self._save()
 
     def list_chats(self):
-        return [f"{cid}: {info['name']} ({len(info['history'])} messages)" for cid, info in self.data['chats'].items()]
+        return [f"{cid}: { info['name']} ({len(info['history'])} messages)" for cid, info in self.data['chats'].items()]
 
     def switch_chat(self, chat_id: str):
         if chat_id in self.data['chats']:
diff --git a/router/__pycache__/ollama_provider.cpython-314.pyc b/router/__pycache__/ollama_provider.cpython-314.pyc
diff --git a/router/ollama_provider.py b/router/ollama_provider.py
@@ -21,25 +21,30 @@ def list_models(self):
         except requests.RequestException:
             return []
 
-    def chat(self, prompt: str, system: Optional[str] = None, history: list = None) -> str:
+    def chat(self, prompt: str, system: Optional[str] = None, history: list = None, temperature: float = 0.7, max_tokens: int = 4096) -> str:
         if not self.model:
             return "[ERROR] No Ollama model selected in config.json."
 
-        # Берем старую историю, если она есть
         messages = history.copy() if history else []
 
-        # Вставляем системный промпт в начало, если его там нет
         if system:
             if not messages or messages[0].get("role") != "system":
                 messages.insert(0, {"role": "system", "content": system})
 
-        # Добавляем новое сообщение пользователя
         messages.append({"role": "user", "content": prompt})
 
         try:
             r = requests.post(
                 f"{self.url}/api/chat",
-                json={"model": self.model, "messages": messages, "stream": False},
+                json={
+                    "model": self.model, 
+                    "messages": messages, 
+                    "stream": False, 
+                    "options": {
+                        "temperature": temperature,
+                        "num_predict": max_tokens
+                    }
+                },
                 timeout=300,
             )
             r.raise_for_status()
diff --git a/runner.py b/runner.py
@@ -59,18 +59,27 @@ def load_config():
 class Runner:
     def __init__(self):
         self.name = "CORTEX"
-        self.version = 'v0.1.0'
+        self.version = 'v0.1.1'
         
-        # 1. Запрашиваем пароль для расшифровки БД
         print("--- DATABASE AUTHENTICATION ---")
         pwd = getpass.getpass("Enter master password: ")
         self.db = Database(pwd)
         print("Database loaded successfully.\n")
         
-        # 2. Загружаем конфиг
         self.config = load_config()
+
+        self.config = load_config()
+        
+   
+        for agent_name, agent_value in self.config["agents"].items():
+            if isinstance(agent_value, str):
+                self.config["agents"][agent_name] = {
+                    "provider": agent_value,
+                    "temperature": 0.7,
+                    "max_tokens": 4096
+                }
+
         
-        # 3. Инициализация провайдеров
         self.providers = {
             "ollama": Ollama(
                 model=self.config["ollama"]["model"], 
@@ -92,13 +101,12 @@ def __init__(self):
             ) if self.config["openrouter"]["api_key"] else None
         }
 
-        # 4. Передаем db в Commands!
         self.commands_handler = Commands(
             providers=self.providers,
             agent_map=self.config["agents"],
             config_path=CONFIG_PATH,
             version=self.version,
-            db=self.db  # ВОТ ТУТ МЫ ПЕРЕДАЕМ БАЗУ
+            db=self.db  
         )
 
     def main(self):
