diff --git a/api/ai/ai_response_model.py b/api/ai/ai_response_model.py index 90ab155..ee1029e 100644 --- a/api/ai/ai_response_model.py +++ b/api/ai/ai_response_model.py @@ -30,16 +30,8 @@ class StreamRequest(BaseModel): prompt: List[str] = Field(min_length=1) model: str = Field(min_length=1) -class ResultMetadata(BaseModel): - batch_id: str - model_used: str - text_type: str - - -class TranslationResult(BaseModel): - input_text: str +class WorkflowResult(BaseModel): output_text: str - metadata: ResultMetadata class ResponseMetadata(BaseModel): @@ -50,6 +42,6 @@ class ResponseMetadata(BaseModel): class StreamResponse(BaseModel): - results: List[TranslationResult] + results: List[WorkflowResult] metadata: ResponseMetadata errors: List[Any] \ No newline at end of file diff --git a/api/ai/ai_service.py b/api/ai/ai_service.py index 6388a5d..7bd08be 100644 --- a/api/ai/ai_service.py +++ b/api/ai/ai_service.py @@ -7,8 +7,7 @@ from api.ai.ai_response_model import ( WorkflowRequest, StreamResponse, - TranslationResult, - ResultMetadata, + WorkflowResult, ResponseMetadata, AvailableModelsResponse, ModelInfo @@ -55,15 +54,7 @@ async def run_workflow_service(assistant_id, target_language, prompt, model): workflow_response = await run_workflow(workflow_request) results = [ - TranslationResult( - input_text=result.input_text, - output_text=result.output_text, - metadata=ResultMetadata( - batch_id=result.metadata.get("batch_id"), - model_used=result.metadata.get("model_used"), - text_type=result.metadata.get("text_type") - ) - ) + WorkflowResult(output_text=result.output_text) for result in workflow_response.get("final_results", []) ] diff --git a/api/langgraph/workflow_stream.py b/api/langgraph/workflow_stream.py index e5c733f..433b8cc 100644 --- a/api/langgraph/workflow_stream.py +++ b/api/langgraph/workflow_stream.py @@ -86,15 +86,7 @@ async def stream_workflow_events( total_processing_time = time.time() - workflow_start_time results_data = [ - { - "input_text": result.input_text, - "output_text": result.output_text, - "metadata": { - "batch_id": result.metadata.get("batch_id"), - "model_used": result.metadata.get("model_used"), - "text_type": result.metadata.get("text_type") - } - } + {"output_text": result.output_text} for result in final_results ] diff --git a/api/ui.py b/api/ui.py index f2636e2..5df7e51 100644 --- a/api/ui.py +++ b/api/ui.py @@ -788,13 +788,7 @@ async def serve_ui(): if (m.results && m.results.length) { m.results.forEach(r => { html += `
`; - html += `
Input:
`; - html += `
${esc(r.input_text)}
`; - html += `
Output:
`; html += `
${esc(r.output_text)}
`; - if (r.metadata) { - html += `
Model: ${esc(r.metadata.model_used)} | Type: ${esc(r.metadata.text_type)}
`; - } html += `
`; }); }