chatgraph 0.5.3__py3-none-any.whl → 0.6.0__py3-none-any.whl

chatgraph/types/end_types.py

@@ -1,4 +1,3 @@
-
 class RedirectResponse:
     """
     Representa uma resposta que redireciona o fluxo do chatbot para uma nova rota.
@@ -16,6 +15,7 @@ class RedirectResponse:
         """
         self.route = route
 
+
 class EndChatResponse:
     """
     Representa uma resposta que indica o fim do chatbot.
@@ -25,21 +25,35 @@ class EndChatResponse:
         observations (str): As observações finais do chatbot.
     """
 
-    def __init__(self, tabulation_id: str, observations:str) -> None:
-        '''
+    def __init__(self, tabulation_id: str, observations: str) -> None:
+        """
         Finzaliza e tabula as informações do chatbot.
-        '''
+        """
         self.tabulation_id = tabulation_id
         self.observations = observations
 
+
 class TransferToHuman:
     """
     Representa uma transferencia para um atendente humano.
     """
-    def __init__(self, campaign_id: str, observations:str) -> None:
-        '''
+
+    def __init__(self, campaign_id: str, observations: str) -> None:
+        """
         Finzaliza e tabula as informações do chatbot.
-        '''
+        """
         self.campaign_id = campaign_id
         self.observations = observations
-    
+
+
+class TransferToMenu:
+    """
+    Representa uma transferencia para outro Menu.
+    """
+
+    def __init__(self, menu: str, user_message: str) -> None:
+        """
+        Finzaliza e tabula as informações do chatbot.
+        """
+        self.menu = menu.lower()
+        self.user_message = user_message

chatgraph/types/image.py

@@ -0,0 +1,36 @@
+import hashlib
+import base64
+from chatgraph.types.message_types import Message
+
+
+class ImageData:
+    def __init__(self, url: str, image_bytes: bytes = None):
+        if not url and not image_bytes:
+            raise ValueError("URL or image bytes must be provided.")
+
+        self.url = url
+        self.image_bytes = image_bytes
+
+        # Create hash of image bytes if available, otherwise use URL
+        hash_input = image_bytes if image_bytes else url.encode("utf-8")
+
+        # Create SHA-256 hash and encode in base64
+        self.image_id = base64.b64encode(hashlib.sha256(hash_input).digest()).decode(
+            "utf-8"
+        )
+
+
+class SendImage:
+    def __init__(self, image: ImageData, message: Message):
+        if not isinstance(image, ImageData):
+            raise TypeError("Expected an instance of ImageData.")
+
+        self.type = "image"
+        self.image = image
+        self.message = message
+
+    def to_dict(self):
+        return {
+            "file_id": self.image.image_id,
+            "message": self.message.to_dict(),
+        }

chatgraph/types/request_types.py

@@ -1,4 +1,5 @@
 from chatgraph.gRPC.gRPCCall import RouterServiceClient
+from chatgraph.types.image import ImageData, SendImage
 from chatgraph.types.message_types import Message, Button, MessageTypes, messageTypes
 from typing import Optional
 from datetime import datetime
@@ -157,7 +158,7 @@ class UserCall:
 
     def __str__(self):
         return f"UserCall({self.type}, {self.__type_message}, {self.__content_message}, {self.__user_state})"
-    
+
     def send(
         self,
         message: messageTypes | Message,
@@ -174,11 +175,27 @@ class UserCall:
                 detail=str(message),
             )
 
+        if isinstance(message, ImageData):
+            message = SendImage(
+                image=message,
+                message=Message(type="message", detail=""),
+            )
+
         if isinstance(message, Message):
             self.__send(message)
+        elif isinstance(message, SendImage):
+            self.__send_image(message)
         else:
             raise ValueError("Tipo de mensagem inválido.")
 
+    def __send_image(self, message: SendImage) -> None:
+        dict_message = message.to_dict()
+        dict_message["message"]["chat_id"] = self.__user_state.chatID.to_dict()
+        response = self.__router_client.send_image(dict_message)
+
+        if not response.status:
+            raise ValueError("Erro ao enviar imagem.")
+
     def __send(self, message: Message) -> None:
 
         dict_message = message.to_dict()
@@ -192,26 +209,41 @@ class UserCall:
             raise ValueError("Erro ao enviar mensagem de botões.")
 
     def transfer_to_human(self, message: str, campaign_name: str) -> None:
-        
+
         campaign = self.__router_client.get_campaign_id({"name": campaign_name})
-        
+
         response = self.__router_client.transfer_to_human(
             {
                 "chat_id": self.__user_state.chatID.to_dict(),
                 "campaign_id": campaign.id,
+                "observation": message,
             }
         )
 
         if not response.status:
             raise ValueError("Erro ao transferir chat para humano.")
 
+    def transfer_to_menu(self, menu: str, message: str) -> None:
+
+        response = self.__router_client.transfer_to_menu(
+            {
+                "chat_id": self.__user_state.chatID.to_dict(),
+                "menu": menu,
+                "user_message": message,
+            }
+        )
+
+        if not response.status:
+            raise ValueError("Erro ao transferir chat para menu.")
+
     def end_chat(self, message: str, tabulation_name: str) -> None:
         tabulation = self.__router_client.get_tabulation_id({"name": tabulation_name})
-        
+
         response = self.__router_client.end_chat(
             {
                 "chat_id": self.__user_state.chatID.to_dict(),
                 "tabulation_id": tabulation.id,
+                "observation": message,
             }
         )
 
@@ -263,15 +295,15 @@ class UserCall:
     @property
     def protocol(self):
         return self.__user_state.protocol
-    
+
     @property
     def observation(self):
         return self.__user_state.observation
-    
+
     @property
     def type_message(self):
         return self.__type_message
-    
+
     @property
     def content_message(self):
         return self.__content_message
@@ -279,12 +311,16 @@ class UserCall:
     @menu.setter
     def menu(self, menu):
 
-        self.update_user_state(menu, self.__user_state.route, self.__user_state.observation)
+        self.update_user_state(
+            menu, self.__user_state.route, self.__user_state.observation
+        )
 
     @route.setter
     def route(self, route):
 
-        self.update_user_state(self.__user_state.menu, route, self.__user_state.observation)
+        self.update_user_state(
+            self.__user_state.menu, route, self.__user_state.observation
+        )
 
     @observation.setter
     def observation(self, observation):

{chatgraph-0.5.3.dist-info → chatgraph-0.6.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: chatgraph
-Version: 0.5.3
+Version: 0.6.0
 Summary: A user-friendly chatbot library
 Home-page: https://github.com/irissonnlima/chatgraph
 License: MIT
@@ -16,6 +16,7 @@ Requires-Dist: grpcio-tools (>=1.67.0,<2.0.0)
 Requires-Dist: matplotlib (>=3.10.0,<4.0.0)
 Requires-Dist: networkx (>=3.4.2,<4.0.0)
 Requires-Dist: pika (>=1.3.2,<2.0.0)
+Requires-Dist: protobuf (>=6.31.1,<7.0.0)
 Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
 Requires-Dist: rich (>=13.8.1,<14.0.0)
 Requires-Dist: typer (>=0.12.5,<0.13.0)

chatgraph-0.6.0.dist-info/RECORD

@@ -0,0 +1,23 @@
+chatgraph/__init__.py,sha256=ENRmvWkLHfwErug_flNYgbTa4MUMrIzMqGeaFSlSNhM,929
+chatgraph/auth/credentials.py,sha256=xsMEpLQnc66novPjL6upocMcnUnvFJ7yxINzUenkxmc,2388
+chatgraph/bot/chatbot_model.py,sha256=wLVXehmHEWItRwvl6J8HM96lC69FXmqsNG8KqplBboI,7373
+chatgraph/bot/chatbot_router.py,sha256=bVbm9dBoC2OesetXlQJpCrkaiLM3GUro0GrOyCSgreI,2586
+chatgraph/cli/__init__.py,sha256=tfgYhGoKy1nD4STN4xDh6J4VV55RICk7v1GZmzAg-bM,7413
+chatgraph/error/chatbot_error.py,sha256=4sEcW8vz0eQk2QDmDygucVM4caHliZW5iH7XJvmLBuk,1897
+chatgraph/error/route_error.py,sha256=CY8m82ap7-Sr-DXPsolltRW50TqD__5RyXBmNNJCWr8,793
+chatgraph/gRPC/gRPCCall.py,sha256=8t3Il6C-x6LqvdiUOZ5absfVwAfEWY7MpEfmEVNMgVY,5776
+chatgraph/messages/message_consumer.py,sha256=yHhfLYjtmXQYG44o7SNO6XUaeM4-jH8BpIpPHI-mt8Y,6116
+chatgraph/pb/router.proto,sha256=DkP_ESSy2BNO7mMqtXAmFafsz4OLtxj-IoKBzeDout4,3308
+chatgraph/pb/router_pb2.py,sha256=ohYjPoSwbYfFJp1JQIvi7F-BlrHkifeIEvF5KzfjD8g,7426
+chatgraph/pb/router_pb2_grpc.py,sha256=-khhvX4epBDOqGUhQEhuJSAX_Kh7iJBG3csC18Aa73g,28287
+chatgraph/types/background_task.py,sha256=j-Bpd1l_LCDiWukhfG8wNr1F0_GUaOL_J2_BCUnTof4,928
+chatgraph/types/end_types.py,sha256=ulVq3XybRs-EGcOxw6e1FF6U2p3nbtPfOv3uRe6oJfo,1638
+chatgraph/types/image.py,sha256=f2PnUo4Jp24Cn1F4NQhgeoGJPacQiMZCogVowjJwrIE,1097
+chatgraph/types/message_types.py,sha256=__LOk09Xfvojum0xiq9n157_3QSSLNaDMc02iq8dGpk,3434
+chatgraph/types/request_types.py,sha256=stiEDOtFxn22z2bpgYlIcfSM0ellqVbPaJC2noY8mwQ,9957
+chatgraph/types/route.py,sha256=FO5INy5UXgicuQ8FKEZKcPv6WS5YH10dPx2OaPu_0sE,2978
+chatgraph-0.6.0.dist-info/entry_points.txt,sha256=bO9_Q-PqE5vCNNo6ke_-3j2gHfKQMDGnKDTkNuCdBuA,48
+chatgraph-0.6.0.dist-info/LICENSE,sha256=rVJozpRzDlplOpvI8A1GvmfVS0ReYdZvMWc1j2jV0v8,1090
+chatgraph-0.6.0.dist-info/METADATA,sha256=L5LzW2b54Jj0GzFDBUOu_95HztOVbtCMB0VVa1ozjsA,12908
+chatgraph-0.6.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+chatgraph-0.6.0.dist-info/RECORD,,

chatgraph-0.5.3.dist-info/RECORD

@@ -1,22 +0,0 @@
-chatgraph/__init__.py,sha256=QtS8l6UjQ7N9gPNruTaPKUTNP8-EvTYeiTBoiSKGp7E,781
-chatgraph/auth/credentials.py,sha256=xsMEpLQnc66novPjL6upocMcnUnvFJ7yxINzUenkxmc,2388
-chatgraph/bot/chatbot_model.py,sha256=PhWCDDTIfT92rwS-USciugVjXuCDKk44NJbiOfEhvRk,7051
-chatgraph/bot/chatbot_router.py,sha256=bVbm9dBoC2OesetXlQJpCrkaiLM3GUro0GrOyCSgreI,2586
-chatgraph/cli/__init__.py,sha256=tfgYhGoKy1nD4STN4xDh6J4VV55RICk7v1GZmzAg-bM,7413
-chatgraph/error/chatbot_error.py,sha256=4sEcW8vz0eQk2QDmDygucVM4caHliZW5iH7XJvmLBuk,1897
-chatgraph/error/route_error.py,sha256=CY8m82ap7-Sr-DXPsolltRW50TqD__5RyXBmNNJCWr8,793
-chatgraph/gRPC/gRPCCall.py,sha256=d8nobBb8yDSKl_eFd1ZUfozvpyDZSoul2HlBIO8Q288,5308
-chatgraph/messages/message_consumer.py,sha256=yHhfLYjtmXQYG44o7SNO6XUaeM4-jH8BpIpPHI-mt8Y,6116
-chatgraph/pb/router.proto,sha256=pw3hTwjdMN8-yJIAlKHNigVKYqM-w2nme9-2bCTBiJ0,2853
-chatgraph/pb/router_pb2.py,sha256=jw-Rg05NMwgOKrRsoubFFb82v_O80KgWY4tIqwUf2Aw,6571
-chatgraph/pb/router_pb2_grpc.py,sha256=KxhVxCMVD9tpuX7CyXqIkzZ4P3zfXeVT92Bt_Hi7mrE,24109
-chatgraph/types/background_task.py,sha256=j-Bpd1l_LCDiWukhfG8wNr1F0_GUaOL_J2_BCUnTof4,928
-chatgraph/types/end_types.py,sha256=--Ty2gM_y7J-yRAvZV26e4HMUpoguAMAhfOIS9-kQTk,1316
-chatgraph/types/message_types.py,sha256=__LOk09Xfvojum0xiq9n157_3QSSLNaDMc02iq8dGpk,3434
-chatgraph/types/request_types.py,sha256=BeUI2l9XZpBf5c8xyPvKI68WZ5u6n98cC8DmUfWKlQ0,8788
-chatgraph/types/route.py,sha256=FO5INy5UXgicuQ8FKEZKcPv6WS5YH10dPx2OaPu_0sE,2978
-chatgraph-0.5.3.dist-info/entry_points.txt,sha256=bO9_Q-PqE5vCNNo6ke_-3j2gHfKQMDGnKDTkNuCdBuA,48
-chatgraph-0.5.3.dist-info/LICENSE,sha256=rVJozpRzDlplOpvI8A1GvmfVS0ReYdZvMWc1j2jV0v8,1090
-chatgraph-0.5.3.dist-info/METADATA,sha256=X4TtQbT5Bcpbaes_XIrLjuoELAMuYU3y105R8F7Ilnw,12866
-chatgraph-0.5.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-chatgraph-0.5.3.dist-info/RECORD,,