lock

langchain-ai · Jan 26, 2024 · 9c4a222 · 9c4a222
1 parent 5e84aa7
commit 9c4a222
Show file tree

Hide file tree

Showing 5 changed files with 119 additions and 18 deletions.
diff --git a/.github/workflows/python_test.yml b/.github/workflows/python_test.yml
@@ -52,7 +52,7 @@ jobs:
       - name: Install poetry
         run: pipx install poetry==$POETRY_VERSION
       - name: Install dependencies
-        run: poetry install
+        run: poetry install --with dev,lint
       - name: Analysing the code with our lint
         run: make lint
 

diff --git a/python/langsmith/wrappers/_openai.py b/python/langsmith/wrappers/_openai.py
@@ -84,7 +84,7 @@ def patch_openai(client: C) -> C:
     client.chat.completions.create = _get_wrapper(  # type: ignore[method-assign]
         client.chat.completions.create, "ChatOpenAI", _reduce_chat
     )
-    client.completions.create = _get_wrapper(
+    client.completions.create = _get_wrapper(  # type: ignore[method-assign]
         client.completions.create, "OpenAI", _reduce_completions
-    )  # type: ignore[method-assign]
+    )
     return client
diff --git a/python/poetry.lock b/python/poetry.lock
diff --git a/python/pyproject.toml b/python/pyproject.toml
@@ -47,6 +47,9 @@ freezegun = "^1.2.2"
 pytest-subtests = "^0.11.0"
 pytest-watcher = "^0.3.4"
 
+[tool.poetry.group.lint.dependencies]
+openai = "^1.10"
+
 
 [build-system]
 requires = ["poetry-core"]

diff --git a/python/tests/integration_tests/wrappers/test_openai.py b/python/tests/integration_tests/wrappers/test_openai.py
@@ -1,4 +1,4 @@
-# mypy: disable-error-code="attr-defined, union-attr, arg-type"
+# mypy: disable-error-code="attr-defined, union-attr, arg-type, call-overload"
 import time
 from unittest import mock