agent/Agent.py

from langchain_core.language_models import BaseChatModel

from emdedd.Embedding import Embedding


class Agent:
    embedding: Embedding
    llm: BaseChatModel

    def __init__(self, embedding, llm):
        self.embedding = embedding
        self.llm = llm