diff --git a/src/shell_history_search/core/search.py b/src/shell_history_search/core/search.py
index 5408031..3303bfd 100644
--- a/src/shell_history_search/core/search.py
+++ b/src/shell_history_search/core/search.py
@@ -1,21 +1,12 @@
 import sqlite3
-from dataclasses import dataclass
 from pathlib import Path
 from typing import Optional
 
 from ..db import init_database, get_db_path
+from ..models import SearchResult
 from .embeddings import EmbeddingService
 
 
-@dataclass
-class SearchResult:
-    command: str
-    shell_type: str
-    timestamp: Optional[int]
-    similarity: float
-    command_id: int
-
-
 class SearchEngine:
     def __init__(
         self,
@@ -115,4 +106,4 @@ class SearchEngine:
         self._conn.execute("DELETE FROM embeddings")
         self._conn.execute("DELETE FROM commands")
         self._conn.execute("DELETE FROM index_state")
-        self._conn.commit()
\ No newline at end of file
+        self._conn.commit()