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,
|