crewplus 0.2.9__tar.gz → 0.2.10__tar.gz

{crewplus-0.2.9 → crewplus-0.2.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: crewplus
-Version: 0.2.9
+Version: 0.2.10
 Summary: Base services for CrewPlus AI applications
 Author-Email: Tim Liu <tim@opsmateai.com>
 License: MIT
@@ -45,7 +45,7 @@ CrewPlus is designed as a modular and extensible ecosystem of packages. This all
 
 -   **Chat Services:** A unified interface for interacting with various chat models (e.g., `GeminiChatModel`).
 -   **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.
+-   **Vector DB Services:** Abstractions for working with popular vector stores for retrieval-augmented generation (RAG).
 
 
 ## Documentation

{crewplus-0.2.9 → crewplus-0.2.10}/README.md

@@ -25,7 +25,7 @@ CrewPlus is designed as a modular and extensible ecosystem of packages. This all
 
 -   **Chat Services:** A unified interface for interacting with various chat models (e.g., `GeminiChatModel`).
 -   **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.
+-   **Vector DB Services:** Abstractions for working with popular vector stores for retrieval-augmented generation (RAG).
 
 
 ## Documentation

crewplus-0.2.10/crewplus/__init__.py

@@ -0,0 +1,10 @@
+from .services.gemini_chat_model import GeminiChatModel
+from .services.model_load_balancer import ModelLoadBalancer
+from .vectorstores.milvus import SchemaMilvus, VDBService
+
+__all__ = [
+    "GeminiChatModel",
+    "ModelLoadBalancer",
+    "SchemaMilvus",
+    "VDBService"
+]

crewplus-0.2.10/crewplus/services/__init__.py

@@ -0,0 +1,6 @@
+from .gemini_chat_model import GeminiChatModel
+from .init_services import init_load_balancer, get_model_balancer
+from .model_load_balancer import ModelLoadBalancer
+
+
+__all__ = ["GeminiChatModel", "init_load_balancer", "get_model_balancer", "ModelLoadBalancer"]

{crewplus-0.2.9 → crewplus-0.2.10}/crewplus/services/init_services.py

@@ -3,15 +3,11 @@ from crewplus.services.model_load_balancer import ModelLoadBalancer
 
 model_balancer = None
 
-def init_load_balancer(config_path: str = None):
+def init_load_balancer():
     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)
+        config_path = os.getenv("MODEL_CONFIG_PATH", "config/models_config.json")
+        model_balancer = ModelLoadBalancer(config_path)
         model_balancer.load_config()  # Load initial configuration synchronously
 
 def get_model_balancer() -> ModelLoadBalancer:

{crewplus-0.2.9 → crewplus-0.2.10}/crewplus/services/model_load_balancer.py

@@ -141,7 +141,6 @@ class ModelLoadBalancer:
             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'],

crewplus-0.2.10/crewplus/utils/__init__.py

@@ -0,0 +1,4 @@
+from .schema_action import Action
+from .schema_document_updater import SchemaDocumentUpdater
+
+__all__ = ["Action", "SchemaDocumentUpdater"]

crewplus-0.2.10/crewplus/vectorstores/milvus/__init__.py

@@ -0,0 +1,5 @@
+from .milvus_schema_manager import MilvusSchemaManager, ZillizSchemaManager
+from .schema_milvus import SchemaMilvus
+from .vdb_service import VDBService
+
+__all__ = ["MilvusSchemaManager", "ZillizSchemaManager", "VDBService"]

{crewplus-0.2.9 → crewplus-0.2.10}/crewplus/vectorstores/milvus/schema_milvus.py

@@ -5,34 +5,9 @@ import json
 from pymilvus import DataType
 from langchain_milvus import Milvus
 from langchain_core.documents import Document
-from crewplus.utils.schema_document_updater import SchemaDocumentUpdater
-from crewplus.utils.schema_action import Action
+from ...utils import SchemaDocumentUpdater, Action
 from .milvus_schema_manager import MilvusSchemaManager
 
-DEFAULT_SCHEMA = """
-{
-    "node_types": {
-        "Document": {
-            "properties": {
-                "pk": {
-                    "type": "INT64",
-                    "is_primary": true,
-                    "auto_id": true
-                },
-                "vector": {
-                    "type": "FLOAT_VECTOR",
-                    "dim": 1536
-                },
-                "text": {
-                    "type": "VARCHAR",
-                    "max_length": 65535,
-                    "description": "The core text of the memory. This could be a user query, a documented fact, a procedural step, or a log of an event."
-                }
-	        }
-	    }
-    }
-}
-"""
 
 class SchemaMilvus(Milvus):
     """