Similarity fix and entity test

database/query.py

@@ -22,6 +22,8 @@ class DatabaseService:
         entities: Optional[List[tuple[str,str]]] = None,
         limit: int = 10
     ) -> List[Dict[str, any]]:
+        
+        print(f"Extracted entities2: {entities}")
         try:
             with psycopg2.connect(
                 user=self.DB_USER,
@@ -150,7 +152,8 @@ class DatabaseService:
                         }
                         for content, distance, art_date, art_topic,url in articles
                     ]
-
+                    cursor.close()
+                    conn.close()
                     return formatted_results
 
         except Exception as e:

models/embedding.py

@@ -7,4 +7,7 @@ class EmbeddingModel:
         self.model = SentenceTransformer("paraphrase-multilingual-MiniLM-L12-v2")
     
     def encode(self, text: str):
-        return self.model.encode(text, device=self.device)
+        return self.model.encode(text, device=self.device)
+
+    #def similarity(self, embeddings1, embeddings2):
+        return self.model.similarity(embeddings1, embeddings2)