refractor agents (#100)

* refractor agents * minor cosmetic, add terminal ui for cli * pump to 0.3.4 * Add temporary path * fix unclose files in tests --------- Co-authored-by: trducng <trungduc1992@gmail.com>
2023-12-06 17:06:29 +07:00
parent d9e925eb75
commit 797df5a69c
21 changed files with 281 additions and 228 deletions
--- a/knowledgehub/base/schema.py
+++ b/knowledgehub/base/schema.py
@@ -114,6 +114,7 @@ class LLMInterface(AIMessage):
    completion_tokens: int = -1
    total_tokens: int = -1
    prompt_tokens: int = -1
+    total_cost: float = 0
    logits: list[list[float]] = Field(default_factory=list)
    messages: list[AIMessage] = Field(default_factory=list)