edsl 0.1.46__py3-none-any.whl → 0.1.48__py3-none-any.whl

edsl/language_models/key_management/KeyLookupCollection.py

@@ -1,38 +0,0 @@
-from collections import UserDict
-
-from edsl.language_models.key_management.KeyLookupBuilder import KeyLookupBuilder
-
-
-class KeyLookupCollection(UserDict):
-    """A singleton class that stores key-lookup objects.
-
-    This is because once a KeyLook is created once, we do not
-    need to keep re-creating it.
-
-    >>> collection = KeyLookupCollection()
-    >>> collection2 = KeyLookupCollection()
-    >>> collection is collection2  # Test singleton pattern
-    True
-    >>> collection.add_key_lookup(("config", "env"))
-    >>> ("config", "env") in collection.data
-    True
-    """
-
-    _instance = None
-
-    def __new__(cls, *args, **kwargs):
-        if cls._instance is None:
-            cls._instance = super().__new__(cls)
-        return cls._instance
-
-    def __init__(self, *args, **kwargs):
-        if not hasattr(self, "_initialized"):
-            self.data = {}
-            self._initialized = True
-            super().__init__(*args, **kwargs)
-
-    def add_key_lookup(self, fetch_order=None):
-        if fetch_order is None:
-            fetch_order = ("config", "env")
-        if fetch_order not in self.data:
-            self.data[fetch_order] = KeyLookupBuilder(fetch_order=fetch_order).build()

edsl/language_models/key_management/models.py

@@ -1,137 +0,0 @@
-from dataclasses import dataclass, asdict
-from typing import Optional
-
-
-@dataclass
-class APIKeyEntry:
-    """A class representing an API key entry.
-
-    >>> entry = APIKeyEntry.example()
-    >>> entry.service
-    'openai'
-    >>> entry.name
-    'OPENAI_API_KEY'
-    >>> entry.value
-    'sk-abcd1234'
-    >>> entry.source
-    'env'
-    """
-
-    service: str
-    name: str
-    value: str
-    source: Optional[str] = None
-
-    @classmethod
-    def example(cls):
-        return APIKeyEntry(
-            service="openai", name="OPENAI_API_KEY", value="sk-abcd1234", source="env"
-        )
-
-
-@dataclass
-class LimitEntry:
-    """A class representing rate limit entries for a service.
-
-    >>> limit = LimitEntry.example()
-    >>> limit.service
-    'openai'
-    >>> limit.rpm
-    60
-    >>> limit.tpm
-    100000
-    >>> limit.rpm_source
-    'config'
-    >>> limit.tpm_source
-    'env'
-    """
-
-    service: str
-    rpm: int
-    tpm: int
-    rpm_source: Optional[str] = None
-    tpm_source: Optional[str] = None
-
-    @classmethod
-    def example(cls):
-        return LimitEntry(
-            service="openai", rpm=60, tpm=100000, rpm_source="config", tpm_source="env"
-        )
-
-
-@dataclass
-class APIIDEntry:
-    """A class representing an API ID entry.
-
-    >>> id_entry = APIIDEntry.example()
-    >>> id_entry.service
-    'bedrock'
-    >>> id_entry.name
-    'AWS_ACCESS_KEY_ID'
-    >>> id_entry.value
-    'AKIA1234'
-    >>> id_entry.source
-    'env'
-    """
-
-    service: str
-    name: str
-    value: str
-    source: Optional[str] = None
-
-    @classmethod
-    def example(cls):
-        return APIIDEntry(
-            service="bedrock", name="AWS_ACCESS_KEY_ID", value="AKIA1234", source="env"
-        )
-
-
-@dataclass
-class LanguageModelInput:
-    """A class representing input configuration for a language model service.
-
-    >>> lm_input = LanguageModelInput.example()
-    >>> lm_input.api_token
-    'sk-abcd123'
-    >>> lm_input.rpm
-    60
-    >>> lm_input.tpm
-    100000
-    >>> lm_input.api_id
-
-
-    Test dictionary conversion:
-    >>> d = lm_input.to_dict()
-    >>> isinstance(d, dict)
-    True
-    >>> LanguageModelInput.from_dict(d).api_token == lm_input.api_token
-    True
-    """
-
-    api_token: str
-    rpm: int
-    tpm: int
-    api_id: Optional[str] = None
-    token_source: Optional[str] = None
-    rpm_source: Optional[str] = None
-    tpm_source: Optional[str] = None
-    id_source: Optional[str] = None
-
-    def to_dict(self):
-        return asdict(self)
-
-    @classmethod
-    def from_dict(cls, d):
-        return cls(**d)
-
-    @classmethod
-    def example(cls):
-        return LanguageModelInput(
-            api_token="sk-abcd123", tpm=100000, rpm=60, api_id=None
-        )
-
-
-if __name__ == "__main__":
-    import doctest
-
-    doctest.testmod()