Update plugins/converterPlugin.py

plugins/converterPlugin.py

@@ -13,9 +13,12 @@ class ConverterPlugin:
         name="query_converters",
         description="Execute SQL query against Cosmos DB converters collection"
     )
-    async def query_converters(self, query: str) -> str:
+    async def query_converters(
+        self, 
+        user_input:Annotated[str,"Natural Language question the user asked"],
+        query: Annotated[str,"SQL generated from NL2SQL plugin"]) -> str:
         try:
-            items = await self.db.query_converters(query)
+            items = await self.db.query_converters(query, user_input)
             self.logger.info(f"Executed query: {query}")
             if not items:
                 return "No items found for the given query."
@@ -76,8 +79,8 @@ class ConverterPlugin:
     async def get_converters_by_dimming(
         self,
         dimming_type: Annotated[str, "Dimming type mentioned like dali, mains, 1-10v"],
-        voltage_current: Annotated[str | None,"Voltage or current specification like 350mA, 24V DC"] = None,
-        lamp_type: Annotated[str | None, "Lamp model (e.g., Haloled, B4)"] = None,
+        voltage_current: Annotated[str | None,"Voltage or current specification like 350mA, 24V DC"],
+        lamp_type: Annotated[str | None, "Lamp model (e.g., Haloled, B4)"],
         threshold: int = 75) -> str:
         """Search converters by dimming type with technical specifications"""
         try:
@@ -121,19 +124,22 @@ class ConverterPlugin:
     )
     async def get_converters_by_voltage_current(
         self,
-        artnr: Annotated[int, ""] = None,
-        current: Annotated[str, "Current like 350mA, 700mA"]=None,
-        input_voltage: Annotated[str, "Input voltage range like '198-464' NEVER ip"] = None,
-        output_voltage: Annotated[str, "Output voltage range like '24', '2-25'"] = None
+        artnr: Annotated[int | None, "Converter artnr"],
+        current: Annotated[str | None, "Current like 350mA, 700mA"],
+        input_voltage: Annotated[str | None, "Input voltage range like '198-464' NEVER ip, null if no voltage"],
+        output_voltage: Annotated[str | None, "Output voltage range like '24', '2-25' null if no voltage"],
+        lamp_type:  Annotated[str | None, "Lamp model (e.g., Haloled, B4)"],
     ) -> str:
         try:
             converters = await self.db.get_converters_by_voltage_current(artnr=artnr,
                                                                 current=current,
                                                                 input_voltage=input_voltage,
-                                                                output_voltage=output_voltage)
+                                                                output_voltage=output_voltage,
+                                                                lamp_type=lamp_type)
             self.logger.info(f"""Used get_converters_by_voltage_current with input_voltage: {input_voltage}
                                                                      output_voltage: {output_voltage}
                                                                      current: {current}
+                                                                     lamp_type: {lamp_type}
                                                                      artnr: {artnr}""")
             if not converters:
                 return "No relavent converters found"