[AUR-361] Setup pre-commit, pytest, GitHub actions, ssh-secret (#3)

Co-authored-by: trducng <trungduc1992@gmail.com>
This commit is contained in:
ian_Cin
2023-08-30 07:22:01 +07:00
committed by GitHub
parent c3c25db48c
commit 5241edbc46
19 changed files with 268 additions and 54 deletions

View File

@@ -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):