thinking-machines-lab/tinker
1
0
Fork 0
mirror of https://github.com/thinking-machines-lab/tinker.git synced 2026-05-01 17:45:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Sync contents

Browse source
This commit is contained in:
Daniel Xu 2025-10-15 16:37:21 +00:00
parent 9ba155a34d
commit 422bc7b9f1
122 changed files with 516 additions and 477 deletions
Download patch file Download diff file Expand all files Collapse all files

1
tests/api_resources/__init__.py
View file

@ -1 +0,0 @@
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.

6
tests/api_resources/test_futures.py
View file

@ -1,5 +1,3 @@
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
from __future__ import annotations
import os
@ -42,7 +40,6 @@ class TestFutures:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
future = response.parse()
assert_matches_type(FutureRetrieveResponse, future, path=["response"])
@ -53,7 +50,6 @@ class TestFutures:
request_id="123e4567-e89b-12d3-a456-426614174000",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
future = response.parse()
assert_matches_type(FutureRetrieveResponse, future, path=["response"])
@ -91,7 +87,6 @@ class TestAsyncFutures:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
future = await response.parse()
assert_matches_type(FutureRetrieveResponse, future, path=["response"])
@ -102,7 +97,6 @@ class TestAsyncFutures:
request_id="123e4567-e89b-12d3-a456-426614174000",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
future = await response.parse()
assert_matches_type(FutureRetrieveResponse, future, path=["response"])

20
tests/api_resources/test_models.py
View file

@ -1,5 +1,3 @@
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
from __future__ import annotations
import os
@ -49,7 +47,6 @@ class TestModels:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
model = response.parse()
assert_matches_type(UntypedAPIFuture, model, path=["response"])
@ -60,7 +57,7 @@ class TestModels:
base_model="meta-llama/Llama-3.2-1B",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
model = response.parse()
assert_matches_type(UntypedAPIFuture, model, path=["response"])
@ -92,7 +89,6 @@ class TestModels:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
model = response.parse()
assert_matches_type(GetInfoResponse, model, path=["response"])
@ -103,7 +99,7 @@ class TestModels:
model_id="123e4567-e89b-12d3-a456-426614174000",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
model = response.parse()
assert_matches_type(GetInfoResponse, model, path=["response"])
@ -135,7 +131,6 @@ class TestModels:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
model = response.parse()
assert_matches_type(UntypedAPIFuture, model, path=["response"])
@ -146,7 +141,7 @@ class TestModels:
model_id="123e4567-e89b-12d3-a456-426614174000",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
model = response.parse()
assert_matches_type(UntypedAPIFuture, model, path=["response"])
@ -188,7 +183,6 @@ class TestAsyncModels:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
model = await response.parse()
assert_matches_type(UntypedAPIFuture, model, path=["response"])
@ -199,7 +193,7 @@ class TestAsyncModels:
base_model="meta-llama/Llama-3.2-1B",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
model = await response.parse()
assert_matches_type(UntypedAPIFuture, model, path=["response"])
@ -231,7 +225,6 @@ class TestAsyncModels:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
model = await response.parse()
assert_matches_type(GetInfoResponse, model, path=["response"])
@ -242,7 +235,7 @@ class TestAsyncModels:
model_id="123e4567-e89b-12d3-a456-426614174000",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
model = await response.parse()
assert_matches_type(GetInfoResponse, model, path=["response"])
@ -274,7 +267,6 @@ class TestAsyncModels:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
model = await response.parse()
assert_matches_type(UntypedAPIFuture, model, path=["response"])
@ -285,7 +277,7 @@ class TestAsyncModels:
model_id="123e4567-e89b-12d3-a456-426614174000",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
model = await response.parse()
assert_matches_type(UntypedAPIFuture, model, path=["response"])

10
tests/api_resources/test_sampling.py
View file

@ -1,5 +1,3 @@
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
from __future__ import annotations
import os
@ -78,7 +76,6 @@ class TestSampling:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
sampling = response.parse()
assert_matches_type(UntypedAPIFuture, sampling, path=["response"])
@ -97,7 +94,6 @@ class TestSampling:
sampling_params={},
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
sampling = response.parse()
assert_matches_type(UntypedAPIFuture, sampling, path=["response"])
@ -164,7 +160,6 @@ class TestSampling:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
sampling = response.parse()
assert_matches_type(SampleResponse, sampling, path=["response"])
@ -183,7 +178,6 @@ class TestSampling:
sampling_params={},
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
sampling = response.parse()
assert_matches_type(SampleResponse, sampling, path=["response"])
@ -256,7 +250,6 @@ class TestAsyncSampling:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
sampling = await response.parse()
assert_matches_type(UntypedAPIFuture, sampling, path=["response"])
@ -275,7 +268,6 @@ class TestAsyncSampling:
sampling_params={},
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
sampling = await response.parse()
assert_matches_type(UntypedAPIFuture, sampling, path=["response"])
@ -342,7 +334,6 @@ class TestAsyncSampling:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
sampling = await response.parse()
assert_matches_type(SampleResponse, sampling, path=["response"])
@ -361,7 +352,6 @@ class TestAsyncSampling:
sampling_params={},
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
sampling = await response.parse()
assert_matches_type(SampleResponse, sampling, path=["response"])

10
tests/api_resources/test_service.py
View file

@ -1,5 +1,3 @@
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
from __future__ import annotations
import os
@ -29,7 +27,6 @@ class TestService:
response = client.service.with_raw_response.get_server_capabilities()
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
service = response.parse()
assert_matches_type(GetServerCapabilitiesResponse, service, path=["response"])
@ -38,7 +35,6 @@ class TestService:
def test_streaming_response_get_server_capabilities(self, client: Tinker) -> None:
with client.service.with_streaming_response.get_server_capabilities() as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
service = response.parse()
assert_matches_type(GetServerCapabilitiesResponse, service, path=["response"])
@ -57,7 +53,6 @@ class TestService:
response = client.service.with_raw_response.health_check()
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
service = response.parse()
assert_matches_type(HealthResponse, service, path=["response"])
@ -66,7 +61,6 @@ class TestService:
def test_streaming_response_health_check(self, client: Tinker) -> None:
with client.service.with_streaming_response.health_check() as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
service = response.parse()
assert_matches_type(HealthResponse, service, path=["response"])
@ -91,7 +85,6 @@ class TestAsyncService:
response = await async_client.service.with_raw_response.get_server_capabilities()
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
service = await response.parse()
assert_matches_type(GetServerCapabilitiesResponse, service, path=["response"])
@ -100,7 +93,6 @@ class TestAsyncService:
async def test_streaming_response_get_server_capabilities(self, async_client: AsyncTinker) -> None:
async with async_client.service.with_streaming_response.get_server_capabilities() as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
service = await response.parse()
assert_matches_type(GetServerCapabilitiesResponse, service, path=["response"])
@ -119,7 +111,6 @@ class TestAsyncService:
response = await async_client.service.with_raw_response.health_check()
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
service = await response.parse()
assert_matches_type(HealthResponse, service, path=["response"])
@ -128,7 +119,6 @@ class TestAsyncService:
async def test_streaming_response_health_check(self, async_client: AsyncTinker) -> None:
async with async_client.service.with_streaming_response.health_check() as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
service = await response.parse()
assert_matches_type(HealthResponse, service, path=["response"])

6
tests/api_resources/test_telemetry.py
View file

@ -1,5 +1,3 @@
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
from __future__ import annotations
import os
@ -56,7 +54,6 @@ class TestTelemetry:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
telemetry = response.parse()
assert_matches_type(TelemetryResponse, telemetry, path=["response"])
@ -78,7 +75,6 @@ class TestTelemetry:
session_id="182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
telemetry = response.parse()
assert_matches_type(TelemetryResponse, telemetry, path=["response"])
@ -129,7 +125,6 @@ class TestAsyncTelemetry:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
telemetry = await response.parse()
assert_matches_type(TelemetryResponse, telemetry, path=["response"])
@ -151,7 +146,6 @@ class TestAsyncTelemetry:
session_id="182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
telemetry = await response.parse()
assert_matches_type(TelemetryResponse, telemetry, path=["response"])

22
tests/api_resources/test_training.py
View file

@ -1,5 +1,3 @@
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
from __future__ import annotations
import os
@ -74,7 +72,6 @@ class TestTraining:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])
@ -85,7 +82,6 @@ class TestTraining:
base_model="meta-llama/Llama-3.2-1B",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])
@ -131,7 +127,6 @@ class TestTraining:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])
@ -167,7 +162,6 @@ class TestTraining:
model_id="123e4567-e89b-12d3-a456-426614174000",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])
@ -240,7 +234,6 @@ class TestTraining:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])
@ -276,7 +269,6 @@ class TestTraining:
model_id="123e4567-e89b-12d3-a456-426614174000",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])
@ -316,7 +308,6 @@ class TestTraining:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])
@ -328,7 +319,6 @@ class TestTraining:
model_id="123e4567-e89b-12d3-a456-426614174000",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])
@ -360,7 +350,6 @@ class TestTraining:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])
@ -371,7 +360,6 @@ class TestTraining:
model_id="123e4567-e89b-12d3-a456-426614174000",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])
@ -441,7 +429,6 @@ class TestAsyncTraining:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = await response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])
@ -452,7 +439,6 @@ class TestAsyncTraining:
base_model="meta-llama/Llama-3.2-1B",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = await response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])
@ -498,7 +484,6 @@ class TestAsyncTraining:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = await response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])
@ -534,7 +519,6 @@ class TestAsyncTraining:
model_id="123e4567-e89b-12d3-a456-426614174000",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = await response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])
@ -607,7 +591,6 @@ class TestAsyncTraining:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = await response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])
@ -643,7 +626,6 @@ class TestAsyncTraining:
model_id="123e4567-e89b-12d3-a456-426614174000",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = await response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])
@ -683,7 +665,6 @@ class TestAsyncTraining:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = await response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])
@ -695,7 +676,6 @@ class TestAsyncTraining:
model_id="123e4567-e89b-12d3-a456-426614174000",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = await response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])
@ -727,7 +707,6 @@ class TestAsyncTraining:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = await response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])
@ -738,7 +717,6 @@ class TestAsyncTraining:
model_id="123e4567-e89b-12d3-a456-426614174000",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
training = await response.parse()
assert_matches_type(UntypedAPIFuture, training, path=["response"])

14
tests/api_resources/test_weights.py
View file

@ -1,5 +1,3 @@
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
from __future__ import annotations
import os
@ -45,7 +43,6 @@ class TestWeights:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
weight = response.parse()
assert_matches_type(UntypedAPIFuture, weight, path=["response"])
@ -57,7 +54,6 @@ class TestWeights:
path="tinker://model-id/weights/step-0123",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
weight = response.parse()
assert_matches_type(UntypedAPIFuture, weight, path=["response"])
@ -90,7 +86,6 @@ class TestWeights:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
weight = response.parse()
assert_matches_type(UntypedAPIFuture, weight, path=["response"])
@ -101,7 +96,6 @@ class TestWeights:
model_id="123e4567-e89b-12d3-a456-426614174000",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
weight = response.parse()
assert_matches_type(UntypedAPIFuture, weight, path=["response"])
@ -134,7 +128,6 @@ class TestWeights:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
weight = response.parse()
assert_matches_type(UntypedAPIFuture, weight, path=["response"])
@ -145,7 +138,6 @@ class TestWeights:
model_id="123e4567-e89b-12d3-a456-426614174000",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
weight = response.parse()
assert_matches_type(UntypedAPIFuture, weight, path=["response"])
@ -186,7 +178,6 @@ class TestAsyncWeights:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
weight = await response.parse()
assert_matches_type(UntypedAPIFuture, weight, path=["response"])
@ -198,7 +189,6 @@ class TestAsyncWeights:
path="tinker://model-id/weights/step-0123",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
weight = await response.parse()
assert_matches_type(UntypedAPIFuture, weight, path=["response"])
@ -231,7 +221,6 @@ class TestAsyncWeights:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
weight = await response.parse()
assert_matches_type(UntypedAPIFuture, weight, path=["response"])
@ -242,7 +231,6 @@ class TestAsyncWeights:
model_id="123e4567-e89b-12d3-a456-426614174000",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
weight = await response.parse()
assert_matches_type(UntypedAPIFuture, weight, path=["response"])
@ -275,7 +263,6 @@ class TestAsyncWeights:
)
assert response.is_closed is True
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
weight = await response.parse()
assert_matches_type(UntypedAPIFuture, weight, path=["response"])
@ -286,7 +273,6 @@ class TestAsyncWeights:
model_id="123e4567-e89b-12d3-a456-426614174000",
) as response:
assert not response.is_closed
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
weight = await response.parse()
assert_matches_type(UntypedAPIFuture, weight, path=["response"])