crewplus 0.2.23__tar.gz → 0.2.24__tar.gz

{crewplus-0.2.23 → crewplus-0.2.24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: crewplus
-Version: 0.2.23
+Version: 0.2.24
 Summary: Base services for CrewPlus AI applications
 Author-Email: Tim Liu <tim@opsmateai.com>
 License: MIT

{crewplus-0.2.23 → crewplus-0.2.24}/crewplus/services/model_load_balancer.py

@@ -63,7 +63,11 @@ class ModelLoadBalancer:
             # Instantiate models
             for model_config in self.models_config:
                 model_id = model_config['id']
-                self.models[model_id] = self._instantiate_model(model_config)
+                model_instance = self._instantiate_model(model_config)
+                if model_instance is not None:
+                    self.models[model_id] = model_instance
+                else:
+                    self.logger.warning(f"Model with id {model_id} was not loaded due to instantiation error.")
 
             self._config_loaded = True
             self.logger.debug("Model balancer: configuration loaded successfully.")
@@ -141,15 +145,21 @@ class ModelLoadBalancer:
                 kwargs['temperature'] = model_config['temperature']
             return ChatOpenAI(**kwargs)
         elif provider == 'azure-openai-embeddings':
-            return AzureOpenAIEmbeddings(
-                model=model_config['model_name'],
-                azure_deployment=model_config['deployment_name'],
-                openai_api_version=model_config['api_version'],
-                api_key=model_config['api_key'],
-                azure_endpoint=model_config['api_base'],
-                dimensions=model_config.get('dimensions', 1536),
-                chunk_size=16, request_timeout=60, max_retries=2
-            )
+            try:
+                emb_kwargs = dict(
+                    model=model_config['model_name'],
+                    azure_deployment=model_config['deployment_name'],
+                    openai_api_version=model_config['api_version'],
+                    api_key=model_config['api_key'],
+                    azure_endpoint=model_config['api_base'],
+                    chunk_size=16, request_timeout=60, max_retries=2
+                )
+                if 'dimensions' in model_config:
+                    emb_kwargs['dimensions'] = model_config['dimensions']
+                return AzureOpenAIEmbeddings(**emb_kwargs)
+            except Exception as e:
+                self.logger.error(f"Failed to instantiate AzureOpenAIEmbeddings: {e}")
+                return None
         elif provider == 'google-genai':
             kwargs = {
                 'google_api_key': model_config['api_key'],

{crewplus-0.2.23 → crewplus-0.2.24}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "pdm.backend"
 
 [project]
 name = "crewplus"
-version = "0.2.23"
+version = "0.2.24"
 description = "Base services for CrewPlus AI applications"
 authors = [
     { name = "Tim Liu", email = "tim@opsmateai.com" },