ws-bom-robot-app 0.0.60__py3-none-any.whl → 0.0.62__py3-none-any.whl

ws_bom_robot_app/llm/vector_store/loader/json_loader.py

@@ -1,25 +1,25 @@
-import json
-from typing import Optional
-from langchain_core.documents import Document
-from langchain_community.document_loaders.base import BaseLoader
-
-class JsonLoader(BaseLoader):
-  def __init__(self, file_path: str, meta_fields:Optional[list[str]] = [],encoding: Optional[str] = "utf-8"):
-    self.file_path = file_path
-    self.meta_fields = meta_fields
-    self.encoding = encoding
-
-  def load(self) -> list[Document]:
-    with open(self.file_path, "r", encoding=self.encoding) as file:
-      data = json.load(file)
-    _list = data if isinstance(data, list) else [data]
-    return [
-      Document(
-        page_content=json.dumps(item),
-        metadata={
-          "source": self.file_path,
-          **{field: item.get(field) for field in self.meta_fields if item.get(field)}
-        }
-      )
-      for item in _list
-    ]
+import json
+from typing import Optional
+from langchain_core.documents import Document
+from langchain_community.document_loaders.base import BaseLoader
+
+class JsonLoader(BaseLoader):
+  def __init__(self, file_path: str, meta_fields:Optional[list[str]] = [],encoding: Optional[str] = "utf-8"):
+    self.file_path = file_path
+    self.meta_fields = meta_fields
+    self.encoding = encoding
+
+  def load(self) -> list[Document]:
+    with open(self.file_path, "r", encoding=self.encoding) as file:
+      data = json.load(file)
+    _list = data if isinstance(data, list) else [data]
+    return [
+      Document(
+        page_content=json.dumps(item),
+        metadata={
+          "source": self.file_path,
+          **{field: item.get(field) for field in self.meta_fields if item.get(field)}
+        }
+      )
+      for item in _list
+    ]

ws_bom_robot_app/task_manager.py

@@ -350,13 +350,15 @@ class DatabaseTaskManagerStrategy(TaskManagerStrategy):
             session.commit()
 #endregion
 
-# global instance
+#region global
 def __get_taskmanager_strategy() -> TaskManagerStrategy:
+    """ Factory function to get the appropriate task manager strategy based on the runtime configuration."""
     if config.runtime_options().is_multi_process:
         return DatabaseTaskManagerStrategy()
     return MemoryTaskManagerStrategy()
 task_manager = __get_taskmanager_strategy()
 _log.info(f"Task manager strategy: {task_manager.__class__.__name__}")
+#endregion
 
 #region api
 router = APIRouter(prefix="/api/task", tags=["task"])

ws_bom_robot_app/util.py

@@ -1,6 +1,6 @@
 import logging.handlers
 import os, logging, json
-from typing import TypeVar, Generic
+from typing import Callable, TypeVar, Generic
 from functools import wraps
 from .config import config
 
@@ -23,25 +23,64 @@ _log: logging.Logger = locals().get("_loc", logger_instance(__name__))
 #endregion
 
 #region cache
-class cache(Generic[T]):
-    def _filepath() -> str:
-        return os.path.join('.data',f'{T.__module__}.{T.__name__}.json')
-    @staticmethod
-    def get() -> list[T]:
-        filepath: str = cache._filepath()
-        if os.path.exists(filepath):
-            with open(filepath, 'r') as file:
-                content = file.read()
-            items: list[T] = json.loads(content)
-            return items
-        return None
-    @staticmethod
-    def set(items: list[T]):
-        with open(cache._filepath(), 'w') as file:
-            file.write(json.dumps(items))
-    @staticmethod
-    def clear():
-        os.remove(cache._filepath())
+_cache = {}
+_cache_timestamps = {}
+
+def cache_with_ttl(ttl_seconds: int):
+    """
+    Decorator for caching async function results with TTL (Time To Live)
+
+    Args:
+        ttl_seconds: Cache expiration time in seconds
+    """
+    import time
+    def decorator(func: Callable) -> Callable:
+        @wraps(func)
+        async def wrapper(*args, **kwargs):
+            # Create cache key from function name and arguments
+            cache_key = f"{func.__name__}:{hash(str(args) + str(sorted(kwargs.items())))}"
+
+            current_time = time.time()
+
+            # Check if cached result exists and is still valid
+            if (cache_key in _cache and
+                cache_key in _cache_timestamps and
+                current_time - _cache_timestamps[cache_key] < ttl_seconds):
+                return _cache[cache_key]
+
+            # Call the original function and cache the result
+            result = await func(*args, **kwargs)
+            _cache[cache_key] = result
+            _cache_timestamps[cache_key] = current_time
+
+            return result
+        return wrapper
+    return decorator
+
+def clear_cache(id: str = None):
+    """Clear the cache by id function"""
+    cache_key_prefix = f"{id}:"
+    keys_to_remove = [key for key in _cache.keys() if key.startswith(cache_key_prefix)]
+    for key in keys_to_remove:
+        _cache.pop(key, None)
+        _cache_timestamps.pop(key, None)
+
+def get_cache_info(id: str) -> dict:
+    """Get information about current cache status"""
+    import time
+    current_time = time.time()
+    cache_info = {}
+
+    for key, timestamp in _cache_timestamps.items():
+        if key.startswith(f"{id}:"):
+            remaining_ttl = 600 - (current_time - timestamp)
+            cache_info[key] = {
+                "cached_at": timestamp,
+                "remaining_ttl": max(0, remaining_ttl),
+                "is_expired": remaining_ttl <= 0
+            }
+
+    return cache_info
 #endregion
 
 def _get_timer_wrapper(is_async=False):

{ws_bom_robot_app-0.0.60.dist-info → ws_bom_robot_app-0.0.62.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ws_bom_robot_app
-Version: 0.0.60
+Version: 0.0.62
 Summary: A FastAPI application serving ws bom/robot/llm platform ai.
 Home-page: https://github.com/websolutespa/bom
 Author: Websolute Spa
@@ -13,24 +13,23 @@ Description-Content-Type: text/markdown
 Requires-Dist: standardwebhooks==1.0.0
 Requires-Dist: apscheduler==3.11.0
 Requires-Dist: aiofiles==24.1.0
-Requires-Dist: pydantic==2.10.6
-Requires-Dist: pydantic-settings==2.7.1
-Requires-Dist: fastapi[standard]==0.115.8
+Requires-Dist: pydantic==2.11.7
+Requires-Dist: pydantic-settings==2.10.1
+Requires-Dist: fastapi[standard]==0.115.14
 Requires-Dist: chevron==0.14.0
-Requires-Dist: trafilatura==2.0.0
-Requires-Dist: langchain==0.3.25
-Requires-Dist: langchain-community==0.3.24
-Requires-Dist: langchain-core==0.3.59
-Requires-Dist: langchain-openai==0.3.16
+Requires-Dist: langchain==0.3.26
+Requires-Dist: langchain-community==0.3.26
+Requires-Dist: langchain-core==0.3.67
+Requires-Dist: langchain-openai==0.3.27
 Requires-Dist: langchain-anthropic==0.3.6
 Requires-Dist: langchain-google-genai==2.0.7
-Requires-Dist: langchain-google-vertexai==2.0.13
-Requires-Dist: langchain-groq==0.3.2
-Requires-Dist: langchain-ollama==0.3.2
-Requires-Dist: faiss-cpu==1.9.0
-Requires-Dist: chromadb==0.6.3
-Requires-Dist: langchain_chroma==0.2.1
-Requires-Dist: fastembed==0.5.1
+Requires-Dist: langchain-google-vertexai==2.0.27
+Requires-Dist: langchain-groq==0.3.5
+Requires-Dist: langchain-ollama==0.3.3
+Requires-Dist: faiss-cpu==1.11.0
+Requires-Dist: chromadb==1.0.13
+Requires-Dist: langchain_chroma==0.2.4
+Requires-Dist: fastembed==0.7.1
 Requires-Dist: langchain-qdrant==0.2.0
 Requires-Dist: lark==1.2.2
 Requires-Dist: unstructured==0.16.21
@@ -48,9 +47,10 @@ Requires-Dist: unstructured-ingest[sftp]
 Requires-Dist: unstructured-ingest[sharepoint]
 Requires-Dist: unstructured-ingest[slack]
 Requires-Dist: html5lib==1.1
-Requires-Dist: markdownify==0.14.1
+Requires-Dist: markdownify==1.1.0
 Requires-Dist: duckduckgo-search==8.0.4
 Requires-Dist: langchain_google_community==2.0.7
+Requires-Dist: trafilatura==2.0.0
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{ws_bom_robot_app-0.0.60.dist-info → ws_bom_robot_app-0.0.62.dist-info}/RECORD

@@ -1,45 +1,46 @@
 ws_bom_robot_app/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ws_bom_robot_app/auth.py,sha256=84nIbmJsMrNs0sxIQGEHbjsjc2P6ZrZZGSn8dkiL6is,895
-ws_bom_robot_app/config.py,sha256=9W3cz92hbogDrtbCcybpCY29zCka7G225CNQEptpx30,4183
-ws_bom_robot_app/cron_manager.py,sha256=0Yt5AMTPGlXZ_M5ck0SKMX8wvzoPsseEezg_s0Q3HKY,9224
+ws_bom_robot_app/config.py,sha256=JFO4HrFnzUjFfv5MH0Tzwda0krllEqmTufHZ6J-BgEI,4147
+ws_bom_robot_app/cron_manager.py,sha256=pFHV7SZtp6GRmmLD9K1Mb1TE9Ev9n5mIiFScrc7tpCo,9221
 ws_bom_robot_app/main.py,sha256=zO3B-v-v9ESASvw8IaQj9Y9hNvNmOxohFmA0R82EybQ,6518
-ws_bom_robot_app/task_manager.py,sha256=Zedzs2R3O-wNSQOqs4jorgFwPRi-ji_0TN4mGfk-VvE,15958
-ws_bom_robot_app/util.py,sha256=b49ItlZgh2Wzw-6K8k5Wa44eVgjQ0JmWQwJnEaQBVGw,3502
+ws_bom_robot_app/task_manager.py,sha256=Q3Il2TtkP0FoG9zHEBu48pZGXzimTtvWQsoH6wdvQs0,16077
+ws_bom_robot_app/util.py,sha256=RjVD6B9sHje788Lndqq5DHy6TJM0KLs9qx3JYt81Wyk,4834
 ws_bom_robot_app/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ws_bom_robot_app/llm/agent_context.py,sha256=uatHJ8wcRly6h0S762BgfzDMpmcwCHwNzwo37aWjeE0,1305
-ws_bom_robot_app/llm/agent_description.py,sha256=5IP0qFSJvaE3zjGS7f0W1DuiegP0RHXRMBoDC5pCofA,4779
-ws_bom_robot_app/llm/agent_handler.py,sha256=4HYP8wbdtJhRi3bk6PvJ3cRDZyLYWt3Ow5tnHpkEg1o,7738
-ws_bom_robot_app/llm/agent_lcel.py,sha256=8d10b43BXqE4rfXE5uh8YGT67o1bw0q0l7QXFT6wPKA,2320
-ws_bom_robot_app/llm/api.py,sha256=1nzQ7g2n_DlX6Ixo5ecS10UvyyKJ42qZQ6aD8-EI7BE,4709
-ws_bom_robot_app/llm/defaut_prompt.py,sha256=D9dn8yPveu0bVwGM1wQWLYftmBs5O76o0R_caLLll8w,1121
-ws_bom_robot_app/llm/main.py,sha256=UK33yI_0zDCdM5zKe9h7c_qzM41PIANvRFCxjGlAzlI,5140
-ws_bom_robot_app/llm/nebuly_handler.py,sha256=1HaBeBNzEhyTsgz9v-15Tt7oAc6UBGtqB_DBujpFIcw,7534
-ws_bom_robot_app/llm/settings.py,sha256=DCLaGZwxlw0xE46LpfUgin_FHD8_XJIthCgI6r2UDlM,121
+ws_bom_robot_app/llm/agent_description.py,sha256=yK4aVU3RNk1oP4bEneV3QPAi-208JwWk4R6qHlzqYIg,4656
+ws_bom_robot_app/llm/agent_handler.py,sha256=BQ-f--Z5QCJDp-7tzSG_CKrANUCqG65S09psgWVNxa4,7597
+ws_bom_robot_app/llm/agent_lcel.py,sha256=GGZcGBKsSBbZQ-_MPI3NUMvT7lTerYgwKs3o74stwSU,2252
+ws_bom_robot_app/llm/api.py,sha256=2bF-UFczY9LuBqPxKObM0TOWYbZgVztX1RiIz5MSorU,5042
+ws_bom_robot_app/llm/defaut_prompt.py,sha256=LlCd_nSMkMmHESfiiiQYfnJyB6Pp-LSs4CEKdYW4vFk,1106
+ws_bom_robot_app/llm/main.py,sha256=-7HVVpVa_SIDrZS1zS92moQJ0SOSaBrMIPKOdhQ1jEQ,5006
+ws_bom_robot_app/llm/nebuly_handler.py,sha256=WDKDx7ItBv39dhAkYtRciA11YWUwZ7HjEOI2cgr-5NI,7851
+ws_bom_robot_app/llm/settings.py,sha256=EkFGCppORenStH9W4e6_dYvQ-5p6xiEMpmUHBqNqG9M,117
 ws_bom_robot_app/llm/feedbacks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ws_bom_robot_app/llm/feedbacks/feedback_manager.py,sha256=bnP0FEJTyrzT0YzqCVE73EC07Eu_4FLxVu3Cy-5Si0o,3211
+ws_bom_robot_app/llm/feedbacks/feedback_manager.py,sha256=WcKgzlOb8VFG7yqHoIOO_R6LAzdzE4YIRFCVOGBSgfM,2856
 ws_bom_robot_app/llm/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ws_bom_robot_app/llm/models/api.py,sha256=266f5jc1ikfX9HnK5Ms4NxXowNRfxFEH8GPlQcDvN5Y,10709
+ws_bom_robot_app/llm/models/api.py,sha256=9GK35esx-jl4EIkpSGbiAOyYPbdhtB59txwoyAgdwnk,10939
 ws_bom_robot_app/llm/models/base.py,sha256=1TqxuTK3rjJEALn7lvgoen_1ba3R2brAgGx6EDTtDZo,152
-ws_bom_robot_app/llm/models/feedback.py,sha256=pYNQGxNOBgeAAfdJLI95l7ePLBI5tVdsgnyjp5oMOQU,1722
+ws_bom_robot_app/llm/models/feedback.py,sha256=zh1jLqPRLzNlxInkCMoiJbfSu0-tiOEYHM7FhC46PkM,1692
 ws_bom_robot_app/llm/models/kb.py,sha256=oVSw6_dmNxikAHrPqcfxDXz9M0ezLIYuxpgvzfs_Now,9514
 ws_bom_robot_app/llm/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ws_bom_robot_app/llm/providers/llm_manager.py,sha256=zIkxgTLYQCcup2Ixf4eWap4mNinuJH2YmkjLjZGDyJM,8371
 ws_bom_robot_app/llm/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ws_bom_robot_app/llm/tools/tool_builder.py,sha256=p_Q32_-OSydcxzj69PgPIuiny816zYv5dVsCHSY0ELc,1188
-ws_bom_robot_app/llm/tools/tool_manager.py,sha256=J8WQnjXpLe3Sv6thUi_kgXIDBmt1Z4C-ASzs4RHsVNg,10253
-ws_bom_robot_app/llm/tools/utils.py,sha256=Ba7ScFZPVJ3ke8KLO8ik1wyR2f_zC99Bikqx0OGnKoI,1924
+ws_bom_robot_app/llm/tools/tool_builder.py,sha256=p5CeLCuAilhtEAbPaiAWKGtuIWl8vfilZjYJ9Kw1dLg,3200
+ws_bom_robot_app/llm/tools/tool_manager.py,sha256=6P8YinxI3RYcwkyFTqZkX9C3Lu43Kl0-o6pwQRj3pPo,13719
+ws_bom_robot_app/llm/tools/utils.py,sha256=tdmOAk8l4HVzw67z3brA9yX-1WLu91paU-WmXHyz4Bg,1883
 ws_bom_robot_app/llm/tools/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ws_bom_robot_app/llm/tools/models/main.py,sha256=pBQNWPd1OZgZ2xkOnUOawNbujQ5oJXLdyuAex1afLWc,579
+ws_bom_robot_app/llm/tools/models/main.py,sha256=1hICqHs-KS2heenkH7b2eH0N2GrPaaNGBrn64cl_A40,827
 ws_bom_robot_app/llm/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ws_bom_robot_app/llm/utils/agent.py,sha256=ISF9faaD5tBi-8sbgQpgfqWT1JIVcgv_lRhyaNAkI2Q,1445
+ws_bom_robot_app/llm/utils/agent.py,sha256=_CY5Dji3UeAIi2iuU7ttz4fml1q8aCFgVWOv970x8Fw,1411
 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/cms.py,sha256=Cg5jVi0Peg3PNKOkWIx3SIISv8vf2zn0FP9JJumyeJE,3384
+ws_bom_robot_app/llm/utils/download.py,sha256=iAUxH_NiCpTPtGzhC4hBtxotd2HPFt2MBhttslIxqiI,3194
 ws_bom_robot_app/llm/utils/kb.py,sha256=jja45WCbNI7SGEgqDS99nErlwB5eY8Ga7BMnhdMHZ90,1279
-ws_bom_robot_app/llm/utils/print.py,sha256=HK3zhZOd4cEyXZ8QcudLtTIfqqtMOERce_yTofS8NXo,803
+ws_bom_robot_app/llm/utils/print.py,sha256=IsPYEWRJqu-dqlJA3F9OnnIS4rOq_EYX1Ljp3BvDnww,774
 ws_bom_robot_app/llm/utils/secrets.py,sha256=-HtqLIDVIJrpvGC5YhPAVyLsq8P4ChVM5g3GOfdwqVk,878
 ws_bom_robot_app/llm/utils/webhooks.py,sha256=LAAZqyN6VhV13wu4X-X85TwdDgAV2rNvIwQFIIc0FJM,2114
 ws_bom_robot_app/llm/vector_store/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ws_bom_robot_app/llm/vector_store/generator.py,sha256=W_hi_UOPaSjnEuazhUFIrMAwTvz64Du8_gpiVAxFlVc,6451
+ws_bom_robot_app/llm/vector_store/generator.py,sha256=9_xdtCKJhmt1OP0GXDjvFERXMP7ozLZT92KuYEBDgC0,6314
 ws_bom_robot_app/llm/vector_store/db/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ws_bom_robot_app/llm/vector_store/db/base.py,sha256=rNIYHPDXhVyoP9AJKRbGT5Vh5HzcKYx8MUIhEuCVGW4,6491
 ws_bom_robot_app/llm/vector_store/db/chroma.py,sha256=3UXR7PZidFxgI5jlC0WWPAJ0NGRI2AqSBVlL9VZOJgw,3356
@@ -64,8 +65,8 @@ ws_bom_robot_app/llm/vector_store/integration/slack.py,sha256=FMjESXm2QetFXI6i8e
 ws_bom_robot_app/llm/vector_store/loader/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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.60.dist-info/METADATA,sha256=qOMWEMQar7dYSZllRX9OOj_F198nzb4Xmd6Ur3MR1FE,8456
-ws_bom_robot_app-0.0.60.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ws_bom_robot_app-0.0.60.dist-info/top_level.txt,sha256=Yl0akyHVbynsBX_N7wx3H3ZTkcMLjYyLJs5zBMDAKcM,17
-ws_bom_robot_app-0.0.60.dist-info/RECORD,,
+ws_bom_robot_app/llm/vector_store/loader/json_loader.py,sha256=qo9ejRZyKv_k6jnGgXnu1W5uqsMMtgqK_uvPpZQ0p74,833
+ws_bom_robot_app-0.0.62.dist-info/METADATA,sha256=b1mmWDRfVWVw1PbJmb75XL68KuGe6VnA5Yv-reMGB-4,8459
+ws_bom_robot_app-0.0.62.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ws_bom_robot_app-0.0.62.dist-info/top_level.txt,sha256=Yl0akyHVbynsBX_N7wx3H3ZTkcMLjYyLJs5zBMDAKcM,17
+ws_bom_robot_app-0.0.62.dist-info/RECORD,,