scale-gp-beta 0.1.0a28__tar.gz → 0.1.0a30__tar.gz

{scale_gp_beta-0.1.0a28 → scale_gp_beta-0.1.0a30}/.gitignore

@@ -1,5 +1,4 @@
 .prism.log
-.vscode
 _dev
 
 __pycache__

scale_gp_beta-0.1.0a30/.release-please-manifest.json

@@ -0,0 +1,3 @@
+{
+  ".": "0.1.0-alpha.30"
+}

{scale_gp_beta-0.1.0a28 → scale_gp_beta-0.1.0a30}/CHANGELOG.md

@@ -1,5 +1,64 @@
 # Changelog
 
+## 0.1.0-alpha.30 (2025-08-26)
+
+Full Changelog: [v0.1.0-alpha.29...v0.1.0-alpha.30](https://github.com/scaleapi/sgp-python-beta/compare/v0.1.0-alpha.29...v0.1.0-alpha.30)
+
+### Features
+
+* **api:** api update ([cb4c1c5](https://github.com/scaleapi/sgp-python-beta/commit/cb4c1c5edb6a15c7e3e998d54c83604f9c6bb3a1))
+* **api:** api update ([32e9415](https://github.com/scaleapi/sgp-python-beta/commit/32e9415aa3fc23ebdae39d438cabf8c6a2496b49))
+* **api:** api update ([8146bf3](https://github.com/scaleapi/sgp-python-beta/commit/8146bf3e9e9c65defb95d3a80525538cb05fa825))
+* **api:** api update ([b112792](https://github.com/scaleapi/sgp-python-beta/commit/b1127924897e97dbf82f342b53f926407cf84591))
+* **api:** api update ([674f2b1](https://github.com/scaleapi/sgp-python-beta/commit/674f2b1c7289ebf7ca41be8b91913df2da04835e))
+* **api:** api update ([48c91fa](https://github.com/scaleapi/sgp-python-beta/commit/48c91fa1edf4b05196ad396e092d26e102a31ae2))
+* **api:** api update ([680a7f9](https://github.com/scaleapi/sgp-python-beta/commit/680a7f9fff37bed71a05bed10928ef0ceae19079))
+* **api:** api update ([d44ffe4](https://github.com/scaleapi/sgp-python-beta/commit/d44ffe4df3e02a55abb1d73abca68e2978a95f07))
+* **api:** api update ([8ce7997](https://github.com/scaleapi/sgp-python-beta/commit/8ce79973d97ec3f4e362301979fd055a4021e046))
+* **api:** api update ([fddd794](https://github.com/scaleapi/sgp-python-beta/commit/fddd7945fa11de496a252c30f950e717dbde2a71))
+* **api:** api update ([491142b](https://github.com/scaleapi/sgp-python-beta/commit/491142b453f9f37b0907982e2a35819b05c79aba))
+* **api:** api update ([4a8519e](https://github.com/scaleapi/sgp-python-beta/commit/4a8519eacadbdf4e7e5094b61671ec5282330772))
+* **api:** api update ([c8bf560](https://github.com/scaleapi/sgp-python-beta/commit/c8bf5604617ddf3b23e35d3fa63f8992e27cc677))
+* **api:** manual updates ([b86c069](https://github.com/scaleapi/sgp-python-beta/commit/b86c0694240b1d7c1925d1ec7ef48a230939a795))
+* **api:** manual updates ([8efcc54](https://github.com/scaleapi/sgp-python-beta/commit/8efcc5412b8f912c8b7fc94eaf72ee6f2de8f9e3))
+* **api:** manual updates ([d8144e7](https://github.com/scaleapi/sgp-python-beta/commit/d8144e7b869b9ccb10e1c23997d3e28bddc5d5df))
+* clean up environment call outs ([8aa4d07](https://github.com/scaleapi/sgp-python-beta/commit/8aa4d075fcebffcf02fc539525c6af434291e823))
+* **client:** support file upload requests ([76d95ab](https://github.com/scaleapi/sgp-python-beta/commit/76d95abb89870b14e6303de8ef52e764bbe4403a))
+* SGP client hooks ([#128](https://github.com/scaleapi/sgp-python-beta/issues/128)) ([f247adb](https://github.com/scaleapi/sgp-python-beta/commit/f247adb2273f081c2d2083cc256e43c08b83c21a))
+
+
+### Bug Fixes
+
+* **parsing:** ignore empty metadata ([a1190e2](https://github.com/scaleapi/sgp-python-beta/commit/a1190e280c32aaac4faad6d496e012eedfeb9490))
+* **parsing:** parse extra field types ([6b64a45](https://github.com/scaleapi/sgp-python-beta/commit/6b64a450d2a0b70ea857b58ccbc3540f1ad7713b))
+
+
+### Chores
+
+* Added CONTRIBUTING.MD for lib ([#127](https://github.com/scaleapi/sgp-python-beta/issues/127)) ([2b1bda7](https://github.com/scaleapi/sgp-python-beta/commit/2b1bda7ab925c4c8c44976f72f9f5723a9e61f30))
+* **internal:** change ci workflow machines ([d7ace41](https://github.com/scaleapi/sgp-python-beta/commit/d7ace4109f8a7f2a98dd8f2cc526fee2b40f85b5))
+* **internal:** fix ruff target version ([057e041](https://github.com/scaleapi/sgp-python-beta/commit/057e041dc7ec6638e68ba66868bbd851aeaba3b8))
+* **internal:** update comment in script ([4760add](https://github.com/scaleapi/sgp-python-beta/commit/4760add2d22f15b47d5d8b4153dc7744523a0c63))
+* **project:** add settings file for vscode ([ae9740b](https://github.com/scaleapi/sgp-python-beta/commit/ae9740bd28d7aaa21a089cd328554caea643a538))
+* **types:** rebuild Pydantic models after all types are defined ([a1d6eec](https://github.com/scaleapi/sgp-python-beta/commit/a1d6eece8eaae0e783eda5aad65dceb62249cdad))
+* update @stainless-api/prism-cli to v5.15.0 ([a15a320](https://github.com/scaleapi/sgp-python-beta/commit/a15a320bf13cdc88ffc312253655ad97dfa517a9))
+* update github action ([f78e671](https://github.com/scaleapi/sgp-python-beta/commit/f78e67191edd058555d76a45f5b08d4bf974a2c4))
+
+## 0.1.0-alpha.29 (2025-07-14)
+
+Full Changelog: [v0.1.0-alpha.28...v0.1.0-alpha.29](https://github.com/scaleapi/sgp-python-beta/compare/v0.1.0-alpha.28...v0.1.0-alpha.29)
+
+### Features
+
+* **api:** Add application_operation_type ([2455e20](https://github.com/scaleapi/sgp-python-beta/commit/2455e20838b22637ddce2a8e523f113d43acab0e))
+* **api:** Add span status to span models ([c3fb91c](https://github.com/scaleapi/sgp-python-beta/commit/c3fb91c773d6ef11e1cc02e4546ff1923fc25c82))
+* **api:** change name of app op type to span_type ([2285f10](https://github.com/scaleapi/sgp-python-beta/commit/2285f10ca3f1d51d3733996724d24daa09e83b88))
+
+
+### Bug Fixes
+
+* Use stainless types auto generated types for Span status and span type. ([#125](https://github.com/scaleapi/sgp-python-beta/issues/125)) ([6a3b070](https://github.com/scaleapi/sgp-python-beta/commit/6a3b07040137cb283711399bad76bedf78affcaf))
+
 ## 0.1.0-alpha.28 (2025-07-11)
 
 Full Changelog: [v0.1.0-alpha.27...v0.1.0-alpha.28](https://github.com/scaleapi/sgp-python-beta/compare/v0.1.0-alpha.27...v0.1.0-alpha.28)

{scale_gp_beta-0.1.0a28 → scale_gp_beta-0.1.0a30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: scale-gp-beta
-Version: 0.1.0a28
+Version: 0.1.0a30
 Summary: The official Python library for the Scale GP API
 Project-URL: Homepage, https://github.com/scaleapi/sgp-python-beta
 Project-URL: Repository, https://github.com/scaleapi/sgp-python-beta
@@ -364,7 +364,6 @@ pip install --pre scale-gp-beta[aiohttp]
 Then you can enable it by instantiating the client with `http_client=DefaultAioHttpClient()`:
 
 ```python
-import os
 import asyncio
 from scale_gp_beta import DefaultAioHttpClient
 from scale_gp_beta import AsyncSGPClient
@@ -373,7 +372,7 @@ from scale_gp_beta import AsyncSGPClient
 async def main() -> None:
     async with AsyncSGPClient(
         account_id="My Account ID",
-        api_key=os.environ.get("SGP_API_KEY"),  # This is the default and can be omitted
+        api_key="My API Key",
         http_client=DefaultAioHttpClient(),
     ) as client:
         completion = await client.chat.completions.create(

{scale_gp_beta-0.1.0a28 → scale_gp_beta-0.1.0a30}/README.md

@@ -329,7 +329,6 @@ pip install --pre scale-gp-beta[aiohttp]
 Then you can enable it by instantiating the client with `http_client=DefaultAioHttpClient()`:
 
 ```python
-import os
 import asyncio
 from scale_gp_beta import DefaultAioHttpClient
 from scale_gp_beta import AsyncSGPClient
@@ -338,7 +337,7 @@ from scale_gp_beta import AsyncSGPClient
 async def main() -> None:
     async with AsyncSGPClient(
         account_id="My Account ID",
-        api_key=os.environ.get("SGP_API_KEY"),  # This is the default and can be omitted
+        api_key="My API Key",
         http_client=DefaultAioHttpClient(),
     ) as client:
         completion = await client.chat.completions.create(

{scale_gp_beta-0.1.0a28 → scale_gp_beta-0.1.0a30}/api.md

@@ -1,3 +1,21 @@
+# Shared Types
+
+```python
+from scale_gp_beta.types import Identity
+```
+
+# Responses
+
+Types:
+
+```python
+from scale_gp_beta.types import Response, ResponseCreateResponse
+```
+
+Methods:
+
+- <code title="post /v5/responses">client.responses.<a href="./src/scale_gp_beta/resources/responses.py">create</a>(\*\*<a href="src/scale_gp_beta/types/response_create_params.py">params</a>) -> <a href="./src/scale_gp_beta/types/response_create_response.py">ResponseCreateResponse</a></code>
+
 # Completions
 
 Types:
@@ -171,7 +189,13 @@ Methods:
 Types:
 
 ```python
-from scale_gp_beta.types import Span, SpanBatchResponse, SpanUpsertBatchResponse
+from scale_gp_beta.types import (
+    Span,
+    SpanStatus,
+    SpanType,
+    SpanBatchResponse,
+    SpanUpsertBatchResponse,
+)
 ```
 
 Methods:

{scale_gp_beta-0.1.0a28 → scale_gp_beta-0.1.0a30}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "scale-gp-beta"
-version = "0.1.0-alpha.28"
+version = "0.1.0-alpha.30"
 description = "The official Python library for the Scale GP API"
 dynamic = ["readme"]
 license = "Apache-2.0"
@@ -159,7 +159,7 @@ reportPrivateUsage = false
 [tool.ruff]
 line-length = 120
 output-format = "grouped"
-target-version = "py37"
+target-version = "py38"
 
 [tool.ruff.format]
 docstring-code-format = true

{scale_gp_beta-0.1.0a28 → scale_gp_beta-0.1.0a30}/src/scale_gp_beta/_base_client.py

@@ -532,7 +532,10 @@ class BaseClient(Generic[_HttpxClientT, _DefaultStreamT]):
         is_body_allowed = options.method.lower() != "get"
 
         if is_body_allowed:
-            kwargs["json"] = json_data if is_given(json_data) else None
+            if isinstance(json_data, bytes):
+                kwargs["content"] = json_data
+            else:
+                kwargs["json"] = json_data if is_given(json_data) else None
             kwargs["files"] = files
         else:
             headers.pop("Content-Type", None)

{scale_gp_beta-0.1.0a28 → scale_gp_beta-0.1.0a30}/src/scale_gp_beta/_client.py

@@ -27,6 +27,7 @@ from .resources import (
     datasets,
     inference,
     questions,
+    responses,
     completions,
     evaluations,
     dataset_items,
@@ -61,6 +62,7 @@ ENVIRONMENTS: Dict[str, str] = {
 
 
 class SGPClient(SyncAPIClient):
+    responses: responses.ResponsesResource
     completions: completions.CompletionsResource
     chat: chat.ChatResource
     inference: inference.InferenceResource
@@ -165,6 +167,7 @@ class SGPClient(SyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
+        self.responses = responses.ResponsesResource(self)
         self.completions = completions.CompletionsResource(self)
         self.chat = chat.ChatResource(self)
         self.inference = inference.InferenceResource(self)
@@ -290,6 +293,7 @@ class SGPClient(SyncAPIClient):
 
 
 class AsyncSGPClient(AsyncAPIClient):
+    responses: responses.AsyncResponsesResource
     completions: completions.AsyncCompletionsResource
     chat: chat.AsyncChatResource
     inference: inference.AsyncInferenceResource
@@ -394,6 +398,7 @@ class AsyncSGPClient(AsyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
+        self.responses = responses.AsyncResponsesResource(self)
         self.completions = completions.AsyncCompletionsResource(self)
         self.chat = chat.AsyncChatResource(self)
         self.inference = inference.AsyncInferenceResource(self)
@@ -520,6 +525,7 @@ class AsyncSGPClient(AsyncAPIClient):
 
 class SGPClientWithRawResponse:
     def __init__(self, client: SGPClient) -> None:
+        self.responses = responses.ResponsesResourceWithRawResponse(client.responses)
         self.completions = completions.CompletionsResourceWithRawResponse(client.completions)
         self.chat = chat.ChatResourceWithRawResponse(client.chat)
         self.inference = inference.InferenceResourceWithRawResponse(client.inference)
@@ -535,6 +541,7 @@ class SGPClientWithRawResponse:
 
 class AsyncSGPClientWithRawResponse:
     def __init__(self, client: AsyncSGPClient) -> None:
+        self.responses = responses.AsyncResponsesResourceWithRawResponse(client.responses)
         self.completions = completions.AsyncCompletionsResourceWithRawResponse(client.completions)
         self.chat = chat.AsyncChatResourceWithRawResponse(client.chat)
         self.inference = inference.AsyncInferenceResourceWithRawResponse(client.inference)
@@ -550,6 +557,7 @@ class AsyncSGPClientWithRawResponse:
 
 class SGPClientWithStreamedResponse:
     def __init__(self, client: SGPClient) -> None:
+        self.responses = responses.ResponsesResourceWithStreamingResponse(client.responses)
         self.completions = completions.CompletionsResourceWithStreamingResponse(client.completions)
         self.chat = chat.ChatResourceWithStreamingResponse(client.chat)
         self.inference = inference.InferenceResourceWithStreamingResponse(client.inference)
@@ -565,6 +573,7 @@ class SGPClientWithStreamedResponse:
 
 class AsyncSGPClientWithStreamedResponse:
     def __init__(self, client: AsyncSGPClient) -> None:
+        self.responses = responses.AsyncResponsesResourceWithStreamingResponse(client.responses)
         self.completions = completions.AsyncCompletionsResourceWithStreamingResponse(client.completions)
         self.chat = chat.AsyncChatResourceWithStreamingResponse(client.chat)
         self.inference = inference.AsyncInferenceResourceWithStreamingResponse(client.inference)

{scale_gp_beta-0.1.0a28 → scale_gp_beta-0.1.0a30}/src/scale_gp_beta/_files.py

@@ -69,12 +69,12 @@ def _transform_file(file: FileTypes) -> HttpxFileTypes:
         return file
 
     if is_tuple_t(file):
-        return (file[0], _read_file_content(file[1]), *file[2:])
+        return (file[0], read_file_content(file[1]), *file[2:])
 
     raise TypeError(f"Expected file types input to be a FileContent type or to be a tuple")
 
 
-def _read_file_content(file: FileContent) -> HttpxFileContent:
+def read_file_content(file: FileContent) -> HttpxFileContent:
     if isinstance(file, os.PathLike):
         return pathlib.Path(file).read_bytes()
     return file
@@ -111,12 +111,12 @@ async def _async_transform_file(file: FileTypes) -> HttpxFileTypes:
         return file
 
     if is_tuple_t(file):
-        return (file[0], await _async_read_file_content(file[1]), *file[2:])
+        return (file[0], await async_read_file_content(file[1]), *file[2:])
 
     raise TypeError(f"Expected file types input to be a FileContent type or to be a tuple")
 
 
-async def _async_read_file_content(file: FileContent) -> HttpxFileContent:
+async def async_read_file_content(file: FileContent) -> HttpxFileContent:
     if isinstance(file, os.PathLike):
         return await anyio.Path(file).read_bytes()
 

{scale_gp_beta-0.1.0a28 → scale_gp_beta-0.1.0a30}/src/scale_gp_beta/_models.py

@@ -208,14 +208,18 @@ class BaseModel(pydantic.BaseModel):
             else:
                 fields_values[name] = field_get_default(field)
 
+        extra_field_type = _get_extra_fields_type(__cls)
+
         _extra = {}
         for key, value in values.items():
             if key not in model_fields:
+                parsed = construct_type(value=value, type_=extra_field_type) if extra_field_type is not None else value
+
                 if PYDANTIC_V2:
-                    _extra[key] = value
+                    _extra[key] = parsed
                 else:
                     _fields_set.add(key)
-                    fields_values[key] = value
+                    fields_values[key] = parsed
 
         object.__setattr__(m, "__dict__", fields_values)
 
@@ -370,6 +374,23 @@ def _construct_field(value: object, field: FieldInfo, key: str) -> object:
     return construct_type(value=value, type_=type_, metadata=getattr(field, "metadata", None))
 
 
+def _get_extra_fields_type(cls: type[pydantic.BaseModel]) -> type | None:
+    if not PYDANTIC_V2:
+        # TODO
+        return None
+
+    schema = cls.__pydantic_core_schema__
+    if schema["type"] == "model":
+        fields = schema["schema"]
+        if fields["type"] == "model-fields":
+            extras = fields.get("extras_schema")
+            if extras and "cls" in extras:
+                # mypy can't narrow the type
+                return extras["cls"]  # type: ignore[no-any-return]
+
+    return None
+
+
 def is_basemodel(type_: type) -> bool:
     """Returns whether or not the given type is either a `BaseModel` or a union of `BaseModel`"""
     if is_union(type_):
@@ -439,7 +460,7 @@ def construct_type(*, value: object, type_: object, metadata: Optional[List[Any]
         type_ = type_.__value__  # type: ignore[unreachable]
 
     # unwrap `Annotated[T, ...]` -> `T`
-    if metadata is not None:
+    if metadata is not None and len(metadata) > 0:
         meta: tuple[Any, ...] = tuple(metadata)
     elif is_annotated_type(type_):
         meta = get_args(type_)[1:]

{scale_gp_beta-0.1.0a28 → scale_gp_beta-0.1.0a30}/src/scale_gp_beta/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "scale_gp_beta"
-__version__ = "0.1.0-alpha.28"  # x-release-please-version
+__version__ = "0.1.0-alpha.30"  # x-release-please-version

scale_gp_beta-0.1.0a30/src/scale_gp_beta/lib/CONTRIBUTING.MD

@@ -0,0 +1,53 @@
+# Custom Code Patch
+You can in theory add custom code patches to anywhere in the repo but at the risk of encountering many merge conflicts with Stainless into the future.
+Stainless will never use the `/lib` and `/examples` directories. When possible try to only modify these directories.
+If you have to add custom code elsewhere, please keep the footprint small and create a library for most of the logic.
+
+
+For information on custom code patching with Stainless see [here](https://www.stainless.com/docs/guides/add-custom-code).
+
+# Process for Adding Features
+Checkout the `next` branch and pull, then create a branch from `next`.
+
+> **_NOTE:_** Stainless uses next to "queue up" updates to the SDK.
+> 
+> Stainless will update next with their own logic updates to the SDK along with any changes in the OpenAPI spec and changes to the Stainless
+config on their SaaS platform.
+
+Make any code changes you need. Ensuring all the tests for the library are passing.
+
+There is strict linting in this repo. Use the following commands in order.
+
+```bash
+rye lint --fix
+```
+
+```bash
+rye run lint | grep /specific_file.py
+```
+
+`rye run lint` will not work if there are errors with `rye lint --fix`.
+I am unsure why but I get many errors which are ignorable in the rest of the repo when running `rye run lint` so I usually use it
+with grep to target the file I am developing.
+
+> **_NOTE:_** The strict linting requires all types to be strictly typed. This can be a pain but is worth considering before developing any new solution.
+> Try and avoid using ignore commands when possible, but sometimes it is unavoidable (see OpenAI tracing Plugin).
+
+When commiting, use the [Conventional Commits](https://www.conventionalcommits.org/en/v1.0.0/) spec. Usually this is something like:
+
+```bash
+git commit -m "feat: Added OpenAI tracing processor"
+```
+This allow Stainless to update the release doc with more useful info.
+
+**When creating a PR, you must manually change the destination from `main` to `next`.**
+
+Once merged, Stainless should pick up the changes and update or create a new PR for `main`.
+You will need to accept changes on this PR, Stainless should then auto merge. Note on occasion I have had to manually merge.
+There was no consequence to this.
+
+These PRs Stainless makes have 4 types of changes, all are merged to main via these automatic PR requests:
+1. Custom Code Changes
+2. Stainless SDK logic changes
+3. Changes to the Stainless Config in the SaaS platform
+4. Changes to the OpenAPI schema

{scale_gp_beta-0.1.0a28 → scale_gp_beta-0.1.0a30}/src/scale_gp_beta/lib/tracing/integrations/openai/openai_span_type_map.py

@@ -1,8 +1,8 @@
 from typing import Dict, Optional
 
-from scale_gp_beta.lib.tracing.types import SpanTypeLiterals
+from scale_gp_beta.types import SpanType
 
-OPENAI_SPAN_TYPE_MAP: Dict[str, SpanTypeLiterals] = {
+OPENAI_SPAN_TYPE_MAP: Dict[str, SpanType] = {
     "generation": "COMPLETION",
     "agent": "AGENT_WORKFLOW",
     "function": "CODE_EXECUTION",
@@ -17,7 +17,7 @@ OPENAI_SPAN_TYPE_MAP: Dict[str, SpanTypeLiterals] = {
 }
 
 
-def openai_span_type_map(span_type: Optional[str]) -> SpanTypeLiterals:
+def openai_span_type_map(span_type: Optional[str]) -> SpanType:
     """
     Maps an OpenAI span type string to its corresponding SGP SpanTypeLiteral.
 

{scale_gp_beta-0.1.0a28 → scale_gp_beta-0.1.0a30}/src/scale_gp_beta/lib/tracing/span.py

@@ -6,11 +6,12 @@ from typing import TYPE_CHECKING, Type, Optional
 from threading import RLock
 from typing_extensions import override
 
+from scale_gp_beta.types import SpanType, SpanStatus
 from scale_gp_beta.types.span_upsert_batch_params import Item as SpanCreateRequest
 
 from .util import iso_timestamp, generate_span_id
 from .scope import Scope
-from .types import SpanInputParam, SpanOutputParam, SpanTypeLiterals, SpanMetadataParam, SpanStatusLiterals
+from .types import SpanInputParam, SpanOutputParam, SpanMetadataParam
 from .exceptions import ParamsCreationError
 
 if TYPE_CHECKING:
@@ -59,7 +60,7 @@ class BaseSpan:
         input: Optional[SpanInputParam] = None,
         output: Optional[SpanOutputParam] = None,
         metadata: Optional[SpanMetadataParam] = None,
-        span_type: SpanTypeLiterals = "STANDALONE"
+        span_type: SpanType = "STANDALONE"
     ):
         self._name = name
         self._trace_id: str = trace_id or "no_trace_id"
@@ -71,8 +72,8 @@ class BaseSpan:
         self._input: SpanInputParam = input or {}
         self._output: SpanOutputParam = output or {}
         self._metadata: SpanMetadataParam = metadata or {}
-        self._span_type: SpanTypeLiterals = span_type
-        self._status: SpanStatusLiterals = "SUCCESS"
+        self._span_type: SpanType = span_type
+        self._status: SpanStatus = "SUCCESS"
         self._queue_manager = queue_manager
 
         self._contextvar_token: Optional[contextvars.Token[Optional[BaseSpan]]] = None
@@ -108,11 +109,11 @@ class BaseSpan:
         return self._parent_span_id
 
     @property
-    def status(self) -> SpanStatusLiterals:
+    def status(self) -> SpanStatus:
         return self._status
 
     @property
-    def span_type(self) -> SpanTypeLiterals:
+    def span_type(self) -> SpanType:
         return self._span_type
 
     # with setters
@@ -291,7 +292,7 @@ class Span(BaseSpan):
         input: Optional[SpanInputParam] = None,
         output: Optional[SpanOutputParam] = None,
         metadata: Optional[SpanMetadataParam] = None,
-        span_type: SpanTypeLiterals = "STANDALONE",
+        span_type: SpanType = "STANDALONE",
     ):
         super().__init__(name, trace_id, queue_manager, span_id, parent_span_id, group_id, input, output, metadata, span_type)
         self._queue_manager: TraceQueueManager = queue_manager

{scale_gp_beta-0.1.0a28 → scale_gp_beta-0.1.0a30}/src/scale_gp_beta/lib/tracing/trace.py

@@ -4,10 +4,12 @@ from types import TracebackType
 from typing import Type, Optional
 from typing_extensions import override
 
+from scale_gp_beta.types import SpanType, SpanStatus
+
 from .span import Span, BaseSpan, NoOpSpan
 from .util import generate_trace_id
 from .scope import Scope
-from .types import SpanInputParam, SpanOutputParam, SpanTypeLiterals, SpanMetadataParam, SpanStatusLiterals
+from .types import SpanInputParam, SpanOutputParam, SpanMetadataParam
 from .trace_queue_manager import TraceQueueManager
 
 log: logging.Logger = logging.getLogger(__name__)
@@ -79,11 +81,11 @@ class BaseTrace:
         return self.root_span.group_id
 
     @property
-    def span_type(self) -> SpanTypeLiterals:
+    def span_type(self) -> SpanType:
         return self.root_span.span_type
 
     @property
-    def status(self) -> SpanStatusLiterals:
+    def status(self) -> SpanStatus:
         return self.root_span.status
 
     def set_error(
@@ -128,7 +130,7 @@ class NoOpTrace(BaseTrace):
             trace_id: Optional[str] = None,
             span_id: Optional[str] = None,
             group_id: Optional[str] = None,
-            span_type: SpanTypeLiterals = "TRACER",
+            span_type: SpanType = "TRACER",
             input: Optional[SpanInputParam] = None,
             output: Optional[SpanOutputParam] = None,
             metadata: Optional[SpanMetadataParam] = None,
@@ -164,7 +166,7 @@ class Trace(BaseTrace):
             trace_id: Optional[str] = None,
             span_id: Optional[str] = None,
             group_id: Optional[str] = None,
-            span_type: SpanTypeLiterals = "TRACER",
+            span_type: SpanType = "TRACER",
             input: Optional[SpanInputParam] = None,
             output: Optional[SpanOutputParam] = None,
             metadata: Optional[SpanMetadataParam] = None,

{scale_gp_beta-0.1.0a28 → scale_gp_beta-0.1.0a30}/src/scale_gp_beta/lib/tracing/trace_queue_manager.py

@@ -11,6 +11,8 @@ from .util import configure, is_disabled
 from .trace_exporter import TraceExporter
 
 if TYPE_CHECKING:
+    import httpx
+
     from .span import Span
     from .trace import Trace
 
@@ -44,6 +46,7 @@ class TraceQueueManager:
         worker_enabled: Optional[bool] = None,
     ):
         self._client = client
+        self.register_client(client) if client else None
         self._attempted_local_client_creation = False
         self._trigger_queue_size = trigger_queue_size
         self._trigger_cadence = trigger_cadence
@@ -68,6 +71,17 @@ class TraceQueueManager:
         log.info("Registering client")
         self._client = client
 
+        original_prepare_request = self._client._prepare_request
+
+        def custom_prepare_request(request: "httpx.Request") -> None:
+            original_prepare_request(request)
+
+            # TODO: Hook logic here, we should check to see if we are in the scope of a span, if so we should inject
+            # appropriate headers into the request
+            # current_span = Scope.get_current_span()
+
+        self._client._prepare_request = custom_prepare_request  # type: ignore
+
     def shutdown(self, timeout: Optional[float] = None) -> None:
         if not self._worker_enabled:
             log.debug("No worker to shutdown")

{scale_gp_beta-0.1.0a28 → scale_gp_beta-0.1.0a30}/src/scale_gp_beta/lib/tracing/tracing.py

@@ -1,11 +1,13 @@
 import logging
 from typing import Optional
 
+from scale_gp_beta.types import SpanType
+
 from .span import Span, BaseSpan, NoOpSpan
 from .util import is_disabled
 from .scope import Scope
 from .trace import Trace, BaseTrace, NoOpTrace
-from .types import SpanInputParam, SpanOutputParam, SpanTypeLiterals, SpanMetadataParam
+from .types import SpanInputParam, SpanOutputParam, SpanMetadataParam
 from .trace_queue_manager import TraceQueueManager, tracing_queue_manager
 
 log: logging.Logger = logging.getLogger(__name__)
@@ -51,7 +53,7 @@ def flush_queue() -> None:
 
 def create_trace(
         name: str,
-        span_type: SpanTypeLiterals = "TRACER",
+        span_type: SpanType = "TRACER",
         input: Optional[SpanInputParam] = None,
         output: Optional[SpanOutputParam] = None,
         metadata: Optional[SpanMetadataParam] = None,
@@ -74,7 +76,7 @@ def create_trace(
 
     Args:
         name: The name of the trace.
-        span_type (Optional[SpanTypeLiterals]): Type of root span.
+        span_type (Optional[SpanType]): Type of root span.
         input (Optional[SpanInputParam]): Input of root span.
         output (Optional[SpanOutputParam]): Output of root span.
         metadata (Optional[SpanMetadataParam]): An optional, user-defined metadata.
@@ -131,7 +133,7 @@ def create_trace(
 
 def create_span(
     name: str,
-    span_type: SpanTypeLiterals = "STANDALONE",
+    span_type: SpanType = "STANDALONE",
     input: Optional[SpanInputParam] = None,
     output: Optional[SpanOutputParam] = None,
     metadata: Optional[SpanMetadataParam] = None,
@@ -160,7 +162,7 @@ def create_span(
     Args:
         name (str): A descriptive name for the span (e.g., "database_query",
                     "http_request").
-        span_type (SpanTypeLiterals): The type of the span.
+        span_type (SpanType): The type of the span.
         input (Optional[SpanInputParam], optional): A dictionary containing
             input data or parameters relevant to this span's operation. Defaults to None.
         output (Optional[SpanOutputParam], optional): A dictionary containing

scale_gp_beta-0.1.0a30/src/scale_gp_beta/lib/tracing/types.py

@@ -0,0 +1,5 @@
+from typing_extensions import Any, Dict
+
+SpanInputParam = Dict[str, Any]
+SpanOutputParam = Dict[str, Any]
+SpanMetadataParam = Dict[str, Any]

{scale_gp_beta-0.1.0a28 → scale_gp_beta-0.1.0a30}/src/scale_gp_beta/resources/__init__.py

@@ -56,6 +56,14 @@ from .questions import (
     QuestionsResourceWithStreamingResponse,
     AsyncQuestionsResourceWithStreamingResponse,
 )
+from .responses import (
+    ResponsesResource,
+    AsyncResponsesResource,
+    ResponsesResourceWithRawResponse,
+    AsyncResponsesResourceWithRawResponse,
+    ResponsesResourceWithStreamingResponse,
+    AsyncResponsesResourceWithStreamingResponse,
+)
 from .completions import (
     CompletionsResource,
     AsyncCompletionsResource,
@@ -90,6 +98,12 @@ from .evaluation_items import (
 )
 
 __all__ = [
+    "ResponsesResource",
+    "AsyncResponsesResource",
+    "ResponsesResourceWithRawResponse",
+    "AsyncResponsesResourceWithRawResponse",
+    "ResponsesResourceWithStreamingResponse",
+    "AsyncResponsesResourceWithStreamingResponse",
     "CompletionsResource",
     "AsyncCompletionsResource",
     "CompletionsResourceWithRawResponse",