Update app.py
Browse files
app.py
CHANGED
@@ -104,7 +104,7 @@ vectordb = Chroma.from_documents(
|
|
104 |
# define retriever
|
105 |
retriever = vectordb.as_retriever(search_kwargs={"k": 2}, search_type="mmr")
|
106 |
|
107 |
-
class FinalAnswer(
|
108 |
question: str = Field(description="the original question")
|
109 |
answer: str = Field(description="the extracted answer")
|
110 |
|
|
|
104 |
# define retriever
|
105 |
retriever = vectordb.as_retriever(search_kwargs={"k": 2}, search_type="mmr")
|
106 |
|
107 |
+
class FinalAnswer(BaseModel):
|
108 |
question: str = Field(description="the original question")
|
109 |
answer: str = Field(description="the extracted answer")
|
110 |
|