wizit-context-ingestor 0.2.4__tar.gz → 0.2.5b1__tar.gz

{wizit_context_ingestor-0.2.4 → wizit_context_ingestor-0.2.5b1}/PKG-INFO

@@ -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 → wizit_context_ingestor-0.2.5b1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "wizit_context_ingestor"
-version = "0.2.4"
+version = "0.2.5-beta-1"
 description = "Contextual Rag with Cloud Solutions"
 readme = "README.md"
 requires-python = ">=3.12"

{wizit_context_ingestor-0.2.4 → wizit_context_ingestor-0.2.5b1}/src/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-0.2.4 → wizit_context_ingestor-0.2.5b1}/src/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-0.2.4 → wizit_context_ingestor-0.2.5b1}/src/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-0.2.4 → wizit_context_ingestor-0.2.5b1}/src/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(