Merge pull request #52 from Aleph-Alpha/support-Python3.7-google-colab

Remove dependency on Literal from typing to support Pyton 3.7

.github/workflows/integration.yml

@@ -5,13 +5,18 @@ on: push
 jobs:
   lint-and-test:
     runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        # We support latest 3.x version and 3.7 because
+        # Google Colab uses 3.7 by default.
+        python-version: [3.7, 3.x]
 
     steps:
       - uses: actions/checkout@v3
       - name: Set up Python
         uses: actions/setup-python@v3
         with:
-          python-version: "3.x"
+          python-version: ${{ matrix.python-version }}
           cache: "pip"
           cache-dependency-path: |
             **/setup.py

Changelog.md

@@ -1,5 +1,12 @@
 # Changelog
 
+## 2.4.1
+
+### Python support
+
+- Minimal supported Python version is now 3.7
+- Previously we only supported version 3.8
+
 ## 2.4.0
 
 ### New feature

aleph_alpha_client/aleph_alpha_client.py

@@ -127,8 +127,22 @@ def available_models(self):
         )
         return self._translate_errors(response).json()
 
+    def available_checkpoints(self):
+        """
+        Queries all checkpoints which are currently available.
+        """
+        response = self.get_request(
+            self.host + "checkpoints_available", headers=self.request_headers
+        )
+        return self._translate_errors(response).json()
+
     def tokenize(
-        self, model: Optional[str], prompt: str, tokens: bool = True, token_ids: bool = True, checkpoint: Optional[str] = None
+        self,
+        model: Optional[str],
+        prompt: str,
+        tokens: bool = True,
+        token_ids: bool = True,
+        checkpoint: Optional[str] = None,
     ):
         """
         Tokenizes the given prompt for the given model.
@@ -153,7 +167,12 @@ def tokenize(
         )
         return self._translate_errors(response).json()
 
-    def detokenize(self, model: Optional[str], token_ids: List[int], checkpoint: Optional[str] = None):
+    def detokenize(
+        self,
+        model: Optional[str],
+        token_ids: List[int],
+        checkpoint: Optional[str] = None,
+    ):
         """
         Detokenizes the given tokens.
         """

aleph_alpha_client/embedding.py

@@ -3,7 +3,6 @@
     Any,
     Dict,
     List,
-    Literal,
     NamedTuple,
     Optional,
     Tuple,
@@ -113,7 +112,7 @@ class SemanticEmbeddingRequest(NamedTuple):
 
     prompt: Prompt
     representation: SemanticRepresentation
-    compress_to_size: Optional[Literal[128]] = None
+    compress_to_size: Optional[int] = None
 
 
 class SemanticEmbeddingResponse(NamedTuple):

aleph_alpha_client/version.py

@@ -1 +1 @@
-__version__ = "2.4.0"
+__version__ = "2.4.1"

tests/test_qa.py

@@ -80,9 +80,10 @@ def test_qa_with_client(client: AlephAlphaClient):
 def test_qa_with_client_against_checkpoint(
     client: AlephAlphaClient, qa_checkpoint_name: str
 ):
-    model_name = "luminous-extended"
     # given a client
-    assert model_name in map(lambda model: model["name"], client.available_models())
+    assert qa_checkpoint_name in map(
+        lambda checkpoint: checkpoint["name"], client.available_checkpoints()
+    )
 
     # when posting a QA request with explicit parameters
     response = client.qa(