v0.1.0

Author: OliverWales

README.md

@@ -1 +1,21 @@
-# python-sdk
+# Arva Python SDK
+
+This repository contains the source code for the Arva Python SDK.
+
+To install it, run:
+
+```SH
+pip install arva-sdk
+```
+
+To get started, create a client with:
+
+```Python
+from arva import Arva
+
+client = Arva(
+    api_key='<YOUR_API_KEY>',
+)
+```
+
+Full docs [here](https://platform.arva-ai.com/docs).

pyproject.toml

@@ -0,0 +1,30 @@
+[build-system]
+requires = ["setuptools>=45", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "arva-sdk"
+version = "0.1.0"
+description = "A Python SDK for the Arva API"
+readme = "README.md"
+authors = [{name = "Oli Wales"}]
+license = {file = "LICENSE"}
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.7",
+    "Programming Language :: Python :: 3.8",
+    "Programming Language :: Python :: 3.9",
+]
+keywords = ["arva", "sdk", "api"]
+dependencies = [
+    "requests",
+    "requests-toolbelt",
+]
+requires-python = ">=3.7"
+
+[project.urls]
+Homepage = "https://github.com/Arva-AI/python-sdk"

requirements.txt

@@ -0,0 +1,2 @@
+requests
+requests-toolbelt

src/arva/__init__.py

@@ -0,0 +1 @@
+from .arva import Arva

src/arva/arva.py

@@ -0,0 +1,76 @@
+from typing import List, Dict, Any, Optional, IO
+import requests
+import json
+import os
+from requests_toolbelt import MultipartEncoder
+
+class Arva:
+    def __init__(self, api_key: str, base_url: str = 'http://platform.arva-ai.com/api/v0') -> None:
+        self.api_key = api_key
+        self.base_url = base_url
+        self.customers = self.Customers(self)
+
+    class Customers:
+        def __init__(self, arva_instance: 'Arva') -> None:
+            self.arva_instance = arva_instance
+
+        def create(self, agent_id: str, registered_name: str, state: str) -> Dict[str, Any]:
+            url = f"{self.arva_instance.base_url}/customer/create"
+            headers = {'Authorization': f'Bearer {self.arva_instance.api_key}'}
+            payload = {
+                'registeredName': registered_name,
+                'state': state,
+                'agentId': agent_id
+            }
+            
+            response = requests.post(url, headers=headers, json=payload)
+            response.raise_for_status()
+            return response.json()
+        
+        def update(self, id: str, user_info_patch: Dict[str, Any], websites: Optional[List[str]] = None, files: Optional[List[IO[bytes]]] = None) -> Dict[str, Any]:
+            url = f"{self.arva_instance.base_url}/customer/update"
+            headers = {'Authorization': f'Bearer {self.arva_instance.api_key}'}
+
+            if websites is None:
+                websites = []
+            if files is None:
+                files = []
+
+            file_tuples = [
+                ('file', (os.path.basename(os.path.abspath(f.name)), f, 'application/pdf')) for f in files
+            ]
+
+            fields = [
+                ('customerId', id),
+                ('userInfoPatch', json.dumps(user_info_patch)),
+                ('websites', json.dumps(websites)),
+            ] + file_tuples
+
+            mp_encoder: MultipartEncoder = MultipartEncoder(fields=fields)
+            headers['Content-Type'] = mp_encoder.content_type
+
+            response = requests.post(url, headers=headers, data=mp_encoder)
+            response.raise_for_status()
+            return response.json()
+        
+        def getById(self, id: str) -> Dict[str, Any]:
+            url = f"{self.arva_instance.base_url}/customer/getById?id={id}"
+            headers = {'Authorization': f'Bearer {self.arva_instance.api_key}'}
+
+            response = requests.get(url, headers=headers)
+            response.raise_for_status()
+            return response.json()
+    
+        def review(self, id: str, verdict: str, reason: str, rfi: Optional[str] = None) -> Dict[str, Any]:
+            url = f"{self.arva_instance.base_url}/customer/review"
+            headers = {'Authorization': f'Bearer {self.arva_instance.api_key}'}
+            payload = {
+                'customerId': id,
+                'verdict': verdict,
+                'reason': reason,
+                'rfi': rfi
+            }
+            
+            response = requests.post(url, headers=headers, json=payload)
+            response.raise_for_status()
+            return response.json()