Update utils.py

utils.py

@@ -360,7 +360,7 @@ def rag_chain(llm, prompt, vectorstore, retriever):
     workflow.add_node("retrieve", retrieve)  # retrieve
     workflow.add_node("grade_documents", grade_documents)  # grade documents
     workflow.add_node("generate", generate)  # generate
-    workflow.add_node("generate_ohne", generate)  # generate ohne dokumente anzuhängen
+    #workflow.add_node("generate_ohne", generate)  # generate ohne dokumente anzuhängen
     workflow.add_node("transform_query", transform_query)  # transform_query
     #momentan  nicht genutzt
     #workflow.add_node("web_search", web_search)  # web search
@@ -376,14 +376,14 @@ def rag_chain(llm, prompt, vectorstore, retriever):
         decide_to_generate,
         {
             "transform_query": "transform_query",
-            "generate_ohne": "generate_ohne",
+            #"generate_ohne": "generate_ohne",
             "generate": "generate",
         },
     )
     workflow.add_edge("transform_query", "retrieve_redirect")
     workflow.add_edge("retrieve_redirect", "retrieve")
     
-    workflow.add_edge("generate_ohne", "generate")
+    #workflow.add_edge("generate_ohne", "generate")
     workflow.add_edge("generate", END)
     # Compile
     app = workflow.compile()
@@ -398,7 +398,7 @@ def rag_chain(llm, prompt, vectorstore, retriever):
         pprint.pprint("\n---\n")
     
     # Final generation
-    pprint.pprint(value['keys']['generation'])
+    return value['keys']['generation']
 
 ############################################
 # rag_chain Alternative für RAg mit Bild-Upload, da hier das llm so nicht genutzt werden kann und der prompt mit den RAG Erweiterungen anders übergeben wird
@@ -1399,7 +1399,7 @@ def decide_to_generate(state):
         else:
             # keine neue frage, sondern generieren - ohne Dokumente anzuhängen
             print("---ENTSCHEIDUNG: Generiere ohne Dokumente---")
-            return "generate_ohne" 
+            return "generate" 
     else:
         # We have relevant documents, so generate answer
         print("---ENTSCHEIDUNG: GENERIERE---")