adding final details

requirements.txt

@@ -3,4 +3,5 @@ textwrap3
 openai
 python-dotenv
 jsonlines
-pytest
+pytest
+numpy

src/chat.py

@@ -3,7 +3,8 @@ import openai
 from dotenv import load_dotenv
 from index import IndexSearchEngine
 from gpt_3_manager import Gpt3Manager
-from prompt import QuestionAnsweringPrompt, PassageSummarizationPrompt
+from prompt import QuestionAnsweringPrompt, PassageSummarizationPrompt, TextPromptLoader
+from pathlib import Path
 
 
 load_dotenv()
@@ -14,28 +15,38 @@ openai.api_key = OPENAI_API_KEY
 
 
 class ChatBot:
-    def __init__(self, index_search_engine: IndexSearchEngine):
+    def __init__(
+        self, index_search_engine: IndexSearchEngine, prompt_loader, gpt_manager
+    ):
         self.index_search_engine = index_search_engine
+        self.prompet_loader = prompt_loader
+        self.gpt_manager = gpt_manager
 
     def ask(self, question):
-        search_result = self.index_search_engine.search(question=question)
+        search_result = self.index_search_engine.search(question=question, count=2)
 
         answers = []
         for result in search_result:
             print("iterating over answering questions")
-
-            question_answering_prompt = QuestionAnsweringPrompt.load(
-                "prompts\question_answering.txt"
+            question_answering_prompt = QuestionAnsweringPrompt(
+                passage=result, question=question, prompt_loader=self.prompet_loader
+            )
+            prompt = question_answering_prompt.load(
+                Path("prompts") / "question_answering.txt"
             )
 
-            answer = Gpt3Manager.get_completion(
-                prompt=question_answering_prompt, max_tokens=80, model="text-curie-001"
+            answer = self.gpt_manager.get_completion(
+                prompt=prompt, max_tokens=80, model="text-curie-001"
             )
             answers.append(answer)
 
-        passage_summarization_prompt = PassageSummarizationPrompt.load(
-            "prompts\passage_summarization.txt"
+        passage_summarization_prompt = PassageSummarizationPrompt(
+            "\n".join(answers), self.prompet_loader
+        )
+
+        prompt = passage_summarization_prompt.load(
+            Path("prompts") / "passage_summarization.txt"
         )
 
-        final_answer = Gpt3Manager.get_completion(prompt=passage_summarization_prompt)
+        final_answer = self.gpt_manager.get_completion(prompt=prompt)
         return final_answer

src/index.py

@@ -1,6 +1,5 @@
 from abc import ABC, abstractmethod
 import jsonlines
-from gpt_3_manager import Gpt3Manager
 from utils import dot_similarity
 
 
@@ -35,4 +34,4 @@ class IndexSearchEngine:
                 simmilarities, key=lambda x: x["score"], reverse=True
             )
 
-        return sorted_similarities[:count]
+        return [result["index"]["content"] for result in sorted_similarities[:count]]

src/main.py

@@ -0,0 +1,29 @@
+import os
+from pathlib import Path
+
+from index import IndexSearchEngine
+from gpt_3_manager import Gpt3Manager
+
+from dotenv import load_dotenv
+from chat import ChatBot
+from index import JsonLinesIndex
+
+from prompt import TextPromptLoader
+
+load_dotenv()
+
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+
+
+path = Path("index") / "index.jsonl"
+
+index = JsonLinesIndex()
+loaded = index.load(path)
+gpt_manager = Gpt3Manager(api_key=OPENAI_API_KEY)
+
+engine = IndexSearchEngine(loaded, gpt_manager=gpt_manager)
+loader = TextPromptLoader()
+chatbot = ChatBot(engine, prompt_loader=loader, gpt_manager=gpt_manager)
+
+answer = chatbot.ask("What does the twitter terms of service does")
+print(answer)

src/prompt.py

@@ -3,7 +3,7 @@ from abc import ABC, abstractmethod
 # Prompt Loaders
 class PromptLoader(ABC):
     @abstractmethod
-    def load_prompt():
+    def load_prompt(self, path):
         pass
 
 
@@ -50,7 +50,7 @@ class PassageSummarizationPrompt(Prompt):
         super().__init__(prompt_loader=prompt_loader)
         self.passage = passage
 
-    # prompt = self.load_prompt(path).replace("<<PASSAGE>>", "\n".join(self.answers))
+    # prompt = self.load_prompt(path).replace("<<PASSAGE>>", )
 
     def load(self, path):
         prompt = self.load_prompt(path).replace("<<PASSAGE>>", self.passage)

src/tests/__init__.py

@@ -1 +0,0 @@
-

src/tests/chat_test.py

@@ -1,28 +1,31 @@
 import os
 from pathlib import Path
+
 from index import IndexSearchEngine
 from gpt_3_manager import Gpt3Manager
+
 from dotenv import load_dotenv
 from chat import ChatBot
 from index import JsonLinesIndex
 
-# load_dotenv()
+from prompt import TextPromptLoader
+
+load_dotenv()
 
-# OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 
 
-# def test_chatbot():
-#     path = Path("index") / "index.jsonl"
+def test_chatbot():
+    path = Path("index") / "index.jsonl"
 
-#     index = JsonLinesIndex()
-#     loaded = index.load(path)
-#     gpt_manager = Gpt3Manager(api_key=OPENAI_API_KEY)
-#     engine = IndexSearchEngine(loaded, gpt_manager=gpt_manager)
+    index = JsonLinesIndex()
+    loaded = index.load(path)
+    gpt_manager = Gpt3Manager(api_key=OPENAI_API_KEY)
 
-#     chatbot = ChatBot(engine)
-#     answer = chatbot.ask("What does the twitter terms of service does")
-#     print(answer)
-#     # assert 0 == 0
+    engine = IndexSearchEngine(loaded, gpt_manager=gpt_manager)
+    loader = TextPromptLoader()
+    chatbot = ChatBot(engine, prompt_loader=loader, gpt_manager=gpt_manager)
 
+    answer = chatbot.ask("What does the twitter terms of service does")
 
-# test_chatbot()
+    assert answer != None