fixed mypy

Author: eavanvalkenburg

python/semantic_kernel/connectors/memory/__init__.py

@@ -6,13 +6,13 @@
     "AzureAISearchCollection": ".azure_ai_search",
     "AzureAISearchSettings": ".azure_ai_search",
     "AzureAISearchStore": ".azure_ai_search",
-    "AzureCosmosDBNoSQLCollection": ".azure_cosmos_db",
-    "AzureCosmosDBNoSQLCompositeKey": ".azure_cosmos_db",
-    "AzureCosmosDBNoSQLSettings": ".azure_cosmos_db",
-    "AzureCosmosDBNoSQLStore": ".azure_cosmos_db",
-    "AzureCosmosDBforMongoDBCollection": ".azure_cosmos_db",
-    "AzureCosmosDBforMongoDBSettings": ".azure_cosmos_db",
-    "AzureCosmosDBforMongoDBStore": ".azure_cosmos_db",
+    "CosmosNoSqlCollection": ".azure_cosmos_db",
+    "CosmosNoSqlCompositeKey": ".azure_cosmos_db",
+    "CosmosNoSqlSettings": ".azure_cosmos_db",
+    "CosmosNoSqlStore": ".azure_cosmos_db",
+    "CosmosMongoCollection": ".azure_cosmos_db",
+    "CosmosMongoSettings": ".azure_cosmos_db",
+    "CosmosMongoStore": ".azure_cosmos_db",
     "ChromaCollection": ".chroma",
     "ChromaStore": ".chroma",
     "PostgresCollection": ".postgres",

python/semantic_kernel/connectors/memory/__init__.pyi

@@ -2,12 +2,12 @@
 
 from .azure_ai_search import AzureAISearchCollection, AzureAISearchSettings, AzureAISearchStore
 from .azure_cosmos_db import (
-    AzureCosmosDBforMongoDBSettings,
-    AzureCosmosDBNoSQLCompositeKey,
-    AzureCosmosDBNoSQLSettings,
     CosmosMongoCollection,
+    CosmosMongoSettings,
     CosmosMongoStore,
     CosmosNoSqlCollection,
+    CosmosNoSqlCompositeKey,
+    CosmosNoSqlSettings,
     CosmosNoSqlStore,
 )
 from .chroma import ChromaCollection, ChromaStore
@@ -25,14 +25,14 @@ __all__ = [
     "AzureAISearchCollection",
     "AzureAISearchSettings",
     "AzureAISearchStore",
-    "AzureCosmosDBNoSQLCompositeKey",
-    "AzureCosmosDBNoSQLSettings",
-    "AzureCosmosDBforMongoDBSettings",
     "ChromaCollection",
     "ChromaStore",
     "CosmosMongoCollection",
+    "CosmosMongoSettings",
     "CosmosMongoStore",
     "CosmosNoSqlCollection",
+    "CosmosNoSqlCompositeKey",
+    "CosmosNoSqlSettings",
     "CosmosNoSqlStore",
     "FaissCollection",
     "FaissStore",

python/semantic_kernel/data/definitions.py

@@ -391,7 +391,7 @@ def _parse_vector_store_record_field_class(
     if (args := getattr(property_type, "__args__", None)) and NoneType in args and len(args) == 2:
         property_type = args[0]
     property_type_name = str(property_type) if hasattr(property_type, "__args__") else property_type.__name__
-    return field_type(name=field.name, type_=property_type_name)
+    return field_type(name=field.name, type=property_type_name)
 
 
 def _parse_vector_store_record_field_instance(

python/semantic_kernel/exceptions/vector_store_exceptions.py

@@ -81,4 +81,5 @@ class VectorSearchOptionsException(VectorStoreOperationException):
     "VectorStoreModelSerializationException",
     "VectorStoreModelValidationError",
     "VectorStoreOperationException",
+    "VectorStoreOperationNotSupportedException",
 ]