from CustomAgents.DocsAgent import DocsAgent
from agentforge.utils.ChromaUtils import ChromaUtils
from agentforge.utils.function_utils import Logger

docs_agent = DocsAgent()
# kb = ChromaUtils()

while True:
    user_input = input("Welcome to the chat with docs!\nQuestion: ")
    results = kb.query_memory(collection_name="docs", query=user_input, num_results=5)
    print(results)
    response = docs_agent.run(docs=results['documents'], query=user_input, whatever='data')
    print(f"Agent: {response}")
    # print(f"Results: {results}")
