deterministic query/file handling

app/main.py

@@ -212,15 +212,46 @@ def generate_node(state: GraphState) -> GraphState:
         })
         return {"result": f"Error: {str(e)}", "metadata": metadata}
 
-workflow = StateGraph(GraphState)
-workflow.add_node("ingest", ingest_node)
-workflow.add_node("retrieve", retrieve_node)
-workflow.add_node("generate", generate_node)
-workflow.add_edge(START, "ingest")
-workflow.add_edge("ingest", "retrieve")
-workflow.add_edge("retrieve", "generate")
-workflow.add_edge("generate", END)
-compiled_graph = workflow.compile()
+def file_only_node(state: GraphState) -> GraphState:
+    """Return ingestor result directly without calling generator"""
+    logger.info("File-only processing: returning ingestor result directly")
+    
+    ingestor_context = state.get("ingestor_context", "")
+    metadata = state.get("metadata", {})
+    metadata.update({
+        "processing_type": "file_only",
+        "result_source": "ingestor"
+    })
+    
+    return {
+        "result": ingestor_context,
+        "metadata": metadata
+    }
+
+# Create separate workflows for different processing types
+def create_file_workflow():
+    """Workflow for file uploads: ingest -> file_only (skip retrieve and generate)"""
+    workflow = StateGraph(GraphState)
+    workflow.add_node("ingest", ingest_node)
+    workflow.add_node("file_only", file_only_node)
+    workflow.add_edge(START, "ingest")
+    workflow.add_edge("ingest", "file_only")
+    workflow.add_edge("file_only", END)
+    return workflow.compile()
+
+def create_query_workflow():
+    """Workflow for queries: retrieve -> generate (skip ingest)"""
+    workflow = StateGraph(GraphState)
+    workflow.add_node("retrieve", retrieve_node)
+    workflow.add_node("generate", generate_node)
+    workflow.add_edge(START, "retrieve")
+    workflow.add_edge("retrieve", "generate")
+    workflow.add_edge("generate", END)
+    return workflow.compile()
+
+# Compile workflows
+file_workflow = create_file_workflow()
+query_workflow = create_query_workflow()
 
 def process_query_core(
     query: str,
@@ -253,7 +284,14 @@ def process_query_core(
             }
         }
         
-        final_state = compiled_graph.invoke(initial_state)
+        # Choose workflow based on whether file is provided
+        if file_content and filename:
+            logger.info("File provided - using file workflow (ingest -> file_only)")
+            final_state = file_workflow.invoke(initial_state)
+        else:
+            logger.info("No file provided - using query workflow (retrieve -> generate)")
+            final_state = query_workflow.invoke(initial_state)
+        
         total_duration = (datetime.now() - start_time).total_seconds()
         
         final_metadata = final_state.get("metadata", {})