[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:
ian_Cin
2023-10-02 16:24:56 +07:00
committed by GitHub
parent 3cceec63ef
commit d83c22aa4e
16 changed files with 114 additions and 69 deletions

View File

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