sonika-langchain-bot 0.0.18__py3-none-any.whl → 0.0.19__py3-none-any.whl

sonika_langchain_bot/langchain_clasificator.py

@@ -2,16 +2,30 @@ from pydantic import BaseModel
 from typing import Dict, Any, Type
 from sonika_langchain_bot.langchain_class import ILanguageModel
 
-# Clase para realizar la clasificación de texto
+class ClassificationResponse(BaseModel):
+    """Respuesta de clasificación con tokens utilizados"""
+    input_tokens: int
+    output_tokens: int
+    result: Dict[str, Any]
+
 class TextClassifier:
     def __init__(self, validation_class: Type[BaseModel], llm: ILanguageModel):
-        self.llm =llm
+        self.llm = llm
         self.validation_class = validation_class
-        #configuramos el modelo para que tenga una estructura de salida
-        self.llm.model =  self.llm.model.with_structured_output(validation_class)
+        # Guardamos ambas versiones del modelo
+        self.original_model = self.llm.model  # Sin structured output
+        self.structured_model = self.llm.model.with_structured_output(validation_class)
 
-    def classify(self, text: str) -> Dict[str, Any]:
-        # Crear el template del prompt
+    def classify(self, text: str) -> ClassificationResponse:
+        """
+        Clasifica el texto según la clase de validación.
+        
+        Args:
+            text: Texto a clasificar
+        
+        Returns:
+            ClassificationResponse: Objeto con result, input_tokens y output_tokens
+        """
         prompt = f"""
         Classify the following text based on the properties defined in the validation class.
         
@@ -19,12 +33,34 @@ class TextClassifier:
         
         Only extract the properties mentioned in the validation class.
         """
-        response = self.llm.invoke(prompt=prompt)
         
-        # Asegurarse de que el `response` es de la clase de validación proporcionada
+        # Primero invocamos el modelo ORIGINAL para obtener metadata de tokens
+        raw_response = self.original_model.invoke(prompt)
+        
+        # Extraer información de tokens del AIMessage original
+        input_tokens = 0
+        output_tokens = 0
+        
+        if hasattr(raw_response, 'response_metadata'):
+            token_usage = raw_response.response_metadata.get('token_usage', {})
+            input_tokens = token_usage.get('prompt_tokens', 0)
+            output_tokens = token_usage.get('completion_tokens', 0)
+        
+        # Ahora invocamos con structured output para obtener el objeto parseado
+        response = self.structured_model.invoke(prompt)
+        
+        # Validar que el response es de la clase correcta
         if isinstance(response, self.validation_class):
-            # Crear el resultado dinámicamente basado en los atributos de la clase de validación
-            result = {field: getattr(response, field) for field in self.validation_class.__fields__.keys()}
-            return result
+            # Crear el resultado dinámicamente basado en los atributos
+            result_data = {
+                field: getattr(response, field) 
+                for field in self.validation_class.__fields__.keys()
+            }
+            
+            return ClassificationResponse(
+                input_tokens=input_tokens,
+                output_tokens=output_tokens,
+                result=result_data
+            )
         else:
-            raise ValueError(f"The response is not of type '{self.validation_class.__name__}'")
+            raise ValueError(f"The response is not of type '{self.validation_class.__name__}'")

{sonika_langchain_bot-0.0.18.dist-info → sonika_langchain_bot-0.0.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sonika-langchain-bot
-Version: 0.0.18
+Version: 0.0.19
 Summary: Agente langchain con LLM
 Author: Erley Blanco Carvajal
 License: MIT License

{sonika_langchain_bot-0.0.18.dist-info → sonika_langchain_bot-0.0.19.dist-info}/RECORD

@@ -3,13 +3,13 @@ sonika_langchain_bot/document_processor.py,sha256=RuHT22Zt-psoe4adFWKwBJ0gi638fq
 sonika_langchain_bot/langchain_bdi.py,sha256=ithc55azP5XSPb8AGRUrDGYnVI6I4IqpqElLNat4BAQ,7024
 sonika_langchain_bot/langchain_bot_agent.py,sha256=wdQrIVUxmkU-VmGfrdIcnDJPdKmRQXgeKYf_50By5cE,24359
 sonika_langchain_bot/langchain_bot_agent_bdi.py,sha256=Ev0hhRQYe6kyGAHiFDhFsfu6QnTwUFaA9oB8DfNV7u4,8613
-sonika_langchain_bot/langchain_clasificator.py,sha256=GR85ZAliymBSoDa5PXB31BvJkuiokGjS2v3RLdXnzzk,1381
+sonika_langchain_bot/langchain_clasificator.py,sha256=h0-H_1bqgA04rF2ZHh5zOg2PinqTuLQMcSK7AGK4uw8,2583
 sonika_langchain_bot/langchain_class.py,sha256=5anB6v_wCzEoAJRb8fV9lPPS72E7-k51y_aeiip8RAw,1114
 sonika_langchain_bot/langchain_files.py,sha256=SEyqnJgBc_nbCIG31eypunBbO33T5AHFOhQZcghTks4,381
 sonika_langchain_bot/langchain_models.py,sha256=vqSSZ48tNofrTMLv1QugDdyey2MuIeSdlLSD37AnzkI,2235
 sonika_langchain_bot/langchain_tools.py,sha256=y7wLf1DbUua3QIvz938Ek-JIMOuQhrOIptJadW8OIsU,466
-sonika_langchain_bot-0.0.18.dist-info/licenses/LICENSE,sha256=O8VZ4aU_rUMAArvYTm2bshcZ991huv_tpfB5BKHH9Q8,1064
-sonika_langchain_bot-0.0.18.dist-info/METADATA,sha256=mfLJal88ZamfkvOouwQZ01VwVz9784vb5ynFOr5Ust0,6508
-sonika_langchain_bot-0.0.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sonika_langchain_bot-0.0.18.dist-info/top_level.txt,sha256=UsTTSZFEw2wrPSVh4ufu01e2m_E7O_QVYT_k4zCQaAE,21
-sonika_langchain_bot-0.0.18.dist-info/RECORD,,
+sonika_langchain_bot-0.0.19.dist-info/licenses/LICENSE,sha256=O8VZ4aU_rUMAArvYTm2bshcZ991huv_tpfB5BKHH9Q8,1064
+sonika_langchain_bot-0.0.19.dist-info/METADATA,sha256=hjo1qMMAqU1CCsc1BbbURDV3sOTQYVdEraEVCiDRXoc,6508
+sonika_langchain_bot-0.0.19.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sonika_langchain_bot-0.0.19.dist-info/top_level.txt,sha256=UsTTSZFEw2wrPSVh4ufu01e2m_E7O_QVYT_k4zCQaAE,21
+sonika_langchain_bot-0.0.19.dist-info/RECORD,,