feat: Models setup, mypy type check

.github/workflows/ci.yaml

@@ -0,0 +1,25 @@
+name: CI
+
+on:
+  push:
+
+jobs:
+  mypy:
+    name: mypy
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - name: Install poetry
+        run: pipx install poetry
+      - uses: actions/setup-python@v5
+        with:
+          python-version: '3.12'
+          cache: 'poetry'
+      - name: Install dependencies
+        run: poetry install
+      - uses: tsuyoshicho/action-mypy@v4
+        with:
+          github_token: ${{ secrets.GITHUB_TOKEN }}
+          reporter: github-check
+          target: libertai_agents
+          execute_command: 'poetry run mypy'

libertai_agents/models.py

@@ -0,0 +1,11 @@
+from transformers import AutoTokenizer, PreTrainedTokenizerFast
+
+
+class Model:
+    tokenizer: PreTrainedTokenizerFast
+
+    def __init__(self, model_id: str):
+        self.tokenizer = AutoTokenizer.from_pretrained(model_id)
+
+
+Hermes2Pro = Model(model_id="NousResearch/Hermes-2-Pro-Llama-3-8B")

mypy.ini

@@ -0,0 +1,4 @@
+[mypy]
+
+[mypy-transformers.*]
+ignore_missing_imports = True