add: update_overview, get_file_names, get_files

Author: maisiukartyom

flow_prompt/prompt/flow_prompt.py

@@ -199,6 +199,59 @@ def add_ideal_answer(
 
         return response
 
+    def update_overview(self, overview: str, user_id: str = None):
+        """Update user's overview
+
+        Args:
+            user_id (str): user id,
+            overview (str): new overview to replace the old one
+        """
+        
+        response = FlowPromptService.update_user_overview(user_id, overview, self.api_token)
+        
+        return response
+    
+    def get_file_names(self, prefix: str, user_id: str = None):
+        """Fetch all filenames of the given user
+
+        Args:
+            prefix (str): s3 bucket folder name to fetch from
+            user_id (str): user identifier
+
+        Returns:
+            list: list of file names
+        """
+        response = FlowPromptService.get_file_names(prefix, user_id, self.api_token)
+        
+        return response
+
+    def get_files(self, paths: list[str], user_id: str = None):
+        """Method to fetch file contents by the provided s3 paths
+
+        Args:
+            paths (list[str]): paths to s3 bucket files
+            user_id (str): user identifier
+        
+        Returns: 
+            dict: key = path, value: file content 
+        """
+        
+        response = FlowPromptService.get_files(paths, user_id, self.api_token)
+        
+        return response
+    
+    def save_files(self, files: dict, user_id: str = None):
+        """Method to save files into FPS S3 bucket
+
+        Args:
+            files (dict): dictionary where key = file_name (relative path), val = file_content
+        """
+        
+        response = FlowPromptService.save_files(files, user_id, self.api_token)
+        
+        return response
+    
+    
     def get_pipe_prompt(self, prompt_id: str, version: str = None) -> PipePrompt:
         """
         if the user has keys:  lib -> service: get_actual_prompt(local_prompt) -> Service:

flow_prompt/services/flow_prompt/__init__.py

@@ -194,3 +194,91 @@ def create_test_with_ideal_answer(
         json_data = json.dumps(data)
         requests.post(url, headers=headers, data=json_data)
         logger.info(f"Created Ci/CD for prompt {prompt_data['prompt_id']}")
+
+    @classmethod
+    def update_user_overview(
+        cls,
+        user_id: str,
+        overview: str,
+        api_token: str
+    ):
+        url = f"{cls.url}lib/files?updateOverview"
+        headers = {"Authorization": f"Token {api_token}"}
+        data = {
+            'user_id': user_id,
+            "overview": overview,
+        }
+        json_data = json.dumps(data)
+        logger.debug(f"Request to {url} with data: {data}")
+        response = requests.post(url, headers=headers, data=json_data)
+        logger.info(f"Update overview of the user: ${user_id}")
+        
+        return response
+    
+    @classmethod
+    def get_file_names(
+        cls, 
+        prefix: str,
+        user_id: str,
+        api_token: str,
+    ):
+        url = f"{cls.url}lib/files?getFileNames&prefix={prefix}&user_id={user_id}"
+        headers = {"Authorization": f"Token {api_token}"}
+
+        logger.debug(f"Request to {url}")
+        response = requests.get(url, headers=headers)
+
+        if response.status_code == 200:
+            logger.info(f"Fetched filenames for user - ${user_id} from {prefix}")
+            return response.json()
+        else:
+            logger.error(response)
+            return response
+        
+    @classmethod
+    def get_files(
+        cls,
+        paths: list,
+        user_id: str,
+        api_token: str,
+    ):
+        url = f"{cls.url}lib/files?getFiles"
+        headers = {"Authorization": f"Token {api_token}"}
+        data = {
+            'user_id': user_id,
+            'paths': paths
+        }
+        
+        json_data = json.dumps(data)
+        response = requests.post(url=url, headers=headers, data=json_data)
+
+        if response.status_code == 200:
+            logger.info(f"Fetched files for user - ${user_id}")
+            return response.json()
+        else:
+            logger.error(response)
+            return response
+    
+    @classmethod
+    def save_files(
+        cls, 
+        files: dict,
+        user_id: str,
+        api_token: str,
+    ):
+        url = f"{cls.url}lib/files?saveFiles"
+        headers = {"Authorization": f"Token {api_token}"}
+        data = {
+            'user_id': user_id,
+            'files': files,
+        }
+        
+        json_data = json.dumps(data)
+        response = requests.post(url=url, headers=headers, data=json_data)
+
+        if response.status_code == 200:
+            logger.info(f"Saved files for user - ${user_id}")
+            return response.json()
+        else:
+            logger.error(response)
+            return response

tests/prompts/test_files.py

@@ -0,0 +1,33 @@
+import logging
+import os
+from pytest import fixture
+from flow_prompt import FlowPrompt
+logger = logging.getLogger(__name__)
+
+@fixture
+def fp():
+    api_token = os.getenv("FLOW_PROMPT_API_TOKEN")
+    flow_prompt = FlowPrompt(
+        api_token=api_token
+    )
+    return flow_prompt
+
+
+def test_elytimes(fp):
+
+    user_id = '94187488-9041-7011-ac07-58581cb3f737'
+    overview = "I'm handsome"
+
+    # response = fp.get_file_names('/user', user_id)
+    response = fp.update_overview(overview)
+    # response = fp.get_files([
+    #     'user/file1.txt', 'company/file1.txt'
+    # ], user_id)
+    
+    #  print(response)
+    # fp.save_files({
+    #     'user/file1.txt': 'Hello world',
+    #     'company/file1.txt': 'Hello company!'
+    # }, user_id)
+    
+    # assert 'file_contents' in response