semantic-kernel 0.2.4.dev0__py3-none-any.whl → 0.2.5.dev0__py3-none-any.whl

semantic_kernel/ai/embeddings/embedding_index_base.py

@@ -1,20 +0,0 @@
-# Copyright (c) Microsoft. All rights reserved.
-
-from abc import ABC, abstractmethod
-from typing import List, Tuple
-
-from numpy import ndarray
-
-from semantic_kernel.memory.memory_record import MemoryRecord
-
-
-class EmbeddingIndexBase(ABC):
-    @abstractmethod
-    async def get_nearest_matches_async(
-        self,
-        collection: str,
-        embedding: ndarray,
-        limit: int,
-        min_relevance_score: float,
-    ) -> List[Tuple[MemoryRecord, float]]:
-        pass

semantic_kernel/ai/open_ai/__init__.py

@@ -1,27 +0,0 @@
-# Copyright (c) Microsoft. All rights reserved.
-
-from semantic_kernel.ai.open_ai.services.azure_chat_completion import (
-    AzureChatCompletion,
-)
-from semantic_kernel.ai.open_ai.services.azure_text_completion import (
-    AzureTextCompletion,
-)
-from semantic_kernel.ai.open_ai.services.azure_text_embedding import AzureTextEmbedding
-from semantic_kernel.ai.open_ai.services.open_ai_chat_completion import (
-    OpenAIChatCompletion,
-)
-from semantic_kernel.ai.open_ai.services.open_ai_text_completion import (
-    OpenAITextCompletion,
-)
-from semantic_kernel.ai.open_ai.services.open_ai_text_embedding import (
-    OpenAITextEmbedding,
-)
-
-__all__ = [
-    "OpenAITextCompletion",
-    "OpenAIChatCompletion",
-    "OpenAITextEmbedding",
-    "AzureTextCompletion",
-    "AzureChatCompletion",
-    "AzureTextEmbedding",
-]

semantic_kernel/memory/storage/data_entry.py

@@ -1,36 +0,0 @@
-# Copyright (c) Microsoft. All rights reserved.
-
-from datetime import datetime
-
-from semantic_kernel.memory.memory_record import MemoryRecord
-
-
-class DataEntry:
-    _key: str
-    _value: MemoryRecord
-    _timestamp: datetime
-
-    def __init__(self, key: str, value: MemoryRecord, timestamp: datetime) -> None:
-        self._key = key
-        self._value = value
-        self._timestamp = timestamp
-
-    @property
-    def key(self) -> str:
-        return self._key
-
-    @property
-    def value(self) -> MemoryRecord:
-        return self._value
-
-    @value.setter
-    def value(self, value: MemoryRecord) -> None:
-        self._value = value
-
-    @property
-    def timestamp(self) -> datetime:
-        return self._timestamp
-
-    @timestamp.setter
-    def timestamp(self, timestamp: datetime) -> None:
-        self._timestamp = timestamp

semantic_kernel/memory/storage/data_store_base.py

@@ -1,36 +0,0 @@
-# Copyright (c) Microsoft. All rights reserved.
-
-from abc import ABC, abstractmethod
-from typing import Any, List, Optional
-
-from semantic_kernel.memory.storage.data_entry import DataEntry
-
-
-class DataStoreBase(ABC):
-    @abstractmethod
-    async def get_collections_async(self) -> List[str]:
-        pass
-
-    @abstractmethod
-    async def get_all_async(self, collection: str) -> List[Any]:
-        pass
-
-    @abstractmethod
-    async def get_async(self, collection: str, key: str) -> Optional[DataEntry]:
-        pass
-
-    @abstractmethod
-    async def put_async(self, collection: str, value: Any) -> DataEntry:
-        pass
-
-    @abstractmethod
-    async def remove_async(self, collection: str, key: str) -> None:
-        pass
-
-    @abstractmethod
-    async def get_value_async(self, collection: str, key: str) -> Any:
-        pass
-
-    @abstractmethod
-    async def put_value_async(self, collection: str, key: str, value: Any) -> None:
-        pass

semantic_kernel/memory/storage/volatile_data_store.py

@@ -1,62 +0,0 @@
-# Copyright (c) Microsoft. All rights reserved.
-
-from datetime import datetime
-from typing import Dict, List, Optional
-
-from semantic_kernel.memory.memory_record import MemoryRecord
-from semantic_kernel.memory.storage.data_entry import DataEntry
-from semantic_kernel.memory.storage.data_store_base import DataStoreBase
-
-
-class VolatileDataStore(DataStoreBase):
-    _store: Dict[str, Dict[str, DataEntry]]
-
-    def __init__(self) -> None:
-        self._store = {}
-
-    async def get_collections_async(self) -> List[str]:
-        return list(self._store.keys())
-
-    async def get_all_async(self, collection: str) -> List[DataEntry]:
-        if collection not in self._store:
-            return []
-
-        return list(self._store[collection].values())
-
-    async def get_async(self, collection: str, key: str) -> Optional[DataEntry]:
-        if collection not in self._store:
-            return None
-        if key not in self._store[collection]:
-            return None
-
-        return self._store[collection][key]
-
-    async def put_async(self, collection: str, value: DataEntry) -> DataEntry:
-        if collection not in self._store:
-            self._store[collection] = {}
-
-        self._store[collection][value.key] = value
-
-        return value
-
-    async def remove_async(self, collection: str, key: str) -> None:
-        if collection not in self._store:
-            return
-        if key not in self._store[collection]:
-            return
-
-        del self._store[collection][key]
-
-    async def get_value_async(self, collection: str, key: str) -> MemoryRecord:
-        entry = await self.get_async(collection, key)
-
-        if entry is None:
-            # TODO: equivalent here?
-            raise Exception(f"Key '{key}' not found in collection '{collection}'")
-
-        return entry.value
-
-    async def put_value_async(
-        self, collection: str, key: str, value: MemoryRecord
-    ) -> None:
-        await self.put_async(collection, DataEntry(key, value, datetime.now()))