oneliai 0.1.0__py3-none-any.whl → 0.3.0__py3-none-any.whl

oneliai/client.py

@@ -1,7 +1,13 @@
 import requests
+import asyncio
+import uuid
+import logging
+import os
+from datetime import datetime
+URL="https://apis.oneli.chat"
+# URL="http://localhost:8085"
 
-# URL="https://apis.oneli.chat"
-URL="http://localhost:8085"
+appurl="http://localhost:3000"
 class AIClient:
     def __init__(self, client_id, client_secret, base_url=f'{URL}/v1/strategy'):
         self.client_id = client_id
@@ -23,10 +29,11 @@ class AIClient:
         else:
             raise Exception('Failed to get access token')
 
-    def generate_response(self, template_id, variables):
+    def generate_response(self, question,template_id, variables):
         response = requests.post(
             f'{self.base_url}/dynamic-response',
             json={
+                'question':question,
                 'template_id': template_id,
                 'variables': variables
             },
@@ -66,6 +73,7 @@ class AIClient:
         if response.status_code == 200:
             return response.json()
         else:
+            
             raise Exception('Failed to start intention query')
         
     def voc(self,productname,text):
@@ -95,62 +103,330 @@ class AIClient:
         if response.status_code == 200:
             return response.json()
         else:
-            raise Exception('Failed to start voc ')
+            res=response.json()
+            raise Exception(f'Failed to start voc,reason:{res["error"]}')
+
+     #选品建议   
+    def suggestion(self, selected_products):
+        response = requests.post(
+            f'{self.base_url}/suggestion',
+            json={
+                "final_selected_products": selected_products
+           
+            },
+            headers={'Authorization': f'Bearer {self.access_token}'}
+        )
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception('Failed to start suggestion')
+        
+    async def competitor_analysis(self,asins,missionid=""):
+        if not missionid:  # Check if missionid is empty
+           missionid = str(uuid.uuid4())  # Generate a random UUID
+        task_id = await self.get_competitive_data(missionid,asins)
+        logging.info(task_id)
+    
+        if task_id:
+            task = await self.check_task_status(task_id)
+            if task['status'] == "SUCCESS":
+                ret=await self.analysis_details(missionid)
+                return ret
+        
+    async def analysis_details(self,missionid):
+        response = requests.post(
+            f'{self.base_url}/conv/analysis/getid', 
+            json={
+                'missionid': missionid
+               
+            },
+            headers={"Content-Type": "application/json"}
+        )
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception('Failed to start suggestion')
+
+        
+    async def get_competitive_data(self,asins):
+        response = requests.post(
+                f'{self.base_url}/competitor_analysis',
+                json={
+                    "asins": asins
+            
+                },
+                headers={'Authorization': f'Bearer {self.access_token}'}
+            )
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception('Failed to start suggestion')
+    
+
+    #获取所有asins
+    async def productList(self,missionid):
+        data=await self.get_token()
+        print(data['token'])
+        response = requests.post(
+            f"{URL}/v1/agent/productList",
+            json={
+                'missionid':missionid
+            },
+            headers={'Authorization': f"Bearer {data['token']}"}
+        )
+
+        print(response)
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception('Failed to request task_status')
+
+
+
+    async def getTaskStatus(self,taskId):
+        response =requests.get(f"{URL}/task/task_status/{taskId}")
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception('Failed to request task_status')
+        
+    async def check_task_status(self,task_id):
+        while True:
+            try:
+                response = await self.get_task_status(task_id)
+                logging.info(response)
+                status = response.get('status')
+                logging.info(f"Task status: {status}")
+                
+                if status == "SUCCESS":
+                    return {"status": "SUCCESS", "result": response.get('result')}
+                    
+            except Exception as error:
+                logging.error(f"Error checking task status: {error}")
+            
+            await asyncio.sleep(1)  # Sleep for 1 second
+     
+
+    def getGoodinfofromAmazon(self, asins,filename=None):
+        if filename:
+            locafile,filename=self.upload_file(filename)
+            print(filename)
+        
+        
+            response=self.create_asin_mission(locafile,filename)
+            asins=response['asins']
+
+        response = requests.post(
+            f'{appurl}/api/run-task',
+            json={
+                "type": "asin",
+                "data":{"asins": asins}
+            
+            },
+        
+        )
+        
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception('Failed to run task')
+        
+    async def getAsinfromAmazon(self,title,keywords):
+        missionid = str(uuid.uuid4()) 
+        await self.update_mission("app_"+missionid, title,keywords)
+        response = requests.post(
+            f'{appurl}/api/run-task',
+            json={
+                "type": "list",
+                "data":{"keywords":keywords,"missionid":"app_"+missionid}
+            
+            },
+        
+        )
+
+
+        # if response.status_code == 202:
+        # task_id = response.json()['task_id']
+        # print(f"Task started with ID: {task_id}")
+        # return task_id
+        #print(response)
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception('Failed to run task')
+
+        
+    def upload_file(self,file_path):
+        url =f"{URL}/v1/task/upload"
+        file_name = os.path.basename(file_path)
+        with open(file_path, 'rb') as f:
+            files = {'file': (file_name, f)}
+            response = requests.post(url, files=files)
+        
+        if response.status_code == 200:
+            data = response.json()
+            print(f"Uploaded filename: {data['filename']}")
+            return file_name,data['filename']
+        else:
+            print(f"Upload failed: {response.text}")
+            return None
+
+
+    async def create_asin_mission(self,locafile,filename):
+        # Start ASIN extraction process
+        task_id = await self.create_asin_mission_api(filename)
+        
+        if task_id:
+            task = await self.check_task_status(task_id)
+            
+            if task["status"] == "SUCCESS" and task["result"]["code"] == 200:
+                missionid = task["result"]["missionid"]
+                
+                # Generate report title using AI
+                gentitle_prompt = f"""
+                以下是选品报告标题的关键要素
+                今天是{datetime.now()}
+                用户了提交了asin 列表,文件名称为{locafile}
+                请生成亚马逊选品分析报告的标题
+                """
+                response_title = await self.call_ai_api([{"role": "user", "content": gentitle_prompt}])
+                title = response_title["data"]
+                
+                # Update mission with generated title
+                task_res = await self.update_mission(missionid, title, "无需关键词")
+                
+                if task_res["code"] == 200:
+                    # Get ASIN list for the mission
+                    res = await self.get_asin_list(missionid)
+                    
+                    if res["code"] == 200:
+                        asinlist = res["data"]
+                        asins = [item["asin"] for item in asinlist]
+                        return {"missionid": missionid, "asins": asins}
+        
+        # Return None if any step fails
+        return None
+    
+
+
+    # Example implementations of the required service functions
+    async def create_asin_mission_api(self,filename: str) -> str:
+     
+        response = requests.post(
+            f"{URL}/task/start_task",
+            json={
+                'name':"create_asin_mission",
+                'data': {"file_name":filename},
+            }
+           
+        )
+        """Mock implementation - replace with actual API call"""
+        if response.status_code == 200:
+            res=response.json()
+            return res['task_id']
+        else:
+            raise Exception('Failed to request task_status')
+        
+
+    async def check_task_status(self,task_id: str) -> dict:
+        response =requests.get(f"{URL}/task/task_status/{task_id}")
+        if response.status_code == 200:
+            ret= response.json()
+            """Mock implementation - replace with actual status check"""
+            return {
+                "status": ret["status"],
+                "result": {
+                    "code": 200,
+                    "missionid": ret["result"]["missionid"]
+                }
+            }
+        else:
+            raise Exception('Failed to request task_status')
+    
+    async def get_token(self):
+        
+        response =requests.get(f"{appurl}/api/gettoken")
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception('Failed to request task_status')
+        # response =requests.get(f"{URL}/system/sdklogin")
+        # if response.status_code == 200:
+        #     return response.json()
+        # else:
+        #     raise Exception('Failed to request task_status')
+
+    async def call_ai_api(self,messages,tools):
+        data=await self.get_token()
+        print(data)
+        response = requests.post(
+            f"{URL}/agent/fn",
+            json={"tools":tools,"messages":messages},
+            headers={'Authorization': f"Bearer {data['token']}"}
+        )
+
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception('Failed to request task_status')
+        
+
+    async def update_mission(self,missionid: str, title: str, keywords: str) :
+        data=await self.get_token()
+        print(data['token'])
+        response = requests.post(
+            f"{URL}/v1/agent/intertMissioinlist",
+            json={
+                'task_id':missionid,
+                'report_title': title,
+                'keywords': keywords,
+                'task_status':'In Progress',
+                'task_status_description':'数据采集任务开始'
+            },
+            headers={'Authorization': f"Bearer {data['token']}"}
+        )
+
+        print(response)
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception('Failed to request task_status')
+        
+
+    async def get_asin_list(self,missionid: str):
+        response = requests.post(
+            f"{URL}/conv/getid",
+            json={
+                'missionid':missionid}
+           )
+
+        
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception('Failed to request task_status')
+        
+
+        
+
+
+    async def getallpageData(self,missionid: str) :
+        data=await self.get_token()
+        print(data['token'])
+        response = requests.post(
+            f"{URL}/v1/agent/test/getallpageData",
+            json={
+                'missionid':missionid
+            },
+            headers={'Authorization': f"Bearer {data['token']}"}
+        )
+
+        print(response)
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception('Failed to request task_status')
+            
         
-    # def registEndpoint(self,  name,endpointpath):
-    #     response = requests.post(
-    #         f'{self.base_url}/createEndpoints',
-    #         json={
-    #             "endpointpath": endpointpath,
-    #             "method": "POST",
-    #             "name":name
-    #         },
-    #         headers={'Authorization': f'Bearer {self.access_token}'}
-    #     )
-    #     if response.status_code == 200:
-    #         return response.json()
-    #     else:
-    #         raise Exception('Failed to createEndpoints')
-        
-
-
-    # def createRoles(self, role_name, description):
-    #     response = requests.post(
-    #         f'{self.base_url}/createRoles',
-    #         json={
-    #             "role_name": role_name,
-    #             "description":description
-    #         },
-    #         headers={'Authorization': f'Bearer {self.access_token}'}
-    #     )
-    #     if response.status_code == 200:
-    #         return response.json()
-    #     else:
-    #         raise Exception('Failed to createRoles')
-        
-    # def roles_endpoint(self, role_id, endpoint_id):
-    #     response = requests.post(
-    #         f'{self.base_url}/roles/{role_id}/endpoints',
-    #         json={
-    #             "endpoint_id": endpoint_id
-    #         },
-    #         headers={'Authorization': f'Bearer {self.access_token}'}
-    #     )
-    #     if response.status_code == 200:
-    #         return response.json()
-    #     else:
-    #         raise Exception('Failed to roles_endpoint')
-        
-
-    # def user_roles(self, user_id, role_id):
-    #     response = requests.post(
-    #         f'{self.base_url}/users/{user_id}/roles',
-    #         json={
-    #             "role_id": role_id
-    #         },
-    #         headers={'Authorization': f'Bearer {self.access_token}'}
-    #     )
-    #     if response.status_code == 200:
-    #         return response.json()
-    #     else:
-    #         raise Exception('Failed to user_roles')
+    
+ 

{oneliai-0.1.0.dist-info → oneliai-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oneliai
-Version: 0.1.0
+Version: 0.3.0
 Summary: SDK for ONELI.AI Customer API
 Author: oneli.ai
 Requires-Python: >=3.6

oneliai-0.3.0.dist-info/RECORD

@@ -0,0 +1,8 @@
+oneliai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+oneliai/client.py,sha256=-sjQayxFFcRd9wueMiaVb7GCkJpSWZqNT-4KpcAyBJ0,14400
+sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sdk/client.py,sha256=FwUhqzY2DkRTloe_XMyaaaEzg7THaIV6OxPDliMQjq0,4267
+oneliai-0.3.0.dist-info/METADATA,sha256=J1A-JNIohocFBTrPrJJ40W3RpvuqMwHBrfF7DnaXyq0,155
+oneliai-0.3.0.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+oneliai-0.3.0.dist-info/top_level.txt,sha256=GqjKHlaU6NsCzFm5EOy-2oJoa3YStKZAX8qonkZqB1M,8
+oneliai-0.3.0.dist-info/RECORD,,

oneliai-0.1.0.dist-info/RECORD

@@ -1,8 +0,0 @@
-oneliai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-oneliai/client.py,sha256=-e3RToSTR2T54C06v9CG-6OH4fXZCvUg3svCbZ8tXC4,5237
-sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sdk/client.py,sha256=FwUhqzY2DkRTloe_XMyaaaEzg7THaIV6OxPDliMQjq0,4267
-oneliai-0.1.0.dist-info/METADATA,sha256=iKDLgkW148wBs4EdcvJS9gCKtSZzPKBmesWZFa_v914,155
-oneliai-0.1.0.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-oneliai-0.1.0.dist-info/top_level.txt,sha256=GqjKHlaU6NsCzFm5EOy-2oJoa3YStKZAX8qonkZqB1M,8
-oneliai-0.1.0.dist-info/RECORD,,