More entities fixes

database/query.py

@@ -1,5 +1,5 @@
 import os
-from typing import List, Dict, Optional
+from typing import List, Dict, Optional,Tuple
 from datetime import datetime
 import psycopg2
 from psycopg2 import sql
@@ -19,7 +19,7 @@ class DatabaseService:
         start_date: Optional[datetime] = None,
         end_date: Optional[datetime] = None,
         topic: Optional[str] = None,
-        entities: Optional[List[tuple[str,str]]] = None,
+        entities: Optional[List[Tuple[str,str]]] = None,
         limit: int = 10
     ) -> List[Dict[str, any]]:
         
@@ -152,8 +152,7 @@ class DatabaseService:
                         }
                         for content, distance, art_date, art_topic,url in articles
                     ]
-                    cursor.close()
-                    conn.close()
+
                     return formatted_results
 
         except Exception as e:

database/query_processor.py

@@ -87,13 +87,12 @@ class QueryProcessor:
     ) -> List[Dict[str, Any]]:
         """Execute search with proper error handling"""
         try:
-            entity_texts = [ent[0] for ent in entities]
             return await self.db_service.semantic_search(
                 query_embedding=query_embedding,
                 start_date=start_date,
                 end_date=end_date,
                 topic=topic,
-                entities=entity_texts
+                entities=entities
             )
         except Exception as e:
             logger.error(f"Semantic search failed: {str(e)}")