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

oneliai/client.py

@@ -121,21 +121,26 @@ class AIClient:
         else:
             raise Exception('Failed to start suggestion')
         
-    async def competitor_analysis(self,asins,missionid=""):
+    async def competitor_analysis(self,missionid="",asins=[]):
         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)
+       
+        # 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', 
+            f'{URL}/v1/conv/analysis/getid', 
             json={
                 'missionid': missionid
                
@@ -145,22 +150,29 @@ class AIClient:
         if response.status_code == 200:
             return response.json()
         else:
-            raise Exception('Failed to start suggestion')
+            raise Exception('Failed to start analysis')
 
         
-    async def get_competitive_data(self,asins):
+    async def get_competitive_data(self,missionid,asins):
+        print(2)
         response = requests.post(
                 f'{self.base_url}/competitor_analysis',
                 json={
-                    "asins": asins
+                    "asins": asins,
+                    "missionid":missionid
             
                 },
                 headers={'Authorization': f'Bearer {self.access_token}'}
             )
+        
         if response.status_code == 200:
-            return response.json()
+        
+           result=response.json()
+              
+           return result['data']['task_id']
         else:
-            raise Exception('Failed to start suggestion')
+            raise Exception('Failed to request get_competitive_data')
+     
     
 
     #获取所有asins
@@ -184,16 +196,24 @@ class AIClient:
 
 
     async def getTaskStatus(self,taskId):
-        response =requests.get(f"{URL}/task/task_status/{taskId}")
+        print("232332")
+        data=await self.get_token()
+        print(data['token'])
+        response =requests.get(f"{URL}/v1/task/task_status/{taskId}",
+                                headers={'Authorization': f"Bearer {data['token']}"})
+        print(response.json())
         if response.status_code == 200:
             return response.json()
         else:
             raise Exception('Failed to request task_status')
         
     async def check_task_status(self,task_id):
+        print(3)
+        
         while True:
             try:
-                response = await self.get_task_status(task_id)
+                
+                response = await self.getTaskStatus(task_id)
                 logging.info(response)
                 status = response.get('status')
                 logging.info(f"Task status: {status}")
@@ -328,20 +348,20 @@ class AIClient:
             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 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):
         

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

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

oneliai-0.4.0.dist-info/RECORD

@@ -0,0 +1,8 @@
+oneliai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+oneliai/client.py,sha256=Uqybrpr7xWouFjZPXYwOFsqTJfrioz8L9kYGTuqqEIY,14909
+sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sdk/client.py,sha256=EenoF_sh_EsovxqLCKfcs0_9mMV1LqC3GnysQupgL1g,4296
+oneliai-0.4.0.dist-info/METADATA,sha256=00szTOVCSu2lDKfo51cCJh4Wh4ntFzh6Y6qRkHxCmww,155
+oneliai-0.4.0.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+oneliai-0.4.0.dist-info/top_level.txt,sha256=7zBSOMx80pMmYiAfyZoqmb8S5_lloaTESpnFRO3KoYk,12
+oneliai-0.4.0.dist-info/RECORD,,

{oneliai-0.3.0.dist-info → oneliai-0.4.0.dist-info}/top_level.txt

@@ -1 +1,2 @@
 oneliai
+sdk

sdk/client.py

@@ -1,4 +1,6 @@
 import requests
+import jwt
+import datetime
 
 # URL="https://apis.oneli.chat"
 URL="http://localhost:8085"

oneliai-0.3.0.dist-info/RECORD

@@ -1,8 +0,0 @@
-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,,