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

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

@@ -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 → csc_cia_stne-0.1.41}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "csc_cia_stne"
-version = "0.1.39"
+version = "0.1.41"
 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.39 → csc_cia_stne-0.1.41}/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) -> 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-0.1.39 → csc_cia_stne-0.1.41}/src/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 → csc_cia_stne-0.1.41}/src/csc_cia_stne.egg-info/PKG-INFO

@@ -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