Nurfog/openccb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: activate AI usage

Browse Source
This commit is contained in:
Juan Enrique Allende Cifuentes
2026-03-19 11:21:59 -03:00
parent a220e9742f
commit 686eb377f0
4 changed files with 182 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files
services/cms-service/src/handlers.rs
+75 -3
View File
@@ -1135,6 +1135,29 @@ pub async fn summarize_lesson(
.unwrap_or("")
.trim();
// Calculate and log token usage
let system_prompt = "You are an expert English Teacher. Summarize the following lesson content.";
let input_tokens = count_tokens(system_prompt) + count_tokens(&content_text);
let output_tokens = count_tokens(summary);
let total_tokens = input_tokens + output_tokens;
let _ = sqlx::query("SELECT log_ai_usage($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11)")
.bind(claims.sub)
.bind(org_ctx.id)
.bind(total_tokens)
.bind(input_tokens)
.bind(output_tokens)
.bind("/lessons/summarize")
.bind(&model)
.bind("summary")
.bind(&json!({
"lesson_id": id,
}))
.bind(&format!("{} - {}", system_prompt, content_text)) // prompt
.bind(summary) // response
.execute(&pool)
.await;
// 3. Update lesson
let updated_lesson =
sqlx::query_as::<_, Lesson>("UPDATE lessons SET summary = $1 WHERE id = $2 RETURNING *")
@@ -1274,8 +1297,8 @@ pub async fn generate_quiz(
let output_tokens = count_tokens(&response_json.to_string());
let total_tokens = input_tokens + output_tokens;
// Log AI usage
let _ = sqlx::query("SELECT log_ai_usage($1, $2, $3, $4, $5, $6, $7, $8, $9)")
// Log AI usage with prompt and response
let _ = sqlx::query("SELECT log_ai_usage($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11)")
.bind(claims.sub)
.bind(org_ctx.id)
.bind(total_tokens)
@@ -1288,6 +1311,8 @@ pub async fn generate_quiz(
"lesson_id": id,
"quiz_type": quiz_req.quiz_type,
}))
.bind(&system_prompt) // prompt
.bind(&response_json.to_string()) // response
.execute(&pool)
.await;
@@ -1804,7 +1829,7 @@ pub async fn generate_mermaid_diagram(
let transcription_str = lesson.transcription.as_ref().and_then(|v| v.as_str());
let summary_str = lesson.summary.as_deref();
let lesson_context = transcription_str.or(summary_str).unwrap_or("Conceptos generales de la lección.");
let user_hint = payload.prompt_hint.unwrap_or_else(|| "Extrae los conceptos y flujos principales de la lección.".to_string());
let user_hint = payload.prompt_hint.clone().unwrap_or_else(|| "Extrae los conceptos y flujos principales de la lección.".to_string());
let system_prompt = format!(
"Eres un experto arquitecto de información y especialista en diagramación usando Mermaid.js.\n\
@@ -1861,6 +1886,29 @@ pub async fn generate_mermaid_diagram(
.strip_prefix("```\n").unwrap_or(ai_response)
.strip_suffix("```").unwrap_or(ai_response).trim();
// Calculate and log token usage
let input_tokens = count_tokens(&system_prompt) + count_tokens("Genera el código Mermaid directamente.");
let output_tokens = count_tokens(cleaned_response);
let total_tokens = input_tokens + output_tokens;
let _ = sqlx::query("SELECT log_ai_usage($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11)")
.bind(_claims.sub)
.bind(org_ctx.id)
.bind(total_tokens)
.bind(input_tokens)
.bind(output_tokens)
.bind("/lessons/generate-mermaid")
.bind(&model)
.bind("diagram-generation")
.bind(&json!({
"lesson_id": lesson_id,
"hint": payload.prompt_hint,
}))
.bind(&system_prompt) // prompt
.bind(cleaned_response) // response
.execute(&pool)
.await;
Ok(Json(serde_json::json!({
"mermaid_code": cleaned_response,
})))
@@ -2116,6 +2164,30 @@ pub async fn generate_hotspots(
}
};
// Calculate and log token usage
let full_prompt = format!("{} - {}", system_prompt, user_prompt);
let input_tokens = count_tokens(&full_prompt) + 500; // Estimate for image tokens
let output_tokens = count_tokens(&hotspots.to_string());
let total_tokens = input_tokens + output_tokens;
let _ = sqlx::query("SELECT log_ai_usage($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11)")
.bind(_claims.sub)
.bind(org_ctx.id)
.bind(total_tokens)
.bind(input_tokens)
.bind(output_tokens)
.bind("/lessons/generate-hotspots")
.bind(&model)
.bind("hotspots-generation")
.bind(&json!({
"lesson_id": lesson_id,
"image_url": payload.image_url,
}))
.bind(&full_prompt) // prompt
.bind(&hotspots.to_string()) // response
.execute(&pool)
.await;
Ok(Json(hotspots))
}
services/cms-service/src/handlers_question_bank.rs
+38 -35
View File
@@ -135,9 +135,9 @@ fn calculate_course_type(plan_name: &str) -> String {
}
}
fn calculate_course_type_from_duration(duracion: Option<i32>) -> String {
fn calculate_course_type_from_duration(duracion: Option<f64>) -> String {
match duracion {
Some(d) if d >= 70 => "intensive".to_string(), // 80h or more = intensive
Some(d) if d as i64 >= 70 => "intensive".to_string(), // 80h or more = intensive
_ => "regular".to_string(), // 40h or less = regular
}
}
@@ -396,7 +396,12 @@ pub async fn import_from_mysql(
)
.fetch_all(&mysql_pool)
.await
.map_err(|e| (StatusCode::INTERNAL_SERVER_ERROR, format!("Failed to fetch plans: {}", e)))?;
.map_err(|e| {
let error_msg = format!("Failed to fetch: {}", e);
tracing::error!("MySQL Error: {}", error_msg);
tracing::error!("Check column names in your MySQL database (plandeestudios, curso tables)");
(StatusCode::INTERNAL_SERVER_ERROR, error_msg)
})?;
tracing::info!("Fetched {} study plans from MySQL", mysql_plans.len());
@@ -408,7 +413,7 @@ pub async fn import_from_mysql(
c.NivelCurso AS nivel_curso,
pe.idPlanDeEstudios AS id_plan_de_estudios,
pe.Nombre AS nombre_plan,
CAST(c.Duracion AS SIGNED INTEGER) AS duracion
c.Duracion AS duracion
FROM curso c
JOIN plandeestudios pe ON c.idPlanDeEstudios = pe.idPlanDeEstudios
WHERE c.Activo = 1
@@ -757,7 +762,7 @@ pub async fn list_mysql_courses(
c.NivelCurso AS nivel_curso,
pe.idPlanDeEstudios AS id_plan_de_estudios,
pe.Nombre AS nombre_plan,
CAST(c.Duracion AS SIGNED INTEGER) AS duracion
c.Duracion AS duracion
FROM curso c
JOIN plandeestudios pe ON c.idPlanDeEstudios = pe.idPlanDeEstudios
WHERE c.Activo = 1
@@ -783,8 +788,8 @@ pub async fn get_mysql_plans(
let plans: Vec<MySqlPlanInfo> = sqlx::query_as(
r#"
SELECT
mysql_id as "idPlanDeEstudios",
name as "NombrePlan"
mysql_id as id_plan_de_estudios,
name as nombre_plan
FROM mysql_study_plans
WHERE organization_id = $1 AND is_active = true
ORDER BY name
@@ -808,12 +813,12 @@ pub async fn get_mysql_courses_by_plan(
let courses: Vec<MySqlCourseInfo> = sqlx::query_as(
r#"
SELECT
c.mysql_id as "idCursos",
c.name as "NombreCurso",
c.level as "NivelCurso",
sp.mysql_id as "idPlanDeEstudios",
sp.name as "NombrePlan",
c.duracion as "Duracion"
c.mysql_id as id_cursos,
c.name as nombre_curso,
c.level as nivel_curso,
sp.mysql_id as id_plan_de_estudios,
sp.name as nombre_plan,
c.duracion as duracion
FROM mysql_courses c
JOIN mysql_study_plans sp ON c.study_plan_id = sp.id
WHERE c.organization_id = $1
@@ -838,34 +843,18 @@ pub struct MySqlCoursesFilters {
#[derive(Debug, Clone, sqlx::FromRow, Serialize)]
pub struct MySqlPlanInfo {
#[sqlx(rename = "idPlanDeEstudios")]
#[serde(rename = "idPlanDeEstudios")]
pub id_plan_de_estudios: i32,
#[sqlx(rename = "NombrePlan")]
#[serde(rename = "NombrePlan")]
pub nombre_plan: String,
}
#[derive(Debug, Clone, Serialize, Deserialize, sqlx::FromRow)]
pub struct MySqlCourseInfo {
#[sqlx(rename = "idCursos")]
#[serde(rename = "idCursos")]
pub id_cursos: i32,
#[sqlx(rename = "NombreCurso")]
#[serde(rename = "NombreCurso")]
pub nombre_curso: String,
#[sqlx(rename = "NivelCurso")]
#[serde(rename = "NivelCurso", skip_serializing_if = "Option::is_none")]
pub nivel_curso: Option<i32>,
#[sqlx(rename = "idPlanDeEstudios")]
#[serde(rename = "idPlanDeEstudios")]
pub id_plan_de_estudios: i32,
#[sqlx(rename = "NombrePlan")]
#[serde(rename = "NombrePlan")]
pub nombre_plan: String,
#[sqlx(rename = "Duracion")]
#[serde(rename = "Duracion", skip_serializing_if = "Option::is_none")]
pub duracion: Option<i32>, // Duration in hours (40=regular, 80=intensive)
pub duracion: Option<f64>, // Duration in hours (40=regular, 80=intensive) - MySQL float type
}
#[derive(Debug, Serialize, Deserialize)]
@@ -878,10 +867,19 @@ pub async fn import_all_from_mysql(
Org(org_ctx): Org,
claims: Claims,
State(pool): State<PgPool>,
Json(payload): Json<ImportAllFromMySQLPayload>,
body: String,
) -> Result<Json<serde_json::Value>, (StatusCode, String)> {
use serde_json::json;
// Parse optional JSON body
let import_metadata_only = if body.trim().is_empty() {
false
} else {
serde_json::from_str::<ImportAllFromMySQLPayload>(&body)
.map(|p| p.import_metadata_only.unwrap_or(false))
.unwrap_or(false)
};
// Connect to MySQL
let mysql_url = std::env::var("MYSQL_DATABASE_URL")
.map_err(|_| (StatusCode::INTERNAL_SERVER_ERROR, "MYSQL_DATABASE_URL not configured".to_string()))?;
@@ -903,7 +901,12 @@ pub async fn import_all_from_mysql(
)
.fetch_all(&mysql_pool)
.await
.map_err(|e| (StatusCode::INTERNAL_SERVER_ERROR, format!("Failed to fetch plans: {}", e)))?;
.map_err(|e| {
let error_msg = format!("Failed to fetch: {}", e);
tracing::error!("MySQL Error: {}", error_msg);
tracing::error!("Check column names in your MySQL database (plandeestudios, curso tables)");
(StatusCode::INTERNAL_SERVER_ERROR, error_msg)
})?;
tracing::info!("Fetched {} study plans from MySQL", mysql_plans.len());
@@ -915,7 +918,7 @@ pub async fn import_all_from_mysql(
c.NivelCurso AS nivel_curso,
pe.idPlanDeEstudios AS id_plan_de_estudios,
pe.Nombre AS nombre_plan,
CAST(c.Duracion AS SIGNED INTEGER) AS duracion
c.Duracion AS duracion
FROM curso c
JOIN plandeestudios pe ON c.idPlanDeEstudios = pe.idPlanDeEstudios
WHERE c.Activo = 1
@@ -936,7 +939,7 @@ pub async fn import_all_from_mysql(
.map_err(|e| (StatusCode::INTERNAL_SERVER_ERROR, format!("Failed to save courses/plans: {}", e)))?;
// If only metadata import is requested, return early
if payload.import_metadata_only.unwrap_or(false) {
if import_metadata_only {
return Ok(Json(json!({
"success": true,
"message": "Metadatos importados exitosamente",
@@ -1305,7 +1308,7 @@ pub async fn import_course_from_mysql(
c.NivelCurso AS nivel_curso,
pe.idPlanDeEstudios AS id_plan_de_estudios,
pe.Nombre AS nombre_plan,
CAST(c.Duracion AS SIGNED INTEGER) AS duracion
c.Duracion AS duracion
FROM curso c
JOIN plandeestudios pe ON c.idPlanDeEstudios = pe.idPlanDeEstudios
WHERE c.idCursos = ? AND c.Activo = 1 AND pe.Activo = 1