crewplus 0.2.9__py3-none-any.whl → 0.2.10__py3-none-any.whl

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/services/__init__.py

@@ -1 +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/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/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/utils/__init__.py

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

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/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):
     """