chatgraph 0.2.2__tar.gz → 0.2.4__tar.gz

{chatgraph-0.2.2 → chatgraph-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: chatgraph
-Version: 0.2.2
+Version: 0.2.4
 Summary: A user-friendly chatbot library
 Home-page: https://github.com/irissonnlima/chatgraph
 License: MIT
@@ -12,6 +12,7 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: pika (>=1.3.2,<2.0.0)
+Requires-Dist: rich (>=13.8.1,<14.0.0)
 Project-URL: Repository, https://github.com/irissonnlima/chatgraph
 Description-Content-Type: text/markdown
 

{chatgraph-0.2.2 → chatgraph-0.2.4}/chatgraph/bot/chatbot_model.py

@@ -2,6 +2,7 @@ import inspect
 from abc import ABC
 from functools import wraps
 from logging import debug
+import json
 
 from ..error.chatbot_error import ChatbotError, ChatbotMessageError
 from ..messages.base_message_consumer import MessageConsumer
@@ -97,8 +98,9 @@ class ChatbotApp(ABC):
         """
         Inicia o consumo de mensagens pelo chatbot, processando cada mensagem recebida.
         """
+        self.__message_consumer.reprer()
         self.__message_consumer.start_consume(self.process_message)
-
+    
     def process_message(self, message: Message):
         """
         Processa uma mensagem recebida, identificando a rota correspondente e executando a função associada.
@@ -146,9 +148,11 @@ class ChatbotApp(ABC):
                 'menu': route,
                 'obs': user_state.obs,
             }
-            return response            
+            return json.dumps(response)            
         elif type(message_response) in (ChatbotResponse, EndChatResponse, TransferToHuman):
-            return message_response.json()
+            response = message_response.json()
+            response['customer_id'] = customer_id
+            return json.dumps(response)
         elif type(message_response) == RedirectResponse:
             route = self.__adjust_route(message_response.route, menu)
             message.user_state.menu = route
@@ -173,4 +177,4 @@ class ChatbotApp(ABC):
         if 'start' not in route:
             route = absolute_route + route
 
-        return route
+        return route

{chatgraph-0.2.2 → chatgraph-0.2.4}/chatgraph/messages/base_message_consumer.py

@@ -31,3 +31,10 @@ class MessageConsumer(ABC):
             MessageConsumer: A instância do consumidor de mensagens configurado.
         """
         pass
+    
+    @abstractmethod
+    def reprer(self) -> str:
+        '''
+        Retorna uma representação textual do consumidor de mensagens.
+        '''
+        pass

{chatgraph-0.2.2 → chatgraph-0.2.4}/chatgraph/messages/rabbitMQ_message_consumer.py

@@ -6,7 +6,10 @@ from typing import Callable
 from ..auth.credentials import Credential
 from ..types.message_types import Message, UserState
 from .base_message_consumer import MessageConsumer
-
+from rich.console import Console
+from rich.table import Table
+from rich.text import Text
+from rich.panel import Panel
 
 class RabbitMessageConsumer(MessageConsumer):
     """
@@ -175,3 +178,34 @@ class RabbitMessageConsumer(MessageConsumer):
             company_phone=message.get('company_phone', ''),
             status=message.get('status'),
         )
+
+    def reprer(self):
+        console = Console()
+
+        # Título "ChatGraph" destacado em vermelho e negrito dentro de um painel
+        title_text = Text("ChatGraph", style="bold red", justify="center")
+        title_panel = Panel.fit(title_text, title=" ", border_style="bold red", padding=(1, 4))
+
+        # Linha separadora com emojis
+        separator = Text("🐇🐇🐇 RabbitMessageConsumer 📨📨📨", style="cyan", justify="center")
+
+        # Criação da tabela com os atributos
+        table = Table(show_header=True, header_style="bold magenta", title="RabbitMQ Consumer")
+        table.add_column("Atributo", justify="center", style="cyan", no_wrap=True)
+        table.add_column("Valor", justify="center", style="magenta")
+
+        table.add_row("Virtual Host", self.__virtual_host)
+        table.add_row("Prefetch Count", str(self.__prefetch_count))
+        table.add_row("Queue Consume", self.__queue_consume)
+        table.add_row("AMQP URL", self.__amqp_url)
+        table.add_row("Username", self.__credentials.username)
+        table.add_row("Password", "******")  # Oculta a senha
+
+        # Imprime o título, separador e a tabela centralizada
+        console.print(title_panel, justify="center")
+        console.print(separator, justify="center")
+        console.print(table, justify="center")
+        
+c = Credential('user', 'pass')	
+consumer = RabbitMessageConsumer(c, "amqp://example.com", "queue_name", 10, "/virtual_host")
+consumer.reprer()

{chatgraph-0.2.2 → chatgraph-0.2.4}/chatgraph/types/output_state.py

@@ -28,10 +28,10 @@ class ChatbotResponse:
         '''
         Retorna o objeto em formato json.
         '''
-        return json.dumps({
+        return {
             'type': 'message',
             'message': self.message,
-        })
+        }
 
 
 class RedirectResponse:
@@ -71,11 +71,11 @@ class EndChatResponse:
         '''
         Retorna o objeto em formato json.
         '''
-        return json.dumps({
+        return {
             'type': 'tabulate',
             'tabulation_id': self.tabulation_id,
             'observations': self.observations,
-        })
+        }
         
 class TransferToHuman:
     """
@@ -92,8 +92,8 @@ class TransferToHuman:
         '''
         Retorna o objeto em formato json.
         '''
-        return json.dumps({
+        return {
             'type': 'transfer',
             'campaign_id': self.campaign_id,
             'observations': self.observations,
-        })
+        }

{chatgraph-0.2.2 → chatgraph-0.2.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "chatgraph"
-version = "0.2.2"
+version = "0.2.4"
 description = "A user-friendly chatbot library"
 authors = ["Irisson N. Lima <irisson.lima@verdecard.com.br>"]
 readme = "README.md"
@@ -12,6 +12,7 @@ license = "MIT"
 [tool.poetry.dependencies]
 python = "^3.12"
 pika = "^1.3.2"
+rich = "^13.8.1"
 
 [tool.poetry.group.dev.dependencies]
 ruff = "^0.5.1"