iatoolkit 0.67.0__tar.gz → 0.76.0__tar.gz

iatoolkit-0.76.0/PKG-INFO

@@ -0,0 +1,196 @@
+Metadata-Version: 2.4
+Name: iatoolkit
+Version: 0.76.0
+Summary: IAToolkit
+Author: Fernando Libedinsky
+License-Expression: MIT
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+License-File: LICENSE
+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==1.79.0
+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
+Dynamic: license-file
+
+<div align="center">
+  <h1>IAToolkit</h1>
+  <p><strong>The Open-Source Framework for Building AI Assistants on Your Private Data</strong></p>
+  <p>
+    <a href="https://www.iatoolkit.com">Website</a> |
+    <a href="./docs/index.md">Full Documentation</a> |
+    <a href="./docs/quickstart.md">Quickstart Guide</a>
+  </p>
+</div>
+
+![IAToolkit Demo](./docs/assets/iatoolkit-demo.gif)
+
+---
+
+## ✨ Why IAToolkit?
+
+IAToolkit is more than a framework — it's a complete foundation for building **enterprise-grade AI assistants** 
+capable of understanding your proprietary data, automating workflows, and integrating seamlessly with 
+your existing systems.
+
+Whether you’re building a production chatbot for your company or learning how real AI applications are architected, 
+IAToolkit gives you everything you need to build secure, production-grade assistants — from data access, 
+multi-tenant architecture, and tool execution to prompt orchestration, semantic search, 
+and full UI integration — all powered by a clean, extensible Python API.
+
+---
+
+### 🔌 Connect to Anything  
+Build AI assistants that truly understand your business.
+
+- Connect to **SQL databases** (PostgreSQL, MySQL, SQLite)
+- Query structured data using natural language
+- Perform **semantic search** on PDFs, DOCX, TXT, XLSX
+- Works as a full RAG engine out-of-the-box
+
+---
+
+### 🏢 Multi-Tenant by Design  
+Ideal for SaaS, agencies, consultancies, or large enterprises.
+
+- Isolated **Company Modules**  
+- Each company defines its own:
+  - data sources  
+  - prompts  
+  - custom tools  
+  - branding + UI theme  
+- Clean separation for multi-client deployments  
+- Perfect for scaling from 1 to 100 customers
+- Configure everything using simple YAML 
+
+---
+
+### 🧩 Extensible & Provider-Agnostic  
+Customize every layer and keep full control.
+
+- Add **custom tools** the LLM can call (SQL, API, Python functions)
+- Swap between **OpenAI (GPT)**, **Google Gemini**, or future models
+- Built on dependency injection for maximum modularity
+- Extend prompts, dispatchers, services, and LLM clients effortlessly
+
+---
+
+### 🛡️ Built for Production  
+Designed for real-world systems — reliable, maintainable, and scalable.
+
+- Integrated authentication and session handling  
+- Secure secret management via environment variables  
+- Detailed logging of prompts, tool calls, and token usage  
+- Structured architecture with **90%+ automated test coverage**  
+- Clean separation of UI, business logic, and LLM orchestration  
+- Runs anywhere: local machine, docker, cloud, serverless  
+
+---
+
+### 🎓 Learn How Real AI Applications Are Built  
+IAToolkit is also an excellent learning resource:
+
+- Understand how to orchestrate multiple LLM calls  
+- Learn how tools, SQL, vector search and prompts interact  
+- Explore a complete architecture for real AI products  
+- Use the Sample Company demo to practice building AI assistants  
+- Perfect for developers, students, and teams exploring enterprise AI
+
+---
+
+## 🚀 Getting Started in 3 Minutes
+
+Get your first AI assistant running locally by following our "Hello World" example.
+Our detailed guide will walk you through setting up your virtual environment, configuring your `.env` file, 
+and launching the application.
+
+➡️ **[Start the Quickstart Guide](./docs/quickstart.md)**
+
+## 📚 Documentation
+
+Our documentation is designed for users of all levels, from initial setup to advanced development.
+
+| Guide                                         | Description                                                                                               |
+| --------------------------------------------- | --------------------------------------------------------------------------------------------------------- |
+| 🚀 **[Quickstart Guide](./docs/quickstart.md)** | The fastest way to install, configure, and run IAToolkit for the first time.                                |
+| ⚙️ **[Companies & Components](./docs/companies_and_components.md)** | A deep dive into the `company.yaml` file, the core of all configuration.                  |
+| 💻 **[Programming Guide](./docs/programming_guide.md)** | Understand the internal architecture, services, and design patterns to extend the framework.      |
+| ☁️ **[Deployment Guide](./docs/deployment_guide.md)** | Learn how to deploy your IAToolkit application to a production environment.                             |
+| 🗃️ **[Database Guide](./docs/database_guide.md)** | An overview of the core database schema used by the IAToolkit framework itself.                         |
+
+➡️ **[Explore all documentation](./docs/index.md)**
+
+## 🤝 Contributing
+
+We welcome contributions of all kinds! Whether it's a new feature, a bug fix, or an improvement to the documentation, your help is appreciated. Please read our **[Contributing Guide](./contributing.md)** to get started.
+
+## 📄 License
+
+IAToolkit is open-source software licensed under the **[MIT License](./LICENSE)**.

{iatoolkit-0.67.0 → iatoolkit-0.76.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "iatoolkit"
-version = "0.67.0"
+version = "0.76.0"
 requires-python = ">=3.12"
 description = "IAToolkit"
 readme = "readme.md"

iatoolkit-0.76.0/readme.md

@@ -0,0 +1,114 @@
+<div align="center">
+  <h1>IAToolkit</h1>
+  <p><strong>The Open-Source Framework for Building AI Assistants on Your Private Data</strong></p>
+  <p>
+    <a href="https://www.iatoolkit.com">Website</a> |
+    <a href="./docs/index.md">Full Documentation</a> |
+    <a href="./docs/quickstart.md">Quickstart Guide</a>
+  </p>
+</div>
+
+![IAToolkit Demo](./docs/assets/iatoolkit-demo.gif)
+
+---
+
+## ✨ Why IAToolkit?
+
+IAToolkit is more than a framework — it's a complete foundation for building **enterprise-grade AI assistants** 
+capable of understanding your proprietary data, automating workflows, and integrating seamlessly with 
+your existing systems.
+
+Whether you’re building a production chatbot for your company or learning how real AI applications are architected, 
+IAToolkit gives you everything you need to build secure, production-grade assistants — from data access, 
+multi-tenant architecture, and tool execution to prompt orchestration, semantic search, 
+and full UI integration — all powered by a clean, extensible Python API.
+
+---
+
+### 🔌 Connect to Anything  
+Build AI assistants that truly understand your business.
+
+- Connect to **SQL databases** (PostgreSQL, MySQL, SQLite)
+- Query structured data using natural language
+- Perform **semantic search** on PDFs, DOCX, TXT, XLSX
+- Works as a full RAG engine out-of-the-box
+
+---
+
+### 🏢 Multi-Tenant by Design  
+Ideal for SaaS, agencies, consultancies, or large enterprises.
+
+- Isolated **Company Modules**  
+- Each company defines its own:
+  - data sources  
+  - prompts  
+  - custom tools  
+  - branding + UI theme  
+- Clean separation for multi-client deployments  
+- Perfect for scaling from 1 to 100 customers
+- Configure everything using simple YAML 
+
+---
+
+### 🧩 Extensible & Provider-Agnostic  
+Customize every layer and keep full control.
+
+- Add **custom tools** the LLM can call (SQL, API, Python functions)
+- Swap between **OpenAI (GPT)**, **Google Gemini**, or future models
+- Built on dependency injection for maximum modularity
+- Extend prompts, dispatchers, services, and LLM clients effortlessly
+
+---
+
+### 🛡️ Built for Production  
+Designed for real-world systems — reliable, maintainable, and scalable.
+
+- Integrated authentication and session handling  
+- Secure secret management via environment variables  
+- Detailed logging of prompts, tool calls, and token usage  
+- Structured architecture with **90%+ automated test coverage**  
+- Clean separation of UI, business logic, and LLM orchestration  
+- Runs anywhere: local machine, docker, cloud, serverless  
+
+---
+
+### 🎓 Learn How Real AI Applications Are Built  
+IAToolkit is also an excellent learning resource:
+
+- Understand how to orchestrate multiple LLM calls  
+- Learn how tools, SQL, vector search and prompts interact  
+- Explore a complete architecture for real AI products  
+- Use the Sample Company demo to practice building AI assistants  
+- Perfect for developers, students, and teams exploring enterprise AI
+
+---
+
+## 🚀 Getting Started in 3 Minutes
+
+Get your first AI assistant running locally by following our "Hello World" example.
+Our detailed guide will walk you through setting up your virtual environment, configuring your `.env` file, 
+and launching the application.
+
+➡️ **[Start the Quickstart Guide](./docs/quickstart.md)**
+
+## 📚 Documentation
+
+Our documentation is designed for users of all levels, from initial setup to advanced development.
+
+| Guide                                         | Description                                                                                               |
+| --------------------------------------------- | --------------------------------------------------------------------------------------------------------- |
+| 🚀 **[Quickstart Guide](./docs/quickstart.md)** | The fastest way to install, configure, and run IAToolkit for the first time.                                |
+| ⚙️ **[Companies & Components](./docs/companies_and_components.md)** | A deep dive into the `company.yaml` file, the core of all configuration.                  |
+| 💻 **[Programming Guide](./docs/programming_guide.md)** | Understand the internal architecture, services, and design patterns to extend the framework.      |
+| ☁️ **[Deployment Guide](./docs/deployment_guide.md)** | Learn how to deploy your IAToolkit application to a production environment.                             |
+| 🗃️ **[Database Guide](./docs/database_guide.md)** | An overview of the core database schema used by the IAToolkit framework itself.                         |
+
+➡️ **[Explore all documentation](./docs/index.md)**
+
+## 🤝 Contributing
+
+We welcome contributions of all kinds! Whether it's a new feature, a bug fix, or an improvement to the documentation, your help is appreciated. Please read our **[Contributing Guide](./contributing.md)** to get started.
+
+## 📄 License
+
+IAToolkit is open-source software licensed under the **[MIT License](./LICENSE)**.

iatoolkit-0.76.0/requirements.txt

@@ -0,0 +1,71 @@
+bcrypt==4.2.1
+boto3==1.36.22
+botocore==1.36.22
+build==1.2.2.post1
+click==8.1.8
+cryptography==44.0.3
+Flask==3.1.0
+Flask-Bcrypt==1.0.1
+flask-cors==6.0.0
+Flask-Injector==0.15.0
+Flask-Session==0.8.0
+flatbuffers==24.3.25
+google-ai-generativelanguage==0.6.15
+google-api-core==2.24.1
+google-api-python-client==2.161.0
+google-auth==2.37.0
+google-auth-httplib2==0.2.0
+google-auth-oauthlib==1.2.1
+google-cloud-core==2.4.1
+google-cloud-storage==3.0.0
+google-crc32c==1.6.0
+google-generativeai==0.8.5
+google-resumable-media==2.7.2
+googleapis-common-protos==1.66.0
+gunicorn==23.0.0
+h11==0.14.0
+httpcore==1.0.7
+httplib2==0.22.0
+httptools==0.6.4
+httpx==0.28.0
+httpx-sse==0.4.0
+huggingface-hub==0.31.4
+humanfriendly==10.0
+idna==3.10
+injector==0.22.0
+Jinja2==3.1.5
+langchain==0.3.19
+langchain-core==0.3.35
+langchain-text-splitters==0.3.6
+markdown2==2.5.3
+openai==1.79.0
+openpyxl==3.1.5
+pandas==2.3.1
+pgvector==0.3.6
+pillow==11.0.0
+psutil==7.0.0
+psycopg2-binary==2.9.10
+PyJWT==2.10.1
+PyMuPDF==1.25.0
+python-dotenv==1.0.1
+pytest==8.3.4
+pytest-cov==5.0.0
+pytest-mock==3.14.0
+python-dateutil==2.9.0.post0
+python-docx==1.1.2
+pytesseract==0.3.13
+pytz==2025.2
+PyYAML==6.0.2
+redis==5.2.1
+regex==2024.11.6
+requests==2.32.3
+requests-oauthlib==2.0.0
+requests-toolbelt==1.0.0
+s3transfer==0.11.2
+sib-api-v3-sdk==7.6.0
+SQLAlchemy==2.0.36
+tiktoken==0.8.0
+tokenizers==0.21.0
+websocket-client==1.8.0
+websockets==14.1
+Werkzeug==3.1.3

{iatoolkit-0.67.0 → iatoolkit-0.76.0}/src/iatoolkit/__init__.py

@@ -5,21 +5,18 @@ IAToolkit Package
 # Expose main classes and functions at the top level of the package
 
 # main IAToolkit class
-from .iatoolkit import IAToolkit, current_iatoolkit, create_app
+from .iatoolkit import IAToolkit, create_app, current_iatoolkit
 
 # for registering the client companies
 from .company_registry import register_company
 from .base_company import BaseCompany
-from iatoolkit.repositories.database_manager import DatabaseManager
 
 # --- Services ---
-from iatoolkit.services.configuration_service import ConfigurationService
 from iatoolkit.services.query_service import QueryService
-from iatoolkit.services.sql_service import SqlService
 from iatoolkit.services.document_service import DocumentService
 from iatoolkit.services.search_service import SearchService
+from iatoolkit.services.sql_service import SqlService
 from iatoolkit.services.load_documents_service import LoadDocumentsService
-from iatoolkit.services.excel_service import ExcelService
 from iatoolkit.infra.call_service import CallServiceClient
 
 __all__ = [
@@ -28,11 +25,8 @@ __all__ = [
     'current_iatoolkit',
     'register_company',
     'BaseCompany',
-    'DatabaseManager',
     'QueryService',
-    'ConfigurationService',
     'SqlService',
-    'ExcelService',
     'DocumentService',
     'SearchService',
     'LoadDocumentsService',

{iatoolkit-0.67.0 → iatoolkit-0.76.0}/src/iatoolkit/base_company.py

@@ -9,7 +9,7 @@ from iatoolkit.repositories.profile_repo import ProfileRepo
 from iatoolkit.repositories.llm_query_repo import LLMQueryRepo
 from iatoolkit.services.prompt_manager_service import PromptService
 from iatoolkit.repositories.models import Company, Function, PromptCategory
-from iatoolkit import IAToolkit
+from .iatoolkit import IAToolkit
 
 
 class BaseCompany(ABC):
@@ -20,6 +20,7 @@ class BaseCompany(ABC):
         self.llm_query_repo: LLMQueryRepo = injector.get(LLMQueryRepo)
         self.prompt_service: PromptService = injector.get(PromptService)
         self.company: Company | None = None
+        self.company_short_name: str
 
     def _create_company(self,
                         short_name: str,
@@ -68,12 +69,6 @@ class BaseCompany(ABC):
             **kwargs
         )
 
-
-    @abstractmethod
-    # get context specific for this company
-    def get_company_context(self, **kwargs) -> str:
-        raise NotImplementedError("La subclase debe implementar el método get_company_context()")
-
     @abstractmethod
     # get context specific for this company
     def get_user_info(self, user_identifier: str) -> dict:
@@ -85,10 +80,6 @@ class BaseCompany(ABC):
         raise NotImplementedError("La subclase debe implementar el método handle_request()")
 
     @abstractmethod
-    # get context specific for the query
-    def get_metadata_from_filename(self, filename: str) -> dict:
-        raise NotImplementedError("La subclase debe implementar el método get_query_context()")
-
     def register_cli_commands(self, app):
         """
         optional method for a company definition of it's cli commands

{iatoolkit-0.67.0 → iatoolkit-0.76.0}/src/iatoolkit/cli_commands.py

@@ -5,7 +5,7 @@
 
 import click
 import logging
-from iatoolkit import IAToolkit
+from .iatoolkit import IAToolkit
 from iatoolkit.services.profile_service import ProfileService
 
 def register_core_commands(app):
@@ -17,26 +17,24 @@ def register_core_commands(app):
         """⚙️ Genera una nueva API key para una compañía ya registrada."""
         try:
             profile_service = IAToolkit.get_instance().get_injector().get(ProfileService)
-            click.echo(f"🔑 Generando API key para '{company_short_name}'...")
+            click.echo(f"🔑 Generating API-KEY for company: '{company_short_name}'...")
             result = profile_service.new_api_key(company_short_name)
 
             if 'error' in result:
                 click.echo(f"❌ Error: {result['error']}")
-                click.echo("👉 Asegúrate de que el nombre de la compañía es correcto y está registrada.")
+                click.echo("👉 Make sure the company is registered and valid.")
             else:
-                click.echo("✅ ¡Configuración lista! Agrega esta variable a tu entorno:")
+                click.echo("✅ ¡Api-key is ready! add this variable to your environment:")
                 click.echo(f"IATOOLKIT_API_KEY='{result['api-key']}'")
         except Exception as e:
             logging.exception(e)
-            click.echo(f"❌ Ocurrió un error inesperado durante la configuración: {e}")
+            click.echo(f"❌ unexpectd error during the configuration: {e}")
 
     @app.cli.command("encrypt-key")
     @click.argument("key")
     def encrypt_llm_api_key(key: str):
         from iatoolkit.common.util import Utility
 
-
-
         util = IAToolkit.get_instance().get_injector().get(Utility)
         try:
             encrypt_key = util.encrypt_key(key)

{iatoolkit-0.67.0 → iatoolkit-0.76.0}/src/iatoolkit/common/exceptions.py

@@ -37,6 +37,7 @@ class IAToolkitException(Exception):
         LOAD_DOCUMENT_ERROR = 25
         INVALID_USER = 26
         VECTOR_STORE_ERROR = 27
+        EMBEDDING_ERROR = 28
 
 
 

{iatoolkit-0.67.0 → iatoolkit-0.76.0}/src/iatoolkit/common/routes.py

@@ -20,13 +20,13 @@ def register_views(injector, app):
     from iatoolkit.views.verify_user_view import VerifyAccountView
     from iatoolkit.views.forgot_password_view import ForgotPasswordView
     from iatoolkit.views.change_password_view import ChangePasswordView
-    from iatoolkit.views.file_store_api_view import FileStoreApiView
+    from iatoolkit.views.load_document_api_view import LoadDocumentApiView
     from iatoolkit.views.user_feedback_api_view import UserFeedbackApiView
     from iatoolkit.views.prompt_api_view import PromptApiView
     from iatoolkit.views.history_api_view import HistoryApiView
     from iatoolkit.views.help_content_api_view import HelpContentApiView
     from iatoolkit.views.profile_api_view import UserLanguageApiView  # <-- Importa la nueva vista
-
+    from iatoolkit.views.embedding_api_view import EmbeddingApiView
     from iatoolkit.views.login_view import LoginView, FinalizeContextView
     from iatoolkit.views.external_login_view import ExternalLoginView, RedeemTokenApiView
     from iatoolkit.views.logout_api_view import LogoutApiView
@@ -97,7 +97,11 @@ def register_views(injector, app):
     app.add_url_rule('/tasks/review/<int:task_id>', view_func=TaskReviewApiView.as_view('tasks-review'))
 
     # this endpoint is for upload documents into the vector store (api-key)
-    app.add_url_rule('/api/load', view_func=FileStoreApiView.as_view('load_api'))
+    app.add_url_rule('/api/load-document', view_func=LoadDocumentApiView.as_view('load-document'), methods=['POST'])
+
+    # this endpoint is for generating embeddings for a given text
+    app.add_url_rule('/<company_short_name>/api/embedding',
+                     view_func=EmbeddingApiView.as_view('embedding_api'))
 
 
     @app.route('/download/<path:filename>')

{iatoolkit-0.67.0 → iatoolkit-0.76.0}/src/iatoolkit/common/util.py

@@ -343,6 +343,6 @@ class Utility:
 
     def is_gemini_model(self, model: str) -> bool:
         gemini_models = [
-            'gemini', 'gemini-2.5-pro'
+            'gemini', 'gemini-2.5-pro', 'gemini-3'
         ]
         return any(gemini_model in model.lower() for gemini_model in gemini_models)