iatoolkit 0.3.9__py3-none-any.whl → 0.107.4__py3-none-any.whl

iatoolkit/views/profile_api_view.py

@@ -0,0 +1,46 @@
+# iatoolkit/views/profile_api_view.py
+from flask import request, jsonify
+from flask.views import MethodView
+from injector import inject
+from iatoolkit.services.auth_service import AuthService
+from iatoolkit.services.profile_service import ProfileService
+
+
+class UserLanguageApiView(MethodView):
+    """
+    API endpoint for managing user language preferences.
+    """
+
+    @inject
+    def __init__(self,
+                 auth_service: AuthService,
+                 profile_service: ProfileService):
+        self.auth_service = auth_service
+        self.profile_service = profile_service
+
+    def post(self):
+        """
+        Handles POST requests to update the user's preferred language.
+        Expects a JSON body with a 'language' key, e.g., {"language": "en"}.
+        """
+        # 1. Authenticate the user from the current session.
+        auth_result = self.auth_service.verify()
+        if not auth_result.get("success"):
+            return jsonify(auth_result), auth_result.get("status_code")
+
+        user_identifier = auth_result.get('user_identifier')
+
+        # 2. Validate request body
+        data = request.get_json()
+        if not data or 'language' not in data:
+            return jsonify({"error_message": "Missing 'language' field in request body"}), 400
+
+        new_lang = data.get('language')
+
+        # 3. Call the service to perform the update
+        update_result = self.profile_service.update_user_language(user_identifier, new_lang)
+
+        if not update_result.get('success'):
+            return jsonify(update_result), 400
+
+        return jsonify({"message": "Language preference updated successfully"}), 200

iatoolkit/views/prompt_api_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 iatoolkit.services.prompt_service import PromptService
+from iatoolkit.services.auth_service import AuthService
+from injector import inject
+import logging
+
+
+class PromptApiView(MethodView):
+    @inject
+    def __init__(self,
+                 auth_service: AuthService,
+                 prompt_service: PromptService ):
+        self.auth_service = auth_service
+        self.prompt_service = prompt_service
+
+    def get(self, company_short_name):
+        try:
+            # get access credentials
+            auth_result = self.auth_service.verify(anonymous=True)
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get('status_code')
+
+            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"unexpected error getting company prompts: {e}")
+            return jsonify({"error_message": str(e)}), 500

iatoolkit/views/root_redirect_view.py

@@ -0,0 +1,22 @@
+from flask import redirect, url_for
+from flask.views import MethodView
+from iatoolkit.company_registry import get_company_registry
+
+
+class RootRedirectView(MethodView):
+    """
+    Vista que redirige la raíz '/' al home de la primera compañía disponible.
+    """
+
+    def get(self):
+        registry = get_company_registry()
+        companies = registry.get_all_company_instances()
+
+        if companies:
+            # Obtener el short_name de la primera compañía registrada.
+            # En Python 3.7+, los diccionarios mantienen el orden de inserción.
+            first_company_short_name = next(iter(companies))
+            return redirect(url_for('home', company_short_name=first_company_short_name))
+
+        # Fallback: Si no hay compañías, ir al index genérico (o a un 404)
+        return redirect(url_for('index'))

iatoolkit/views/signup_view.py

@@ -0,0 +1,100 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask.views import MethodView
+from flask import render_template, request, url_for, redirect, flash
+from iatoolkit.services.profile_service import ProfileService
+from iatoolkit.services.branding_service import BrandingService
+from iatoolkit.services.i18n_service import I18nService
+from injector import inject
+from itsdangerous import URLSafeTimedSerializer
+import os
+
+
+class SignupView(MethodView):
+    @inject
+    def __init__(self, profile_service: ProfileService,
+                 branding_service: BrandingService,
+                 i18n_service: I18nService):
+        self.profile_service = profile_service
+        self.branding_service = branding_service # 3. Guardar la instancia
+        self.i18n_service = i18n_service
+
+        self.serializer = URLSafeTimedSerializer(os.getenv("IATOOLKIT_SECRET_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=self.i18n_service.t('errors.templates.company_not_found')), 404
+
+        branding_data = self.branding_service.get_company_branding(company_short_name)
+        current_lang = request.args.get("lang") or "en"
+
+        return render_template('signup.html',
+                               company_short_name=company_short_name,
+                               branding=branding_data,
+                               lang=current_lang)
+
+    def post(self, company_short_name: str):
+        try:
+            company = self.profile_service.get_company_by_short_name(company_short_name)
+            if not company:
+                return render_template('error.html',
+                                     message=self.i18n_service.t('errors.templates.company_not_found')), 404
+
+            branding_data = self.branding_service.get_company_branding(company_short_name)
+
+            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')
+
+            # get the language from the form, then query
+            current_lang = request.form.get("lang") or request.args.get("lang")
+
+            # 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:
+                flash(response["error"], 'error')
+                return render_template(
+                    'signup.html',
+                    company_short_name=company_short_name,
+                    branding=branding_data,
+                    lang=current_lang,
+                    form_data={
+                        "first_name": first_name,
+                        "last_name": last_name,
+                        "email": email,
+                        "password": password,
+                        "confirm_password": confirm_password
+                    }), 400
+
+            flash(response["message"], 'success')
+            return redirect(url_for('home', company_short_name=company_short_name, lang=current_lang))
+
+        except Exception as e:
+            message = self.i18n_service.t('errors.templates.processing_error', error=str(e))
+            return render_template(
+                "error.html",
+                company_short_name=company_short_name,
+                branding=branding_data,
+                message=message,
+                lang=current_lang
+            ), 500

iatoolkit/views/static_page_view.py

@@ -0,0 +1,27 @@
+from flask import render_template
+from flask.views import MethodView
+from injector import inject
+
+
+class StaticPageView(MethodView):
+    """
+    View genérica para servir páginas estáticas simples (sin lógica de negocio compleja).
+    """
+
+    @inject
+    def __init__(self):
+        pass
+
+    def get(self, page_name: str):
+        # Mapeo seguro de nombres de página a plantillas
+        # Esto evita que se intente cargar cualquier archivo arbitrario
+        valid_pages = {
+            'foundation': 'docs/foundation.html',
+            'mini_project': 'docs/mini_project.html'
+        }
+
+        if page_name not in valid_pages:
+            # Si la página no existe, podríamos retornar un 404 o redirigir al index
+            return render_template('error.html', message=f"Página no encontrada: {page_name}"), 404
+
+        return render_template(valid_pages[page_name])

iatoolkit/views/user_feedback_api_view.py

@@ -0,0 +1,60 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask import request, jsonify
+from flask.views import MethodView
+from iatoolkit.services.user_feedback_service import UserFeedbackService
+from iatoolkit.services.auth_service import AuthService
+from injector import inject
+import logging
+
+
+class UserFeedbackApiView(MethodView):
+    @inject
+    def __init__(self,
+                 auth_service: AuthService,
+                 user_feedback_service: UserFeedbackService ):
+        self.auth_service = auth_service
+        self.user_feedback_service = user_feedback_service
+
+    def post(self, company_short_name):
+        try:
+            # get access credentials
+            auth_result = self.auth_service.verify()
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get("status_code")
+
+            user_identifier = auth_result.get('user_identifier')
+
+            data = request.get_json()
+            if not data:
+                return jsonify({"error_message": "invalid json body"}), 402
+
+            # these validations are performed also in the frontend
+            # the are localized in the front
+            message = data.get("message")
+            if not message:
+                return jsonify({"error_message": "missing feedback message"}), 400
+
+            rating = data.get("rating")
+            if not rating:
+                return jsonify({"error_message": "missing rating"}), 400
+
+            response = self.user_feedback_service.new_feedback(
+                company_short_name=company_short_name,
+                message=message,
+                user_identifier=user_identifier,
+                rating=rating
+            )
+
+            if "error" in response:
+                return {'error_message': response["error"]}, 402
+
+            return response, 200
+        except Exception as e:
+            logging.exception(
+                f"unexpected error processing feedback for {company_short_name}: {e}")
+            return jsonify({"error_message": str(e)}), 500
+

iatoolkit/views/users_api_view.py

@@ -0,0 +1,33 @@
+from flask.views import MethodView
+from flask import jsonify
+from injector import inject
+from iatoolkit.services.auth_service import AuthService
+from iatoolkit.services.profile_service import ProfileService
+import logging
+
+
+class UsersApiView(MethodView):
+    """
+    list company users and their roles
+    """
+    @inject
+    def __init__(self,
+                 auth_service: AuthService,
+                 profile_service: ProfileService):
+        self.auth_service = auth_service
+        self.profile_service = profile_service
+
+    def get(self, company_short_name: str):
+        try:
+            auth_result = self.auth_service.verify(anonymous=True)
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get("status_code", 401)
+
+            # get users of the company
+            users = self.profile_service.get_company_users(company_short_name)
+
+            return jsonify(users), 200
+
+        except Exception as e:
+            logging.exception(f"Error fetching users for {company_short_name}: {e}")
+            return jsonify({"error": "Unexpected error", "details": str(e)}), 500

iatoolkit/views/verify_user_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 render_template, url_for, redirect, session, flash
+from iatoolkit.services.profile_service import ProfileService
+from itsdangerous import URLSafeTimedSerializer, SignatureExpired
+from iatoolkit.services.branding_service import BrandingService
+from iatoolkit.services.i18n_service import I18nService
+from injector import inject
+import os
+
+
+class VerifyAccountView(MethodView):
+    @inject
+    def __init__(self,
+                 profile_service: ProfileService,
+                 branding_service: BrandingService,
+                 i18n_service: I18nService):
+        self.profile_service = profile_service
+        self.branding_service = branding_service
+        self.i18n_service = i18n_service
+        self.serializer = URLSafeTimedSerializer(os.getenv("IATOOLKIT_SECRET_KEY"))
+
+    def get(self, company_short_name: str, token: str):
+        try:
+            # get company info
+            company = self.profile_service.get_company_by_short_name(company_short_name)
+            if not company:
+                return render_template('error.html',
+                                       message=self.i18n_service.t('errors.templates.company_not_found')), 404
+
+            branding_data = self.branding_service.get_company_branding(company_short_name)
+            try:
+                # decode the token from the URL
+                email = self.serializer.loads(token, salt='email-confirm', max_age=3600*5)
+            except SignatureExpired:
+                flash(self.i18n_service.t('errors.verification.token_expired'), 'error')
+                return render_template('signup.html',
+                                       company_short_name=company_short_name,
+                                       branding=branding_data,
+                                       token=token), 400
+
+            response = self.profile_service.verify_account(email)
+            if "error" in response:
+                flash(response["error"], 'error')
+                return render_template(
+                    'signup.html',
+                    company_short_name=company_short_name,
+                    branding=branding_data,
+                    token=token), 400
+
+            flash(response['message'], 'success')
+            return redirect(url_for('home', company_short_name=company_short_name))
+
+        except Exception as e:
+            flash(self.i18n_service.t('errors.general.unexpected_error', error=str(e)), 'error')
+            return redirect(url_for('home', company_short_name=company_short_name))

iatoolkit-0.107.4.dist-info/METADATA

@@ -0,0 +1,268 @@
+Metadata-Version: 2.4
+Name: iatoolkit
+Version: 0.107.4
+Summary: IAToolkit
+Author: Fernando Libedinsky
+License-Expression: MIT
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+License-File: LICENSE
+License-File: LICENSE_COMMUNITY.md
+Requires-Dist: bcrypt==4.2.1
+Requires-Dist: boto3==1.36.22
+Requires-Dist: botocore==1.36.22
+Requires-Dist: build==1.2.2.post1
+Requires-Dist: click==8.1.8
+Requires-Dist: cryptography==44.0.3
+Requires-Dist: Flask==3.1.0
+Requires-Dist: Flask-Bcrypt==1.0.1
+Requires-Dist: flask-cors==6.0.0
+Requires-Dist: Flask-Injector==0.15.0
+Requires-Dist: Flask-Session==0.8.0
+Requires-Dist: flatbuffers==24.3.25
+Requires-Dist: google-ai-generativelanguage==0.6.15
+Requires-Dist: google-api-core==2.24.1
+Requires-Dist: google-api-python-client==2.161.0
+Requires-Dist: google-auth==2.37.0
+Requires-Dist: google-auth-httplib2==0.2.0
+Requires-Dist: google-auth-oauthlib==1.2.1
+Requires-Dist: google-cloud-core==2.4.1
+Requires-Dist: google-cloud-storage==3.0.0
+Requires-Dist: google-crc32c==1.6.0
+Requires-Dist: google-generativeai==0.8.5
+Requires-Dist: google-resumable-media==2.7.2
+Requires-Dist: googleapis-common-protos==1.66.0
+Requires-Dist: gunicorn==23.0.0
+Requires-Dist: h11==0.14.0
+Requires-Dist: httpcore==1.0.7
+Requires-Dist: httplib2==0.22.0
+Requires-Dist: httptools==0.6.4
+Requires-Dist: httpx==0.28.0
+Requires-Dist: httpx-sse==0.4.0
+Requires-Dist: huggingface-hub==0.31.4
+Requires-Dist: humanfriendly==10.0
+Requires-Dist: idna==3.10
+Requires-Dist: injector==0.22.0
+Requires-Dist: Jinja2==3.1.5
+Requires-Dist: langchain==0.3.19
+Requires-Dist: langchain-core==0.3.35
+Requires-Dist: langchain-text-splitters==0.3.6
+Requires-Dist: markdown2==2.5.3
+Requires-Dist: openai==2.8.1
+Requires-Dist: openpyxl==3.1.5
+Requires-Dist: pandas==2.3.1
+Requires-Dist: pgvector==0.3.6
+Requires-Dist: pillow==11.0.0
+Requires-Dist: psutil==7.0.0
+Requires-Dist: psycopg2-binary==2.9.10
+Requires-Dist: PyJWT==2.10.1
+Requires-Dist: PyMuPDF==1.25.0
+Requires-Dist: python-dotenv==1.0.1
+Requires-Dist: pytest==8.3.4
+Requires-Dist: pytest-cov==5.0.0
+Requires-Dist: pytest-mock==3.14.0
+Requires-Dist: python-dateutil==2.9.0.post0
+Requires-Dist: python-docx==1.1.2
+Requires-Dist: pytesseract==0.3.13
+Requires-Dist: pytz==2025.2
+Requires-Dist: PyYAML==6.0.2
+Requires-Dist: redis==5.2.1
+Requires-Dist: regex==2024.11.6
+Requires-Dist: requests==2.32.3
+Requires-Dist: requests-oauthlib==2.0.0
+Requires-Dist: requests-toolbelt==1.0.0
+Requires-Dist: s3transfer==0.11.2
+Requires-Dist: sib-api-v3-sdk==7.6.0
+Requires-Dist: SQLAlchemy==2.0.36
+Requires-Dist: tiktoken==0.8.0
+Requires-Dist: tokenizers==0.21.0
+Requires-Dist: websocket-client==1.8.0
+Requires-Dist: websockets==14.1
+Requires-Dist: Werkzeug==3.1.3
+Requires-Dist: pyjwt[crypto]>=2.8.0
+Dynamic: license-file
+
+# 🧠 IAToolkit — Open-Source Framework for Real-World AI Assistants
+
+Build private, production-grade AI assistants that run entirely inside your environment and speak the language of your business.
+
+IAToolkit is not a demo wrapper or a prompt playground — it is a **full architecture** for implementing intelligent systems that combine LLMs, SQL data, internal documents, tools, workflows, and multi-tenant business logic.
+
+---
+
+## 🚀 Why IAToolkit?
+
+Modern AI development is fragmented: LangChain handles chains, LlamaIndex handles documents, 
+your backend handles SQL, your frontend handles chats, and your devs glue everything together.
+
+**IAToolkit brings all of this into one unified, production-ready framework.**
+
+It focuses on:
+
+- **real-world data** (SQL + documents)
+- **real workflows** (LLM tools + python services)
+- **real multi-tenant architecture** (1 company → many companies)
+- **real constraints** (security, reproducibility, governance)
+- **real deployment** (your servers, your infrastructure)
+
+IAToolkit lets you build the assistant that *your* organization needs — not a generic chatbot.
+
+---
+
+## 🧩 Architecture in a Nutshell
+
+IAToolkit is a structured, layered framework:
+
+Interfaces (Web & API)
+        ↓
+Intelligence Layer (prompts, tools, SQL orchestration, RAG)
+        ↓
+Execution Layer (services, workflows, validation)
+        ↓
+Data Access (SQLAlchemy, connectors)
+        ↓
+Company Modules (company.yaml + custom tools)
+
+### ✔ Interfaces  
+Chat UI, REST API, auth, sessions, JSON/HTML responses.
+
+### ✔ Intelligence Layer  
+Core logic: prompt rendering, SQL orchestration, RAG, LLM tool dispatching.
+
+### ✔ Execution Layer  
+Python services that implement real workflows: querying data, generating reports, retrieving documents, executing business logic.
+
+### ✔ Data Access  
+A clean repository pattern using SQLAlchemy.
+
+### ✔ Company Modules  
+Each company has:
+
+- its own `company.yaml`
+- its own prompts
+- its own tools
+- its own services
+- its own vector store & SQL context
+
+This modularity allows **true multi-tenancy**.
+
+---
+
+## 🔌 Connect to Anything
+
+IAToolkit integrates naturally with:
+
+- **SQL databases** (PostgreSQL, MySQL, SQL Server, etc.)
+- **Document retrieval** (PDF, text, embeddings)
+- **External APIs**
+- **Internal microservices**
+- **Custom Python tools**
+
+It also includes a **production-grade RAG pipeline**, combining:
+
+- embeddings  
+- chunking  
+- hybrid search  
+- SQL queries + document retrieval  
+- tool execution  
+
+Everything orchestrated through the Intelligence Layer.
+
+---
+
+## 🏢 Multi-Tenant Architecture
+
+A single installation of IAToolkit can power assistants for multiple companies, departments, or customers.
+```text
+companies/
+    company_a
+    company_b
+    company_c
+```
+Each Company is fully isolated:
+
+- prompts  
+- tools  
+- credentials  
+- documents  
+- SQL contexts  
+- business rules  
+
+This makes IAToolkit ideal for SaaS products, agencies, consultancies, and organizations with multiple business units.
+
+---
+
+## 🆓 Community Edition vs Enterprise Edition
+
+IAToolkit follows a modern **open-core** model:
+
+### 🟦 Community Edition (MIT License)
+- Full Open-Source Core  
+- SQL + Basic RAG  
+- One Company  
+- Custom Python tools  
+- Self-managed deployment  
+
+Perfect for developers, small teams, single-business use cases, and experimentation.
+
+### 🟥 Enterprise Edition (Commercial License)
+- Unlimited Companies (multi-tenant)  
+- Payment services integration 
+- Enterprise Agent Workflows
+- SSO integration
+- Priority support & continuous updates  
+- Activation via **License Key**  
+
+👉 Licensing information:  
+- [Community Edition (MIT)](LICENSE_COMMUNITY.md)  
+- [Enterprise License](ENTERPRISE_LICENSE.md)
+
+---
+
+## 🧩 Who Is IAToolkit For?
+
+- Companies building internal “ChatGPT for the business”  
+- SaaS products adding AI assistants for multiple customers  
+- AI teams that need reproducible prompts and controlled tools  
+- Developers who want real workflows, not toy demos  
+- Organizations requiring privacy, security, and self-hosting  
+- Teams working with SQL-heavy business logic  
+- Consultancies deploying AI for multiple clients  
+
+---
+
+## ⭐ Key Differentiators
+
+- prioritizes **architecture-first design**, not chains or wrappers  
+- supports **multi-company** out of the box  
+- integrates **SQL, RAG, and tools** into a single intelligence layer  
+- keeps **business logic isolated** inside Company modules  
+- runs entirely **on your own infrastructure**  
+- ships with a **full web chat**, and API.  
+- is built for **production**, not prototypes  
+
+---
+
+## 📚 Documentation
+
+- 🚀 **[Quickstart](docs/quickstart.md)** – Set up your environment and run the project
+- ☁️ **[Deployment Guide](docs/deployment_guide.md)** – Production deployment instructions
+- 🏗️ **[Companies & Components](docs/companies_and_components.md)** – how Company modules work
+- 🧠 **[Programming Guide](docs/programming_guide.md)** – services, intelligence layer, dispatching
+- 🗃️ **[Database Guide](docs/database_guide.md)** – internal schema overview
+- 🌱 **[Foundation Article](https://iatoolkit.com/pages/foundation)** – the “Why” behind the architecture
+- 📘 **[Mini-Project (3 months)](https://iatoolkit.com/pages/mini_project)** – how to deploy a corporate AI assistant
+
+
+---
+
+## 🤝 Contributing
+
+IAToolkit is open-source and community-friendly.  
+PRs, issues, ideas, and feedback are always welcome.
+
+---
+
+## ⭐ Support the Project
+
+If you find IAToolkit useful, please **star the GitHub repo** — it helps visibility and adoption.