osbot-utils 2.38.0__py3-none-any.whl → 2.40.0__py3-none-any.whl

osbot_utils/helpers/llms/actions/LLM_Request__Execute.py

@@ -26,6 +26,6 @@ class LLM_Request__Execute(Type_Safe):
         llm_response                          = Schema__LLM_Response(response_data=response_data)   # Create response object
 
         if self.use_cache:                                                                          # Cache the response if enabled
-            self.llm_cache.add(llm_request, llm_response)
+            self.llm_cache.add(request=llm_request, response=llm_response, payload=llm_payload)
 
         return llm_response

osbot_utils/helpers/llms/cache/LLM_Request__Cache.py

@@ -25,14 +25,16 @@ class LLM_Request__Cache(Type_Safe):
         return Safe_Str__Hash(hash_value)
 
     @type_safe
-    def add(self, request     : Schema__LLM_Request ,                                       # Request to cache
-                  response    : Schema__LLM_Response                                        # Response to store
+    def add(self, request  : Schema__LLM_Request ,                                          # Request to cache
+                  response : Schema__LLM_Response,                                          # Response to store
+                  payload  : dict                                                           # Payload to store
              ) -> Obj_Id:                                                                   # returns cache_id
 
         hash_request          = self.compute_request_hash (request)                         # calculate request hash
         cache_entry           = Schema__LLM_Response__Cache(cache_id                = Obj_Id()             ,              # Create a cache entry
                                                             llm_request             = request              ,
                                                             llm_response            = response             ,
+                                                            llm_payload             = payload          ,
                                                             hash__request           = hash_request         )
         cache_id             = cache_entry.cache_id
 
@@ -46,15 +48,30 @@ class LLM_Request__Cache(Type_Safe):
 
         if request_hash in self.cache_index.cache_id__from__hash__request:                                                              # Check if we have an exact match
             cache_id    = self.cache_index.cache_id__from__hash__request[request_hash]
-            cache_entry = self.get_cache_entry(cache_id)
+            cache_entry = self.get__cache_entry__from__cache_id(cache_id)
             if cache_entry:
                 return cache_entry.llm_response
 
         return None
 
     @type_safe
-    def get_cache_entry(self, cache_id: Obj_Id) -> Optional[Schema__LLM_Response__Cache]:                               # Get cache entry by ID
-        return self.cache_entries.get(cache_id)
+    def get__cache_entry__from__cache_id(self, cache_id: Obj_Id) -> Optional[Schema__LLM_Response__Cache]:                               # Get cache entry by ID
+        if cache_id:
+            return self.cache_entries.get(cache_id)
+
+    @type_safe
+    def get__cache_entry__from__request(self, request: Schema__LLM_Request):
+        cache_id = self.get__cache_id__from__request(request)
+        return self.get__cache_entry__from__cache_id(cache_id)
+
+    @type_safe
+    def get__cache_id__from__request(self, request: Schema__LLM_Request):
+        request_hash = self.compute_request_hash(request)
+        return self.get__cache_id__from__request_hash(request_hash)
+
+    @type_safe
+    def get__cache_id__from__request_hash(self, request_hash: Safe_Str__Hash):
+        return self.cache_index.cache_id__from__hash__request.get(request_hash)
 
     def exists(self, request: Schema__LLM_Request) -> bool:                                                             # True if in cache
         request_hash = self.compute_request_hash(request)
@@ -76,7 +93,7 @@ class LLM_Request__Cache(Type_Safe):
         return self.save()
 
     def get_by_id(self, cache_id : Obj_Id)-> Optional[Schema__LLM_Response]:                                            # Cached response or None
-        cache_entry = self.get_cache_entry(cache_id)
+        cache_entry = self.get__cache_entry__from__cache_id(cache_id)
         if cache_entry:
             return cache_entry.llm_response
         return None

osbot_utils/helpers/llms/cache/LLM_Request__Cache__File_System.py

@@ -169,12 +169,13 @@ class LLM_Request__Cache__File_System(LLM_Request__Cache):
         return path
 
     @type_safe
-    def add(self, request  : Schema__LLM_Request,
+    def add(self, request  : Schema__LLM_Request ,
                   response : Schema__LLM_Response,
+                  payload  : dict                ,
                   now      : datetime = None
              ) -> Obj_Id:                                                                           # Save an LLM request/response pair using temporal organization.
 
-        cache_id        = super().add(request, response)                                            # First use standard add() to handle in-memory caching
+        cache_id        = super().add(request=request, response=response, payload=payload)                                            # First use standard add() to handle in-memory caching
         cache_entry     = self.cache_entries[cache_id]                                              # get the cache entry (which will exist since it was added on super().add(request, response)  )
         request_domains = self.extract_domains_from_request(request)                                # Extract domains and areas for organization
         domains         = self.shared_domains + request_domains
@@ -219,7 +220,7 @@ class LLM_Request__Cache__File_System(LLM_Request__Cache):
     #                 cache_id_str = os.path.splitext(os.path.basename(item_path))[0]
     #                 if is_obj_id(cache_id_str):
     #                     cache_id = Obj_Id(cache_id_str)
-    #                     cache_entry = self.get_cache_entry(cache_id)
+    #                     cache_entry = self.get_cache_entry__from__cache_id(cache_id)
     #                     if cache_entry:
     #                         results.append(cache_entry)
     #

osbot_utils/helpers/llms/schemas/Schema__LLM_Response__Cache.py

@@ -10,4 +10,5 @@ class Schema__LLM_Response__Cache(Type_Safe):
     hash__request          : Safe_Str__Hash       = None
     llm_response           : Schema__LLM_Response = None
     llm_request            : Schema__LLM_Request  = None
+    llm_payload            : dict
     timestamp              : Timestamp_Now

osbot_utils/version

@@ -1 +1 @@
-v2.38.0
+v2.40.0

{osbot_utils-2.38.0.dist-info → osbot_utils-2.40.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: osbot_utils
-Version: 2.38.0
+Version: 2.40.0
 Summary: OWASP Security Bot - Utils
 License: MIT
 Author: Dinis Cruz
@@ -23,7 +23,7 @@ Description-Content-Type: text/markdown
 
 Powerful Python util methods and classes that simplify common apis and tasks.
 
-![Current Release](https://img.shields.io/badge/release-v2.38.0-blue)
+![Current Release](https://img.shields.io/badge/release-v2.40.0-blue)
 [![codecov](https://codecov.io/gh/owasp-sbot/OSBot-Utils/graph/badge.svg?token=GNVW0COX1N)](https://codecov.io/gh/owasp-sbot/OSBot-Utils)
 
 

{osbot_utils-2.38.0.dist-info → osbot_utils-2.40.0.dist-info}/RECORD

@@ -190,7 +190,7 @@ osbot_utils/helpers/html/Tag__Link.py,sha256=rQ-gZN8EkSv5x1S-smdjvFflwMQHACHQXiO
 osbot_utils/helpers/html/Tag__Style.py,sha256=LPPlIN7GyMvfCUlbs2eXVMUr9jS0PX5M94A5Ig_jXIs,846
 osbot_utils/helpers/html/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 osbot_utils/helpers/llms/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-osbot_utils/helpers/llms/actions/LLM_Request__Execute.py,sha256=vR_7ManwTSyUOIF_eJK3sft1wBVadaGHkZOzspVc7T4,1987
+osbot_utils/helpers/llms/actions/LLM_Request__Execute.py,sha256=F921XqKGdH7M77E1OKrGcJnYBoqMKeOdXp98VVCJhKE,2025
 osbot_utils/helpers/llms/actions/Type_Safe__Schema_For__LLMs.py,sha256=em9RoSZqSSo6BQBZvEKH8Qv8f8f8oubNpy0LIDsak-E,12024
 osbot_utils/helpers/llms/actions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 osbot_utils/helpers/llms/builders/LLM_Request__Builder.py,sha256=c8MN66ijFn9DugbBIGmG0bm9ujS-0kvZDGqJcZsHHYc,3385
@@ -198,8 +198,8 @@ osbot_utils/helpers/llms/builders/LLM_Request__Builder__Open_AI.py,sha256=JwyPDe
 osbot_utils/helpers/llms/builders/LLM_Request__Factory.py,sha256=bpFXVTKpalBL7ZONjaHU5c0-2Rwzzd2vgdD1FpYxfGw,6291
 osbot_utils/helpers/llms/builders/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 osbot_utils/helpers/llms/cache/LLM_Cache__Path_Generator.py,sha256=7kDY-aaUWR4Da6lj5RPwjqTLUTGrOUYhbNfOPM30e40,4442
-osbot_utils/helpers/llms/cache/LLM_Request__Cache.py,sha256=8Ayz26bteYhc4GNLpItNYlTwF4_yKvhFridDVA61fPo,6783
-osbot_utils/helpers/llms/cache/LLM_Request__Cache__File_System.py,sha256=m4guw6yoo8Q05PQea6SJnET5xpJMwmmXhdN-kaRvX5s,15306
+osbot_utils/helpers/llms/cache/LLM_Request__Cache.py,sha256=xmHZ8hV1w6aUjPp8zmHIydjr84keUT--Ug_ABpNSVlE,7684
+osbot_utils/helpers/llms/cache/LLM_Request__Cache__File_System.py,sha256=4xYBhzydIoZI0g9CAswSGdFtwp6dcItm8_vnMfeSoTg,15408
 osbot_utils/helpers/llms/cache/LLM_Request__Cache__Storage.py,sha256=0ok7z2kGiK3edlgD0hwD5z2hYFZ91Viar1JgD7SNoxg,5434
 osbot_utils/helpers/llms/cache/Virtual_Storage__Local__Folder.py,sha256=xNM2xy0-2ROvhEFXa_jwqth7eApIcRTqcRUDxo0THXI,3214
 osbot_utils/helpers/llms/cache/Virtual_Storage__Sqlite.py,sha256=j8EWuKFVMtDBxrCEi8yNdpzwvTxO6Bri2Bi4-KwO-3s,4638
@@ -214,7 +214,7 @@ osbot_utils/helpers/llms/schemas/Schema__LLM_Request__Function_Call.py,sha256=VJ
 osbot_utils/helpers/llms/schemas/Schema__LLM_Request__Message__Content.py,sha256=nl-16yz4G_72ViACKE9CvGStrKxw2Gm_JcaU8wVcJXI,521
 osbot_utils/helpers/llms/schemas/Schema__LLM_Request__Message__Role.py,sha256=T99w0cRrDPXQqPT-Nw7_14tMr4vKpUlhw74UJZL6w6w,168
 osbot_utils/helpers/llms/schemas/Schema__LLM_Response.py,sha256=R62zjoUqu9sbtbFcOX7vG1t0vTaAhS_3Z9CkxESrTjI,298
-osbot_utils/helpers/llms/schemas/Schema__LLM_Response__Cache.py,sha256=ZlD3DeKeS-29n9qiMVrsnj0M4lkZWknnP-Wn8wanu6E,786
+osbot_utils/helpers/llms/schemas/Schema__LLM_Response__Cache.py,sha256=jRV637LYbj9HhRtipYsd0JBxcvRnVHcF0OFJHObGhHA,820
 osbot_utils/helpers/llms/schemas/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 osbot_utils/helpers/pubsub/Event__Queue.py,sha256=bCtIdVlAuG-jvFEnz14oNhgRScEUrd8v9BqLcZleGks,5038
 osbot_utils/helpers/pubsub/PubSub__Client.py,sha256=6K3l4H-Tc0DhktrxpYzLVur1uZ532pQsHWprLNRXFJE,2316
@@ -398,8 +398,8 @@ osbot_utils/utils/Toml.py,sha256=Rxl8gx7mni5CvBAK-Ai02EKw-GwtJdd3yeHT2kMloik,166
 osbot_utils/utils/Version.py,sha256=Ww6ChwTxqp1QAcxOnztkTicShlcx6fbNsWX5xausHrg,422
 osbot_utils/utils/Zip.py,sha256=pR6sKliUY0KZXmqNzKY2frfW-YVQEVbLKiyqQX_lc-8,14052
 osbot_utils/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-osbot_utils/version,sha256=5wV6C9vRtevBlgAl6UzZ00kufvcHsbgatM8FaiKy5Us,8
-osbot_utils-2.38.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-osbot_utils-2.38.0.dist-info/METADATA,sha256=1oshM3ZIVLYEZhR_CgTKhCw7m2wWz62PvYqgCgLm22A,1329
-osbot_utils-2.38.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-osbot_utils-2.38.0.dist-info/RECORD,,
+osbot_utils/version,sha256=EA_osWNMH5Sn_inc9L5qYPjfsYR_EbU1rahws7uQB0o,8
+osbot_utils-2.40.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+osbot_utils-2.40.0.dist-info/METADATA,sha256=YPJG2T0zO_VSq9HLF-n45Imxo2bf9vq2VxWN6QBc5ko,1329
+osbot_utils-2.40.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+osbot_utils-2.40.0.dist-info/RECORD,,