crewplus 0.2.26__tar.gz → 0.2.28__tar.gz

{crewplus-0.2.26 → crewplus-0.2.28}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: crewplus
-Version: 0.2.26
+Version: 0.2.28
 Summary: Base services for CrewPlus AI applications
 Author-Email: Tim Liu <tim@opsmateai.com>
 License: MIT
@@ -14,6 +14,7 @@ Requires-Dist: langchain-openai==0.3.24
 Requires-Dist: google-genai==1.21.1
 Requires-Dist: langchain-milvus<0.3.0,>=0.2.1
 Requires-Dist: langfuse<4.0.0,>=3.1.3
+Requires-Dist: langchain-mcp-adapters>=0.1.4
 Description-Content-Type: text/markdown
 
 # CrewPlus
@@ -34,16 +35,17 @@ This repository, `crewplus-base`, contains the core `crewplus` Python package. I
 CrewPlus is designed as a modular and extensible ecosystem of packages. This allows you to adopt only the components you need for your specific use case.
 
 -   **`crewplus` (This package):** The core package containing foundational services for chat, model load balancing, and vector stores.
--   **`crewplus-agents`:** An extension for creating and managing autonomous AI agents.
+-   **`crewplus-agent`:** crewplus agent core: agentic task planner and executor, with context-aware memory.
 -   **`crewplus-ingestion`:** Provides robust pipelines for knowledge ingestion and data processing.
 -   **`crewplus-memory`:** Provides agent memory services for Crewplus AI Agents.
 -   **`crewplus-integrations`:** A collection of third-party integrations to connect CrewPlus with other services and platforms.
 
 ## Features
 
--   **Chat Services:** A unified interface for interacting with various chat models (e.g., `GeminiChatModel`).
+-   **Chat Services:** A unified interface for interacting with various chat models (e.g., `GeminiChatModel`, `TracedAzureChatOpenAI`).
 -   **Model Load Balancer:** Intelligently distribute requests across multiple LLM endpoints.
 -   **Vector DB Services:** working with popular vector stores (e.g. Milvus, Zilliz Cloud) for retrieval-augmented generation (RAG) and agent memory.
+-   **Observability & Tracing:** Automatic integration with tracing tools like Langfuse, with an extensible design for adding others (e.g., Helicone, ...).
 
 
 ## Documentation
@@ -92,15 +94,17 @@ crewplus-base/                    # GitHub repo name
 │   └──  services/
 │       └──  __init__.py
 │       └──  gemini_chat_model.py
+│       └──  azure_chat_model.py
 │       └──  model_load_balancer.py
+│       └──  tracing_manager.py
 │       └──  ...
 │   └──  vectorstores/milvus
 │       └──  __init__.py
 │       └──  schema_milvus.py
 │       └──  vdb_service.py
-│   └──  core/
+│   └──  utils/
 │       └──  __init__.py
-│       └──  config.py
+│       └──  schema_action.py
 │       └──  ...
 ├── tests/
 │   └── ...

{crewplus-0.2.26 → crewplus-0.2.28}/README.md

@@ -16,16 +16,17 @@ This repository, `crewplus-base`, contains the core `crewplus` Python package. I
 CrewPlus is designed as a modular and extensible ecosystem of packages. This allows you to adopt only the components you need for your specific use case.
 
 -   **`crewplus` (This package):** The core package containing foundational services for chat, model load balancing, and vector stores.
--   **`crewplus-agents`:** An extension for creating and managing autonomous AI agents.
+-   **`crewplus-agent`:** crewplus agent core: agentic task planner and executor, with context-aware memory.
 -   **`crewplus-ingestion`:** Provides robust pipelines for knowledge ingestion and data processing.
 -   **`crewplus-memory`:** Provides agent memory services for Crewplus AI Agents.
 -   **`crewplus-integrations`:** A collection of third-party integrations to connect CrewPlus with other services and platforms.
 
 ## Features
 
--   **Chat Services:** A unified interface for interacting with various chat models (e.g., `GeminiChatModel`).
+-   **Chat Services:** A unified interface for interacting with various chat models (e.g., `GeminiChatModel`, `TracedAzureChatOpenAI`).
 -   **Model Load Balancer:** Intelligently distribute requests across multiple LLM endpoints.
 -   **Vector DB Services:** working with popular vector stores (e.g. Milvus, Zilliz Cloud) for retrieval-augmented generation (RAG) and agent memory.
+-   **Observability & Tracing:** Automatic integration with tracing tools like Langfuse, with an extensible design for adding others (e.g., Helicone, ...).
 
 
 ## Documentation
@@ -74,15 +75,17 @@ crewplus-base/                    # GitHub repo name
 │   └──  services/
 │       └──  __init__.py
 │       └──  gemini_chat_model.py
+│       └──  azure_chat_model.py
 │       └──  model_load_balancer.py
+│       └──  tracing_manager.py
 │       └──  ...
 │   └──  vectorstores/milvus
 │       └──  __init__.py
 │       └──  schema_milvus.py
 │       └──  vdb_service.py
-│   └──  core/
+│   └──  utils/
 │       └──  __init__.py
-│       └──  config.py
+│       └──  schema_action.py
 │       └──  ...
 ├── tests/
 │   └── ...

crewplus-0.2.28/crewplus/services/init_services.py

@@ -0,0 +1,37 @@
+import os
+from .model_load_balancer import ModelLoadBalancer
+
+model_balancer = None
+
+def init_load_balancer(config_path: str = 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.
+    """
+    global model_balancer
+    if model_balancer is None:
+        # Use parameter if provided, otherwise check env var, then default
+        final_config_path = config_path or os.getenv(
+            "MODEL_CONFIG_PATH", 
+            "config/models_config.json"
+        )
+        try:
+            # 1. Create a local instance first.
+            balancer = ModelLoadBalancer(final_config_path)
+            # 2. Attempt to load its configuration.
+            balancer.load_config()
+            # 3. Only assign to the global variable on full success.
+            model_balancer = balancer
+        except Exception as e:
+            # If any step fails, the global model_balancer remains None,
+            # allowing for another initialization attempt later.
+            # Re-raise the exception to notify the caller of the failure.
+            raise RuntimeError(f"Failed to initialize and configure ModelLoadBalancer from {final_config_path}: {e}") from e
+
+def get_model_balancer() -> ModelLoadBalancer:
+    if model_balancer is None:
+        raise RuntimeError("ModelLoadBalancer not initialized. Please call init_load_balancer() first.")
+    return model_balancer

{crewplus-0.2.26 → crewplus-0.2.28}/crewplus/services/tracing_manager.py

@@ -104,9 +104,10 @@ class TracingManager:
         """
         Adds all registered tracing handlers to the request configuration.
         
-        This method is idempotent; it will not add a handler if one of the
-        same type is already present in the configuration's callback list. It
-        also respects a `tracing_disabled` flag in the metadata.
+        This method is robust and handles three scenarios for the 'callbacks' key:
+        1. A list of callbacks.
+        2. A LangChain CallbackManager instance.
+        3. None or a missing key.
         
         Args:
             config: The request configuration dictionary from a LangChain call.
@@ -121,16 +122,27 @@ class TracingManager:
         if not self._handlers or config.get("metadata", {}).get("tracing_disabled"):
             return config
         
-        # Use an empty list as a safe default if 'callbacks' is missing or None.
-        callbacks = config.get("callbacks") or []
-        new_callbacks = list(callbacks)
+        callbacks = config.get("callbacks")
+        
+        # Case 1: The 'callbacks' key holds a CallbackManager instance
+        if hasattr(callbacks, 'add_handler') and hasattr(callbacks, 'handlers'):
+            for handler in self._handlers:
+                if not any(isinstance(cb, type(handler)) for cb in callbacks.handlers):
+                    callbacks.add_handler(handler, inherit=True)
+            return config # Return the original, now-mutated config
+        
+        # Case 2: The 'callbacks' key holds a list or is None
+        current_callbacks = callbacks or []
+        new_callbacks = list(current_callbacks)
         
         for handler in self._handlers:
             if not any(isinstance(cb, type(handler)) for cb in new_callbacks):
                 new_callbacks.append(handler)
         
-        # Only create a new config dictionary if callbacks were actually added.
-        if len(new_callbacks) > len(callbacks):
+        if len(new_callbacks) > len(current_callbacks):
+            # Create a new dictionary with the updated callbacks list.
+            # This is a safe operation that overwrites the existing 'callbacks'
+            # key and avoids mutating the original config object.
             return {**config, "callbacks": new_callbacks}
         
         return config

{crewplus-0.2.26 → crewplus-0.2.28}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "pdm.backend"
 
 [project]
 name = "crewplus"
-version = "0.2.26"
+version = "0.2.28"
 description = "Base services for CrewPlus AI applications"
 authors = [
     { name = "Tim Liu", email = "tim@opsmateai.com" },
@@ -19,6 +19,7 @@ dependencies = [
     "google-genai==1.21.1",
     "langchain-milvus (>=0.2.1,<0.3.0)",
     "langfuse (>=3.1.3,<4.0.0)",
+    "langchain-mcp-adapters>=0.1.4",
 ]
 
 [project.license]

crewplus-0.2.26/crewplus/services/init_services.py

@@ -1,20 +0,0 @@
-import os
-from .model_load_balancer import ModelLoadBalancer
-
-model_balancer = None
-
-def init_load_balancer(config_path: str = None):
-    global model_balancer
-    if model_balancer is None:
-        # Use parameter if provided, otherwise check env var, then default
-        final_config_path = config_path or os.getenv(
-            "MODEL_CONFIG_PATH", 
-            "config/models_config.json"  # Fixed default path
-        )
-        model_balancer = ModelLoadBalancer(final_config_path)
-        model_balancer.load_config()  # Load initial configuration synchronously
-
-def get_model_balancer() -> ModelLoadBalancer:
-    if model_balancer is None:
-        raise RuntimeError("ModelLoadBalancer not initialized")
-    return model_balancer