Add get user chatxbt id rpc method

src/tools/user_profile_toolkit.py

@@ -1,7 +1,10 @@
+import asyncio
 import chainlit as cl
 from phi.tools import Toolkit
 from phi.utils.log import logger
 
+from src.libs.rpc_client import rpc_call
+
 
 class UserProfileToolkit(Toolkit):
     def __init__(self):
@@ -16,6 +19,7 @@ class UserProfileToolkit(Toolkit):
         self.register(self.update_user_name)
         self.register(self.update_user_email)
         self.register(self.update_user_picture)
+        self.register(self.get_user_id)
 
     @cl.on_chat_start
     def get_user_info(self, info_type: str) -> str:
@@ -163,7 +167,7 @@ class UserProfileToolkit(Toolkit):
         str: Confirmation message.
 
         Example:
-        >>> update_user_email('jane.doe@example.com')
+        >>> update_user_email('jane.doe@chatxbt.com')
         """
         response = self.update_user_info('email', email)
         return f"{response}"
@@ -183,3 +187,24 @@ class UserProfileToolkit(Toolkit):
         """
         response = self.update_user_info('picture', picture)
         return f"{response}"
+    
+    def get_user_id(self, user_email: str) -> str:
+        """
+        Fetches the ChatXBT ID of the user associated with the given email.
+
+        Args:
+        user_email (str): The email address of the user whose ChatXBT ID needs to be fetched.
+
+        Returns:
+        str]: A string containing the ChatXBT ID of the user. If the user is not found, an empty string is returned.
+
+        Example:
+        >>> get_user_id('jane.doe@chatxbt.com')
+        """
+        logger.info("Fetching user ChatXBT ID")
+
+        params = {
+            "userEmail": user_email.lower()
+        }
+        response = asyncio.run(rpc_call(method_name="getUserId", params=params))
+        return f"{response}"