mixpeek 0.6.26__tar.gz → 0.6.28__tar.gz

{mixpeek-0.6.26 → mixpeek-0.6.28}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mixpeek
-Version: 0.6.26
+Version: 0.6.28
 Summary: Mixpeek Python SDK
 Home-page: https://github.com/mixpeek/mixpeek-python
 Author: Ethan Steininger
@@ -101,6 +101,15 @@ mixpeek.connections.data.delete(
 )
 ```
 
+Upload file
+
+```python
+mixpeek.connections.storage.upload(
+    connection_id="conn_321",
+    file_path="/my/local/file.mp4"
+)
+```
+
 ### Tools
 
 ```python

{mixpeek-0.6.26 → mixpeek-0.6.28}/README.md

@@ -86,6 +86,15 @@ mixpeek.connections.data.delete(
 )
 ```
 
+Upload file
+
+```python
+mixpeek.connections.storage.upload(
+    connection_id="conn_321",
+    file_path="/my/local/file.mp4"
+)
+```
+
 ### Tools
 
 ```python

{mixpeek-0.6.26 → mixpeek-0.6.28}/mixpeek/client.py

@@ -11,7 +11,7 @@ from .endpoints.pipelines import Pipelines
 class Mixpeek:
     def __init__(self, api_key: str):
         self.api_key = api_key
-        self.base_url = "http://localhost:8000/"
+        self.base_url = "https://api.mixpeek.com/"
         self.headers = {
             "Authorization": f"Bearer {self.api_key}",
             "Content-Type": "application/json"
@@ -19,6 +19,8 @@ class Mixpeek:
         self.extract = Extract(self.base_url, self.headers)
         self.embed = Embed(self.base_url, self.headers)
         self.generate = Generate(self.base_url, self.headers)
-        self.connections = Connections(self.base_url, self.headers)
+
+        # we include api_key as well because the header is different for upload
+        self.connections = Connections(self.base_url, self.headers, self.api_key)
         self.tools = Tools(self.base_url, self.headers)
         self.pipelines = Pipelines(self.base_url, self.headers)

mixpeek-0.6.28/mixpeek/endpoints/connections.py

@@ -0,0 +1,69 @@
+import requests
+import os
+
+class Connections:
+    def __init__(self, base_url, headers, api_key):
+        self.base_url = base_url
+        self.headers = headers
+        self.api_key = api_key
+        self.data = self.Data(self)
+        self.storage = self.Storage(self)
+
+    def create(self, alias: str, engine: str, details: dict):
+        url = f"{self.base_url}connections/"
+        data = {
+            "alias": alias,
+            "engine": engine,
+            "details": details
+        }
+        response = requests.post(url, json=data, headers=self.headers)
+        return response.json()
+
+    class Data:
+        def __init__(self, parent):
+            self.base_url = parent.base_url
+            self.headers = parent.headers
+        
+        # mixpeek.connections.data.insert(connection_id="123", payload={"key": "value"})
+        def insert(self, connection_id: str, payload: list):
+            pass
+
+        # mixpeek.connections.data.delete(connection_id="123", filters={"key": "value"})
+        def delete(self, connection_id: str, filters: dict):
+            pass
+
+        # mixpeek.connections.data.upsert(connection_id="123", payload={"key": "value"}, filters={"key": "value"})
+        def upsert(self, connection_id: str, payload: dict, filters: dict):
+            pass
+
+        
+    class Storage:
+        def __init__(self, parent):
+            self.base_url = parent.base_url
+            self.headers = parent.headers
+            self.api_key = parent.api_key
+
+        def upload(self, connection_id: str, file_path: str):
+            url = f"{self.base_url}connections/storage?connection_id={connection_id}"
+
+            files=[
+                ('file',(os.path.basename(file_path),open(file_path,'rb'),'application/octet-stream'))
+            ]
+            headers = {
+                'Authorization': f'Bearer {self.api_key}'
+            }
+
+            response = requests.request("POST", url, headers=headers, files=files)
+
+            return response.json()
+
+
+        # mixpeek.connections.storage.delete(connection_id="123", file_name="example.txt")
+        def delete(self, connection_id: str, file_name: str):
+            url = f"{self.base_url}storage/{connection_id}/delete/{file_name}"
+            response = requests.delete(url, headers=self.headers)
+            return response.json()
+
+
+            
+        

mixpeek-0.6.28/mixpeek/endpoints/extract.py

@@ -0,0 +1,55 @@
+import requests
+
+class Extract:
+    def __init__(self, base_url, headers):
+        self.base_url = base_url
+        self.headers = headers
+
+    def automatic(self, input: str, input_type: str):
+        url = f"{self.base_url}extract/"
+        data = {
+            "input": input,
+            "input_type": input_type
+        }
+        response = requests.post(url, json=data, headers=self.headers)
+        return response.json()
+
+    def video(self, input: str, input_type: str):
+        url = f"{self.base_url}extract/"
+        data = {
+            "modality": "video",
+            "input": input,
+            "input_type": input_type
+        }
+        response = requests.post(url, json=data, headers=self.headers)
+        return response.json()
+    
+    def audio(self, input: str, input_type: str):
+        url = f"{self.base_url}extract/"
+        data = {
+            "modality": "audio",
+            "input": input,
+            "input_type": input_type
+        }
+        response = requests.post(url, json=data, headers=self.headers)
+        return response.json()
+
+    def image(self, input: str, input_type: str):
+        url = f"{self.base_url}extract/"
+        data = {
+            "modality": "image",
+            "input": input,
+            "input_type": input_type
+        }
+        response = requests.post(url, json=data, headers=self.headers)
+        return response.json()
+
+    def text(self, input: str, input_type: str):
+        url = f"{self.base_url}extract/"
+        data = {
+            "modality": "text",
+            "input": input,
+            "input_type": input_type
+        }
+        response = requests.post(url, json=data, headers=self.headers)
+        return response.json()

{mixpeek-0.6.26 → mixpeek-0.6.28}/mixpeek.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mixpeek
-Version: 0.6.26
+Version: 0.6.28
 Summary: Mixpeek Python SDK
 Home-page: https://github.com/mixpeek/mixpeek-python
 Author: Ethan Steininger
@@ -101,6 +101,15 @@ mixpeek.connections.data.delete(
 )
 ```
 
+Upload file
+
+```python
+mixpeek.connections.storage.upload(
+    connection_id="conn_321",
+    file_path="/my/local/file.mp4"
+)
+```
+
 ### Tools
 
 ```python

{mixpeek-0.6.26 → mixpeek-0.6.28}/setup.py

@@ -6,7 +6,7 @@ with open('requirements.txt') as f:
 
 setup(
     name='mixpeek',
-    version='0.6.26',
+    version='0.6.28',
     author='Ethan Steininger',
     author_email='ethan@mixpeek.com',
     description='Mixpeek Python SDK',

mixpeek-0.6.26/mixpeek/endpoints/connections.py

@@ -1,31 +0,0 @@
-import requests
-
-class Connections:
-    def __init__(self, base_url, headers):
-        self.base_url = base_url
-        self.headers = headers
-        self.data = self.Data(self)
-
-    def create(self, alias: str, engine: str, details: dict):
-        url = f"{self.base_url}connections/"
-        data = {
-            "alias": alias,
-            "engine": engine,
-            "details": details
-        }
-        response = requests.post(url, json=data, headers=self.headers)
-        return response.json()
-
-    # mixpeek.connections.data.insert(connection_id="123", payload={"key": "value"})
-    class Data:
-        def __init__(self, parent):
-            self.base_url = parent.base_url
-            self.headers = parent.headers
-        
-        def insert(self, connection_id: str, payload: list):
-            pass
-
-        def delete(self, connection_id: str, filters: dict):
-            pass
-            
-        

mixpeek-0.6.26/mixpeek/endpoints/extract.py

@@ -1,16 +0,0 @@
-import requests
-
-class Extract:
-    def __init__(self, base_url, headers):
-        self.base_url = base_url
-        self.headers = headers
-
-    def extract(self, input: str, input_type: str, modality: str = None):
-        url = f"{self.base_url}extract/"
-        data = {
-            "modality": modality,
-            "input": input,
-            "input_type": input_type
-        }
-        response = requests.post(url, json=data, headers=self.headers)
-        return response.json()