oneliai 0.3.0__tar.gz → 0.4.0__tar.gz

{oneliai-0.3.0 → oneliai-0.4.0}/PKG-INFO

@@ -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.3.0 → oneliai-0.4.0}/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 → oneliai-0.4.0}/oneliai.egg-info/PKG-INFO

@@ -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.3.0 → oneliai-0.4.0}/oneliai.egg-info/SOURCES.txt

@@ -5,4 +5,6 @@ oneliai.egg-info/PKG-INFO
 oneliai.egg-info/SOURCES.txt
 oneliai.egg-info/dependency_links.txt
 oneliai.egg-info/requires.txt
-oneliai.egg-info/top_level.txt
+oneliai.egg-info/top_level.txt
+sdk/__init__.py
+sdk/client.py

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

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

oneliai-0.4.0/sdk/client.py

@@ -0,0 +1,129 @@
+import requests
+import jwt
+import datetime
+
+# URL="https://apis.oneli.chat"
+URL="http://localhost:8085"
+class AICustomerClient:
+    def __init__(self, client_id, client_secret, base_url=f'{URL}/v1/strategy'):
+        self.client_id = client_id
+        self.client_secret = client_secret
+        self.base_url = base_url
+        self.access_token = self._get_access_token()
+
+    def _get_access_token(self):
+        response = requests.post(
+            f'{self.base_url}/auth/token',
+            json={
+                'client_id': self.client_id,
+                'client_secret': self.client_secret,
+                'grant_type': 'client_credentials'
+            }
+        )
+        if response.status_code == 200:
+            return response.json().get('access_token')
+        else:
+            raise Exception('Failed to get access token')
+
+    def generate_response(self, template_id, variables):
+        response = requests.post(
+            f'{self.base_url}/dynamic-response',
+            json={
+                'template_id': template_id,
+                'variables': variables
+            },
+            headers={'Authorization': f'Bearer {self.access_token}'}
+        )
+        if response.status_code == 200:
+            return response.json().get('response')
+        else:
+            return response.json()
+            # raise Exception('Failed to generate response')
+
+    def query_data(self, arg, template_id):
+        response = requests.post(
+            f'{self.base_url}/query-data',
+            json={
+                'arg': arg,
+                'template_id': template_id
+            },
+            headers={'Authorization': f'Bearer {self.access_token}'}
+        )
+        if response.status_code == 200:
+            return response.json()
+        else:
+            res=response.json()
+            raise Exception(res['error'])
+        
+
+    def query_intention(self, question):
+        response = requests.post(
+            f'{self.base_url}/query-intention',
+            json={
+                'question': question
+             
+            },
+            headers={'Authorization': f'Bearer {self.access_token}'}
+        )
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception('Failed to start intention query')
+        
+    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.3.0 → oneliai-0.4.0}/setup.py

@@ -19,7 +19,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='oneliai',  
-    version="0.3.0",          
+    version="0.4.0",          
     author="oneli.ai",
     description='SDK for ONELI.AI Customer API',
     # long_description=open("README.md").read(),