Update CosmosDBHandlers/cosmosConnector.py

CosmosDBHandlers/cosmosConnector.py

@@ -24,7 +24,7 @@ class CosmosLampHandler:
         )
         self.chat_memory_handler = ChatMemoryHandler()
         self.database = self.client.get_database_client("TAL_DB")
-        self.container = self.database.get_container_client("Converters")
+        self.container = self.database.get_container_client("Converters_with_embeddings")
         self.logger = logging.Logger("test")
         # self.logger = logger
         self.embedding_model = AzureOpenAIEmbeddings(
@@ -289,14 +289,19 @@ class CosmosLampHandler:
             output_min, output_max = self._parse_voltage(output_voltage) if output_voltage else (None, None)
             normalized_lamp_type = self._normalize_lamp_name(lamp_type) if lamp_type else None
 
-            # Build query
             query_parts = []
-            if input_min and input_max:
-                query_parts.append(f"c.nom_input_voltage_v.min = {input_max} AND c.nom_input_voltage_v.max = {input_min}")
-                self.logger.info(f"c.nom_input_voltage_v.min = {input_max} AND c.nom_input_voltage_v.max = {input_min}")
             if output_min and output_max:
-                query_parts.append(f"c.output_voltage_v.min = {output_max} AND c.output_voltage_v.max = {output_min}")
-                self.logger.info(f"c.nom_input_voltage_v.min = {output_max} AND c.nom_input_voltage_v.max = {output_min}")
+                if output_min==output_max and output_max in [(24.0,24.0), (48.0,48.0)]:
+                    query_parts.append(f"c.output_voltage_v.min = {output_max} AND c.output_voltage_v.max = {output_min}")
+                    self.logger.info(f"c.nom_input_voltage_v.min = {output_max} AND c.nom_input_voltage_v.max = {output_min}")
+                else:
+                    query_parts.append(f"c.output_voltage_v.min <= {output_max} AND c.output_voltage_v.max >= {output_min}")
+                    self.logger.info(f"c.nom_input_voltage_v.min <= {output_max} AND c.nom_input_voltage_v.max >= {output_min}")
+
+            if input_min and input_max:
+                query_parts.append(f"c.nom_input_voltage_v.min <= {input_max} AND c.nom_input_voltage_v.max >= {input_min}")
+                self.logger.info(f"c.nom_input_voltage_v.min <= {input_max} AND c.nom_input_voltage_v.max >= {input_min}")
+            
             if current:
                  query_parts.append(f"c.type LIKE '%{current}%'")