Add CI

.github/workflows/pyright.yml

@@ -0,0 +1,47 @@
+name: Pyright
+
+on:
+  push:
+  pull_request:
+    branches: [ main ]
+
+env:
+  WORKING_DIRECTORY: "."
+  PYRIGHT_OUTPUT_FILENAME: "pyright.log"
+
+jobs:
+Pyright:
+  runs-on: ${{ matrix.os }}
+  strategy:
+    matrix:
+      os: [ubuntu-latest, macos-latest]
+  steps:
+    - name: Checkout code
+      uses: actions/checkout@v3
+
+    - name: Set up Python
+      uses: actions/setup-python@v4
+      with:
+        python-version: ${{ matrix.python-version }}
+
+    - name: Cache Poetry virtualenv
+      uses: actions/cache@v2
+      with:
+        path: ~/.cache/pypoetry/virtualenvs
+        key: ${{ runner.os }}-poetry-${{ hashFiles('**/poetry.lock') }}
+        restore-keys: |
+          ${{ runner.os }}-poetry-
+
+    - name: Install Poetry
+      uses: snok/install-poetry@v1.3.1
+
+    - name: Run Static Type Checking with Pyright
+      run: |
+        set -e -o pipefail
+        poetry run pyright > ${{ env.WORKING_DIRECTORY }}/${{ env.PYRIGHT_OUTPUT_FILENAME }}
+
+    - name: Upload Artifacts
+      uses: actions/upload-artifact@v3
+      with:
+        name: pyright-log
+        path: ${{ env.WORKING_DIRECTORY }}/${{ env.PYRIGHT_OUTPUT_FILENAME }}

instructor/cli/usage.py

@@ -46,13 +46,7 @@ async def get_usage_for_past_n_days(n_days: int) -> list[dict[str, Any]]:
 
 
 # Define the cost per unit for each model
-# Add temporary body type hint here because mypy may infer the dict type
-# from the first few items (?) in the dict, which may not be representative of
-# the entire dict.
-MODEL_COSTS: dict[
-    ModelNames,
-    Union[dict[str, float], float],
-] = {
+MODEL_COSTS = {
     "gpt-4-0125-preview": {"prompt": 0.01 / 1000, "completion": 0.03 / 1000},
     "gpt-4-turbo-preview": {"prompt": 0.01 / 1000, "completion": 0.03 / 1000},
     "gpt-4-1106-preview": {"prompt": 0.01 / 1000, "completion": 0.03 / 1000},
@@ -114,11 +108,11 @@ def calculate_cost(
 
 def group_and_sum_by_date_and_snapshot(usage_data: list[dict[str, Any]]) -> Table:
     """Group and sum the usage data by date and snapshot, including costs."""
-    summary: defaultdict[
-        str, defaultdict[str, dict[str, Union[int, float]]]
-    ] = defaultdict(
-        lambda: defaultdict(
-            lambda: {"total_requests": 0, "total_tokens": 0, "total_cost": 0.0}
+    summary: defaultdict[str, defaultdict[str, dict[str, Union[int, float]]]] = (
+        defaultdict(
+            lambda: defaultdict(
+                lambda: {"total_requests": 0, "total_tokens": 0, "total_cost": 0.0}
+            )
         )
     )
 

instructor/client.py

@@ -346,7 +346,7 @@ async def create_with_completion(
             strict=strict,
             **kwargs,
         )
-        return response, response._raw_response  # type: ignore
+        return response, response._raw_response
 
 
 @overload