crewplus 0.2.34__tar.gz → 0.2.36__tar.gz

{crewplus-0.2.34 → crewplus-0.2.36}/PKG-INFO

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

{crewplus-0.2.34 → crewplus-0.2.36}/crewplus/services/gemini_chat_model.py

@@ -294,6 +294,7 @@ class GeminiChatModel(BaseChatModel):
         if creds is None:
             # Get service account file from env if not provided
             sa_file = self.service_account_file or os.getenv("GCP_SERVICE_ACCOUNT_FILE")
+            self.logger.debug(f"Service account file: {sa_file}")
             if sa_file:
                 try:
                     creds = service_account.Credentials.from_service_account_file(
@@ -307,15 +308,21 @@ class GeminiChatModel(BaseChatModel):
         
         # If creds is still None, the client will use Application Default Credentials (ADC).
         
-        self._client = genai.Client(
-            project=self.project_id,
-            location=self.location,
-            credentials=creds,
-        )
-        self.logger.info(
-            f"Initialized GeminiChatModel with model: {self.model_name} for Vertex AI "
-            f"(Project: {self.project_id}, Location: {self.location})"
-        )
+        try:
+            self._client = genai.Client(
+                vertexai=True,
+                project=self.project_id,
+                location=self.location,
+                credentials=creds,
+            )
+            self.logger.info(
+                f"Initialized GeminiChatModel with model: {self.model_name} for Vertex AI "
+                f"(Project: {self.project_id}, Location: {self.location})"
+            )
+        except Exception as e:
+            error_msg = f"Failed to initialize GenAI Client for Vertex AI: {e}"
+            self.logger.error(error_msg, exc_info=True)
+            raise ValueError(error_msg)
 
     def get_model_identifier(self) -> str:
         """Return a string identifying this model for tracing and logging."""

{crewplus-0.2.34 → crewplus-0.2.36}/crewplus/services/init_services.py

@@ -1,15 +1,27 @@
 import os
+import logging
+from typing import Optional
 from .model_load_balancer import ModelLoadBalancer
 
 model_balancer = None
 
-def init_load_balancer(config_path: str = None):
+def init_load_balancer(
+    config_path: Optional[str] = None, 
+    logger: Optional[logging.Logger] = None
+):
     """
     Initializes the global ModelLoadBalancer instance.
 
     This function is idempotent. If the balancer is already initialized,
     it does nothing. It follows a safe initialization pattern where the
     global instance is only assigned after successful configuration loading.
+    
+    Args:
+        config_path (Optional[str]): The path to the model configuration file.
+            If not provided, it's determined by the `MODEL_CONFIG_PATH`
+            environment variable, or defaults to "config/models_config.json".
+        logger (Optional[logging.Logger]): An optional logger instance to be
+            used by the model balancer.
     """
     global model_balancer
     if model_balancer is None:
@@ -20,7 +32,10 @@ def init_load_balancer(config_path: str = None):
         )
         try:
             # 1. Create a local instance first.
-            balancer = ModelLoadBalancer(final_config_path)
+            balancer = ModelLoadBalancer(
+                config_path=final_config_path, 
+                logger=logger
+            )
             # 2. Attempt to load its configuration.
             balancer.load_config()
             # 3. Only assign to the global variable on full success.

{crewplus-0.2.34 → crewplus-0.2.36}/pyproject.toml

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