csc-cia-stne 0.1.39__py3-none-any.whl → 0.1.41__py3-none-any.whl

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) -> 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) -> Dict[str, Any]:
         """
         Envia prompts e opcionalmente múltiplas imagens para a IA via Databricks.
         
@@ -190,6 +190,8 @@ class JerryClient:
                 - mime_type: Tipo MIME da imagem (deve ser image/*)
                 - name: Nome/descrição da imagem (opcional, para logs)
                 - 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
         
         Tipos de imagem suportados:
             ✅ image/jpeg, image/jpg, image/png, image/gif, image/webp, image/bmp, image/tiff
@@ -461,7 +463,13 @@ class JerryClient:
                 "images_validated": images_stats
             }
         
-        # 4. REQUISIÇÃO PARA O DATABRICKS
+        # 4. PREPARAÇÃO E ENVIO COM TOOLS (SE FORNECIDO)
+        if tools:
+            self.log.info(f"{jerry_prompt}🔧 Ferramentas fornecidas para a IA:")
+            for tool in tools:
+                self.log.info(f"{jerry_prompt}   - {tool['function']['name']}: {tool['function']['description']}")
+
+        # 5. REQUISIÇÃO PARA O DATABRICKS
         max_retries = 3
         for attempt in range(max_retries):
             try:
@@ -497,7 +505,9 @@ class JerryClient:
                 self.log.info(f"{jerry_prompt}🚀 Enviando para Databricks (modelo: {self.ia_model})...")
                 response = self.client.chat.completions.create(
                     model=self.ia_model,
-                    messages=messages
+                    tools=tools,
+                    temperature=temperature,
+                    messages=messages,
                 )
 
                 # ✅ SUCESSO
@@ -506,7 +516,7 @@ class JerryClient:
                 self.log.info(f"{jerry_prompt}📄 Resposta bruta: {json.dumps(response.model_dump(), ensure_ascii=False)}")
 
                 content_processado = response.model_dump()
-                content_processado = content_processado["choices"][0]["message"]["content"]
+                #content_processado = content_processado["choices"][0]["message"]["content"]
 
                 return {
                     "success": True, 

csc_cia_stne/utilitarios/validations/GcpBigQueryValidator.py

@@ -20,18 +20,13 @@ class InitParamsValidator(BaseModel):
         return value
     
     @model_validator(mode="after")
-    def check_others_input(cls, model):
-        creds_dict = model.creds_dict
-        creds_file = model.creds_file
+    def check_others_input(self):
+        creds_dict = isinstance(self.creds_dict, dict)
+        creds_file = isinstance(self.creds_file, str)
 
-        if isinstance(creds_dict, dict):
-            return model
-
-        elif isinstance(creds_file, str) and creds_file.strip():
-            return model
-
-        else:
-            raise ValueError("Pelo menos um dos parâmetros 'creds_dict' ou 'creds_file' deve ser fornecido.")
+        if not creds_dict and not creds_file:
+            raise ValueError("É necessário fornecer 'creds_dict' ou 'creds_file' para autenticação")
+        return self
         
 
 class tryQueryValidator(BaseModel):

{csc_cia_stne-0.1.39.dist-info → csc_cia_stne-0.1.41.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: csc_cia_stne
-Version: 0.1.39
+Version: 0.1.41
 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.39.dist-info → csc_cia_stne-0.1.41.dist-info}/RECORD

@@ -7,7 +7,7 @@ csc_cia_stne/gcp_bigquery.py,sha256=foq8azvvv_f7uikMDslX9RcUIrx7RAS-Sn0AGW0QFQc,
 csc_cia_stne/gcp_bucket.py,sha256=vMALWiW7IoBCuJAR8bUCpOV6BuBzI9AhRRk3b72OdMk,11515
 csc_cia_stne/gcp_document_ai.py,sha256=Dzlk7YR3M_LxE0sHn-Lxz-PA1NsUZN2hgY5PyUfs0IQ,4506
 csc_cia_stne/google_drive.py,sha256=7qwx4_RPEoSJgeVI02aLYNXA7o69_Z3qONvX5bfA4V0,44500
-csc_cia_stne/jerry.py,sha256=JVph_WBZOxmxs3J1gaP0hRw9mgurlic3UyUZLzpd9QY,27804
+csc_cia_stne/jerry.py,sha256=TNdMz2B7c9NiW2USmIHcZcZT5KMuux8CJWzNBRKACFE,28451
 csc_cia_stne/karavela.py,sha256=jJCYX43D49gGuzmwwK6bN9XVnv2dXdp9iHnnV5H1LMQ,4794
 csc_cia_stne/logger_json.py,sha256=CXxSCOFGMymDi8XE9SKnPKjW4D0wJLqDLnxqePS26i8,3187
 csc_cia_stne/logger_rich.py,sha256=fklgkBb4rblKQd7YZ3q-eWfhGg9eflO2k2-z4pGh_yo,5201
@@ -29,7 +29,7 @@ csc_cia_stne/utilitarios/functions/func_recriar_pastas.py,sha256=4whZpB3aJQaCPJ3
 csc_cia_stne/utilitarios/functions/func_settings.py,sha256=XwlfqdcfocXQ8kTsDKZ6GsAtpzr0_u44AOTIMtdem7U,2059
 csc_cia_stne/utilitarios/functions/func_titulo.py,sha256=bH4tYxovTARF-g2kWUK_GIzzXt8egbVdp6mWD6fc_3I,5345
 csc_cia_stne/utilitarios/functions/func_validate_json.py,sha256=mAc2J5Eab8ZpNYqsDLJMKkKIeyrSbhytncdJ5G0jztA,1306
-csc_cia_stne/utilitarios/validations/GcpBigQueryValidator.py,sha256=PfCeeQquheZkrm07HTIjobleh3QQOjljRFGdxbQ1amQ,4630
+csc_cia_stne/utilitarios/validations/GcpBigQueryValidator.py,sha256=A58HNGbOSm95ijWmeVeZ2HX6WrhiQz0WbrZhdYrJAgU,4545
 csc_cia_stne/utilitarios/validations/GoogleDriveValidator.py,sha256=Q1oZTae4hDJ2TQ4sUL5Q5TkDNPoJo-DZQt6wIue2jwM,4811
 csc_cia_stne/utilitarios/validations/ServiceNowValidator.py,sha256=yleKUIo1ZfyloP9fDPNjv3JJXdLcocT81WIgRSYmqEA,14423
 csc_cia_stne/utilitarios/validations/__init__.py,sha256=O_qyEU2ji3u6LHUXZCXvUFsMpoMWL625qqHTXyXivTA,106
@@ -41,8 +41,8 @@ csc_cia_stne/utilitarios/web_screen/__init__.py,sha256=5QcOPXKd95SvP2DoZiHS0gaU6
 csc_cia_stne/utilitarios/web_screen/web_screen_abstract.py,sha256=PjL8Vgfj_JdKidia7RFyCkro3avYLQu4RZRos41sh3w,3241
 csc_cia_stne/utilitarios/web_screen/web_screen_botcity.py,sha256=Xi5YJjl2pcxlX3OimqcBWRNXZEpAE7asyUjDJ4Oho5U,12297
 csc_cia_stne/utilitarios/web_screen/web_screen_selenium.py,sha256=JLIcPJE9ZX3Pd6zG6oTRMqqUAY063UzLY3ReRlxmiSM,15581
-csc_cia_stne-0.1.39.dist-info/licenses/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
-csc_cia_stne-0.1.39.dist-info/METADATA,sha256=X6h8HAerxIkqx6xz5FXBIj9lcJz9k3wflrklQw7Wjho,1574
-csc_cia_stne-0.1.39.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-csc_cia_stne-0.1.39.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
-csc_cia_stne-0.1.39.dist-info/RECORD,,
+csc_cia_stne-0.1.41.dist-info/licenses/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
+csc_cia_stne-0.1.41.dist-info/METADATA,sha256=rL3cmx7Qg2zObO69ij_1LGd3pHtFNap3xfSt7yW2H9A,1574
+csc_cia_stne-0.1.41.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+csc_cia_stne-0.1.41.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
+csc_cia_stne-0.1.41.dist-info/RECORD,,