Fix ci

Author: viartemev

.github/workflows/pylint.yml

@@ -0,0 +1,22 @@
+[MAIN]
+# Указываем pylint, где искать модули, чтобы исправить ошибки импорта.
+init-hook='import sys; sys.path.append(".venv/lib/python3.12/site-packages")'
+
+[MESSAGES CONTROL]
+# Отключаем менее критичные или спорные проверки
+disable=
+    missing-module-docstring,
+    missing-class-docstring,
+    missing-function-docstring,
+    too-few-public-methods,
+    redefined-outer-name,
+    protected-access,
+    line-too-long,
+    trailing-whitespace,
+    trailing-newlines,
+    missing-final-newline,
+    wrong-import-order
+
+[FORMAT]
+# Устанавливаем максимальную длину строки
+max-line-length=120 

.pylintrc

@@ -1,19 +1,22 @@
+"""
+Этот модуль определяет основного AI-агента, его логику и цикл взаимодействия с пользователем.
+"""
 from typing import Callable, Tuple, Optional, List, Dict, Any
 import os
 import json
 import logging
-from openai import OpenAI
+from openai import OpenAI, APIError
 from openai.types.chat import ChatCompletionMessage
 from dotenv import load_dotenv
 from app.agents.tools import ToolDefinition
 
 # Четкий и инструктивный системный промпт
 SYSTEM_PROMPT = (
-    """You are a helpful AI assistant. You have access to a set of tools to answer the user’s questions.
-    When the user asks a question, first consider whether a tool is needed.
-    If you decide to use a tool, call it. Once you receive the result from the tool,
-    use that result to formulate a final response to the user.
-    Don’t just repeat the tool’s output — instead, provide a helpful and detailed answer that incorporates the retrieved data."""
+    "Ты — полезный ИИ-ассистент. У тебя есть доступ к набору инструментов для ответов на вопросы пользователя. "
+    "Когда пользователь задает вопрос, сначала подумай, нужно ли использовать инструмент. "
+    "Если решишь использовать инструмент, вызови его. После получения результата от инструмента, "
+    "используй этот результат для формулирования окончательного ответа пользователю. "
+    "Не просто констатируй вывод инструмента, а дай полезный и развернутый ответ, который включает в себя полученные данные."
 )
 
 # Настройка логирования
@@ -42,32 +45,31 @@ def __init__(
 
     def _execute_tool(self, tool_name: str, tool_args: Dict[str, Any]) -> str:
         """Выполняет инструмент и возвращает его результат в виде строки."""
-        logging.info(f"Выполнение инструмента '{tool_name}' с аргументами: {tool_args}")
-        if tool_name in self.tools:
-            tool = self.tools[tool_name]
-            try:
-                result = tool.function(tool_args)
-                logging.info(f"Инструмент '{tool_name}' успешно выполнен.")
-                return str(result)
-            except Exception as e:
-                logging.error(f"Ошибка при выполнении инструмента '{tool_name}': {e}", exc_info=True)
-                return f"Ошибка: Не удалось выполнить инструмент '{tool_name}'. Причина: {e}"
-        else:
-            logging.warning(f"Попытка вызова неизвестного инструмента: '{tool_name}'")
+        logging.info("Выполнение инструмента '%s' с аргументами: %s", tool_name, tool_args)
+        tool = self.tools.get(tool_name)
+        if not tool:
+            logging.warning("Попытка вызова неизвестного инструмента: '%s'", tool_name)
             return f"Ошибка: Инструмент '{tool_name}' не найден."
+        try:
+            result = tool.function(tool_args)
+            logging.info("Инструмент '%s' успешно выполнен.", tool_name)
+            return str(result)
+        except Exception as e:
+            logging.error("Непредвиденная ошибка при выполнении инструмента '%s': %s", tool_name, e, exc_info=True)
+            return f"Ошибка: Не удалось выполнить инструмент '{tool_name}'. Причина: {e}"
 
     def _get_user_input(self) -> Optional[str]:
         """Обрабатывает получение ввода от пользователя."""
         print("\033[94mВы:\033[0m ", end="")
         if self.user_input_handler:
             user_input, ok = self.user_input_handler()
             return user_input if ok else None
-        else:
-            try:
-                user_input = input()
-                return user_input if user_input.strip() else None
-            except EOFError:
-                return None
+        
+        try:
+            user_input = input()
+            return user_input if user_input.strip() else None
+        except EOFError:
+            return None
 
     def run(self) -> None:
         """Запускает основной цикл общения с агентом."""
@@ -84,17 +86,21 @@ def run(self) -> None:
 
                 max_tool_calls = 5
                 for _ in range(max_tool_calls):
-                    logging.info(f"Вызов OpenAI с историей из {len(conversation)} сообщений:\n{json.dumps(conversation, indent=2, ensure_ascii=False)}")
+                    logging.info("Вызов OpenAI с историей из %d сообщений:\n%s", 
+                                 len(conversation), 
+                                 json.dumps(conversation, indent=2, ensure_ascii=False))
 
-                    response: ChatCompletionMessage = self.client.chat.completions.create(
-                        model=self.model,
-                        messages=conversation,
-                        tools=[
-                            {"type": "function", "function": tool.to_openai_spec()}
-                            for tool in self.tools.values()
-                        ] if self.tools else None,
-                        tool_choice="auto" if self.tools else None,
-                    ).choices[0].message
+                    try:
+                        response: ChatCompletionMessage = self.client.chat.completions.create(
+                            model=self.model,
+                            messages=conversation,
+                            tools=[tool.to_openai_spec() for tool in self.tools.values()] if self.tools else None,
+                            tool_choice="auto" if self.tools else None,
+                        ).choices[0].message
+                    except APIError as e:
+                        logging.error("Ошибка OpenAI API: %s", e)
+                        print(f"\033[91m{self.name}: Произошла ошибка при обращении к OpenAI. Попробуйте еще раз.\033[0m")
+                        break
 
                     if not response.tool_calls:
                         assistant_response = response.content or ""
@@ -111,7 +117,7 @@ def run(self) -> None:
                             tool_args = json.loads(tool_args_str)
                             tool_result = self._execute_tool(tool_name, tool_args)
                         except json.JSONDecodeError:
-                            logging.error(f"Не удалось декодировать аргументы для '{tool_name}': {tool_args_str}")
+                            logging.error("Не удалось декодировать аргументы для '%s': %s", tool_name, tool_args_str)
                             tool_result = f"Ошибка: Неверные аргументы для инструмента '{tool_name}'."
 
                         conversation.append({

app/agents/agent.py

@@ -1,5 +1,10 @@
+"""
+Этот модуль определяет инструменты (tools), которые может использовать AI-агент.
+Каждый инструмент представлен классом ToolDefinition и соответствующей функцией.
+"""
 from typing import Callable, Any, Dict
 import os
+import logging
 
 class ToolDefinition:
     """Определяет структуру инструмента, его описание и функцию."""
@@ -18,9 +23,12 @@ def __init__(
     def to_openai_spec(self) -> Dict[str, Any]:
         """Преобразует определение инструмента в формат, ожидаемый OpenAI API."""
         return {
-            "name": self.name,
-            "description": self.description,
-            "parameters": self.input_schema,
+            "type": "function",
+            "function": {
+                "name": self.name,
+                "description": self.description,
+                "parameters": self.input_schema,
+            }
         }
 
 def read_file_tool(input_data: Dict[str, Any]) -> str:
@@ -41,7 +49,8 @@ def read_file_tool(input_data: Dict[str, Any]) -> str:
             return f.read()
     except FileNotFoundError:
         return f"Ошибка: Файл не найден по пути '{path}'."
-    except Exception as e:
+    except IOError as e:
+        logging.error("Ошибка ввода-вывода при чтении файла %s: %s", path, e)
         return f"Ошибка: Не удалось прочитать файл '{path}': {e}"
 
 read_file_definition = ToolDefinition(
@@ -85,7 +94,8 @@ def list_files_tool(input_data: Dict[str, Any]) -> str:
             for f in files:
                 output += f'{sub_indent}{f}\n'
         return output.strip()
-    except Exception as e:
+    except OSError as e:
+        logging.error("Ошибка ОС при листинге файлов в %s: %s", path, e)
         return f"Ошибка: Не удалось получить список файлов для '{path}': {e}"
 
 list_files_definition = ToolDefinition(
@@ -128,7 +138,8 @@ def edit_file_tool(input_data: Dict[str, Any]) -> str:
         with open(path, "w", encoding="utf-8") as f:
             f.write(content)
         return f"Файл '{path}' успешно сохранен."
-    except Exception as e:
+    except IOError as e:
+        logging.error("Ошибка ввода-вывода при записи в файл %s: %s", path, e)
         return f"Ошибка: Не удалось записать в файл '{path}': {e}"
 
 edit_file_definition = ToolDefinition(
@@ -171,7 +182,8 @@ def delete_file_tool(input_data: Dict[str, Any]) -> str:
 
         os.remove(path)
         return f"Файл '{path}' успешно удален."
-    except Exception as e:
+    except OSError as e:
+        logging.error("Ошибка ОС при удалении файла %s: %s", path, e)
         return f"Ошибка: Не удалось удалить файл '{path}': {e}"
 
 delete_file_definition = ToolDefinition(

app/agents/tools.py

@@ -9,4 +9,5 @@ loguru==0.7.3
 openai==1.85.0
 ruff>=0.4.0
 pytest>=7.0.0
-pytest-mock>=3.12.0
+pytest-mock>=3.12.0
+pylint>=3.0.0