iatoolkit 0.3.2__tar.gz → 0.3.4__tar.gz

{iatoolkit-0.3.2 → iatoolkit-0.3.4}/PKG-INFO

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 0.3.2
+Version: 0.3.4
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT
-Requires-Python: >=3.10
+Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 Requires-Dist: aiohappyeyeballs==2.4.4
 Requires-Dist: aiohttp==3.11.9
@@ -206,3 +206,47 @@ Requires-Dist: wrapt==1.17.0
 Requires-Dist: yarl==1.18.3
 Requires-Dist: zipp==3.21.0
 Requires-Dist: zstandard==0.23.0
+
+# iatoolkit
+
+IAToolkit is a comprehensive, open-source framework designed for building enterprise-grade 
+AI chatbots and conversational applications. 
+Built on Flask with dependency injection, it provides a robust foundation for scalable AI solutions.
+
+## 🚀 Key Features
+- **Universal LLM Integration**: OpenAI GPT, Google Gemini
+- **Template System**: Jinja2-powered prompt templates with variables
+- **Context Management**: Maintain conversation context across sessions
+
+### 🔒 **Enterprise Security**
+- **JWT Authentication**: Secure token-based authentication
+- **Session Management**: Redis-backed secure sessions
+- **CORS Configuration**: Flexible cross-origin resource sharing
+
+### 🛠 **Function Calling & Tools**
+- **Native Function Calls**: Direct integration with LLM function calling
+- **Custom Tools**: Build and register custom tools for your chatbot
+- **SQL Query Generation**: Natural language to SQL conversion
+- **API Integrations**: Connect to external services and APIs
+
+### 🗄 **Database & Storage**
+- **Multi-Database Support**: PostgreSQL, MySQL, SQLite via SQLAlchemy
+- **Vector Store Integration**: Semantic search and retrieval
+- **Document Processing**: PDF, Word, Excel, and text file handling
+
+### 📊 **Analytics & Monitoring**
+- **Query Logging**: Track all LLM interactions
+- **Performance Metrics**: Response times, token usage, costs
+- **Benchmarking**: Compare model performance
+- **Task Management**: Async task processing with status tracking
+
+### 🔧 **Developer Experience**
+- **Dependency Injection**: Clean, testable architecture
+- **CLI Tools**: Command-line interface for common tasks
+- **Hot Reloading**: Development-friendly configuration
+- **Comprehensive Logging**: Debug and monitor easily
+
+## License
+MIT License
+
+

{iatoolkit-0.3.2 → iatoolkit-0.3.4}/pyproject.toml

@@ -4,10 +4,10 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "iatoolkit"
-version = "0.3.2"
-requires-python = ">=3.10"
+version = "0.3.4"
+requires-python = ">=3.11"
 description = "IAToolkit"
-readme = "README.md"
+readme = "readme.md"
 license = "MIT"
 authors = [{ name = "Fernando Libedinsky" }]
 

iatoolkit-0.3.4/readme.md

@@ -0,0 +1,43 @@
+# iatoolkit
+
+IAToolkit is a comprehensive, open-source framework designed for building enterprise-grade 
+AI chatbots and conversational applications. 
+Built on Flask with dependency injection, it provides a robust foundation for scalable AI solutions.
+
+## 🚀 Key Features
+- **Universal LLM Integration**: OpenAI GPT, Google Gemini
+- **Template System**: Jinja2-powered prompt templates with variables
+- **Context Management**: Maintain conversation context across sessions
+
+### 🔒 **Enterprise Security**
+- **JWT Authentication**: Secure token-based authentication
+- **Session Management**: Redis-backed secure sessions
+- **CORS Configuration**: Flexible cross-origin resource sharing
+
+### 🛠 **Function Calling & Tools**
+- **Native Function Calls**: Direct integration with LLM function calling
+- **Custom Tools**: Build and register custom tools for your chatbot
+- **SQL Query Generation**: Natural language to SQL conversion
+- **API Integrations**: Connect to external services and APIs
+
+### 🗄 **Database & Storage**
+- **Multi-Database Support**: PostgreSQL, MySQL, SQLite via SQLAlchemy
+- **Vector Store Integration**: Semantic search and retrieval
+- **Document Processing**: PDF, Word, Excel, and text file handling
+
+### 📊 **Analytics & Monitoring**
+- **Query Logging**: Track all LLM interactions
+- **Performance Metrics**: Response times, token usage, costs
+- **Benchmarking**: Compare model performance
+- **Task Management**: Async task processing with status tracking
+
+### 🔧 **Developer Experience**
+- **Dependency Injection**: Clean, testable architecture
+- **CLI Tools**: Command-line interface for common tasks
+- **Hot Reloading**: Development-friendly configuration
+- **Comprehensive Logging**: Debug and monitor easily
+
+## License
+MIT License
+
+

{iatoolkit-0.3.2 → iatoolkit-0.3.4}/src/iatoolkit/__init__.py

@@ -22,7 +22,11 @@ from services.dispatcher_service import Dispatcher
 from services.document_service import DocumentService
 from services.search_service import SearchService
 from repositories.profile_repo import ProfileRepo
+from repositories.llm_query_repo import LLMQueryRepo
 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
 
 
 __all__ = [
@@ -37,5 +41,12 @@ __all__ = [
     'DocumentService',
     'SearchService',
     'ProfileRepo',
+    'LLMQueryRepo',
     'DatabaseManager',
+    'CallServiceClient',
+    'Utility',
+    'Company',
+    'Function',
+    'TaskType',
+    'Base',
 ]

{iatoolkit-0.3.2 → iatoolkit-0.3.4}/src/iatoolkit/company_registry.py

@@ -8,10 +8,8 @@ import logging
 
 class CompanyRegistry:
     """
-    Registro centralizado de empresas para iatoolkit.
-
-    Permite a los clientes registrar sus clases de empresa de forma explícita
-    en lugar de usar autodiscovery.
+    Company registry with dependency injection support.
+    Allow the client to register companies and instantiate them with dependency injection.
     """
 
     def __init__(self):
@@ -19,22 +17,6 @@ class CompanyRegistry:
         self._company_instances: Dict[str, BaseCompany] = {}
         self._injector = None
 
-    def register_company(self, name: str, company_class: Type[BaseCompany]) -> None:
-        """
-        Registra una clase de empresa.
-
-        Args:
-            name: Nombre de la empresa (ej: 'maxxa')
-            company_class: Clase que hereda de BaseCompany
-        """
-        if not issubclass(company_class, BaseCompany):
-            raise ValueError(f"La clase {company_class.__name__} debe heredar de BaseCompany")
-
-        company_key = name.lower()
-        self._company_classes[company_key] = company_class
-
-        logging.info(f"Empresa registrada: {company_key} -> {company_class.__name__}")
-
     def set_injector(self, injector) -> None:
         """Establece el injector para crear instancias con dependencias"""
         self._injector = injector
@@ -65,34 +47,33 @@ class CompanyRegistry:
         return self._company_instances.copy()
 
     def get_registered_companies(self) -> Dict[str, Type[BaseCompany]]:
-        """Retorna las clases registradas"""
         return self._company_classes.copy()
 
-    def get_company_instances(self) -> Dict[str, BaseCompany]:
-        """Retorna las instancias de empresas"""
-        return self._company_instances.copy()
-
     def clear(self) -> None:
         """Limpia el registro (útil para tests)"""
         self._company_classes.clear()
         self._company_instances.clear()
 
 
-# Instancia global del registry
+# global instance of the company registry
 _company_registry = CompanyRegistry()
 
 
 def register_company(name: str, company_class: Type[BaseCompany]) -> None:
     """
-    Función pública para registrar empresas.
+    Public function to register a company.
 
     Args:
-        name: Nombre de la empresa
-        company_class: Clase que hereda de BaseCompany
+        name: Name of the company
+        company_class: Class that inherits from BaseCompany
     """
-    _company_registry.register_company(name, company_class)
+    if not issubclass(company_class, BaseCompany):
+        raise ValueError(f"La clase {company_class.__name__} debe heredar de BaseCompany")
+
+    company_key = name.lower()
+    _company_registry._company_classes[company_key] = company_class
 
 
 def get_company_registry() -> CompanyRegistry:
-    """Obtiene el registry global"""
+    """get the global company registry instance"""
     return _company_registry

{iatoolkit-0.3.2 → iatoolkit-0.3.4}/src/iatoolkit/iatoolkit.py

@@ -19,7 +19,6 @@ import click
 from typing import Optional, Dict, Any
 from repositories.database_manager import DatabaseManager
 from injector import Binder, singleton, Injector
-from .toolkit_config import IAToolkitConfig
 
 VERSION = "2.0.0"
 
@@ -51,9 +50,9 @@ class IAToolkit:
             return
 
         self.config = config or {}
-        self.app: Optional[Flask] = None
-        self.db_manager: Optional[DatabaseManager] = None
-        self._injector: Optional[Injector] = None
+        self.app = None
+        self.db_manager = None
+        self._injector = None
 
     @classmethod
     def get_instance(cls) -> 'IAToolkit':
@@ -68,7 +67,7 @@ class IAToolkit:
     def create_iatoolkit(self):
         """
             Creates, configures, and returns the Flask application instance.
-            his is the main entry point for the application factory.
+            this is the main entry point for the application factory.
         """
         self._setup_logging()
 
@@ -78,12 +77,8 @@ class IAToolkit:
         # Step 2: Set up the core components that DI depends on
         self._setup_database()
 
-        # Step 3: Create the Injector with CORE dependencies (NO VIEWS)
-        toolkit_config_module = IAToolkitConfig(app=self.app, db_manager=self.db_manager)
-        self._injector = Injector([
-            toolkit_config_module,
-            self._configure_core_dependencies # This method binds services, repos, etc.
-        ])
+        # Step 3: Create the Injector and configure all dependencies in one place
+        self._injector = Injector(self._configure_core_dependencies)
 
         # Step 4: Register routes using the fully configured injector
         self._register_routes()
@@ -102,7 +97,6 @@ class IAToolkit:
         logging.info(f"🎉 IAToolkit v{VERSION} inicializado correctamente")
         return self.app
 
-
     def _get_config_value(self, key: str, default=None):
         """Obtiene un valor de configuración, primero del dict config, luego de env vars"""
         return self.config.get(key, os.getenv(key, default))
@@ -228,7 +222,8 @@ class IAToolkit:
         """⚙️ Configures all system dependencies."""
         try:
             # Core dependencies
-            binder.bind(Injector, to=self._injector, scope=singleton)
+            binder.bind(Flask, to=self.app, scope=singleton)
+            binder.bind(DatabaseManager, to=self.db_manager, scope=singleton)
 
             # Bind all application components by calling the specific methods
             self._bind_repositories(binder)
@@ -317,16 +312,17 @@ class IAToolkit:
 
     def _setup_cli_commands(self):
         """⌨️ Configura comandos CLI básicos"""
+        from services.dispatcher_service import Dispatcher
+        from services.profile_service import ProfileService
 
         @self.app.cli.command("init-db")
         def init_db():
             """🗄️ Inicializa la base de datos del sistema"""
             try:
-                from services.dispatcher_service import Dispatcher
-                dispatcher = self._get_injector().get(Dispatcher)
+                dispatcher = self.get_injector().get(Dispatcher)
 
                 click.echo("🚀 Inicializando base de datos...")
-                dispatcher.init_db()
+                dispatcher.setup_all_companies()
                 click.echo("✅ Base de datos inicializada correctamente")
 
             except Exception as e:
@@ -334,6 +330,34 @@ class IAToolkit:
                 click.echo(f"❌ Error: {e}")
 
 
+        @self.app.cli.command("setup-company")
+        @click.argument("company_short_name")
+        def setup_company(company_short_name: str):
+            """⚙️ Ejecuta el proceso de configuración para una nueva empresa."""
+            try:
+                # step 1: init the database
+                dispatcher = self.get_injector().get(Dispatcher)
+                click.echo("🚀 step 1 of 2: init companies in the database...")
+                dispatcher.setup_all_companies()
+                click.echo("✅ database is ready.")
+
+                # step 2: generate the api key
+                profile_service = self.get_injector().get(ProfileService)
+                click.echo(f"🔑 step 2 of 2: generating api-key for use in '{company_short_name}'...")
+                result = profile_service.new_api_key(company_short_name)
+
+                if 'error' in result:
+                    click.echo(f"❌ Error in step 2: {result['error']}")
+                    click.echo("👉 Make sure company name is correct and it's initialized in your app.")
+                else:
+                    click.echo("Configuration es 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}")
+
+
     def _setup_context_processors(self):
         # Configura context processors para templates
         @self.app.context_processor
@@ -362,7 +386,7 @@ class IAToolkit:
         except:
             return 'templates'
 
-    def _get_injector(self) -> Injector:
+    def get_injector(self) -> Injector:
         """Obtiene el injector actual"""
         if not self._injector:
             raise IAToolkitException(

{iatoolkit-0.3.2 → iatoolkit-0.3.4}/src/iatoolkit.egg-info/PKG-INFO

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 0.3.2
+Version: 0.3.4
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT
-Requires-Python: >=3.10
+Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 Requires-Dist: aiohappyeyeballs==2.4.4
 Requires-Dist: aiohttp==3.11.9
@@ -206,3 +206,47 @@ Requires-Dist: wrapt==1.17.0
 Requires-Dist: yarl==1.18.3
 Requires-Dist: zipp==3.21.0
 Requires-Dist: zstandard==0.23.0
+
+# iatoolkit
+
+IAToolkit is a comprehensive, open-source framework designed for building enterprise-grade 
+AI chatbots and conversational applications. 
+Built on Flask with dependency injection, it provides a robust foundation for scalable AI solutions.
+
+## 🚀 Key Features
+- **Universal LLM Integration**: OpenAI GPT, Google Gemini
+- **Template System**: Jinja2-powered prompt templates with variables
+- **Context Management**: Maintain conversation context across sessions
+
+### 🔒 **Enterprise Security**
+- **JWT Authentication**: Secure token-based authentication
+- **Session Management**: Redis-backed secure sessions
+- **CORS Configuration**: Flexible cross-origin resource sharing
+
+### 🛠 **Function Calling & Tools**
+- **Native Function Calls**: Direct integration with LLM function calling
+- **Custom Tools**: Build and register custom tools for your chatbot
+- **SQL Query Generation**: Natural language to SQL conversion
+- **API Integrations**: Connect to external services and APIs
+
+### 🗄 **Database & Storage**
+- **Multi-Database Support**: PostgreSQL, MySQL, SQLite via SQLAlchemy
+- **Vector Store Integration**: Semantic search and retrieval
+- **Document Processing**: PDF, Word, Excel, and text file handling
+
+### 📊 **Analytics & Monitoring**
+- **Query Logging**: Track all LLM interactions
+- **Performance Metrics**: Response times, token usage, costs
+- **Benchmarking**: Compare model performance
+- **Task Management**: Async task processing with status tracking
+
+### 🔧 **Developer Experience**
+- **Dependency Injection**: Clean, testable architecture
+- **CLI Tools**: Command-line interface for common tasks
+- **Hot Reloading**: Development-friendly configuration
+- **Comprehensive Logging**: Debug and monitor easily
+
+## License
+MIT License
+
+

{iatoolkit-0.3.2 → iatoolkit-0.3.4}/src/iatoolkit.egg-info/SOURCES.txt

@@ -1,4 +1,5 @@
 pyproject.toml
+readme.md
 requirements.txt
 src/iatoolkit/__init__.py
 src/iatoolkit/base_company.py

iatoolkit-0.3.4/src/services/api_service.py

@@ -0,0 +1,75 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Producto: IAToolkit
+# Todos los derechos reservados.
+# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
+
+from infra.call_service import CallServiceClient
+from injector import inject
+from common.exceptions import IAToolkitException
+import json
+from typing import Optional, Dict, Any, Union
+
+class ApiService:
+    @inject
+    def __init__(self, call_service: CallServiceClient):
+        self.call_service = call_service
+
+    def call_api(
+        self,
+        endpoint: str,
+        method: str,
+        headers: Optional[Dict[str, str]] = None,
+        params: Optional[Dict[str, Union[str, int, float, bool]]] = None,
+        body: Optional[Dict[str, Any]] = None,
+        files: Optional[Dict[str, Any]] = None,
+        timeout: Union[int, float, tuple] = 10
+    ) -> str:
+        """
+        Ejecuta una llamada HTTP genérica.
+
+        - endpoint: URL completa
+        - method: GET | POST | PUT | DELETE (case-insensitive)
+        - headers: dict opcional de cabeceras
+        - params: dict opcional de query string
+        - body: dict opcional para JSON (POST/PUT/DELETE)
+        - files: dict opcional para multipart/form-data (prioriza POST files)
+        - timeout: segundos (int/float) o tuple (connect, read)
+        """
+        m = (method or "").strip().lower()
+
+        if m == "get":
+            response, status_code = self.call_service.get(
+                endpoint, params=params, headers=headers, timeout=timeout
+            )
+        elif m == "post":
+            # Si vienen files → multipart; si no → JSON
+            if files:
+                response, status_code = self.call_service.post_files(
+                    endpoint, data=files, params=params, headers=headers, timeout=timeout
+                )
+            else:
+                response, status_code = self.call_service.post(
+                    endpoint=endpoint, json_dict=body, params=params, headers=headers, timeout=timeout
+                )
+        elif m == "put":
+            response, status_code = self.call_service.put(
+                endpoint, json_dict=body, params=params, headers=headers, timeout=timeout
+            )
+        elif m == "delete":
+            response, status_code = self.call_service.delete(
+                endpoint, json_dict=body, params=params, headers=headers, timeout=timeout
+            )
+        else:
+            raise IAToolkitException(
+                IAToolkitException.ErrorType.INVALID_PARAMETER,
+                f"API error: método '{method}' no soportado"
+            )
+
+        if status_code < 200 or status_code >= 300:
+            raise IAToolkitException(
+                IAToolkitException.ErrorType.CALL_ERROR,
+                f"API {endpoint} error: {status_code}"
+            )
+
+        # Normalizamos a string JSON (para que el LLM lo consuma consistente)
+        return json.dumps(response)

{iatoolkit-0.3.2 → iatoolkit-0.3.4}/src/services/dispatcher_service.py

@@ -6,6 +6,7 @@
 from iatoolkit import current_iatoolkit
 from common.exceptions import IAToolkitException
 from services.prompt_manager_service import PromptService
+from services.api_service import ApiService
 from repositories.llm_query_repo import LLMQueryRepo
 from repositories.models import Company, Function
 from services.excel_service import ExcelService
@@ -23,10 +24,12 @@ class Dispatcher:
                  prompt_service: PromptService,
                  llmquery_repo: LLMQueryRepo,
                  util: Utility,
+                 api_service: ApiService,
                  excel_service: ExcelService,
                  mail_service: MailService):
         self.prompt_service = prompt_service
         self.llmquery_repo = llmquery_repo
+        self.api_service = api_service
         self.util = util
         self.excel_service = excel_service
         self.mail_service = mail_service
@@ -36,41 +39,42 @@ class Dispatcher:
         # Use the global registry
         self.company_registry = get_company_registry()
 
-        # The dispatcher starts "empty" and will be initialized later.
+        # load into the dispatcher the configured companies
         self.company_classes = {}
         self.initialize_companies()
 
+        # run the statrtup logic for all companies
+        self.start_execution()
+
         self.tool_handlers = {
             "iat_generate_excel": self.excel_service.excel_generator,
             "iat_send_email": self.mail_service.send_mail,
+            "iat_api_call": self.api_service.call_api
         }
 
     def initialize_companies(self):
         """
         Initializes and instantiates all registered company classes.
-        This method should be called *after* the main injector is fully configured.
+        This method should be called *after* the main injector is fully configured
+        and the company registry is populated.
         """
         if self.company_classes: # Prevent re-initialization
             return
 
         # ✅ NOW it is safe to get the injector and instantiate companies.
-        injector = current_iatoolkit()._get_injector()
+        injector = current_iatoolkit().get_injector()
         self.company_registry.set_injector(injector)
         self.company_classes = self.company_registry.instantiate_companies()
 
     def start_execution(self):
         """Runs the startup logic for all registered companies."""
-        # Ensure companies are initialized before starting them
-        if not self.company_classes:
-            self.initialize_companies()
-
         for company_name, company_instance in self.company_classes.items():
             logging.info(f'Starting execution for company: {company_name}')
             company_instance.start_execution()
 
         return True
 
-    def init_db(self):
+    def setup_all_companies(self):
         # create system functions
         for function in self.system_functions:
             self.llmquery_repo.create_or_update_function(
@@ -96,10 +100,9 @@ class Dispatcher:
 
         # initialize the database for every company class
         for company in self.company_classes.values():
-            print(f'inicializando clase: {company.__class__.__name__}')
+            print(f'company: {company.__class__.__name__}')
             company.init_db()
 
-
     def dispatch(self, company_name: str, action: str, **kwargs) -> str:
         company_key = company_name.lower()
 

{iatoolkit-0.3.2 → iatoolkit-0.3.4}/src/services/profile_service.py

@@ -266,7 +266,7 @@ class ProfileService:
 
         api_key = ApiKey(key=key, company_id=company.id)
         self.profile_repo.create_api_key(api_key)
-        return {"message": f"La nueva clave de API para {company_short_name} es: {key}"}
+        return {"api-key": key}
 
 
     def send_verification_email(self, new_user: User, company_short_name):

{iatoolkit-0.3.2 → iatoolkit-0.3.4}/src/services/prompt_manager_service.py

@@ -11,6 +11,7 @@ from collections import defaultdict
 from repositories.models import Prompt, PromptCategory, Company
 import os
 from common.exceptions import IAToolkitException
+from pathlib import Path
 
 
 class PromptService:
@@ -32,7 +33,7 @@ class PromptService:
 
         prompt_filename = prompt_name.lower() + '.prompt'
         if is_system_prompt:
-            template_dir = 'prompts'
+            template_dir = 'src/system_prompts'
         else:
             template_dir = f'companies/{company.short_name}/prompts'
 
@@ -103,16 +104,15 @@ class PromptService:
             system_prompt_content = []
 
             # get the filepaths for all system prompts
-            current_dir = os.path.dirname(os.path.abspath(__file__))
-            src_dir = os.path.dirname(current_dir)  # ../src
-            system_prompt_dir = os.path.join(src_dir, "prompts")
+            current_dir = Path(__file__).resolve().parent
+            project_root = current_dir.parent.parent
 
-            # Obtener, ordenar y leer los system prompts
+            # read all the system prompts from the database
             system_prompts = self.llm_query_repo.get_system_prompts()
 
             for prompt in system_prompts:
-                # Construir la ruta absoluta para leer el archivo
-                absolute_filepath = os.path.join(system_prompt_dir, prompt.filepath)
+                # build the absolute filepath for reading it
+                absolute_filepath = os.path.join(project_root, prompt.filepath)
                 if not os.path.exists(absolute_filepath):
                     logging.warning(f"El archivo para el prompt de sistema no existe: {absolute_filepath}")
                     continue
@@ -123,7 +123,7 @@ class PromptService:
                     raise IAToolkitException(IAToolkitException.ErrorType.FILE_IO_ERROR,
                                        f"Error leyendo el archivo de prompt del sistema {absolute_filepath}: {e}")
 
-            # Unir todo el contenido en un solo string
+            # join the system prompts into a single string
             return "\n".join(system_prompt_content)
 
         except IAToolkitException:

iatoolkit-0.3.2/src/services/api_service.py

@@ -1,30 +0,0 @@
-# Copyright (c) 2024 Fernando Libedinsky
-# Producto: IAToolkit
-# Todos los derechos reservados.
-# En trámite de registro en el Registro de Propiedad Intelectual de Chile.
-
-from infra.call_service import CallServiceClient
-from injector import inject
-from common.exceptions import IAToolkitException
-import json
-
-
-class ApiService:
-    @inject
-    def __init__(self, call_service: CallServiceClient):
-        self.call_service = call_service
-
-    def call_api(self, endpoint: str, method: str, **kwargs):
-        if method == 'get':
-            response, status_code = self.call_service.get(endpoint)
-        elif method == 'post':
-            response, status_code = self.call_service.post(endpoint=endpoint, json_dict=kwargs)
-        else:
-            raise IAToolkitException(IAToolkitException.ErrorType.INVALID_PARAMETER,
-                               f'API error, {method} not supported')
-
-        if status_code != 200:
-            raise IAToolkitException(IAToolkitException.ErrorType.CALL_ERROR,
-                               f'API {endpoint} error: {status_code}')
-
-        return json.dumps(response)