v0.3.2

api.md

@@ -1,8 +1,8 @@
-# Top Level
+# Anthropic
 
-Custom Methods:
+Methods:
 
-- `count_tokens`
+- <code>client.<a href="./src/anthropic/_client.py">count_tokens</a>(\*args) -> int</code>
 
 # Completions
 
@@ -15,4 +15,3 @@ from anthropic.types import Completion
 Methods:
 
 - <code title="post /v1/complete">client.completions.<a href="./src/anthropic/resources/completions.py">create</a>(\*\*<a href="src/anthropic/types/completion_create_params.py">params</a>) -> <a href="./src/anthropic/types/completion.py">Completion</a></code>
-- <code title="post /v1/complete">client.completions.<a href="./src/anthropic/resources/completions.py">create</a>(\*\*<a href="src/anthropic/types/completion_create_params.py">params</a>) -> <a href="./src/anthropic/types/completion.py">Completion</a></code>

pyproject.toml

@@ -13,7 +13,7 @@ packages = [
 [tool.poetry.dependencies]
 python = "^3.7"
 httpx = ">= 0.23.0"
-pydantic = ">= 1.9.0, < 2.0"
+pydantic = "^1.9.0"
 typing-extensions = ">= 4.1.1"
 anyio = ">= 3.5.0"
 distro = ">= 1.7.0"

tests/test_tokenizer.py

@@ -0,0 +1,76 @@
+import asyncio
+import threading
+import concurrent.futures
+from multiprocessing import Pool
+
+import pytest
+
+from anthropic import _tokenizers
+
+
+@pytest.fixture(autouse=True)
+def before_test() -> None:
+    # clear cache
+    _tokenizers._tokenizer = None
+
+
+def _sync_tokenizer_test() -> None:
+    tokenizer = _tokenizers.sync_get_tokenizer()
+    encoded_text = tokenizer.encode("hello world")  # type: ignore
+    assert len(encoded_text.ids) == 2  # type: ignore
+
+
+def test_threading() -> None:
+    failed = False
+
+    def target() -> None:
+        nonlocal failed
+
+        try:
+            _sync_tokenizer_test()
+        except Exception:
+            failed = True
+            raise
+
+    t1 = threading.Thread(target=target)
+    t2 = threading.Thread(target=target)
+
+    t1.start()
+    t2.start()
+    t1.join()
+    t2.join()
+
+    assert not failed
+
+
+def test_concurrent_futures() -> None:
+    with concurrent.futures.ThreadPoolExecutor() as executor:
+        future1 = executor.submit(_sync_tokenizer_test)
+        future2 = executor.submit(_sync_tokenizer_test)
+
+        future1.result()
+        future2.result()
+
+
+def test_multiprocessing() -> None:
+    with Pool(processes=10) as pool:
+        pool.apply(_sync_tokenizer_test)
+
+
+async def _async_tokenizer_test() -> None:
+    tokenizer = await _tokenizers.async_get_tokenizer()
+    encoded_text = tokenizer.encode("hello world")  # type: ignore
+    assert len(encoded_text.ids) == 2  # type: ignore
+
+
+async def test_asyncio_tasks() -> None:
+    loop = asyncio.get_event_loop()
+
+    task1 = loop.create_task(_async_tokenizer_test())
+    task2 = loop.create_task(_async_tokenizer_test())
+
+    await asyncio.gather(task1, task2)
+
+
+async def test_asyncio_gather() -> None:
+    await asyncio.gather(_async_tokenizer_test(), _async_tokenizer_test())