iatoolkit 0.7.5__py3-none-any.whl → 0.7.6__py3-none-any.whl

views/llmquery_view.py

@@ -0,0 +1,65 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask import request, jsonify, render_template
+from flask.views import MethodView
+from services.query_service import QueryService
+from common.auth import IAuthentication
+from injector import inject
+import logging
+
+
+class LLMQueryView(MethodView):
+    @inject
+    def __init__(self,
+                 iauthentication: IAuthentication,
+                 query_service: QueryService,
+                 ):
+        self.iauthentication = iauthentication
+        self.query_service = query_service
+
+    def post(self, company_short_name):
+        data = request.get_json()
+        if not data:
+            return jsonify({"error_message": "Cuerpo de la solicitud JSON inválido o faltante"}), 400
+
+        # get access credentials
+        iaut = self.iauthentication.verify(company_short_name, data.get("external_user_id"))
+        if not iaut.get("success"):
+            return jsonify(iaut), 401
+
+        company_id = iaut.get("company_id")
+        external_user_id = iaut.get("external_user_id")
+        local_user_id = iaut.get("local_user_id")
+
+        # now check the form
+        question = data.get("question")
+        files = data.get("files", [])
+        client_data = data.get("client_data", {})
+        prompt_name = data.get("prompt_name")
+        if not question and not prompt_name:
+            return jsonify({"error_message": "Falta la consulta o el prompt_name"}), 400
+
+        try:
+            response = self.query_service.llm_query(
+                company_short_name=company_short_name,
+                external_user_id=external_user_id,
+                local_user_id=local_user_id,
+                question=question,
+                prompt_name=prompt_name,
+                client_data=client_data,
+                files=files)
+            if "error" in response:
+                return {'error_message': response.get("error_message", '')}, 401
+
+            return response, 200
+        except Exception as e:
+            logging.exception(
+                f"Error inesperado al procesar llm_query para company {company_short_name}: {e}")
+            if local_user_id:
+                return render_template("error.html",
+                                       message="Ha ocurrido un error inesperado."), 500
+            else:
+                return jsonify({"error_message": str(e)}), 500

views/login_view.py

@@ -0,0 +1,60 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask.views import MethodView
+from flask import request, redirect, render_template, url_for
+from injector import inject
+from services.profile_service import ProfileService
+
+class LoginView(MethodView):
+    @inject
+    def __init__(self, profile_service: ProfileService):
+        self.profile_service = profile_service
+
+    def get(self, company_short_name: str):
+        # get company info
+        company = self.profile_service.get_company_by_short_name(company_short_name)
+        if not company:
+            return render_template('error.html', message="Empresa no encontrada"), 404
+
+        return render_template('login.html',
+                               company=company,
+                               company_short_name=company_short_name)
+
+    def post(self, company_short_name: str):
+        # get company info
+        company = self.profile_service.get_company_by_short_name(company_short_name)
+        if not company:
+            return render_template('error.html', message="Empresa no encontrada"), 404
+
+        email = request.form.get('email')
+        try:
+            password = request.form.get('password')
+
+            response = self.profile_service.login(
+                company_short_name=company_short_name,
+                email=email,
+                password=password
+                )
+
+            if "error" in response:
+                return render_template(
+                'login.html',
+                                    company_short_name=company_short_name,
+                                    company=company,
+                                    form_data={
+                                           "email": email,
+                                           "password": password,
+                                       },
+                                    alert_message=response["error"]), 400
+
+            return redirect(url_for('chat', company_short_name=company_short_name))
+
+        except Exception as e:
+            return render_template("error.html",
+                                   company=company,
+                                   company_short_name=company_short_name,
+                                   message="Ha ocurrido un error inesperado."), 500
+

views/prompt_view.py

@@ -0,0 +1,37 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask import jsonify
+from flask.views import MethodView
+from services.prompt_manager_service import PromptService
+from common.auth import IAuthentication
+from injector import inject
+import logging
+
+
+class PromptView(MethodView):
+    @inject
+    def __init__(self,
+                 iauthentication: IAuthentication,
+                 prompt_service: PromptService ):
+        self.iauthentication = iauthentication
+        self.prompt_service = prompt_service
+
+    def get(self, company_short_name):
+        # get access credentials
+        iaut = self.iauthentication.verify(company_short_name)
+        if not iaut.get("success"):
+            return jsonify(iaut), 401
+
+        try:
+            response = self.prompt_service.get_user_prompts(company_short_name)
+            if "error" in response:
+                return {'error_message': response["error"]}, 402
+
+            return response, 200
+        except Exception as e:
+            logging.exception(
+                f"Error inesperado al obtener el historial de consultas para company {company_short_name}: {e}")
+            return jsonify({"error_message": str(e)}), 500

views/signup_view.py

@@ -0,0 +1,87 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask.views import MethodView
+from flask import render_template
+from services.profile_service import ProfileService
+from injector import inject
+from itsdangerous import URLSafeTimedSerializer
+from flask import url_for, request
+import os
+
+
+class SignupView(MethodView):
+    @inject
+    def __init__(self, profile_service: ProfileService):
+        self.profile_service = profile_service
+        self.serializer = URLSafeTimedSerializer(os.getenv("USER_VERIF_KEY"))
+
+
+    def get(self, company_short_name: str):
+        # get company info
+        company = self.profile_service.get_company_by_short_name(company_short_name)
+        if not company:
+            return render_template('error.html', message="Empresa no encontrada"), 404
+
+        user_agent = request.user_agent
+        is_mobile = user_agent.platform in ["android", "iphone", "ipad"] or "mobile" in user_agent.string.lower()
+        return render_template('signup.html',
+                               company=company,
+                               company_short_name=company_short_name,
+                               is_mobile=is_mobile)
+
+    def post(self, company_short_name: str):
+        # get company info
+        company = self.profile_service.get_company_by_short_name(company_short_name)
+        if not company:
+            return render_template('error.html', message="Empresa no encontrada"), 404
+
+        try:
+            first_name = request.form.get('first_name')
+            last_name = request.form.get('last_name')
+            email = request.form.get('email')
+            password = request.form.get('password')
+            confirm_password = request.form.get('confirm_password')
+
+            # create verification token and url for verification
+            token = self.serializer.dumps(email, salt='email-confirm')
+            verification_url = url_for('verify_account',
+                                       company_short_name=company_short_name,
+                                       token=token, _external=True)
+
+            response = self.profile_service.signup(
+                company_short_name=company_short_name,
+                email=email,
+                first_name=first_name, last_name=last_name,
+                password=password, confirm_password=confirm_password,
+                verification_url=verification_url)
+
+            if "error" in response:
+                return render_template(
+                'signup.html',
+                                    company=company,
+                                    company_short_name=company_short_name,
+                                    form_data={
+                                           "first_name": first_name,
+                                           "last_name": last_name,
+                                           "email": email,
+                                            "password": password,
+                                            "confirm_password": confirm_password
+                                       },
+                                    alert_message=response["error"]), 400
+
+            # all is OK
+            return render_template(
+                'login.html',
+                            company=company,
+                            company_short_name=company_short_name,
+                                   alert_icon='success',
+                                   alert_message=response["message"]), 200
+        except Exception as e:
+            return render_template("error.html",
+                                   company=company,
+                                   company_short_name=company_short_name,
+                                   message="Ha ocurrido un error inesperado."), 500
+

views/tasks_review_view.py

@@ -0,0 +1,83 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask.views import MethodView
+from flask import request, jsonify
+from services.tasks_service import TaskService
+from repositories.profile_repo import ProfileRepo
+from injector import inject
+import logging
+from typing import Optional
+
+
+class TaskReviewView(MethodView):
+    @inject
+    def __init__(self, task_service: TaskService, profile_repo: ProfileRepo):
+        self.task_service = task_service
+        self.profile_repo = profile_repo
+
+
+    def _authenticate_requesting_company_via_api_key(self) -> tuple[
+        Optional[int], Optional[str], Optional[tuple[dict, int]]]:
+        """
+        Autentica a la compañía  usando su API Key.
+        Retorna (company_id, company_short_name, None) en éxito.
+        Retorna (None, None, (error_json, status_code)) en fallo.
+        """
+        api_key_header = request.headers.get('Authorization')
+        if not api_key_header or not api_key_header.startswith('Bearer '):
+            return None, None, ({"error": "API Key faltante o mal formada en el header Authorization"}, 401)
+
+        api_key_value = api_key_header.split('Bearer ')[1]
+        try:
+            api_key_entry = self.profile_repo.get_active_api_key_entry(api_key_value)
+            if not api_key_entry:
+                return None, None, ({"error": "API Key inválida o inactiva"}, 401)
+
+            # api_key_entry.company ya está cargado por joinedload en get_active_api_key_entry
+            if not api_key_entry.company:  # Sanity check
+                logging.error(
+                    f"ChatTokenRequest: API Key {api_key_value[:5]}... no tiene compañía asociada a pesar de ser válida.")
+                return None, None, ({"error": "Error interno del servidor al verificar API Key"}, 500)
+
+            return api_key_entry.company_id, api_key_entry.company.short_name, None
+
+        except Exception as e:
+            logging.exception(f"ChatTokenRequest: Error interno durante validación de API Key: {e}")
+            return None, None, ({"error": "Error interno del servidor al validar API Key"}, 500)
+
+
+    def post(self, task_id: int):
+        try:
+            # only requests with valid api-key are allowed
+            auth_company_id, auth_company_short_name, error = self._authenticate_requesting_company_via_api_key()
+            if error:
+                return jsonify(error[0]), error[1]
+
+            req_data = request.get_json()
+
+            required_fields = ['review_user', 'approved']
+            for field in required_fields:
+                if field not in req_data:
+                    return jsonify({"error": f"El campo {field} es requerido"}), 400
+
+            review_user = req_data.get('review_user', '')
+            approved = req_data.get('approved', False)
+            comment = req_data.get('comment', '')
+
+            new_task = self.task_service.review_task(
+                task_id=task_id,
+                review_user=review_user,
+                approved=approved,
+                comment=comment)
+
+            return jsonify({
+                "task_id": new_task.id,
+                "status": new_task.status.name
+            }), 200
+
+        except Exception as e:
+            logging.exception("Error al revisar la tarea: %s", str(e))
+            return jsonify({"error": str(e)}), 500

views/tasks_view.py

@@ -0,0 +1,98 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask.views import MethodView
+from flask import request, jsonify
+from services.tasks_service import TaskService
+from repositories.profile_repo import ProfileRepo
+from injector import inject
+from datetime import datetime
+import logging
+from typing import Optional
+
+
+class TaskView(MethodView):
+    @inject
+    def __init__(self, task_service: TaskService, profile_repo: ProfileRepo):
+        self.task_service = task_service
+        self.profile_repo = profile_repo
+
+    def _authenticate_requesting_company_via_api_key(self) -> tuple[
+        Optional[int], Optional[str], Optional[tuple[dict, int]]]:
+        """
+        Autentica a la compañía  usando su API Key.
+        Retorna (company_id, company_short_name, None) en éxito.
+        Retorna (None, None, (error_json, status_code)) en fallo.
+        """
+        api_key_header = request.headers.get('Authorization')
+        if not api_key_header or not api_key_header.startswith('Bearer '):
+            return None, None, ({"error": "API Key faltante o mal formada en el header Authorization"}, 401)
+
+        api_key_value = api_key_header.split('Bearer ')[1]
+        try:
+            api_key_entry = self.profile_repo.get_active_api_key_entry(api_key_value)
+            if not api_key_entry:
+                return None, None, ({"error": "API Key inválida o inactiva"}, 401)
+
+            # api_key_entry.company ya está cargado por joinedload en get_active_api_key_entry
+            if not api_key_entry.company:  # Sanity check
+                logging.error(
+                    f"ChatTokenRequest: API Key {api_key_value[:5]}... no tiene compañía asociada a pesar de ser válida.")
+                return None, None, ({"error": "Error interno del servidor al verificar API Key"}, 500)
+
+            return api_key_entry.company_id, api_key_entry.company.short_name, None
+
+        except Exception as e:
+            logging.exception(f"ChatTokenRequest: Error interno durante validación de API Key: {e}")
+            return None, None, ({"error": "Error interno del servidor al validar API Key"}, 500)
+
+
+    def post(self):
+        try:
+            # only requests with valid api-key are allowed
+            auth_company_id, auth_company_short_name, error = self._authenticate_requesting_company_via_api_key()
+            if error:
+                return jsonify(error[0]), error[1]
+
+            req_data = request.get_json()
+            files = request.files.getlist('files')
+
+            required_fields = ['company', 'task_type', 'client_data']
+            for field in required_fields:
+                if field not in req_data:
+                    return jsonify({"error": f"El campo {field} es requerido"}), 400
+
+            company_short_name = req_data.get('company', '')
+            task_type = req_data.get('task_type', '')
+            client_data = req_data.get('client_data', {})
+            company_task_id = req_data.get('company_task_id', 0)
+            execute_at = req_data.get('execute_at', None)
+
+            # validate date format is parameter is present
+            if execute_at:
+                try:
+                    # date in iso format
+                    execute_at = datetime.fromisoformat(execute_at)
+                except ValueError:
+                    return jsonify({
+                        "error": "El formato de execute_at debe ser YYYY-MM-DD HH:MM:SS"
+                    }), 400
+
+            new_task = self.task_service.create_task(
+                company_short_name=company_short_name,
+                task_type_name=task_type,
+                client_data=client_data,
+                company_task_id=company_task_id,
+                execute_at=execute_at,
+                files=files)
+
+            return jsonify({
+                "task_id": new_task.id,
+                "status": new_task.status.name
+            }), 201
+
+        except Exception as e:
+            logging.exception("Error al crear la tarea: %s", str(e))
+            return jsonify({"error": str(e)}), 500

views/user_feedback_view.py

@@ -0,0 +1,74 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask import request, jsonify, render_template
+from flask.views import MethodView
+from services.user_feedback_service import UserFeedbackService
+from common.auth import IAuthentication
+from injector import inject
+import logging
+
+
+class UserFeedbackView(MethodView):
+    @inject
+    def __init__(self,
+                 iauthentication: IAuthentication,
+                 user_feedback_service: UserFeedbackService ):
+        self.iauthentication = iauthentication
+        self.user_feedback_service = user_feedback_service
+
+    def post(self, company_short_name):
+        data = request.get_json()
+        if not data:
+            return jsonify({"error_message": "Cuerpo de la solicitud JSON inválido o faltante"}), 400
+
+        # get access credentials
+        iaut = self.iauthentication.verify(company_short_name, data.get("external_user_id"))
+        if not iaut.get("success"):
+            return jsonify(iaut), 401
+
+        message = data.get("message")
+        if not message:
+            return jsonify({"error_message": "Falta el mensaje de feedback"}), 400
+        
+        space = data.get("space")
+        if not space:
+            return jsonify({"error_message": "Falta el espacio de Google Chat"}), 400
+        
+        type = data.get("type")
+        if not type:
+            return jsonify({"error_message": "Falta el tipo de feedback"}), 400
+        
+        rating = data.get("rating")
+        if not rating:
+            return jsonify({"error_message": "Falta la calificación"}), 400
+
+        external_user_id = data.get("external_user_id")
+        local_user_id = data.get("local_user_id", 0)
+
+        try:
+            response = self.user_feedback_service.new_feedback(
+                company_short_name=company_short_name,
+                message=message,
+                external_user_id=external_user_id,
+                local_user_id=local_user_id,
+                space=space,
+                type=type,
+                rating=rating
+            )
+
+            if "error" in response:
+                return {'error_message': response["error"]}, 402
+
+            return response, 200
+        except Exception as e:
+            logging.exception(
+                f"Error inesperado al procesar feedback para company {company_short_name}: {e}")
+            if local_user_id:
+                return render_template("error.html",
+                                       message="Ha ocurrido un error inesperado."), 500
+            else:
+                return jsonify({"error_message": str(e)}), 500
+

views/verify_user_view.py

@@ -0,0 +1,55 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask.views import MethodView
+from flask import render_template
+from services.profile_service import ProfileService
+from itsdangerous import URLSafeTimedSerializer, SignatureExpired
+from injector import inject
+import os
+
+
+class VerifyAccountView(MethodView):
+    @inject
+    def __init__(self, profile_service: ProfileService):
+        self.profile_service = profile_service
+        self.serializer = URLSafeTimedSerializer(os.getenv("USER_VERIF_KEY"))
+
+    def get(self, company_short_name: str, token: str):
+        # get company info
+        company = self.profile_service.get_company_by_short_name(company_short_name)
+        if not company:
+            return render_template('error.html', message="Empresa no encontrada"), 404
+
+        try:
+            # decode the token from the URL
+            email = self.serializer.loads(token, salt='email-confirm', max_age=3600*5)
+        except SignatureExpired:
+            return render_template('signup.html',
+                                   company=company,
+                                   company_short_name=company_short_name,
+                                   token=token,
+                                   alert_message="El enlace de verificación ha expirado. Por favor, solicita uno nuevo."), 400
+
+        try:
+            response = self.profile_service.verify_account(email)
+            if "error" in response:
+                return render_template(
+                    'signup.html',
+                    company=company,
+                    company_short_name=company_short_name,
+                    token=token,
+                    alert_message=response["error"]), 400
+
+            return render_template('login.html',
+                                   company=company,
+                                   company_short_name=company_short_name,
+                                   alert_icon='success',
+                                   alert_message=response['message'])
+        except Exception as e:
+            return render_template("error.html",
+                                   company=company,
+                                   company_short_name=company_short_name,
+                                   message="Ha ocurrido un error inesperado."), 500

iatoolkit-0.7.5.dist-info/RECORD

@@ -1,36 +0,0 @@
-common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-common/auth.py,sha256=kKBvZoIm8RPpPFZ6KEm1oowHJfJwSO0mf10Yac8SG9k,8456
-common/exceptions.py,sha256=EXx40n5htp7UiOM6P1xfJ9U6NMcADqm62dlFaKz7ICU,1154
-common/routes.py,sha256=_1ywiXorwEeJ8IAamQwSxU9z8jlAAL_U64tMb01-zFg,4188
-common/session_manager.py,sha256=7D_RuJs60w-1zDr3fOGEz9JW7IZlSXuUHgUT87CzaUo,472
-common/util.py,sha256=5zw1wL_FdPBibGJbQAl1kozgTjUpsOZlPqT3tbWsHp4,15515
-iatoolkit/__init__.py,sha256=GkFxAQHKPifz4Kd8M73Rc8TWRVIxjxkl1N0nsPvb_sU,1743
-iatoolkit/base_company.py,sha256=WmD4o0qFC1n5DW5eRsRsuNfaGot9nxGFcJe3LmibSuE,4259
-iatoolkit/cli_commands.py,sha256=BGuThg19eoSssrBJIqzBNaWpMmyy7u4yRUz0JA7d-vc,2270
-iatoolkit/company_registry.py,sha256=tduqt3oV8iDX_IB1eA7KIgvIxE4edTcy-3qZIXh3Lzw,2549
-iatoolkit/iatoolkit.py,sha256=bbwLkLS_xZGNS0Dki3zX73RhAYMS32IshXDIdr_RVXg,15776
-iatoolkit/system_prompts/format_styles.prompt,sha256=MSMe1qvR3cF_0IbFshn8R0z6Wx6VCHQq1p37rpu5wwk,3576
-iatoolkit/system_prompts/query_main.prompt,sha256=w_9ybgWgiQH4V_RbAXqsvz0M7oOuiyhxcwf-D0CgfA4,3017
-iatoolkit/system_prompts/sql_rules.prompt,sha256=y4nURVnb9AyFwt-lrbMNBHHtZlhk6kC9grYoOhRnrJo,59174
-services/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
-services/benchmark_service.py,sha256=_ruKh9YzrTLtR0ZKrRNxqJQW0HdbwWuFz1gfLzJ9owA,5850
-services/dispatcher_service.py,sha256=y2J-TfrkoCcsOESr3E6E3q5bleJfRibT4ONxs9AxUSg,13959
-services/document_service.py,sha256=8MsJz0pihM9E9Z92PrPqDgQnpMAmpFrbogXr5HscMWM,5926
-services/excel_service.py,sha256=ATPaeAvkLwQAkPZ3AKIUpO73RVyRg0D8c6i37_mcql0,3559
-services/file_processor_service.py,sha256=98yWYF7nIq1nm7nh6IzMmTKaOMTIeqCFWYwXVtV-ZJI,4102
-services/history_service.py,sha256=j0QCqcIIyw7DBy3GrZrEZNk0I4m-uuRoG5g0Z2RCcOE,1586
-services/jwt_service.py,sha256=YoZ9h7_o9xBko-arNQv4MbcwnxoSWVNj4VbZmMo_QGY,3908
-services/load_documents_service.py,sha256=UGfomYz7seWFXawbDuk2t6CyoEr1vggR8vmrCUAeLBg,7190
-services/mail_service.py,sha256=_67pctxZO46DHnWBip51ayuYtWd4bEoS1kg29ACO7_I,2162
-services/profile_service.py,sha256=vTK9TvH_2AFdqgL3sGjhck9LyLGIbvdC2USoaRx82G8,17561
-services/prompt_manager_service.py,sha256=7SMC6N_T4BP4m5-ZNYAL3Y2YWHwl1bytXSgnEqu5bWI,8301
-services/query_service.py,sha256=gvUnq0Vpn4gv5ycQk8-fklh7NDFIXpO1Vt1lT2ugO6Q,15283
-services/search_service.py,sha256=bB3FWFxJi1iYsOdBxyu9tzIO406nQxcyeQzEowpmpjY,1803
-services/sql_service.py,sha256=s84K1ADlvMtum949wgMh8jsmqlOUeL-m_SWfAM4Wsv4,2141
-services/tasks_service.py,sha256=1DdbERlAxIkCpGEylnHDKC-KAsXRJugbaRSzRbPfL58,6790
-services/user_feedback_service.py,sha256=_LeNBYz4hHFapXfYTQVfkkD34gE8j2UeKnyOZ8H0nWo,2442
-services/user_session_context_service.py,sha256=GluNSgqP6W_hFke4oslSnfGnU_b-ph28BHH6jf3EIm0,3797
-iatoolkit-0.7.5.dist-info/METADATA,sha256=LPCUPFDxvY1hpeWfbGx5Whlxh2baz1XoUwMAZv0NjYk,9300
-iatoolkit-0.7.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-iatoolkit-0.7.5.dist-info/top_level.txt,sha256=iJOJr1k6nTgm_uuxyeWodC3pMsTebfQ2qEUz3rkbpB4,26
-iatoolkit-0.7.5.dist-info/RECORD,,