diff --git a/ai-hub/app/core/orchestration/harness_evaluator.py b/ai-hub/app/core/orchestration/harness_evaluator.py index 7d2c8e3..1d33a20 100644 --- a/ai-hub/app/core/orchestration/harness_evaluator.py +++ b/ai-hub/app/core/orchestration/harness_evaluator.py @@ -135,7 +135,9 @@ ): if event["type"] == "content": final_answer += event["content"] - if self.assistant: self.assistant.write(self.mesh_node_id, ".cortex/feedback.md", f"# Audit in Progress...\n\n{final_answer}", session_id=self.sync_workspace_id) + + if self.assistant: + self.assistant.write(self.mesh_node_id, ".cortex/feedback.md", f"# Finalizing Audit...\n\n{final_answer}", session_id=self.sync_workspace_id) score_match = FINAL_SCORE.search(final_answer) return {"score": int(score_match.group(1)) if score_match else 0, "justification": final_answer} diff --git a/ai-hub/app/core/services/rag.py b/ai-hub/app/core/services/rag.py index 245b412..8d90b11 100644 --- a/ai-hub/app/core/services/rag.py +++ b/ai-hub/app/core/services/rag.py @@ -35,10 +35,11 @@ provider_name: str, load_faiss_retriever: bool = False, user_service=None, - user_id: Optional[str] = None + user_id: Optional[str] = None, + save_prompt: bool = True ) -> AsyncGenerator[Dict[str, Any], None]: """Entry point for the RAG pipeline.""" - session = self._resolve_session(db, session_id, prompt) + session = self._resolve_session(db, session_id, prompt, save_prompt=save_prompt) llm_provider, resolved_provider_name = self._resolve_provider(db, session, provider_name, user_service) context_chunks = []