qwak-core 0.4.403__py3-none-any.whl → 0.4.405__py3-none-any.whl

qwak/__init__.py

@@ -1,7 +1,7 @@
 """Top-level package for qwak-core."""
 
 __author__ = "Qwak.ai"
-__version__ = "0.4.403"
+__version__ = "0.4.405"
 
 from qwak.inner.di_configuration import wire_dependencies
 from qwak.model.experiment_tracking import log_metric, log_param

qwak/llmops/generation/chat/openai/types/chat/chat_completion.py

@@ -16,8 +16,6 @@
 from dataclasses import dataclass
 from typing import List, Optional
 
-from typing_extensions import Literal
-
 from qwak.llmops.generation.base import ModelResponse
 from .chat_completion_message import ChatCompletionMessage
 from .chat_completion_token_logprob import ChatCompletionTokenLogprob
@@ -34,9 +32,7 @@ class ChoiceLogprobs:
 
 @dataclass
 class Choice:
-    finish_reason: Literal[
-        "stop", "length", "tool_calls", "content_filter", "function_call"
-    ]
+    finish_reason: str
     """The reason the model stopped generating tokens.
 
     This will be `stop` if the model hit a natural stop point or a provided stop
@@ -55,6 +51,21 @@ class Choice:
     logprobs: Optional[ChoiceLogprobs] = None
     """Log probability information for the choice."""
 
+    def __post_init__(self):
+        """Validates that finish_reason is one of the allowed values."""
+        allowed_reasons = {
+            "stop",
+            "length",
+            "tool_calls",
+            "content_filter",
+            "function_call",
+        }
+        if self.finish_reason not in allowed_reasons:
+            raise ValueError(
+                f"Invalid finish_reason: '{self.finish_reason}'. "
+                f"Must be one of {allowed_reasons}"
+            )
+
 
 @dataclass
 class ChatCompletion(ModelResponse):
@@ -73,7 +84,7 @@ class ChatCompletion(ModelResponse):
     model: str
     """The model used for the chat completion."""
 
-    object: Literal["chat.completion"]
+    object: str
     """The object type, which is always `chat.completion`."""
 
     system_fingerprint: Optional[str] = None
@@ -85,3 +96,10 @@ class ChatCompletion(ModelResponse):
 
     usage: Optional[CompletionUsage] = None
     """Usage statistics for the completion request."""
+
+    def __post_init__(self):
+        """Validates that the object type is correct."""
+        if self.object != "chat.completion":
+            raise ValueError(
+                f"Invalid object type: '{self.object}'. Must be 'chat.completion'"
+            )

qwak/llmops/generation/chat/openai/types/chat/chat_completion_chunk.py

@@ -16,8 +16,6 @@
 from dataclasses import dataclass
 from typing import List, Optional
 
-from typing_extensions import Literal
-
 from qwak.llmops.generation.base import ModelResponse
 from .chat_completion_token_logprob import ChatCompletionTokenLogprob
 
@@ -69,9 +67,16 @@ class ChoiceDeltaToolCall:
 
     function: Optional[ChoiceDeltaToolCallFunction] = None
 
-    type: Optional[Literal["function"]] = None
+    type: Optional[str] = None
     """The type of the tool. Currently, only `function` is supported."""
 
+    def __post_init__(self):
+        """Validates that type is 'function' if present."""
+        if self.type is not None and self.type != "function":
+            raise ValueError(
+                f"Invalid type: '{self.type}'. Must be 'function' or None."
+            )
+
 
 @dataclass
 class ChoiceDelta:
@@ -85,11 +90,21 @@ class ChoiceDelta:
     model.
     """
 
-    role: Optional[Literal["system", "user", "assistant", "tool"]] = None
+    role: Optional[str] = None
     """The role of the author of this message."""
 
     tool_calls: Optional[List[ChoiceDeltaToolCall]] = None
 
+    def __post_init__(self):
+        """Validates that role is one of the allowed values if present."""
+        if self.role is not None:
+            allowed_roles = {"system", "user", "assistant", "tool"}
+            if self.role not in allowed_roles:
+                raise ValueError(
+                    f"Invalid role: '{self.role}'. "
+                    f"Must be one of {allowed_roles} or None."
+                )
+
 
 @dataclass
 class ChoiceLogprobs:
@@ -105,9 +120,7 @@ class Choice:
     index: int
     """The index of the choice in the list of choices."""
 
-    finish_reason: Optional[
-        Literal["stop", "length", "tool_calls", "content_filter", "function_call"]
-    ] = None
+    finish_reason: Optional[str] = None
     """The reason the model stopped generating tokens.
 
     This will be `stop` if the model hit a natural stop point or a provided stop
@@ -120,6 +133,22 @@ class Choice:
     logprobs: Optional[ChoiceLogprobs] = None
     """Log probability information for the choice."""
 
+    def __post_init__(self):
+        """Validates that finish_reason is one of the allowed values if present."""
+        if self.finish_reason is not None:
+            allowed_reasons = {
+                "stop",
+                "length",
+                "tool_calls",
+                "content_filter",
+                "function_call",
+            }
+            if self.finish_reason not in allowed_reasons:
+                raise ValueError(
+                    f"Invalid finish_reason: '{self.finish_reason}'. "
+                    f"Must be one of {allowed_reasons} or None."
+                )
+
 
 @dataclass
 class ChatCompletionChunk(ModelResponse):
@@ -141,7 +170,7 @@ class ChatCompletionChunk(ModelResponse):
     model: str
     """The model to generate the completion."""
 
-    object: Literal["chat.completion.chunk"]
+    object: str
     """The object type, which is always `chat.completion.chunk`."""
 
     system_fingerprint: Optional[str] = None
@@ -150,3 +179,10 @@ class ChatCompletionChunk(ModelResponse):
     Can be used in conjunction with the `seed` request parameter to understand when
     backend changes have been made that might impact determinism.
     """
+
+    def __post_init__(self):
+        """Validates that the object type is correct."""
+        if self.object != "chat.completion.chunk":
+            raise ValueError(
+                f"Invalid object type: '{self.object}'. Must be 'chat.completion.chunk'"
+            )

qwak/llmops/generation/chat/openai/types/chat/chat_completion_message.py

@@ -16,8 +16,6 @@
 from dataclasses import dataclass
 from typing import List, Optional
 
-from typing_extensions import Literal
-
 from .chat_completion_message_tool_call import ChatCompletionMessageToolCall
 
 __all__ = ["ChatCompletionMessage", "FunctionCall"]
@@ -39,7 +37,7 @@ class FunctionCall:
 
 @dataclass
 class ChatCompletionMessage:
-    role: Literal["assistant"]
+    role: str
     """The role of the author of this message."""
 
     content: Optional[str] = None
@@ -54,3 +52,8 @@ class ChatCompletionMessage:
 
     tool_calls: Optional[List[ChatCompletionMessageToolCall]] = None
     """The tool calls generated by the model, such as function calls."""
+
+    def __post_init__(self):
+        """Validates that the role type is correct."""
+        if self.role != "assistant":
+            raise ValueError(f"Invalid object type: '{self.role}'. Must be 'assistant'")

{qwak_core-0.4.403.dist-info → qwak_core-0.4.405.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: qwak-core
-Version: 0.4.403
+Version: 0.4.405
 Summary: Qwak Core contains the necessary objects and communication tools for using the Qwak Platform
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model
@@ -22,7 +22,7 @@ Requires-Dist: PyYAML (>=6.0.2)
 Requires-Dist: cachetools
 Requires-Dist: chevron (==0.14.0)
 Requires-Dist: cloudpickle (==2.2.1) ; extra == "feature-store"
-Requires-Dist: dacite (==1.8.1)
+Requires-Dist: dacite (==1.9.2)
 Requires-Dist: dependency-injector (>=4.0)
 Requires-Dist: filelock
 Requires-Dist: grpcio (>=1.71.2)
@@ -32,11 +32,11 @@ Requires-Dist: protobuf (>=4.25.8,<5)
 Requires-Dist: pyarrow (>=20.0.0) ; extra == "feature-store"
 Requires-Dist: pyathena (>=2.2.0,!=2.18.0) ; extra == "feature-store"
 Requires-Dist: pydantic
-Requires-Dist: pyspark (==3.4.2) ; extra == "feature-store"
+Requires-Dist: pyspark (==3.5.7) ; extra == "feature-store"
 Requires-Dist: python-jose[cryptography] (>=3.4.0)
 Requires-Dist: python-json-logger (>=2.0.2)
 Requires-Dist: requests
-Requires-Dist: retrying (==1.3.4)
+Requires-Dist: retrying (==1.4.2)
 Requires-Dist: tqdm
 Requires-Dist: typeguard (>=2,<3)
 Requires-Dist: typer

{qwak_core-0.4.403.dist-info → qwak_core-0.4.405.dist-info}/RECORD

@@ -622,7 +622,7 @@ _qwak_proto/qwak/workspace/workspace_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXH
 _qwak_proto/qwak/workspace/workspace_service_pb2.py,sha256=93uNm83VVrkLG_XVsuBOBTPs4UJF91YD1xJTOEbRyig,9239
 _qwak_proto/qwak/workspace/workspace_service_pb2.pyi,sha256=nKKCHwnovZhsy8TSVmdz-Vtl0nviOOoX56HD-41Xo08,13726
 _qwak_proto/qwak/workspace/workspace_service_pb2_grpc.py,sha256=yKGuexxTBza99Ihe0DSTniV2ZSd_AG47inHenqfi890,27193
-qwak/__init__.py,sha256=jbJpnK5qaR8QUf28ZAElDwpseRvDwYcBf85PQMIs89c,587
+qwak/__init__.py,sha256=9jl_ZNELtQVatPBtj2D-NKNFWYPVHaOnUSf8EEM8PJk,587
 qwak/automations/__init__.py,sha256=qFZRvCxUUn8gcxkJR0v19ulHW2oJ0x6-Rif7HiheDP4,1522
 qwak/automations/automation_executions.py,sha256=5MeH_epYYWb8NKXgAozwT_jPyyUDednBHG7izloi7RY,3228
 qwak/automations/automations.py,sha256=2Wyrgj2mW5VmJzTKAXj3vQM_svYCuq_Bsu3_Vu9SdR4,12732
@@ -895,13 +895,13 @@ qwak/llmops/generation/chat/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMp
 qwak/llmops/generation/chat/openai/LICENSE.txt,sha256=d0M6HDjQ76tf255XPlAGkIoECMe688MXcGEYsOFySfI,11336
 qwak/llmops/generation/chat/openai/types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak/llmops/generation/chat/openai/types/chat/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-qwak/llmops/generation/chat/openai/types/chat/chat_completion.py,sha256=GNabt8TN-N890KrZSUkwkG87aCYobVmtVZRpEV_1BKg,2922
+qwak/llmops/generation/chat/openai/types/chat/chat_completion.py,sha256=tkrAUs112xAOVbwsYPZj0JIacGMNWAZ_7E1D_Hm0NN8,3519
 qwak/llmops/generation/chat/openai/types/chat/chat_completion_assistant_message_param.py,sha256=Z-RpaLK5p7Mf87HXXEkhf7nIFsrq7GtJ-bZoAByat_k,2155
-qwak/llmops/generation/chat/openai/types/chat/chat_completion_chunk.py,sha256=K5yqRScwwgvQ_i67G63JKDJOOCqdHAa_md10XbYzw-k,4696
+qwak/llmops/generation/chat/openai/types/chat/chat_completion_chunk.py,sha256=YeQx2PFkM0SqUPZE1QfsbdD3GeyWIC3sMcYgwm0dzbg,6045
 qwak/llmops/generation/chat/openai/types/chat/chat_completion_content_part_text_param.py,sha256=N-880rUhsmzQX0gliLoc0GEzRYG6b999abPdiv5CeSc,945
 qwak/llmops/generation/chat/openai/types/chat/chat_completion_function_call_option_param.py,sha256=T_dfvBpEupTdwHVxBX9nAJu5M0YmH7GNNwofQ-DT1kY,881
 qwak/llmops/generation/chat/openai/types/chat/chat_completion_function_message_param.py,sha256=23WF1T86LoWrQaVg5DAASlxlpE82UUZYa58l3MlEF1Y,1108
-qwak/llmops/generation/chat/openai/types/chat/chat_completion_message.py,sha256=GjQx7m7cJ6HnjYewh83a51gDQBZfL45vS5BJAP0eZTw,1800
+qwak/llmops/generation/chat/openai/types/chat/chat_completion_message.py,sha256=8DIPVH_HLwLxUtLY8hc85WvMb2vsktvXJ6aop9SmSPY,1955
 qwak/llmops/generation/chat/openai/types/chat/chat_completion_message_param.py,sha256=wjJTJwOOZhFx8KgmBHl9iZJDMutzEma5R0q3FlJ2vUI,1354
 qwak/llmops/generation/chat/openai/types/chat/chat_completion_message_tool_call.py,sha256=HrmsJw6D41H3a-8lNZsYRz0MJy3_8YGnX9-HEAYO3d4,1417
 qwak/llmops/generation/chat/openai/types/chat/chat_completion_message_tool_call_param.py,sha256=1PKfzKEV1WjtN7xkvGSLyS7H9wILM5LLQEbr5eg6Q2g,1525
@@ -1095,6 +1095,6 @@ qwak_services_mock/mocks/workspace_manager_service_mock.py,sha256=O9ZSwln4T4kHVk
 qwak_services_mock/services_mock.py,sha256=0BWwS2re9ryO3JrJJgSNyEQ0lDOjyrpV36oa8t7Pd7A,19163
 qwak_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
-qwak_core-0.4.403.dist-info/METADATA,sha256=fY7VcDYvh5uQEGyspc7RGMLjrM8DA0hgng54VRMrTgo,2088
-qwak_core-0.4.403.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-qwak_core-0.4.403.dist-info/RECORD,,
+qwak_core-0.4.405.dist-info/METADATA,sha256=0AlgtO69BiHj6tBFUGDI32YjzE2SWlkYrpOoWiZoPgI,2088
+qwak_core-0.4.405.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+qwak_core-0.4.405.dist-info/RECORD,,