fix: change dto schema and update some query logic

src/models/dtos/keyframes/mod.rs

@@ -4,7 +4,9 @@ use crate::models::entities::{keyframes::KeyframeEntity, videos::VideoEntity};
 pub struct KeyframeDto {
     id: i64,
     name: String,
+    video_id: i64,
     video_name: String,
+    watch_url: String,
     frame_index: i64,
     frame_timestamp: f32,
 }
@@ -13,13 +15,17 @@ impl From<(KeyframeEntity, VideoEntity)> for KeyframeDto {
     fn from((mut keyframe, mut video): (KeyframeEntity, VideoEntity)) -> Self {
         let id = keyframe.id();
         let name = std::mem::take(keyframe.name_mut());
+        let video_id = video.id();
         let video_name = std::mem::take(video.name_mut());
+        let watch_url = std::mem::take(video.watch_url_mut());
         let frame_index = keyframe.frame_index();
         let frame_timestamp = keyframe.frame_timestamp();
         Self {
             id,
             name,
+            video_id,
             video_name,
+            watch_url,
             frame_index,
             frame_timestamp,
         }

src/models/dtos/vectors/keyframes/requests/mod.rs

@@ -13,8 +13,4 @@ impl VectorizedKeyframeRequestDto {
     pub fn top_k(&self) -> u64 {
         self.top_k
     }
-    
-    // pub fn object_threshold(&self) -> f64 {
-    //     self.object_threshold
-    // }
 }

src/models/states/mod.rs

@@ -32,11 +32,13 @@ impl AppState {
     }
 
     async fn qdrant_client_helper() -> anyhow::Result<Arc<Qdrant>> {
-        let client = Qdrant::from_url(&format!(
-            "http://{}:{}",
-            std::env::var("QDRANT_HOST")?,
-            std::env::var("QDRANT_GRPC_PORT")?
-        ))
+        let client = Qdrant::from_url(
+            &std::env::var("QDRANT_URL").expect("`QDRANT_URL` environment variable must be set"),
+        )
+        .api_key(
+            std::env::var("QDRANT_API_KEY")
+                .expect("`QDRANT_API_KEY` environment variable must be set"),
+        )
         .build()?;
         Ok(Arc::new(client))
     }

src/services/queries/keyframes/mod.rs

@@ -1,6 +1,6 @@
 use qdrant_client::{
     Qdrant,
-    qdrant::{PrefetchQueryBuilder, Query, QueryPointsBuilder},
+    qdrant::{Query, QueryPointsBuilder},
 };
 use rayon::iter::{IntoParallelIterator, ParallelIterator};
 use translators::{GoogleTranslator, Translator};
@@ -44,14 +44,8 @@ impl<'a> VectorizedKeyframeService<'a> {
             .client
             .query(
                 QueryPointsBuilder::new(QDRANT_KEYFRAME_COLLECTION_NAME)
-                    .add_prefetch(
-                        PrefetchQueryBuilder::default()
-                            .query(Query::new_nearest(embeddings.clone()))
-                            .using("images")
-                            .limit(top_k * 10),
-                    )
                     .query(Query::new_nearest(embeddings))
-                    .using("objects")
+                    .using("images")
                     .limit(top_k),
             )
             .await?