wizit-context-ingestor 0.2.4__py3-none-any.whl → 0.2.5b1__py3-none-any.whl

wizit_context_ingestor/infra/aws_model.py

@@ -37,7 +37,6 @@ class AWSModels(AiApplicationService):
         temperature: float = 0.7,
         max_tokens: int = 8000,
         region_name: str = "us-east-1") -> ChatBedrockConverse:
-
         """
         Load an AWS AI chat model for text generation.
 

wizit_context_ingestor/infra/persistence/local_storage.py

@@ -1,5 +1,6 @@
 from ...application.interfaces import PersistenceService
 from ...domain.models import ParsedDoc
+from typing import Optional
 import logging
 import os
 logger = logging.getLogger(__name__)
@@ -40,7 +41,7 @@ class LocalStorageService(PersistenceService):
             raise
 
 
-    def save_parsed_document(self, file_key: str, parsed_document: ParsedDoc):
+    def save_parsed_document(self, file_key: str, parsed_document: ParsedDoc, file_tags: Optional[dict] = {}):
         """Save a parsed document."""
         with open(f"{self.tmp_folder}/{file_key}", "w", encoding="utf-8") as f:
             f.write(parsed_document.document_text)

wizit_context_ingestor/infra/persistence/s3_storage.py

@@ -4,6 +4,7 @@ from boto3 import client as boto3_client
 import logging
 import os
 from botocore.exceptions import ClientError
+from typing import Optional
 
 logger = logging.getLogger(__name__)
 
@@ -77,8 +78,7 @@ class S3StorageService(PersistenceService):
             raise
 
 
-
-    def save_parsed_document(self, file_key: str, parsed_document: ParsedDoc, file_tags: dict = {}):
+    def save_parsed_document(self, file_key: str, parsed_document: ParsedDoc, file_tags: Optional[dict] = {}):
         """Save a parsed document to S3.
 
         Args:

wizit_context_ingestor/main.py

@@ -113,8 +113,11 @@ class DeelabRedisChunksManager:
         try:
             rag_chunker = SemanticChunks(self.embeddings_model)
             redis_embeddings_manager = RedisEmbeddingsManager(
-                embeddings_model=self.embeddings_model,
-                redis_connection_string=self.redis_connection_string
+                self.embeddings_model,
+                self.redis_connection_string,
+                {
+                    "file_key": file_key
+                }
             )
             local_persistence_service = LocalStorageService()
             context_chunks_in_document_service = ContextChunksInDocumentService(

{wizit_context_ingestor-0.2.4.dist-info → wizit_context_ingestor-0.2.5b1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: wizit-context-ingestor
-Version: 0.2.4
+Version: 0.2.5b1
 Summary: Contextual Rag with Cloud Solutions
 Requires-Dist: anthropic[vertex]>=0.66.0
 Requires-Dist: boto3>=1.40.23

{wizit_context_ingestor-0.2.4.dist-info → wizit_context_ingestor-0.2.5b1.dist-info}/RECORD

@@ -10,22 +10,22 @@ wizit_context_ingestor/domain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5N
 wizit_context_ingestor/domain/models.py,sha256=DV83PArMyh-VoUqnVF_ohcgStsk549ixdYw98B8o2GI,381
 wizit_context_ingestor/domain/services.py,sha256=R14ImkiDzv-X1o1qYl9J__FBclnt9Uj85xbo0wgOqCg,3407
 wizit_context_ingestor/infra/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wizit_context_ingestor/infra/aws_model.py,sha256=P0iQKYjERGV38Lc1WtBnZtVmKiHkktwUb4G4dqJz6-s,2540
+wizit_context_ingestor/infra/aws_model.py,sha256=glIaewSdv6PDBXoCe6QgCUIzLCjtM7KlayEERXRNFwo,2539
 wizit_context_ingestor/infra/persistence/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wizit_context_ingestor/infra/persistence/local_storage.py,sha256=Spw-Zmwj2OTY-TU9WokvNinnu1D5R87SpUV0S7MJZN8,1647
-wizit_context_ingestor/infra/persistence/s3_storage.py,sha256=c39pMpkoFtfFXzRFxVgHzKVsT3IoJzBsCe6-CydAQXM,4754
+wizit_context_ingestor/infra/persistence/local_storage.py,sha256=sDFat-FMN123FUWZp_ztwoVjl0HrzChCDZmicFemy5o,1707
+wizit_context_ingestor/infra/persistence/s3_storage.py,sha256=HYO3gWNE64ECSYYrxrIi9-2jWv1vwwGEE5QX-ZqpOCs,4791
 wizit_context_ingestor/infra/rag/pg_embeddings.py,sha256=5m4R4GmwMU3C7AR3Je3nCdgO-2jyIaCG4QN9phGD68Q,8072
 wizit_context_ingestor/infra/rag/redis_embeddings.py,sha256=wlgSBedq_kcrZ3SF4vGVTWM0B350kkd8C894i4mMUA8,4828
 wizit_context_ingestor/infra/rag/semantic_chunks.py,sha256=tM6bSacBvu-VWb3VkxgQNrfskz3zFxOOAU23D2kZWD8,2255
 wizit_context_ingestor/infra/secrets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wizit_context_ingestor/infra/secrets/aws_secrets_manager.py,sha256=1k_R_uzLabptiZ1GXAoqAgYpk8EykXIb-pUDdidUDJQ,1202
 wizit_context_ingestor/infra/vertex_model.py,sha256=Izpz2ZQ4Koh4PSrHAj_0iUv4Rx354SlUqqw-LrLXCOE,7256
-wizit_context_ingestor/main.py,sha256=8pTXFqLHrs02YxhHaNm9v49HIykilhefkonJxKkBsT4,7929
+wizit_context_ingestor/main.py,sha256=x1bieggNs3uzLkNHXZk4fBz0fYlOyff1TGvjV3Af2yM,7966
 wizit_context_ingestor/services/.DS_Store,sha256=1lFlJ5EFymdzGAUAaI30vcaaLHt3F1LwpG7xILf9jsM,6148
 wizit_context_ingestor/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wizit_context_ingestor/services/chunks.py,sha256=tQQsdsOscZWzqVY5WxVxr3ii62FOJ3nMARaJJz6CvjQ,2011
 wizit_context_ingestor/services/parse_doc.py,sha256=5EIQnGWSyXwepazcO4lj-MKBaBWdGzIQIwKDE1soZTc,4577
 wizit_context_ingestor/services/pg_embeddings_manager.py,sha256=n1HOmu_Z_Z71H-rVAyJS3FdPKbBckm5W8_XethY8nuM,4998
-wizit_context_ingestor-0.2.4.dist-info/WHEEL,sha256=F3mArEuDT3LDFEqo9fCiUx6ISLN64aIhcGSiIwtu4r8,79
-wizit_context_ingestor-0.2.4.dist-info/METADATA,sha256=-vIELbGOA3J_qLY0iJyz4GXH5UyB4WBmUNL4tJjUKHI,3575
-wizit_context_ingestor-0.2.4.dist-info/RECORD,,
+wizit_context_ingestor-0.2.5b1.dist-info/WHEEL,sha256=F3mArEuDT3LDFEqo9fCiUx6ISLN64aIhcGSiIwtu4r8,79
+wizit_context_ingestor-0.2.5b1.dist-info/METADATA,sha256=yMnfleIUWwt32cVfndgq8OrMC7TkJZw0gbJzN05qwfY,3577
+wizit_context_ingestor-0.2.5b1.dist-info/RECORD,,