[AUR-361] Setup pre-commit, pytest, GitHub actions, ssh-secret (#3)
Co-authored-by: trducng <trungduc1992@gmail.com>
This commit is contained in:
@@ -1,15 +1,16 @@
|
||||
from dataclasses import dataclass, field
|
||||
from typing import List
|
||||
|
||||
from ..components import BaseComponent
|
||||
|
||||
|
||||
@dataclass
|
||||
class LLMInterface:
|
||||
text: list[str]
|
||||
text: List[str]
|
||||
completion_tokens: int = -1
|
||||
total_tokens: int = -1
|
||||
prompt_tokens: int = -1
|
||||
logits: list[list[float]] = field(default_factory=list)
|
||||
logits: List[List[float]] = field(default_factory=list)
|
||||
|
||||
|
||||
class PromptTemplate(BaseComponent):
|
||||
|
Reference in New Issue
Block a user