Move Document and other interface into base/schema (#69)

knowledgehub/llms/base.py

@@ -1,13 +0,0 @@
-from typing import List
-
-from pydantic import Field
-
-from kotaemon.documents.base import Document
-
-
-class LLMInterface(Document):
-    candidates: List[str] = Field(default_factory=list)
-    completion_tokens: int = -1
-    total_tokens: int = -1
-    prompt_tokens: int = -1
-    logits: List[List[float]] = Field(default_factory=list)

knowledgehub/llms/chats/base.py

@@ -8,7 +8,7 @@ from langchain.schema.messages import BaseMessage, HumanMessage
 from theflow.base import Param
 
 from ...base import BaseComponent
-from ..base import LLMInterface
+from ...base.schema import LLMInterface
 
 logger = logging.getLogger(__name__)
 

knowledgehub/llms/completions/base.py

@@ -5,7 +5,7 @@ from langchain.llms.base import BaseLLM
 from theflow.base import Param
 
 from ...base import BaseComponent
-from ..base import LLMInterface
+from ...base.schema import LLMInterface
 
 logger = logging.getLogger(__name__)