mixpeek 0.1.2__py3-none-any.whl → 0.5__py3-none-any.whl

mixpeek/__init__.py

@@ -1,9 +1,86 @@
-"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+# mixpeek.py
+import requests
+import json
 
-from ._version import __title__, __version__
-from .sdk import *
-from .sdkconfiguration import *
-from .models import *
 
+class Mixpeek:
+    def __init__(self, api_key):
+        self.base_url = "http://localhost:8000"
+        self.headers = {
+            "Authorization": f"Bearer {api_key}",
+            "Content-Type": "application/json",
+        }
 
-VERSION: str = __version__
+        self.generate = self.Generate(self)
+        self.parse = self.Parse(self)
+        self.embed = self.Embed(self)
+
+    def _send_post(self, url, payload):
+        return requests.post(url, headers=self.headers, data=json.dumps(payload)).json()
+
+    def _send_get(self, url, payload=None):
+        return requests.get(url, headers=self.headers, data=json.dumps(payload)).json()
+
+    class Embed:
+        def __init__(self, mixpeek):
+            self.mixpeek = mixpeek
+            self.embed_url = f"{self.mixpeek.base_url}/embed"
+
+        def text(self, text, additional_data=None):
+            payload = {"input": text}
+            if additional_data is not None:
+                payload.update(additional_data)
+            return self.mixpeek._send_get(self.embed_url, payload)
+
+    class Generate:
+        def __init__(self, mixpeek):
+            self.mixpeek = mixpeek
+            self.openai = self.OpenAI(self.mixpeek)
+
+        class OpenAI:
+            def __init__(self, mixpeek):
+                self.mixpeek = mixpeek
+                self.generate_openai_url = f"{self.mixpeek.base_url}/generate"
+
+            def _construct_request_format(
+                self, model, context, response_format, settings=None
+            ):
+                payload = {
+                    "model": {"provider": "gpt", "model": model},
+                    "context": context,
+                    "messages": [{"role": "user", "content": ""}],
+                    **settings,
+                }
+
+                if response_format is not None:
+                    json_schema = response_format.model_json_schema()
+                    payload["response_format"] = json_schema
+
+                return payload
+
+            def chat(self, model, response_format, context, settings=None):
+                payload = self._construct_request_format(
+                    model, context, response_format, settings
+                )
+                return self.mixpeek._send_post(self.generate_openai_url, payload)
+
+    class Parse:
+        def __init__(self, mixpeek):
+            self.mixpeek = mixpeek
+            self.text = self.Text(self.mixpeek)
+
+        class Text:
+            def __init__(self, mixpeek):
+                self.mixpeek = mixpeek
+                self.parse_url = f"{self.mixpeek.base_url}/parsers"
+
+            def extract(self, file_url, should_chunk=True):
+                url = self.parse_url + "?should_chunk=" + str(should_chunk)
+                return self.mixpeek._send_post(url, {"file_url": file_url})
+
+            def chunk(self, corpus):
+                pass
+
+        class Index:
+            def __init__(self):
+                pass

mixpeek-0.5.dist-info/LICENSE.rst

@@ -0,0 +1,20 @@
+The MIT License (MIT)
+Copyright (c) 2021 mixpeek
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
+DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
+OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE
+OR OTHER DEALINGS IN THE SOFTWARE.

mixpeek-0.5.dist-info/METADATA

@@ -0,0 +1,12 @@
+Metadata-Version: 2.1
+Name: mixpeek
+Version: 0.5
+Summary: full-text file search API
+Home-page: https://github.com/mixpeek/mixpeek-python
+Author: Ethan Steininger
+Author-email: info@mixpeek.com
+License: MIT
+Keywords: file search
+License-File: LICENSE.rst
+Requires-Dist: requests
+

mixpeek-0.5.dist-info/RECORD

@@ -0,0 +1,6 @@
+mixpeek/__init__.py,sha256=SNRaL77AXYHrVoK5AaxvaFkehqAHjnj7LU9sNTWSiWM,2927
+mixpeek-0.5.dist-info/LICENSE.rst,sha256=HHFCinDHL38osjHNrBgNZqExYuEWJ3hNZxzcr1mWWeo,1072
+mixpeek-0.5.dist-info/METADATA,sha256=eepojmh-AImTMqu4cPebkRgoXnbqal21tP9AC4-bZlM,279
+mixpeek-0.5.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+mixpeek-0.5.dist-info/top_level.txt,sha256=EJ8Jc4IhqyUwnUlBwKbs498Ju4O9a-IDh2kXc_lo6Vg,8
+mixpeek-0.5.dist-info/RECORD,,

{mixpeek-0.1.2.dist-info → mixpeek-0.5.dist-info}/WHEEL

@@ -1,4 +1,5 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.1
+Generator: bdist_wheel (0.43.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
+

mixpeek-0.5.dist-info/top_level.txt

@@ -0,0 +1 @@
+mixpeek

mixpeek/_hooks/__init__.py

@@ -1,5 +0,0 @@
-"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
-
-from .sdkhooks import *
-from .types import *
-from .registration import *

mixpeek/_hooks/registration.py

@@ -1,13 +0,0 @@
-from .types import Hooks
-
-
-# This file is only ever generated once on the first generation and then is free to be modified.
-# Any hooks you wish to add should be registered in the init_hooks function. Feel free to define them
-# in this file or in separate files in the hooks folder.
-
-
-def init_hooks(hooks: Hooks):
-    # pylint: disable=unused-argument
-    """Add hooks by calling hooks.register{sdk_init/before_request/after_success/after_error}Hook
-    with an instance of a hook that implements that specific Hook interface
-    Hooks are registered per SDK instance, and are valid for the lifetime of the SDK instance"""

mixpeek/_hooks/sdkhooks.py

@@ -1,76 +0,0 @@
-"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
-
-import httpx
-from .types import (
-    SDKInitHook,
-    BeforeRequestContext,
-    BeforeRequestHook,
-    AfterSuccessContext,
-    AfterSuccessHook,
-    AfterErrorContext,
-    AfterErrorHook,
-    Hooks,
-)
-from .registration import init_hooks
-from typing import List, Optional, Tuple
-from mixpeek.httpclient import HttpClient
-
-
-class SDKHooks(Hooks):
-    def __init__(self) -> None:
-        self.sdk_init_hooks: List[SDKInitHook] = []
-        self.before_request_hooks: List[BeforeRequestHook] = []
-        self.after_success_hooks: List[AfterSuccessHook] = []
-        self.after_error_hooks: List[AfterErrorHook] = []
-        init_hooks(self)
-
-    def register_sdk_init_hook(self, hook: SDKInitHook) -> None:
-        self.sdk_init_hooks.append(hook)
-
-    def register_before_request_hook(self, hook: BeforeRequestHook) -> None:
-        self.before_request_hooks.append(hook)
-
-    def register_after_success_hook(self, hook: AfterSuccessHook) -> None:
-        self.after_success_hooks.append(hook)
-
-    def register_after_error_hook(self, hook: AfterErrorHook) -> None:
-        self.after_error_hooks.append(hook)
-
-    def sdk_init(self, base_url: str, client: HttpClient) -> Tuple[str, HttpClient]:
-        for hook in self.sdk_init_hooks:
-            base_url, client = hook.sdk_init(base_url, client)
-        return base_url, client
-
-    def before_request(
-        self, hook_ctx: BeforeRequestContext, request: httpx.Request
-    ) -> httpx.Request:
-        for hook in self.before_request_hooks:
-            out = hook.before_request(hook_ctx, request)
-            if isinstance(out, Exception):
-                raise out
-            request = out
-
-        return request
-
-    def after_success(
-        self, hook_ctx: AfterSuccessContext, response: httpx.Response
-    ) -> httpx.Response:
-        for hook in self.after_success_hooks:
-            out = hook.after_success(hook_ctx, response)
-            if isinstance(out, Exception):
-                raise out
-            response = out
-        return response
-
-    def after_error(
-        self,
-        hook_ctx: AfterErrorContext,
-        response: Optional[httpx.Response],
-        error: Optional[Exception],
-    ) -> Tuple[Optional[httpx.Response], Optional[Exception]]:
-        for hook in self.after_error_hooks:
-            result = hook.after_error(hook_ctx, response, error)
-            if isinstance(result, Exception):
-                raise result
-            response, error = result
-        return response, error

mixpeek/_hooks/types.py

@@ -1,94 +0,0 @@
-"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
-
-from abc import ABC, abstractmethod
-import httpx
-from mixpeek.httpclient import HttpClient
-from typing import Any, Callable, List, Optional, Tuple, Union
-
-
-class HookContext:
-    operation_id: str
-    oauth2_scopes: Optional[List[str]] = None
-    security_source: Optional[Union[Any, Callable[[], Any]]] = None
-
-    def __init__(
-        self,
-        operation_id: str,
-        oauth2_scopes: Optional[List[str]],
-        security_source: Optional[Union[Any, Callable[[], Any]]],
-    ):
-        self.operation_id = operation_id
-        self.oauth2_scopes = oauth2_scopes
-        self.security_source = security_source
-
-
-class BeforeRequestContext(HookContext):
-    def __init__(self, hook_ctx: HookContext):
-        super().__init__(
-            hook_ctx.operation_id, hook_ctx.oauth2_scopes, hook_ctx.security_source
-        )
-
-
-class AfterSuccessContext(HookContext):
-    def __init__(self, hook_ctx: HookContext):
-        super().__init__(
-            hook_ctx.operation_id, hook_ctx.oauth2_scopes, hook_ctx.security_source
-        )
-
-
-class AfterErrorContext(HookContext):
-    def __init__(self, hook_ctx: HookContext):
-        super().__init__(
-            hook_ctx.operation_id, hook_ctx.oauth2_scopes, hook_ctx.security_source
-        )
-
-
-class SDKInitHook(ABC):
-    @abstractmethod
-    def sdk_init(self, base_url: str, client: HttpClient) -> Tuple[str, HttpClient]:
-        pass
-
-
-class BeforeRequestHook(ABC):
-    @abstractmethod
-    def before_request(
-        self, hook_ctx: BeforeRequestContext, request: httpx.Request
-    ) -> Union[httpx.Request, Exception]:
-        pass
-
-
-class AfterSuccessHook(ABC):
-    @abstractmethod
-    def after_success(
-        self, hook_ctx: AfterSuccessContext, response: httpx.Response
-    ) -> Union[httpx.Response, Exception]:
-        pass
-
-
-class AfterErrorHook(ABC):
-    @abstractmethod
-    def after_error(
-        self,
-        hook_ctx: AfterErrorContext,
-        response: Optional[httpx.Response],
-        error: Optional[Exception],
-    ) -> Union[Tuple[Optional[httpx.Response], Optional[Exception]], Exception]:
-        pass
-
-
-class Hooks(ABC):
-    @abstractmethod
-    def register_sdk_init_hook(self, hook: SDKInitHook):
-        pass
-
-    @abstractmethod
-    def register_before_request_hook(self, hook: BeforeRequestHook):
-        pass
-
-    @abstractmethod
-    def register_after_success_hook(self, hook: AfterSuccessHook):
-        pass
-
-    @abstractmethod
-    def register_after_error_hook(self, hook: AfterErrorHook):
-        pass

mixpeek/_version.py

@@ -1,12 +0,0 @@
-"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
-
-import importlib.metadata
-
-__title__: str = "mixpeek"
-__version__: str = "0.1.2"
-
-try:
-    if __package__ is not None:
-        __version__ = importlib.metadata.version(__package__)
-except importlib.metadata.PackageNotFoundError:
-    pass