-
Notifications
You must be signed in to change notification settings - Fork 96
/
schema.py
29 lines (19 loc) · 1.13 KB
/
schema.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
from typing import Optional, Union
from pydantic import Field
from genai.schema import ModerationParameters, PromptTemplateData, TextGenerationParameters, TextTokenizationParameters
from genai.schema._api import _TextGenerationCreateRequest, _TextTokenizationCreateRequest
class TextGenerationCreateRequest(_TextGenerationCreateRequest):
"""Following properties are required for local server"""
input: str
model_id: str
parameters: TextGenerationParameters = Field(default_factory=lambda: TextGenerationParameters())
data: Optional[PromptTemplateData] = None
input: Optional[str] = Field(None, examples=["How are you"], title="Input string")
moderations: Optional[ModerationParameters] = None
class TextTokenizationCreateRequest(_TextTokenizationCreateRequest):
"""Following properties are required for local server"""
model_id: str = Field(...)
parameters: TextTokenizationParameters = Field(default_factory=lambda: TextTokenizationParameters())
input: Optional[Union[str, list[str]]] = None
prompt_id: Optional[str] = None
__all__ = ["TextGenerationCreateRequest", "TextTokenizationCreateRequest"]