thinking-machine 0.1.5__tar.gz → 0.1.7__tar.gz

{thinking_machine-0.1.5/src/thinking_machine.egg-info → thinking_machine-0.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: thinking-machine
-Version: 0.1.5
+Version: 0.1.7
 Summary: A Machine that thinks.
 Author-email: Machina Ratiocinatrix <machina.ratio@gmail.com>, Alexander Fedotov <alex.fedotov@aol.com>
 Project-URL: Homepage, https://github.com/machina-ratiocinatrix/thinking-machine

{thinking_machine-0.1.5 → thinking_machine-0.1.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "thinking-machine"
-version = "0.1.5"
+version = "0.1.7"
 authors = [
     {name="Machina Ratiocinatrix", email="machina.ratio@gmail.com"},
     {name="Alexander Fedotov", email="alex.fedotov@aol.com"}

{thinking_machine-0.1.5 → thinking_machine-0.1.7}/src/thinking_machine/cli.py

@@ -103,6 +103,13 @@ def run():
         else:
             if config.provider == '':
                 raise ValueError(f"Unrecognized API key prefix and no provider specified.")
+            else:
+                if config.provider == 'Baseten':
+                    os.environ['BASETEN_API_KEY'] = args.provider_api_key
+                elif config.provider == 'Lightning':
+                    os.environ['LIGHTNING_API_KEY'] = args.provider_api_key
+                else:
+                    raise ValueError(f"Unsupported provider specified.")
                 
         config.provider_api_key = args.provider_api_key
         

{thinking_machine-0.1.5 → thinking_machine-0.1.7}/src/thinking_machine/machine.py

@@ -152,6 +152,46 @@ def machine(plato_text, config, **kwargs):
 
         return thoughts, text
 
+    elif provider == 'Baseten':
+        # Transform plato_text to CMJ format
+        messages = plato_text_to_cmj(plato_text=plato_text,
+                                     machine_name=name)
+        # Call OpenAI API via opehaina
+        environ['BASETEN_API_KEY'] = api_key
+        try:
+            from .providers import basta
+        except ImportError:
+            print("Baseten module is missing.", file=sys.stderr)
+            sys.exit(1)
+
+        thoughts, text = basta.respond(
+            messages=messages,
+            instructions=system_prompt,
+            **kwargs
+        )
+
+        return thoughts, text
+
+    elif provider == 'Lightning':
+        # Transform plato_text to CMJ format
+        messages = plato_text_to_cmj(plato_text=plato_text,
+                                     machine_name=name)
+        # Call OpenAI API via opehaina
+        environ['LIGHTNING_API_KEY'] = api_key
+        try:
+            from .providers import light
+        except ImportError:
+            print("Lightning module is missing.", file=sys.stderr)
+            sys.exit(1)
+
+        thoughts, text = light.respond(
+            messages=messages,
+            instructions=system_prompt,
+            **kwargs
+        )
+
+        return thoughts, text
+
     elif provider == 'Meta':
         # Transform plato_text to CMJ format
         messages = plato_text_to_cmj(plato_text=plato_text,

thinking_machine-0.1.7/src/thinking_machine/providers/basta.py

@@ -0,0 +1,82 @@
+# -*- coding: utf-8 -*-
+# Python
+
+"""Copyright (c) Alexander Fedotov.
+This source code is licensed under the license found in the
+LICENSE file in the root directory of this source tree.
+"""
+import urllib.request
+import urllib.error
+import json
+from os import environ
+
+
+def respond(messages, instructions, **kwargs):
+    """
+    """
+    api_key = environ.get("BASETEN_API_KEY")
+    api_base = environ.get("BASETEN_API_BASE", "https://inference.baseten.co/v1")
+    default_model = environ.get("BASETEN_DEFAULT_MODEL", "deepseek-ai/DeepSeek-V4-Pro")
+
+    instruction = kwargs.get('system_instruction', instructions)
+    first_message = [dict(role='system', content=instruction)] if instruction else []
+
+    # add contents and user text to the first (instruction) message
+    first_message.extend(messages)
+    instruction_and_contents = first_message
+
+    # Define the payload
+    payload = {
+        "model":            kwargs.get("model", default_model),
+        "messages":         instruction_and_contents,
+        "max_tokens":       kwargs.get("max_tokens", 32000),
+        "temperature":      kwargs.get("temperature", 1.0),
+        "reasoning_effort": kwargs.get("reasoning_effort", "max"),
+        "thinking": {
+            "type": "enabled"
+        }
+    }
+
+    # Convert data dictionary to JSON and encode it to bytes
+    data_bytes = json.dumps(payload).encode('utf-8')
+
+    # Set the mandatory headers
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Api-Key {api_key}",
+        "User-Agent": "Thinking-Machine"
+    }
+
+    # Create the Request object
+    req = urllib.request.Request(
+        f'{api_base}/chat/completions',
+        data=data_bytes,
+        headers=headers,
+        method="POST")
+
+    try:
+        # Execute the request
+        with urllib.request.urlopen(req, timeout=300) as response:
+            response_data = response.read().decode('utf-8')
+            output = json.loads(response_data)
+            message = output['choices'][0]['message']
+            text = message['content']
+            thoughts = message['reasoning_content']
+
+        return thoughts, text
+
+    except urllib.error.HTTPError as e:
+        # Handle HTTP errors (e.g., 401 Unauthorized, 400 Bad Request)
+        error_info = e.read().decode('utf-8', errors='ignore')
+        print(f"HTTP Error {e.code}: {e.reason}")
+        print(f"Error Details: {error_info}")
+        return '', ''
+
+    except urllib.error.URLError as e:
+        # Handle network/connection errors
+        print(f"Failed to reach the server: {e.reason}")
+        return '', ''
+
+
+if __name__ == '__main__':
+    ...

thinking_machine-0.1.7/src/thinking_machine/providers/light.py

@@ -0,0 +1,82 @@
+# -*- coding: utf-8 -*-
+# Python
+
+"""Copyright (c) Alexander Fedotov.
+This source code is licensed under the license found in the
+LICENSE file in the root directory of this source tree.
+"""
+import urllib.request
+import urllib.error
+import json
+from os import environ
+
+
+def respond(messages, instructions, **kwargs):
+    """
+    """
+    api_key = environ.get("LIGHNING_API_KEY")
+    api_base = environ.get("LIGHTNING_API_BASE", "https://lightning.ai/api/v1")
+    default_model = environ.get("LIGHTNING_DEFAULT_MODEL", "openai/gpt-5.5-2026-04-23")
+
+    instruction = kwargs.get('system_instruction', instructions)
+    first_message = [dict(role='system', content=instruction)] if instruction else []
+
+    # add contents and user text to the first (instruction) message
+    first_message.extend(messages)
+    instruction_and_contents = first_message
+
+    # Define the payload
+    payload = {
+        "model":            kwargs.get("model", default_model),
+        "messages":         instruction_and_contents,
+        "max_tokens":       kwargs.get("max_tokens", 32000),
+        "temperature":      kwargs.get("temperature", 1.0),
+        "reasoning_effort": kwargs.get("reasoning_effort", "high"),
+        "thinking": {
+            "type": "enabled"
+        }
+    }
+
+    # Convert data dictionary to JSON and encode it to bytes
+    data_bytes = json.dumps(payload).encode('utf-8')
+
+    # Set the mandatory headers
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {api_key}",
+        "User-Agent": "Thinking-Machine"
+    }
+
+    # Create the Request object
+    req = urllib.request.Request(
+        f'{api_base}/chat/completions',
+        data=data_bytes,
+        headers=headers,
+        method="POST")
+
+    try:
+        # Execute the request
+        with urllib.request.urlopen(req, timeout=300) as response:
+            response_data = response.read().decode('utf-8')
+            output = json.loads(response_data)
+            message = output['choices'][0]['message']
+            text = message['content']
+            thoughts = message['reasoning_content']
+
+        return thoughts, text
+
+    except urllib.error.HTTPError as e:
+        # Handle HTTP errors (e.g., 401 Unauthorized, 400 Bad Request)
+        error_info = e.read().decode('utf-8', errors='ignore')
+        print(f"HTTP Error {e.code}: {e.reason}")
+        print(f"Error Details: {error_info}")
+        return '', ''
+
+    except urllib.error.URLError as e:
+        # Handle network/connection errors
+        print(f"Failed to reach the server: {e.reason}")
+        return '', ''
+
+
+if __name__ == '__main__':
+    ...

{thinking_machine-0.1.5 → thinking_machine-0.1.7/src/thinking_machine.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: thinking-machine
-Version: 0.1.5
+Version: 0.1.7
 Summary: A Machine that thinks.
 Author-email: Machina Ratiocinatrix <machina.ratio@gmail.com>, Alexander Fedotov <alex.fedotov@aol.com>
 Project-URL: Homepage, https://github.com/machina-ratiocinatrix/thinking-machine

{thinking_machine-0.1.5 → thinking_machine-0.1.7}/src/thinking_machine.egg-info/SOURCES.txt

@@ -16,10 +16,12 @@ src/thinking_machine.egg-info/entry_points.txt
 src/thinking_machine.egg-info/requires.txt
 src/thinking_machine.egg-info/top_level.txt
 src/thinking_machine/providers/__init__.py
+src/thinking_machine/providers/basta.py
 src/thinking_machine/providers/camelids.py
 src/thinking_machine/providers/castor_pollux.py
 src/thinking_machine/providers/depsek.py
 src/thinking_machine/providers/electroid.py
+src/thinking_machine/providers/light.py
 src/thinking_machine/providers/openai.py
 src/thinking_machine/providers/qrog.py
 src/thinking_machine/providers/strangelove.py