from enum import Enum

from typing import (

Dict,

List,

Optional,

)

from pydantic import BaseModel

class CodegateFunction(BaseModel):

name: Optional[str] = None

arguments: Optional[str] = None

class CodegateChatCompletionDeltaToolCall(BaseModel):

index: Optional[int] = None

class CodegateDelta(BaseModel):

role: str

content: Optional[str] = None

tool_calls: Optional[List[CodegateChatCompletionDeltaToolCall]] = None

class CodegateStreamingChoices(BaseModel):

delta: CodegateDelta

index: Optional[int] = None

finish_reason: Optional[str] = None

class CodegateModelResponseStream(BaseModel):

payload: Optional[Dict] = None

class MessageTypeFilter(Enum):

USER = "user"