Spaces:
Sleeping
Sleeping
Update agents/rlhf_workflows.py
#4
by
sahil-1-garg
- opened
- 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 |
}
|