5a9d6f75be
- Migrate the MVP into kotaemon. - Preliminary include the pipeline within chatbot interface. - Organize MVP as an application. Todo: - Add an info panel to view the planning of agents -> Fix streaming agents' output. Resolve: #60 Resolve: #61 Resolve: #62
44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
from abc import abstractmethod
|
|
from typing import Iterator
|
|
|
|
from theflow import Function, Node, Param, lazy
|
|
|
|
from kotaemon.base.schema import Document
|
|
|
|
|
|
class BaseComponent(Function):
|
|
"""A component is a class that can be used to compose a pipeline.
|
|
|
|
!!! tip "Benefits of component"
|
|
- Auto caching, logging
|
|
- Allow deployment
|
|
|
|
!!! tip "For each component, the spirit is"
|
|
- Tolerate multiple input types, e.g. str, Document, List[str], List[Document]
|
|
- Enforce single output type. Hence, the output type of a component should be
|
|
as generic as possible.
|
|
"""
|
|
|
|
inflow = None
|
|
|
|
def flow(self):
|
|
if self.inflow is None:
|
|
raise ValueError("No inflow provided.")
|
|
|
|
if not isinstance(self.inflow, BaseComponent):
|
|
raise ValueError(
|
|
f"inflow must be a BaseComponent, found {type(self.inflow)}"
|
|
)
|
|
|
|
return self.__call__(self.inflow.flow())
|
|
|
|
@abstractmethod
|
|
def run(
|
|
self, *args, **kwargs
|
|
) -> Document | list[Document] | Iterator[Document] | None:
|
|
"""Run the component."""
|
|
...
|
|
|
|
|
|
__all__ = ["BaseComponent", "Param", "Node", "lazy"]
|