Skip to content

Commit 563f456

Browse files
authored
Update chat_completions.py
1 parent 5b87c03 commit 563f456

1 file changed

Lines changed: 9 additions & 2 deletions

File tree

src/together/types/chat_completions.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ class MessageRole(str, Enum):
2828
class ResponseFormatType(str, Enum):
2929
JSON_OBJECT = "json_object"
3030
JSON_SCHEMA = "json_schema"
31+
REGEX = "regex"
3132

3233

3334
class FunctionCall(BaseModel):
@@ -71,9 +72,15 @@ class ChatCompletionMessage(BaseModel):
7172
class ResponseFormat(BaseModel):
7273
type: ResponseFormatType
7374
schema_: Dict[str, Any] | None = None
74-
75+
pattern: str | None = None
76+
7577
def to_dict(self) -> Dict[str, Any]:
76-
return {"schema": self.schema_, "type": self.type}
78+
result = {"type": self.type}
79+
if self.schema_ is not None:
80+
result["schema"] = self.schema_
81+
if self.pattern is not None:
82+
result["pattern"] = self.pattern
83+
return result
7784

7885

7986
class FunctionTool(BaseModel):

0 commit comments

Comments
 (0)