csc-cia-stne 0.1.41__tar.gz → 0.1.43__tar.gz

{csc_cia_stne-0.1.41 → csc_cia_stne-0.1.43}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: csc_cia_stne
-Version: 0.1.41
+Version: 0.1.43
 Summary: Biblioteca do time CSC-CIA utilizada no desenvolvimento de RPAs
 License: MIT
 Keywords: karavela,csc,cia,stone,rpa,botcity,stne

{csc_cia_stne-0.1.41 → csc_cia_stne-0.1.43}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "csc_cia_stne"
-version = "0.1.41"
+version = "0.1.43"
 license = { text = "MIT" }
 description = "Biblioteca do time CSC-CIA utilizada no desenvolvimento de RPAs"
 keywords = ["karavela", "csc", "cia", "stone", "rpa", "botcity", "stne"]

{csc_cia_stne-0.1.41 → csc_cia_stne-0.1.43}/src/csc_cia_stne/jerry.py

@@ -176,7 +176,7 @@ class JerryClient:
             self.error = f"Erro na autenticação: {e}"
             return None
 
-    def v1_enviar_para_ia(self, system_prompt: str, user_prompt: str, arquivos: Optional[List[Dict[str, str]]] = None, tools: Optional[List[Dict[str, Any]]] = None, temperature: float = 0.3) -> Dict[str, Any]:
+    def v1_enviar_para_ia(self, system_prompt: str, user_prompt: str, arquivos: Optional[List[Dict[str, str]]] = None, tools: Optional[List[Dict[str, Any]]] = None, temperature: float = 0.3, tool_choice: Optional[str] = None) -> Dict[str, Any]:
         """
         Envia prompts e opcionalmente múltiplas imagens para a IA via Databricks.
         
@@ -192,6 +192,7 @@ class JerryClient:
                 - file_path: Caminho original da imagem (opcional, para logs)
             temperature (float): Temperatura para a geração de texto (default = 0.3)
             tools (Optional[List[Dict[str, Any]]]): Lista de ferramentas (tools) para fornecer à IA
+            tool_choice (Optional[str]): required, auto
         
         Tipos de imagem suportados:
             ✅ image/jpeg, image/jpg, image/png, image/gif, image/webp, image/bmp, image/tiff
@@ -386,9 +387,9 @@ class JerryClient:
                 image_size = len(decoded_data)
                 total_size += image_size
                 
-                # Limite por imagem: 20MB
-                if image_size > 20 * 1024 * 1024:
-                    error = f"Imagem '{arquivo_name}': Tamanho muito grande ({image_size:,} bytes). Máximo: 20MB"
+                # Limite por imagem: 4MB
+                if image_size > 4 * 1024 * 1024:
+                    error = f"Imagem '{arquivo_name}': Tamanho muito grande ({image_size:,} bytes). Máximo: 4MB"
                     validation_errors.append(error)
                     images_stats["rejected"] += 1
                     images_stats["rejected_files"].append({"filename": arquivo_name, "error": error})
@@ -410,8 +411,8 @@ class JerryClient:
                 self.log.info(f"{jerry_prompt}   ✅ {arquivo_name} ({arquivo_mime}, {image_size:,} bytes)")
             
             # 2.9 Validar tamanho total
-            if total_size > 50 * 1024 * 1024:  # 50MB total
-                error = f"Tamanho total das imagens muito grande ({total_size:,} bytes). Máximo: 50MB"
+            if total_size > 4 * 1024 * 1024:  # 4MB total
+                error = f"Tamanho total das imagens muito grande ({total_size:,} bytes). Máximo: 4MB"
                 validation_errors.append(error)
                 self.log.error(f"{jerry_prompt}❌ {error}")
 
@@ -506,6 +507,7 @@ class JerryClient:
                 response = self.client.chat.completions.create(
                     model=self.ia_model,
                     tools=tools,
+                    tool_choice=tool_choice,
                     temperature=temperature,
                     messages=messages,
                 )

{csc_cia_stne-0.1.41 → csc_cia_stne-0.1.43}/src/csc_cia_stne.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: csc_cia_stne
-Version: 0.1.41
+Version: 0.1.43
 Summary: Biblioteca do time CSC-CIA utilizada no desenvolvimento de RPAs
 License: MIT
 Keywords: karavela,csc,cia,stone,rpa,botcity,stne