gen sdk GCP

Author: aac228

.speakeasy/workflow.lock

@@ -8,8 +8,8 @@ sources:
             - latest
     mistral-google-cloud-source:
         sourceNamespace: mistral-google-cloud-source
-        sourceRevisionDigest: sha256:d3e3d15303dcc1acb27b8895aa3064328bd5b8013ea635c2bce553b6e647b498
-        sourceBlobDigest: sha256:db72004ee842a27c3e77980be28a727811e0581daa7a51ad34d142302f8ba2f3
+        sourceRevisionDigest: sha256:e0fd58ce2dbba068f375d3a23d758b8678c2a68cf4fc7bc46ea7e1b37abe0647
+        sourceBlobDigest: sha256:0707d8d2566a9ef4ef286bb0abe467f8696ccf83ba73091065d7caf627a06611
         tags:
             - latest
     mistral-openapi:
@@ -29,10 +29,10 @@ targets:
     mistralai-gcp-sdk:
         source: mistral-google-cloud-source
         sourceNamespace: mistral-google-cloud-source
-        sourceRevisionDigest: sha256:d3e3d15303dcc1acb27b8895aa3064328bd5b8013ea635c2bce553b6e647b498
-        sourceBlobDigest: sha256:db72004ee842a27c3e77980be28a727811e0581daa7a51ad34d142302f8ba2f3
+        sourceRevisionDigest: sha256:e0fd58ce2dbba068f375d3a23d758b8678c2a68cf4fc7bc46ea7e1b37abe0647
+        sourceBlobDigest: sha256:0707d8d2566a9ef4ef286bb0abe467f8696ccf83ba73091065d7caf627a06611
         codeSamplesNamespace: mistral-openapi-google-cloud-code-samples
-        codeSamplesRevisionDigest: sha256:7d95ba7aa230088b9975be341ba638e51cc574b6e863bd3a0f53e9c5ee261bba
+        codeSamplesRevisionDigest: sha256:0657ec41e473356a5a0eeaca3dff137e9ff16080ec1fb50e72553245aa86ffe5
     mistralai-sdk:
         source: mistral-openapi
         sourceNamespace: mistral-openapi

packages/mistralai_gcp/.speakeasy/gen.lock

@@ -1,18 +1,18 @@
 lockVersion: 2.0.0
 id: ec60f2d8-7869-45c1-918e-773d41a8cf74
 management:
-  docChecksum: e6c0a4254e61b1f171b409862f717867
+  docChecksum: d50a06ac34844141709fa2e57cc940c5
   docVersion: 0.0.2
-  speakeasyVersion: 1.440.1
-  generationVersion: 2.460.1
+  speakeasyVersion: 1.451.1
+  generationVersion: 2.470.1
   releaseVersion: 1.2.3
   configChecksum: 3fc99d7ec7ee057a323b593ebf8fdb8c
   published: true
 features:
   python:
     additionalDependencies: 1.0.0
     constsAndDefaults: 1.0.5
-    core: 5.6.5
+    core: 5.6.8
     defaultEnabledRetries: 0.2.0
     enumUnions: 0.1.0
     envVarSecurityUsage: 0.3.2
@@ -29,11 +29,10 @@ features:
     responseFormat: 1.0.1
     retries: 3.0.2
     sdkHooks: 1.0.0
-    serverEvents: 1.0.4
+    serverEvents: 1.0.7
     serverEventsSentinels: 0.1.0
     serverIDs: 3.0.0
-    tests: 1.6.0
-    unions: 3.0.3
+    unions: 3.0.4
 generatedFiles:
   - .gitattributes
   - .python-version
@@ -187,3 +186,4 @@ examples:
         "200":
           application/json: {"id": "cmpl-e5cc70bb28c444948073e77776eb30ef", "object": "chat.completion", "model": "codestral-latest", "usage": {"prompt_tokens": 16, "completion_tokens": 34, "total_tokens": 50}, "created": 1702256327, "choices": []}
         "422": {}
+generatedTests: {}

packages/mistralai_gcp/pyproject.toml

@@ -22,12 +22,12 @@ google-auth = "2.27.0"
 httpx = "^0.27.0"
 jsonpath-python = "^1.0.6"
 pydantic = "~2.9.2"
-python-dateutil = "2.8.2"
+python-dateutil = "^2.8.2"
 requests = "^2.32.3"
 typing-inspect = "^0.9.0"
 
 [tool.poetry.group.dev.dependencies]
-mypy = "==1.10.1"
+mypy = "==1.13.0"
 pylint = "==3.2.3"
 pytest = "^8.2.2"
 pytest-asyncio = "^0.23.7"

packages/mistralai_gcp/src/mistralai_gcp/chat.py

@@ -5,7 +5,7 @@
 from mistralai_gcp._hooks import HookContext
 from mistralai_gcp.types import Nullable, OptionalNullable, UNSET
 from mistralai_gcp.utils import eventstreaming
-from typing import Any, AsyncGenerator, Generator, List, Optional, Union
+from typing import Any, List, Optional, Union
 
 
 class Chat(BaseSDK):
@@ -40,7 +40,7 @@ def stream(
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
-    ) -> Optional[Generator[models.CompletionEvent, None, None]]:
+    ) -> Optional[eventstreaming.EventStream[models.CompletionEvent]]:
         r"""Stream chat completion
 
         Mistral AI provides the ability to stream responses back to a client in order to allow partial results for certain requests. Tokens will be sent as data-only server-sent events as they become available, with the stream terminated by a data: [DONE] message. Otherwise, the server will hold the request open until the timeout or until completion, with the response containing the full result as JSON.
@@ -132,7 +132,7 @@ def stream(
 
         data: Any = None
         if utils.match_response(http_res, "200", "text/event-stream"):
-            return eventstreaming.stream_events(
+            return eventstreaming.EventStream(
                 http_res,
                 lambda raw: utils.unmarshal_json(raw, models.CompletionEvent),
                 sentinel="[DONE]",
@@ -185,7 +185,7 @@ async def stream_async(
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
-    ) -> Optional[AsyncGenerator[models.CompletionEvent, None]]:
+    ) -> Optional[eventstreaming.EventStreamAsync[models.CompletionEvent]]:
         r"""Stream chat completion
 
         Mistral AI provides the ability to stream responses back to a client in order to allow partial results for certain requests. Tokens will be sent as data-only server-sent events as they become available, with the stream terminated by a data: [DONE] message. Otherwise, the server will hold the request open until the timeout or until completion, with the response containing the full result as JSON.
@@ -277,7 +277,7 @@ async def stream_async(
 
         data: Any = None
         if utils.match_response(http_res, "200", "text/event-stream"):
-            return eventstreaming.stream_events_async(
+            return eventstreaming.EventStreamAsync(
                 http_res,
                 lambda raw: utils.unmarshal_json(raw, models.CompletionEvent),
                 sentinel="[DONE]",

packages/mistralai_gcp/src/mistralai_gcp/fim.py

@@ -5,7 +5,7 @@
 from mistralai_gcp._hooks import HookContext
 from mistralai_gcp.types import Nullable, OptionalNullable, UNSET
 from mistralai_gcp.utils import eventstreaming
-from typing import Any, AsyncGenerator, Generator, Optional, Union
+from typing import Any, Optional, Union
 
 
 class Fim(BaseSDK):
@@ -32,7 +32,7 @@ def stream(
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
-    ) -> Optional[Generator[models.CompletionEvent, None, None]]:
+    ) -> Optional[eventstreaming.EventStream[models.CompletionEvent]]:
         r"""Stream fim completion
 
         Mistral AI provides the ability to stream responses back to a client in order to allow partial results for certain requests. Tokens will be sent as data-only server-sent events as they become available, with the stream terminated by a data: [DONE] message. Otherwise, the server will hold the request open until the timeout or until completion, with the response containing the full result as JSON.
@@ -112,7 +112,7 @@ def stream(
 
         data: Any = None
         if utils.match_response(http_res, "200", "text/event-stream"):
-            return eventstreaming.stream_events(
+            return eventstreaming.EventStream(
                 http_res,
                 lambda raw: utils.unmarshal_json(raw, models.CompletionEvent),
                 sentinel="[DONE]",
@@ -157,7 +157,7 @@ async def stream_async(
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
-    ) -> Optional[AsyncGenerator[models.CompletionEvent, None]]:
+    ) -> Optional[eventstreaming.EventStreamAsync[models.CompletionEvent]]:
         r"""Stream fim completion
 
         Mistral AI provides the ability to stream responses back to a client in order to allow partial results for certain requests. Tokens will be sent as data-only server-sent events as they become available, with the stream terminated by a data: [DONE] message. Otherwise, the server will hold the request open until the timeout or until completion, with the response containing the full result as JSON.
@@ -237,7 +237,7 @@ async def stream_async(
 
         data: Any = None
         if utils.match_response(http_res, "200", "text/event-stream"):
-            return eventstreaming.stream_events_async(
+            return eventstreaming.EventStreamAsync(
                 http_res,
                 lambda raw: utils.unmarshal_json(raw, models.CompletionEvent),
                 sentinel="[DONE]",

packages/mistralai_gcp/src/mistralai_gcp/httpclient.py

@@ -41,6 +41,9 @@ def build_request(
     ) -> httpx.Request:
         pass
 
+    def close(self) -> None:
+        pass
+
 
 @runtime_checkable
 class AsyncHttpClient(Protocol):
@@ -76,3 +79,6 @@ def build_request(
         extensions: Optional[httpx._types.RequestExtensions] = None,
     ) -> httpx.Request:
         pass
+
+    async def aclose(self) -> None:
+        pass

packages/mistralai_gcp/src/mistralai_gcp/models/assistantmessage.py

@@ -12,13 +12,17 @@
 )
 from pydantic import model_serializer
 from typing import List, Literal, Optional, Union
-from typing_extensions import NotRequired, TypedDict
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-AssistantMessageContentTypedDict = Union[str, List[ContentChunkTypedDict]]
+AssistantMessageContentTypedDict = TypeAliasType(
+    "AssistantMessageContentTypedDict", Union[str, List[ContentChunkTypedDict]]
+)
 
 
-AssistantMessageContent = Union[str, List[ContentChunk]]
+AssistantMessageContent = TypeAliasType(
+    "AssistantMessageContent", Union[str, List[ContentChunk]]
+)
 
 
 AssistantMessageRole = Literal["assistant"]

packages/mistralai_gcp/src/mistralai_gcp/models/chatcompletionrequest.py

@@ -19,23 +19,30 @@
 from mistralai_gcp.utils import get_discriminator
 from pydantic import Discriminator, Tag, model_serializer
 from typing import List, Optional, Union
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
-ChatCompletionRequestStopTypedDict = Union[str, List[str]]
+ChatCompletionRequestStopTypedDict = TypeAliasType(
+    "ChatCompletionRequestStopTypedDict", Union[str, List[str]]
+)
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-ChatCompletionRequestStop = Union[str, List[str]]
+ChatCompletionRequestStop = TypeAliasType(
+    "ChatCompletionRequestStop", Union[str, List[str]]
+)
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-ChatCompletionRequestMessagesTypedDict = Union[
-    SystemMessageTypedDict,
-    UserMessageTypedDict,
-    AssistantMessageTypedDict,
-    ToolMessageTypedDict,
-]
+ChatCompletionRequestMessagesTypedDict = TypeAliasType(
+    "ChatCompletionRequestMessagesTypedDict",
+    Union[
+        SystemMessageTypedDict,
+        UserMessageTypedDict,
+        AssistantMessageTypedDict,
+        ToolMessageTypedDict,
+    ],
+)
 
 
 ChatCompletionRequestMessages = Annotated[
@@ -49,10 +56,15 @@
 ]
 
 
-ChatCompletionRequestToolChoiceTypedDict = Union[ToolChoiceTypedDict, ToolChoiceEnum]
+ChatCompletionRequestToolChoiceTypedDict = TypeAliasType(
+    "ChatCompletionRequestToolChoiceTypedDict",
+    Union[ToolChoiceTypedDict, ToolChoiceEnum],
+)
 
 
-ChatCompletionRequestToolChoice = Union[ToolChoice, ToolChoiceEnum]
+ChatCompletionRequestToolChoice = TypeAliasType(
+    "ChatCompletionRequestToolChoice", Union[ToolChoice, ToolChoiceEnum]
+)
 
 
 class ChatCompletionRequestTypedDict(TypedDict):

packages/mistralai_gcp/src/mistralai_gcp/models/chatcompletionstreamrequest.py

@@ -19,23 +19,26 @@
 from mistralai_gcp.utils import get_discriminator
 from pydantic import Discriminator, Tag, model_serializer
 from typing import List, Optional, Union
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
-StopTypedDict = Union[str, List[str]]
+StopTypedDict = TypeAliasType("StopTypedDict", Union[str, List[str]])
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-Stop = Union[str, List[str]]
+Stop = TypeAliasType("Stop", Union[str, List[str]])
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-MessagesTypedDict = Union[
-    SystemMessageTypedDict,
-    UserMessageTypedDict,
-    AssistantMessageTypedDict,
-    ToolMessageTypedDict,
-]
+MessagesTypedDict = TypeAliasType(
+    "MessagesTypedDict",
+    Union[
+        SystemMessageTypedDict,
+        UserMessageTypedDict,
+        AssistantMessageTypedDict,
+        ToolMessageTypedDict,
+    ],
+)
 
 
 Messages = Annotated[
@@ -49,12 +52,15 @@
 ]
 
 
-ChatCompletionStreamRequestToolChoiceTypedDict = Union[
-    ToolChoiceTypedDict, ToolChoiceEnum
-]
+ChatCompletionStreamRequestToolChoiceTypedDict = TypeAliasType(
+    "ChatCompletionStreamRequestToolChoiceTypedDict",
+    Union[ToolChoiceTypedDict, ToolChoiceEnum],
+)
 
 
-ChatCompletionStreamRequestToolChoice = Union[ToolChoice, ToolChoiceEnum]
+ChatCompletionStreamRequestToolChoice = TypeAliasType(
+    "ChatCompletionStreamRequestToolChoice", Union[ToolChoice, ToolChoiceEnum]
+)
 
 
 class ChatCompletionStreamRequestTypedDict(TypedDict):

packages/mistralai_gcp/src/mistralai_gcp/models/contentchunk.py

@@ -6,10 +6,12 @@
 from mistralai_gcp.utils import get_discriminator
 from pydantic import Discriminator, Tag
 from typing import Union
-from typing_extensions import Annotated
+from typing_extensions import Annotated, TypeAliasType
 
 
-ContentChunkTypedDict = Union[TextChunkTypedDict, ReferenceChunkTypedDict]
+ContentChunkTypedDict = TypeAliasType(
+    "ContentChunkTypedDict", Union[TextChunkTypedDict, ReferenceChunkTypedDict]
+)
 
 
 ContentChunk = Annotated[