23 lines
400 B
Python
23 lines
400 B
Python
|
"""Type definitions"""
|
||
|
|
|
||
|
|
from dataclasses import dataclass
|
||
|
|
from typing import Optional
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class ChatResult:
|
||
|
|
success: bool
|
||
|
|
text_response: str
|
||
|
|
# TODO: switch stats to tokens instead of words!!!
|
||
|
|
words_sent: int
|
||
|
|
words_received: int
|
||
|
|
|
||
|
|
|
||
|
|
class Llm:
|
||
|
|
def chat(
|
||
|
|
self,
|
||
|
|
message: str,
|
||
|
|
system_message: Optional[str])\
|
||
|
|
-> ChatResult:
|
||
|
|
pass
|