Spaces:

variphx
/

aic-server

Runtime error

aic-server / src /models /dtos /vectors /keyframes /mod.rs

add/fix: refactor some error handling and add keyframes semantic search functionalities

a8c4fcf about 1 year ago

880 Bytes

use qdrant_client::qdrant::{ScoredPoint, point_id::PointIdOptions};

mod requests;

pub use requests::VectorizedKeyframeRequestDto;

#[derive(Debug, Clone, Copy, serde::Serialize, utoipa::ToSchema)]
pub struct VectorizedKeyframeDto {
    id: u64,
    score: f32,
}

impl TryFrom<ScoredPoint> for VectorizedKeyframeDto {
    type Error = anyhow::Error;
    fn try_from(value: ScoredPoint) -> Result<Self, Self::Error> {
        Ok(Self {
            id: value
                .id
                .and_then(|x| x.point_id_options)
                .and_then(|x| {
                    if let PointIdOptions::Num(x) = x {
                        Some(x)
                    } else {
                        None
                    }
                })
                .ok_or_else(|| anyhow::anyhow!("scored point should have id"))?,
            score: value.score,
        })
    }
}

	use qdrant_client::qdrant::{ScoredPoint, point_id::PointIdOptions};

	mod requests;

	pub use requests::VectorizedKeyframeRequestDto;

	#[derive(Debug, Clone, Copy, serde::Serialize, utoipa::ToSchema)]
	pub struct VectorizedKeyframeDto {
	id: u64,
	score: f32,
	}

	impl TryFrom<ScoredPoint> for VectorizedKeyframeDto {
	type Error = anyhow::Error;
	fn try_from(value: ScoredPoint) -> Result<Self, Self::Error> {
	Ok(Self {
	id: value
	.id
	.and_then(\|x\| x.point_id_options)
	.and_then(\|x\| {
	if let PointIdOptions::Num(x) = x {
	Some(x)
	} else {
	None
	}
	})
	.ok_or_else(\|\| anyhow::anyhow!("scored point should have id"))?,
	score: value.score,
	})
	}
	}