ArturG9 commited on
Commit
fc71a07
1 Parent(s): 49a3241

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -7
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
- retrieval_grader = retrieval_grader()
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,