oskardotid commited on
Commit
8f5fc25
·
1 Parent(s): 288b789

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -26
app.py CHANGED
@@ -52,32 +52,6 @@ User Query:
52
 
53
  user_prompt = UserRolePrompt(USER_PROMPT_TEMPLATE)
54
 
55
- class RetrievalAugmentedQAPipeline:
56
- def __init__(self, llm: ChatOpenAI(), vector_db_retriever: VectorDatabase) -> None:
57
- self.llm = llm
58
- self.vector_db_retriever = vector_db_retriever
59
-
60
- def run_pipeline(self, user_query: str) -> str:
61
- context_list = self.vector_db_retriever.search_by_text(user_query, k=4)
62
-
63
- context_prompt = ""
64
- for context in context_list:
65
- context_prompt += context[0] + "\n"
66
-
67
- formatted_system_prompt = raqa_prompt.create_message(context=context_prompt)
68
-
69
- formatted_user_prompt = user_prompt.create_message(user_query=user_query)
70
-
71
- return self.llm.run([formatted_system_prompt, formatted_user_prompt])
72
-
73
-
74
- chat_openai = ChatOpenAI()
75
-
76
- retrieval_augmented_qa_pipeline = RetrievalAugmentedQAPipeline(
77
- vector_db_retriever=vector_db,
78
- llm=chat_openai
79
- )
80
-
81
  @cl.on_chat_start
82
  async def start_chat():
83
  settings = {
 
52
 
53
  user_prompt = UserRolePrompt(USER_PROMPT_TEMPLATE)
54
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
55
  @cl.on_chat_start
56
  async def start_chat():
57
  settings = {