Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -110,10 +110,7 @@ def main():
|
|
| 110 |
)
|
| 111 |
|
| 112 |
retriever = create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type=search_type, k=k, chunk_size=550, chunk_overlap=40)
|
| 113 |
-
|
| 114 |
-
toxicity_grader = toxicity_grader()
|
| 115 |
-
hallucination_grader = hallucination_grader()
|
| 116 |
-
answer_grader = answer_grader()
|
| 117 |
|
| 118 |
|
| 119 |
# Graph
|
|
@@ -122,7 +119,7 @@ def main():
|
|
| 122 |
# Define the nodes
|
| 123 |
workflow.add_node("ask_question", lambda state: ask_question(state, retriever))
|
| 124 |
workflow.add_node("retrieve", lambda state: retrieve(state, retriever))
|
| 125 |
-
workflow.add_node("grade_documents", lambda state: grade_documents(state, retrieval_grader )) # grade documents
|
| 126 |
workflow.add_node("generate", generate) # generatae
|
| 127 |
workflow.add_node("web_search", web_search) # web search
|
| 128 |
workflow.add_node("transform_query", transform_query)
|
|
@@ -132,7 +129,7 @@ def main():
|
|
| 132 |
workflow.set_entry_point("ask_question")
|
| 133 |
workflow.add_conditional_edges(
|
| 134 |
"ask_question",
|
| 135 |
-
lambda state: grade_question_toxicity(state, toxicity_grader),
|
| 136 |
|
| 137 |
{
|
| 138 |
"good": "retrieve",
|
|
@@ -154,7 +151,7 @@ def main():
|
|
| 154 |
workflow.add_edge("web_search", "generate")
|
| 155 |
workflow.add_conditional_edges(
|
| 156 |
"generate",
|
| 157 |
-
lambda state: grade_generation_v_documents_and_question(state, hallucination_grader, answer_grader),
|
| 158 |
{
|
| 159 |
"not supported": "generate",
|
| 160 |
"useful": END,
|
|
|
|
| 110 |
)
|
| 111 |
|
| 112 |
retriever = create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type=search_type, k=k, chunk_size=550, chunk_overlap=40)
|
| 113 |
+
|
|
|
|
|
|
|
|
|
|
| 114 |
|
| 115 |
|
| 116 |
# Graph
|
|
|
|
| 119 |
# Define the nodes
|
| 120 |
workflow.add_node("ask_question", lambda state: ask_question(state, retriever))
|
| 121 |
workflow.add_node("retrieve", lambda state: retrieve(state, retriever))
|
| 122 |
+
workflow.add_node("grade_documents", lambda state: grade_documents(state, retrieval_grader() )) # grade documents
|
| 123 |
workflow.add_node("generate", generate) # generatae
|
| 124 |
workflow.add_node("web_search", web_search) # web search
|
| 125 |
workflow.add_node("transform_query", transform_query)
|
|
|
|
| 129 |
workflow.set_entry_point("ask_question")
|
| 130 |
workflow.add_conditional_edges(
|
| 131 |
"ask_question",
|
| 132 |
+
lambda state: grade_question_toxicity(state, toxicity_grader()),
|
| 133 |
|
| 134 |
{
|
| 135 |
"good": "retrieve",
|
|
|
|
| 151 |
workflow.add_edge("web_search", "generate")
|
| 152 |
workflow.add_conditional_edges(
|
| 153 |
"generate",
|
| 154 |
+
lambda state: grade_generation_v_documents_and_question(state, hallucination_grader(), answer_grader()),
|
| 155 |
{
|
| 156 |
"not supported": "generate",
|
| 157 |
"useful": END,
|