[AUR-395, AUR-415] Adopt Example1 Injury pipeline; add .flow() for enabling bottom-up pipeline execution (#32)
* add example1/injury pipeline example * add dotenv * update various api
This commit is contained in:
@@ -1,25 +1,13 @@
|
||||
from dataclasses import dataclass, field
|
||||
from typing import List
|
||||
|
||||
from ..base import BaseComponent
|
||||
from pydantic import Field
|
||||
|
||||
from kotaemon.documents.base import Document
|
||||
|
||||
|
||||
@dataclass
|
||||
class LLMInterface:
|
||||
text: List[str]
|
||||
class LLMInterface(Document):
|
||||
candidates: List[str]
|
||||
completion_tokens: int = -1
|
||||
total_tokens: int = -1
|
||||
prompt_tokens: int = -1
|
||||
logits: List[List[float]] = field(default_factory=list)
|
||||
|
||||
|
||||
class PromptTemplate(BaseComponent):
|
||||
pass
|
||||
|
||||
|
||||
class Extract(BaseComponent):
|
||||
pass
|
||||
|
||||
|
||||
class PromptNode(BaseComponent):
|
||||
pass
|
||||
logits: List[List[float]] = Field(default_factory=list)
|
||||
|
Reference in New Issue
Block a user