csc-cia-stne 0.1.40__py3-none-any.whl → 0.1.42__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, tool_choice: Optional[str] = None) -> Dict[str, Any]:
         """
         Envia prompts e opcionalmente múltiplas imagens para a IA via Databricks.
         
@@ -190,6 +190,9 @@ 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
+            tool_choice (Optional[str]): required, auto
         
         Tipos de imagem suportados:
             ✅ image/jpeg, image/jpg, image/png, image/gif, image/webp, image/bmp, image/tiff
@@ -461,7 +464,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 +506,10 @@ 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,
+                    tool_choice=tool_choice,
+                    temperature=temperature,
+                    messages=messages,
                 )
 
                 # ✅ SUCESSO
@@ -506,7 +518,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-0.1.40.dist-info → csc_cia_stne-0.1.42.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: csc_cia_stne
-Version: 0.1.40
+Version: 0.1.42
 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.40.dist-info → csc_cia_stne-0.1.42.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=Bmu0A35tLPVBVHbjnwJkit4W-yZpOaWB5hG80E5f_gk,28587
 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
@@ -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.40.dist-info/licenses/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
-csc_cia_stne-0.1.40.dist-info/METADATA,sha256=nfrk3fUIulnqzXv-2LSOgrUxR8714-uErQJ6mfSQWDw,1574
-csc_cia_stne-0.1.40.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-csc_cia_stne-0.1.40.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
-csc_cia_stne-0.1.40.dist-info/RECORD,,
+csc_cia_stne-0.1.42.dist-info/licenses/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
+csc_cia_stne-0.1.42.dist-info/METADATA,sha256=_4_CSj4MqzkzQD-sQ4s4o48vF1JgVZCn_z4N2fhh46g,1574
+csc_cia_stne-0.1.42.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+csc_cia_stne-0.1.42.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
+csc_cia_stne-0.1.42.dist-info/RECORD,,