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

iatoolkit/iatoolkit.py

@@ -19,7 +19,7 @@ from werkzeug.middleware.proxy_fix import ProxyFix
 from injector import Binder, Injector, singleton
 from importlib.metadata import version as _pkg_version, PackageNotFoundError
 
-IATOOLKIT_VERSION = "0.71.2"
+IATOOLKIT_VERSION = "0.71.4"
 
 # global variable for the unique instance of IAToolkit
 _iatoolkit_instance: Optional['IAToolkit'] = None

iatoolkit/services/embedding_service.py

@@ -50,7 +50,6 @@ class EmbeddingClientFactory:
     def __init__(self, config_service: ConfigurationService):
         self.config_service = config_service
         self._clients = {}  # Cache for storing initialized client wrappers
-        self._lock = Lock()
 
     def get_client(self, company_short_name: str) -> EmbeddingClientWrapper:
         """
@@ -60,42 +59,42 @@ class EmbeddingClientFactory:
         if company_short_name in self._clients:
             return self._clients[company_short_name]
 
-        with self._lock:
-            if company_short_name in self._clients:
-                return self._clients[company_short_name]
-
-            embedding_config = self.config_service.get_configuration(company_short_name, 'embedding_provider')
-            if not embedding_config:
-                raise ValueError(f"Embedding provider not configured for company '{company_short_name}'.")
-
-            provider = embedding_config.get('provider')
-            if not provider:
-                raise ValueError(f"Embedding provider not configured for company '{company_short_name}'.")
-            model = embedding_config.get('model')
-
-            api_key_name = embedding_config.get('api_key_name')
-            api_key = os.getenv(api_key_name)
-            if not api_key:
-                raise ValueError(f"Environment variable '{api_key_name}' is not set.")
-
-            # Logic to handle multiple providers
-            wrapper = None
-            if provider == 'huggingface':
-                if not model:
-                    model='sentence-transformers/all-MiniLM-L6-v2'
-                client = InferenceClient(model=model, token=api_key)
-                wrapper = HuggingFaceClientWrapper(client, model)
-            elif provider == 'openai':
-                client = OpenAI(api_key=api_key)
-                if not model:
-                    model='text-embedding-ada-002'
-                wrapper = OpenAIClientWrapper(client, model)
-            else:
-                raise NotImplementedError(f"Embedding provider '{provider}' is not implemented.")
-
-            logging.info(f"Embedding client for '{company_short_name}' created with model: {model} via {provider}")
-            self._clients[company_short_name] = wrapper
-            return wrapper
+        # Get the embedding provider and model from the company.yaml
+        embedding_config = self.config_service.get_configuration(company_short_name, 'embedding_provider')
+        if not embedding_config:
+            raise ValueError(f"Embedding provider not configured for company '{company_short_name}'.")
+
+        provider = embedding_config.get('provider')
+        if not provider:
+            raise ValueError(f"Embedding provider not configured for company '{company_short_name}'.")
+        model = embedding_config.get('model')
+
+        api_key_name = embedding_config.get('api_key_name')
+        if not api_key_name:
+            raise ValueError(f"Missiong configuration for embedding_provider:api_key_name en config.yaml.")
+
+        api_key = os.getenv(api_key_name)
+        if not api_key:
+            raise ValueError(f"Environment variable '{api_key_name}' is not set.")
+
+        # Logic to handle multiple providers
+        wrapper = None
+        if provider == 'huggingface':
+            if not model:
+                model='sentence-transformers/all-MiniLM-L6-v2'
+            client = InferenceClient(model=model, token=api_key)
+            wrapper = HuggingFaceClientWrapper(client, model)
+        elif provider == 'openai':
+            client = OpenAI(api_key=api_key)
+            if not model:
+                model='text-embedding-ada-002'
+            wrapper = OpenAIClientWrapper(client, model)
+        else:
+            raise NotImplementedError(f"Embedding provider '{provider}' is not implemented.")
+
+        logging.debug(f"Embedding client for '{company_short_name}' created with model: {model} via {provider}")
+        self._clients[company_short_name] = wrapper
+        return wrapper
 
 class EmbeddingService:
     """

{iatoolkit-0.71.2.dist-info → iatoolkit-0.71.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 0.71.2
+Version: 0.71.4
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT

{iatoolkit-0.71.2.dist-info → iatoolkit-0.71.4.dist-info}/RECORD

@@ -2,7 +2,7 @@ iatoolkit/__init__.py,sha256=lWMmZiXzVqOAD2cnLZTjoX5_EMoDljjU7oAZTQbKu5c,951
 iatoolkit/base_company.py,sha256=U0KWZf7zLrJCw87NrEt2MCCHfd78HS22D7ht_BLeoL0,3522
 iatoolkit/cli_commands.py,sha256=bTn4iXiDM4P2Nyn6oPT2wCL4pUQy34hVTL6c4fyEQdY,2303
 iatoolkit/company_registry.py,sha256=NAsrCJyvgkdG8yGfqA7EdPUTHDzzcPzKu0PKXniub50,2490
-iatoolkit/iatoolkit.py,sha256=Z5aPSnKlepfqWqsChSju4aQVZttDI-hxUy1p0Eo1WIc,19439
+iatoolkit/iatoolkit.py,sha256=Svj5NGzx5JpI_KXC-Z6DY5y8aBKweoDfj7vYJIN4K6A,19439
 iatoolkit/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 iatoolkit/common/exceptions.py,sha256=EXx40n5htp7UiOM6P1xfJ9U6NMcADqm62dlFaKz7ICU,1154
 iatoolkit/common/routes.py,sha256=oFL8ivzenGpNgiDjV0pda_NRZ5XrlEHSIHPWW53zj8k,6529
@@ -43,7 +43,7 @@ iatoolkit/services/company_context_service.py,sha256=ME9dshCY1J1L6Kz_anrvoSfWy2Z
 iatoolkit/services/configuration_service.py,sha256=TmWRqne9M9vf06c9_Hlfc8VMB--s99iK39hf8JSyO3k,5746
 iatoolkit/services/dispatcher_service.py,sha256=wHZV4AU4okYKuFwH5afB3ktYDE2Ej4XVvxmUlL-IsmM,12635
 iatoolkit/services/document_service.py,sha256=whW3B9g7lCaNOqezYmoHkSAov0K7EH90u-KVANO0mPg,6057
-iatoolkit/services/embedding_service.py,sha256=AxcNVEhUeUXXYU80kafS70D-kFBlnN2_EnR6doae-cE,6110
+iatoolkit/services/embedding_service.py,sha256=wVW2OXT26KPwiAzFmBAi2gHuq-vY0UFhX_SBuq8RLks,6043
 iatoolkit/services/excel_service.py,sha256=MGVFjSo1NOL8PL7_hVgEEFzvXfA4NmF-JFEr8FBSjOA,4090
 iatoolkit/services/file_processor_service.py,sha256=h8Tg2Xwvjv_HOIUo51Rnu5H4tUmkJ8W9tKvmFfgKPms,3792
 iatoolkit/services/history_service.py,sha256=HoJs5CRpXturrdycznlalL6TOPbzC2npziTs86mOIkg,1375
@@ -115,8 +115,8 @@ iatoolkit/views/tasks_api_view.py,sha256=wGnuwuuL83ByQ1Yre6ytRVztA0OGQjGrwMjB1_G
 iatoolkit/views/tasks_review_api_view.py,sha256=wsCpzqyRyUdCXWAhyGlBe3eNZZ6A1DQG7TblN_GZNfM,1894
 iatoolkit/views/user_feedback_api_view.py,sha256=QOTBtpNqYAmewXDgVAoaIprnVjvf1xM0ExWxSFp3ICI,2098
 iatoolkit/views/verify_user_view.py,sha256=TTC9BNjyjT8ZT5KnQ2f5-8UF1Y_sJII2wJ501xWSFvY,2653
-iatoolkit-0.71.2.dist-info/licenses/LICENSE,sha256=5tOLQdjoCvSXEx_7Lr4bSab3ha4NSwzamvua0fwCXi8,1075
-iatoolkit-0.71.2.dist-info/METADATA,sha256=en0VLL6vKIP1H7lXkakyLlzXotm7FNDJZqStOOHfvt0,9963
-iatoolkit-0.71.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-iatoolkit-0.71.2.dist-info/top_level.txt,sha256=V_w4QvDx0b1RXiy8zTCrD1Bp7AZkFe3_O0-9fMiwogg,10
-iatoolkit-0.71.2.dist-info/RECORD,,
+iatoolkit-0.71.4.dist-info/licenses/LICENSE,sha256=5tOLQdjoCvSXEx_7Lr4bSab3ha4NSwzamvua0fwCXi8,1075
+iatoolkit-0.71.4.dist-info/METADATA,sha256=7leXcUn7QdAkVAk-d3XTd18X-jNcisMR7-PtZw1JXZg,9963
+iatoolkit-0.71.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+iatoolkit-0.71.4.dist-info/top_level.txt,sha256=V_w4QvDx0b1RXiy8zTCrD1Bp7AZkFe3_O0-9fMiwogg,10
+iatoolkit-0.71.4.dist-info/RECORD,,