iatoolkit 0.66.2__py3-none-any.whl → 0.71.2__py3-none-any.whl

iatoolkit/views/embedding_api_view.py

@@ -0,0 +1,65 @@
+# iatoolkit/views/embedding_api_view.py
+# 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.embedding_service import EmbeddingService
+from iatoolkit.services.auth_service import AuthService
+from injector import inject
+import logging
+
+class EmbeddingApiView(MethodView):
+    """
+    Handles API requests to generate an embedding for a given text.
+    Authentication is based on the active Flask session.
+    """
+
+    @inject
+    def __init__(self,
+                 auth_service: AuthService,
+                 embedding_service: EmbeddingService):
+        self.auth_service = auth_service
+        self.embedding_service = embedding_service
+
+    def post(self, company_short_name: str):
+        """
+        Generates an embedding for the text provided in the request body.
+        Expects a JSON payload with a "text" key.
+        """
+        try:
+            # 1. Authenticate the user from the current session
+            auth_result = self.auth_service.verify(anonymous=True)
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get("status_code", 401)
+
+            # 2. Validate incoming request data
+            if not request.is_json:
+                return jsonify({"error": "Request must be JSON"}), 400
+
+            data = request.get_json()
+            text = data.get('text')
+
+            if not text:
+                return jsonify({"error": "The 'text' key is required."}), 400
+
+            # 3. Call the embedding service, now passing the company_short_name
+            embedding_b64 = self.embedding_service.embed_text(
+                company_short_name=company_short_name,
+                text=text,
+                to_base64=True
+            )
+
+            model_name = self.embedding_service.get_model_name(company_short_name)
+            response = {
+                "embedding": embedding_b64,
+                "model": model_name
+            }
+            return jsonify(response), 200
+
+        except Exception as e:
+            logging.exception(f"Unexpected error in EmbeddingApiView: {e}")
+            # Return a generic error message to the client
+            return jsonify({"error": "An internal error occurred while generating the embedding."}), 500

iatoolkit/views/external_login_view.py

@@ -47,7 +47,7 @@ class ExternalLoginView(BaseLoginView):
 
         # 5. Delegate the path decision to the centralized logic.
         try:
-            return self._handle_login_path(company, user_identifier, target_url, redeem_token)
+            return self._handle_login_path(company_short_name, user_identifier, target_url, redeem_token)
         except Exception as e:
             logging.exception(f"Error processing external login path for {company_short_name}/{user_identifier}: {e}")
             return jsonify({"error": f"Internal server error while starting chat. {str(e)}"}), 500

iatoolkit/views/file_store_api_view.py

@@ -48,7 +48,7 @@ class FileStoreApiView(MethodView):
             # get the file content from base64
             content = base64.b64decode(base64_content)
 
-            new_document = self.doc_service.load_file_callback(
+            new_document = self.doc_service._file_processing_callback(
                 filename=filename,
                 content=content,
                 company=company,

iatoolkit/views/forgot_password_view.py

@@ -30,9 +30,8 @@ class ForgotPasswordView(MethodView):
             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)
+        branding_data = self.branding_service.get_company_branding(company_short_name)
         return render_template('forgot_password.html',
-                               company=company,
                                company_short_name=company_short_name,
                                branding=branding_data
                                )
@@ -45,7 +44,7 @@ class ForgotPasswordView(MethodView):
                 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)
+            branding_data = self.branding_service.get_company_branding(company_short_name)
             email = request.form.get('email')
 
             # create a safe token and url for it
@@ -59,7 +58,6 @@ class ForgotPasswordView(MethodView):
                 flash(response["error"], 'error')
                 return render_template(
                     'forgot_password.html',
-                    company=company,
                     company_short_name=company_short_name,
                     branding=branding_data,
                     form_data={"email": email}), 400

iatoolkit/views/help_content_api_view.py

@@ -5,7 +5,7 @@
 
 from flask import request, jsonify
 from flask.views import MethodView
-from iatoolkit.services.help_content_service import HelpContentService
+from iatoolkit.services.configuration_service import ConfigurationService
 from iatoolkit.services.i18n_service import I18nService
 from iatoolkit.services.auth_service import AuthService
 from injector import inject
@@ -21,10 +21,10 @@ class HelpContentApiView(MethodView):
     @inject
     def __init__(self,
                  auth_service: AuthService,
-                 help_content_service: HelpContentService,
+                 config_service: ConfigurationService,
                  i18n_service: I18nService):
         self.auth_service = auth_service
-        self.help_content_service = help_content_service
+        self.config_service = config_service
         self.i18n_service = i18n_service
 
     def post(self, company_short_name: str):
@@ -36,10 +36,10 @@ class HelpContentApiView(MethodView):
 
             user_identifier = auth_result.get('user_identifier')
 
-            # 2. Call the history service with the unified identifier.
-            # The service's signature should now only expect user_identifier.
-            response = self.help_content_service.get_content(
-                company_short_name=company_short_name
+            # 2. Call the config service with the unified identifier.
+            response = self.config_service.get_configuration(
+                company_short_name=company_short_name,
+                content_key='help_content'      # specific key for this service
             )
 
             if "error" in response:
@@ -50,5 +50,5 @@ class HelpContentApiView(MethodView):
 
         except Exception as e:
             logging.exception(
-                f"Unexpected error fetching help_content for {company_short_name}/{user_identifier}: {e}")
-            return jsonify({"error_message": self.i18n_service.t('errors.general.unexpected_error')}), 500
+                f"Unexpected error fetching help_content for {company_short_name}: {e}")
+            return jsonify({"error_message": self.i18n_service.t('errors.general.unexpected_error', error=str(e))}), 500

iatoolkit/views/history_api_view.py

@@ -53,4 +53,4 @@ class HistoryApiView(MethodView):
         except Exception as e:
             logging.exception(
                 f"Unexpected error: {e}")
-            return jsonify({"error_message": self.i18n_service.t('errors.general.unexpected_error')}), 500
+            return jsonify({"error_message": self.i18n_service.t('errors.general.unexpected_error', error=str(e))}), 500

iatoolkit/views/home_view.py

@@ -25,13 +25,14 @@ class HomeView(MethodView):
         self.util = utility
 
     def get(self, company_short_name: str):
+        branding_data = {}
         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)
+            branding_data = self.branding_service.get_company_branding(company_short_name)
             home_template = self.util.get_company_template(company_short_name, "home.html")
 
             # 2. Verificamos si el archivo de plantilla personalizado no existe.
@@ -47,7 +48,6 @@ class HomeView(MethodView):
             # 3. Si el archivo existe, intentamos leerlo y renderizarlo.
             return render_template_string(
                 home_template,
-                company=company,
                 company_short_name=company_short_name,
                 branding=branding_data,
             )

iatoolkit/views/init_context_api_view.py

@@ -4,7 +4,7 @@ from iatoolkit.services.query_service import QueryService
 from iatoolkit.services.profile_service import ProfileService
 from iatoolkit.services.auth_service import AuthService
 from iatoolkit.services.i18n_service import I18nService
-from flask import jsonify
+from flask import jsonify, request
 import logging
 
 
@@ -39,29 +39,30 @@ class InitContextApiView(MethodView):
 
             user_identifier = auth_result.get('user_identifier')
 
-            # 2. Execute the forced rebuild sequence using the unified identifier.
-            self.query_service.session_context.clear_all_context(company_short_name, user_identifier)
-            logging.info(f"Context for {company_short_name}/{user_identifier} has been cleared.")
+            # check if model was sent as a parameter
+            data = request.get_json(silent=True) or {}
+            model = data.get('model', '')
 
-            # LLM context is clean, now we can load it again
-            self.query_service.prepare_context(
+            # reinit the LLM context
+            response = self.query_service.init_context(
                 company_short_name=company_short_name,
-                user_identifier=user_identifier
-            )
+                user_identifier=user_identifier,
+                model=model)
 
-            self.query_service.finalize_context_rebuild(
-                company_short_name=company_short_name,
-                user_identifier=user_identifier
-            )
-
-            # 3. Respond with JSON, as this is an API endpoint.
+            # Respond with JSON, as this is an API endpoint.
             success_message = self.i18n_service.t('api_responses.context_reloaded_success')
-            return jsonify({'status': 'OK', 'message': success_message}), 200
+            response_message = {'status': 'OK', 'message': success_message}
+
+            # if received a response ID with the context, return it
+            if response.get('response_id'):
+                response_message['response_id'] = response['response_id']
+
+            return jsonify(response_message), 200
 
         except Exception as e:
             logging.exception(f"errors while reloading context: {e}")
-            error_message = self.i18n_service.t('errors.general.unexpected_error')
-            return jsonify({"error_message": error_message}), 500
+            error_message = self.i18n_service.t('errors.general.unexpected_error', error=str(e))
+            return jsonify({"error_message": error_message}), 406
 
     def options(self, company_short_name):
         """

iatoolkit/views/llmquery_api_view.py

@@ -44,14 +44,15 @@ class LLMQueryApiView(MethodView):
                 question=data.get('question', ''),
                 prompt_name=data.get('prompt_name'),
                 client_data=data.get('client_data', {}),
+                response_id = data.get('response_id'),
                 files=data.get('files', [])
             )
             if 'error' in result:
-                return jsonify(result), 400
+                return jsonify(result), 407
 
             return jsonify(result), 200
 
         except Exception as e:
             logging.exception(
                 f"Unexpected error: {e}")
-            return jsonify({"error_message": self.i18n_service.t('errors.general.unexpected_error')}), 500
+            return jsonify({"error_message": self.i18n_service.t('errors.general.unexpected_error', error=str(e))}), 500

iatoolkit/views/login_simulation_view.py

@@ -10,17 +10,29 @@ from flask.views import MethodView
 from flask import render_template, request, Response
 from injector import inject
 from iatoolkit.services.profile_service import ProfileService
+from iatoolkit.services.branding_service import BrandingService
 
 
 class LoginSimulationView(MethodView):
     @inject
     def __init__(self,
-                 profile_service: ProfileService):
+                 profile_service: ProfileService,
+                 branding_service: BrandingService):
         self.profile_service = profile_service
+        self.branding_service = branding_service
+
 
     def get(self, company_short_name: str = None):
-        """Muestra el formulario para iniciar la simulación."""
+        company = self.profile_service.get_company_by_short_name(company_short_name)
+        if not company:
+            return render_template('error.html',
+                                   company_short_name=company_short_name,
+                                   message="Empresa no encontrada"), 404
+
+        branding_data = self.branding_service.get_company_branding(company_short_name)
+
         return render_template('login_simulation.html',
+                               branding=branding_data,
                                company_short_name=company_short_name
                                )
 

iatoolkit/views/login_view.py

@@ -12,7 +12,7 @@ from iatoolkit.services.jwt_service import JWTService
 from iatoolkit.services.query_service import QueryService
 from iatoolkit.services.prompt_manager_service import PromptService
 from iatoolkit.services.branding_service import BrandingService
-from iatoolkit.services.onboarding_service import OnboardingService
+from iatoolkit.services.configuration_service import ConfigurationService
 from iatoolkit.services.i18n_service import I18nService
 from iatoolkit.views.base_login_view import BaseLoginView
 import logging
@@ -29,7 +29,7 @@ class LoginView(BaseLoginView):
             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)
+        branding_data = self.branding_service.get_company_branding(company_short_name)
         email = request.form.get('email')
         password = request.form.get('password')
 
@@ -61,7 +61,7 @@ class LoginView(BaseLoginView):
 
         # 2. Delegate the path decision to the centralized logic.
         try:
-            return self._handle_login_path(company, user_identifier, target_url)
+            return self._handle_login_path(company_short_name, user_identifier, target_url)
         except Exception as e:
             message = self.i18n_service.t('errors.templates.processing_error', error=str(e))
             return render_template(
@@ -83,7 +83,7 @@ class FinalizeContextView(MethodView):
                  query_service: QueryService,
                  prompt_service: PromptService,
                  branding_service: BrandingService,
-                 onboarding_service: OnboardingService,
+                 config_service: ConfigurationService,
                  jwt_service: JWTService,
                  i18n_service: I18nService
                  ):
@@ -92,7 +92,7 @@ class FinalizeContextView(MethodView):
         self.query_service = query_service
         self.prompt_service = prompt_service
         self.branding_service = branding_service
-        self.onboarding_service = onboarding_service
+        self.config_service = config_service
         self.i18n_service = i18n_service
 
     def get(self, company_short_name: str, token: str = None):
@@ -111,7 +111,7 @@ class FinalizeContextView(MethodView):
 
                 user_identifier = payload.get('user_identifier')
             else:
-                logging.warning("Fallo crítico: missing session information or auth token")
+                logging.error("missing session information or auth token")
                 return redirect(url_for('home', company_short_name=company_short_name))
 
             company = self.profile_service.get_company_by_short_name(company_short_name)
@@ -119,17 +119,17 @@ class FinalizeContextView(MethodView):
                 return render_template('error.html',
                             company_short_name=company_short_name,
                             message="Empresa no encontrada"), 404
-            branding_data = self.branding_service.get_company_branding(company)
+            branding_data = self.branding_service.get_company_branding(company_short_name)
 
             # 2. Finalize the context rebuild (the heavy task).
-            self.query_service.finalize_context_rebuild(
+            self.query_service.set_context_for_llm(
                 company_short_name=company_short_name,
                 user_identifier=user_identifier
             )
 
             # 3. render the chat page.
             prompts = self.prompt_service.get_user_prompts(company_short_name)
-            onboarding_cards = self.onboarding_service.get_onboarding_cards(company)
+            onboarding_cards = self.config_service.get_configuration(company_short_name, 'onboarding_cards')
 
             # Get the entire 'js_messages' block in the correct language.
             js_translations = self.i18n_service.get_translation_block('js_messages')

iatoolkit/views/signup_view.py

@@ -32,9 +32,8 @@ class SignupView(MethodView):
             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)
+        branding_data = self.branding_service.get_company_branding(company_short_name)
         return render_template('signup.html',
-                               company=company,
                                company_short_name=company_short_name,
                                branding=branding_data)
 
@@ -45,7 +44,7 @@ class SignupView(MethodView):
                 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)
+            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')
@@ -70,7 +69,6 @@ class SignupView(MethodView):
                 flash(response["error"], 'error')
                 return render_template(
                     'signup.html',
-                    company=company,
                     company_short_name=company_short_name,
                     branding=branding_data,
                     form_data={

iatoolkit/views/verify_user_view.py

@@ -32,14 +32,13 @@ class VerifyAccountView(MethodView):
                 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)
+            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=company,
                                        company_short_name=company_short_name,
                                        branding=branding_data,
                                        token=token), 400
@@ -49,7 +48,6 @@ class VerifyAccountView(MethodView):
                 flash(response["error"], 'error')
                 return render_template(
                     'signup.html',
-                    company=company,
                     company_short_name=company_short_name,
                     branding=branding_data,
                     token=token), 400
@@ -58,5 +56,5 @@ class VerifyAccountView(MethodView):
             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')
+            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.66.2.dist-info → iatoolkit-0.71.2.dist-info}/METADATA

@@ -1,11 +1,12 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 0.66.2
+Version: 0.71.2
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
+License-File: LICENSE
 Requires-Dist: aiohappyeyeballs==2.4.4
 Requires-Dist: aiohttp==3.11.9
 Requires-Dist: aiosignal==1.3.1
@@ -206,47 +207,64 @@ Requires-Dist: wrapt==1.17.0
 Requires-Dist: yarl==1.18.3
 Requires-Dist: zipp==3.21.0
 Requires-Dist: zstandard==0.23.0
+Dynamic: license-file
 
 
 <div align="center">
-  <h1>IAToolkit</h1>
+<h1>
+    IAToolkit
+</h1>
+
   <p><strong>The Open-Source Framework for Building AI Chatbots on Your Private Data.</strong></p>
+<h4>
+  <a href="https://www.iatoolkit.com" target="_blank" style="text-decoration: none; color: inherit;">
+    www.iatoolkit.com
+  </a>
+</h4>
 </div>
 
-IAToolkit is a comprehensive, open-source framework designed for building enterprise-grade 
-AI chatbots and conversational applications. 
+IAToolkit is a comprehensive, Python open-source framework designed for building enterprise-grade 
+AI chatbots and conversational applications. It bridges the gap between the power of 
+Large Language Models (LLMs) and the valuable, 
+private data locked within your organization's databases and documents.
+
 With IAToolkit, you can build production-ready, context-aware chatbots and agents that 
 can query relational databases, perform semantic searches on documents, 
 and connect to your internal APIs in minutes.
 
-IAToolkit bridges the gap between powerful LLMs and your company's data. 
+Create secure, branded chat interfaces that can reason over your data, answer questions, and execute custom business logic, 
+all powered by leading models from OpenAI, Google Gemini, and more.
 
 
 ## 🚀 Key Features
 
-*   **🔗 Unified Data Connection**:
-    *   **Natural Language to SQL**: Let your chatbot query relational databases (PostgreSQL, MySQL, SQLite) using everyday language.
-    *   **Semantic Document Search**: Automatically chunk, embed, and search across your private documents (PDFs, Word, etc.) to provide contextually accurate answers.
+* **🔗 Unified Data Connection**
+    * **Natural Language to SQL**: Let your chatbot query relational databases (PostgreSQL, MySQL, SQLite) using everyday language.
+    * **Semantic Document Search**: Automatically chunk, embed, and search across your private documents (PDFs, Word, etc.) to provide contextually accurate answers.
 
-*   **🏢 Enterprise-Ready Multi-Tenancy**:
-    *   Deploy isolated "Company" modules, each with its own data, tools, and context. Perfect for SaaS products or internal departmental agents.
+* **🏢 Enterprise-Ready Multi-Tenancy**
+    * Deploy isolated "Company" modules, each with its own data, tools, and context. 
+    * Perfect for SaaS products or internal departmental agents.
 
-*   **🧠 LLM Agnostic**:
-    *   Switch between **OpenAI (GPT-*)** and **Google (Gemini-*)** with a single line change in your configuration. No code refactoring needed.
+* **🎨 Fully Brandable UI**
+    * Customize the look and feel for each "Company" with its own logos, colors, and even language settings (i18n).
+    * Provides a white-labeled experience for your users.
 
-*   **🛠️ Developer-First Experience**:
-    *   Built with a clean, **Dependency Injection** architecture.
-    *   High-quality code base with **90%+ test coverage**.
-    *   Powerful Flask-based **CLI** for database setup, API key generation, and more.
+* **🧠 LLM Agnostic**
+    * Switch between **OpenAI (GPT-*)** and **Google (Gemini-*)** with a single line change in your configuration. 
+    * No code refactoring needed.
 
-*   **🔒 Security & Observability Built-In**:
-    *   Comes with JWT-based authentication, user management, and secure session handling out of the box.
-    *   Full traceability with detailed logging of all queries, function calls, token usage, and costs.
+* **🛠️ Developer-First Experience**
+    * Built with a clean **Dependency Injection** architecture.  
+    * High-quality code base with **90%+ test coverage**.
 
-## ⚡ Quick Start: Create a Custom Tool in 30 Seconds
+* **🔒 Security & Observability Built-In**
+    * Comes with integrated user authentication, API keys, and secure session handling out of the box.  
+    * Full traceability with detailed logging of all queries, function calls, token usage, and costs.
+## ⚡ Quick Start: Try our 'hello world' example
 
-See how easy it is to give your AI a new skill. Just define a method inside your Company class and describe it. 
-IAToolkit handles the rest.
+Ready to see it in action? Our Quickstart Guide will walk you through downloading, configuring, and launching your first AI assistant in just a few minutes. 
+It's the best way to experience the toolkit's capabilities firsthand.
 
 ## 🤝 Contributing