chatgraph 0.3.1__py3-none-any.whl → 0.3.2a0__py3-none-any.whl

chatgraph/__init__.py

@@ -2,8 +2,9 @@ from .auth.credentials import Credential
 from .bot.chatbot_model import ChatbotApp
 from .bot.chatbot_router import ChatbotRouter
 from .messages.message_consumer import MessageConsumer
-from .types.message_types import UserCall, UserState, Element
-from .types.output_state import ChatbotResponse, RedirectResponse, EndChatResponse, TransferToHuman
+from .types.request_types import UserCall, UserState
+from .types.end_types import RedirectResponse, EndChatResponse, TransferToHuman
+from .types.message_types import Message, Button, ListElements
 from .types.route import Route
 
 __all__ = [
@@ -18,5 +19,7 @@ __all__ = [
     'EndChatResponse',
     'TransferToHuman',
     'UserState',
-    'Element',
+    'Message',
+    'Button',
+    'ListElements'
 ]

chatgraph/bot/chatbot_model.py

@@ -6,8 +6,9 @@ from logging import error
 
 from ..error.chatbot_error import ChatbotError, ChatbotMessageError
 from ..messages.message_consumer import MessageConsumer
-from ..types.message_types import UserCall
-from ..types.output_state import ChatbotResponse, RedirectResponse, EndChatResponse, TransferToHuman
+from ..types.request_types import UserCall
+from ..types.message_types import messageTypes, Message, Button, ListElements
+from ..types.end_types import RedirectResponse, EndChatResponse, TransferToHuman
 from ..types.route import Route
 from .chatbot_router import ChatbotRouter
 
@@ -140,16 +141,33 @@ class ChatbotApp:
             kwargs[route_state_name] = Route(route, list(self.__routes.keys()))
 
         userCall_response = func(**kwargs)
+        
+        if isinstance(userCall_response, (list, tuple)):
+            for response in userCall_response:
+                self.__process_func_response(response, userCall)
 
+    def __process_func_response(self, userCall_response, userCall: UserCall):
+        
         if isinstance(userCall_response, (str, float, int)):
-            return self.__create_response(userCall_response, customer_id, route, menu, obs)
+            userCall.send(Message(text=userCall_response))
+            return
 
-        elif isinstance(userCall_response, ChatbotResponse):
-            adjusted_route = self.__adjust_route(userCall_response.route, route)
-            return self.__create_response(userCall_response.json(), customer_id, adjusted_route, menu, obs)
+        elif isinstance(userCall_response, Route):
+            userCall.route = userCall_response.current
+            return
+            
+        elif isinstance(userCall_response, (Message, Button, ListElements)):
+            userCall.send(userCall_response)
+            
+            return
 
-        elif isinstance(userCall_response, (EndChatResponse, TransferToHuman)):
-            return self.__end_chat_response(userCall_response, customer_id)
+        elif isinstance(userCall_response, EndChatResponse):
+            userCall.end_chat(userCall_response.obersevations, userCall_response.tabulation_id)
+            return
+        
+        elif isinstance(userCall_response, TransferToHuman):
+            userCall.transfer_to_human(userCall_response.observations, userCall_response.campaign_id)
+            return
 
         elif isinstance(userCall_response, RedirectResponse):
             route = self.__adjust_route(userCall_response.route, route)
@@ -157,7 +175,7 @@ class ChatbotApp:
             return self.process_message(userCall)
 
         elif not userCall_response:
-            return self.__create_response('', customer_id, None, None, None)
+            return
 
         else:
             error('Tipo de retorno inválido!')
@@ -181,30 +199,4 @@ class ChatbotApp:
         if 'start' not in route:
             route = absolute_route + route
 
-        return route
-    
-    def __create_response(self, response, customer_id, route, menu, obs):
-        """
-        Cria uma resposta padronizada em formato JSON com o estado do usuário.
-        """
-        response = ChatbotResponse(response).json() if not isinstance(response, dict) else response
-        response['user_state'] = {
-            'customer_id': customer_id,
-            'route': route,
-            'menu': menu,
-            'obs': obs,
-        }
-        return json.dumps(response)
-
-    def __end_chat_response(self, response, customer_id):
-        """
-        Gera a resposta de finalização ou transferência de chat.
-        """
-        response = response.json()
-        response['user_state'] = {
-            'customer_id': customer_id,
-            'menu': None,
-            'route': None,
-            'obs': None,
-        }
-        return json.dumps(response)
+        return route

chatgraph/cli/__init__.py

@@ -0,0 +1,189 @@
+import typer
+from rich.console import Console
+from rich.table import Table
+from rich.text import Text
+from rich.panel import Panel
+from dotenv import load_dotenv
+from ..gRPC.gRPCCall import WhatsappServiceClient, UserStateServiceClient
+import os, re
+
+load_dotenv()
+app = typer.Typer()
+
+@app.command()
+def configs():
+    """Recupera os IDs necessários."""
+    console = Console()
+    
+    username = os.getenv('RABBIT_USER')
+    password = os.getenv('RABBIT_PASS')
+    url = os.getenv('RABBIT_URI')
+    queue = os.getenv('RABBIT_QUEUE')
+    prefetch = os.getenv('RABBIT_PREFETCH', 1)
+    vhost = os.getenv('RABBIT_VHOST', '/')
+    grpc = os.getenv('GRPC_URI')
+
+    
+    tableRabbit = Table(show_header=True, header_style="bold magenta", title="RabbitMQ Consumer")
+    tableRabbit.add_column("Atributo", justify="center", style="cyan", no_wrap=True)
+    tableRabbit.add_column("Valor", justify="center", style="magenta")
+
+    tableRabbit.add_row("Virtual Host", vhost)
+    tableRabbit.add_row("Prefetch Count", str(prefetch))
+    tableRabbit.add_row("Queue Consume", queue)
+    tableRabbit.add_row("AMQP URL", url)
+    tableRabbit.add_row("Username", username)
+    tableRabbit.add_row("Password", "******")
+    
+    
+    tableGRPC = Table(show_header=True, header_style="bold magenta", title="gRPC Consumer")
+    tableGRPC.add_column("Atributo", justify="center", style="cyan", no_wrap=True)
+    tableGRPC.add_column("Valor", justify="center", style="magenta")
+    
+    tableGRPC.add_row("URI", grpc)
+    
+    console.print(tableGRPC, justify="center")
+    console.print(tableRabbit, justify="center")
+
+@app.command()
+def campaigns(regex: str = typer.Option(None, "--regex", "-r", help="Filtro regex para campanhas.")):
+    """Recupera as campanhas cadastradas."""
+    grpc = os.getenv('GRPC_URI')
+    
+    wwp = WhatsappServiceClient(grpc)
+    campaigns = wwp.get_all_campaigns()
+    
+    console = Console()
+    tableGRPC = Table(show_header=True, header_style="bold magenta", title="Campaigns")
+    tableGRPC.add_column("campaign_id", justify="center", style="cyan", no_wrap=True)
+    tableGRPC.add_column("campaign_name", justify="center", style="magenta")
+    tableGRPC.add_column("last_update", justify="center", style="magenta")
+    
+    # Aplicar filtro de regex, se fornecido
+    filtered_campaigns = campaigns.campaigns
+    if regex:
+        pattern = re.compile(regex)
+        filtered_campaigns = [
+            campaign for campaign in campaigns.campaigns
+            if pattern.search(campaign.campaign_name)
+        ]
+    
+    for campaign in filtered_campaigns:
+        tableGRPC.add_row(campaign.campaign_id, campaign.campaign_name, campaign.last_update)
+    
+    console.print(tableGRPC, justify="center")
+    
+@app.command()
+def tabulations(
+    regex: str = typer.Option(None, "--regex", "-r", help="Filtro regex para as tabulações.")
+):
+    """Recupera as tabulações cadastradas."""
+    grpc = os.getenv('GRPC_URI')
+    
+    wwp = WhatsappServiceClient(grpc)
+    tabulations = wwp.get_all_tabulations()
+    
+    console = Console()
+    tableGRPC = Table(show_header=True, header_style="bold magenta", title="Tabulations")
+    tableGRPC.add_column("tabulation_id", justify="center", style="cyan", no_wrap=True)
+    tableGRPC.add_column("tabulation_name", justify="center", style="magenta")
+    tableGRPC.add_column("tabulation_type", justify="center", style="magenta")
+    tableGRPC.add_column("group_name", justify="center", style="magenta")
+    tableGRPC.add_column("customer_service_survey_id", justify="center", style="magenta")
+    tableGRPC.add_column("last_update", justify="center", style="magenta")
+    
+    # Aplicar filtro de regex, se fornecido
+    filtered_tabulations = tabulations.tabulations
+    if regex:
+        pattern = re.compile(regex, re.IGNORECASE)
+        filtered_tabulations = [
+            tabulation for tabulation in tabulations.tabulations
+            if (
+                pattern.search(tabulation.tabulation_id) or
+                pattern.search(tabulation.tabulation_name) or
+                pattern.search(tabulation.tabulation_type) or
+                pattern.search(tabulation.group_name) or
+                (tabulation.customer_service_survey_id and pattern.search(tabulation.customer_service_survey_id)) or
+                pattern.search(tabulation.last_update)
+            )
+        ]
+    
+    for tabulation in filtered_tabulations:
+        tableGRPC.add_row(
+            tabulation.tabulation_id,
+            tabulation.tabulation_name,
+            tabulation.tabulation_type,
+            tabulation.group_name,
+            tabulation.customer_service_survey_id or "N/A",
+            tabulation.last_update,
+        )
+    
+    console.print(tableGRPC, justify="center")
+
+
+@app.command("ustate")
+def user_state(
+    regex: str = typer.Option(None, "--regex", "-r", help="Filtro regex para os estados do usuário.")
+):
+    """Recupera os UserState em operação no momento."""
+    grpc = os.getenv('GRPC_URI')
+    
+    ustate = UserStateServiceClient(grpc)
+    userstates = ustate.get_all_user_states()
+    
+    console = Console()
+    tableGRPC = Table(show_header=True, header_style="bold magenta", title="User States")
+    tableGRPC.add_column("user_id", justify="center", style="cyan", no_wrap=True)
+    tableGRPC.add_column("menu_id", justify="center", style="magenta")
+    tableGRPC.add_column("route", justify="center", style="magenta")
+    tableGRPC.add_column("obs", justify="center", style="magenta")
+    tableGRPC.add_column("date", justify="center", style="magenta")
+    tableGRPC.add_column("direction", justify="center", style="magenta")
+    
+    # Aplicar filtro de regex, se fornecido
+    filtered_user_states = userstates.user_states
+    if regex:
+        pattern = re.compile(regex, re.IGNORECASE)
+        filtered_user_states = [
+            userstate for userstate in userstates.user_states
+            if (
+                pattern.search(userstate.user_id) or
+                pattern.search(userstate.menu_id) or
+                pattern.search(userstate.route) or
+                pattern.search(userstate.obs) or
+                pattern.search(userstate.date) or
+                pattern.search(str(userstate.direction))
+            )
+        ]
+    
+    for userstate in filtered_user_states:
+        tableGRPC.add_row(
+            userstate.user_id,
+            userstate.menu_id,
+            userstate.route,
+            userstate.obs,
+            userstate.date,
+            str(userstate.direction),
+        )
+    
+    console.print(tableGRPC, justify="center")
+
+@app.command("del-ustate")
+def delete_user_state(user_id: str = typer.Argument(..., help="ID do UserState a ser deletado.")):
+    """Deleta um UserState em operação no momento."""
+    grpc = os.getenv('GRPC_URI')
+    
+    ustate = UserStateServiceClient(grpc)
+    
+    # Chama o método para deletar o UserState usando o ID fornecido
+    try:
+        success = ustate.delete_user_state(user_id)
+        if success:
+            typer.echo(f"UserState com ID '{user_id}' deletado com sucesso.")
+        else:
+            typer.echo(f"Falha ao deletar UserState com ID '{user_id}'.", err=True)
+    except Exception as e:
+        typer.echo(f"Erro ao tentar deletar UserState: {e}", err=True)
+    
+def main():
+    app()

chatgraph/gRPC/gRPCCall.py

@@ -22,6 +22,7 @@ class WhatsappServiceClient:
         
         # Cria o stub (client) para o serviço gRPC
         self.stub = whatsapp_pb2_grpc.MessageServiceStub(self.channel)
+        self.actions_stub = whatsapp_pb2_grpc.ActionsServiceStub(self.channel)
 
     def send_button(self, message_data):
         # Cria o request para o método SendButton
@@ -58,6 +59,78 @@ class WhatsappServiceClient:
         except grpc.RpcError as e:
             print(f"Erro ao fazer a requisição gRPC SendText: {e}")
             return None
+    
+    def transfer_to_human(self, message_data):
+        # Cria o request para o método TransferToHuman
+        request = whatsapp_pb2.MessageRequest(**message_data)
+
+        # Faz a chamada ao serviço gRPC
+        try:
+            response = self.stub.TransferToHuman(request)
+            return response
+        except grpc.RpcError as e:
+            print(f"Erro ao fazer a requisição gRPC TransferToHuman: {e}")
+            return None
+    
+    def end_chat(self, message_data):
+        # Cria o request para o método EndChat
+        request = whatsapp_pb2.MessageRequest(**message_data)
+
+        # Faz a chamada ao serviço gRPC
+        try:
+            response = self.stub.EndChat(request)
+            return response
+        except grpc.RpcError as e:
+            print(f"Erro ao fazer a requisição gRPC EndChat: {e}")
+            return None
+    
+    def get_campaign_id(self, campaign_name):
+        # Cria o request para o método GetCampaignID
+        request = whatsapp_pb2.CampaignName(campaign_name=campaign_name)
+
+        # Faz a chamada ao serviço gRPC
+        try:
+            response = self.actions_stub.GetCampaignID(request)
+            return response
+        except grpc.RpcError as e:
+            print(f"Erro ao fazer a requisição gRPC GetCampaignID: {e}")
+            return None
+        
+    def get_tabulation_id(self, tabulation_name):
+        # Cria o request para o método GetTabulationID
+        request = whatsapp_pb2.TabulationName(tabulation_name=tabulation_name)
+
+        # Faz a chamada ao serviço gRPC
+        try:
+            response = self.actions_stub.GetTabulationID(request)
+            return response
+        except grpc.RpcError as e:
+            print(f"Erro ao fazer a requisição gRPC GetTabulationID: {e}")
+            return None
+
+    def get_all_campaigns(self):
+        # Cria o request para o método GetAllCampaigns
+        request = whatsapp_pb2.Void()
+
+        # Faz a chamada ao serviço gRPC
+        try:
+            response = self.actions_stub.GetCampaignsList(request)
+            return response
+        except grpc.RpcError as e:
+            print(f"Erro ao fazer a requisição gRPC GetAllCampaigns: {e}")
+            return None
+    
+    def get_all_tabulations(self):
+        # Cria o request para o método GetAllTabulations
+        request = whatsapp_pb2.Void()
+
+        # Faz a chamada ao serviço gRPC
+        try:
+            response = self.actions_stub.GetTabulationsList(request)
+            return response
+        except grpc.RpcError as e:
+            print(f"Erro ao fazer a requisição gRPC GetAllTabulations: {e}")
+            return None
 
 class UserStateServiceClient:
     def __init__(self, grpc_uri=None):
@@ -122,4 +195,16 @@ class UserStateServiceClient:
             return response
         except grpc.RpcError as e:
             print(f"Erro ao fazer a requisição gRPC DeleteUserState: {e}")
+            return None
+    
+    def get_all_user_states(self):
+        # Cria o request para o método GetAllUserStates
+        request = userstate_pb2.Void()
+
+        # Faz a chamada ao serviço gRPC
+        try:
+            response = self.stub.GetAllUserStates(request)
+            return response
+        except grpc.RpcError as e:
+            print(f"Erro ao fazer a requisição gRPC GetAllUserStates: {e}")
             return None

chatgraph/messages/message_consumer.py

@@ -4,7 +4,7 @@ import os
 import pika
 from typing import Callable
 from ..auth.credentials import Credential
-from ..types.message_types import UserCall, UserState
+from ..types.request_types import UserCall, UserState
 from rich.console import Console
 from rich.table import Table
 from rich.text import Text
@@ -146,16 +146,8 @@ class MessageConsumer:
         message = body.decode()
         message_json = json.loads(message)
         pure_message = self.__transform_message(message_json)
-        response = process_message(pure_message)
+        process_message(pure_message)
 
-        ch.basic_publish(
-            exchange='',
-            routing_key=props.reply_to,
-            properties=pika.BasicProperties(
-                correlation_id=props.correlation_id
-            ),
-            body=str(response),
-        )
         ch.basic_ack(delivery_tag=method.delivery_tag)
 
     def __transform_message(self, message: dict) -> UserCall:
@@ -170,6 +162,10 @@ class MessageConsumer:
         """
         
         user_state = message.get('user_state', {})
+        obs = user_state.get('obs', {})
+        if isinstance(obs, str):
+            obs = json.loads(obs)
+            
         return UserCall(
             type=message.get('type', ''),
             text=message.get('text', ''),
@@ -178,7 +174,7 @@ class MessageConsumer:
                 menu=user_state.get('menu', ''),
                 route=user_state.get('route', ''),
                 lst_update=user_state.get('lst_update', ''),
-                obs=user_state.get('obs', {}),
+                obs=obs,
             ),
             channel=message.get('channel', ''),
             customer_phone=message.get('customer_phone', ''),

chatgraph/pb/userstate.proto

@@ -9,6 +9,11 @@ service UserStateService {
     rpc InsertUserState(UserState) returns (RequestStatus);
     rpc UpdateUserState(UserState) returns (RequestStatus);
     rpc DeleteUserState(UserStateId) returns (RequestStatus);
+
+    rpc GetAllUserStates(Void) returns (UserStatesList);
+}
+
+message Void {
 }
 
 message UserState {
@@ -17,6 +22,7 @@ message UserState {
     string route = 3;
     string obs = 4;
     string date = 5;
+    string direction = 6;
 }
 
 message UserStateId {
@@ -26,4 +32,8 @@ message UserStateId {
 message RequestStatus {
     bool status = 1;
     string message = 2;
+}
+
+message UserStatesList {
+    repeated UserState user_states = 1;
 }

chatgraph/pb/userstate_pb2.py

@@ -24,7 +24,7 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0fuserstate.proto\x12\tuserstate\"W\n\tUserState\x12\x0f\n\x07user_id\x18\x01 \x01(\t\x12\x0f\n\x07menu_id\x18\x02 \x01(\t\x12\r\n\x05route\x18\x03 \x01(\t\x12\x0b\n\x03obs\x18\x04 \x01(\t\x12\x0c\n\x04\x64\x61te\x18\x05 \x01(\t\"\x1e\n\x0bUserStateId\x12\x0f\n\x07user_id\x18\x01 \x01(\t\"0\n\rRequestStatus\x12\x0e\n\x06status\x18\x01 \x01(\x08\x12\x0f\n\x07message\x18\x02 \x01(\t2\x9e\x02\n\x10UserStateService\x12?\n\x0fSelectUserState\x12\x16.userstate.UserStateId\x1a\x14.userstate.UserState\x12\x41\n\x0fInsertUserState\x12\x14.userstate.UserState\x1a\x18.userstate.RequestStatus\x12\x41\n\x0fUpdateUserState\x12\x14.userstate.UserState\x1a\x18.userstate.RequestStatus\x12\x43\n\x0f\x44\x65leteUserState\x12\x16.userstate.UserStateId\x1a\x18.userstate.RequestStatusB\x10Z\x0e./pb/userstateb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0fuserstate.proto\x12\tuserstate\"\x06\n\x04Void\"j\n\tUserState\x12\x0f\n\x07user_id\x18\x01 \x01(\t\x12\x0f\n\x07menu_id\x18\x02 \x01(\t\x12\r\n\x05route\x18\x03 \x01(\t\x12\x0b\n\x03obs\x18\x04 \x01(\t\x12\x0c\n\x04\x64\x61te\x18\x05 \x01(\t\x12\x11\n\tdirection\x18\x06 \x01(\t\"\x1e\n\x0bUserStateId\x12\x0f\n\x07user_id\x18\x01 \x01(\t\"0\n\rRequestStatus\x12\x0e\n\x06status\x18\x01 \x01(\x08\x12\x0f\n\x07message\x18\x02 \x01(\t\";\n\x0eUserStatesList\x12)\n\x0buser_states\x18\x01 \x03(\x0b\x32\x14.userstate.UserState2\xde\x02\n\x10UserStateService\x12?\n\x0fSelectUserState\x12\x16.userstate.UserStateId\x1a\x14.userstate.UserState\x12\x41\n\x0fInsertUserState\x12\x14.userstate.UserState\x1a\x18.userstate.RequestStatus\x12\x41\n\x0fUpdateUserState\x12\x14.userstate.UserState\x1a\x18.userstate.RequestStatus\x12\x43\n\x0f\x44\x65leteUserState\x12\x16.userstate.UserStateId\x1a\x18.userstate.RequestStatus\x12>\n\x10GetAllUserStates\x12\x0f.userstate.Void\x1a\x19.userstate.UserStatesListB\x10Z\x0e./pb/userstateb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -32,12 +32,16 @@ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'userstate_pb2', _globals)
 if not _descriptor._USE_C_DESCRIPTORS:
   _globals['DESCRIPTOR']._loaded_options = None
   _globals['DESCRIPTOR']._serialized_options = b'Z\016./pb/userstate'
-  _globals['_USERSTATE']._serialized_start=30
-  _globals['_USERSTATE']._serialized_end=117
-  _globals['_USERSTATEID']._serialized_start=119
-  _globals['_USERSTATEID']._serialized_end=149
-  _globals['_REQUESTSTATUS']._serialized_start=151
-  _globals['_REQUESTSTATUS']._serialized_end=199
-  _globals['_USERSTATESERVICE']._serialized_start=202
-  _globals['_USERSTATESERVICE']._serialized_end=488
+  _globals['_VOID']._serialized_start=30
+  _globals['_VOID']._serialized_end=36
+  _globals['_USERSTATE']._serialized_start=38
+  _globals['_USERSTATE']._serialized_end=144
+  _globals['_USERSTATEID']._serialized_start=146
+  _globals['_USERSTATEID']._serialized_end=176
+  _globals['_REQUESTSTATUS']._serialized_start=178
+  _globals['_REQUESTSTATUS']._serialized_end=226
+  _globals['_USERSTATESLIST']._serialized_start=228
+  _globals['_USERSTATESLIST']._serialized_end=287
+  _globals['_USERSTATESERVICE']._serialized_start=290
+  _globals['_USERSTATESERVICE']._serialized_end=640
 # @@protoc_insertion_point(module_scope)

chatgraph/pb/userstate_pb2_grpc.py

@@ -5,7 +5,7 @@ import warnings
 
 import chatgraph.pb.userstate_pb2 as userstate__pb2
 
-GRPC_GENERATED_VERSION = '1.66.2'
+GRPC_GENERATED_VERSION = '1.67.0'
 GRPC_VERSION = grpc.__version__
 _version_not_supported = False
 
@@ -54,6 +54,11 @@ class UserStateServiceStub(object):
                 request_serializer=userstate__pb2.UserStateId.SerializeToString,
                 response_deserializer=userstate__pb2.RequestStatus.FromString,
                 _registered_method=True)
+        self.GetAllUserStates = channel.unary_unary(
+                '/userstate.UserStateService/GetAllUserStates',
+                request_serializer=userstate__pb2.Void.SerializeToString,
+                response_deserializer=userstate__pb2.UserStatesList.FromString,
+                _registered_method=True)
 
 
 class UserStateServiceServicer(object):
@@ -83,6 +88,12 @@ class UserStateServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def GetAllUserStates(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
 
 def add_UserStateServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -106,6 +117,11 @@ def add_UserStateServiceServicer_to_server(servicer, server):
                     request_deserializer=userstate__pb2.UserStateId.FromString,
                     response_serializer=userstate__pb2.RequestStatus.SerializeToString,
             ),
+            'GetAllUserStates': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetAllUserStates,
+                    request_deserializer=userstate__pb2.Void.FromString,
+                    response_serializer=userstate__pb2.UserStatesList.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'userstate.UserStateService', rpc_method_handlers)
@@ -224,3 +240,30 @@ class UserStateService(object):
             timeout,
             metadata,
             _registered_method=True)
+
+    @staticmethod
+    def GetAllUserStates(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/userstate.UserStateService/GetAllUserStates',
+            userstate__pb2.Void.SerializeToString,
+            userstate__pb2.UserStatesList.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

chatgraph/pb/voll.proto

@@ -8,6 +8,19 @@ service MessageService {
   rpc SendButton(MessageRequest) returns (MessageResponse);
   rpc SendList(MessageRequest) returns (MessageResponse);
   rpc SendText(MessageRequest) returns (MessageResponse);
+  rpc TransferToHuman(MessageRequest) returns (MessageResponse);
+  rpc EndChat(MessageRequest) returns (MessageResponse);
+}
+
+service ActionsService {
+  rpc GetCampaignID(CampaignName) returns (CampaignDetails);
+  rpc GetCampaignsList(Void) returns (CampaignsList);
+
+  rpc GetTabulationID(TabulationName) returns (TabulationDetails);
+  rpc GetTabulationsList(Void) returns (TabulationsList);
+}
+
+message Void {
 }
 
 message MessageRequest {
@@ -16,9 +29,12 @@ message MessageRequest {
   string unique_customer_id = 3;
   string message_title = 4;
   string message_text = 5;
-  string message_caption = 6;
-  string button_title = 7;
-  repeated QuickOption options = 8;
+  string platform = 6;
+  string message_caption = 7;
+  string button_title = 8;
+  string campaign_name = 9;
+  string tabulation_name = 10;
+  repeated QuickOption options = 11;
 }
 
 message MessageResponse {
@@ -29,4 +45,45 @@ message MessageResponse {
 message QuickOption {
   string title = 1;
   string description = 2;
+}
+
+message CampaignName {
+  string campaign_name = 1;
+}
+
+message TabulationName {
+  string tabulation_name = 1;
+}
+
+message CampaignDetails{
+  string campaign_name = 1;
+  string campaign_id = 2;
+}
+
+message CampaignElement{
+  string campaign_id = 1;
+  string campaign_name = 2;
+  string last_update = 3;
+}
+
+message CampaignsList{
+  repeated CampaignElement campaigns = 1;
+}
+
+message TabulationDetails{
+  string tabulation_name = 1;
+  string tabulation_id = 2;
+}
+
+message TabulationElement{
+  string tabulation_id = 1;
+  string tabulation_name = 2;
+  string tabulation_type = 3;
+  string group_name = 4;
+  string customer_service_survey_id = 5;
+  string last_update = 6;
+}
+
+message TabulationsList{
+  repeated TabulationElement tabulations = 1;
 }