mixpeek 0.6.26__py3-none-any.whl → 0.6.28__py3-none-any.whl

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/endpoints/connections.py

@@ -1,10 +1,13 @@
 import requests
+import os
 
 class Connections:
-    def __init__(self, base_url, headers):
+    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/"
@@ -16,16 +19,51 @@ class Connections:
         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
         
+        # 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/endpoints/extract.py

@@ -5,12 +5,51 @@ class Extract:
         self.base_url = base_url
         self.headers = headers
 
-    def extract(self, input: str, input_type: str, modality: str = None):
+    def automatic(self, input: str, input_type: str):
         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()
+
+    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.dist-info → mixpeek-0.6.28.dist-info}/METADATA

@@ -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.dist-info → mixpeek-0.6.28.dist-info}/RECORD

@@ -1,14 +1,14 @@
 mixpeek/__init__.py,sha256=XDdcK7wTEOEcF1cp-GeWmgPJ21Ny1R9pB0PPNrdDTMo,28
-mixpeek/client.py,sha256=aZ6zzSrqbV3RxD2G-W0Aqy06QWDfRux_0DfLr-awdG0,895
+mixpeek/client.py,sha256=ifN-rCXtCbitTj6coUquQjpHgd98V3EVwORzNo8tIMY,992
 mixpeek/exceptions.py,sha256=Orhdo5UFLn3fcWVJtlgkznW8Iy5ndL96h0qTY8zOlDA,235
 mixpeek/endpoints/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mixpeek/endpoints/connections.py,sha256=-SdwLhHY1KNnNT9u_V2hx0kiKjYrDXKLzceedZBqa14,917
+mixpeek/endpoints/connections.py,sha256=TLpaJBKvzCNpvb7ul8xfmkZP5rWyl4eTl6ec_5a2qiM,2344
 mixpeek/endpoints/embed.py,sha256=8ds_FinxZRW-ZQyv6LjDAX6Zoek2Cv4OYhwIgSBqwTs,1598
-mixpeek/endpoints/extract.py,sha256=7-d-8Mc7Uj20JyQRURoXeawpvUGvwMeBI8nwtjCYe8k,483
+mixpeek/endpoints/extract.py,sha256=PRY1ZjPPwJ3xEKNKVdWIm_WZeRNc68oJKcwVXNkv-S8,1706
 mixpeek/endpoints/generate.py,sha256=SFjVYfgeuIt4wO0I5ItnB4TEHhRkLgZOvQfWlEioye8,594
 mixpeek/endpoints/pipelines.py,sha256=X2mRsWgOx6FgWbInrAdjLQJbjXjorNqAo9mjFawLpoE,1210
 mixpeek/endpoints/tools.py,sha256=Ni9AYm2-jpDS1SLUohNLTOkXDUxLPnNLpTlhWJLfofs,791
-mixpeek-0.6.26.dist-info/METADATA,sha256=PKf1HAukMjpd1omrd6Rxxegqz9_-r-5XEZq93FUjRsE,1952
-mixpeek-0.6.26.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-mixpeek-0.6.26.dist-info/top_level.txt,sha256=EJ8Jc4IhqyUwnUlBwKbs498Ju4O9a-IDh2kXc_lo6Vg,8
-mixpeek-0.6.26.dist-info/RECORD,,
+mixpeek-0.6.28.dist-info/METADATA,sha256=IidDmjtBsjxat5OdnRqZp5wBxFN_n6a0ofD85JCtU2E,2083
+mixpeek-0.6.28.dist-info/WHEEL,sha256=cpQTJ5IWu9CdaPViMhC9YzF8gZuS5-vlfoFihTBC86A,91
+mixpeek-0.6.28.dist-info/top_level.txt,sha256=EJ8Jc4IhqyUwnUlBwKbs498Ju4O9a-IDh2kXc_lo6Vg,8
+mixpeek-0.6.28.dist-info/RECORD,,

{mixpeek-0.6.26.dist-info → mixpeek-0.6.28.dist-info}/WHEEL

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