iatoolkit 0.71.4__py3-none-any.whl → 0.91.1__py3-none-any.whl

iatoolkit/locales/es.yaml

@@ -1,163 +1,172 @@
-    # locales/es.yaml
-    ui:
-      login_widget:
-        title: "Iniciar Sesión"
-        welcome_message: "Ingresa tus credenciales o registrate para acceder a la plataforma IA de Sample Company."
-        email_placeholder: "Correo electrónico"
-        login_button: "Acceder"
-        forgot_password_link: "¿Olvidaste tu contraseña?"
-        no_account_prompt: "¿No tienes una cuenta?"
-        signup_link: "Regístrate"
-
-      signup:
-        title: "Crear una Cuenta"
-        subtitle: "Comienza tu viaje con nosotros hoy."
-        first_name_label: "Nombre"
-        last_name_label: "Apellido"
-        email_label: "Correo Electrónico"
-        password_label: "Contraseña"
-        confirm_password_label: "Confirmar Contraseña"
-        signup_button: "Crear Cuenta"
-        already_have_account: "¿Ya tienes una cuenta?"
-        login_link: "Inicia Sesión"
-        disclaimer: "🔒 Valoramos tu privacidad. Tus datos se usarán exclusivamente para el funcionamiento de la plataforma."
-
-      forgot_password:
-        title: "Recuperar Contraseña"
-        subtitle: "Ingresa tu correo electrónico y te enviaremos un enlace para restablecer tu contraseña."
-        submit_button: "Enviar Enlace de Recuperación"
-        back_to_login: "Volver a Iniciar Sesión"
-
-      change_password:
-        title: "Crear Nueva Contraseña"
-        subtitle: "Estás cambiando la contraseña para <strong>{email}</strong>."
-        temp_code_label: "Código Temporal"
-        temp_code_placeholder: "Revisa tu correo electrónico"
-        new_password_label: "Nueva Contraseña"
-        password_instructions: "Debe contener al menos 8 caracteres, mayúscula, minúscula, número y un carácter especial."
-        confirm_password_label: "Confirmar Nueva Contraseña"
-        save_button: "Guardar Contraseña"
-        back_to_home: "Volver al inicio"
-
-      chat:
-        welcome_message: "¡Hola! ¿En qué te puedo ayudar hoy?"
-        input_placeholder: "Escribe tu consulta aquí..."
-        prompts_available: "Prompts disponibles"
-
-      tooltips:
-        history: "Historial con mis consultas"
-        reload_context: "Forzar Recarga de Contexto"
-        feedback: "Tu feedback es muy importante"
-        usage_guide: "Guía de Uso"
-        onboarding: "Cómo preguntar mejor"
-        logout: "Cerrar sesión"
-        use_prompt_assistant: "Usar Asistente de Prompts"
-        attach_files: "Adjuntar archivos"
-        view_attached_files: "Ver archivos adjuntos"
-      modals:
-        files_title: "Archivos Cargados"
-        feedback_title: "Tu Opinión es Importante"
-        feedback_prompt: "¿Qué tan útil fue la respuesta del asistente?"
-        feedback_comment_label: "Tu comentario nos ayuda a mejorar:"
-        feedback_comment_placeholder: "Escribe aquí tu opinión, sugerencias o comentarios..."
-        history_title: "Historial de Consultas"
-        history_table_date: "Fecha"
-        history_table_query: "Consulta"
-        loading_history: "Cargando historial..."
-        no_history_found: "No se encontró historial de consultas."
-        help_title: "Guía de uso del Asistente IA"
-
-      buttons:
-        cancel: "Cerrar"
-        send: "Enviar"
-        stop: "Detener"
-
-    errors:
-      company_not_found: "La empresa {company_short_name} no existe."
-      timeout: "El tiempo de espera ha expirado."
-
-
-      auth:
-        invalid_password: "La contraseña proporcionada es incorrecta."
-        user_not_found: "No se encontró un usuario con ese correo."
-        invalid_or_expired_token: "Token inválido o expirado."
-        session_creation_failed: "No se pudo crear la sesión del usuario."
-        authentication_required: "Autenticación requerida. No se proporcionó cookie de sesión o clave de API."
-        invalid_api_key: "Clave de API inválida o inactiva."
-        no_user_identifier_api: "No se proporcionó user_identifier para la llamada a la API."
-      templates:
-        company_not_found: "Empresa no encontrada."
-        home_template_not_found: "La plantilla de la página de inicio para la empresa '{company_short_name}' no está configurada."
-        processing_error: "Error al procesar el template: {error}"
-        template_not_found: "No se encontro el template: '{template_name}'."
-
-      general:
-        unexpected_error: "Ha ocurrido un error inesperado: {error}."
-        unsupported_language: "El idioma seleccionado no es válido."
-      signup:
-        company_not_found: "La empresa {company_short_name} no existe."
-        incorrect_password_for_existing_user: "La contraseña para el usuario {email} es incorrecta."
-        user_already_registered: "El usuario con email '{email}' ya existe en esta empresa."
-        password_mismatch: "Las contraseñas no coinciden. Por favor, inténtalo de nuevo."
-      change_password:
-        token_expired: "El enlace de cambio de contraseña ha expirado. Por favor, solicita uno nuevo."
-        password_mismatch: "Las contraseñas no coinciden. Por favor, inténtalo nuevamente."
-        invalid_temp_code: "El código temporal no es válido. Por favor, verifica o solicita uno nuevo."
-      forgot_password:
-        user_not_registered: "El usuario con correo {email} no está registrado."
-      verification:
-        token_expired: "El enlace de verificación ha expirado. Por favor, contacta a soporte si necesitas uno nuevo."
-        user_not_found: "El usuario que intentas verificar no existe."
-      validation:
-        password_too_short: "La contraseña debe tener al menos 8 caracteres."
-        password_no_uppercase: "La contraseña debe tener al menos una letra mayúscula."
-        password_no_lowercase: "La contraseña debe tener al menos una letra minúscula."
-        password_no_digit: "La contraseña debe tener al menos un número."
-        password_no_special_char: "La contraseña debe tener al menos un carácter especial."
-
-      services:
-        no_text_file: "El archivo no es texto o la codificación no es UTF-8"
-        no_output_file: "falta el nombre del archivo de salida"
-        no_data_for_excel: "faltan los datos  o no es una lista de diccionarios"
-        no_download_directory: "no esta configurado el directorio temporal para guardar excels"
-        cannot_create_excel: "no se pudo crear el archivo excel"
-        invalid_filename: "Nombre de archivo inválido"
-        file_not_exist : "Archivo no encontrado"
-        path_is_not_a_file : "La ruta no corresponde a un archivo"
-        file_validation_error : "Error validando archivo"
-        user_not_authorized: "Usuario no esta autorizado para esta empresa"
-        account_not_verified: "Tu cuenta no ha sido verificada. Por favor, revisa tu correo."
-        missing_response_id: "No se encontró 'previous_response_id' para '{company_short_name}/{user_identifier}'. Reinicia el contexto."
-
-    api_responses:
-      context_reloaded_success: "El contexto se ha recargado con éxito."
-
-    services:
-      mail_sent: "mail enviado exitosamente."
-      start_query: "Hola, cual es tu pregunta?"
-
-    flash_messages:
-      password_changed_success: "Tu contraseña ha sido restablecida exitosamente. Ahora puedes iniciar sesión."
-      signup_success: "Registro exitoso. Por favor, revisa tu correo para verificar tu cuenta."
-      user_associated_success: "Usuario existente asociado exitosamente a la nueva empresa."
-      account_verified_success: "Tu cuenta ha sido verificada exitosamente. ¡Bienvenido!"
-      forgot_password_success: "Si tu correo está registrado, recibirás un enlace para restablecer tu contraseña."
-
-    # Claves específicas para JavaScript
-    js_messages:
-      feedback_sent_success_title: "Feedback Enviado"
-      feedback_sent_success_body: "¡Gracias por tu comentario!"
-      feedback_sent_error: "No se pudo enviar el feedback, por favor intente nuevamente."
-      feedback_rating_error: "Por favor, califica al asistente con las estrellas."
-      feedback_comment_error: "Por favor, escribe tu comentario antes de enviar."
-      context_reloaded: "El contexto ha sido recargado."
-      error_loading_history: "Ocurrió un error al cargar el historial."
-      request_aborted: "La generación de la respuesta ha sido detenida."
-      processing_error: "Ocurrió un error al procesar la solicitud."
-      server_comm_error: "Error de comunicación con el servidor ({status}). Por favor, intente de nuevo más tarde."
-      network_error: "Ocurrió un error de red. Por favor, inténtalo de nuevo en unos momentos."
-      unknown_server_error: "Error desconocido del servidor."
-      loading: "Cargando..."
-      reload_init: "Iniciando recarga de contexto en segundo plano..."
-      no_history_found: "No existe historial de consultas."
+# locales/es.yaml
+ui:
+  login_widget:
+    title: "Iniciar Sesión"
+    welcome_message: "Ingresa tus credenciales o registrate para acceder a la plataforma."
+    email_placeholder: "Correo electrónico"
+    login_button: "Acceder"
+    forgot_password_link: "¿Olvidaste tu contraseña?"
+    no_account_prompt: "¿No tienes una cuenta?"
+    signup_link: "Regístrate"
+
+  signup:
+    title: "Crear una Cuenta"
+    subtitle: "Comienza tu viaje con nosotros hoy."
+    first_name_label: "Nombre"
+    last_name_label: "Apellido"
+    email_label: "Correo Electrónico"
+    password_label: "Contraseña"
+    confirm_password_label: "Confirmar Contraseña"
+    signup_button: "Crear Cuenta"
+    already_have_account: "¿Ya tienes una cuenta?"
+    login_link: "Inicia Sesión"
+    disclaimer: "🔒 Valoramos tu privacidad. Tus datos se usarán exclusivamente para el funcionamiento de la plataforma."
+
+  forgot_password:
+    title: "Recuperar Contraseña"
+    subtitle: "Ingresa tu correo electrónico y te enviaremos un enlace para restablecer tu contraseña."
+    submit_button: "Enviar Enlace de Recuperación"
+    back_to_login: "Volver a Iniciar Sesión"
+
+  change_password:
+    title: "Crear Nueva Contraseña"
+    subtitle: "Estás cambiando la contraseña para <strong>{email}</strong>."
+    temp_code_label: "Código Temporal"
+    temp_code_placeholder: "Revisa tu correo electrónico"
+    new_password_label: "Nueva Contraseña"
+    password_instructions: "Debe contener al menos 8 caracteres, mayúscula, minúscula, número y un carácter especial."
+    confirm_password_label: "Confirmar Nueva Contraseña"
+    save_button: "Guardar Contraseña"
+    back_to_home: "Volver al inicio"
+
+  chat:
+    welcome_message: "¡Hola! ¿En qué te puedo ayudar hoy?"
+    input_placeholder: "Escribe aquí..."
+    prompts_available: "Prompts disponibles"
+    init_context: "Inicializando el contexto de la IA ..."
+
+
+  tooltips:
+    history: "Historial con mis consultas"
+    reload_context: "Forzar Recarga de Contexto"
+    feedback: "Tu feedback es muy importante"
+    usage_guide: "Guía de Uso"
+    onboarding: "Cómo preguntar mejor"
+    logout: "Cerrar sesión"
+    use_prompt_assistant: "Usar Asistente de Prompts"
+    attach_files: "Adjuntar archivos"
+    view_attached_files: "Ver archivos adjuntos"
+  modals:
+    files_title: "Archivos Cargados"
+    feedback_title: "Tu Opinión es Importante"
+    feedback_prompt: "¿Qué tan útil fue la respuesta del asistente?"
+    feedback_comment_label: "Tu comentario nos ayuda a mejorar:"
+    feedback_comment_placeholder: "Escribe aquí tu opinión, sugerencias o comentarios..."
+    history_title: "Historial de Consultas"
+    history_table_date: "Fecha"
+    history_table_query: "Consulta"
+    loading_history: "Cargando historial..."
+    no_history_found: "No se encontró historial de consultas."
+    help_title: "Guía de uso del Asistente IA"
+
+  buttons:
+    cancel: "Cerrar"
+    send: "Enviar"
+    stop: "Detener"
+
+errors:
+  company_not_found: "La empresa {company_short_name} no existe."
+  timeout: "El tiempo de espera ha expirado."
+
+
+  auth:
+    invalid_password: "La contraseña proporcionada es incorrecta."
+    user_not_found: "No se encontró un usuario con ese correo."
+    invalid_or_expired_token: "Token inválido o expirado."
+    session_creation_failed: "No se pudo crear la sesión del usuario."
+    authentication_required: "Autenticación requerida. No se proporcionó cookie de sesión o clave de API."
+    invalid_api_key: "Clave de API inválida o inactiva."
+    no_user_identifier_api: "No se proporcionó user_identifier para la llamada a la API."
+  templates:
+    company_not_found: "Empresa no encontrada."
+    home_template_not_found: "La plantilla de la página de inicio para la empresa '{company_short_name}' no está configurada."
+    processing_error: "Error al procesar el template: {error}"
+    template_not_found: "No se encontro el template: '{template_name}'."
+
+  general:
+    unexpected_error: "Ha ocurrido un error inesperado: {error}."
+    unsupported_language: "El idioma seleccionado no es válido."
+  signup:
+    company_not_found: "La empresa {company_short_name} no existe."
+    incorrect_password_for_existing_user: "La contraseña para el usuario {email} es incorrecta."
+    user_already_registered: "El usuario con email '{email}' ya existe en esta empresa."
+    password_mismatch: "Las contraseñas no coinciden. Por favor, inténtalo de nuevo."
+  change_password:
+    token_expired: "El enlace de cambio de contraseña ha expirado. Por favor, solicita uno nuevo."
+    password_mismatch: "Las contraseñas no coinciden. Por favor, inténtalo nuevamente."
+    invalid_temp_code: "El código temporal no es válido. Por favor, verifica o solicita uno nuevo."
+  forgot_password:
+    user_not_registered: "El usuario con correo {email} no está registrado."
+  verification:
+    token_expired: "El enlace de verificación ha expirado. Por favor, contacta a soporte si necesitas uno nuevo."
+    user_not_found: "El usuario que intentas verificar no existe."
+  validation:
+    password_too_short: "La contraseña debe tener al menos 8 caracteres."
+    password_no_uppercase: "La contraseña debe tener al menos una letra mayúscula."
+    password_no_lowercase: "La contraseña debe tener al menos una letra minúscula."
+    password_no_digit: "La contraseña debe tener al menos un número."
+    password_no_special_char: "La contraseña debe tener al menos un carácter especial."
+
+  services:
+    no_text_file: "El archivo no es texto o la codificación no es UTF-8"
+    no_output_file: "falta el nombre del archivo de salida"
+    no_data_for_excel: "faltan los datos  o no es una lista de diccionarios"
+    no_download_directory: "no esta configurado el directorio temporal para guardar excels"
+    cannot_create_excel: "no se pudo crear el archivo excel"
+    invalid_filename: "Nombre de archivo inválido"
+    file_not_exist : "Archivo no encontrado"
+    path_is_not_a_file : "La ruta no corresponde a un archivo"
+    file_validation_error : "Error validando archivo"
+    user_not_authorized: "Usuario no esta autorizado para esta empresa"
+    account_not_verified: "Tu cuenta no ha sido verificada. Por favor, revisa tu correo."
+    missing_response_id: "No se encontró 'previous_response_id' para '{company_short_name}/{user_identifier}'. Reinicia el contexto."
+    context_rebuild_failed: "No se pudo reconstruir el contexto de la empresa."
+    cannot_read_excel: "No se pudo leer el archivo excel"
+    cannot_read_csv: "No se pudo leer el archivo CSV"
+
+
+api_responses:
+  context_reloaded_success: "El contexto se ha recargado con éxito."
+
+services:
+  mail_sent: "mail enviado exitosamente."
+  start_query: "Hola, cual es tu pregunta?"
+  mail_change_password: "se envio mail para cambio de clave"
+
+flash_messages:
+  password_changed_success: "Tu contraseña ha sido restablecida exitosamente. Ahora puedes iniciar sesión."
+  signup_success: "Registro exitoso. Por favor, revisa tu correo para verificar tu cuenta."
+  signup_success_no_verification: "Registro exitoso."
+  user_associated_success: "Usuario existente asociado exitosamente a la nueva empresa."
+  account_verified_success: "Tu cuenta ha sido verificada exitosamente. ¡Bienvenido!"
+  forgot_password_success: "Si tu correo está registrado, recibirás un enlace para restablecer tu contraseña."
+
+# Claves específicas para JavaScript
+js_messages:
+  feedback_sent_success_title: "Feedback Enviado"
+  feedback_sent_success_body: "¡Gracias por tu comentario!"
+  feedback_sent_error: "No se pudo enviar el feedback, por favor intente nuevamente."
+  feedback_rating_error: "Por favor, califica al asistente con las estrellas."
+  feedback_comment_error: "Por favor, escribe tu comentario antes de enviar."
+  context_reloaded: "El contexto ha sido recargado."
+  error_loading_history: "Ocurrió un error al cargar el historial."
+  request_aborted: "La generación de la respuesta ha sido detenida."
+  processing_error: "Ocurrió un error al procesar la solicitud."
+  server_comm_error: "Error de comunicación con el servidor ({status}). Por favor, intente de nuevo más tarde."
+  network_error: "Ocurrió un error de red. Por favor, inténtalo de nuevo en unos momentos."
+  unknown_server_error: "Error desconocido del servidor."
+  loading: "Cargando..."
+  reload_init: "Iniciando recarga de contexto en segundo plano..."
+  no_history_found: "No existe historial de consultas."
+  example: "Ejemplo:"
 

iatoolkit/repositories/database_manager.py

@@ -4,7 +4,7 @@
 # IAToolkit is open source software.
 
 # database_manager.py
-from sqlalchemy import create_engine, event, inspect
+from sqlalchemy import create_engine, event, inspect, text
 from sqlalchemy.orm import sessionmaker, scoped_session
 from sqlalchemy.engine.url import make_url
 from iatoolkit.repositories.models import Base
@@ -14,14 +14,26 @@ from pgvector.psycopg2 import register_vector
 
 class DatabaseManager:
     @inject
-    def __init__(self, database_url: str, register_pgvector: bool = True):
+    def __init__(self,
+                 database_url: str,
+                 schema: str | None = None,
+                 register_pgvector: bool = True):
         """
         Inicializa el gestor de la base de datos.
         :param database_url: URL de la base de datos.
+        :param schema: Esquema por defecto para la conexión (search_path).
         :param echo: Si True, habilita logs de SQL.
         """
+
+        self.schema = schema
+
+        # FIX HEROKU: replace postgres:// by postgresql:// for compatibility with SQLAlchemy 1.4+
+        if database_url and database_url.startswith("postgres://"):
+            database_url = database_url.replace("postgres://", "postgresql://", 1)
+
         self.url = make_url(database_url)
-        if database_url.startswith('sqlite'):       # for tests
+
+        if database_url.startswith('sqlite'):
             self._engine = create_engine(database_url, echo=False)
         else:
             self._engine = create_engine(
@@ -40,9 +52,30 @@ class DatabaseManager:
                                            expire_on_commit=False)
         self.scoped_session = scoped_session(self.SessionFactory)
 
-        # REGISTRAR pgvector para cada nueva conexión solo en postgres
-        if register_pgvector and self.url.get_backend_name() == 'postgresql':
-            event.listen(self._engine, 'connect', self.on_connect)
+        # Register pgvector for each new connection
+        backend = self.url.get_backend_name()
+        if backend == 'postgresql' or backend == 'postgres':
+            if register_pgvector:
+                event.listen(self._engine, 'connect', self.on_connect)
+
+            # if there is a schema, configure the search_path for each connection
+            if self.schema:
+                event.listen(self._engine, 'checkout', self.set_search_path)
+
+    def set_search_path(self, dbapi_connection, connection_record, connection_proxy):
+        # Configure the search_path for this connection
+        cursor = dbapi_connection.cursor()
+
+        # The defined schema is first, and then public by default
+        try:
+            cursor.execute(f"SET search_path TO {self.schema}, public")
+            cursor.close()
+
+            # commit for persist the change in the session
+            dbapi_connection.commit()
+        except Exception:
+            # if failed, rollback to avoid invalidating the connection
+            dbapi_connection.rollback()
 
     @staticmethod
     def on_connect(dbapi_connection, connection_record):
@@ -62,6 +95,12 @@ class DatabaseManager:
         return self._engine
 
     def create_all(self):
+        # if there is a schema defined, make sure it exists before creating tables
+        backend = self.url.get_backend_name()
+        if self.schema and (backend == 'postgresql' or backend == 'postgres'):
+            with self._engine.begin() as conn:
+                conn.execute(text(f"CREATE SCHEMA IF NOT EXISTS {self.schema}"))
+
         Base.metadata.create_all(self._engine)
 
     def drop_all(self):
@@ -73,22 +112,24 @@ class DatabaseManager:
     def get_all_table_names(self) -> list[str]:
         # Returns a list of all table names in the database
         inspector = inspect(self._engine)
-        return inspector.get_table_names()
+        return inspector.get_table_names(schema=self.schema)
 
     def get_table_schema(self,
                          table_name: str,
-                         schema_name: str | None = None,
+                         db_schema: str,
+                         schema_object_name: str | None = None,
                          exclude_columns: list[str] | None = None) -> str:
         inspector = inspect(self._engine)
 
-        if table_name not in inspector.get_table_names():
-            raise RuntimeError(f"Table '{table_name}' does not exist.")
+        # search the table in the specified schema
+        if table_name not in inspector.get_table_names(schema=db_schema):
+            raise RuntimeError(f"Table '{table_name}' does not exist in database schema '{db_schema}'.")
 
         if exclude_columns is None:
             exclude_columns = []
 
-        # get all thre table columns
-        columns = inspector.get_columns(table_name)
+        # get all the table columns
+        columns = inspector.get_columns(table_name, schema=db_schema)
 
         # construct a json dictionary with the table definition
         json_dict = {
@@ -96,8 +137,12 @@ class DatabaseManager:
             "description": f"Definición de la tabla {table_name}.",
             "fields": []
         }
-        if schema_name:
-            json_dict["description"] += f"Los detalles de cada campo están en el objeto **`{schema_name}`**."
+        if schema_object_name:
+            json_dict["description"] += f"Los detalles de cada campo están en el objeto **`{schema_object_name}`**."
+
+        if db_schema:
+            json_dict["schema"] = db_schema
+            json_dict["description"] += f"Pertenece al esquema **`{db_schema}`**."
 
         # now add every column to the json dictionary
         for col in columns:

iatoolkit/repositories/llm_query_repo.py

@@ -3,7 +3,7 @@
 #
 # IAToolkit is open source software.
 
-from iatoolkit.repositories.models import LLMQuery, Function, Company, Prompt, PromptCategory
+from iatoolkit.repositories.models import LLMQuery, Tool, Company, Prompt, PromptCategory
 from injector import inject
 from iatoolkit.repositories.database_manager import DatabaseManager
 from sqlalchemy import or_
@@ -13,38 +13,50 @@ class LLMQueryRepo:
     def __init__(self, db_manager: DatabaseManager):
         self.session = db_manager.get_session()
 
+    def commit(self):
+        self.session.commit()
+
+    def rollback(self):
+        self.session.rollback()
+
     def add_query(self, query: LLMQuery):
         self.session.add(query)
         self.session.commit()
         return query
 
 
-    def get_company_functions(self, company: Company) -> list[Function]:
+    def get_company_tools(self, company: Company) -> list[Tool]:
         return (
-            self.session.query(Function)
+            self.session.query(Tool)
             .filter(
-                Function.is_active.is_(True),
+                Tool.is_active.is_(True),
                 or_(
-                    Function.company_id == company.id,
-                    Function.system_function.is_(True)
+                    Tool.company_id == company.id,
+                    Tool.system_function.is_(True)
                 )
             )
             .all()
         )
 
-    def create_or_update_function(self, new_function: Function):
-        function = self.session.query(Function).filter_by(company_id=new_function.company_id,
-                                                 name=new_function.name).first()
-        if function:
-            function.description = new_function.description
-            function.parameters = new_function.parameters
-            function.system_function = new_function.system_function
+    def delete_system_tools(self):
+        self.session.query(Tool).filter_by(system_function=True).delete(synchronize_session=False)
+
+    def create_or_update_tool(self, new_tool: Tool):
+        tool = self.session.query(Tool).filter_by(company_id=new_tool.company_id,
+                                                  name=new_tool.name).first()
+        if tool:
+            tool.description = new_tool.description
+            tool.parameters = new_tool.parameters
+            tool.system_function = new_tool.system_function
         else:
-            self.session.add(new_function)
-            function = new_function
+            self.session.add(new_tool)
+            tool = new_tool
 
-        self.session.commit()
-        return function
+        self.session.flush()
+        return tool
+
+    def delete_tool(self, tool: Tool):
+        self.session.query(Tool).filter_by(id=tool.id).delete(synchronize_session=False)
 
     def create_or_update_prompt(self, new_prompt: Prompt):
         prompt = self.session.query(Prompt).filter_by(company_id=new_prompt.company_id,
@@ -53,7 +65,6 @@ class LLMQueryRepo:
             prompt.category_id = new_prompt.category_id
             prompt.description = new_prompt.description
             prompt.order = new_prompt.order
-            prompt.active = new_prompt.active
             prompt.is_system_prompt = new_prompt.is_system_prompt
             prompt.filename = new_prompt.filename
             prompt.custom_fields = new_prompt.custom_fields
@@ -61,7 +72,7 @@ class LLMQueryRepo:
             self.session.add(new_prompt)
             prompt = new_prompt
 
-        self.session.commit()
+        self.session.flush()
         return prompt
 
     def create_or_update_prompt_category(self, new_category: PromptCategory):
@@ -73,7 +84,7 @@ class LLMQueryRepo:
             self.session.add(new_category)
             category = new_category
 
-        self.session.commit()
+        self.session.flush()
         return category
 
     def get_history(self, company: Company, user_identifier: str) -> list[LLMQuery]:
@@ -84,8 +95,9 @@ class LLMQueryRepo:
     def get_prompts(self, company: Company) -> list[Prompt]:
         return self.session.query(Prompt).filter_by(company_id=company.id, is_system_prompt=False).all()
 
+    def get_prompt_by_name(self, company: Company, prompt_name: str):
+        return self.session.query(Prompt).filter_by(company_id=company.id, name=prompt_name).first()
+
     def get_system_prompts(self) -> list[Prompt]:
         return self.session.query(Prompt).filter_by(is_system_prompt=True, active=True).order_by(Prompt.order).all()
 
-    def get_prompt_by_name(self, company: Company, prompt_name: str):
-        return self.session.query(Prompt).filter_by(company_id=company.id, name=prompt_name).first()