ws-bom-robot-app 0.0.46__py3-none-any.whl → 0.0.47__py3-none-any.whl

ws_bom_robot_app/llm/utils/agent.py

@@ -5,11 +5,16 @@ from ws_bom_robot_app.llm.utils.print import HiddenPrints
 from ws_bom_robot_app.llm.vector_store.db.manager import VectorDbManager
 import warnings
 
-async def get_rules(embeddings: Embeddings, rules: LlmRules, input: str) -> str:
+async def get_rules(embeddings: Embeddings, rules: LlmRules, input: str | list) -> str:
   with warnings.catch_warnings():
     warnings.simplefilter("ignore", category=Warning)
+    # check if the input is multimodal and convert it to text
+    if isinstance(input, list):
+      input = " ".join(obj.get("text", "") for obj in input)
+    # check if the input is empty or the rules are not provided
     if any([input=="",rules is None,rules and rules.vector_db == "",rules and not os.path.exists(rules.vector_db)]):
       return ""
+    # get the rules from the vector db and return prompt with rules
     rules_prompt = ""
     rules_doc = await VectorDbManager.get_strategy(rules.vector_type).invoke(embeddings, rules.vector_db,input,search_type="similarity_score_threshold", search_kwargs={"score_threshold": rules.threshold}) #type: ignore
     if len(rules_doc) > 0:

{ws_bom_robot_app-0.0.46.dist-info → ws_bom_robot_app-0.0.47.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: ws_bom_robot_app
-Version: 0.0.46
+Version: 0.0.47
 Summary: A FastAPI application serving ws bom/robot/llm platform ai.
 Home-page: https://github.com/websolutespa/bom
 Author: Websolute Spa

{ws_bom_robot_app-0.0.46.dist-info → ws_bom_robot_app-0.0.47.dist-info}/RECORD

@@ -26,7 +26,7 @@ ws_bom_robot_app/llm/tools/utils.py,sha256=LEId1UolLvJsMYbl-awm7h-gJ9Up9DrrnH6HT
 ws_bom_robot_app/llm/tools/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ws_bom_robot_app/llm/tools/models/main.py,sha256=o3Rwbn5nsugKOgLG0FUIuvtPPHYhfVpqG4E3BQB2nWM,388
 ws_bom_robot_app/llm/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ws_bom_robot_app/llm/utils/agent.py,sha256=Tc89ZExE1aJ8cV1_3b9lfICCxgjKHHPPADs0fWi3NOo,1023
+ws_bom_robot_app/llm/utils/agent.py,sha256=_GdD8kfEzR-xNCkXv8gtFzjeVsHtuSi6_QO9vsePiNE,1322
 ws_bom_robot_app/llm/utils/chunker.py,sha256=N7570xBYlObneg-fsvDhPAJ-Pv8C8OaYZOBK6q7LmMI,607
 ws_bom_robot_app/llm/utils/download.py,sha256=GaRypPgkx16HfYRj-upX9kvmjfAdFFb5TP4P97scWeA,3273
 ws_bom_robot_app/llm/utils/kb.py,sha256=jja45WCbNI7SGEgqDS99nErlwB5eY8Ga7BMnhdMHZ90,1279
@@ -60,7 +60,7 @@ ws_bom_robot_app/llm/vector_store/loader/__init__.py,sha256=47DEQpj8HBSa-_TImW-5
 ws_bom_robot_app/llm/vector_store/loader/base.py,sha256=L_ugekNuAq0N9O-24wtlHSNHkqSeD-KsJrfGt_FX9Oc,5340
 ws_bom_robot_app/llm/vector_store/loader/docling.py,sha256=yP0zgXLeFAlByaYuj-6cYariuknckrFds0dxdRcnVz8,3456
 ws_bom_robot_app/llm/vector_store/loader/json_loader.py,sha256=LDppW0ZATo4_1hh-KlsAM3TLawBvwBxva_a7k5Oz1sc,858
-ws_bom_robot_app-0.0.46.dist-info/METADATA,sha256=bpE8xHYYEMkLlx7_ZQKMXJ3UuOdKyI-1EZpUY-wisT4,8348
-ws_bom_robot_app-0.0.46.dist-info/WHEEL,sha256=beeZ86-EfXScwlR_HKu4SllMC9wUEj_8Z_4FJ3egI2w,91
-ws_bom_robot_app-0.0.46.dist-info/top_level.txt,sha256=Yl0akyHVbynsBX_N7wx3H3ZTkcMLjYyLJs5zBMDAKcM,17
-ws_bom_robot_app-0.0.46.dist-info/RECORD,,
+ws_bom_robot_app-0.0.47.dist-info/METADATA,sha256=qAE4WJjYXevHh1uAdIrTs-gfDDaHIvSGw8k8DFnqABY,8348
+ws_bom_robot_app-0.0.47.dist-info/WHEEL,sha256=tTnHoFhvKQHCh4jz3yCn0WPTYIy7wXx3CJtJ7SJGV7c,91
+ws_bom_robot_app-0.0.47.dist-info/top_level.txt,sha256=Yl0akyHVbynsBX_N7wx3H3ZTkcMLjYyLJs5zBMDAKcM,17
+ws_bom_robot_app-0.0.47.dist-info/RECORD,,

{ws_bom_robot_app-0.0.46.dist-info → ws_bom_robot_app-0.0.47.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (76.1.0)
+Generator: setuptools (77.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any