Jina-13.py
· 177 B · Python
Raw
from docarray import BaseDoc
class PromptDocument(BaseDoc):
prompt: str
max_tokens: int
class ModelOutputDocument(BaseDoc):
token_id: int
generated_text: str
| 1 | from docarray import BaseDoc |
| 2 | |
| 3 | |
| 4 | class PromptDocument(BaseDoc): |
| 5 | prompt: str |
| 6 | max_tokens: int |
| 7 | |
| 8 | |
| 9 | class ModelOutputDocument(BaseDoc): |
| 10 | token_id: int |
| 11 | generated_text: str |