oneliai 0.1.0__py3-none-any.whl

oneliai/client.py

@@ -0,0 +1,156 @@
+import requests
+
+# URL="https://apis.oneli.chat"
+URL="http://localhost:8085"
+class AIClient:
+    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 voc(self,productname,text):
+        response = requests.post(
+            f'{self.base_url}/voc',
+            json={
+                'productname': productname,
+                'text':text
+            },
+            headers={'Authorization': f'Bearer {self.access_token}'}
+        )
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception('Failed to start voc ')
+        
+
+    def spec(self,asin):
+        response = requests.post(
+            f'{self.base_url}/spec', 
+            json={
+                'asin': asin
+               
+            },
+            headers={'Authorization': f'Bearer {self.access_token}'}
+        )
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception('Failed to start voc ')
+        
+    # 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/METADATA

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

oneliai-0.1.0.dist-info/RECORD

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

oneliai-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: bdist_wheel (0.37.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

oneliai-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+oneliai

sdk/client.py

@@ -0,0 +1,127 @@
+import requests
+
+# 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')