after help
Browse files- .gitignore +1 -0
- Dockerfile +1 -1
- app.py +2 -2
.gitignore
CHANGED
|
@@ -1 +1,2 @@
|
|
| 1 |
__pycache__/
|
|
|
|
|
|
| 1 |
__pycache__/
|
| 2 |
+
.env
|
Dockerfile
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
FROM python:3.9
|
| 2 |
RUN useradd -m -u 1000 user
|
| 3 |
USER user
|
| 4 |
ENV HOME=/home/user \
|
|
|
|
| 1 |
+
FROM python:3.11.9
|
| 2 |
RUN useradd -m -u 1000 user
|
| 3 |
USER user
|
| 4 |
ENV HOME=/home/user \
|
app.py
CHANGED
|
@@ -27,7 +27,7 @@ Question:
|
|
| 27 |
user_role_prompt = UserRolePrompt(user_prompt_template)
|
| 28 |
|
| 29 |
class RetrievalAugmentedQAPipeline:
|
| 30 |
-
def __init__(self, llm
|
| 31 |
self.llm = llm
|
| 32 |
self.vector_db_retriever = vector_db_retriever
|
| 33 |
|
|
@@ -125,7 +125,7 @@ async def on_chat_start():
|
|
| 125 |
vector_db = VectorDatabase()
|
| 126 |
vector_db = await vector_db.abuild_from_list(texts)
|
| 127 |
|
| 128 |
-
|
| 129 |
|
| 130 |
# Create a chain
|
| 131 |
retrieval_augmented_qa_pipeline = RetrievalAugmentedQAPipeline(
|
|
|
|
| 27 |
user_role_prompt = UserRolePrompt(user_prompt_template)
|
| 28 |
|
| 29 |
class RetrievalAugmentedQAPipeline:
|
| 30 |
+
def __init__(self, llm, vector_db_retriever: VectorDatabase) -> None:
|
| 31 |
self.llm = llm
|
| 32 |
self.vector_db_retriever = vector_db_retriever
|
| 33 |
|
|
|
|
| 125 |
vector_db = VectorDatabase()
|
| 126 |
vector_db = await vector_db.abuild_from_list(texts)
|
| 127 |
|
| 128 |
+
chat_openai = ChatOpenAI()
|
| 129 |
|
| 130 |
# Create a chain
|
| 131 |
retrieval_augmented_qa_pipeline = RetrievalAugmentedQAPipeline(
|