vllora 0.1.23

AI gateway for managing and routing LLM requests - Govern, Secure, and Optimize your AI Traffic.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133

use actix_web::{web, HttpRequest, HttpResponse, Result};
use serde::{Deserialize, Serialize};
use uuid::Uuid;
use vllora_core::metadata::models::project::NewProjectDTO;
use vllora_core::metadata::pool::DbPool;
use vllora_core::metadata::services::project::ProjectServiceImpl;
use vllora_core::types::metadata::project::Project;
use vllora_core::types::metadata::services::project::ProjectService;

use vllora_core::ok_json;

#[derive(Deserialize)]
pub struct CreateProjectRequest {
    pub name: String,
    pub description: Option<String>,
    pub settings: Option<serde_json::Value>,
}

#[derive(Deserialize)]
pub struct UpdateProjectRequest {
    pub name: Option<String>,
    pub description: Option<String>,
    pub settings: Option<serde_json::Value>,
    pub is_default: Option<bool>,
}

#[derive(Serialize)]
pub struct GetProjectResponse {
    pub project: Project,
}

pub async fn list_projects(_req: HttpRequest, db_pool: web::Data<DbPool>) -> Result<HttpResponse> {
    let project_service = ProjectServiceImpl::new(db_pool.get_ref().clone());

    // Use a dummy owner_id for now (you might want to get this from auth context)
    let owner_id = Uuid::nil();

    ok_json!(project_service.list(owner_id))
}

pub async fn create_project(
    req: web::Json<CreateProjectRequest>,
    db_pool: web::Data<DbPool>,
) -> Result<HttpResponse> {
    let project_service = ProjectServiceImpl::new(db_pool.get_ref().clone());

    // Use a dummy owner_id for now (you might want to get this from auth context)
    let owner_id = Uuid::nil();

    let new_project = NewProjectDTO {
        name: req.name.clone(),
        description: req.description.clone(),
        settings: req.settings.clone(),
        private_model_prices: None,
        usage_limit: None,
    };

    Ok(project_service
        .create(new_project, owner_id)
        .map(|project| HttpResponse::Created().json(project))?)
}

pub async fn get_project(
    project_id: web::Path<Uuid>,
    db_pool: web::Data<DbPool>,
) -> Result<HttpResponse> {
    let project_id = project_id.into_inner();

    let project_service = ProjectServiceImpl::new(db_pool.get_ref().clone());

    // Use a dummy owner_id for now (you might want to get this from auth context)
    let owner_id = Uuid::nil();

    ok_json!(project_service
        .get_by_id(project_id, owner_id)
        .map(|project| GetProjectResponse { project }))
}

pub async fn delete_project(
    project_id: web::Path<Uuid>,
    db_pool: web::Data<DbPool>,
) -> Result<HttpResponse> {
    let project_id = project_id.into_inner();

    let project_service = ProjectServiceImpl::new(db_pool.get_ref().clone());

    // Use a dummy owner_id for now (you might want to get this from auth context)
    let owner_id = Uuid::nil();

    Ok(project_service
        .delete(project_id, owner_id)
        .map(|_result| {
            HttpResponse::Ok().json(serde_json::json!({
                "message": "Project deleted successfully"
            }))
        })?)
}

pub async fn update_project(
    project_id: web::Path<Uuid>,
    req: web::Json<UpdateProjectRequest>,
    db_pool: web::Data<DbPool>,
) -> Result<HttpResponse> {
    let project_id = project_id.into_inner();
    let project_service = ProjectServiceImpl::new(db_pool.get_ref().clone());

    // Use a dummy owner_id for now (you might want to get this from auth context)
    let owner_id = Uuid::nil();

    // Convert UpdateProjectRequest to UpdateProjectDTO
    let update_data = vllora_core::metadata::models::project::UpdateProjectDTO {
        name: req.name.clone(),
        description: req.description.clone(),
        settings: req.settings.clone(),
        is_default: req.is_default,
    };

    ok_json!(project_service.update(project_id, owner_id, update_data))
}

pub async fn set_default_project(
    project_id: web::Path<Uuid>,
    db_pool: web::Data<DbPool>,
) -> Result<HttpResponse> {
    let project_id = project_id.into_inner();

    let project_service = ProjectServiceImpl::new(db_pool.get_ref().clone());

    // Use a dummy owner_id for now (you might want to get this from auth context)
    let owner_id = Uuid::nil();

    ok_json!(project_service.set_default(project_id, owner_id))
}