responder.py

class Responder():
    def __init__(self, collection) -> None:
        self.llm = OpenaiLanguageModel(anonymize=False)
        self.collection = collection

    def __call__(self, question) -> Any:
        results = self.collection.query(
                    query_texts=["This is a query document"],
                    n_results=10
                )

        # Joining all the text content of documents into one string
        merged_text = ' '.join([doc for doc in results['documents'][0]])

        prompt = f"""
        Answer this question: {question}, using these informations from the document: {merged_text}
        """

        response = self.llm.generate(prompt)

        return response