Update agents/rlhf_workflows.py

#4
Files changed (1) hide show
  1. agents/rlhf_workflows.py +3 -2
agents/rlhf_workflows.py CHANGED
@@ -60,7 +60,7 @@ def get_rlhf_manager() -> RLHFWorkflowManager:
60
  """Get the RLHF workflow manager instance"""
61
  return RLHFWorkflowManager()
62
 
63
- def run_rlhf_workflow(file_path: str, kind: str) -> Dict[str, Any]:
64
  """Run RLHF-enhanced workflow (placeholder - simplified)"""
65
  # For now, just return a basic structure
66
  # This can be enhanced later with actual RLHF logic
@@ -68,5 +68,6 @@ def run_rlhf_workflow(file_path: str, kind: str) -> Dict[str, Any]:
68
  "status": "error",
69
  "error": "RLHF workflow not implemented for this endpoint",
70
  "file_path": file_path,
71
- "kind": kind
 
72
  }
 
60
  """Get the RLHF workflow manager instance"""
61
  return RLHFWorkflowManager()
62
 
63
+ def run_rlhf_workflow(file_path: str, kind: str, user_api_key: Optional[str] = None) -> Dict[str, Any]:
64
  """Run RLHF-enhanced workflow (placeholder - simplified)"""
65
  # For now, just return a basic structure
66
  # This can be enhanced later with actual RLHF logic
 
68
  "status": "error",
69
  "error": "RLHF workflow not implemented for this endpoint",
70
  "file_path": file_path,
71
+ "kind": kind,
72
+ "user_api_key": user_api_key
73
  }