l

app/main.py

@@ -1,10 +1,8 @@
 import os
-# --- START: FIX FOR ALL CACHE PERMISSION ERRORS ---
-# We set the cache for DSPY to /tmp
-os.environ["DSPY_CACHE_DIR"] = "/tmp/.dspy_cache"
+# --- KEEP THE SENTENCE-TRANSFORMERS FIX ---
 # We set the HF cache to the folder we create in the Dockerfile
 os.environ["HF_HOME"] = "/code/.cache/huggingface"
-# --- END: FIX ---
+# --- REMOVED THE DSPY ENV VAR ---
 
 # --- ALL OTHER IMPORTS MUST BE BELOW THIS FIX ---
 import json
@@ -18,7 +16,8 @@ from typing import Dict, Any
 # We need to make sure the path is correct
 from app.services.guardrails import check_input_guardrail, check_output_guardrail
 from app.services.rag_pipeline import generate_solution
-from app.services.dspy_feedback import refine_solution_with_dspy
+# --- REMOVED THE DSPY IMPORT ---
+# from app.services.dspy_feedback import refine_solution_with_dspy 
 from app.schemas import (
     AskRequest, AskResponse, FeedbackRequest, FeedbackResponse
 )
@@ -61,8 +60,13 @@ async def ask_math_question(request: AskRequest):
 
 @app.post("/feedback/", response_model=FeedbackResponse, status_code=200)
 async def give_feedback(request: FeedbackRequest):
+    """
+    Endpoint to receive feedback.
+    The DSPy refinement has been removed.
+    """
     print(f"--- HITL: Received Feedback for {request.thread_id} ---")
     
+    # 1. Log the feedback
     try:
         feedback_entry = request.model_dump()
         feedback_entry["timestamp"] = datetime.utcnow().isoformat()
@@ -71,36 +75,22 @@ async def give_feedback(request: FeedbackRequest):
     except Exception as e:
         print(f"--- HITL: Error logging feedback: {e} ---")
 
+    # --- START: MODIFIED SECTION ---
+    # Since dspy is removed, we just log the feedback and
+    # return the original solution, no matter if it's "good" or "bad".
+    
     if request.rating == "bad" and request.feedback_text:
-        print(f"--- HITL: Rating is 'bad'. Generating refinement... ---")
-        try:
-            refined_solution = refine_solution_with_dspy(
-                question=request.question,
-                original_solution=request.original_solution,
-                user_feedback=request.feedback_text
-            )
-            
-            is_safe, message = check_output_guardrail(refined_solution)
-            if not is_safe:
-                raise HTTPException(status_code=500, detail=f"Refined output blocked: {message}")
-
-            return FeedbackResponse(
-                solution=message,
-                source="refined", # New source
-                thread_id=request.thread_id,
-                question=request.question
-            )
-        except Exception as e:
-            print(f"--- HITL: Error during refinement: {e} ---")
-            raise HTTPException(status_code=500, detail="Error processing feedback.")
+        print(f"--- HITL: Rating is 'bad'. Logging only (refinement disabled). ---")
+    else:
+        print("--- HITL: Rating is 'good'. Logging only. ---")
     
-    print("--- HITL: Rating is 'good'. Logging only. ---")
     return FeedbackResponse(
         solution=request.original_solution,
         source="feedback_logged",
         thread_id=request.thread_id,
         question=request.question
     )
+    # --- END: MODIFIED SECTION ---
 
 @app.get("/")
 def read_root():