iatoolkit 0.7.9__py3-none-any.whl → 0.7.11__py3-none-any.whl

iatoolkit/__init__.py

@@ -16,20 +16,24 @@ from .base_company import BaseCompany
 
 # --- Services ---
 # Assuming they are in a 'services' sub-package
-from services.sql_service import SqlService
-from services.excel_service import ExcelService
-from services.dispatcher_service import Dispatcher
-from services.document_service import DocumentService
-from services.search_service import SearchService
-from services.load_documents_service import LoadDocumentsService
-from repositories.profile_repo import ProfileRepo
-from repositories.llm_query_repo import LLMQueryRepo
-from services.query_service import QueryService
-from services.prompt_manager_service import PromptService
-from repositories.database_manager import DatabaseManager
-from infra.call_service import CallServiceClient
-from common.util import Utility
-from repositories.models import Base, Company, Function, TaskType, Prompt, PromptCategory
+from iatoolkit.services.sql_service import SqlService
+from iatoolkit.services.excel_service import ExcelService
+from iatoolkit.services.dispatcher_service import Dispatcher
+from iatoolkit.services.document_service import DocumentService
+from iatoolkit.services.search_service import SearchService
+from iatoolkit.services.load_documents_service import LoadDocumentsService
+from iatoolkit.repositories.profile_repo import ProfileRepo
+from iatoolkit.repositories.llm_query_repo import LLMQueryRepo
+
+from iatoolkit.services.query_service import QueryService
+from iatoolkit.services.prompt_manager_service import PromptService
+from iatoolkit.repositories.database_manager import DatabaseManager
+
+from iatoolkit.infra.call_service import CallServiceClient
+from iatoolkit.common.util import Utility
+
+
+from iatoolkit.repositories.models import Base, Company, Function, TaskType, Prompt, PromptCategory
 
 
 __all__ = [

iatoolkit/base_company.py

@@ -5,11 +5,11 @@
 
 # companies/base_company.py
 from abc import ABC, abstractmethod
-from typing import Any
-from repositories.profile_repo import ProfileRepo
-from repositories.llm_query_repo import LLMQueryRepo
-from services.prompt_manager_service import PromptService
-from repositories.models import Company, Function, PromptCategory
+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
 
 

iatoolkit/cli_commands.py

@@ -6,7 +6,7 @@
 import click
 import logging
 from iatoolkit import IAToolkit
-from services.profile_service import ProfileService
+from iatoolkit.services import ProfileService
 
 def register_core_commands(app):
     """Registra los comandos CLI del núcleo de IAToolkit."""
@@ -33,7 +33,9 @@ def register_core_commands(app):
     @app.cli.command("encrypt-key")
     @click.argument("key")
     def encrypt_llm_api_key(key: str):
-        from common.util import Utility
+        from iatoolkit.common.util import Utility
+
+
 
         util = IAToolkit.get_instance().get_injector().get(Utility)
         try:
@@ -46,7 +48,7 @@ def register_core_commands(app):
     @app.cli.command("exec-tasks")
     @click.argument("company_short_name")
     def exec_pending_tasks(company_short_name: str):
-        from services.tasks_service import TaskService
+        from iatoolkit.services.tasks_service import TaskService
         task_service = IAToolkit.get_instance().get_injector().get(TaskService)
 
         try:

{common → iatoolkit/common}/auth.py

@@ -4,11 +4,11 @@
 # IAToolkit is open source software.
 
 from flask import redirect, url_for
-from common.session_manager import SessionManager
+from iatoolkit.common.session_manager import SessionManager
 from datetime import datetime, timezone
 from injector import inject
-from repositories.profile_repo import ProfileRepo
-from services.jwt_service import JWTService
+from iatoolkit.repositories.profile_repo import ProfileRepo
+from iatoolkit.services.jwt_service import JWTService
 import logging
 from flask import request
 from typing import Optional

{common → iatoolkit/common}/routes.py

@@ -4,9 +4,9 @@
 # IAToolkit is open source software.
 
 from flask import render_template, redirect, flash, url_for,send_from_directory, current_app
-from common.session_manager import SessionManager
+from iatoolkit.common.session_manager import SessionManager
 from flask import jsonify
-from views.history_view import HistoryView
+from iatoolkit.views.history_view import HistoryView
 import os
 
 
@@ -22,23 +22,23 @@ def logout(company_short_name: str):
 # this function register all the views
 def register_views(injector, app):
 
-    from views.llmquery_view import LLMQueryView
-    from views.tasks_view import TaskView
-    from views.tasks_review_view import TaskReviewView
-    from views.home_view import HomeView
-    from views.chat_view import ChatView
-    from views.login_view import LoginView
-    from views.external_chat_login_view import ExternalChatLoginView
-    from views.signup_view import SignupView
-    from views.verify_user_view import VerifyAccountView
-    from views.forgot_password_view import ForgotPasswordView
-    from views.change_password_view import ChangePasswordView
-    from views.file_store_view import FileStoreView
-    from views.user_feedback_view import UserFeedbackView
-    from views.prompt_view import PromptView
-    from views.chat_token_request_view import ChatTokenRequestView
-    from views.external_login_view import ExternalLoginView
-    from views.download_file_view import DownloadFileView
+    from iatoolkit.views.llmquery_view import LLMQueryView
+    from iatoolkit.views.tasks_view import TaskView
+    from iatoolkit.views.tasks_review_view import TaskReviewView
+    from iatoolkit.views.home_view import HomeView
+    from iatoolkit.views.chat_view import ChatView
+    from iatoolkit.views.login_view import LoginView
+    from iatoolkit.views.external_chat_login_view import ExternalChatLoginView
+    from iatoolkit.views.signup_view import SignupView
+    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_view import FileStoreView
+    from iatoolkit.views.user_feedback_view import UserFeedbackView
+    from iatoolkit.views.prompt_view import PromptView
+    from iatoolkit.views.chat_token_request_view import ChatTokenRequestView
+    from iatoolkit.views.external_login_view import ExternalLoginView
+    from iatoolkit.views.download_file_view import DownloadFileView
 
     app.add_url_rule('/', view_func=HomeView.as_view('home'))
 

{common → iatoolkit/common}/util.py

@@ -5,11 +5,11 @@
 
 import logging
 from typing import List
-from common.exceptions import IAToolkitException
+from iatoolkit.common.exceptions import IAToolkitException
 from injector import inject
 import os
 from jinja2 import Environment, FileSystemLoader
-from common.session_manager import SessionManager
+from iatoolkit.common.session_manager import SessionManager
 from datetime import datetime, date
 from decimal import Decimal
 import yaml

iatoolkit/iatoolkit.py

@@ -8,13 +8,14 @@ from flask_session import Session
 from flask_injector import FlaskInjector
 from flask_bcrypt import Bcrypt
 from flask_cors import CORS
-from common.exceptions import IAToolkitException
+from iatoolkit.common.exceptions import IAToolkitException
 from urllib.parse import urlparse
 import redis
 import logging
 import os
 from typing import Optional, Dict, Any
-from repositories.database_manager import DatabaseManager
+from iatoolkit.repositories.database_manager import DatabaseManager
+
 from injector import Binder, singleton, Injector
 from importlib.metadata import version as _pkg_version, PackageNotFoundError
 
@@ -122,7 +123,7 @@ class IAToolkit:
 
     def _register_routes(self):
         """Registers routes by passing the configured injector."""
-        from common.routes import register_views
+        from iatoolkit.common.routes import register_views
 
         # Pass the injector to the view registration function
         register_views(self._injector, self.app)
@@ -249,11 +250,12 @@ class IAToolkit:
             )
 
     def _bind_repositories(self, binder: Binder):
-        from repositories.document_repo import DocumentRepo
-        from repositories.profile_repo import ProfileRepo
-        from repositories.llm_query_repo import LLMQueryRepo
-        from repositories.vs_repo import VSRepo
-        from repositories.tasks_repo import TaskRepo
+        from iatoolkit.repositories.document_repo import DocumentRepo
+        from iatoolkit.repositories.profile_repo import ProfileRepo
+        from iatoolkit.repositories.llm_query_repo import LLMQueryRepo
+
+        from iatoolkit.repositories.vs_repo import VSRepo
+        from iatoolkit.repositories.tasks_repo import TaskRepo
 
         binder.bind(DocumentRepo, to=DocumentRepo)
         binder.bind(ProfileRepo, to=ProfileRepo)
@@ -262,17 +264,17 @@ class IAToolkit:
         binder.bind(TaskRepo, to=TaskRepo)
 
     def _bind_services(self, binder: Binder):
-        from services.query_service import QueryService
-        from services.tasks_service import TaskService
-        from services.benchmark_service import BenchmarkService
-        from services.document_service import DocumentService
-        from services.prompt_manager_service import PromptService
-        from services.excel_service import ExcelService
-        from services.mail_service import MailService
-        from services.load_documents_service import LoadDocumentsService
-        from services.profile_service import ProfileService
-        from services.jwt_service import JWTService
-        from services.dispatcher_service import Dispatcher
+        from iatoolkit.services import QueryService
+        from iatoolkit.services.tasks_service import TaskService
+        from iatoolkit.services import BenchmarkService
+        from iatoolkit.services.document_service import DocumentService
+        from iatoolkit.services.prompt_manager_service import PromptService
+        from iatoolkit.services.excel_service import ExcelService
+        from iatoolkit.services import MailService
+        from iatoolkit.services.load_documents_service import LoadDocumentsService
+        from iatoolkit.services import ProfileService
+        from iatoolkit.services.jwt_service import JWTService
+        from iatoolkit.services.dispatcher_service import Dispatcher
 
         binder.bind(QueryService, to=QueryService)
         binder.bind(TaskService, to=TaskService)
@@ -287,12 +289,14 @@ class IAToolkit:
         binder.bind(Dispatcher, to=Dispatcher)
 
     def _bind_infrastructure(self, binder: Binder):
-        from infra.llm_client import llmClient
-        from infra.llm_proxy import LLMProxy
-        from infra.google_chat_app import GoogleChatApp
-        from infra.mail_app import MailApp
-        from common.auth import IAuthentication
-        from common.util import Utility
+        from iatoolkit.infra.llm_client import llmClient
+        from iatoolkit.infra import LLMProxy
+        from iatoolkit.infra import GoogleChatApp
+        from iatoolkit.infra.mail_app import MailApp
+        from iatoolkit.common import IAuthentication
+        from iatoolkit.common.util import Utility
+
+
 
         binder.bind(LLMProxy, to=LLMProxy, scope=singleton)
         binder.bind(llmClient, to=llmClient, scope=singleton)
@@ -303,10 +307,10 @@ class IAToolkit:
 
     def _bind_views(self, binder: Binder):
         """Vincula las vistas después de que el injector ha sido creado"""
-        from views.llmquery_view import LLMQueryView
-        from views.home_view import HomeView
-        from views.chat_view import ChatView
-        from views.change_password_view import ChangePasswordView
+        from iatoolkit.views import LLMQueryView
+        from iatoolkit.views import HomeView
+        from iatoolkit.views import ChatView
+        from iatoolkit.views import ChangePasswordView
 
         binder.bind(HomeView, to=HomeView)
         binder.bind(ChatView, to=ChatView)
@@ -320,7 +324,7 @@ class IAToolkit:
 
     def _init_dispatcher_and_company_instances(self):
         from iatoolkit.company_registry import get_company_registry
-        from services.dispatcher_service import Dispatcher
+        from iatoolkit.services.dispatcher_service import Dispatcher
 
         # instantiate all the registered companies
         get_company_registry().instantiate_companies(self._injector)
@@ -351,7 +355,7 @@ class IAToolkit:
         # Configura context processors para templates
         @self.app.context_processor
         def inject_globals():
-            from common.session_manager import SessionManager
+            from iatoolkit.common import SessionManager
             return {
                 'url_for': url_for,
                 'iatoolkit_version': self.version,
@@ -386,7 +390,7 @@ class IAToolkit:
         return self._injector
 
     def get_dispatcher(self):
-        from services.dispatcher_service import Dispatcher
+        from iatoolkit.services.dispatcher_service import Dispatcher
         if not self._injector:
             raise IAToolkitException(
                 IAToolkitException.ErrorType.CONFIG_ERROR,
@@ -406,14 +410,9 @@ class IAToolkit:
 def current_iatoolkit() -> IAToolkit:
     return IAToolkit.get_instance()
 
-# 🚀 Función de conveniencia para inicialización rápida
+# Función de conveniencia para inicialización rápida
 def create_app(config: Optional[Dict[str, Any]] = None) -> Flask:
     toolkit = IAToolkit(config)
     toolkit.create_iatoolkit()
 
     return toolkit.app
-
-if __name__ == "__main__":
-    app = create_app()
-    if app:
-        app.run(debug=True)

{infra → iatoolkit/infra}/call_service.py

@@ -3,7 +3,7 @@
 #
 # IAToolkit is open source software.
 
-from common.exceptions import IAToolkitException
+from iatoolkit.common.exceptions import IAToolkitException
 from injector import inject
 # call_service.py
 import requests

{infra → iatoolkit/infra}/connectors/file_connector_factory.py

@@ -3,11 +3,11 @@
 #
 # IAToolkit is open source software.
 
-from infra.connectors.file_connector import FileConnector
-from infra.connectors.local_file_connector import LocalFileConnector
-from infra.connectors.s3_connector import S3Connector
-from infra.connectors.google_drive_connector import GoogleDriveConnector
-from infra.connectors.google_cloud_storage_connector import GoogleCloudStorageConnector
+from iatoolkit.infra.connectors.file_connector import FileConnector
+from iatoolkit.infra.connectors.local_file_connector import LocalFileConnector
+from iatoolkit.infra.connectors.s3_connector import S3Connector
+from iatoolkit.infra.connectors.google_drive_connector import GoogleDriveConnector
+from iatoolkit.infra.connectors.google_cloud_storage_connector import GoogleCloudStorageConnector
 from typing import Dict
 import os
 

{infra → iatoolkit/infra}/connectors/google_cloud_storage_connector.py

@@ -3,7 +3,7 @@
 #
 # IAToolkit is open source software.
 
-from infra.connectors.file_connector import FileConnector
+from iatoolkit.infra.connectors.file_connector import FileConnector
 from google.cloud import storage
 from typing import List
 

{infra → iatoolkit/infra}/connectors/google_drive_connector.py

@@ -3,7 +3,7 @@
 #
 # IAToolkit is open source software.
 
-from infra.connectors.file_connector import FileConnector
+from iatoolkit.infra.connectors.file_connector import FileConnector
 from googleapiclient.discovery import build
 from googleapiclient.http import MediaIoBaseDownload
 from google.oauth2.service_account import Credentials

{infra → iatoolkit/infra}/connectors/local_file_connector.py

@@ -4,9 +4,9 @@
 # IAToolkit is open source software.
 
 import os
-from infra.connectors.file_connector import FileConnector
+from iatoolkit.infra.connectors.file_connector import FileConnector
 from typing import List
-from common.exceptions import IAToolkitException
+from iatoolkit.common.exceptions import IAToolkitException
 
 
 class LocalFileConnector(FileConnector):

{infra → iatoolkit/infra}/connectors/s3_connector.py

@@ -4,7 +4,7 @@
 # IAToolkit is open source software.
 
 import boto3
-from infra.connectors.file_connector import FileConnector
+from iatoolkit.infra.connectors.file_connector import FileConnector
 from typing import List
 
 

{infra → iatoolkit/infra}/gemini_adapter.py

@@ -3,11 +3,11 @@
 #
 # IAToolkit is open source software.
 
-from infra.llm_response import LLMResponse, ToolCall, Usage
+from iatoolkit.infra.llm_response import LLMResponse, ToolCall, Usage
 from typing import Dict, List, Optional
 from google.generativeai.types import HarmCategory, HarmBlockThreshold
 from google.protobuf.json_format import MessageToDict
-from common.exceptions import IAToolkitException
+from iatoolkit.common.exceptions import IAToolkitException
 import logging
 import json
 import uuid

{infra → iatoolkit/infra}/google_chat_app.py

@@ -4,7 +4,7 @@
 # IAToolkit is open source software.
 
 from injector import inject
-from infra.call_service import CallServiceClient
+from iatoolkit.infra.call_service import CallServiceClient
 import logging
 import os
 from typing import Dict, Any

{infra → iatoolkit/infra}/llm_client.py

@@ -3,23 +3,23 @@
 #
 # IAToolkit is open source software.
 
-from infra.llm_proxy import LLMProxy
-from repositories.models import Company, LLMQuery
-from repositories.llm_query_repo import LLMQueryRepo
+from iatoolkit.infra.llm_proxy import LLMProxy
+from iatoolkit.repositories.models import Company, LLMQuery
+from iatoolkit.repositories.llm_query_repo import LLMQueryRepo
 from sqlalchemy.exc import SQLAlchemyError, OperationalError
-from common.util import Utility
+from iatoolkit.common.util import Utility
 from injector import inject
 import time
 import markdown2
 import os
 import logging
 import json
-from common.exceptions import IAToolkitException
+from iatoolkit.common.exceptions import IAToolkitException
 import threading
 import re
 import tiktoken
 from typing import Dict, Optional, List
-from services.dispatcher_service import Dispatcher
+from iatoolkit.services.dispatcher_service import Dispatcher
 
 
 class llmClient:
@@ -55,7 +55,7 @@ class llmClient:
         if self._dispatcher is None:
             # Import what you need, right when you need it.
             from iatoolkit import current_iatoolkit
-            from services.dispatcher_service import Dispatcher
+            from iatoolkit.services.dispatcher_service import Dispatcher
             # Use the global context proxy to get the injector, then get the service
             self._dispatcher = current_iatoolkit().get_injector().get(Dispatcher)
         return self._dispatcher

{infra → iatoolkit/infra}/llm_proxy.py

@@ -5,12 +5,12 @@
 
 from typing import Dict, List, Any
 from abc import ABC, abstractmethod
-from common.util import Utility
-from infra.llm_response import LLMResponse
-from infra.openai_adapter import OpenAIAdapter
-from infra.gemini_adapter import GeminiAdapter
-from common.exceptions import IAToolkitException
-from repositories.models import Company
+from iatoolkit.common.util import Utility
+from iatoolkit.infra.llm_response import LLMResponse
+from iatoolkit.infra.openai_adapter import OpenAIAdapter
+from iatoolkit.infra.gemini_adapter import GeminiAdapter
+from iatoolkit.common.exceptions import IAToolkitException
+from iatoolkit.repositories.models import Company
 from openai import OpenAI
 import google.generativeai as genai
 import os

{infra → iatoolkit/infra}/mail_app.py

@@ -5,7 +5,7 @@
 
 import sib_api_v3_sdk
 from sib_api_v3_sdk.rest import ApiException
-from common.exceptions import IAToolkitException
+from iatoolkit.common.exceptions import IAToolkitException
 import os
 import base64
 import logging

{infra → iatoolkit/infra}/openai_adapter.py

@@ -5,8 +5,8 @@
 
 import logging
 from typing import Dict, List, Optional
-from infra.llm_response import LLMResponse, ToolCall, Usage
-from common.exceptions import IAToolkitException
+from iatoolkit.infra.llm_response import LLMResponse, ToolCall, Usage
+from iatoolkit.common.exceptions import IAToolkitException
 
 
 class OpenAIAdapter:

{repositories → iatoolkit/repositories}/database_manager.py

@@ -7,7 +7,7 @@
 from sqlalchemy import create_engine, event, inspect
 from sqlalchemy.orm import sessionmaker, scoped_session
 from sqlalchemy.engine.url import make_url
-from repositories.models import Base
+from iatoolkit.repositories.models import Base
 from injector import inject
 from pgvector.psycopg2 import register_vector
 

{repositories → iatoolkit/repositories}/document_repo.py

@@ -3,10 +3,10 @@
 #
 # IAToolkit is open source software.
 
-from repositories.models import Document
+from iatoolkit.repositories.models import Document
 from injector import inject
-from repositories.database_manager import DatabaseManager
-from common.exceptions import IAToolkitException
+from iatoolkit.repositories.database_manager import DatabaseManager
+from iatoolkit.common.exceptions import IAToolkitException
 
 
 class DocumentRepo:

{repositories → iatoolkit/repositories}/llm_query_repo.py

@@ -3,9 +3,9 @@
 #
 # IAToolkit is open source software.
 
-from repositories.models import LLMQuery, Function, Company, Prompt, PromptCategory
+from iatoolkit.repositories.models import LLMQuery, Function, Company, Prompt, PromptCategory
 from injector import inject
-from repositories.database_manager import DatabaseManager
+from iatoolkit.repositories.database_manager import DatabaseManager
 from sqlalchemy import or_
 
 class LLMQueryRepo:

{repositories → iatoolkit/repositories}/profile_repo.py

@@ -3,9 +3,9 @@
 #
 # IAToolkit is open source software.
 
-from repositories.models import User, Company, ApiKey, UserFeedback
+from iatoolkit.repositories.models import User, Company, ApiKey, UserFeedback
 from injector import inject
-from repositories.database_manager import DatabaseManager
+from iatoolkit.repositories.database_manager import DatabaseManager
 from sqlalchemy.orm import joinedload # Para cargar la relación eficientemente
 
 

{repositories → iatoolkit/repositories}/tasks_repo.py

@@ -5,8 +5,8 @@
 
 from injector import inject
 from datetime import datetime
-from repositories.models import Task, TaskStatus, TaskType
-from repositories.database_manager import DatabaseManager
+from iatoolkit.repositories.models import Task, TaskStatus, TaskType
+from iatoolkit.repositories.database_manager import DatabaseManager
 from sqlalchemy import or_
 
 

{repositories → iatoolkit/repositories}/vs_repo.py

@@ -6,9 +6,9 @@
 from sqlalchemy import  text
 from huggingface_hub import InferenceClient
 from injector import inject
-from common.exceptions import IAToolkitException
-from repositories.database_manager import DatabaseManager
-from repositories.models import Document, VSDoc
+from iatoolkit.common.exceptions import IAToolkitException
+from iatoolkit.repositories.database_manager import DatabaseManager
+from iatoolkit.repositories.models import Document, VSDoc
 import os
 import logging
 

{services → iatoolkit/services}/benchmark_service.py

@@ -7,9 +7,9 @@ import pandas as pd
 import time
 import logging
 from injector import inject
-from services.query_service import QueryService
-from repositories.profile_repo import ProfileRepo
-from common.exceptions import IAToolkitException
+from iatoolkit.services.query_service import QueryService
+from iatoolkit.repositories.profile_repo import ProfileRepo
+from iatoolkit.common.exceptions import IAToolkitException
 
 
 class BenchmarkService:

{services → iatoolkit/services}/dispatcher_service.py

@@ -3,14 +3,15 @@
 #
 # IAToolkit is open source software.
 
-from common.exceptions import IAToolkitException
-from services.prompt_manager_service import PromptService
-from repositories.llm_query_repo import LLMQueryRepo
-from repositories.models import Company, Function
-from services.excel_service import ExcelService
-from services.mail_service import MailService
-from common.session_manager import SessionManager
-from common.util import Utility
+from iatoolkit.common.exceptions import IAToolkitException
+from iatoolkit.services.prompt_manager_service import PromptService
+from iatoolkit.repositories.llm_query_repo import LLMQueryRepo
+
+from iatoolkit.repositories.models import Company, Function
+from iatoolkit.services.excel_service import ExcelService
+from iatoolkit.services.mail_service import MailService
+from iatoolkit.common.session_manager import SessionManager
+from iatoolkit.common.util import Utility
 from injector import inject
 import logging
 import os

{services → iatoolkit/services}/document_service.py

@@ -10,7 +10,7 @@ import io
 import os
 import pytesseract
 from injector import inject
-from common.exceptions import IAToolkitException
+from iatoolkit.common.exceptions import IAToolkitException
 
 class DocumentService:
     @inject

{services → iatoolkit/services}/excel_service.py

@@ -3,11 +3,11 @@
 #
 # IAToolkit is open source software.
 
-from common.util import Utility
+from iatoolkit.common.util import Utility
 import pandas as pd
 from uuid import uuid4
 from pathlib import Path
-from common.exceptions import IAToolkitException
+from iatoolkit.common.exceptions import IAToolkitException
 from injector import inject
 import os
 import logging