Sync changes for the ParamType fix (#90)

# What does this PR do?

Sync changes for the ParamType fix

Author: dineshyv

src/llama_stack_client/types/shared/param_type.py

@@ -5,11 +5,70 @@
 
 from ..._models import BaseModel
 
-__all__ = ["ParamType", "Type"]
+__all__ = [
+    "ParamType",
+    "StringType",
+    "NumberType",
+    "BooleanType",
+    "ArrayType",
+    "ObjectType",
+    "JsonType",
+    "UnionType",
+    "ChatCompletionInputType",
+    "CompletionInputType",
+    "AgentTurnInputType",
+]
 
 
-class Type(BaseModel):
+class StringType(BaseModel):
     type: Literal["string"]
 
 
-ParamType: TypeAlias = Union[Type, Type, Type, Type, Type, Type, Type, Type, Type, Type]
+class NumberType(BaseModel):
+    type: Literal["number"]
+
+
+class BooleanType(BaseModel):
+    type: Literal["boolean"]
+
+
+class ArrayType(BaseModel):
+    type: Literal["array"]
+
+
+class ObjectType(BaseModel):
+    type: Literal["object"]
+
+
+class JsonType(BaseModel):
+    type: Literal["json"]
+
+
+class UnionType(BaseModel):
+    type: Literal["union"]
+
+
+class ChatCompletionInputType(BaseModel):
+    type: Literal["chat_completion_input"]
+
+
+class CompletionInputType(BaseModel):
+    type: Literal["completion_input"]
+
+
+class AgentTurnInputType(BaseModel):
+    type: Literal["agent_turn_input"]
+
+
+ParamType: TypeAlias = Union[
+    StringType,
+    NumberType,
+    BooleanType,
+    ArrayType,
+    ObjectType,
+    JsonType,
+    UnionType,
+    ChatCompletionInputType,
+    CompletionInputType,
+    AgentTurnInputType,
+]

src/llama_stack_client/types/shared_params/param_type.py

@@ -5,11 +5,70 @@
 from typing import Union
 from typing_extensions import Literal, Required, TypeAlias, TypedDict
 
-__all__ = ["ParamType", "Type"]
+__all__ = [
+    "ParamType",
+    "StringType",
+    "NumberType",
+    "BooleanType",
+    "ArrayType",
+    "ObjectType",
+    "JsonType",
+    "UnionType",
+    "ChatCompletionInputType",
+    "CompletionInputType",
+    "AgentTurnInputType",
+]
 
 
-class Type(TypedDict, total=False):
+class StringType(TypedDict, total=False):
     type: Required[Literal["string"]]
 
 
-ParamType: TypeAlias = Union[Type, Type, Type, Type, Type, Type, Type, Type, Type, Type]
+class NumberType(TypedDict, total=False):
+    type: Required[Literal["number"]]
+
+
+class BooleanType(TypedDict, total=False):
+    type: Required[Literal["boolean"]]
+
+
+class ArrayType(TypedDict, total=False):
+    type: Required[Literal["array"]]
+
+
+class ObjectType(TypedDict, total=False):
+    type: Required[Literal["object"]]
+
+
+class JsonType(TypedDict, total=False):
+    type: Required[Literal["json"]]
+
+
+class UnionType(TypedDict, total=False):
+    type: Required[Literal["union"]]
+
+
+class ChatCompletionInputType(TypedDict, total=False):
+    type: Required[Literal["chat_completion_input"]]
+
+
+class CompletionInputType(TypedDict, total=False):
+    type: Required[Literal["completion_input"]]
+
+
+class AgentTurnInputType(TypedDict, total=False):
+    type: Required[Literal["agent_turn_input"]]
+
+
+ParamType: TypeAlias = Union[
+    StringType,
+    NumberType,
+    BooleanType,
+    ArrayType,
+    ObjectType,
+    JsonType,
+    UnionType,
+    ChatCompletionInputType,
+    CompletionInputType,
+    AgentTurnInputType,
+]