fix #584

mediar-ai · Nov 9, 2024 · 7d85f5d · 7d85f5d
1 parent f7ca07b
commit 7d85f5d
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 3 deletions.
diff --git a/screenpipe-app-tauri/components/search-chat.tsx b/screenpipe-app-tauri/components/search-chat.tsx
@@ -515,6 +515,12 @@ export function SearchChat({
       return;
     }
 
+    // Track AI usage metrics
+    posthog.capture("ai_chat_usage", {
+      agent: selectedAgent.name,
+      total_chars: floatingInput.length + selectedContentLength
+    });
+
     const userMessage = {
       id: generateId(),
       role: "user" as const,

diff --git a/screenpipe-core/src/language.rs b/screenpipe-core/src/language.rs
@@ -2,7 +2,7 @@ use clap::ValueEnum;
 use serde::Serialize;
 use std::fmt;
 
-#[derive(ValueEnum, Clone, Debug, Serialize)]
+#[derive(ValueEnum, Clone, Debug, Serialize, Hash, Eq, PartialEq)]
 #[serde(rename_all = "kebab-case")]
 #[repr(usize)]
 pub enum Language {

diff --git a/screenpipe-server/src/bin/screenpipe-server.rs b/screenpipe-server/src/bin/screenpipe-server.rs
@@ -321,7 +321,8 @@ async fn main() -> anyhow::Result<()> {
         cli.monitor_id.clone()
     };
 
-    let languages = cli.language.clone();
+    let languages = cli.unique_languages().unwrap();
+    let languages_clone = languages.clone();
 
     let ocr_engine_clone = cli.ocr_engine.clone();
     let vad_engine = cli.vad_engine.clone();
@@ -535,7 +536,7 @@ async fn main() -> anyhow::Result<()> {
         println!("│ {:<19} │ {:<34} │", "", "all languages");
     } else {
         let total_languages = cli.language.len();
-        for (_, language) in cli.language.iter().enumerate().take(MAX_ITEMS_TO_DISPLAY) {
+        for (_, language) in languages_clone.iter().enumerate().take(MAX_ITEMS_TO_DISPLAY) {
             let language_str = format!("id: {}", language);
             let formatted_language = format_cell(&language_str, VALUE_WIDTH);
             println!("│ {:<19} │ {:<34} │", "", formatted_language);

diff --git a/screenpipe-server/src/cli.rs b/screenpipe-server/src/cli.rs
@@ -248,6 +248,20 @@ pub struct Cli {
 
 }
 
+
+
+impl Cli {
+    pub fn unique_languages(&self) -> Result<Vec<Language>, String> {
+        let mut unique_langs = std::collections::HashSet::new();
+        for lang in &self.language {
+            if !unique_langs.insert(lang.clone()) {
+                // continue don't care
+            }
+        }
+        Ok(unique_langs.into_iter().collect())
+    }
+}
+
 #[derive(Subcommand)]
 pub enum Command {
     /// Pipe management commands