Spaces:
No application file
No application file
File size: 1,601 Bytes
5a1b165 fa46f05 bcd324a 5a1b165 fa46f05 5a1b165 fa46f05 5a1b165 fa46f05 5a1b165 fa46f05 bcd324a 5a1b165 fa46f05 5a1b165 fa46f05 bcd324a 5a1b165 fa46f05 5a1b165 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
import os
import openai
from pathlib import Path
from dotenv import load_dotenv
from TwitterChatBot.index import IndexSearchEngine
from TwitterChatBot.prompt import (
QuestionAnsweringPrompt,
PassageSummarizationPrompt,
TextPromptLoader,
)
load_dotenv()
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
openai.api_key = OPENAI_API_KEY
class ChatBot:
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, count=2)
answers = []
for result in search_result:
question_answering_prompt = QuestionAnsweringPrompt(
passage=result, question=question, prompt_loader=self.prompet_loader
)
prompt = question_answering_prompt.load(
Path("../prompts") / "question_answering.txt"
)
answer = self.gpt_manager.get_completion(
prompt=prompt, max_tokens=80, model="text-curie-001"
)
answers.append(answer)
passage_summarization_prompt = PassageSummarizationPrompt(
"\n".join(answers), self.prompet_loader
)
prompt = passage_summarization_prompt.load(
Path("../prompts") / "passage_summarization.txt"
)
final_answer = self.gpt_manager.get_completion(prompt=prompt)
return final_answer
|